fix imitate api invalid model in response

linweiyuan · Jul 23, 2023 · 8fc4b7a · 8fc4b7a
1 parent d2e4bbf
commit 8fc4b7a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/api/imitate/api.go b/api/imitate/api.go
@@ -116,7 +116,7 @@ func CreateChatCompletions(c *gin.Context) {
 	}
 
 	if !originalRequest.Stream {
-		c.JSON(200, newChatCompletion(fullResponse))
+		c.JSON(200, newChatCompletion(fullResponse, translatedRequest.Model))
 	} else {
 		c.String(200, "data: [DONE]\n\n")
 	}

diff --git a/api/imitate/response.go b/api/imitate/response.go
@@ -124,12 +124,12 @@ type StringStruct struct {
 }
 
 //goland:noinspection SpellCheckingInspection
-func newChatCompletion(fullTest string) ChatCompletion {
+func newChatCompletion(fullTest, model string) ChatCompletion {
 	return ChatCompletion{
 		ID:      "chatcmpl-QXlha2FBbmROaXhpZUFyZUF3ZXNvbWUK",
 		Object:  "chat.completion",
 		Created: int64(0),
-		Model:   "gpt-3.5-turbo-0301",
+		Model:   model,
 		Usage: usage{
 			PromptTokens:     0,
 			CompletionTokens: 0,