fastapi

package

v0.11.1 Latest Latest Go to latest Published: Jul 30, 2025 License: MIT Imports: 9 Imported by: 0

Details

This section is empty.

This section is empty.

func New(serverURL string, opts ...Option) (embeddings.Embedder, error)

New creates a new FastAPI embedder client.

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

Embedder communicates with a remote FastAPI server to generate embeddings. It directly implements the embeddings.Embedder interface.

func (e *Embedder) EmbedDocuments(ctx context.Context, texts []string) ([][]float32, error)

EmbedDocuments sends a batch of documents to the FastAPI server for embedding.

func (e *Embedder) EmbedQueries(ctx context.Context, texts []string) ([][]float32, error)

EmbedQueries embeds a batch of queries.

func (e *Embedder) EmbedQuery(ctx context.Context, text string) ([]float32, error)

EmbedQuery embeds a single query.

func (e *Embedder) GetDimension(ctx context.Context) (int, error)

GetDimension lazily fetches the embedding dimension by sending a test query.

type Option func(*options)

Option defines a function type for configuring the embedder.

func WithAPIKey(key string) Option

func WithHTTPClient(client *http.Client) Option

WithHTTPClient allows providing a custom http.Client.

func WithLogger(logger *slog.Logger) Option

WithLogger sets a custom logger.

func WithTask(task string) Option

WithTask allows overriding the default task description sent to the API.