openairetryclient

Documentation

Index

• type OpenAIRetryClient
  • func (c *OpenAIRetryClient) CreateChatCompletionWithRetry(ctx context.Context, request openai.ChatCompletionRequest) (*openai.ChatCompletionResponse, error)
  • func (c *OpenAIRetryClient) CreateEmbeddingsWithRetry(ctx context.Context, request openai.EmbeddingRequest) (*openai.EmbeddingResponse, error)

Types

type OpenAIRetryClient

type OpenAIRetryClient struct {
	openai.Client
	Config struct {
		Timeout     time.Duration
		MaxAttempts uint
	}
}

func (*OpenAIRetryClient) CreateChatCompletionWithRetry

func (c *OpenAIRetryClient) CreateChatCompletionWithRetry(
	ctx context.Context,
	request openai.ChatCompletionRequest,
) (*openai.ChatCompletionResponse, error)

func (*OpenAIRetryClient) CreateEmbeddingsWithRetry

func (c *OpenAIRetryClient) CreateEmbeddingsWithRetry(
	ctx context.Context,
	request openai.EmbeddingRequest,
) (*openai.EmbeddingResponse, error)