From 8fc4b7a49d359c67cd24c07f314d17517fb1beb7 Mon Sep 17 00:00:00 2001 From: linweiyuan Date: Sun, 23 Jul 2023 19:48:31 +0800 Subject: [PATCH] fix imitate api invalid model in response --- api/imitate/api.go | 2 +- api/imitate/response.go | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/api/imitate/api.go b/api/imitate/api.go index d56e89883..37c7193de 100644 --- 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 index 1e9adc269..c081469a3 100644 --- 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,