Kaynağa Gözat

完成其他模型联网

tsurumure 5 ay önce
ebeveyn
işleme
fdb4e0c293

+ 7 - 1
src/main/java/com/backendsys/modules/sdk/deepseek/service/impl/DeepSeekClientImpl.java

@@ -114,7 +114,7 @@ public class DeepSeekClientImpl implements DeepSeekClient {
             // -- [博查] Web Search API ----------------------------------------------
             if (internet) {
 
-                // 博查、统计接口时间、设置返回参数
+                // 远程查询、统计接口时间、设置返回参数
                 long internetStartTime = System.currentTimeMillis();
                 JsonNode searchResult = bochaService.WebSearch(new BochaParam(prompt));
                 String context = bochaService.WebSearchToString(searchResult);
@@ -124,6 +124,12 @@ public class DeepSeekClientImpl implements DeepSeekClient {
 
                 // 将搜索结果作为上下文添加到消息中
                 messages.add(new DSRequestMessage("system", context));
+                messages.add(new DSRequestMessage("user", "在回答时引用以上全部数据进行分析")); // 的 "name"、"summary"
+                messages.add(new DSRequestMessage("assistant", "好的"));
+
+                // [SSE] 发送消息
+                ChatSseMessage chatSearchSseMessage = new ChatSseMessage("SEARCH", context, null, history_code);
+                sseUtil.send(user_id, new SseResponse(SseResponseEnum.OLLAMA, chatSearchSseMessage).toJsonStr());
             }
             // -----------------------------------------------------------------------
 

+ 7 - 1
src/main/java/com/backendsys/modules/sdk/tencentcloud/huanyuan/service/impl/HunYuanClientImpl.java

@@ -108,7 +108,7 @@ public class HunYuanClientImpl implements HunYuanClient {
             // -- [博查] Web Search API ----------------------------------------------
             if (internet) {
 
-                // 博查、统计接口时间、设置返回参数
+                // 远程查询、统计接口时间、设置返回参数
                 long internetStartTime = System.currentTimeMillis();
                 JsonNode searchResult = bochaService.WebSearch(new BochaParam(prompt));
                 String context = bochaService.WebSearchToString(searchResult);
@@ -118,6 +118,12 @@ public class HunYuanClientImpl implements HunYuanClient {
 
                 // 将搜索结果作为上下文添加到消息中
                 messages.add(setMessage("system", context));
+                messages.add(setMessage("user", "在回答时引用以上全部数据进行分析")); // 的 "name"、"summary"
+                messages.add(setMessage("assistant", "好的"));
+
+                // [SSE] 发送消息
+                ChatSseMessage chatSearchSseMessage = new ChatSseMessage("SEARCH", context, null, history_code);
+                sseUtil.send(user_id, new SseResponse(SseResponseEnum.OLLAMA, chatSearchSseMessage).toJsonStr());
             }
             // -----------------------------------------------------------------------