openai

func NewAdvancedMockOpenAIEmbedder ¶ added in v1.9.1

func NewAdvancedMockOpenAIEmbedder(dimension int) *AdvancedMockOpenAIEmbedder

NewAdvancedMockOpenAIEmbedder creates a new advanced mock with configurable behavior.

func NewOpenAIEmbedder ¶

func NewOpenAIEmbedder(config *Config, tracer trace.Tracer) (*OpenAIEmbedder, error)

NewOpenAIEmbedder creates a new OpenAIEmbedder with the given configuration. It creates an OpenAI client from the configuration for generating text embeddings.

Parameters:

• config: Configuration containing API key, model name, base URL, and other settings
• tracer: OpenTelemetry tracer for observability (can be nil)

Returns:

• *OpenAIEmbedder: A new OpenAI embedder instance
• error: Configuration validation errors or client creation errors

Example:

config := &openai.Config{
    APIKey:  "your-api-key",
    Model:   "text-embedding-3-small",
    Timeout: 30 * time.Second,
}
embedder, err := openai.NewOpenAIEmbedder(config, tracer)
if err != nil {
    log.Fatal(err)
}
embeddings, err := embedder.EmbedDocuments(ctx, texts)

Example usage can be found in examples/rag/simple/main.go.

func NewOpenAIEmbedderWithClient ¶

func NewOpenAIEmbedderWithClient(config *Config, tracer trace.Tracer, client Client) (*OpenAIEmbedder, error)

NewOpenAIEmbedderWithClient creates a new OpenAIEmbedder with a provided client. This is primarily used for testing with mocked clients or when you need to inject a custom client implementation.

Parameters:

• config: Configuration containing API key, model name, and other settings
• tracer: OpenTelemetry tracer for observability (can be nil)
• client: OpenAI client implementation (must not be nil)

Returns:

• *OpenAIEmbedder: A new OpenAI embedder instance
• error: Configuration validation errors or if client is nil

Example:

mockClient := &MockOpenAIClient{}
embedder, err := openai.NewOpenAIEmbedderWithClient(config, tracer, mockClient)

Example usage can be found in examples/rag/simple/main.go.