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 ClaudeContent struct {
	Type string `json:"type"`
	Text string `json:"text"`
}

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

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

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

type ClaudeRequest struct {
	Model     string          `json:"model"`
	Messages  []ClaudeMessage `json:"messages"`
	System    string          `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 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 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 Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

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

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