fix

Author: braisedpork1964

lagent/actions/mcp_client.py

@@ -42,9 +42,6 @@ def _start_loop(loop):
     return event_loop
 
 
-logger = logging.getLogger(__file__)
-
-
 class TokenBucket:
     def __init__(self, rate_limit: float):
         self.rate_limit = rate_limit  # tokens per second
@@ -404,7 +401,8 @@ async def run(self, **kwargs) -> ActionReturn:
             # 否则直接作为字符串返回
             try:
                 result = self._parser.parse_outputs(outputs)
-            except:
+            except Exception as exc:
+                logger.warning(f"Failed to parse MCP Action {self.name} output: {exc}")
                 result = str(outputs)
 
             action_return = ActionReturn(fallback_args, type=self.name, result=result)

lagent/actions/web_visitor.py

@@ -70,7 +70,7 @@ class WebVisitor(AsyncActionMixin, BaseAction):
 2. **Key Extraction for Evidence**: Identify and extract the **most relevant information** from the content, you never miss any important information, output the **full original context** of the content as far as possible, it can be more than three paragraphs.
 3. **Summary Output for Summary**: Organize into a concise paragraph with logical flow, prioritizing clarity and judge the contribution of the information to the goal.
 
-**Final Output Format using JSON format has "rational", "evidence", "summary" feilds**
+**Final Output Format using JSON format has "rational", "evidence", "summary" fields**
 """
 
     def __init__(

lagent/agents/fc_agent.py

@@ -78,10 +78,8 @@ def __init__(
     async def forward(self, env_message: AgentMessage, session_id: str | int, **kwargs):
         selection_message: AgentMessage = None
         current_turn = 0
-        while (
-            self.finish_condition is None
-            or not self.finish_condition(selection_message, env_message)
-            and (self.max_turn is None or current_turn < self.max_turn)
+        while (self.finish_condition is None or not self.finish_condition(selection_message, env_message)) and (
+            self.max_turn is None or current_turn < self.max_turn
         ):
             selection_message = await self.select_agent(env_message, session_id=session_id, **kwargs)
             if selection_message.stream_state == AgentStatusCode.SERVER_ERR:
@@ -90,13 +88,15 @@ async def forward(self, env_message: AgentMessage, session_id: str | int, **kwar
                 for _ in range(2):  # remove the last two messages
                     self.select_agent.memory.get(session_id).delete(-1)
                 return AgentMessage(
-                    role='env', content='Exceeded context length limit', finish_reason=selection_message.finish_reason
+                    sender=self.name,
+                    content='Exceeded context length limit',
+                    finish_reason=selection_message.finish_reason,
                 )
             if selection_message.finish_reason == 'abort':
-                return AgentMessage(role='env', content='Aborted request', finish_reason='abort')
+                return AgentMessage(sender=self.name, content='Aborted request', finish_reason='abort')
             env_message = await self.env_agent(selection_message, session_id=session_id)
             current_turn += 1
-        return AgentMessage(role="env", content="Finished", finish_reason='stop')
+        return AgentMessage(sender=self.name, content="Finished", finish_reason='stop')
 
 
 class EnvAgent(AsyncAgent):

lagent/memory/base_memory.py

@@ -38,7 +38,7 @@ def delete(self, index: Union[List[int], int]) -> None:
             for i in sorted(index, reverse=True):
                 del self.memory[i]
 
-    def load(self, memories: Union[str, dict, List], overwrite: bool = True) -> None:
+    def load(self, memories: Union[dict, List], overwrite: bool = True) -> None:
         if overwrite:
             self.memory = []
         if isinstance(memories, dict):

lagent/schema.py

@@ -96,7 +96,7 @@ class AgentMessage(BaseModel):
     tool_calls: Optional[List[dict]] = None
     tool_calls_ids: Optional[List[str]] = None
     formatted: Optional[Any] = None
-    extra_info: dict = {}
+    extra_info: dict = Field(default_factory=dict)
     type: Optional[str] = None
     receiver: Optional[str] = None
     stream_state: Union[ModelStatusCode, AgentStatusCode] = AgentStatusCode.END

requirements/runtime.txt

@@ -12,7 +12,9 @@ jsonschema
 jupyter==1.0.0
 jupyter_client==8.6.2
 jupyter_core==5.7.2
+openai
 pydantic
+ray
 requests
 tenacity
 termcolor