api

type Agent struct {
	Name         string
	Code         bool
	SystemPrompt string
	Toolbox      []ToolSpec
}

type ChatCompletionProvider interface {
	// CreateChatCompletion generates a chat completion response to the
	// provided messages.
	CreateChatCompletion(
		ctx context.Context,
		params RequestParameters,
		messages []Message,
	) (*Message, error)

	// CreateChatCompletionStream streams the response via the output callback.
	CreateChatCompletionStream(
		ctx context.Context,
		params RequestParameters,
		messages []Message,
		output func(Chunk),
	) (*Message, error)
}

type Chunk struct {
	Kind       ChunkKind
	Text       string
	TokenCount uint
}

type ChunkKind int

type ContentPart struct {
	Type     ContentPartType `json:"type"`
	Text     string          `json:"text,omitempty"`     // text and thinking parts
	MIMEType string          `json:"mimeType,omitempty"` // image parts (e.g. "image/png")
	Data     []byte          `json:"data,omitempty"`     // raw bytes - base64 encoded when serialized
}

type ContentPartType string

type Event struct {
	Type    EventType
	Chunk   Chunk    // EventChunk
	Message *Message // EventResponseComplete
	Error   error    // EventError
}

type EventType int

type Items struct {
	Type string `json:"type"` // e.g. "string"
}

type Message struct {
	Parts       []ContentPart
	Role        MessageRole
	ToolCalls   []ToolCall
	ToolResults []ToolResult
	Metadata    MessageMeta
}

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

type MessageMeta struct {
	GenerationProvider *string         `json:"generation_provider,omitempty"`
	GenerationModel    *string         `json:"generation_model,omitempty"`
	TimeToFirstToken   *time.Duration  `json:"time_to_first_token,omitempty"`
	TotalDuration      *time.Duration  `json:"total_duration,omitempty"`
	ThinkingSignature  *string         `json:"thinking_signature,omitempty"`
	ReasoningDetails   json.RawMessage `json:"reasoning_details,omitempty"` // preserved reasoning_detail blocks (OpenRouter)
}

type MessageRole string

type Model struct {
	Name        string
	Provider    *Provider
	MaxTokens   *int
	Temperature *float32
}

type Provider struct {
	Name    string
	Display string
	Client  ChatCompletionProvider
	Models  []Model
}

type ReasoningEffort string

type RequestParameters struct {
	Model       *Model
	Agent       *Agent
	Effort      ReasoningEffort
	Temperature float32
	MaxTokens   int
}

type ToolCall struct {
	ID         string         `json:"id"`
	Name       string         `json:"name"`
	Parameters map[string]any `json:"parameters"`
}

type ToolConfirmation struct {
	Approved bool
	Reason   string
}

type ToolParameter struct {
	Name        string   `json:"name"`
	Type        string   `json:"type"` // "string", "integer", "boolean", "array"
	Required    bool     `json:"required"`
	Description string   `json:"description"`
	Enum        []string `json:"enum,omitempty"`
	Items       *Items   `json:"items,omitempty"` // for array types
}

type ToolPermission string

type ToolResult struct {
	ToolCallID string `json:"toolCallID"`
	ToolName   string `json:"toolName,omitempty"`
	Result     string `json:"result,omitempty"`
}

type ToolSpec struct {
	Name        string
	Description string
	Permission  ToolPermission
	Parameters  []ToolParameter
	Impl        func(*ToolSpec, map[string]any) (string, error)
}