msg

type AfterToolCallFn func(ctx AfterToolContext) (*AfterToolResult, error)

type AfterToolContext struct {
	Assistant AssistantMessage
	ToolCall  ToolCallPart
	Args      map[string]any
	Result    ToolResultMessage
	Context   *Context
}

type AfterToolResult struct {
	Parts     []ContentPart
	IsError   *bool
	Terminate bool
}

type AgentMessage interface {
	Role() MessageRole
}

type AssistantMessage struct {
	Parts      []ContentPart
	Model      string
	StopReason string
	Usage      *Usage
	Timestamp  time.Time
	// TurnDurationMs is elapsed milliseconds for the full agent turn (LLM + tools).
	TurnDurationMs int64
	// ThinkingDurationMs is elapsed milliseconds for the reasoning stream phase.
	ThinkingDurationMs int64
	// ThinkingText is accumulated reasoning output for UI replay after refresh.
	ThinkingText string
	// RunDurationMs is total run milliseconds; set on the final assistant message of a run.
	RunDurationMs int64
}

type BeforeToolCallFn func(ctx BeforeToolContext) (*BeforeToolResult, error)

type BeforeToolContext struct {
	Assistant AssistantMessage
	ToolCall  ToolCallPart
	Args      map[string]any
	Context   *Context
}

type BeforeToolResult struct {
	Block  bool
	Reason string
}

type BranchSummaryMessage struct {
	Summary   string
	FromID    string
	Timestamp time.Time
}

type CompactionSummaryMessage struct {
	Summary      string
	TokensBefore int
	Timestamp    time.Time
}

type Config struct {
	MaxSteps      int
	ToolExecution ToolExecutionMode
	ModelName     string
	ChatModel     string
	ToolModel     string
	Temperature   float64
	MaxTokens     int
	// ThinkingLevel controls reasoning intensity for supported models (default/off/low/medium/high).
	ThinkingLevel       string
	TransformContext    TransformContextFn
	ConvertToLLM        ConvertToLLMFn
	PrepareNextTurn     PrepareNextTurnFn
	ShouldStopAfterTurn ShouldStopAfterTurnFn
	BeforeToolCall      BeforeToolCallFn
	AfterToolCall       AfterToolCallFn
	GetSteeringMessages GetMessagesFn
	GetFollowUpMessages GetMessagesFn
	SteeringMode        QueueMode
	FollowUpMode        QueueMode
	// LLMRetryMaxAttempts overrides default LLM retries when > 0.
	LLMRetryMaxAttempts int
	// LLMRetryInitialInterval overrides the first retry delay when > 0.
	LLMRetryInitialInterval time.Duration
	// LLMRetryMaxInterval caps the delay between retries when > 0.
	LLMRetryMaxInterval time.Duration
	// LLMRetryMultiplier controls delay growth when > 0.
	LLMRetryMultiplier float64
}

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

type Context struct {
	SystemPrompt string
	Messages     []AgentMessage
	ModelName    string
}

type ConvertToLLMFn func(messages []AgentMessage) ([]llms.MessageContent, error)

type CustomMessage struct {
	CustomType         string
	Parts              []ContentPart
	DisplayOnly        bool
	ExcludeFromContext bool
	Timestamp          time.Time
}

type GetMessagesFn func() ([]AgentMessage, error)

type ImagePart struct {
	MIMEType string
	Data     []byte
	URL      string
}

type MessageRole string

type PrepareNextTurnFn func(ctx TurnContext) (*TurnUpdate, error)

type QueueMode string

type ShouldStopAfterTurnFn func(ctx TurnContext) (bool, error)

type StopReason string

type TextPart struct {
	Text string
}

type ToolCallPart struct {
	ID        string
	Name      string
	Arguments string
}

type ToolExecutionMode string

type ToolResultMessage struct {
	ToolCallID string
	Name       string
	Parts      []ContentPart
	IsError    bool
	Timestamp  time.Time
	// DurationMs is elapsed milliseconds for tool execution.
	DurationMs int64
}

type TransformContextFn func(messages []AgentMessage) ([]AgentMessage, error)

type TurnContext struct {
	Message     AssistantMessage
	ToolResults []ToolResultMessage
	Context     *Context
	NewMessages []AgentMessage
}

type TurnUpdate struct {
	Context   *Context
	ModelName string
}

type Usage struct {
	PromptTokens     int
	CompletionTokens int
	TotalTokens      int
	CacheRead        int
	CacheWrite       int
}

type UserMessage struct {
	Parts     []ContentPart
	Timestamp time.Time
}