universal-mcp-agents 0.1.22__py3-none-any.whl → 0.1.23__py3-none-any.whl

universal_mcp/agents/__init__.py

@@ -9,22 +9,16 @@ from universal_mcp.agents.simple import SimpleAgent
 
 
 def get_agent(
-    agent_name: Literal["react", "simple", "builder", "bigtool", "codeact-script", "codeact-repl"],
+    agent_name: Literal["react", "simple", "builder", "bigtool", "codeact-repl"],
 ):
     if agent_name == "react":
         return ReactAgent
     elif agent_name == "simple":
         return SimpleAgent
-    elif agent_name == "builder":
-        return BuilderAgent
-    elif agent_name == "bigtool":
-        return BigToolAgent
     elif agent_name == "codeact-repl":
         return CodeActPlaybookAgent
     else:
-        raise ValueError(
-            f"Unknown agent: {agent_name}. Possible values:  react, simple, builder, bigtool, codeact-script, codeact-repl"
-        )
+        raise ValueError(f"Unknown agent: {agent_name}. Possible values:  react, simple, codeact-repl")
 
 
 __all__ = [

universal_mcp/agents/base.py

@@ -1,4 +1,4 @@
-# agents/base.py
+import asyncio
 from typing import cast
 from uuid import uuid4
 
@@ -57,40 +57,50 @@ class BaseAgent:
         }
 
         last_ai_chunk = None
-        async for event, meta in self._graph.astream(
-            {"messages": [{"role": "user", "content": user_input}]},
-            config=run_config,
-            context={"system_prompt": self.instructions, "model": self.model},
-            stream_mode=["messages", "custom"],
-            stream_usage=True,
-        ):
-            if event == "messages" and isinstance(meta, (tuple, list)) and len(meta) == 2:
-                payload, meta_dict = meta
-                is_playbook = isinstance(meta_dict, dict) and meta_dict.get("langgraph_node") == "playbook"
-                additional_kwargs = getattr(payload, "additional_kwargs", {}) or {}
-                if is_playbook and not additional_kwargs.get("stream"):
-                    continue
-                if isinstance(payload, AIMessageChunk):
-                    last_ai_chunk = payload
-                    aggregate = payload if aggregate is None else aggregate + payload
-                if "finish_reason" in payload.response_metadata:
-                    logger.debug(
-                        f"Finish event: {payload}, reason: {payload.response_metadata['finish_reason']}, Metadata: {meta_dict}"
+        try:
+            async for event, meta in self._graph.astream(
+                {"messages": [{"role": "user", "content": user_input}]},
+                config=run_config,
+                context={"system_prompt": self.instructions, "model": self.model},
+                stream_mode=["messages", "custom"],
+                stream_usage=True,
+            ):
+                if event == "messages" and isinstance(meta, (tuple, list)) and len(meta) == 2:  # noqa: PLR2004
+                    payload, meta_dict = meta
+                    if meta_dict.get("tags") and "quiet" in meta_dict.get("tags"):
+                        continue
+                    is_agent_builder = (
+                        isinstance(meta_dict, dict) and meta_dict.get("langgraph_node") == "agent_builder"
                     )
-                    pass
-                logger.debug(f"Event: {payload}, Metadata: {meta_dict}")
-                yield payload
-
-            if event == "custom":
-                yield meta
-
-        # Send a final finished message if we saw any AI chunks (to carry usage)
-        if last_ai_chunk is not None and aggregate is not None:
-            event = cast(AIMessageChunk, last_ai_chunk)
-            event.usage_metadata = aggregate.usage_metadata
-            logger.debug(f"Usage metadata: {event.usage_metadata}")
-            event.content = ""  # Clear the message since it would have already been streamed above
-            yield event
+                    additional_kwargs = getattr(payload, "additional_kwargs", {}) or {}
+                    if is_agent_builder and not additional_kwargs.get("stream"):
+                        continue
+                    if isinstance(payload, AIMessageChunk):
+                        last_ai_chunk = payload
+                        aggregate = payload if aggregate is None else aggregate + payload
+                    if "finish_reason" in payload.response_metadata:
+                        logger.debug(
+                            f"Finish event: {payload}, reason: {payload.response_metadata['finish_reason']}, Metadata: {meta_dict}"
+                        )
+                        pass
+                    logger.debug(f"Event: {payload}, Metadata: {meta_dict}")
+                    yield payload
+
+                if event == "custom":
+                    yield meta
+
+        except asyncio.CancelledError:
+            logger.info(f"Stream for thread_id {thread_id} was cancelled by the user.")
+            # Perform any cleanup here if necessary
+        finally:
+            # This block will run whether the stream finished normally or was cancelled
+            # Send a final finished message if we saw any AI chunks (to carry usage)
+            if last_ai_chunk is not None and aggregate is not None:
+                event = cast(AIMessageChunk, last_ai_chunk)
+                event.usage_metadata = aggregate.usage_metadata
+                logger.debug(f"Usage metadata: {event.usage_metadata}")
+                event.content = ""  # Clear the message
+                yield event
 
     async def stream_interactive(self, thread_id: str, user_input: str):
         await self.ainit()

universal_mcp/agents/bigtool/state.py

@@ -1,6 +1,6 @@
 from typing import Annotated
 
-from langgraph.prebuilt.chat_agent_executor import AgentState
+from langchain.agents import AgentState
 
 
 def _enqueue(left: list, right: list) -> list:

universal_mcp/agents/cli.py

@@ -18,14 +18,14 @@ app = Typer()
     mcp client run --config client_config.json
     """,
 )
-def run(name: str = "react"):
+def run(name: str = "codeact-repl"):
     """Run the agent CLI"""
 
     setup_logger(log_file=None, level="ERROR")
     client = AgentrClient()
     params = {
         "instructions": "You are a helpful assistant",
-        "model": "anthropic/claude-sonnet-4-20250514",
+        "model": "anthropic:claude-4-sonnet-20250514",
         "registry": AgentrRegistry(client=client),
         "memory": MemorySaver(),
     }

universal_mcp/agents/codeact0/__main__.py

@@ -13,15 +13,12 @@ async def main():
     agent = CodeActPlaybookAgent(
         name="CodeAct Agent",
         instructions="Be very concise in your answers.",
-        model="anthropic:claude-4-sonnet-20250514",
-        tools={"google_mail": ["list_messages"]},
+        model="azure/gpt-4.1",
         registry=AgentrRegistry(),
         memory=memory,
     )
     print("Starting agent...")
-    result = await agent.invoke(
-        user_input="Fetch unsubscribe links from my Gmail inbox for promo emails I have received in the last 7 days"
-    )
+    result = await agent.invoke(user_input="Check my google calendar and show my todays agenda")
     print(messages_to_list(result["messages"]))