openai

README

openai

API Integration with OpenAI

Documentation

Index

• Constants
• type OpenAIClient
  • func NewOpenAIClient(model string) (*OpenAIClient, error)
  • func (c *OpenAIClient) Prompt(ctx context.Context, messages []openai.ChatCompletionMessage) (string, error)

Constants

const (
	OPENAI_TOKEN  = "OPENAI_TOKEN"
	OPENAI_PREFIX = "openai-"
)

Types

type OpenAIClient

type OpenAIClient struct {
	Client *openai.Client
	Model  string
}

func NewOpenAIClient

func NewOpenAIClient(model string) (*OpenAIClient, error)

NewClient creates a new OpenAI API client using an environment variable.

func (*OpenAIClient) Prompt

func (c *OpenAIClient) Prompt(ctx context.Context, messages []openai.ChatCompletionMessage) (string, error)

Prompt prompts the OpenAI API with a list of messages using 0.000001 temperature and a retry mechanism.