Jelajahi Sumber

针对dify工作流的流式输出修改

Zhangbw 2 bulan lalu
induk
melakukan
35d31b491c

+ 5 - 1
ruoyi-modules/yp-talk/src/main/java/org/dromara/talk/service/impl/DifyServiceImpl.java

@@ -17,7 +17,11 @@ import lombok.extern.slf4j.Slf4j;
 @RequiredArgsConstructor
 public class DifyServiceImpl implements IDifyService {
     private final DifyConfig difyConfig;
-    private final OkHttpClient httpClient = new OkHttpClient();
+    private final OkHttpClient httpClient = new OkHttpClient.Builder()
+        .connectTimeout(30, java.util.concurrent.TimeUnit.SECONDS)
+        .readTimeout(60, java.util.concurrent.TimeUnit.SECONDS)
+        .writeTimeout(30, java.util.concurrent.TimeUnit.SECONDS)
+        .build();
 
     public Map<String, String> callWorkflow(String userMessage, Map<String, Object> inputs, Long userId, String conversationId) {
         try {