openai

Documentation

Overview

Package openai is the OpenAI llm provider

Index

• Constants
• type Provider
  • func NewProvider(apiKey string, model string) *Provider
  • func (p *Provider) GetChatCompletions(ctx context.Context, req openai.ChatCompletionRequest, _ metadata.M) (openai.ChatCompletionResponse, error)
  • func (p *Provider) GetChatCompletionsStream(ctx context.Context, req openai.ChatCompletionRequest, _ metadata.M) (bridgeai.ResponseRecver, error)
  • func (p *Provider) Name() string

Constants

const APIEndpoint = "https://api.openai.com/v1/chat/completions"

APIEndpoint is the endpoint for OpenAI

Types

type Provider

type Provider struct {
	// APIKey is the API key for OpenAI
	APIKey string
	// Model is the model for OpenAI
	// eg. "gpt-3.5-turbo-1106", "gpt-4-turbo-preview", "gpt-4-vision-preview", "gpt-4"
	Model string
	// contains filtered or unexported fields
}

Provider is the provider for OpenAI

func NewProvider

func NewProvider(apiKey string, model string) *Provider

NewProvider creates a new OpenAIProvider

func (*Provider) GetChatCompletions

func (p *Provider) GetChatCompletions(ctx context.Context, req openai.ChatCompletionRequest, _ metadata.M) (openai.ChatCompletionResponse, error)

GetChatCompletions implements ai.LLMProvider.

func (*Provider) GetChatCompletionsStream

func (p *Provider) GetChatCompletionsStream(ctx context.Context, req openai.ChatCompletionRequest, _ metadata.M) (bridgeai.ResponseRecver, error)

GetChatCompletionsStream implements ai.LLMProvider.

func (*Provider) Name

func (p *Provider) Name() string

Name returns the name of the provider