llm

type Choice struct {
	Index        int       `json:"index"`
	Message      Message   `json:"message"`
	Delta        Message   `json:"delta,omitempty"`
	FinishReason string    `json:"finish_reason"`
	Logprobs     *Logprobs `json:"logprobs,omitempty"`
}

type CompletionRequest struct {
	Messages      []Message        `json:"messages"`
	Model         string           `json:"model,omitempty"`
	Temperature   float64          `json:"temperature,omitempty"`
	MaxTokens     int              `json:"max_tokens,omitempty"`
	Tools         []ToolDefinition `json:"tools,omitempty"`
	ToolChoice    interface{}      `json:"tool_choice,omitempty"`
	Stream        bool             `json:"stream,omitempty"`
	SystemPrompt  string           `json:"system_prompt,omitempty"`
	StopSequences []string         `json:"stop_sequences,omitempty"`
}

type CompletionResponse struct {
	ID                string                 `json:"id"`
	Object            string                 `json:"object"`
	Created           int64                  `json:"created"`
	Model             string                 `json:"model"`
	Choices           []Choice               `json:"choices"`
	Usage             Usage                  `json:"usage"`
	SystemFingerprint string                 `json:"system_fingerprint,omitempty"`
	Metadata          map[string]interface{} `json:"metadata,omitempty"`
}

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

type Function struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description"`
	Parameters  map[string]interface{} `json:"parameters"`
}

type FunctionCall struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

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

type Logprobs struct {
	Tokens        []string             `json:"tokens"`
	TokenLogprobs []float64            `json:"token_logprobs"`
	TopLogprobs   []map[string]float64 `json:"top_logprobs"`
}

type Message struct {
	Role       string                 `json:"role"` // "system", "user", "assistant", "tool"
	Content    string                 `json:"content"`
	Name       string                 `json:"name,omitempty"`
	ToolCalls  []ToolCall             `json:"tool_calls,omitempty"`
	ToolCallID string                 `json:"tool_call_id,omitempty"`
	Metadata   map[string]interface{} `json:"metadata,omitempty"`
}

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

type OllamaModelInfo struct {
	Name       string    `json:"name"`
	ModifiedAt time.Time `json:"modified_at"`
	Size       int64     `json:"size"`
	Digest     string    `json:"digest"`
}

type OllamaModelsResponse struct {
	Models []OllamaModelInfo `json:"models"`
}

type OllamaOptions struct {
	Temperature float64  `json:"temperature,omitempty"`
	TopP        float64  `json:"top_p,omitempty"`
	TopK        int      `json:"top_k,omitempty"`
	NumPredict  int      `json:"num_predict,omitempty"`
	Stop        []string `json:"stop,omitempty"`
}

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

type OllamaRequest struct {
	Model     string          `json:"model"`
	Messages  []OllamaMessage `json:"messages"`
	Stream    bool            `json:"stream,omitempty"`
	Options   OllamaOptions   `json:"options,omitempty"`
	Format    string          `json:"format,omitempty"`
	KeepAlive string          `json:"keep_alive,omitempty"`
}

type OllamaResponse struct {
	Model     string        `json:"model"`
	CreatedAt time.Time     `json:"created_at"`
	Message   OllamaMessage `json:"message"`
	Done      bool          `json:"done"`
	Error     string        `json:"error,omitempty"`
}

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

type Provider interface {
	// GetName returns the provider name
	GetName() string

	// GetModels returns available models
	GetModels(ctx context.Context) ([]string, error)

	// Complete generates a completion (auto-detects streaming based on request)
	Complete(ctx context.Context, req CompletionRequest) (*CompletionResponse, error)

	// CompleteStream generates a streaming completion
	CompleteStream(ctx context.Context, req CompletionRequest, callback StreamCallback) error

	// CompleteWithMode generates a completion with explicit streaming mode
	CompleteWithMode(ctx context.Context, req CompletionRequest, mode StreamMode) (*CompletionResponse, error)

	// CompleteStreamWithMode generates a streaming completion with explicit mode
	CompleteStreamWithMode(ctx context.Context, req CompletionRequest, callback StreamCallback, mode StreamMode) error

	// IsHealthy checks if the provider is healthy
	IsHealthy(ctx context.Context) error

	// GetConfig returns provider configuration
	GetConfig() map[string]interface{}

	// SetConfig updates provider configuration
	SetConfig(config map[string]interface{}) error

	// SupportsStreaming returns true if the provider supports streaming
	SupportsStreaming() bool

	// GetStreamingConfig returns the current streaming configuration
	GetStreamingConfig() *StreamingConfig

	// SetStreamingConfig updates the streaming configuration
	SetStreamingConfig(config *StreamingConfig) error

	// Close closes the provider and cleans up resources
	Close() error
}

type ProviderConfig struct {
	Name        string                 `json:"name"`
	Type        string                 `json:"type"`
	Endpoint    string                 `json:"endpoint,omitempty"`
	APIKey      string                 `json:"api_key,omitempty"`
	Model       string                 `json:"model,omitempty"`
	Temperature float64                `json:"temperature,omitempty"`
	MaxTokens   int                    `json:"max_tokens,omitempty"`
	Timeout     time.Duration          `json:"timeout,omitempty"`
	RetryCount  int                    `json:"retry_count,omitempty"`
	RetryDelay  time.Duration          `json:"retry_delay,omitempty"`
	Headers     map[string]string      `json:"headers,omitempty"`
	Streaming   *StreamingConfig       `json:"streaming,omitempty"`
	Metadata    map[string]interface{} `json:"metadata,omitempty"`
}

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

type SimpleTokenCounter struct{}

type StreamCallback func(chunk CompletionResponse) error

type StreamMode string

type StreamingConfig struct {
	Enabled    bool       `json:"enabled"`
	Mode       StreamMode `json:"mode"`
	ChunkSize  int        `json:"chunk_size,omitempty"`
	BufferSize int        `json:"buffer_size,omitempty"`
	FlushDelay int        `json:"flush_delay_ms,omitempty"` // milliseconds
	KeepAlive  bool       `json:"keep_alive,omitempty"`
}

type TokenCounter interface {
	CountTokens(text string) (int, error)
	CountMessagesTokens(messages []Message) (int, error)
}

type ToolCall struct {
	ID       string                 `json:"id"`
	Type     string                 `json:"type"`
	Function FunctionCall           `json:"function"`
	Metadata map[string]interface{} `json:"metadata,omitempty"`
}

type ToolDefinition struct {
	Type     string   `json:"type"`
	Function Function `json:"function"`
}

type Usage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}