Просмотр исходного кода

fix: add missing id,object,created

CaIon 1 год назад
Родитель
Сommit
3e03c5a742
2 измененных файлов с 8 добавлено и 1 удалено
  1. 6 1
      dto/text_response.go
  2. 2 0
      relay/channel/openai/relay-openai.go

+ 6 - 1
dto/text_response.go

@@ -1,8 +1,10 @@
 package dto
 package dto
 
 
 type TextResponseWithError struct {
 type TextResponseWithError struct {
-	Choices []OpenAITextResponseChoice    `json:"choices"`
+	Id      string                        `json:"id"`
 	Object  string                        `json:"object"`
 	Object  string                        `json:"object"`
+	Created int64                         `json:"created"`
+	Choices []OpenAITextResponseChoice    `json:"choices"`
 	Data    []OpenAIEmbeddingResponseItem `json:"data"`
 	Data    []OpenAIEmbeddingResponseItem `json:"data"`
 	Model   string                        `json:"model"`
 	Model   string                        `json:"model"`
 	Usage   `json:"usage"`
 	Usage   `json:"usage"`
@@ -10,6 +12,9 @@ type TextResponseWithError struct {
 }
 }
 
 
 type TextResponse struct {
 type TextResponse struct {
+	Id      string                     `json:"id"`
+	Object  string                     `json:"object"`
+	Created int64                      `json:"created"`
 	Model   string                     `json:"model"`
 	Model   string                     `json:"model"`
 	Choices []OpenAITextResponseChoice `json:"choices"`
 	Choices []OpenAITextResponseChoice `json:"choices"`
 	Usage   `json:"usage"`
 	Usage   `json:"usage"`

+ 2 - 0
relay/channel/openai/relay-openai.go

@@ -192,6 +192,8 @@ func OpenaiHandler(c *gin.Context, resp *http.Response, promptTokens int, model
 			}
 			}
 		}
 		}
 		textResponse := &dto.TextResponse{
 		textResponse := &dto.TextResponse{
+			Id:      responseWithError.Id,
+			Object:  responseWithError.Object,
 			Choices: responseWithError.Choices,
 			Choices: responseWithError.Choices,
 			Model:   responseWithError.Model,
 			Model:   responseWithError.Model,
 			Usage:   *usage,
 			Usage:   *usage,