Documentation
¶
Index ¶
- func ConvertParametersToSchema(params any) (map[string]any, error)
- type Client
- func (c *Client) CreateBatchEmbedding(ctx context.Context, texts []string) (*base.BatchEmbeddingResult, error)
- func (c *Client) CreateChatCompletionStream(ctx context.Context, messages []chat.Message, requestTools []tools.Tool) (chat.MessageStream, error)
- func (c *Client) CreateEmbedding(ctx context.Context, text string) (*base.EmbeddingResult, error)
- func (c *Client) CreateResponseStream(ctx context.Context, messages []chat.Message, requestTools []tools.Tool) (chat.MessageStream, error)
- func (c *Client) Rerank(ctx context.Context, query string, documents []types.Document, criteria string) ([]float64, error)
- type ResponseStreamAdapter
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
Types ¶
type Client ¶
Client represents an OpenAI client wrapper It implements the provider.Provider interface
func NewClient ¶
func NewClient(ctx context.Context, cfg *latest.ModelConfig, env environment.Provider, opts ...options.Opt) (*Client, error)
NewClient creates a new OpenAI client from the provided configuration
func (*Client) CreateBatchEmbedding ¶ added in v1.9.16
func (c *Client) CreateBatchEmbedding(ctx context.Context, texts []string) (*base.BatchEmbeddingResult, error)
CreateBatchEmbedding generates embedding vectors for multiple texts.
OpenAI supports up to 2048 inputs per request
func (*Client) CreateChatCompletionStream ¶
func (c *Client) CreateChatCompletionStream( ctx context.Context, messages []chat.Message, requestTools []tools.Tool, ) (chat.MessageStream, error)
CreateChatCompletionStream creates a streaming chat completion request It returns a stream that can be iterated over to get completion chunks
func (*Client) CreateEmbedding ¶ added in v1.9.16
CreateEmbedding generates an embedding vector for the given text
func (*Client) CreateResponseStream ¶ added in v1.9.18
func (*Client) Rerank ¶ added in v1.9.22
func (c *Client) Rerank(ctx context.Context, query string, documents []types.Document, criteria string) ([]float64, error)
Rerank scores documents by relevance to the query using an OpenAI chat model. It returns relevance scores in the same order as input documents.
type ResponseStreamAdapter ¶ added in v1.9.18
type ResponseStreamAdapter struct {
// contains filtered or unexported fields
}
ResponseStreamAdapter adapts the OpenAI responses stream to our interface
func (*ResponseStreamAdapter) Close ¶ added in v1.9.18
func (a *ResponseStreamAdapter) Close()
Close closes the stream
func (*ResponseStreamAdapter) Recv ¶ added in v1.9.18
func (a *ResponseStreamAdapter) Recv() (chat.MessageStreamResponse, error)
Recv gets the next completion chunk
Source Files