From 579fc8129e32f5c631756ba08baddfa740802bd1 Mon Sep 17 00:00:00 2001
From: CalciumIon <1808837298@qq.com>
Date: Wed, 10 Jul 2024 15:36:17 +0800
Subject: [PATCH] fix: dify (close #355)

---
 relay/channel/dify/dto.go        | 2 +-
 relay/channel/dify/relay-dify.go | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/relay/channel/dify/dto.go b/relay/channel/dify/dto.go
index 74ef604..bc641a2 100644
--- a/relay/channel/dify/dto.go
+++ b/relay/channel/dify/dto.go
@@ -21,7 +21,7 @@ type DifyData struct {
 
 type DifyChatCompletionResponse struct {
 	ConversationId string       `json:"conversation_id"`
-	Answers        string       `json:"answers"`
+	Answer         string       `json:"answer"`
 	CreateAt       int64        `json:"create_at"`
 	MetaData       DifyMetaData `json:"metadata"`
 }
diff --git a/relay/channel/dify/relay-dify.go b/relay/channel/dify/relay-dify.go
index e3fe933..c674ba1 100644
--- a/relay/channel/dify/relay-dify.go
+++ b/relay/channel/dify/relay-dify.go
@@ -117,6 +117,7 @@ func difyStreamHandler(c *gin.Context, resp *http.Response, info *relaycommon.Re
 func difyHandler(c *gin.Context, resp *http.Response, info *relaycommon.RelayInfo) (*dto.OpenAIErrorWithStatusCode, *dto.Usage) {
 	var difyResponse DifyChatCompletionResponse
 	responseBody, err := io.ReadAll(resp.Body)
+
 	if err != nil {
 		return service.OpenAIErrorWrapper(err, "read_response_body_failed", http.StatusInternalServerError), nil
 	}
@@ -134,7 +135,7 @@ func difyHandler(c *gin.Context, resp *http.Response, info *relaycommon.RelayInf
 		Created: common.GetTimestamp(),
 		Usage:   difyResponse.MetaData.Usage,
 	}
-	content, _ := json.Marshal(difyResponse.Answers)
+	content, _ := json.Marshal(difyResponse.Answer)
 	choice := dto.OpenAITextResponseChoice{
 		Index: 0,
 		Message: dto.Message{