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"`
	MaxBudgetTokens          int                                                               `json:"maxBudgetTokens"`
	RemainPromptTokens       func() int                                                        `json:"-"`
	EnableMemoryCompress     bool                                                              `json:"enableMemoryCompress"`
	MemoryCompressThreshold  int                                                               `json:"memoryCompressThreshold"` // message count incl. each assistant + tool row from tool rounds
	CompactAfterTurns        int                                                               `json:"compactAfterTurns"`       // completed Execute/ExecuteStream saves before compress; 0=off
	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"`
	// PromptCaching enables provider prompt caching (Anthropic cache_control
	// breakpoints) and cache usage backfill. Default on: it is a pure cost
	// optimization with no correctness impact. Set false for providers/proxies
	// that choke on cache_control (R9 escape hatch).
	PromptCaching        bool `json:"promptCaching,omitempty"`
	ToolParallelismLimit int  `json:"toolParallelismLimit,omitempty"` // 0=默认 max(4, GOMAXPROCS*2) 封顶 32；>0 用该值
	StreamBufferSize     int  `json:"streamBufferSize,omitempty"`     // 0=默认 50；>0 用该值
	// ChunkMergeFlushInterval batches consecutive stream "chunk" results and
	// flushes them as one merged chunk. 0=default 50ms. Disable merging by
	// setting a value <0 (each fragment is sent immediately). Merging never
	// affects end/error/tool_event delivery — those bypass the buffer.
	ChunkMergeFlushInterval time.Duration `json:"chunkMergeFlushInterval,omitempty"` // 0=默认 50ms；<0 禁用合并；>0 用该值
	// CompactionPrefix enables prefix-preserving compaction: trimHistoryToTokenBudget
	// keeps a head cache anchor, replaces the middle with a tail summary, and
	// preserves the recent tail verbatim, so prompt-cache breakpoints 1-2
	// (system+tools) survive compaction (P3.1, prompt caching Step 4).
	// Default off: with it off, trimHistoryToTokenBudget keeps today's behavior
	// and the GetSummary head injection is unchanged (byte-identical).
	CompactionPrefix bool `json:"compactionPrefix,omitempty"`
	// CacheAnchorTokens caps the head-cache-anchor budget (tokens): the head of
	// history, counted from index 0, is kept verbatim up to this many tokens so
	// it participates in the cache prefix. 0 = no anchor (the head is trimmed
	// like today, the middle is still replaced by the summary). Default 0.
	CacheAnchorTokens int `json:"cacheAnchorTokens,omitempty"`
}

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"`
	StopCause         AgentStopCause    `json:"stop_cause,omitempty"`
}

type AgentStopCause string

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

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

type FatalToolError struct {
	Err    error
	Reason string
}

type FatalToolErrorKind interface {
	error
	FatalToolErrorKind()
}

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

	// GetChatHistory returns history sized for LLM context: implementations MUST window (e.g. match GetMessages(limit<=0): maxHistoryMessages cap + optional summary). Returning the full unbounded store will blow prompts; the engine does not apply a second message-count cut. Optional: implement StoredMessageCount(ctx) for compress gating (see agent/engine).
	GetChatHistory(ctx context.Context) ([]Message, error)

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

type MemoryReplay interface {
	ReplayMessages(ctx context.Context, messages []Message) 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 OutputHeader struct {
	ChunkID          string        `json:"chunk_id,omitempty"`
	WallTime         time.Duration `json:"wall_time,omitempty"`       // execution duration
	ExitCode         *int          `json:"exit_code,omitempty"`       // bash etc. when available
	OriginalBytes    int           `json:"original_bytes,omitempty"`  // raw output bytes before truncation
	OriginalTokens   int           `json:"original_tokens,omitempty"` // approximate, via RoughTokenEstimate
	TotalLines       int           `json:"total_lines,omitempty"`     // line count of the raw output
	SavedPath        string        `json:"saved_path,omitempty"`      // when output was written to disk instead
	OmittedCharCount int           `json:"omitted_chars,omitempty"`   // chars elided by TruncateMiddle
}

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

type PromptCacheConfigurer interface {
	SetPromptCacheOptions(PromptCacheOptions)
	PromptCacheOptions() PromptCacheOptions
}

type PromptCacheOptions struct {
	Enabled          bool // master switch; when false the request body is byte-identical to pre-caching
	SystemBreakpoint bool // 1 breakpoint on the system block
	ToolsBreakpoint  bool // 1 breakpoint on the LAST tool only
	HistoryEveryN    int  // history breakpoint budget (0 = no history breakpoints)
	MinCacheTokens   int  // history segment below this (estimated tokens) gets no history breakpoint
}

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
	// StopCause classifies Error (e.g. context_window) for harness consumers.
	StopCause AgentStopCause
}

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 ToolExposure string

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表示使用默认值
	Exposure            ToolExposure           `json:"exposure,omitempty"`            // E1：模型可见性；空值=direct（向后兼容）
	SearchKeywords      []string               `json:"searchKeywords,omitempty"`      // E2：tool_search 索引关键词；空则用 Name+Description 分词
	Extra               map[string]interface{} `json:"extra,omitempty"`
}

type TruncationMeta struct {
	Truncated     bool
	SavedFilePath string
	OriginalBytes int
}

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 (0.1x price)
	CacheCreationTokens int `json:"cache_creation_tokens,omitempty"` // Cache write tokens (1.25x price)
}