t
18c9354095
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力，补充业务读写工具与客户快捷操作契约。 - 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现，将身份权限与业务主体交由宿主管理。 - 使用 libSQL 重构向量存储，并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。 - 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议，补齐模块初始化、业务动作和公共载荷等回归测试。
78 lines
3.0 KiB
Go
78 lines
3.0 KiB
Go
// Package readtools executes deterministic, read-only graph tools through the
|
|
// shared Tool Registry boundary.
|
|
package readtools
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"code.tczkiot.com/wlw/ai-agent/internal/ai/runtime/graphs"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/ai/runtime/retrievers"
|
|
aitooling "code.tczkiot.com/wlw/ai-agent/internal/ai/tooling"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/models"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/pkg/toolx"
|
|
)
|
|
|
|
func ExecuteGraphTool(ctx context.Context, conversation models.Conversation, toolCode string, arguments map[string]any, policy aitooling.Policy) (aitooling.Definition, string, error) {
|
|
toolCode = toolx.NormalizeToolCodeAlias(strings.TrimSpace(toolCode))
|
|
if toolCode != toolx.GraphTriageServiceRequest.Code && toolCode != toolx.GraphAnalyzeConversation.Code {
|
|
return aitooling.Definition{}, "", fmt.Errorf("tool is not a graph read tool")
|
|
}
|
|
definition, err := aitooling.DefaultRegistry.Resolve(toolCode)
|
|
if err != nil {
|
|
return aitooling.Definition{}, "", err
|
|
}
|
|
if err := aitooling.DefaultPolicyGuard.Authorize(aitooling.Invocation{
|
|
Definition: definition,
|
|
Arguments: arguments,
|
|
Policy: policy,
|
|
}); err != nil {
|
|
return definition, "", err
|
|
}
|
|
if definition.TimeoutMS > 0 {
|
|
var cancel context.CancelFunc
|
|
ctx, cancel = context.WithTimeout(ctx, time.Duration(definition.TimeoutMS)*time.Millisecond)
|
|
defer cancel()
|
|
}
|
|
data, err := json.Marshal(arguments)
|
|
if err != nil {
|
|
return definition, "", err
|
|
}
|
|
switch toolCode {
|
|
case toolx.GraphTriageServiceRequest.Code:
|
|
result, err := graphs.NewTriageServiceRequestGraph(conversation).Run(ctx, string(data))
|
|
return definition, result, err
|
|
case toolx.GraphAnalyzeConversation.Code:
|
|
result, err := graphs.NewAnalyzeConversationGraph(conversation).Run(ctx, string(data))
|
|
return definition, result, err
|
|
}
|
|
return definition, "", fmt.Errorf("tool is not a graph read tool")
|
|
}
|
|
|
|
// RetrieveKnowledge executes the built-in knowledge tool after the same
|
|
// registry policy and timeout checks used by graph tools.
|
|
func RetrieveKnowledge(ctx context.Context, agent models.AIAgent, knowledgeBaseIDs []int64, query string, policy aitooling.Policy) (aitooling.Definition, *retrievers.KnowledgeRetrieveResult, error) {
|
|
definition, err := aitooling.DefaultRegistry.Resolve(toolx.BuiltinKnowledgeRetrieve.Code)
|
|
if err != nil {
|
|
return aitooling.Definition{}, nil, err
|
|
}
|
|
arguments := map[string]any{"query": strings.TrimSpace(query), "knowledge_base_ids": knowledgeBaseIDs}
|
|
if err := aitooling.DefaultPolicyGuard.Authorize(aitooling.Invocation{
|
|
Definition: definition,
|
|
Arguments: arguments,
|
|
Policy: policy,
|
|
}); err != nil {
|
|
return definition, nil, err
|
|
}
|
|
if definition.TimeoutMS > 0 {
|
|
var cancel context.CancelFunc
|
|
ctx, cancel = context.WithTimeout(ctx, time.Duration(definition.TimeoutMS)*time.Millisecond)
|
|
defer cancel()
|
|
}
|
|
result, err := retrievers.NewKnowledgeRetriever(agent, knowledgeBaseIDs).RetrieveContext(ctx, strings.TrimSpace(query))
|
|
return definition, result, err
|
|
}
|