init

Author: CC11001100

.gitignore

@@ -0,0 +1 @@
+.idea

README.assets/image-20221207201626228.png

@@ -0,0 +1,109 @@
+# ChatGPT API  For Golang 
+
+# 一、这是什么
+
+这是一个ChatGPT的Golang API库，让你能够把ChatGPT集成到你的Go应用中，注意这是基于逆向工程实现的并不是一个官方库。
+
+优势：
+
+- 只需要提供自己的JWT就能够像在网页上使用一样在Go程序中使用ChatGPT，很方便的就可以把ChatGPT集成到您开发的各种Go应用中 
+
+- 会话维持，ChatGPT是可以保持部分上下文的（这一点上比智障小爱同学强多了...），对于这一点本API库提供了支持，比如：
+
+  - ```
+    我：你好，我的名字叫陈二！
+    ChatGPT: 你好，陈二！很高兴认识你。我是 Assistant，一个大型语言模型，旨在帮助人们了解更多关于世界的信息。如果你有什么问题，我将尽力回答。
+    我：我的名字叫什么呀？
+    ChatGPT：你告诉我你的名字叫陈二。你的名字是一个很普通的中国姓氏和名字，在中国文化中，许多人都有一个姓氏和一个名字。例如，陈是一个常见的姓氏，而二是一个常见的名字。不过，每个人的名字都是独一无二的，所以你的名字只属于你自己。
+    ```
+
+    甚至在你的JWT次数用尽之后可以换一个JWT继续维持会话。
+
+# 二、安装
+
+```bash
+go get -u github.com/golang-infrastructure/go-ChatGPT
+```
+
+# 三、如何使用
+
+## 3.1 如何获取ChatGPT的JWT？
+
+首先你要注册成功一个openapi的账号（教程此处不提供，请自行Google），并且能够在网页上正常使用ChatGPT，然后在ChatGPT的聊天页面，地址是这个：
+
+```
+https://chat.openai.com/chat
+```
+
+按F12打开控制台，粘贴如下代码，即可看到自己的JWT Token：
+
+```js
+JSON.parse(document.getElementById("__NEXT_DATA__").text).props.pageProps.accessToken
+```
+
+如图：
+
+![image-20221207201626228](README.assets/image-20221207201626228.png)
+
+## 3.2 API代码示例
+
+```go
+package main
+
+import (
+	"fmt"
+	chatgpt "github.com/golang-infrastructure/go-ChatGPT"
+)
+
+func main() {
+
+	// 把JWT放到这里
+	jwt := "xxx"
+
+	chat := chatgpt.NewChat(jwt)
+	talk, err := chat.Talk("你好，我的名字叫陈二！")
+	if err != nil {
+		fmt.Println(err.Error())
+		return
+	}
+	fmt.Println(talk.Message.Content)
+
+	talk, err = chat.Talk("我的名字叫什么呀？")
+	if err != nil {
+		fmt.Println(err.Error())
+		return
+	}
+	fmt.Println(talk.Message.Content)
+
+	// Output:
+	// {text [你好，陈二！很高兴认识你。我是 Assistant，一个大型语言模型，旨在帮助人们了解更多关于世界的信息。如果你有什么问题，我将尽力回答。]}
+	// {text [你告诉我你的名字叫陈二。你的名字是一个很普通的中国姓氏和名字，在中国文化中，许多人都有一个姓氏和一个名字。例如，陈是一个常见的姓氏，而二是一个常见的名字。不过，每个人的名字都是独一无二的，所以你的名字只属于你自己。]}
+
+}
+
+```
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+

README.md

