Documentation
¶
Index ¶
Constants ¶
View Source
const ( DefaultEmbeddingModel = "text-embedding-004" ModelContextVar = "model" APIKeyEnvVar = "GEMINI_API_KEY" )
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Client ¶
type Client struct {
DefaultModel embeddings.EmbeddingModel
Client *genai.Client
DefaultContext *context.Context
MaxBatchSize int
// contains filtered or unexported fields
}
func NewGeminiClient ¶
func (*Client) CreateEmbedding ¶
type GeminiEmbeddingFunction ¶
type GeminiEmbeddingFunction struct {
// contains filtered or unexported fields
}
func NewGeminiEmbeddingFunction ¶
func NewGeminiEmbeddingFunction(opts ...Option) (*GeminiEmbeddingFunction, error)
func (*GeminiEmbeddingFunction) EmbedDocuments ¶
func (e *GeminiEmbeddingFunction) EmbedDocuments(ctx context.Context, documents []string) ([]embeddings.Embedding, error)
func (*GeminiEmbeddingFunction) EmbedQuery ¶
func (e *GeminiEmbeddingFunction) EmbedQuery(ctx context.Context, document string) (embeddings.Embedding, error)
type Option ¶
func WithClient ¶
WithClient sets the generative AI client for the client
func WithDefaultModel ¶
func WithDefaultModel(model embeddings.EmbeddingModel) Option
WithDefaultModel sets the default model for the client
func WithEnvAPIKey ¶
func WithEnvAPIKey() Option
WithEnvAPIKey sets the API key for the client from the environment variable GOOGLE_API_KEY
func WithMaxBatchSize ¶
WithMaxBatchSize sets the max batch size for the client - this acts as a limit for the number of embeddings that can be sent in a single request
Source Files
Click to show internal directories.
Click to hide internal directories.