refactor: 移除 agent 批量重试逻辑中的多余 try 块并优化缩进

This commit is contained in:
jxxghp
2026-04-10 15:03:49 +08:00
parent 8cd12ab584
commit bf12a8679d
+4 -3
View File
@@ -70,7 +70,7 @@ class _ThinkTagStripper:
on_output(self.buffer[:start_idx]) on_output(self.buffer[:start_idx])
emitted = True emitted = True
self.in_think_tag = True self.in_think_tag = True
self.buffer = self.buffer[start_idx + 7 :] self.buffer = self.buffer[start_idx + 7:]
else: else:
# 检查是否以 <think> 的不完整前缀结尾 # 检查是否以 <think> 的不完整前缀结尾
partial_match = False partial_match = False
@@ -90,7 +90,7 @@ class _ThinkTagStripper:
end_idx = self.buffer.find("</think>") end_idx = self.buffer.find("</think>")
if end_idx != -1: if end_idx != -1:
self.in_think_tag = False self.in_think_tag = False
self.buffer = self.buffer[end_idx + 8 :] self.buffer = self.buffer[end_idx + 8:]
else: else:
# 检查是否以 </think> 的不完整前缀结尾 # 检查是否以 </think> 的不完整前缀结尾
partial_match = False partial_match = False
@@ -875,7 +875,6 @@ class AgentManager:
if not history_ids: if not history_ids:
return return
try:
session_id = f"__agent_retry_transfer_batch_{uuid.uuid4().hex[:8]}__" session_id = f"__agent_retry_transfer_batch_{uuid.uuid4().hex[:8]}__"
user_id = "system" user_id = "system"
@@ -935,6 +934,8 @@ class AgentManager:
f"Respond in Chinese (中文)." f"Respond in Chinese (中文)."
) )
try:
await self.process_message( await self.process_message(
session_id=session_id, session_id=session_id,
user_id=user_id, user_id=user_id,