embedder

type Config struct {
	Provider Provider      `yaml:"provider"`
	Ollama   *OllamaConfig `yaml:"ollama,omitempty"`
	OpenAI   *OpenAIConfig `yaml:"openai,omitempty"`
}

type EmbedOutcome struct {
	Embedding Embedding
	Err       error
}

type EmbedRequest struct {
	ToolName string
	Args     json.RawMessage
}

type Embedder interface {
	Embed(ctx context.Context, req EmbedRequest) (Embedding, error)
	Provider() Provider
	Close() error
}

type Embedding struct {
	Vector []float32
	Model  string
}

type OllamaConfig struct {
	URL   string `yaml:"url"`
	Model string `yaml:"model"`
}

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

type OpenAIConfig struct {
	Model  string `yaml:"model"`
	APIKey string `yaml:"api_key,omitempty"`
}

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

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

type PoolConfig struct {
	Size  int `yaml:"size"`
	Queue int `yaml:"queue"`
}

type Provider string