commit

chat.go

@@ -0,0 +1,313 @@
+package edge
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"errors"
+	"github.com/RomiChan/websocket"
+	"github.com/google/uuid"
+	"io"
+	"net/http"
+	"net/url"
+	"strconv"
+	"strings"
+	"time"
+)
+
+var H = map[string]string{
+	"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0",
+}
+
+type kv = map[string]string
+
+func New(token, agency string) (*Chat, error) {
+	var bu string
+	var ws string
+	if agency != "" {
+		u, err := url.Parse(agency)
+		if err != nil {
+			return nil, err
+		}
+
+		bu = agency + "/turing/conversation/create"
+		ws = "wss://" + u.Hostname() + "/sydney/ChatHub"
+	} else {
+		bu = DefaultCreate
+		ws = DefaultChatHub
+	}
+
+	return NewChat(Options{
+		Retry:     2,
+		WebSock:   ws,
+		CreateURL: bu,
+		Model:     Creative,
+		Headers: map[string]string{
+			"Cookie": "_U=" + token,
+		},
+	}), nil
+}
+
+func NewChat(opt Options) *Chat {
+	has := func(key string) bool {
+		for k, _ := range opt.Headers {
+			if strings.ToLower(k) == key {
+				return true
+			}
+		}
+		return false
+	}
+
+	for k, v := range H {
+		if !has(strings.ToLower(k)) {
+			opt.Headers[k] = v
+		}
+	}
+
+	chat := Chat{Options: opt}
+	return &chat
+}
+
+func (c *Chat) Reply(ctx context.Context, prompt string) (chan PartialResponse, error) {
+	c.mu.Lock()
+	if c.Session.ConversationId == "" {
+		conv, err := c.newConversation()
+		if err != nil {
+			c.mu.Unlock()
+			return nil, err
+		}
+		c.Session = *conv
+	}
+
+	h, err := newHub(c.Model, c.Session, prompt)
+	if err != nil {
+		c.mu.Unlock()
+		return nil, err
+	}
+
+	hub := map[string]any{
+		"arguments":    []any{h},
+		"invocationId": strconv.Itoa(c.Session.InvocationId),
+		"target":       "chat",
+		"type":         4,
+	}
+
+	conn, err := c.newConn()
+	if err != nil {
+		c.mu.Unlock()
+		return nil, err
+	}
+
+	marshal, err := json.Marshal(hub)
+	if err != nil {
+		c.mu.Unlock()
+		return nil, err
+	}
+
+	err = conn.WriteMessage(websocket.TextMessage, bytes.Join([][]byte{marshal, []byte(Delimiter)}, []byte{}))
+	if err != nil {
+		c.mu.Unlock()
+		return nil, err
+	}
+
+	message := make(chan PartialResponse)
+	go c.resolve(ctx, conn, message)
+	//go func() {
+	//	for {
+	//		// 15秒执行一次
+	//		<-time.After(15 * time.Second)
+	//		err = conn.WriteMessage(websocket.TextMessage, ping)
+	//		if err != nil {
+	//			return
+	//		}
+	//	}
+	//}()
+	return message, nil
+}
+
+// 解析回复信息
+func (c *Chat) resolve(ctx context.Context, conn *websocket.Conn, message chan PartialResponse) {
+	defer close(message)
+	defer c.mu.Unlock()
+	handle := func() bool {
+		// 轮询回复消息
+		_, marshal, err := conn.ReadMessage()
+		if err != nil {
+			message <- PartialResponse{
+				Error: err,
+			}
+			return true
+		}
+
+		// 是心跳应答
+		if bytes.Equal(marshal, ping) {
+			return false
+		}
+
+		//fmt.Println(string(marshal))
+		var response PartialResponse
+		slice := bytes.Split(marshal, []byte(Delimiter))
+
+		err = json.Unmarshal(slice[0], &response)
+		if err != nil {
+			message <- PartialResponse{
+				Error: err,
+			}
+			return false
+		}
+
+		// 结束本次应答
+		if response.Type == 2 {
+			if response.Item.Result.Value != "Success" {
+				response.Error = errors.New("消息响应失败：" + response.Item.Result.Message)
+				message <- response
+			}
+			_ = conn.Close()
+
+			c.Session.InvocationId++
+			messages := response.Item.Messages
+			if messages != nil && len(*messages) > 1 {
+				response.Text = (*messages)[1].Text
+				message <- response
+			}
+			return true
+		}
+
+		if len(response.Arguments) == 0 {
+			return false
+		}
+
+		// 处理消息
+		argument := response.Arguments[0]
+		if argument.Messages == nil || len(*argument.Messages) == 0 {
+			return false
+		}
+
+		m := (*argument.Messages)[0]
+		if m.MessageType == "InternalSearchQuery" ||
+			m.MessageType == "InternalSearchResult" ||
+			m.MessageType == "InternalLoaderMessage" {
+			return false
+		}
+
+		response.Text = m.Text
+		message <- response
+		return false
+	}
+
+	for {
+		select {
+		case <-ctx.Done():
+			message <- PartialResponse{
+				Error: errors.New("请求超时"),
+			}
+			_ = conn.Close()
+			return
+		default:
+			if handle() {
+				return
+			}
+		}
+	}
+}
+
+// 创建websocket
+func (c *Chat) newConn() (*websocket.Conn, error) {
+	header := http.Header{}
+	for k, v := range c.Headers {
+		header.Add(k, v)
+	}
+
+	conn, _, err := websocket.DefaultDialer.Dial(c.WebSock, header)
+	if err != nil {
+		return nil, err
+	}
+
+	if e := conn.WriteMessage(websocket.TextMessage, Schema); e != nil {
+		return nil, e
+	}
+
+	if _, _, e := conn.ReadMessage(); e != nil {
+		return nil, e
+	}
+
+	if e := conn.WriteMessage(websocket.TextMessage, ping); e != nil {
+		return nil, e
+	}
+
+	return conn, nil
+}
+
+// 构建对接参数
+func newHub(model string, conv Conversation, prompt string) (map[string]any, error) {
+	var hub map[string]any
+	if err := json.Unmarshal(chatHub, &hub); err != nil {
+		return nil, err
+	}
+
+	messageId := uuid.NewString()
+
+	if model == "Sydney" {
+		hub["sliceIds"] = sSliceIds
+		// TODO -
+	} else {
+		hub["sliceIds"] = sliceIds
+		hub["tone"] = model
+	}
+
+	hub["traceId"] = conv.TraceId
+	hub["requestId"] = messageId
+	hub["conversationSignature"] = conv.Signature
+	hub["conversationId"] = conv.ConversationId
+	hub["participant"] = kv{
+		"id": conv.ClientId,
+	}
+
+	message, ok := hub["message"].(map[string]any)
+	if !ok {
+		return nil, errors.New("reflect `message` fail")
+	}
+
+	message["timestamp"] = time.Now().Format("2006-01-02T15:04:05+08:00")
+	message["requestId"] = messageId
+	message["messageId"] = messageId
+	message["text"] = prompt
+
+	if conv.InvocationId == 0 {
+		hub["isStartOfSession"] = true
+	} else {
+		delete(hub, "previousMessages")
+	}
+	return hub, nil
+}
+
+// 创建会话
+func (c *Chat) newConversation() (*Conversation, error) {
+	request, err := http.NewRequest(http.MethodGet, c.CreateURL, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	for k, v := range c.Headers {
+		request.Header.Add(k, v)
+	}
+
+	r, err := http.DefaultClient.Do(request)
+	if err != nil {
+		return nil, err
+	}
+
+	marshal, err := io.ReadAll(r.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	var conv Conversation
+	if e := json.Unmarshal(marshal, &conv); e != nil {
+		return nil, e
+	}
+
+	conv.TraceId = strings.ReplaceAll(uuid.NewString(), "-", "")
+	conv.InvocationId = 0
+	return &conv, nil
+}

chat.json

@@ -0,0 +1,56 @@
+{
+  "source":"cib",
+  "optionsSets":[
+    "deepleo",
+    "nlu_direct_response_filter",
+    "disable_emoji_spoken_text",
+    "responsible_ai_policy_235",
+    "enablemm",
+    "harmonyv3",
+    "h3imaginative",
+    "rcsprtsalwlst",
+    "dv3sugg",
+    "autosave",
+    "gencontentv3"
+  ],
+  "sliceIds":[],
+  "verbosity":"verbose",
+  "traceId":"648d851d68f547239e9d458285edae5d",
+  "isStartOfSession":false,
+  "requestId":"",
+  "message":{
+    "locale":"zh-CN",
+    "market":"zh-CN",
+    "region":"HK",
+    "location":"lat:47.639557;long:-122.128159;re=1000m;",
+    "locationHints":[
+      {
+        "country":"Hong Kong",
+        "state":"Hong Kong",
+        "city":"Aberdeen",
+        "timezoneoffset":8,
+        "countryConfidence":8,
+        "cityConfidence":5,
+        "Center":{
+          "Latitude":22.2478,
+          "Longitude":114.1517
+        },
+        "RegionType":2,
+        "SourceType":1
+      }
+    ],
+    "timestamp":"2023-06-17T18:04:16+08:00",
+    "author":"user",
+    "inputMethod":"Keyboard",
+    "text":"",
+    "messageType":"Chat",
+    "requestId":"",
+    "messageId":""
+  },
+  "tone":"Creative",
+  "conversationSignature":"",
+  "participant":{ "id":"" },
+  "spokenTextMode":"None",
+  "conversationId":"",
+  "previousMessages":[]
+}