From 5dc2c8179ff246c8c03e6e5b5fc8006a8a3d69ac Mon Sep 17 00:00:00 2001
From: qfrjava <13402782+qfrjava@user.noreply.gitee.com>
Date: 星期四, 27 三月 2025 18:16:59 +0800
Subject: [PATCH] feat(SMTMetricsManagerControl):增强维度列表解析功能- 添加对 dim_value_list 字段的详细解析逻辑 - 支持解析并返回 SQL 查询和值列表 - 优化 alias 数据的处理方式 - 移除未使用的类型转换代码
---
JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/control/SMTKnowledgeControl.java | 42 +++++++++++++++++++++---------------------
1 files changed, 21 insertions(+), 21 deletions(-)
diff --git a/JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/control/SMTKnowledgeControl.java b/JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/control/SMTKnowledgeControl.java
index e7f88a9..a9dc942 100644
--- a/JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/control/SMTKnowledgeControl.java
+++ b/JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/control/SMTKnowledgeControl.java
@@ -2,11 +2,7 @@
import java.io.BufferedReader;
import java.io.StringReader;
-import java.util.ArrayList;
-import java.util.Date;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.multipart.MultipartFile;
@@ -423,23 +419,27 @@
public ModelAndView addKnowledgeFile(SMTAIServerRequest tranReq, @RequestParam(value = "file", required = false) MultipartFile file) throws Exception
{
- String fileName = file.getOriginalFilename();
- String groupId = tranReq.convParamToString("group_id", true);
- String fileText = SMTStatic.readTextStream(file.getInputStream());
+ String fileText = "";
+ String fileName = file.getOriginalFilename();
+ String groupId = tranReq.convParamToString("group_id", true);
+ fileText = SMTStatic.readTextStream(file.getInputStream());
-
- String fileType;
- List<String> listBlock = null;
- if(fileName.endsWith(".md"))
- {
- fileType = "markdown";
- listBlock = splitMarkdownToVectorBlock(fileText);
- }
- else
- {
- return tranReq.returnJsonState(false, "鏂囦欢鏍煎紡涓嶆敮鎸�", null);
- }
-
+ String fileType = "";
+ List<String> listBlock = null;
+
+ // 鍏堝垽鏂枃浠跺悕鏄惁涓虹┖锛岄伩鍏嶇┖鎸囬拡寮傚父
+ if (fileName != null && fileName.contains(".")) {
+ fileType = fileName.substring(fileName.lastIndexOf(".")).toLowerCase(); // 缁熶竴杞崲涓哄皬鍐欙紝閬垮厤澶у皬鍐欓棶棰�
+ }
+
+ // 鐢� Set 瀛樺偍鏀寔鐨勬枃浠剁被鍨嬶紝鎻愰珮鍙鎬�
+ Set<String> supportedTypes = new HashSet<>(Arrays.asList(".md", ".pdf", ".docx", ".doc"));
+ if (supportedTypes.contains(fileType)) {
+ fileText = SMTAIServerApp.fileTranslTxt(file);
+ } else {
+ return tranReq.returnJsonState(false, "鏂囦欢鏍煎紡涓嶆敮鎸�", null);
+ }
+ listBlock = splitMarkdownToVectorBlock(fileText);
List<String> listVector = new ArrayList<>();
SMTLLMConnect llm = SMTAIServerApp.getApp().allocLLMConnect(null);
try
--
Gitblit v1.9.3