mirrors2/opencatd-open
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: header

Browse Source
This commit is contained in:
Sakurasan
2024-04-17 16:00:21 +08:00
parent 8d6c90c550
commit 4cd019e11e
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
pkg/claude/chat.go
View File

@@ -117,7 +117,12 @@ type ClaudeStreamResponse struct {
} }
func ChatMessages(c *gin.Context, chatReq *openai.ChatCompletionRequest) { func ChatMessages(c *gin.Context, chatReq *openai.ChatCompletionRequest) {
// var haveImages bool
onekey, err := store.SelectKeyCache("openai")
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
return
}
usagelog := store.Tokens{Model: chatReq.Model} usagelog := store.Tokens{Model: chatReq.Model}
var claudReq ChatRequest var claudReq ChatRequest
@@ -177,6 +182,10 @@ func ChatMessages(c *gin.Context, chatReq *openai.ChatCompletionRequest) {
usagelog.PromptCount = tokenizer.NumTokensFromStr(prompt, chatReq.Model) usagelog.PromptCount = tokenizer.NumTokensFromStr(prompt, chatReq.Model)
req, _ := http.NewRequest("POST", MessageEndpoint, strings.NewReader(fmt.Sprintf("%v", bytes.NewReader(claudReq.ByteJson())))) req, _ := http.NewRequest("POST", MessageEndpoint, strings.NewReader(fmt.Sprintf("%v", bytes.NewReader(claudReq.ByteJson()))))
req.Header.Set("x-api-key", onekey.Key)
req.Header.Set("anthropic-version", "2023-06-01")
req.Header.Set("Content-Type", "application/json")
client := http.DefaultClient client := http.DefaultClient
rsp, err := client.Do(req) rsp, err := client.Do(req)
if err != nil { if err != nil {