From 1385b165c9a62a486b4494ebb11320d3c1117a5e Mon Sep 17 00:00:00 2001
From: ageer <ageerle@163.com>
Date: 星期日, 02 三月 2025 11:19:29 +0800
Subject: [PATCH] feat(知识库): 增加知识库模块
---
ruoyi-modules/ruoyi-fusion/src/main/java/org/ruoyi/fusion/controller/ChatController.java | 17 +++++++++++++++++
1 files changed, 17 insertions(+), 0 deletions(-)
diff --git a/ruoyi-modules/ruoyi-fusion/src/main/java/org/ruoyi/fusion/controller/ChatController.java b/ruoyi-modules/ruoyi-fusion/src/main/java/org/ruoyi/fusion/controller/ChatController.java
index 0f717ed..dc5592f 100644
--- a/ruoyi-modules/ruoyi-fusion/src/main/java/org/ruoyi/fusion/controller/ChatController.java
+++ b/ruoyi-modules/ruoyi-fusion/src/main/java/org/ruoyi/fusion/controller/ChatController.java
@@ -17,7 +17,9 @@
import org.ruoyi.common.mybatis.core.page.PageQuery;
import org.ruoyi.common.mybatis.core.page.TableDataInfo;
import org.ruoyi.common.satoken.utils.LoginHelper;
+import org.ruoyi.knowledge.service.EmbeddingService;
import org.ruoyi.system.domain.bo.ChatMessageBo;
+import org.ruoyi.system.domain.request.translation.TranslationRequest;
import org.ruoyi.system.domain.vo.ChatMessageVo;
import org.ruoyi.system.service.IChatMessageService;
import org.ruoyi.system.service.ISseService;
@@ -46,12 +48,16 @@
private final IChatMessageService chatMessageService;
+ private final EmbeddingService embeddingService;
/**
* 鑱婂ぉ鎺ュ彛
*/
@PostMapping("/send")
@ResponseBody
public SseEmitter sseChat(@RequestBody @Valid ChatRequest chatRequest, HttpServletRequest request) {
+ if (chatRequest.getModel().startsWith("ollama")) {
+ return ISseService.ollamaChat(chatRequest);
+ }
return ISseService.sseChat(chatRequest,request);
}
@@ -89,6 +95,17 @@
return ISseService.textToSpeed(textToSpeech);
}
+ /**
+ * 鏂囨湰缈昏瘧
+ *
+ * @param
+ */
+ @PostMapping("/translation")
+ @ResponseBody
+ public String translation(@RequestBody TranslationRequest translationRequest) {
+ return ISseService.translation(translationRequest);
+ }
+
@PostMapping("/dall3")
@ResponseBody
public R<List<Item>> dall3(@RequestBody @Valid Dall3Request request) {
--
Gitblit v1.9.3