feat: support dify agent

Calcium-Ion · Jul 30, 2024 · fbce362 · fbce362
1 parent a6b6bcf
commit fbce362
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/relay/channel/dify/relay-dify.go b/relay/channel/dify/relay-dify.go
@@ -53,7 +53,7 @@ func streamResponseDify2OpenAI(difyResponse DifyChunkChatCompletionResponse) *dt
 		choice.Delta.SetContentString("Workflow: " + difyResponse.Data.WorkflowId + "\n")
 	} else if constant.DifyDebug && difyResponse.Event == "node_started" {
 		choice.Delta.SetContentString("Node: " + difyResponse.Data.NodeId + "\n")
-	} else if difyResponse.Event == "message" {
+	} else if difyResponse.Event == "message" || difyResponse.Event == "agent_message" {
 		choice.Delta.SetContentString(difyResponse.Answer)
 	}
 	response.Choices = append(response.Choices, choice)