xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-17 16:06:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: dify (close #355)

Browse Source
This commit is contained in:
CalciumIon 2024-07-10 15:36:17 +08:00
parent f55f63f412
commit 579fc8129e
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
relay/channel/dify/dto.go
View File

@ -21,7 +21,7 @@ type DifyData struct {
type DifyChatCompletionResponse struct { type DifyChatCompletionResponse struct {
ConversationId string `json:"conversation_id"` ConversationId string `json:"conversation_id"`
Answers string `json:"answers"` Answer string `json:"answer"`
CreateAt int64 `json:"create_at"` CreateAt int64 `json:"create_at"`
MetaData DifyMetaData `json:"metadata"` MetaData DifyMetaData `json:"metadata"`
} }

3
relay/channel/dify/relay-dify.go
View File

@ -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) { func difyHandler(c *gin.Context, resp *http.Response, info *relaycommon.RelayInfo) (*dto.OpenAIErrorWithStatusCode, *dto.Usage) {
var difyResponse DifyChatCompletionResponse var difyResponse DifyChatCompletionResponse
responseBody, err := io.ReadAll(resp.Body) responseBody, err := io.ReadAll(resp.Body)
if err != nil { if err != nil {
return service.OpenAIErrorWrapper(err, "read_response_body_failed", http.StatusInternalServerError), 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(), Created: common.GetTimestamp(),
Usage: difyResponse.MetaData.Usage, Usage: difyResponse.MetaData.Usage,
} }
content, _ := json.Marshal(difyResponse.Answers) content, _ := json.Marshal(difyResponse.Answer)
choice := dto.OpenAITextResponseChoice{ choice := dto.OpenAITextResponseChoice{
Index: 0, Index: 0,
Message: dto.Message{ Message: dto.Message{