@@ -0,0 +1,197 @@
+package chatgpt
+
+import (
+	"encoding/json"
+	"errors"
+	"github.com/PuerkitoBio/goquery"
+	"github.com/go-resty/resty/v2"
+	if_expression "github.com/golang-infrastructure/go-if-expression"
+	"github.com/golang-infrastructure/go-pointer"
+	"github.com/google/uuid"
+	"strings"
+)
+
+// ------------------------------------------------- 请求相关 ------------------------------------------------------------
+
+type Request struct {
+	Action          string           `json:"action"`
+	Messages        []RequestMessage `json:"messages"`
+	ConversationID  *string          `json:"conversation_id"`
+	ParentMessageID *string          `json:"parent_message_id"`
+	Model           string           `json:"model"`
+}
+
+type RequestMessage struct {
+	ID      string         `json:"id"`
+	Role    string         `json:"role"`
+	Content RequestContent `json:"content"`
+}
+
+type RequestContent struct {
+	ContentType string   `json:"content_type"`
+	Parts       []string `json:"parts"`
+}
+
+func NewRequest(question, conversationID, parentMessageID string) *Request {
+	return &Request{
+		Action:         "next",
+		ConversationID: pointer.ToPointerOrNil(conversationID),
+		Messages: []RequestMessage{
+			{
+				ID:   uuid.New().String(),
+				Role: "user",
+				Content: RequestContent{
+					ContentType: "text",
+					Parts:       []string{question},
+				},
+			},
+		},
+		ParentMessageID: pointer.ToPointerOrNil(parentMessageID),
+		Model:           "text-davinci-002-render",
+	}
+}
+
+// ------------------------------------------------- 响应相关 ------------------------------------------------------------
+
+type Response struct {
+	Message        ResponseMessage `json:"message"`
+	ConversationID string          `json:"conversation_id"`
+	Error          any             `json:"error"`
+}
+
+type ResponseMessage struct {
+	ID         string           `json:"id"`
+	Role       string           `json:"role"`
+	User       any              `json:"user"`
+	CreateTime any              `json:"create_time"`
+	UpdateTime any              `json:"update_time"`
+	Content    ResponseContent  `json:"content"`
+	EndTurn    any              `json:"end_turn"`
+	Weight     float64          `json:"weight"`
+	Metadata   ResponseMetadata `json:"metadata"`
+	Recipient  string           `json:"recipient"`
+}
+
+type ResponseContent struct {
+	ContentType string   `json:"content_type"`
+	Parts       []string `json:"parts"`
+}
+
+type ResponseMetadata struct {
+}
+
+// ------------------------------------------------- --------------------------------------------------------------------
+
+const ConversationAPIURL = "https://chat.openai.com/backend-api/conversation"
+
+type ChatGPT struct {
+	jwt           string
+	authorization string
+
+	userAgent string
+
+	conversationID  string
+	parentMessageID string
+}
+
+func NewChat(jwt string) *ChatGPT {
+	return &ChatGPT{
+		jwt:           jwt,
+		authorization: "Bearer " + jwt,
+		userAgent:     "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/105.0.0.0 Safari/537.36",
+	}
+}
+
+func (x *ChatGPT) Talk(question string) (*Response, error) {
+	conversationID := if_expression.Return(x.conversationID != "", x.conversationID, "")
+	parentMessageID := if_expression.Return(x.parentMessageID != "", x.parentMessageID, uuid.New().String())
+	response, err := x.sendConversationMessage(question, conversationID, parentMessageID)
+	if err != nil {
+		return nil, err
+	}
+	x.conversationID = response.ConversationID
+	x.parentMessageID = response.Message.ID
+	return response, nil
+}
+
+func (x *ChatGPT) sendConversationMessage(question, conversationID, parentMessageID string) (*Response, error) {
+	request := NewRequest(question, conversationID, parentMessageID)
+	requestBytes, err := json.Marshal(request)
+	if err != nil {
+		return nil, err
+	}
+	response, err := resty.
+		New().
+		R().
+		SetHeader("User-Agent", x.userAgent).
+		SetHeader("Authorization", x.authorization).
+		SetHeader("Content-Type", "application/json").
+		SetBody(string(requestBytes)).
+		Post(ConversationAPIURL)
+	if err != nil {
+		return nil, err
+	}
+	responseString := response.String()
+	arr := strings.Split(responseString, "\n\n")
+	index := len(arr) - 3
+	if index >= len(arr) || index < 1 {
+		return nil, x.buildErrMessage(responseString)
+	}
+	arr = strings.Split(arr[index], "data: ")
+	if len(arr) < 2 {
+		return nil, x.buildErrMessage(responseString)
+	}
+	chatResponse := &Response{}
+	if err = json.Unmarshal([]byte(arr[1]), chatResponse); err != nil {
+		return nil, err
+	}
+	return chatResponse, nil
+}
+
+// 在调用失败的时候尝试返回更友好的信息
+func (x *ChatGPT) buildErrMessage(responseString string) error {
+	errContent := x.tryExtractErrorMessage(responseString)
+	if errContent != "" {
+		return errors.New(errContent)
+	} else {
+		return errors.New(responseString)
+	}
+}
+
+// 尝试从响应中抽取出错误信息
+func (x *ChatGPT) tryExtractErrorMessage(responseString string) string {
+	doc, err := goquery.NewDocumentFromReader(strings.NewReader(responseString))
+	if err != nil {
+		return ""
+	}
+	return doc.Find("#content").Text()
+}
+
+func (x *ChatGPT) GetConversationID() string {
+	return x.conversationID
+}
+
+func (x *ChatGPT) SetConversationID(conversationID string) {
+	x.conversationID = conversationID
+}
+
+func (x *ChatGPT) GetParentMessageID() string {
+	return x.parentMessageID
+}
+
+func (x *ChatGPT) SetParentMessageID(parentMessageID string) {
+	x.parentMessageID = parentMessageID
+}
+
+func (x *ChatGPT) GetUserAgent() string {
+	return x.userAgent
+}
+
+func (x *ChatGPT) SetUserAgent(userAgent string) {
+	x.userAgent = userAgent
+}
+
+func (x *ChatGPT) SetJWT(jwt string) {
+	x.jwt = jwt
+	x.authorization = "Bearer " + jwt
+}

