From 4cd019e11eafd1c6a08c1627e71a15b52e45d9bb Mon Sep 17 00:00:00 2001
From: Sakurasan <26715255+Sakurasan@users.noreply.github.com>
Date: Wed, 17 Apr 2024 16:00:21 +0800
Subject: [PATCH] fix: header

---
 pkg/claude/chat.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/pkg/claude/chat.go b/pkg/claude/chat.go
index 64d5f01..e37f3c9 100644
--- a/pkg/claude/chat.go
+++ b/pkg/claude/chat.go
@@ -117,7 +117,12 @@ type ClaudeStreamResponse struct {
 }
 
 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}
 	var claudReq ChatRequest
@@ -177,6 +182,10 @@ func ChatMessages(c *gin.Context, chatReq *openai.ChatCompletionRequest) {
 	usagelog.PromptCount = tokenizer.NumTokensFromStr(prompt, chatReq.Model)
 
 	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
 	rsp, err := client.Do(req)
 	if err != nil {