types

type ActionMetadata struct {
	ItemIndex int `json:"itemIndex"`
}

type ActionResponse struct {
	Action *ToolAction `json:"action"`
	Data   interface{} `json:"data"`
	Error  string      `json:"error,omitempty"`
}

type AgentConfig struct {
	MaxIterations            int                                                               `json:"maxIterations"`
	SystemMessage            string                                                            `json:"systemMessage"`
	ChatMessageRole          string                                                            `json:"chatMessageRole,omitempty"`
	Temperature              float32                                                           `json:"temperature"`
	MaxTokens                int                                                               `json:"maxTokens,omitempty"`
	MaxCompletionTokens      int                                                               `json:"maxCompletionTokens,omitempty"`
	TopP                     float32                                                           `json:"topP"`
	FrequencyPenalty         float32                                                           `json:"frequencyPenalty"`
	PresencePenalty          float32                                                           `json:"presencePenalty"`
	StopSequences            []string                                                          `json:"stopSequences"`
	Timeout                  time.Duration                                                     `json:"timeout"`
	ToolExecutionTimeout     time.Duration                                                     `json:"toolExecutionTimeout"`
	ToolTimeouts             map[string]time.Duration                                          `json:"toolTimeouts"`
	RetryAttempts            int                                                               `json:"retryAttempts"`
	RetryDelay               time.Duration                                                     `json:"retryDelay"`
	EnableToolRetry          bool                                                              `json:"enableToolRetry"`
	MaxHistoryMessages       int                                                               `json:"maxHistoryMessages"`
	EnableMemoryCompress     bool                                                              `json:"enableMemoryCompress"`
	MemoryCompressThreshold  int                                                               `json:"memoryCompressThreshold"` // message count incl. each assistant + tool row from tool rounds
	MemoryCompressRatio      float32                                                           `json:"memoryCompressRatio"`
	LogSilent                bool                                                              `json:"logSilent"`
	LogFile                  string                                                            `json:"logFile"`
	DoomLoopThreshold        int                                                               `json:"doomLoopThreshold"`
	OnDoomLoop               func(toolName string, input map[string]interface{}) bool          `json:"-"`
	ToolTimeoutCalculator    func(toolName string, input map[string]interface{}) time.Duration `json:"-"`
	MaxToolCallsPerIteration int                                                               `json:"maxToolCallsPerIteration"`
	ToolResultWriteDir       string                                                            `json:"toolResultWriteDir"`
	DefaultToolResultMaxLen  int                                                               `json:"defaultToolResultMaxLen"`
	ToolErrorMaxLen          int                                                               `json:"toolErrorMaxLen"`
	ReasoningEffort          string                                                            `json:"reasoningEffort"`
}

type AgentInput struct {
	Text  string        `json:"text,omitempty"`
	Parts []MessagePart `json:"parts,omitempty"`
}

type AgentResult struct {
	Output            string            `json:"output"`
	ToolCalls         []ToolCallRequest `json:"tool_calls"`
	IntermediateSteps []ToolCallData    `json:"intermediate_steps"`
	Usage             Usage             `json:"usage"`
}

type EngineRequest struct {
	Actions  []ToolAction            `json:"actions"`
	Metadata RequestResponseMetadata `json:"metadata"`
}

type EngineResponse struct {
	ActionResponses []ActionResponse        `json:"actionResponses"`
	Metadata        RequestResponseMetadata `json:"metadata"`
}

type ImageDataPart struct {
	Data     []byte `json:"data"`
	MIMEType string `json:"mime_type"`
}

type ImageURLPart struct {
	URL    string `json:"url"`
	Detail string `json:"detail,omitempty"` // "low", "high", "auto"
}

type LLMProvider interface {
	// Basic chat functionality
	Chat(ctx context.Context, messages []Message) (Message, error)
	ChatStream(ctx context.Context, messages []Message) (<-chan StreamMessage, error)

	// Tool call support
	ChatWithTools(ctx context.Context, messages []Message, tools []Tool) (Message, error)
	ChatWithToolsStream(ctx context.Context, messages []Message, tools []Tool) (<-chan StreamMessage, error)

	// Model information
	GetModelName() string
	GetModelMetadata() ModelMetadata
}

type LangChainToolWrapper struct {
	// contains filtered or unexported fields
}

type MemoryProvider interface {
	// Load memory variables
	LoadMemoryVariables(ctx context.Context) (map[string]interface{}, error)

	// Save context
	SaveContext(ctx context.Context, input, output map[string]interface{}) error

	// Clear memory
	Clear(ctx context.Context) error

	// Get chat history
	GetChatHistory(ctx context.Context) ([]Message, error)

	// Compress memory (optional, for memory compression)
	CompressMemory(ctx context.Context, llm LLMProvider, maxMessages int) error
}

type Message struct {
	Role       string        `json:"role"` // "system", "user", "assistant", "tool"
	Content    string        `json:"content"`
	Name       string        `json:"name,omitempty"`
	ToolCalls  []ToolCall    `json:"tool_calls,omitempty"`
	ToolCallID string        `json:"tool_call_id,omitempty"`
	Parts      []MessagePart `json:"parts,omitempty"` // Multi-modal content support
	Usage      Usage         `json:"usage,omitempty"` // Token usage information
}

type MessagePart interface {
	// contains filtered or unexported methods
}

