jack 6 ngày trước cách đây
mục cha
commit
bea4b136a8
2 tập tin đã thay đổi với 13 bổ sung3 xóa
  1. 2 1
      app/core/tools/file_search_tool.py
  2. 11 2
      main.py

+ 2 - 1
app/core/tools/file_search_tool.py

@@ -102,7 +102,8 @@ class FileSearchTool(BaseTool):
         print(self.__keys)
         print(self.__keys)
         print(self.__dirkeys)
         print(self.__dirkeys)
         files = []
         files = []
-        if self.index == 0:
+        ## 必须有总结的内容query和才能触发
+        if self.index == 0 and query:
             files = FileService.search_in_files(
             files = FileService.search_in_files(
                 query=query, file_keys=self.__keys, folder_keys=self.__dirkeys
                 query=query, file_keys=self.__keys, folder_keys=self.__dirkeys
             )
             )

+ 11 - 2
main.py

@@ -1,6 +1,6 @@
 import logging
 import logging
-
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import JSONResponse
 from app.providers import (
 from app.providers import (
     logging_provider,
     logging_provider,
     app_provider,
     app_provider,
@@ -37,6 +37,15 @@ def boot(_app, provider):
     logging.info(provider.__name__ + " booted")
     logging.info(provider.__name__ + " booted")
 
 
 
 
+# 添加全局异常处理器
+@app.exception_handler(HTTPException)
+async def http_exception_handler(request, exc: HTTPException):
+    return JSONResponse(
+        status_code=exc.status_code,
+        content={"detail": exc.detail, "message": "Resource not found."},
+    )
+
+
 app = create_app()
 app = create_app()