bigmodel

Documentation

Index

• Constants
• type BigModel
  • func NewChatGPT(apiKey string, opts ...Option) BigModel
• type ChatGPT
  • func (gpt *ChatGPT) Chat(messages []*Message) chan Result
• type Message
  • func AssistantMessage(msg string) *Message
  • func Messages(messages ...*Message) []*Message
  • func SystemMessage(msg string) *Message
  • func UserMessage(msg string) *Message
• type Option
  • func WithSpecifyBaseURL(url string) Option
  • func WithSpecifyModel(model string) Option
• type Result

Constants

const (
	TypeData = iota
	TypeDone
	TypeError
)

const (
	RoleUser      = "user"
	RoleAssistant = "assistant"
	RoleSystem    = "system"
)

Types

type BigModel

type BigModel interface {
	// Chat Receive a query for large model calls and write the output results to the Result channel in real time
	Chat(messages []*Message) chan Result
}

BigModel interface

func NewChatGPT

func NewChatGPT(apiKey string, opts ...Option) BigModel

type ChatGPT

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

ChatGPT big model

func (*ChatGPT) Chat

func (gpt *ChatGPT) Chat(messages []*Message) chan Result

type Message

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

func AssistantMessage

func AssistantMessage(msg string) *Message

func Messages

func Messages(messages ...*Message) []*Message

func SystemMessage

func SystemMessage(msg string) *Message

func UserMessage

func UserMessage(msg string) *Message

type Option

type Option func(*ChatGPT)

func WithSpecifyBaseURL

func WithSpecifyBaseURL(url string) Option

WithSpecifyBaseURL specify OPENAI API base url

func WithSpecifyModel

func WithSpecifyModel(model string) Option

WithSpecifyModel specify ChatGPT model

type Result

type Result struct {
	Type    int
	Content string
}