hera

Documentation

Index

• func InitHeraOpenAI(bearer string)
• type OpenAI
  • func (ai *OpenAI) MakeCodeGenRequest(ctx context.Context, userID int, params OpenAIParams) (openai.CompletionResponse, error)
• type OpenAIParams

Functions

func InitHeraOpenAI

func InitHeraOpenAI(bearer string)

Types

type OpenAI

type OpenAI struct {
	*openai.Client
}

var HeraOpenAI OpenAI

func (*OpenAI) MakeCodeGenRequest

func (ai *OpenAI) MakeCodeGenRequest(ctx context.Context, userID int, params OpenAIParams) (openai.CompletionResponse, error)

type OpenAIParams

type OpenAIParams struct {
	Model     string
	MaxTokens int
	Prompt    string
}