type ModelMetadata struct {
	Name      string                 `json:"name"`
	Version   string                 `json:"version"`
	MaxTokens int                    `json:"maxTokens"`
	Tools     []Tool                 `json:"tools,omitempty"`
	Extra     map[string]interface{} `json:"extra,omitempty"`
}

type OutputParser interface {
	Parse(output string) (interface{}, error)
	GetFormatInstructions() string
}

type ReasoningMessage struct {
	Content   string `json:"content"`
	Reasoning string `json:"reasoning,omitempty"`
}

type RequestResponseMetadata struct {
	ItemIndex        int            `json:"itemIndex,omitempty"`
	PreviousRequests []ToolCallData `json:"previousRequests,omitempty"`
	IterationCount   int            `json:"iterationCount,omitempty"`
}

type StreamEvent struct {
	Type       string      `json:"type"`
	Content    string      `json:"content,omitempty"`
	ToolResult interface{} `json:"toolResult,omitempty"`
	EventName  string      `json:"eventName,omitempty"`
	Data       interface{} `json:"data,omitempty"`
}

type StreamMessage struct {
	Type      string     `json:"type"` // "chunk", "end", "error", "tool_calls", "reasoning"
	Content   string     `json:"content,omitempty"`
	Reasoning string     `json:"reasoning,omitempty"` // Thinking/reasoning content
	Error     string     `json:"error,omitempty"`
	ToolCalls []ToolCall `json:"tool_calls,omitempty"`
	Usage     *Usage     `json:"usage,omitempty"`
}

type StreamResult struct {
	Type    string
	Content string
	Result  *AgentResult
	Error   error
	// Tool event fields
	ToolEvent *ToolEvent
}

type TextPart struct {
	Text string `json:"text"`
}

type Tool interface {
	// Tool basic information
	Name() string
	Description() string
	Schema() map[string]interface{}

	// Tool execution
	Execute(ctx context.Context, input map[string]interface{}) (interface{}, error)

	// Tool metadata
	Metadata() ToolMetadata
}

type ToolAction struct {
	NodeName string                 `json:"nodeName"`
	Input    map[string]interface{} `json:"input"`
	Type     string                 `json:"type"`
	ID       string                 `json:"id"`
	Metadata ActionMetadata         `json:"metadata"`
}

type ToolActionStep struct {
	Tool       string      `json:"tool"`
	ToolInput  interface{} `json:"toolInput"`
	Log        interface{} `json:"log"`
	ToolCallID interface{} `json:"toolCallId"`
	Type       interface{} `json:"type"`
}

type ToolCacheEntry struct {
	Result    interface{}
	Err       error
	Timestamp time.Time
	Prev      *ToolCacheEntry
	Next      *ToolCacheEntry
	Key       string
}

type ToolCall struct {
	ID       string       `json:"id"`
	Type     string       `json:"type"`
	Function ToolFunction `json:"function"`
}

type ToolCallData struct {
	Action      ToolActionStep `json:"action"`
	Observation string         `json:"observation"`
}

type ToolCallRequest struct {
	Tool       string                 `json:"tool"`
	ToolInput  map[string]interface{} `json:"toolInput"`
	ToolCallID string                 `json:"toolCallId"`
	Type       string                 `json:"type,omitempty"`
	Log        string                 `json:"log,omitempty"`
	MessageLog []interface{}          `json:"messageLog,omitempty"`
}

type ToolCallback interface {
	OnToolCall(toolName string, toolCallID string, input map[string]interface{})
	OnToolInputStart(toolName string, toolCallID string, input map[string]interface{})
	OnToolInputEnd(toolName string, toolCallID string, input map[string]interface{})
	OnToolResult(toolName string, toolCallID string, output interface{})
	OnToolError(toolName string, toolCallID string, err error)
}

type ToolCallbackFunc struct {
	// contains filtered or unexported fields
}

type ToolEvent struct {
	Event      string                 // Event type: tool_call, tool_input_start, tool_input_end, tool_result, tool_error
	ToolName   string                 // Tool name
	ToolCallID string                 // Tool call ID
	State      string                 // State: pending, running, completed, error
	Input      map[string]interface{} // Tool input arguments
	Output     interface{}            // Tool execution result
	Error      string                 // Error message if failed
	Duration   time.Duration          // Execution duration
}

type ToolFunction struct {
	Name      string                 `json:"name"`
	Arguments map[string]interface{} `json:"arguments"`
}

type ToolMetadata struct {
	SourceNodeName      string                 `json:"sourceNodeName"`
	IsFromToolkit       bool                   `json:"isFromToolkit"`
	ToolType            string                 `json:"toolType"`                      // "mcp","http","builtin"
	Priority            int                    `json:"priority,omitempty"`            // 优先级，数字越大优先级越高
	Dependencies        []string               `json:"dependencies,omitempty"`        // 依赖的工具名称列表
	MaxTruncationLength int                    `json:"maxTruncationLength,omitempty"` // 工具结果截断长度，0表示使用默认值
	Extra               map[string]interface{} `json:"extra,omitempty"`
}

type Usage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
	ReasoningTokens  int `json:"reasoning_tokens,omitempty"` // Thinking tokens (o1, o3 series etc.)
	CachedTokens     int `json:"cached_tokens,omitempty"`    // Cache read tokens
}