Documentation
¶
Overview ¶
Package embeddings provides Ollama embedding client implementation
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type EmbeddingClient ¶
type EmbeddingClient interface {
// Embed generates embeddings for multiple texts
Embed(ctx context.Context, texts []string) ([][]float32, error)
// EmbedOne generates an embedding for a single text
EmbedOne(ctx context.Context, text string) ([]float32, error)
}
EmbeddingClient defines the interface for generating text embeddings
type OllamaClient ¶ added in v1.1.1
type OllamaClient struct {
// contains filtered or unexported fields
}
OllamaClient implements EmbeddingClient using local Ollama API
func NewOllamaClient ¶ added in v1.1.1
func NewOllamaClient(baseURL, model string) *OllamaClient
NewOllamaClient creates a new Ollama embedding client baseURL is typically "http://localhost:11434" model is the embedding model name, e.g. "nomic-embed-text"
type OpenAIClient ¶
OpenAIClient implements EmbeddingClient using OpenAI's API
func NewOpenAIClient ¶
func NewOpenAIClient(apiKey string) *OpenAIClient
NewOpenAIClient creates a new OpenAI embedding client
Click to show internal directories.
Click to hide internal directories.