universal-mcp-agents 0.1.10__tar.gz → 0.1.12__tar.gz

{universal_mcp_agents-0.1.10 → universal_mcp_agents-0.1.12}/GEMINI.md

@@ -36,6 +36,7 @@ Usage in this repo
 - When adding libraries or changing versions, propose `uv add ...` changes that update both `pyproject.toml` and `uv.lock`, then run `uv run pytest -q` to validate
 - Prefer minimal diffs, explain the plan, apply changes, and run tests/tooling via `uv run`
 - If build/test fails, inspect error context, adjust constraints or code, and re-run via `uv run`
+- After every file change, run `uv run ruff check .` to confirm the changes and ensure code quality.
 
 Common commands (copy/paste)
 - Initialize: `uv init`  |  Install deps: `uv sync`
@@ -43,4 +44,8 @@ Common commands (copy/paste)
 - Remove: `uv remove <pkg>`
 - Run app: `uv run python -m <your_module>` or `uv run main.py`
 - Tests: `uv run pytest -q`
-- Lint/format: `uv run ruff check .` and/or `uv run ruff format .`
+- Lint/format: `uv run ruff check .` and/or `uv run ruff format .`
+
+
+
+NEVER commit or push changes without asking explicitly.

{universal_mcp_agents-0.1.10 → universal_mcp_agents-0.1.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: universal-mcp-agents
-Version: 0.1.10
+Version: 0.1.12
 Summary: Add your description here
 Project-URL: Homepage, https://github.com/universal-mcp/applications
 Project-URL: Repository, https://github.com/universal-mcp/applications

universal_mcp_agents-0.1.12/dataset_code.py

@@ -0,0 +1,83 @@
+import asyncio
+
+from dotenv import load_dotenv
+from langsmith import Client
+
+from universal_mcp.agentr.registry import AgentrRegistry
+from universal_mcp.agents.codeact0 import CodeActAgent
+
+load_dotenv()
+
+client = Client()
+
+async def create_examples(user_input: str, tools_list: list[str]):
+    """Run the agent and create a LangSmith dataset example"""
+
+    # Create/get dataset
+    dataset_name = "codeagent-tests"
+    try:
+        dataset = client.create_dataset(
+            dataset_name,
+            description="Dataset for the codeagent"
+        )
+    except Exception:
+        dataset = client.read_dataset(dataset_name=dataset_name)
+
+
+    # Define the input
+    # user_input = "Send an email to Manoj from my google mail account, manoj@agentr.dev, with the subject 'Hello from auto agent' and the body 'testing'"
+
+    # Capture initial state
+    initial_state = {
+        "messages": [{"role": "user", "content": user_input}],
+        "tools": tools_list
+    }
+    #result = await agent.ainvoke(initial_state, context={"model": "anthropic/claude-4-sonnet-20250514", "system_time": system_time})
+
+    # Extract the final state from the result
+    # Note: Adjust these based on your actual result structure
+
+    # Create the dataset example with actual results
+    example = client.create_example(
+        inputs=initial_state,
+        outputs=None,
+        dataset_id=dataset.id
+    )
+
+    print(f"✅ Created dataset example with ID: {example.id}")
+    print(f"Dataset: {dataset_name}")
+    print(f"Input: {user_input}")
+
+    return example
+
+import yaml
+import os
+
+if __name__ == "__main__":
+    usecases_dir = os.path.join("src", "universal_mcp", "agents", "codeact0", "usecases")
+    async def main():
+        for name in sorted(os.listdir(usecases_dir)):
+            if not name.endswith(".yaml"):
+                continue
+            path = os.path.join(usecases_dir, name)
+            with open(path, encoding="utf-8") as f:
+                content = f.read()
+            data = yaml.safe_load(content) or {}
+            base_prompt = data.get("base_prompt")
+            tools = data.get("tools")
+            if not base_prompt:
+                continue
+            # Normalize tools to a flat list[str]
+            tools_list: list[str] = []
+            if isinstance(tools, list):
+                tools_list = tools
+            elif isinstance(tools, dict):
+                for v in tools.values():
+                    if isinstance(v, list):
+                        tools_list.extend(v)
+                    elif isinstance(v, str):
+                        tools_list.append(v)
+            print(f"Creating example for {name} with {len(tools_list)} tools…")
+            await create_examples(base_prompt, tools_list)
+
+    asyncio.run(main())

{universal_mcp_agents-0.1.10 → universal_mcp_agents-0.1.12}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "universal-mcp-agents"
-version = "0.1.10"
+version = "0.1.12"
 description = "Add your description here"
 readme = "README.md"
 authors = [
@@ -79,3 +79,8 @@ pythonpath = [
 asyncio_mode = "strict"
 asyncio_default_fixture_loop_scope = "module"
 
+[dependency-groups]
+dev = [
+    "ruff>=0.13.0",
+]
+

{universal_mcp_agents-0.1.10 → universal_mcp_agents-0.1.12}/src/tests/test_agents.py

@@ -8,7 +8,7 @@ from universal_mcp.types import ToolFormat
 
 from universal_mcp.agents import get_agent
 from universal_mcp.agents.base import BaseAgent
-from universal_mcp.agents.builder import BuilderAgent
+from universal_mcp.agents.builder.builder import BuilderAgent
 from universal_mcp.agents.llm import load_chat_model
 from universal_mcp.agents.shared.tool_node import build_tool_node_graph
 
@@ -284,9 +284,9 @@ class TestToolFinderGraph:
         "react",
         "simple",
         "builder",
-        "planner",
-        "bigtoolcache",
-        "bigtool2",
+        "bigtool",
+        # "codeact-script",
+        # "codeact-repl",
     ],
 )
 class TestAgents:
@@ -312,7 +312,7 @@ class TestAgents:
         await agent.ainit()
         # Invoke the agent graph to get the final state
         final_state = await agent.invoke(
-            task,
+            user_input=task,
             thread_id=thread_id,
         )
 
@@ -335,7 +335,7 @@ class TestAgentBuilder:
         agent = BuilderAgent(
             name="Test Builder Agent",
             instructions="Test instructions for builder",
-            model="gemini/gemini-1.5-flash",
+            model="gemini/gemini-2.5-flash",
             registry=registry,
         )
         yield agent

universal_mcp_agents-0.1.12/src/universal_mcp/agents/__init__.py

@@ -0,0 +1,39 @@
+from typing import Literal
+
+from universal_mcp.agents.base import BaseAgent
+from universal_mcp.agents.bigtool import BigToolAgent
+from universal_mcp.agents.builder.builder import BuilderAgent
+from universal_mcp.agents.codeact0 import CodeActAgent as CodeActRepl
+from universal_mcp.agents.codeact import CodeActAgent as CodeActScript
+from universal_mcp.agents.react import ReactAgent
+from universal_mcp.agents.simple import SimpleAgent
+
+
+def get_agent(agent_name: Literal["react", "simple", "builder", "bigtool", "codeact-script", "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-script":
+        return CodeActScript
+    elif agent_name == "codeact-repl":
+        return CodeActRepl
+    else:
+        raise ValueError(
+            f"Unknown agent: {agent_name}. Possible values:  react, simple, builder, bigtool, codeact-script, codeact-repl"
+        )
+
+
+__all__ = [
+    "BaseAgent",
+    "ReactAgent",
+    "SimpleAgent",
+    "BuilderAgent",
+    "BigToolAgent",
+    "CodeActScript",
+    "CodeActRepl",
+]

{universal_mcp_agents-0.1.10 → universal_mcp_agents-0.1.12}/src/universal_mcp/agents/base.py

@@ -2,7 +2,7 @@
 from typing import cast
 from uuid import uuid4
 
-from langchain_core.messages import AIMessage, AIMessageChunk
+from langchain_core.messages import AIMessageChunk
 from langgraph.checkpoint.base import BaseCheckpointSaver
 from langgraph.graph import StateGraph
 from langgraph.types import Command
@@ -90,7 +90,7 @@ class BaseAgent:
     async def stream_interactive(self, thread_id: str, user_input: str):
         await self.ainit()
         with self.cli.display_agent_response_streaming(self.name) as stream_updater:
-            async for event in self.stream(thread_id, user_input):
+            async for event in self.stream(thread_id=thread_id, user_input=user_input):
                 if isinstance(event.content, list):
                     thinking_content = "".join([c.get("thinking", "") for c in event.content])
                     stream_updater.update(thinking_content, type_="thinking")
@@ -112,6 +112,7 @@ class BaseAgent:
             run_metadata.update(metadata)
 
         run_config = {
+            "recursion_limit": 25,
             "configurable": {"thread_id": thread_id},
             "metadata": run_metadata,
         }
@@ -123,6 +124,11 @@ class BaseAgent:
         )
         return result
 
+    async def get_state(self, thread_id: str):
+        await self.ainit()
+        state = await self._graph.aget_state(config={"configurable": {"thread_id": thread_id}})
+        return state
+
     async def run_interactive(self, thread_id: str = str(uuid4())):
         """Main application loop"""
 
@@ -133,7 +139,7 @@ class BaseAgent:
         # Main loop
         while True:
             try:
-                state = self._graph.get_state(config={"configurable": {"thread_id": thread_id}})
+                state = await self.get_state(thread_id=thread_id)
                 if state.interrupts:
                     value = self.cli.handle_interrupt(state.interrupts[0])
                     self._graph.invoke(
@@ -168,14 +174,11 @@ class BaseAgent:
                         continue
 
                 # Process with agent
-                await self.stream_interactive(thread_id, user_input)
+                await self.stream_interactive(thread_id=thread_id, user_input=user_input)
 
             except KeyboardInterrupt:
                 self.cli.display_info("\nGoodbye! 👋")
                 break
             except Exception as e:
-                import traceback
-
-                traceback.print_exc()
                 self.cli.display_error(f"An error occurred: {str(e)}")
                 break

{universal_mcp_agents-0.1.10/src/universal_mcp/agents/bigtoolcache → universal_mcp_agents-0.1.12/src/universal_mcp/agents/bigtool}/__init__.py

@@ -13,7 +13,7 @@ from .prompts import SYSTEM_PROMPT
 from .tools import create_meta_tools
 
 
-class BigToolAgentCache(BaseAgent):
+class BigToolAgent(BaseAgent):
     def __init__(
         self,
         registry: ToolRegistry,
@@ -63,4 +63,4 @@ class BigToolAgentCache(BaseAgent):
         return self._graph
 
 
-__all__ = ["BigToolAgentCache"]
+__all__ = ["BigToolAgent"]

{universal_mcp_agents-0.1.10/src/universal_mcp/agents/bigtoolcache → universal_mcp_agents-0.1.12/src/universal_mcp/agents/bigtool}/__main__.py

@@ -2,7 +2,6 @@ import asyncio
 
 from loguru import logger
 from universal_mcp.agentr.registry import AgentrRegistry
-
 from universal_mcp.agents.bigtoolcache import BigToolAgentCache
 
 

{universal_mcp_agents-0.1.10/src/universal_mcp/agents/bigtoolcache → universal_mcp_agents-0.1.12/src/universal_mcp/agents/bigtool}/agent.py

@@ -1,5 +1,4 @@
 from universal_mcp.agentr.registry import AgentrRegistry
-
 from universal_mcp.agents.bigtoolcache import BigToolAgentCache
 
 

{universal_mcp_agents-0.1.10/src/universal_mcp/agents/bigtoolcache → universal_mcp_agents-0.1.12/src/universal_mcp/agents/bigtool}/graph.py

@@ -8,7 +8,6 @@ from langchain_core.messages import AIMessage, SystemMessage, ToolMessage
 from langchain_core.tools import BaseTool
 from langgraph.graph import StateGraph
 from langgraph.types import Command
-from loguru import logger
 from universal_mcp.tools.registry import ToolRegistry
 from universal_mcp.types import ToolFormat
 
@@ -35,7 +34,11 @@ def build_graph(
             current_tools = await registry.export_tools(tools=state["selected_tool_ids"], format=ToolFormat.LANGCHAIN)
         else:
             current_tools = []
-        all_tools = [meta_tools["search_tools"], meta_tools["load_tools"], meta_tools.get("web_search")] + default_tools + current_tools
+        all_tools = (
+            [meta_tools["search_tools"], meta_tools["load_tools"], meta_tools.get("web_search")]
+            + default_tools
+            + current_tools
+        )
 
         # Remove duplicates based on tool name
         seen_names = set()
@@ -81,7 +84,7 @@ def build_graph(
                 valid_tools = await get_valid_tools(tool_ids=tool_call["args"]["tool_ids"], registry=registry)
                 new_tool_ids.extend(valid_tools)
                 # Create tool message response
-                tool_result=f"Successfully loaded {len(valid_tools)} tools: {valid_tools}"
+                tool_result = f"Successfully loaded {len(valid_tools)} tools: {valid_tools}"
             elif tool_call["name"] == "search_tools":
                 tool_result = await meta_tools["search_tools"].ainvoke(tool_call["args"])
             elif tool_call["name"] == "web_search":
@@ -99,8 +102,6 @@ def build_graph(
 
         return Command(goto="agent", update={"messages": tool_messages, "selected_tool_ids": new_tool_ids})
 
-
-
     # Define the graph
     workflow = StateGraph(State)
 

universal_mcp_agents-0.1.12/src/universal_mcp/agents/builder/__main__.py

@@ -0,0 +1,125 @@
+import asyncio
+import json
+from uuid import uuid4
+
+from langgraph.checkpoint.memory import MemorySaver
+from loguru import logger
+from universal_mcp.agentr.registry import AgentrRegistry
+
+from universal_mcp.agents.builder.builder import BuilderAgent
+
+
+async def run_interactive_build():
+    """Simulates a multi-turn conversation to build and then modify an agent."""
+    logger.info("--- SCENARIO 1: INTERACTIVE AGENT BUILD & MODIFY ---")
+
+    registry = AgentrRegistry()
+    memory = MemorySaver()
+    agent = BuilderAgent(
+        name="Builder Agent",
+        instructions="You are a builder agent that creates other agents.",
+        model="anthropic/claude-4-sonnet-20250514",
+        registry=registry,
+        memory=memory,
+    )
+
+    thread_id = str(uuid4())
+
+    conversation_script = [
+        "Send an email to manoj@agentr.dev with the subject 'Hello' and body 'This is a test of the Gmail agent.' from my Gmail account.",
+        "Use outlook instead of gmail",
+    ]
+
+    final_result = {}
+    for i, user_input in enumerate(conversation_script):
+        logger.info(f"\n--- Conversation Turn {i + 1} ---")
+        logger.info(f"User Request: '{user_input}'")
+
+        result = await agent.invoke(user_input=user_input, thread_id=thread_id)
+        final_result.update(result)  # Keep updating the final result
+
+        generated_agent = final_result.get("generated_agent")
+        tool_config = final_result.get("tool_config")
+
+        if generated_agent:
+            logger.info("--- Generated/Modified Agent ---")
+            logger.info(f"Name: {generated_agent.name}")
+            logger.info(f"Description: {generated_agent.description}")
+            logger.info(f"Expertise: {generated_agent.expertise}")
+            logger.info(f"Instructions:\n{generated_agent.instructions}")
+
+        if tool_config:
+            logger.info("--- Selected Tools ---")
+            tools_str = "\n".join(f"- {app}: {', '.join(tool_ids)}" for app, tool_ids in tool_config.items())
+            logger.info(tools_str)
+        else:
+            logger.info("--- Selected Tools ---")
+            logger.info("No tools selected for this agent yet.")
+
+
+async def run_conversation_build():
+    """Simulates a one-shot agent build from a conversation history payload."""
+    logger.info("\n\n--- SCENARIO 2: AGENT BUILD FROM CONVERSATION HISTORY ---")
+
+    registry = AgentrRegistry()
+    agent = BuilderAgent(
+        name="Builder Agent",
+        instructions="You build agents from conversation transcripts.",
+        model="anthropic/claude-4-sonnet-20250514",
+        registry=registry,
+    )
+
+    sample_conversation_history = [
+        {
+            "type": "human",
+            "content": "Hey, can you look at our main branch on the universal-mcp repo and tell me what the last 3 pull requests were?",
+        },
+        {
+            "type": "ai",
+            "content": "Of course. The last 3 pull requests are: #101 'Fix login bug', #102 'Update documentation', and #103 'Add new chart component'.",
+        },
+        {
+            "type": "human",
+            "content": "Awesome, thanks. Now can you draft a new Google Doc and put that list in there for me?",
+        },
+        {"type": "ai", "content": "Done. I have created a new Google Doc with the list of the last 3 pull requests."},
+    ]
+    sample_tool_config = {"github": ["get_pull_requests"], "google_docs": ["create_document"]}
+    wingman_payload = {"conversation_history": sample_conversation_history, "tool_config": sample_tool_config}
+
+    logger.info(f"Payload Conversation History Length: {len(sample_conversation_history)} messages")
+    logger.info(f"Payload Tools Provided: {list(sample_tool_config.keys())}")
+
+    # The payload must be passed as a JSON string in the 'user_input'
+    payload_str = json.dumps(wingman_payload)
+    thread_id = str(uuid4())
+    result = await agent.invoke(user_input=payload_str, thread_id=thread_id)
+
+    generated_agent = result.get("generated_agent")
+    tool_config = result.get("tool_config")
+
+    if generated_agent:
+        logger.info("\n--- Generated Agent Profile ---")
+        logger.info(f"Name: {generated_agent.name}")
+        logger.info(f"Description: {generated_agent.description}")
+        logger.info(f"Expertise: {generated_agent.expertise}")
+        logger.info(f"Instructions:\n{generated_agent.instructions}")
+        logger.info(f"Schedule: {generated_agent.schedule}")
+    else:
+        logger.error("Error: Agent profile was not generated.")
+
+    if tool_config:
+        logger.info("--- Final Tool Configuration ---")
+        tools_str = "\n".join(f"- {app}: {', '.join(tool_ids)}" for app, tool_ids in tool_config.items())
+        logger.info(tools_str)
+    else:
+        logger.error("Error: Tool configuration is missing.")
+
+
+async def main():
+    await run_interactive_build()
+    await run_conversation_build()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())