Files
ai-agent/internal/ai/runtime/graphs/triage_service_request_graph.go
T
mlogclub 5d7c10aeab refactor: rename agent widget references to AI agent for consistency
- Updated runtime configuration to use __CS_AI_AGENT_WIDGET_CONFIG__ instead of __CS_AGENT_WIDGET_CONFIG__.
- Changed message types in support host bridge from "cs-agent" to "cs-ai-agent".
- Minified SDK script updated to reflect new AI agent naming conventions.
- Adjusted scrollbar styles in main.scss to use .cs-ai-agent-scrollbar instead of .cs-agent-scrollbar.
2026-05-30 21:19:36 +08:00

81 lines
2.7 KiB
Go

package graphs
import (
"context"
"encoding/json"
"fmt"
"strings"
"cs-ai-agent/internal/models"
"cs-ai-agent/internal/services"
)
type TriageServiceRequestInput struct {
Goal string `json:"goal"`
ObservedIssue string `json:"observedIssue"`
NeedTicket bool `json:"needTicket"`
NeedHumanHandoff bool `json:"needHumanHandoff"`
AdditionalContext string `json:"additionalContext"`
}
type TriageServiceRequestResult struct {
Analysis AnalyzeConversationResult `json:"analysis"`
TicketDraft *PrepareTicketDraftResult `json:"ticketDraft,omitempty"`
RecommendedAction string `json:"recommendedAction"`
Ready bool `json:"ready"`
}
type TriageServiceRequestGraph struct {
conversation models.Conversation
}
func NewTriageServiceRequestGraph(conversation models.Conversation) *TriageServiceRequestGraph {
return &TriageServiceRequestGraph{conversation: conversation}
}
func (g *TriageServiceRequestGraph) Run(_ context.Context, argumentsInJSON string) (string, error) {
input, err := g.parseInput(argumentsInJSON)
if err != nil {
return "", err
}
messages, _, _ := services.MessageService.FindByConversationIDCursor(g.conversation.ID, 0, 8, "", "")
analysis := buildAnalyzeConversationResult(g.conversation, messages, AnalyzeConversationInput{
Goal: input.Goal,
ObservedIssue: input.ObservedIssue,
NeedTicket: input.NeedTicket,
NeedHumanHandoff: input.NeedHumanHandoff,
AdditionalContext: input.AdditionalContext,
})
result := TriageServiceRequestResult{
Analysis: analysis,
RecommendedAction: analysis.RecommendedNextAction,
Ready: analysis.RecommendedNextAction == "continue_answering" || analysis.RecommendedNextAction == "handoff_to_human",
}
if analysis.RecommendedNextAction == "prepare_ticket" {
draft := buildPrepareTicketDraftResult(g.conversation, messages, PrepareTicketDraftInput{
Issue: input.ObservedIssue,
})
result.TicketDraft = &draft
result.Ready = draft.Ready
}
buf, err := json.Marshal(result)
if err != nil {
return "", err
}
return string(buf), nil
}
func (g *TriageServiceRequestGraph) parseInput(argumentsInJSON string) (TriageServiceRequestInput, error) {
var input TriageServiceRequestInput
if strings.TrimSpace(argumentsInJSON) == "" {
return input, nil
}
if err := json.Unmarshal([]byte(argumentsInJSON), &input); err != nil {
return input, fmt.Errorf("invalid triage service request arguments: %w", err)
}
input.Goal = strings.TrimSpace(input.Goal)
input.ObservedIssue = strings.TrimSpace(input.ObservedIssue)
input.AdditionalContext = strings.TrimSpace(input.AdditionalContext)
return input, nil
}