'commit'

dsRag/StartEs.py

@@ -41,13 +41,14 @@ async def lifespan(app: FastAPI):
     # 初始化阿里云大模型工具
     app.state.aliyun_util = ALiYunUtil()
     yield
+
+
 app = FastAPI(lifespan=lifespan)
 
 # 挂载静态文件目录
 app.mount("/static", StaticFiles(directory="Static"), name="static")
 
 
-
 class QueryRequest(BaseModel):
     query: str = Field(..., description="用户查询的问题")
     documents: List[str] = Field(..., description="用户上传的文档")
@@ -237,11 +238,14 @@ async def rag_stream(request: Request):
             # 调用阿里云大模型
             if len(context) > 0:
                 # 调用大模型生成回答
+                logger.info("正在调用阿里云大模型生成回答...")
                 html_content = aliyun_util.chat(prompt)
+                logger.info(f"调用阿里云大模型生成回答成功完成!")
                 return {"data": html_content}
             else:
                 logger.warning(f"未找到查询'{query}'的相关数据，tags: {query_tags}")
-            return {"data": "没有在知识库中找到相关的信息，无法回答此问题。", "debug": {"query": query, "tags": query_tags}}
+            return {"data": "没有在知识库中找到相关的信息，无法回答此问题。",
+                    "debug": {"query": query, "tags": query_tags}}
         except Exception as e:
             return {"data": f"生成报告时出错: {str(e)}"}
 
@@ -254,6 +258,5 @@ async def rag_stream(request: Request):
         raise HTTPException(status_code=500, detail=str(e))
 
 
-
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=8000)