kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

添加日志显示

Browse Source
This commit is contained in:
Wenjie Zhang 2025-03-18 04:58:20 +08:00
parent 8e9fc6521d
commit 5c59741848
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/routers/chat_router.py
View File

@ -51,7 +51,7 @@ def chat_post(
try: try:
modified_query, refs = retriever(modified_query, history_manager.messages, meta) modified_query, refs = retriever(modified_query, history_manager.messages, meta)
except Exception as e: except Exception as e:
logger.error(f"Retriever error: {e}") logger.error(f"Retriever error: {e}, {traceback.format_exc()}")
yield make_chunk(message=f"Retriever error: {e}", status="error") yield make_chunk(message=f"Retriever error: {e}", status="error")
return return
@ -106,6 +106,7 @@ async def call(query: str = Body(...), meta: dict = Body(None)):
@chat.post("/call_lite") @chat.post("/call_lite")
async def call(query: str = Body(...), meta: dict = Body(None)): async def call(query: str = Body(...), meta: dict = Body(None)):
meta = meta or {}
async def predict_async(query): async def predict_async(query):
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
model_provider = meta.get("model_provider", config.model_provider_lite) model_provider = meta.get("model_provider", config.model_provider_lite)