Documentation
¶
Overview ¶
Package openai provides an implementation of the Provider interface for OpenAI.
Index ¶
- func NewOpenAIProvider(config *core.ProviderConfig) (core.Provider, error)
- type OpenAIProvider
- func (p *OpenAIProvider) ChatCompletion(ctx context.Context, request *core.ChatCompletionRequest) (*core.ChatCompletionResponse, error)
- func (p *OpenAIProvider) Close() error
- func (p *OpenAIProvider) CountTokens(ctx context.Context, text string, modelID string) (int, error)
- func (p *OpenAIProvider) CreateEmbedding(ctx context.Context, request *core.EmbeddingRequest) (*core.EmbeddingResponse, error)
- func (p *OpenAIProvider) GetModels(ctx context.Context) ([]core.ModelInfo, error)
- func (p *OpenAIProvider) StreamingChatCompletion(ctx context.Context, request *core.ChatCompletionRequest, ...) error
- func (p *OpenAIProvider) TextCompletion(ctx context.Context, request *core.TextCompletionRequest) (*core.TextCompletionResponse, error)
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func NewOpenAIProvider ¶
func NewOpenAIProvider(config *core.ProviderConfig) (core.Provider, error)
NewOpenAIProvider creates a new OpenAI provider
Types ¶
type OpenAIProvider ¶
type OpenAIProvider struct {
*core.BaseProvider
// contains filtered or unexported fields
}
OpenAIProvider is an implementation of the Provider interface for OpenAI
func (*OpenAIProvider) ChatCompletion ¶
func (p *OpenAIProvider) ChatCompletion(ctx context.Context, request *core.ChatCompletionRequest) (*core.ChatCompletionResponse, error)
ChatCompletion generates a chat completion
func (*OpenAIProvider) Close ¶
func (p *OpenAIProvider) Close() error
Close closes the provider and releases any resources
func (*OpenAIProvider) CountTokens ¶
CountTokens counts the number of tokens in a text
func (*OpenAIProvider) CreateEmbedding ¶
func (p *OpenAIProvider) CreateEmbedding(ctx context.Context, request *core.EmbeddingRequest) (*core.EmbeddingResponse, error)
CreateEmbedding creates an embedding
func (*OpenAIProvider) StreamingChatCompletion ¶
func (p *OpenAIProvider) StreamingChatCompletion(ctx context.Context, request *core.ChatCompletionRequest, callback func(response *core.ChatCompletionResponse) error) error
StreamingChatCompletion generates a streaming chat completion
func (*OpenAIProvider) TextCompletion ¶
func (p *OpenAIProvider) TextCompletion(ctx context.Context, request *core.TextCompletionRequest) (*core.TextCompletionResponse, error)
TextCompletion generates a text completion
Source Files
Click to show internal directories.
Click to hide internal directories.