chatgpt

Documentation

Index

• type ChatGPTConversion
  • func NewChatGPTConversion(secretKey string) *ChatGPTConversion
  • func (chat *ChatGPTConversion) Ask(ctx context.Context, prompt string, ...) (err error)
  • func (chat *ChatGPTConversion) Init() error
  • func (chat *ChatGPTConversion) RefreshProxy(proxy string) error
  • func (chat *ChatGPTConversion) RefreshSecretKey(secretKey string) error
  • func (chat *ChatGPTConversion) SetBaseURL(baseURL string)
  • func (chat *ChatGPTConversion) SetProxy(proxy string) error
  • func (chat *ChatGPTConversion) SetSystemMsg(content string)
• type ChatMsg
  • func NewChatMsg(role, content, name string) *ChatMsg
• type Request
  • func NewRequest() *Request
  • func (req *Request) GetMessage(options ...string) []openai.ChatCompletionMessage
  • func (req *Request) PopMsg() *ChatMsg
  • func (req *Request) PutSystemMsg(content, name string) string
  • func (req *Request) PutUserMsg(content, name string) (string, string)
  • func (req *Request) SetRes(id string, response *openai.ChatCompletionResponse)
  • func (req *Request) SetResStream(id string, text string, responseStream *openai.ChatCompletionStreamResponse)

Types

type ChatGPTConversion

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

func NewChatGPTConversion

func NewChatGPTConversion(secretKey string) *ChatGPTConversion

func (*ChatGPTConversion) Ask

func (chat *ChatGPTConversion) Ask(ctx context.Context, prompt string, callback func(answer *params.Answer, err error)) (err error)

// ask chatgpt

callback := func(params *chatgpt.CallParams, err error) {
	if params == nil {
		return
	}
	if err != nil {
		log.Println(params.MsgId, err)
	}
	if params.Done {
		log.Println("answer: ", params.MsgId, params.Text)
	}else{
		log.Println("answer: ", params.MsgId, params.Chunk)
	}
}

func (*ChatGPTConversion) Init

func (chat *ChatGPTConversion) Init() error

init client

func (*ChatGPTConversion) RefreshProxy

func (chat *ChatGPTConversion) RefreshProxy(proxy string) error

func (*ChatGPTConversion) RefreshSecretKey

func (chat *ChatGPTConversion) RefreshSecretKey(secretKey string) error

func (*ChatGPTConversion) SetBaseURL

func (chat *ChatGPTConversion) SetBaseURL(baseURL string)

set base URL, default openai URL

func (*ChatGPTConversion) SetProxy

func (chat *ChatGPTConversion) SetProxy(proxy string) error

set proxy chat.SetProxy("socks5://127.0.0.1:3126") chat.SetProxy("http://127.0.0.1:3127")

func (*ChatGPTConversion) SetSystemMsg

func (chat *ChatGPTConversion) SetSystemMsg(content string)

set system role message

type ChatMsg

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

func NewChatMsg

func NewChatMsg(role, content, name string) *ChatMsg

type Request

type Request struct {
	sync.RWMutex
	// contains filtered or unexported fields
}

func NewRequest

func NewRequest() *Request

func (*Request) GetMessage

func (req *Request) GetMessage(options ...string) []openai.ChatCompletionMessage

get message for send ask

func (*Request) PopMsg

func (req *Request) PopMsg() *ChatMsg

func (*Request) PutSystemMsg

func (req *Request) PutSystemMsg(content, name string) string

return msg_id

func (*Request) PutUserMsg

func (req *Request) PutUserMsg(content, name string) (string, string)

renturn msg_id, parent_id

func (*Request) SetRes

func (req *Request) SetRes(id string, response *openai.ChatCompletionResponse)

func (*Request) SetResStream

func (req *Request) SetResStream(id string, text string, responseStream *openai.ChatCompletionStreamResponse)