chatgpt

Documentation

Index

• func New(opts ...ai.ClientOption) (ai.AI, error)
• func NewWithClient(client *openai.Client, model string) ai.AI
• type ChatGPT
  • func (ai *ChatGPT) Chat(ctx context.Context, messages ...string) (ai.ChatResponse, error)
  • func (ai *ChatGPT) ChatSession() ai.ChatSession
  • func (ai *ChatGPT) ChatStream(ctx context.Context, messages ...string) (ai.ChatStream, error)
  • func (ai *ChatGPT) Close() error
  • func (ChatGPT) LLMs() ai.LLMs
  • func (chatgpt *ChatGPT) Model(_ context.Context) (string, error)
  • func (ai *ChatGPT) SetCount(i int32)
  • func (ai *ChatGPT) SetJSONResponse(b bool)
  • func (chatgpt *ChatGPT) SetLimit(limit rate.Limit)
  • func (ai *ChatGPT) SetMaxTokens(i int32)
  • func (ai *ChatGPT) SetModel(model string)
  • func (ai *ChatGPT) SetTemperature(f float32)
  • func (ai *ChatGPT) SetTopP(f float32)
• type ChatGPTResponse
• type ChatResponse
  • func (resp *ChatResponse[Response]) Results() (res []string)
  • func (resp *ChatResponse[Response]) String() string
  • func (resp *ChatResponse[Response]) TokenCount() (res ai.TokenCount)
• type ChatSession
  • func (session *ChatSession) Chat(ctx context.Context, messages ...string) (ai.ChatResponse, error)
  • func (session *ChatSession) ChatStream(ctx context.Context, messages ...string) (ai.ChatStream, error)
  • func (session *ChatSession) History() (history []ai.Message)
• type ChatStream
  • func (stream *ChatStream) Close() error
  • func (stream *ChatStream) Next() (ai.ChatResponse, error)

Functions

func New

func New(opts ...ai.ClientOption) (ai.AI, error)

func NewWithClient

func NewWithClient(client *openai.Client, model string) ai.AI

Types

type ChatGPT

type ChatGPT struct {
	*openai.Client
	// contains filtered or unexported fields
}

func (*ChatGPT) Chat

func (ai *ChatGPT) Chat(ctx context.Context, messages ...string) (ai.ChatResponse, error)

func (*ChatGPT) ChatSession

func (ai *ChatGPT) ChatSession() ai.ChatSession

func (*ChatGPT) ChatStream

func (ai *ChatGPT) ChatStream(ctx context.Context, messages ...string) (ai.ChatStream, error)

func (*ChatGPT) Close

func (ai *ChatGPT) Close() error

func (ChatGPT) LLMs

func (ChatGPT) LLMs() ai.LLMs

func (*ChatGPT) Model

func (chatgpt *ChatGPT) Model(_ context.Context) (string, error)

func (*ChatGPT) SetCount

func (ai *ChatGPT) SetCount(i int32)

func (*ChatGPT) SetJSONResponse

func (ai *ChatGPT) SetJSONResponse(b bool)

func (*ChatGPT) SetLimit

func (chatgpt *ChatGPT) SetLimit(limit rate.Limit)

func (*ChatGPT) SetMaxTokens

func (ai *ChatGPT) SetMaxTokens(i int32)

func (*ChatGPT) SetModel

func (ai *ChatGPT) SetModel(model string)

func (*ChatGPT) SetTemperature

func (ai *ChatGPT) SetTemperature(f float32)

func (*ChatGPT) SetTopP

func (ai *ChatGPT) SetTopP(f float32)

type ChatGPTResponse

type ChatGPTResponse interface {
	openai.ChatCompletionResponse | openai.ChatCompletionStreamResponse
}

type ChatResponse

type ChatResponse[Response ChatGPTResponse] struct {
	// contains filtered or unexported fields
}

func (*ChatResponse[Response]) Results

func (resp *ChatResponse[Response]) Results() (res []string)

func (*ChatResponse[Response]) String

func (resp *ChatResponse[Response]) String() string

func (*ChatResponse[Response]) TokenCount

func (resp *ChatResponse[Response]) TokenCount() (res ai.TokenCount)

type ChatSession

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

func (*ChatSession) Chat

func (session *ChatSession) Chat(ctx context.Context, messages ...string) (ai.ChatResponse, error)

func (*ChatSession) ChatStream

func (session *ChatSession) ChatStream(ctx context.Context, messages ...string) (ai.ChatStream, error)

func (*ChatSession) History

func (session *ChatSession) History() (history []ai.Message)

type ChatStream

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

func (*ChatStream) Close

func (stream *ChatStream) Close() error

func (*ChatStream) Next

func (stream *ChatStream) Next() (ai.ChatResponse, error)