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

style(lint): 优化代码格式和错误处理

Browse Source
This commit is contained in:
Wenjie Zhang 2025-12-19 21:24:19 +08:00
parent 558d26853f
commit 8d97a33ea4
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/routers/chat_router.py
View File

@ -581,7 +581,8 @@ async def chat_agent(
async for msg, metadata in agent.stream_messages(messages, input_context=input_context): async for msg, metadata in agent.stream_messages(messages, input_context=input_context):
if isinstance(msg, AIMessageChunk): if isinstance(msg, AIMessageChunk):
full_msg = msg if not full_msg else full_msg + msg full_msg = msg if not full_msg else full_msg + msg
if conf.enable_content_guard and await content_guard.check_with_keywords(full_msg.content[-20:]): content_for_check = full_msg.content[-20:]
if conf.enable_content_guard and await content_guard.check_with_keywords(content_for_check):
logger.warning("Sensitive content detected in stream") logger.warning("Sensitive content detected in stream")
await save_partial_message(conv_manager, thread_id, full_msg, "content_guard_blocked") await save_partial_message(conv_manager, thread_id, full_msg, "content_guard_blocked")
meta["time_cost"] = asyncio.get_event_loop().time() - start_time meta["time_cost"] = asyncio.get_event_loop().time() - start_time
@ -601,7 +602,8 @@ async def chat_agent(
agent_state = _extract_agent_state(getattr(state, "values", {})) if state else {} agent_state = _extract_agent_state(getattr(state, "values", {})) if state else {}
if agent_state: if agent_state:
yield make_chunk(status="agent_state", agent_state=agent_state, meta=meta) yield make_chunk(status="agent_state", agent_state=agent_state, meta=meta)
except Exception: except Exception as e:
logger.error(f"Error processing tool message: {e}")
pass pass
if ( if (

1
src/knowledge/adapters/lightrag.py
View File

@ -1,6 +1,7 @@
from typing import Any from typing import Any
from src.utils import logger from src.utils import logger
from .base import GraphAdapter from .base import GraphAdapter