From e6938bd236528648b2db8a956fbf3e49307a911f Mon Sep 17 00:00:00 2001
From: JustSong <songquanpeng@foxmail.com>
Date: Sun, 23 Jul 2023 00:58:14 +0800
Subject: [PATCH] fix: fix PaLM2 not billed

---
 controller/relay-palm.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/controller/relay-palm.go b/controller/relay-palm.go
index d9a82498..74624c7f 100644
--- a/controller/relay-palm.go
+++ b/controller/relay-palm.go
@@ -131,6 +131,9 @@ func palmStreamHandler(c *gin.Context, resp *http.Response) (*OpenAIErrorWithSta
 		fullTextResponse := streamResponsePaLM2OpenAI(&palmResponse)
 		fullTextResponse.Id = responseId
 		fullTextResponse.Created = createdTime
+		if len(palmResponse.Candidates) > 0 {
+			responseText = palmResponse.Candidates[0].Content
+		}
 		jsonResponse, err := json.Marshal(fullTextResponse)
 		if err != nil {
 			common.SysError("error marshalling stream response: " + err.Error())