agent

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

type CompletionReason string

type IterationCallback func(iteration int, result *IterationResult)

type IterationResult struct {
	Iteration   int
	Response    *provider.ChatResponse
	ToolCalls   []provider.ToolCall
	ToolResults []ToolResult
	Usage       provider.Usage
	// Messages contains all messages generated during this iteration
	// (assistant messages with tool_calls + tool result messages)
	Messages []provider.Message
}

type LoopResult struct {
	FinalText        string
	Iterations       int
	CompletionReason CompletionReason
	Reason           string
	AllResults       []*IterationResult
	TotalUsage       provider.Usage
}

type Settings struct {
	MaxToolSteps     int
	StopConditions   []StopCondition
	VerifyCompletion VerifyCompletion
	OnIterationStart IterationCallback
	OnIterationEnd   IterationCallback
	OnToolStart      ToolStartCallback
	OnToolEnd        ToolEndCallback
}

type StopCondition func(ctx StopConditionContext) bool

type StopConditionContext struct {
	Iteration  int
	TotalUsage provider.Usage
	LastResult *IterationResult
	AllResults []*IterationResult
}

type StreamEvent struct {
	Type       string `json:"type"` // "token", "tool_start", "tool_end", "error", "done"
	Content    string `json:"content,omitempty"`
	ToolName   string `json:"tool_name,omitempty"`
	ToolArgs   string `json:"tool_args,omitempty"`
	ToolResult string `json:"tool_result,omitempty"`
	Error      string `json:"error,omitempty"`
	SessionID  string `json:"session_id,omitempty"`
}

type TokenCallback func(token string)

type ToolEndCallback func(toolName string, args string, result *ToolResult)

type ToolResult struct {
	ToolCallID string
	ToolName   string
	Result     string
	Error      error
}

type ToolStartCallback func(toolName string, args string)

type VerificationContext struct {
	LastResult     *IterationResult
	Iteration      int
	AllResults     []*IterationResult
	OriginalPrompt string
	TotalUsage     provider.Usage
}

type VerificationResult struct {
	Complete bool
	Reason   string
}

type VerifyCompletion func(ctx VerificationContext) VerificationResult