api

type BaseProvider struct {
	ProviderName string
	APIKey       string
	APIURL       string
	HTTPClient   *http.Client
}

type ChatRequest struct {
	Model           string         `json:"model"`
	Messages        []Message      `json:"messages"`
	Stream          bool           `json:"stream"`
	StreamOptions   *StreamOptions `json:"stream_options,omitempty"`   // ストリーミング時のオプション
	ReasoningEffort string         `json:"reasoning_effort,omitempty"` // OpenAI Extended Thinking用
	Tools           []OpenAITool   `json:"tools,omitempty"`            // Function Calling用
	ToolChoice      string         `json:"tool_choice,omitempty"`      // "auto", "none", "required"
}

type ChatResponse struct {
	Choices []Choice `json:"choices"`
}

type Choice struct {
	Message Message `json:"message"`
}

type Client struct {
	Provider Provider
	Timeout  time.Duration
}

type CompactCapable interface {
	// CompactHistory は会話履歴を圧縮する
	CompactHistory(ctx context.Context, input []InputItem, model, instructions string) (*CompactResponse, error)

	// SupportsCompact は Compact API 対応を返す
	SupportsCompact() bool
}

type CompactResponse struct {
	Output []InputItem   `json:"output"` // 圧縮済みアイテム（次の /responses に使用）
	Model  string        `json:"model"`
	Usage  *CompactUsage `json:"usage,omitempty"`
}

type CompactUsage struct {
	InputTokens  int `json:"input_tokens"`
	OutputTokens int `json:"output_tokens"`
	TotalTokens  int `json:"total_tokens"`
}

type Delta struct {
	Content   string           `json:"content,omitempty"`
	ToolCalls []OpenAIToolCall `json:"tool_calls,omitempty"` // Function Calling用
}

type GeminiFunctionCall struct {
	Name string         `json:"name"`
	Args map[string]any `json:"args"`
}

type GeminiFunctionDeclaration struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description"`
	Parameters  *GeminiParameterSchema `json:"parameters,omitempty"`
}

type GeminiParameterSchema struct {
	Type       string                       `json:"type"`
	Properties map[string]GeminiPropertyDef `json:"properties,omitempty"`
	Required   []string                     `json:"required,omitempty"`
}

type GeminiPropertyDef struct {
	Type        string             `json:"type"`
	Description string             `json:"description,omitempty"`
	Enum        []string           `json:"enum,omitempty"`
	Items       *GeminiPropertyDef `json:"items,omitempty"` // array型用
}

type GeminiToolConfig struct {
	FunctionDeclarations []GeminiFunctionDeclaration `json:"function_declarations"`
}

type ImageData struct {
	Path      string // 元のファイルパス
	MediaType string // "image/png", "image/jpeg" 等
	Base64    string // Base64エンコードされた画像データ
	Size      int64  // ファイルサイズ（バイト）
}

type InputContentPart struct {
	Type     string `json:"type"`                // "input_text" or "input_image"
	Text     string `json:"text,omitempty"`      // type="input_text"の場合
	ImageURL string `json:"image_url,omitempty"` // type="input_image"の場合（data:image/...形式）
}

type InputItem struct {
	Type    string      `json:"type"`              // "message", "compacted", "function_call", "function_call_output"
	Role    string      `json:"role,omitempty"`    // "user", "assistant"
	Content interface{} `json:"content,omitempty"` // string or []InputContentPart

	// アシスタント応答の完全情報（Compact API用）
	ID     string `json:"id,omitempty"`     // "msg_xxx" (アシスタント応答のID)
	Status string `json:"status,omitempty"` // "completed"

	// 圧縮済みアイテム用
	Data string `json:"data,omitempty"` // 暗号化データ（type="compacted"の場合）

	// Function Calling 用（type="function_call"の場合）
	CallID    string `json:"call_id,omitempty"`   // ツール呼び出しID
	Name      string `json:"name,omitempty"`      // ツール名
	Arguments string `json:"arguments,omitempty"` // 引数（JSON文字列）

	// Function Calling 結果用（type="function_call_output"の場合）
	Output string `json:"output,omitempty"` // ツール実行結果
}

