mirror of
https://github.com/Ed1s0nZ/CyberStrikeAI.git
synced 2026-08-28 13:50:25 +02:00
102 lines
2.6 KiB
Go
102 lines
2.6 KiB
Go
package einomcp
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"cyberstrike-ai/internal/agent"
|
|
|
|
"github.com/cloudwego/eino/components/tool"
|
|
"github.com/cloudwego/eino/schema"
|
|
"github.com/eino-contrib/jsonschema"
|
|
)
|
|
|
|
// ExecutionRecorder 可选，在 MCP 工具成功返回且带有 execution id 时回调（用于汇总 mcpExecutionIds）。
|
|
type ExecutionRecorder func(executionID string)
|
|
|
|
// ToolsFromDefinitions 将单 Agent 使用的 OpenAI 风格工具定义转为 Eino InvokableTool，执行时走 Agent 的 MCP 路径。
|
|
func ToolsFromDefinitions(ag *agent.Agent, holder *ConversationHolder, defs []agent.Tool, rec ExecutionRecorder) ([]tool.BaseTool, error) {
|
|
out := make([]tool.BaseTool, 0, len(defs))
|
|
for _, d := range defs {
|
|
if d.Type != "function" || d.Function.Name == "" {
|
|
continue
|
|
}
|
|
info, err := toolInfoFromDefinition(d)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("tool %q: %w", d.Function.Name, err)
|
|
}
|
|
out = append(out, &mcpBridgeTool{
|
|
info: info,
|
|
name: d.Function.Name,
|
|
agent: ag,
|
|
holder: holder,
|
|
record: rec,
|
|
})
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
func toolInfoFromDefinition(d agent.Tool) (*schema.ToolInfo, error) {
|
|
fn := d.Function
|
|
raw, err := json.Marshal(fn.Parameters)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var js jsonschema.Schema
|
|
if len(raw) > 0 && string(raw) != "null" && string(raw) != "{}" {
|
|
if err := json.Unmarshal(raw, &js); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
if js.Type == "" {
|
|
js.Type = string(schema.Object)
|
|
}
|
|
if js.Properties == nil && js.Type == string(schema.Object) {
|
|
// 空参数对象
|
|
}
|
|
return &schema.ToolInfo{
|
|
Name: fn.Name,
|
|
Desc: fn.Description,
|
|
ParamsOneOf: schema.NewParamsOneOfByJSONSchema(&js),
|
|
}, nil
|
|
}
|
|
|
|
type mcpBridgeTool struct {
|
|
info *schema.ToolInfo
|
|
name string
|
|
agent *agent.Agent
|
|
holder *ConversationHolder
|
|
record ExecutionRecorder
|
|
}
|
|
|
|
func (m *mcpBridgeTool) Info(ctx context.Context) (*schema.ToolInfo, error) {
|
|
_ = ctx
|
|
return m.info, nil
|
|
}
|
|
|
|
func (m *mcpBridgeTool) InvokableRun(ctx context.Context, argumentsInJSON string, opts ...tool.Option) (string, error) {
|
|
_ = opts
|
|
var args map[string]interface{}
|
|
if argumentsInJSON != "" && argumentsInJSON != "null" {
|
|
if err := json.Unmarshal([]byte(argumentsInJSON), &args); err != nil {
|
|
return "", fmt.Errorf("invalid tool arguments JSON: %w", err)
|
|
}
|
|
}
|
|
if args == nil {
|
|
args = map[string]interface{}{}
|
|
}
|
|
conv := m.holder.Get()
|
|
res, err := m.agent.ExecuteMCPToolForConversation(ctx, conv, m.name, args)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if res == nil {
|
|
return "", nil
|
|
}
|
|
if res.ExecutionID != "" && m.record != nil {
|
|
m.record(res.ExecutionID)
|
|
}
|
|
return res.Result, nil
|
|
}
|