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"`                   // 温度参数 (0.0-1.0)
	MaxTokens                int                                                      `json:"maxTokens,omitempty"`           // 最大token数
	MaxCompletionTokens      int                                                      `json:"maxCompletionTokens,omitempty"` // 最大完成token数
	TopP                     float32                                                  `json:"topP"`                          // Top P采样
	FrequencyPenalty         float32                                                  `json:"frequencyPenalty"`              // 频率惩罚
	PresencePenalty          float32                                                  `json:"presencePenalty"`               // 存在惩罚
	StopSequences            []string                                                 `json:"stopSequences"`                 // 停止序列
	Timeout                  time.Duration                                            `json:"timeout"`                       // 超时时间
	ToolExecutionTimeout     time.Duration                                            `json:"toolExecutionTimeout"`          // 工具执行超时时间
	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"`       // 记忆压缩阈值（消息数量）
	MemoryCompressRatio      float32                                                  `json:"memoryCompressRatio"`           // 记忆压缩比例（0.0-1.0）
	LogSilent                bool                                                     `json:"logSilent"`                     // 是否静默日志
	LogFile                  string                                                   `json:"logFile"`                       // 日志文件路径，为空则输出到终端
	DoomLoopThreshold        int                                                      `json:"doomLoopThreshold"`             // 同一 tool+同一参数连续出现次数阈值，0 表示关闭
	OnDoomLoop               func(toolName string, input map[string]interface{}) bool `json:"-"`                             // 检测到 doom loop 时回调，返回 false 则终止迭代
	MaxToolCallsPerIteration int                                                      `json:"maxToolCallsPerIteration"`      // 单轮最大 tool 调用数，0 表示不限制
	ToolResultWriteDir       string                                                   `json:"toolResultWriteDir"`            // 工具大结果落盘目录，为空则仅内存截断
	DefaultToolResultMaxLen  int                                                      `json:"defaultToolResultMaxLen"`       // 工具结果默认截断长度，0 表示用 types.MaxTruncationLength
	ToolErrorMaxLen          int                                                      `json:"toolErrorMaxLen"`               // 工具错误信息截断长度，0 表示用 types.ToolErrorMaxLen
}

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 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"
	Content   string     `json:"content,omitempty"`
	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
}

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 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"`
}