xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-07 17:53:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: 支持讯飞大模型 v3.0

Browse Source
This commit is contained in:
RockYang
2023-11-23 07:11:13 +08:00
parent 13fbbc190a
commit 191e3b7d2c
8 changed files with 41 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
api/handler/chatimpl/xunfei_handler.go
View File

@@ -48,6 +48,12 @@ type xunFeiResp struct {
} `json:"payload"`
}
var Model2URL = map[string]string{
"generalv1": "1.1",
"generalv2": "v2.1",
"generalv3": "v3.1",
}
// 科大讯飞消息发送实现
func (h *ChatHandler) sendXunFeiMessage(
@@ -82,13 +88,7 @@ func (h *ChatHandler) sendXunFeiMessage(
return nil
}
var apiURL string
if req.Model == "generalv2" {
apiURL = strings.Replace(h.App.ChatConfig.XunFei.ApiURL, "{version}", "v2.1", 1)
} else {
apiURL = strings.Replace(h.App.ChatConfig.XunFei.ApiURL, "{version}", "v1.1", 1)
}
apiURL := strings.Replace(h.App.ChatConfig.XunFei.ApiURL, "{version}", Model2URL[req.Model], 1)
wsURL, err := assembleAuthUrl(apiURL, key[1], key[2])
//握手并建立websocket 连接
conn, resp, err := d.Dial(wsURL, nil)