wlw/ai-agent
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: refactor tool definitions to use structured ToolSpec and update related references

Browse Source
This commit is contained in:
mlogclub
2026-04-12 22:27:09 +08:00
parent 54f5a684c3
commit 11252d31da
16 changed files with 280 additions and 204 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/ai/runtime/registry/registry.go
+9 -9
View File
@@ -59,24 +59,24 @@ func isAllowedToolCode(toolCode string, allowedToolCodes map[string]struct{}) bo
if isAlwaysAllowedToolCode(toolCode) {
return true
}
if strings.TrimSpace(toolCode) == toolx.GraphTriageServiceRequestToolCode {
if _, ok := allowedToolCodes[toolx.GraphCreateTicketConfirmToolCode]; ok {
if strings.TrimSpace(toolCode) == toolx.GraphTriageServiceRequest.Code {
if _, ok := allowedToolCodes[toolx.GraphCreateTicketConfirm.Code]; ok {
return true
}
if _, ok := allowedToolCodes[toolx.GraphHandoffConversationToolCode]; ok {
if _, ok := allowedToolCodes[toolx.GraphHandoffConversation.Code]; ok {
return true
}
}
if strings.TrimSpace(toolCode) == toolx.GraphAnalyzeConversationToolCode {
if _, ok := allowedToolCodes[toolx.GraphCreateTicketConfirmToolCode]; ok {
if strings.TrimSpace(toolCode) == toolx.GraphAnalyzeConversation.Code {
if _, ok := allowedToolCodes[toolx.GraphCreateTicketConfirm.Code]; ok {
return true
}
if _, ok := allowedToolCodes[toolx.GraphHandoffConversationToolCode]; ok {
if _, ok := allowedToolCodes[toolx.GraphHandoffConversation.Code]; ok {
return true
}
}
if strings.TrimSpace(toolCode) == toolx.GraphPrepareTicketDraftToolCode {
_, ok := allowedToolCodes[toolx.GraphCreateTicketConfirmToolCode]
if strings.TrimSpace(toolCode) == toolx.GraphPrepareTicketDraft.Code {
_, ok := allowedToolCodes[toolx.GraphCreateTicketConfirm.Code]
return ok
}
return false
@@ -98,5 +98,5 @@ func makeAllowedToolCodeSet(input []string) map[string]struct{} {
}
func isAlwaysAllowedToolCode(toolCode string) bool {
return strings.TrimSpace(toolCode) == toolx.GraphHandoffConversationToolCode
return strings.TrimSpace(toolCode) == toolx.GraphHandoffConversation.Code
}