embedding

type Embedding interface {
	// Embed 把单段查询文本转成向量（语义视为 query）
	Embed(ctx context.Context, text string) ([]float32, error)

	// EmbedBatch 批量向量化（语义视为 document，索引时用）
	EmbedBatch(ctx context.Context, texts []string) ([][]float32, error)

	// EmbedTyped 显式指定 task type 的批量调用。supplier 不支持 task type
	// 时退化为普通 EmbedBatch；支持的供应商（Gemini/Voyage）会按 task 优化。
	EmbedTyped(ctx context.Context, texts []string, task TaskType) ([][]float32, error)

	Dimension() int
	Name() string
}

type GeminiConfig struct {
	APIKey string
	Model  string // 默认 text-embedding-004
	// 限额覆盖（0 = 用 paid tier 默认；用 free tier 时调小，如 RPM=100 TPM=30000）
	RPM int
	TPM int
}

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

type OllamaConfig struct {
	Model   string // 默认 nomic-embed-text
	BaseURL string // 默认 http://localhost:11434
	Dim     int    // 向量维度（不同模型不同）
}

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

type OpenAIConfig struct {
	APIKey  string
	Model   string // OpenAI 默认 text-embedding-3-small；自托管按服务实际 model id 填
	BaseURL string // 默认 https://api.openai.com/v1；LM Studio 一般是 http://host:1234/v1
	Dim     int    // 当未知模型或自托管时手动指定；0 = 用内置映射（仅 OpenAI 已知模型）
}

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

type TaskType int

type VoyageAIConfig struct {
	APIKey string
	Model  string // 默认 voyage-code-3
}

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