type MCPToolProvider interface {
	// SetMCPEnabled はMCPが有効かどうかを設定する（レガシー、互換性のため）
	SetMCPEnabled(enabled bool)

	// SetMCPTools はMCPツールの定義を設定する
	SetMCPTools(tools []GeminiFunctionDeclaration)
}

type Message struct {
	Role       string           `json:"role"`
	Content    string           `json:"content"`
	ToolCallID string           `json:"tool_call_id,omitempty"` // Function Calling: ツール結果用
	ToolCalls  []OpenAIToolCall `json:"tool_calls,omitempty"`   // Function Calling: assistant の tool_calls
}

type ModelLister interface {
	// ListModels はインストール済み/利用可能なモデルの一覧を取得
	ListModels() ([]string, error)
}

type MultimodalMessage struct {
	Role    string     `json:"role"`
	Content string     `json:"content"`
	Image   *ImageData `json:"-"` // JSON保存しない（一時的なもの）
}

type OpenAIMCPToolProvider interface {
	// SetMCPTools はMCPツールの定義を設定する
	SetMCPTools(tools []OpenAIToolFunction)
}

type OpenAITool struct {
	Type     string              `json:"type"` // "function"
	Function *OpenAIToolFunction `json:"function"`
}

type OpenAIToolCall struct {
	Index    int                    `json:"index"`    // ストリーミング用インデックス
	ID       string                 `json:"id"`       // ツール呼び出しID
	Type     string                 `json:"type"`     // "function"
	Function OpenAIToolCallFunction `json:"function"` // 関数呼び出し情報
}

type OpenAIToolCallFunction struct {
	Name      string `json:"name"`      // ツール名
	Arguments string `json:"arguments"` // 引数（JSON文字列）
}

type OpenAIToolFunction struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description,omitempty"`
	Parameters  map[string]interface{} `json:"parameters,omitempty"`
	Strict      bool                   `json:"strict,omitempty"`
}

type Provider interface {
	// Name はプロバイダー名を返す
	Name() string

	// ChatWithTools はツール対応の会話を行う（ストリーミング）
	ChatWithTools(ctx context.Context, systemPrompt string, history []Message, model string) (string, error)

	// SupportsImages はプロバイダーが画像入力に対応しているかを返す
	SupportsImages() bool

	// ChatWithImage は画像付きメッセージで会話を行う
	// imageがnilまたはプロバイダーが画像非対応の場合、テキストのみで会話する
	ChatWithImage(ctx context.Context, systemPrompt string, history []Message, userMessage string, image *ImageData, model string) (string, error)

	// IsFunctionCallingEnabled は Function Calling が有効かを返す
	// true の場合、System Prompt からツール定義を削除して重複を避ける
	IsFunctionCallingEnabled() bool
}

type ProviderFactory func(apiKey string) (Provider, error)

type StreamChoice struct {
	Delta        Delta  `json:"delta"`
	FinishReason string `json:"finish_reason,omitempty"` // "stop", "tool_calls" など
}

type StreamOptions struct {
	IncludeUsage bool `json:"include_usage"` // trueでusage情報を最終チャンクに含める
}

type StreamParser func(line string) (content string, done bool, err error)

type StreamResponse struct {
	Choices []StreamChoice   `json:"choices"`
	Usage   *StreamUsageInfo `json:"usage,omitempty"` // 最終チャンクに含まれる
}

type StreamUsageInfo struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
}

type Usage struct {
	InputTokens  int
	OutputTokens int

	// キャッシュ関連（プロバイダーにより対応状況が異なる）
	CachedInputTokens   int // キャッシュから読み取ったトークン数（割引対象）
	CacheCreationTokens int // キャッシュ作成に使用したトークン数（Claude: 1.25x課金）
}

type UsageCallback func(usage Usage)

type UsageReporter interface {
	// SetUsageCallback は使用量レポートのコールバックを設定する
	SetUsageCallback(callback UsageCallback)
}