package callbacks

type ToolTraceItem struct {
	ToolCode      string         `json:"toolCode"`
	ServerCode    string         `json:"serverCode"`
	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"`
}

type ToolSearchTraceItem struct {
	Action             string   `json:"action,omitempty"`
	Query              string   `json:"query,omitempty"`
	TargetToolCode     string   `json:"targetToolCode,omitempty"`
	TargetServerCode   string   `json:"targetServerCode,omitempty"`
	TargetToolName     string   `json:"targetToolName,omitempty"`
	CandidateToolCodes []string `json:"candidateToolCodes,omitempty"`
	Status             string   `json:"status,omitempty"`
	ErrorMessage       string   `json:"errorMessage,omitempty"`
}

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"`
	RecommendedAction string         `json:"recommendedAction,omitempty"`
	RiskLevel         string         `json:"riskLevel,omitempty"`
	TicketDraftReady  bool           `json:"ticketDraftReady,omitempty"`
}

type RetrieverTraceItem struct {
	Query           string  `json:"query,omitempty"`
	KnowledgeBaseID int64   `json:"knowledgeBaseId,omitempty"`
	DocumentID      int64   `json:"documentId,omitempty"`
	DocumentTitle   string  `json:"documentTitle,omitempty"`
	Score           float64 `json:"score,omitempty"`
	LatencyMs       int64   `json:"latencyMs,omitempty"`
}

type RetrieverTraceSummary struct {
	TopK             int
	ScoreThreshold   float64
	ContextMaxTokens int
	MaxContextItems  int
	HitCount         int
	ContextCount     int
	EmbeddingMs      int64
	VectorSearchMs   int64
	HydrateMs        int64
	Policies         []RetrieverPolicyTraceItem
}

type RetrieverPolicyTraceItem struct {
	KnowledgeBaseID int64   `json:"knowledgeBaseId,omitempty"`
	TopK            int     `json:"topK,omitempty"`
	ScoreThreshold  float64 `json:"scoreThreshold,omitempty"`
}

type InstructionTraceSummary struct {
	SectionTitles     []string
	HasGovernanceRule bool
	HasAgentRule      bool
	HasSkillRule      bool
	HasToolRule       bool
}

type RuntimeTraceData struct {
	Version   string         `json:"version"`
	Status    string         `json:"status"`
	RunID     string         `json:"runId,omitempty"`
	Skill     SkillTraceData `json:"skill,omitempty"`
	Interrupt struct {
		CheckPointID string                  `json:"checkPointId,omitempty"`
		Items        []InterruptTraceContext `json:"items,omitempty"`
	} `json:"interrupt"`
	Model struct {
		Provider string `json:"provider,omitempty"`
		Name     string `json:"name,omitempty"`
	} `json:"model"`
	Instruction struct {
		SectionTitles     []string `json:"sectionTitles,omitempty"`
		HasGovernanceRule bool     `json:"hasGovernanceRule,omitempty"`
		HasAgentRule      bool     `json:"hasAgentRule,omitempty"`
		HasSkillRule      bool     `json:"hasSkillRule,omitempty"`
		HasToolRule       bool     `json:"hasToolRule,omitempty"`
	} `json:"instruction"`
	Input struct {
		HistoryMessageCount       int      `json:"historyMessageCount,omitempty"`
		KnowledgeBaseIDs          []int64  `json:"knowledgeBaseIds,omitempty"`
		ToolCodes                 []string `json:"toolCodes,omitempty"`
		StaticToolCodes           []string `json:"staticToolCodes,omitempty"`
		DynamicToolCodes          []string `json:"dynamicToolCodes,omitempty"`
		ToolSearchEnabled         bool     `json:"toolSearchEnabled,omitempty"`
		CurrentUserMessagePreview string   `json:"currentUserMessagePreview,omitempty"`
	} `json:"input"`
	Retriever struct {
		Count            int                        `json:"count,omitempty"`
		TopK             int                        `json:"topK,omitempty"`
		ScoreThreshold   float64                    `json:"scoreThreshold,omitempty"`
		ContextMaxTokens int                        `json:"contextMaxTokens,omitempty"`
		MaxContextItems  int                        `json:"maxContextItems,omitempty"`
		ContextCount     int                        `json:"contextCount,omitempty"`
		EmbeddingMs      int64                      `json:"embeddingMs,omitempty"`
		VectorSearchMs   int64                      `json:"vectorSearchMs,omitempty"`
		HydrateMs        int64                      `json:"hydrateMs,omitempty"`
		Policies         []RetrieverPolicyTraceItem `json:"policies,omitempty"`
		Items            []RetrieverTraceItem       `json:"items,omitempty"`
	} `json:"retriever"`
	Tools struct {
		Count int             `json:"count,omitempty"`
		Items []ToolTraceItem `json:"items,omitempty"`
	} `json:"tools"`
	ToolSearch struct {
		Count int                   `json:"count,omitempty"`
		Items []ToolSearchTraceItem `json:"items,omitempty"`
	} `json:"toolSearch"`
	GraphTools struct {
		Count int                  `json:"count,omitempty"`
		Items []GraphToolTraceItem `json:"items,omitempty"`
	} `json:"graphTools"`
	Output struct {
		ReplyText    string `json:"replyText,omitempty"`
		FinishReason string `json:"finishReason,omitempty"`
	} `json:"output"`
	Error struct {
		Message string `json:"message,omitempty"`
		Stage   string `json:"stage,omitempty"`
	} `json:"error"`
}

type SkillTraceData struct {
	Code               string   `json:"code,omitempty"`
	Name               string   `json:"name,omitempty"`
	RouteReason        string   `json:"routeReason,omitempty"`
	RouteTrace         string   `json:"routeTrace,omitempty"`
	AllowedToolCodes   []string `json:"allowedToolCodes,omitempty"`
	MiddlewareEnabled  bool     `json:"middlewareEnabled,omitempty"`
	MiddlewareToolName string   `json:"middlewareToolName,omitempty"`
}

type InterruptTraceContext struct {
	Type        string `json:"type,omitempty"`
	ID          string `json:"id"`
	InfoPreview string `json:"infoPreview,omitempty"`
}