azopenai

Documentation

Overview

Package azopenai is used to provide the Azure OpenAI service

Index

• type Provider
  • func NewProvider(apiKey string, apiEndpoint string, deploymentID string, apiVersion 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) (ai.ResponseRecver, error)
  • func (p *Provider) Name() string

Types

type Provider

type Provider struct {
	APIKey       string
	APIEndpoint  string
	DeploymentID string
	APIVersion   string
	// contains filtered or unexported fields
}

Provider is the provider for Azure OpenAI

func NewProvider

func NewProvider(apiKey string, apiEndpoint string, deploymentID string, apiVersion string) *Provider

NewProvider creates a new AzureOpenAIProvider

func (*Provider) GetChatCompletions

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

GetChatCompletions get chat completions for ai service

func (*Provider) GetChatCompletionsStream

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

GetChatCompletionsStream implements ai.LLMProvider.

func (*Provider) Name

func (p *Provider) Name() string

Name returns the name of the provider