Files
ai-agent/internal/ai/runtime/tools/analyze_conversation_tool.go
T
t 18c9354095 refactor: 将客服后端重构为宿主可嵌入模块
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力,补充业务读写工具与客户快捷操作契约。

- 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现,将身份权限与业务主体交由宿主管理。

- 使用 libSQL 重构向量存储,并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。

- 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议,补齐模块初始化、业务动作和公共载荷等回归测试。
2026-08-28 22:23:13 +08:00

104 lines
3.3 KiB
Go

package tools
import (
"context"
"code.tczkiot.com/wlw/ai-agent/internal/ai/runtime/graphs"
"code.tczkiot.com/wlw/ai-agent/internal/ai/runtime/registry"
"code.tczkiot.com/wlw/ai-agent/internal/models"
"code.tczkiot.com/wlw/ai-agent/internal/pkg/i18nx"
"code.tczkiot.com/wlw/ai-agent/internal/pkg/toolx"
einotool "github.com/cloudwego/eino/components/tool"
"github.com/cloudwego/eino/schema"
einojsonschema "github.com/eino-contrib/jsonschema"
orderedmap "github.com/wk8/go-ordered-map/v2"
)
type AnalyzeConversationTool struct {
conversation models.Conversation
}
func NewAnalyzeConversationTool() *AnalyzeConversationTool {
return &AnalyzeConversationTool{}
}
func (t *AnalyzeConversationTool) Spec() toolx.ToolSpec {
return toolx.GraphAnalyzeConversation
}
func (t *AnalyzeConversationTool) Name() string {
return toolx.GraphAnalyzeConversation.Name
}
func (t *AnalyzeConversationTool) Code() string {
return toolx.GraphAnalyzeConversation.Code
}
func (t *AnalyzeConversationTool) Enabled(ctx registry.Context) bool {
return true
}
func (t *AnalyzeConversationTool) Build(ctx registry.Context) (einotool.BaseTool, error) {
if !t.Enabled(ctx) {
return nil, nil
}
return &AnalyzeConversationTool{conversation: ctx.Conversation}, nil
}
func (t *AnalyzeConversationTool) Info(ctx context.Context) (*schema.ToolInfo, error) {
return &schema.ToolInfo{
Name: toolx.GraphAnalyzeConversation.Name,
Desc: i18nx.Get("tool.graph.analyzeConversation.info"),
ParamsOneOf: schema.NewParamsOneOfByJSONSchema(&einojsonschema.Schema{
Version: einojsonschema.Version,
Type: "object",
Properties: orderedmap.New[string, *einojsonschema.Schema](orderedmap.WithInitialData(
orderedmap.Pair[string, *einojsonschema.Schema]{
Key: "goal",
Value: &einojsonschema.Schema{
Type: "string",
Description: i18nx.Get("tool.graph.analyzeConversation.param.goal"),
},
},
orderedmap.Pair[string, *einojsonschema.Schema]{
Key: "observed_issue",
Value: &einojsonschema.Schema{
Type: "string",
Description: i18nx.Get("tool.graph.analyzeConversation.param.observedIssue"),
},
},
orderedmap.Pair[string, *einojsonschema.Schema]{
Key: "need_human_handoff",
Value: &einojsonschema.Schema{
Type: "boolean",
Description: i18nx.Get("tool.graph.triageServiceRequest.param.needHumanHandoff"),
},
},
orderedmap.Pair[string, *einojsonschema.Schema]{
Key: "need_quality_check",
Value: &einojsonschema.Schema{
Type: "boolean",
Description: i18nx.Get("tool.graph.analyzeConversation.param.needQualityCheck"),
},
},
orderedmap.Pair[string, *einojsonschema.Schema]{
Key: "additional_context",
Value: &einojsonschema.Schema{
Type: "string",
Description: i18nx.Get("tool.graph.analyzeConversation.param.additionalContext"),
},
},
)),
}),
Extra: map[string]any{
"tool_code": toolx.GraphAnalyzeConversation.Code,
"source_type": toolx.GraphAnalyzeConversation.SourceType,
},
}, nil
}
func (t *AnalyzeConversationTool) InvokableRun(ctx context.Context, argumentsInJSON string, opts ...einotool.Option) (string, error) {
return graphs.NewAnalyzeConversationGraph(t.conversation).Run(ctx, argumentsInJSON)
}