provider

type BatchEmbedResult struct {
	Results []EmbedResult
	Errors  int
}

type EmbedRequest struct {
	ID      int
	Content string
}

type EmbedResult struct {
	ID        int
	Embedding []float32
	Error     error
}

type EmbeddingProvider interface {
	// Name returns the provider name (e.g., "gemini", "ollama")
	Name() string

	// Dimensions returns the embedding vector dimensions
	Dimensions() int

	// BatchSize returns the maximum batch size for embedding requests
	BatchSize() int

	// MaxTokens returns the maximum token limit for this model
	MaxTokens() int

	// EmbedChunks embeds multiple chunks in a batch
	EmbedChunks(ctx context.Context, chunks []EmbedRequest) BatchEmbedResult

	// EmbedQuery embeds a single query for search
	EmbedQuery(ctx context.Context, query string) []float32

	// Validate tests the provider connection/credentials
	Validate(ctx context.Context) error
}

type EmbeddingProviderFactory func(cfg config.ProviderConfig, apiKey string) (EmbeddingProvider, error)

type LLMError struct {
	Message    string
	StatusCode int
}

type LLMProvider interface {
	// Name returns the provider name
	Name() string

	// Call makes a non-streaming LLM request
	Call(ctx context.Context, req LLMRequest) (*LLMResult, *LLMError)

	// Stream makes a streaming LLM request, calling onChunk for each token
	Stream(ctx context.Context, req LLMRequest, onChunk func(string)) (*LLMResult, *LLMError)

	// Validate tests the provider connection/credentials
	Validate(ctx context.Context) error
}

type LLMProviderFactory func(cfg config.ProviderConfig, apiKey string) (LLMProvider, error)

type LLMRequest struct {
	Prompt       string
	SystemPrompt string
	MaxTokens    int
	Temperature  float32
}

type LLMResult struct {
	Text         string
	InputTokens  int
	OutputTokens int
}

type MockEmbeddingProvider struct {
	NameVal       string
	DimensionsVal int
	BatchSizeVal  int
	MaxTokensVal  int
	EmbedFunc     func(ctx context.Context, chunks []EmbedRequest) BatchEmbedResult
	QueryFunc     func(ctx context.Context, query string) []float32
}

type MockLLMProvider struct {
	NameVal    string
	CallFunc   func(ctx context.Context, req LLMRequest) (*LLMResult, *LLMError)
	StreamFunc func(ctx context.Context, req LLMRequest, onChunk func(string)) (*LLMResult, *LLMError)
}