添加向量库

qianbao
1 parent ea8b3b64
Showing 4 changed files with 85 additions and 66 deletions
src/main/java/com/xly/exception/TtsExceptionFilter.java
src/main/java/com/xly/ocr/service/OcrService.java
src/main/java/com/xly/tool/DynamicToolProvider.java
src/main/resources/application.yml
-package com.xly.exception;
-
-import com.fasterxml.jackson.databind.ObjectMapper;
-import com.xly.tts.bean.TTSResponseDTO;
-import jakarta.servlet.*;
-import jakarta.servlet.http.HttpServletRequest;
-import jakarta.servlet.http.HttpServletResponse;
-import lombok.RequiredArgsConstructor;
-import lombok.extern.slf4j.Slf4j;
-import org.springframework.core.annotation.Order;
-import org.springframework.http.HttpStatus;
-import org.springframework.http.MediaType;
-import org.springframework.stereotype.Component;
-
-import java.io.IOException;
-
-@Slf4j
-@Component
-@Order(1)
-@RequiredArgsConstructor
-public class TtsExceptionFilter implements Filter {
-
-    private final ObjectMapper objectMapper;
-
-    @Override
-    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
-            throws IOException, ServletException {
-
-        HttpServletRequest httpRequest = (HttpServletRequest) request;
-        HttpServletResponse httpResponse = (HttpServletResponse) response;
-
-        // 只处理 TTS 相关接口
-        if (!httpRequest.getRequestURI().contains("/api/tts/")) {
-            chain.doFilter(request, response);
-            return;
-        }
-
-        try {
-            chain.doFilter(request, response);
-        } catch (Exception ex) {
-            log.error("TTS接口异常: {}", httpRequest.getRequestURI(), ex);
-
-            // 清除之前的响应
-            if (httpResponse.isCommitted()) {
-                return;
-            }
-
-            httpResponse.reset();
-            httpResponse.setStatus(HttpStatus.INTERNAL_SERVER_ERROR.value());
-            httpResponse.setContentType(MediaType.APPLICATION_JSON_VALUE);
-            httpResponse.setCharacterEncoding("UTF-8");
-
-            TTSResponseDTO errorResult = TTSResponseDTO.error(
-                    HttpStatus.INTERNAL_SERVER_ERROR.value(),
-                    "语音合成服务异常: " + ex.getMessage()
-            );
-
-            String jsonResponse = objectMapper.writeValueAsString(errorResult);
-            httpResponse.getWriter().write(jsonResponse);
-            httpResponse.getWriter().flush();
-        }
-    }
-}
 \ No newline at end of file
+//package com.xly.exception;
+//
+//import com.fasterxml.jackson.databind.ObjectMapper;
+//import com.xly.tts.bean.TTSResponseDTO;
+//import jakarta.servlet.*;
+//import jakarta.servlet.http.HttpServletRequest;
+//import jakarta.servlet.http.HttpServletResponse;
+//import lombok.RequiredArgsConstructor;
+//import lombok.extern.slf4j.Slf4j;
+//import org.springframework.core.annotation.Order;
+//import org.springframework.http.HttpStatus;
+//import org.springframework.http.MediaType;
+//import org.springframework.stereotype.Component;
+//
+//import java.io.IOException;
+//
+//@Slf4j
+//@Component
+//@Order(1)
+//@RequiredArgsConstructor
+//public class TtsExceptionFilter implements Filter {
+//
+//    private final ObjectMapper objectMapper;
+//
+//    @Override
+//    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
+//            throws IOException, ServletException {
+//
+//        HttpServletRequest httpRequest = (HttpServletRequest) request;
+//        HttpServletResponse httpResponse = (HttpServletResponse) response;
+//
+//        // 只处理 TTS 相关接口
+//        if (!httpRequest.getRequestURI().contains("/api/tts/")) {
+//            chain.doFilter(request, response);
+//            return;
+//        }
+//
+//        try {
+//            chain.doFilter(request, response);
+//        } catch (Exception ex) {
+//            log.error("TTS接口异常: {}", httpRequest.getRequestURI(), ex);
+//
+//            // 清除之前的响应
+//            if (httpResponse.isCommitted()) {
+//                return;
+//            }
+//
+//            httpResponse.reset();
+//            httpResponse.setStatus(HttpStatus.INTERNAL_SERVER_ERROR.value());
+//            httpResponse.setContentType(MediaType.APPLICATION_JSON_VALUE);
+//            httpResponse.setCharacterEncoding("UTF-8");
+//
+//            TTSResponseDTO errorResult = TTSResponseDTO.error(
+//                    HttpStatus.INTERNAL_SERVER_ERROR.value(),
+//                    "语音合成服务异常: " + ex.getMessage()
+//            );
+//
+//            String jsonResponse = objectMapper.writeValueAsString(errorResult);
+//            httpResponse.getWriter().write(jsonResponse);
+//            httpResponse.getWriter().flush();
+//        }
+//    }
+//}
 \ No newline at end of file
@@ -19,7 +19,7 @@ public class OcrService {
     private static final List<String> ALLOWED_EXTENSIONS = Arrays.asList(
             ".jpg", ".jpeg", ".png", ".bmp", ".tiff", ".gif"
     );
-    private static final long MAX_FILE_SIZE = 10 * 1024 * 1024; // 10MB
+    private static final long MAX_FILE_SIZE = 100 * 1024 * 1024; // 10MB
     /**
      * 从 MultipartFile 中提取文字
@@ -909,8 +909,20 @@ public class DynamicToolProvider implements ToolProvider {
     }
     private Boolean checkMiss(Map<String,Object> returnMap,ParamRule pd) {
+        if(ObjectUtil.isEmpty(returnMap)){
+            returnMap = new HashMap<>();
+        }
         Boolean bBhcs = (ObjectUtil.isEmpty(returnMap.get(pd.getSParam())));
-        Boolean bDbZero = (pd.getSParamValue().startsWith("d") && 0 == Double.valueOf(returnMap.get(pd.getSParam()).toString()));
+        Boolean bDbZero = false;
+        if(pd.getSParamValue().startsWith("d")){
+            try{
+                String sParam = pd.getSParam().toString();
+                Object sParamV = returnMap.get(sParam);
+                bDbZero = (ObjectUtil.isEmpty(sParamV) || 0 ==Double.valueOf(sParamV.toString()));
+            }catch (Exception e){
+                bDbZero = true;
+            }
+        }
         return  bDbZero || bBhcs  || (!returnMap.containsKey(pd.getSParamValue()) || (ObjectUtil.isEmpty(returnMap.get(pd.getSParamValue()))));
     }
@@ -9,7 +9,6 @@ logging:
     org.springframework: warn
     ai.djl: DEBUG
     dev.langchain4j: DEBUG
-
 server:
   port: 8099
   servlet:
@@ -21,6 +20,14 @@ server:
     enabled: true
     mime-types: audio/mpeg
 spring:
+  # 将 multipart
+  servlet:
+    multipart:
+      max-file-size: 100MB
+      max-request-size: 100MB
+#      enabled: true
+#      location: D:/xlyweberp/ai/upload-temp  # 可选：指定临时文件目录
+#      file-size-threshold: 2MB  # 可选：超过2MB才写入磁盘
   main:
     allow-bean-definition-overriding: true
   application: