feat: enhance graph tool handling by adding recommended action, risk level, and ticket draft readiness

2026-04-12 13:43:09 +08:00
parent 89227c02c4
commit 37a713f3b8
3 changed files with 101 additions and 24 deletions
@@ -65,17 +65,21 @@ func (h *RuntimeTraceHandler) WrapInvokableToolCall(_ context.Context, endpoint
 		}
 		h.collector.AddToolItem(item)
 		if metadata, ok := h.resolveToolMetadata(item.ToolName); ok && strings.TrimSpace(metadata.SourceType) == toolx.GraphToolCatalogServerCode {
+			recommendedAction, riskLevel, ticketDraftReady := parseGraphToolOutcome(item.ToolCode, result)
 			h.collector.AddGraphToolItem(GraphToolTraceItem{
-				ToolCode:      item.ToolCode,
-				ToolName:      item.ToolName,
-				Arguments:     item.Arguments,
-				ResultPreview: item.ResultPreview,
-				ResultReduced: item.ResultReduced,
-				OriginalChars: item.OriginalChars,
-				KeptChars:     item.KeptChars,
-				LatencyMs:     item.LatencyMs,
-				Status:        item.Status,
-				ErrorMessage:  item.ErrorMessage,
+				ToolCode:          item.ToolCode,
+				ToolName:          item.ToolName,
+				Arguments:         item.Arguments,
+				ResultPreview:     item.ResultPreview,
+				ResultReduced:     item.ResultReduced,
+				OriginalChars:     item.OriginalChars,
+				KeptChars:         item.KeptChars,
+				LatencyMs:         item.LatencyMs,
+				Status:            item.Status,
+				ErrorMessage:      item.ErrorMessage,
+				RecommendedAction: recommendedAction,
+				RiskLevel:         riskLevel,
+				TicketDraftReady:  ticketDraftReady,
 			})
 		}
 		if metadata, ok := h.resolveToolMetadata(item.ToolName); ok && strings.TrimSpace(metadata.ToolCode) == toolx.BuiltinToolSearchToolCode {
@@ -85,6 +89,33 @@ func (h *RuntimeTraceHandler) WrapInvokableToolCall(_ context.Context, endpoint
 	}, nil
 }

+func parseGraphToolOutcome(toolCode string, result string) (recommendedAction, riskLevel string, ticketDraftReady bool) {
+	toolCode = strings.TrimSpace(toolCode)
+	if toolCode == "" || strings.TrimSpace(result) == "" {
+		return "", "", false
+	}
+	payload := make(map[string]any)
+	if err := json.Unmarshal([]byte(strings.TrimSpace(result)), &payload); err != nil {
+		return "", "", false
+	}
+	switch toolCode {
+	case toolx.GraphAnalyzeConversationToolCode:
+		return strings.TrimSpace(readToolSearchString(payload, "recommendedNextAction")), strings.TrimSpace(readToolSearchString(payload, "riskLevel")), false
+	case toolx.GraphTriageServiceRequestToolCode:
+		recommendedAction = strings.TrimSpace(readToolSearchString(payload, "recommendedAction"))
+		if analysis, ok := payload["analysis"].(map[string]any); ok {
+			riskLevel = strings.TrimSpace(readToolSearchString(analysis, "riskLevel"))
+		}
+		if ticketDraft, ok := payload["ticketDraft"].(map[string]any); ok {
+			ready, _ := ticketDraft["ready"].(bool)
+			ticketDraftReady = ready
+		}
+		return recommendedAction, riskLevel, ticketDraftReady
+	default:
+		return "", "", false
+	}
+}
+
 func (h *RuntimeTraceHandler) resolveToolMetadata(modelToolName string) (ToolMetadata, bool) {
 	if h == nil || h.toolMetadataBy == nil {
 		return ToolMetadata{}, false
@@ -26,16 +26,19 @@ type ToolSearchTraceItem struct {
 }

 type GraphToolTraceItem struct {
-	ToolCode      string         `json:"toolCode"`
-	ToolName      string         `json:"toolName"`
-	Arguments     map[string]any `json:"arguments,omitempty"`
-	ResultPreview string         `json:"resultPreview,omitempty"`
-	ResultReduced bool           `json:"resultReduced,omitempty"`
-	OriginalChars int            `json:"originalChars,omitempty"`
-	KeptChars     int            `json:"keptChars,omitempty"`
-	LatencyMs     int64          `json:"latencyMs,omitempty"`
-	Status        string         `json:"status,omitempty"`
-	ErrorMessage  string         `json:"errorMessage,omitempty"`
+	ToolCode          string         `json:"toolCode"`
+	ToolName          string         `json:"toolName"`
+	Arguments         map[string]any `json:"arguments,omitempty"`
+	ResultPreview     string         `json:"resultPreview,omitempty"`
+	ResultReduced     bool           `json:"resultReduced,omitempty"`
+	OriginalChars     int            `json:"originalChars,omitempty"`
+	KeptChars         int            `json:"keptChars,omitempty"`
+	LatencyMs         int64          `json:"latencyMs,omitempty"`
+	Status            string         `json:"status,omitempty"`
+	ErrorMessage      string         `json:"errorMessage,omitempty"`
+	RecommendedAction string         `json:"recommendedAction,omitempty"`
+	RiskLevel         string         `json:"riskLevel,omitempty"`
+	TicketDraftReady  bool           `json:"ticketDraftReady,omitempty"`
 }

 type RetrieverTraceItem struct {