chatgpt.go

@@ -0,0 +1,26 @@
+package chatgpt
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestFoo(t *testing.T) {
+
+	// 把JWT Token放到这里
+	jwt := "xxxx"
+
+	chat := NewChat(jwt)
+	talk, err := chat.Talk("你好，我的名字叫陈二！")
+	assert.Nil(t, err)
+	t.Log(talk.Message.Content)
+
+	talk, err = chat.Talk("我的名字叫什么呀？")
+	assert.Nil(t, err)
+	t.Log(talk.Message.Content)
+
+	// Output:
+	//     chatgpt_test.go:15: {text [你好陈二，很高兴认识你！]}
+	//    chatgpt_test.go:19: {text [你已经告诉我你的名字是陈二了。你可以直接说：“我的名字叫陈二。”]}
+
+}

chatgpt_test.go

@@ -0,0 +1,47 @@
+<!DOCTYPE html PUBLIC '-//W3C//DTD XHTML 1.0 Transitional//EN'
+        'http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd'>
+<html xmlns='http://www.w3.org/1999/xhtml'>
+<head>
+    <meta content='text/html; charset=utf-8' http-equiv='content-type'/>
+    <style type='text/css'>body {
+        font-family: Arial;
+        margin-left: 40px;
+    }
+
+    img {
+        border: 0 none;
+    }
+
+    #content {
+        margin-left: auto;
+        margin-right: auto
+    }
+
+    #message h2 {
+        font-size: 20px;
+        font-weight: normal;
+        color: #000000;
+        margin: 34px 0px 0px 0px
+    }
+
+    #message p {
+        font-size: 13px;
+        color: #000000;
+        margin: 7px 0px 0 px0px
+    }
+
+    #errorref {
+        font-size: 11px;
+        color: #737373;
+        margin-top: 41px
+    }</style>
+    <title>Service unavailable</title></head>
+<body>
+<div id='content'>
+    <div id='message'><h2>Our services aren't available right now</h2>
+        <p>We're working to restore all services as soon as possible. Please check back soon.</p></div>
+    <div id='errorref'><span>0b36QYwAAAABq02pIbjvPR56/Jwe1K0HZV1NURURHRTAxMDcAZTY2YjhiMDMtMDc5My00NDA5LTk3NzMtMmU2MTJlNzFhMWUz</span>
+    </div>
+</div>
+</body>
+</html>