api

type ChatRequest struct {
	Model    string    `json:"model"`
	Messages []Message `json:"messages"`
	Stream   bool      `json:"stream"`
}

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

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

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

type ClaudeContent struct {
	Type string `json:"type"`
	Text string `json:"text"`
}

type ClaudeContentPart struct {
	Type   string             `json:"type"`             // "text" or "image"
	Text   string             `json:"text,omitempty"`   // type="text"の場合
	Source *ClaudeImageSource `json:"source,omitempty"` // type="image"の場合
}

type ClaudeDelta struct {
	Type string `json:"type"`
	Text string `json:"text"`
}

type ClaudeImageSource struct {
	Type      string `json:"type"`       // "base64"
	MediaType string `json:"media_type"` // "image/png", "image/jpeg" etc
	Data      string `json:"data"`       // Base64エンコードされたデータ
}

type ClaudeMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type ClaudeMultimodalMessage struct {
	Role    string              `json:"role"`
	Content []ClaudeContentPart `json:"content"`
}

type ClaudeMultimodalRequest struct {
	Model    string        `json:"model"`
	Messages []interface{} `json:"messages"` // ClaudeMessage or ClaudeMultimodalMessage
	// System can be either string (legacy) or []ClaudeSystemBlock (prompt caching).
	System    interface{} `json:"system,omitempty"`
	MaxTokens int         `json:"max_tokens"`
	Stream    bool        `json:"stream"`
}

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

type ClaudeRequest struct {
	Model    string          `json:"model"`
	Messages []ClaudeMessage `json:"messages"`
	// System can be either string (legacy) or []ClaudeSystemBlock (prompt caching).
	System    interface{} `json:"system,omitempty"`
	MaxTokens int         `json:"max_tokens"`
	Stream    bool        `json:"stream"`
}

type ClaudeResponse struct {
	Content []ClaudeContent `json:"content"`
}

type ClaudeStreamEvent struct {
	Type  string      `json:"type"`
	Delta ClaudeDelta `json:"delta"`
}

type ClaudeSystemBlock struct {
	Type         string              `json:"type"` // "text"
	Text         string              `json:"text"`
	CacheControl *ClaudeCacheControl `json:"cache_control,omitempty"`
}

type Client struct {
	Provider Provider
	Timeout  time.Duration
}

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

type Delta struct {
	Content string `json:"content"`
}

type GeminiCandidate struct {
	Content GeminiContent `json:"content"`
}

type GeminiContent struct {
	Parts []GeminiPart `json:"parts"`
	Role  string       `json:"role,omitempty"` // "user" or "model"
}

type GeminiInlineData struct {
	MimeType string `json:"mime_type"` // "image/png", "image/jpeg" etc
	Data     string `json:"data"`      // Base64エンコードされたデータ
}

type GeminiMultimodalContent struct {
	Parts []GeminiMultimodalPart `json:"parts"`
	Role  string                 `json:"role,omitempty"` // "user" or "model"
}

type GeminiMultimodalPart struct {
	Text       string            `json:"text,omitempty"`
	InlineData *GeminiInlineData `json:"inline_data,omitempty"`
}

type GeminiMultimodalRequest struct {
	Contents []interface{} `json:"contents"` // GeminiContent or GeminiMultimodalContent
}

type GeminiPart struct {
	Text string `json:"text"`
}

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

type GeminiRequest struct {
	Contents []GeminiContent `json:"contents"`
}

type GeminiResponse struct {
	Candidates []GeminiCandidate `json:"candidates"`
}

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

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

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

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

type OllamaMessageContent struct {
	Content string `json:"content"`
}

type OllamaModel struct {
	Name string `json:"name"`
}

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

type OllamaRequest struct {
	Model    string    `json:"model"`
	Messages []Message `json:"messages"`
	Stream   bool      `json:"stream"`
}

type OllamaStreamResponse struct {
	Message OllamaMessageContent `json:"message"`
	Done    bool                 `json:"done"`
}

type OllamaTagsResponse struct {
	Models []OllamaModel `json:"models"`
}

type OpenAIContentPart struct {
	Type     string          `json:"type"`                // "text" or "image_url"
	Text     string          `json:"text,omitempty"`      // type="text"の場合
	ImageURL *OpenAIImageURL `json:"image_url,omitempty"` // type="image_url"の場合
}

type OpenAIImageURL struct {
	URL string `json:"url"` // "data:image/png;base64,..." 形式
}

type OpenAIMultimodalMessage struct {
	Role    string              `json:"role"`
	Content []OpenAIContentPart `json:"content"`
}

type OpenAIMultimodalRequest struct {
	Model    string        `json:"model"`
	Messages []interface{} `json:"messages"` // Message or OpenAIMultimodalMessage
	Stream   bool          `json:"stream"`
}

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

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

type RAGRequest struct {
	Query  string `json:"query"`
	UserID string `json:"user_id"`
	Model  string `json:"model"`
	TopK   int    `json:"top_k"`
}

type RAGResponse struct {
	Results []RAGResult `json:"results"`
	Query   string      `json:"query"`
	Count   int         `json:"count"`
}

type RAGResult struct {
	ID            string  `json:"id"`
	Content       string  `json:"content"`
	DocumentID    string  `json:"document_id"`
	DocumentTitle string  `json:"document_title"`
	DocumentType  string  `json:"document_type"`
	Similarity    float64 `json:"similarity"`
}

type SerperResponse struct {
	Organic []SerperSearchResult `json:"organic"`
}

type SerperSearchRequest struct {
	Q  string `json:"q"`
	Gl string `json:"gl,omitempty"` // 地域コード (optional)
	Hl string `json:"hl,omitempty"` // 言語コード (optional)
}

type SerperSearchResult struct {
	Title   string `json:"title"`
	Link    string `json:"link"`
	Snippet string `json:"snippet"`
}

type StreamChoice struct {
	Delta Delta `json:"delta"`
}

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

type StreamResponse struct {
	Choices []StreamChoice `json:"choices"`
}