core

Documentation

Index

• Variables
• func DoPrompt(promptName string, in string, variables map[string]string) (string, []openai.ChatCompletionMessage)
• func RunWithHistory(history []openai.ChatCompletionMessage) (string, []openai.ChatCompletionMessage)
• type Chat
  • func (c *Chat) AddMessage(msg openai.ChatCompletionMessage)
  • func (c *Chat) CreateResponse()
  • func (c *Chat) Init()
• type OpenAI
  • func (o *OpenAI) CreateChatCompletion(ctx context.Context, messages []openai.ChatCompletionMessage) (string, Usage, error)
  • func (o *OpenAI) SetTemperature(t float32)
  • func (o *OpenAI) StreamChatCompletion(ctx context.Context, messages []openai.ChatCompletionMessage) (*openai.ChatCompletionStream, error)
• type Usage
  • func (u *Usage) Add(other Usage)

Variables

var (
	Temperature float32
	Model       string
)

Functions

func DoPrompt

func DoPrompt(promptName string, in string, variables map[string]string) (string, []openai.ChatCompletionMessage)

func RunWithHistory

func RunWithHistory(history []openai.ChatCompletionMessage) (string, []openai.ChatCompletionMessage)

Types

type Chat

type Chat struct {
	History []openai.ChatCompletionMessage
	Usage   Usage
	Single  bool
}

func (*Chat) AddMessage

func (c *Chat) AddMessage(msg openai.ChatCompletionMessage)

func (*Chat) CreateResponse

func (c *Chat) CreateResponse()

func (*Chat) Init

func (c *Chat) Init()

type OpenAI

type OpenAI struct {
	// contains filtered or unexported fields
}

func (*OpenAI) CreateChatCompletion

func (o *OpenAI) CreateChatCompletion(ctx context.Context, messages []openai.ChatCompletionMessage) (string, Usage, error)

func (*OpenAI) SetTemperature

func (o *OpenAI) SetTemperature(t float32)

func (*OpenAI) StreamChatCompletion

func (o *OpenAI) StreamChatCompletion(ctx context.Context, messages []openai.ChatCompletionMessage) (*openai.ChatCompletionStream, error)

type Usage

type Usage openai.Usage

func (*Usage) Add

func (u *Usage) Add(other Usage)