feat: add support for Graph Tools in AIAgent model and related services

2026-04-14 15:10:50 +08:00
parent 8d560bb6bd
commit aa7f94e08f
9 changed files with 213 additions and 19 deletions
@@ -281,6 +281,12 @@ func IsAgentDirectToolCode(toolCode string) bool {
 	return ok && spec.DirectAccess
 }

+func IsAgentDirectGraphToolCode(toolCode string) bool {
+	toolCode = NormalizeToolCodeAlias(strings.TrimSpace(toolCode))
+	spec, ok := GetRegisteredToolSpec(toolCode)
+	return ok && spec.DirectAccess && spec.SourceType == enums.ToolSourceTypeGraph
+}
+
 func NormalizeToolCodeAlias(toolCode string) string {
 	toolCode = strings.TrimSpace(toolCode)
 	if canonical, ok := toolAliasToCanonical[toolCode]; ok {