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