universal-mcp-agents 0.1.5__tar.gz → 0.1.6__tar.gz

{universal_mcp_agents-0.1.5 → universal_mcp_agents-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: universal-mcp-agents
-Version: 0.1.5
+Version: 0.1.6
 Summary: Add your description here
 Project-URL: Homepage, https://github.com/universal-mcp/applications
 Project-URL: Repository, https://github.com/universal-mcp/applications
@@ -11,7 +11,6 @@ Requires-Dist: langchain-anthropic>=0.3.19
 Requires-Dist: langchain-google-genai>=2.1.10
 Requires-Dist: langchain-openai>=0.3.32
 Requires-Dist: langgraph>=0.6.6
-Requires-Dist: universal-mcp-applications>=0.1.4
 Requires-Dist: universal-mcp>=0.1.24rc17
 Provides-Extra: dev
 Requires-Dist: pre-commit; extra == 'dev'

universal_mcp_agents-0.1.6/builder_tools.py

@@ -0,0 +1,34 @@
+from universal_mcp.agentr.registry import AgentrRegistry
+from universal_mcp.agents.builder import BuilderAgent
+import json
+
+
+def load_tasks():
+    with open("src/evals/datasets/tasks.jsonl", "r") as f:
+        for line in f:
+            yield json.loads(line)
+
+
+async def main():
+    registry = AgentrRegistry()
+    builder = BuilderAgent(
+        name="Builder Agent",
+        instructions="You are a builder agent that creates other agents.",
+        model="gemini/gemini-1.5-pro",
+        registry=registry,
+    )
+    updated_tasks = []
+    tasks = load_tasks()
+    for task in tasks:
+        print(task["user_input"])
+        result = await builder.invoke(task["user_input"])
+        tools = result["tool_config"] or {}
+        updated_tasks.append({**task, "required_tools": tools})
+    with open("src/evals/datasets/tasks_with_tools.jsonl", "w") as f:
+        for task in updated_tasks:
+            f.write(json.dumps(task) + "\n")
+
+
+if __name__ == "__main__":
+    import asyncio
+    asyncio.run(main())

{universal_mcp_agents-0.1.5 → universal_mcp_agents-0.1.6}/bump_and_release.sh

@@ -9,7 +9,7 @@ uv sync --all-extras
 
 # Run tests with pytest
 echo "Running tests with pytest..."
-# uv run pytest # --cov=src --cov-report=term-missing
+uv run pytest
 
 echo "Tests passed!"
 

{universal_mcp_agents-0.1.5 → universal_mcp_agents-0.1.6}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "universal-mcp-agents"
-version = "0.1.5"
+version = "0.1.6"
 description = "Add your description here"
 readme = "README.md"
 authors = [
@@ -18,8 +18,7 @@ dependencies = [
     "langchain-google-genai>=2.1.10",
     "langchain-openai>=0.3.32",
     "langgraph>=0.6.6",
-    "universal-mcp >= 0.1.24rc17",
-    "universal-mcp-applications>=0.1.4",
+    "universal-mcp>=0.1.24rc17",
 ]
 
 [project.license]

{universal_mcp_agents-0.1.5 → universal_mcp_agents-0.1.6}/src/evals/datasets/tasks.jsonl

@@ -1,22 +1,22 @@
-{"user_input": "Send an email to manoj@agentr.dev from my Gmail account", "difficulty": 1}
-{"user_input": "Show me events from today's Google Calendar.", "difficulty": 1}
-{"user_input": "Create a Google Doc summarizing the last 5 merged pull requests in my GitHub repo- universal-mcp/universal-mcp, including links and commit highlights.", "difficulty": 4}
-{"user_input": "Summarize the key insights from all marketing emails received this week from my Gmail and add a section in a Google Doc with action points.", "difficulty": 4}
-{"user_input": "Create a Google Sheet of the best cafes and restaurants near IIT Bombay", "difficulty": 3}
-{"user_input": "Track the top posts in r/startups over the past 7 days using Reddit and create a trend report on what's being discussed most (e.g., hiring, funding, MVPs) in a Google Doc.", "difficulty": 5}
-{"user_input": "Find the best restaurants in Goa using perplexity web search", "difficulty": 2}
-{"user_input": "List the unread emails from the last 24 hours from my Gmail, sorted by sender.", "difficulty": 2}
-{"user_input": "Tell me how many meetings I have tomorrow and when they start from my Google Calendar.", "difficulty": 1}
-{"user_input": "Create a meeting with aditakarsh@example.com on the topic of the latest trends in AI at 8PM today using Google Calendar.", "difficulty": 2}
-{"user_input": "What are the topics of my meetings today from Google Calendar and who are the attendees? Give a 1-line context for each attendee using LinkedIn or web search.", "difficulty": 4}
-{"user_input": "Fetch my last inbox mail from Microsoft Outlook", "difficulty": 1}
-{"user_input": "Fetch unsubscribe links from my Gmail inbox for promo emails I have received in the last 7 days", "difficulty": 3}
-{"user_input": "Fetch all unread emails from Gmail and new tickets from ClickUp for me from last night", "difficulty": 4}
-{"user_input": "Give me a report on the earnings of Oklo using web search, and projections for the company revenue, stock price", "difficulty": 4}
-{"user_input": "Create a weekly expense report from my credit card transactions and categorize spending by type (food, transport, entertainment, etc.) in a Google Sheet", "difficulty": 3}
-{"user_input": "Generate a comparison table of SaaS tools for project management using web search, including pricing, features, and user ratings in a Google Sheet", "difficulty": 4}
-{"user_input": "Research the top 10 Y Combinator startups from the latest batch using web search and create a report on their industries and funding status in Google Docs", "difficulty": 5}
-{"user_input": "Find and summarize the key takeaways from the latest earnings calls of FAANG companies using web search and create a report in Google Docs", "difficulty": 5}
-{"user_input": "Draft personalized LinkedIn outreach messages for 10 potential collaborators in the fintech space based on their recent posts using LinkedIn data in a Google Sheet", "difficulty": 5}
-{"user_input": "Monitor my Twitter mentions and DMs from the past 48 hours and create a response priority list in Google Sheets", "difficulty": 4}
-{"user_input": "Create a content calendar for next month with trending AI/ML topics using web search and optimal posting times based on my audience analytics in Google Sheets", "difficulty": 5}
+{"user_input": "Send an email to manoj@agentr.dev from my Gmail account", "difficulty": 1, "required_tools": {"google_mail": ["send_email"]}}
+{"user_input": "Show me events from today's Google Calendar.", "difficulty": 1, "required_tools": {"google_calendar": ["list_events"]}}
+{"user_input": "Create a Google Doc summarizing the last 5 merged pull requests in my GitHub repo- universal-mcp/universal-mcp, including links and commit highlights.", "difficulty": 4, "required_tools": {"github": ["get_pull_request"], "google_docs": ["get_document"]}}
+{"user_input": "Summarize the key insights from all marketing emails received this week from my Gmail and add a section in a Google Doc with action points.", "difficulty": 4, "required_tools": {"google_mail": ["create_filters: Set up new Gmail filter with criteria and automated actions"], "google_docs": [], "tavily": []}}
+{"user_input": "Search for best cafes near IIT bombay using exa and make a google sheet out of it", "difficulty": 3, "required_tools": {"exa": ["search"], "google_sheet": ["create_spreadsheet"]}}
+{"user_input": "Track the top posts in r/startups over the past 7 days using Reddit and create a trend report on what's being discussed most (e.g., hiring, funding, MVPs) in a Google Doc.", "difficulty": 5, "required_tools": {"reddit": [], "google_docs": []}}
+{"user_input": "Find the best restaurants in Goa using perplexity web search", "difficulty": 2, "required_tools": {"perplexity": []}}
+{"user_input": "List the unread emails from the last 24 hours from my Gmail, sorted by sender.", "difficulty": 2, "required_tools": {"google_mail": ["list_messages"]}}
+{"user_input": "Tell me how many meetings I have tomorrow and when they start from my Google Calendar.", "difficulty": 1, "required_tools": {"google_calendar": ["get_today_events"]}}
+{"user_input": "Create a meeting with aditakarsh@example.com on the topic of the latest trends in AI at 8PM today using Google Calendar.", "difficulty": 2, "required_tools": {"google_calendar": ["add_an_event"]}}
+{"user_input": "What are the topics of my meetings today from Google Calendar and who are the attendees? Give a 1-line context for each attendee using LinkedIn or web search.", "difficulty": 4, "required_tools": {"google_calendar": ["get_event: Retrieves detailed information about a specific Google Calendar event by its ID"], "linkedin": ["get_your_info"]}}
+{"user_input": "Fetch my last inbox mail from Microsoft Outlook", "difficulty": 1, "required_tools": {"outlook": ["user_get_mail_folder"]}}
+{"user_input": "Fetch unsubscribe links from my Gmail inbox for promo emails I have received in the last 7 days", "difficulty": 3, "required_tools": {"google_mail": []}}
+{"user_input": "Fetch all unread emails from Gmail and new tickets from ClickUp for me from last night", "difficulty": 4, "required_tools": {"google_mail": [], "clickup": []}}
+{"user_input": "Give me a report on the earnings of Oklo using web search, and projections for the company revenue, stock price", "difficulty": 4, "required_tools": {"serpapi": ["search"], "perplexity": [], "e2b": []}}
+{"user_input": "Create a weekly expense report from my credit card transactions and categorize spending by type (food, transport, entertainment, etc.) in a Google Sheet", "difficulty": 3, "required_tools": {}}
+{"user_input": "Generate a comparison table of SaaS tools for project management using web search, including pricing, features, and user ratings in a Google Sheet", "difficulty": 4, "required_tools": {"serpapi": ["search"], "google_sheet": []}}
+{"user_input": "Research the top 10 Y Combinator startups from the latest batch using web search and create a report on their industries and funding status in Google Docs", "difficulty": 5, "required_tools": {"serpapi": [], "google_docs": ["create_document"]}}
+{"user_input": "Find and summarize the key takeaways from the latest earnings calls of FAANG companies using web search and create a report in Google Docs", "difficulty": 5, "required_tools": {"serpapi": ["search"], "google_docs": []}}
+{"user_input": "Draft personalized LinkedIn outreach messages for 10 potential collaborators in the fintech space based on their recent posts using LinkedIn data in a Google Sheet", "difficulty": 5, "required_tools": {"linkedin": [], "scraper": ["linkedin_list_all_posts"], "google_sheet": [], "openai": []}}
+{"user_input": "Monitor my Twitter mentions and DMs from the past 48 hours and create a response priority list in Google Sheets", "difficulty": 4, "required_tools": {"twitter": [], "google_sheet": []}}
+{"user_input": "Create a content calendar for next month with trending AI/ML topics using web search and optimal posting times based on my audience analytics in Google Sheets", "difficulty": 5, "required_tools": {"serpapi": ["search"], "google_sheet": [], "google_calendar": []}}

{universal_mcp_agents-0.1.5 → universal_mcp_agents-0.1.6}/src/evals/evaluators.py

@@ -3,6 +3,7 @@ from agentevals.trajectory.llm import (
     TRAJECTORY_ACCURACY_PROMPT,
     create_trajectory_llm_as_judge,
 )
+from google.ai.generativelanguage_v1beta import ToolConfig
 from langsmith.evaluation import EvaluationResult, run_evaluator
 from langsmith.schemas import Example, Run
 from openevals.llm import create_llm_as_judge
@@ -51,3 +52,26 @@ trajectory_evaluator = create_trajectory_llm_as_judge(
     prompt=TRAJECTORY_ACCURACY_PROMPT,
     model="anthropic:claude-4-sonnet-20250514",
 )
+
+@run_evaluator
+def tool_node_evaluator(run: Run, example: Example | None = None) -> EvaluationResult:
+    """
+    A simple evaluator that checks if the agent used the required tools.
+    """
+    try:
+        if example is None or example.outputs is None or "required_tools" not in example.outputs:
+            return EvaluationResult(key="tool_node", score=0, comment="No required tools provided. Example: " + str(example))
+        required_tools : ToolConfig = example.outputs["required_tools"]
+        agent_response_raw : ToolConfig = run.outputs.get("tool_config", {})
+        # Flatten the tool_configs to a single set of tool_ids
+        required_tool_ids = [f"{app_id}___{tool_id}" for app_id, tools in required_tools.items() for tool_id in tools]
+        agent_tool_ids = [f"{app_id}___{tool_id}" for app_id, tools in agent_response_raw.items() for tool_id in tools]
+        if set(required_tool_ids).issubset(set(agent_tool_ids)):
+            return EvaluationResult(key="tool_node", score=1, comment="Tool usage: " + str(required_tools))
+        else:
+            return EvaluationResult(key="tool_node", score=0, comment="Tool usage: " + str(required_tools))
+    except Exception as e:
+        print(f"Error evaluating tool usage: {str(e)}")
+        print(run.outputs)
+        print(example.outputs)
+        return EvaluationResult(key="tool_node", score=0, comment=f"Error evaluating tool usage: {str(e)}")

{universal_mcp_agents-0.1.5 → universal_mcp_agents-0.1.6}/src/evals/run.py

@@ -12,6 +12,7 @@ from evals.evaluators import (
     correctness_evaluator,
     exact_match_evaluator,
     trajectory_evaluator,
+    tool_node_evaluator,
 )
 from universal_mcp.agents import get_agent
 from universal_mcp.agents.base import BaseAgent
@@ -39,6 +40,7 @@ EVALUATORS: dict[str, Any] = {
     "llm_as_judge": correctness_evaluator,
     "exact_match": exact_match_evaluator,
     "trajectory": trajectory_evaluator,
+    "tool_node": tool_node_evaluator,
 }
 
 
@@ -55,13 +57,16 @@ def get_evaluator(evaluator_name: str) -> RunEvaluator:
 
 
 
-async def agent_runner(agent: BaseAgent, inputs: dict):
+async def agent_runner(agent: BaseAgent, inputs: dict) -> dict:
     """
     Runs the agent and returns a dictionary with the final output.
     """
     result = await agent.invoke(user_input=inputs["user_input"])
     messages = messages_to_list(result["messages"])
-    return {"output": messages}
+    return_result = {"output": messages}
+    if "tool_config" in result:
+        return_result["tool_config"] = result["tool_config"]
+    return return_result
 
 async def main(agent_name: str, dataset_path: str, evaluator_name: str):
     """
@@ -93,9 +98,10 @@ async def main(agent_name: str, dataset_path: str, evaluator_name: str):
         for example in dataset_examples:
             client.create_example(
                 inputs={"user_input": example["user_input"]},
-                outputs={"expected_output": example.get("expected_output")}
-                if "expected_output" in example
-                else None,
+                outputs={
+                    "expected_output": example.get("expected_output", ""), 
+                    "required_tools": example.get("required_tools", {})
+                },
                 dataset_id=dataset.id,
             )
     except Exception:

{universal_mcp_agents-0.1.5 → universal_mcp_agents-0.1.6}/src/universal_mcp/agents/base.py

@@ -73,7 +73,7 @@ class BaseAgent:
             # Ignore intermeddite finish messages
             if "finish_reason" in event.response_metadata:
                 # Got LLM finish reason ignore it
-                logger.debug(f"Finish event: {event}, Metadata: {metadata}")
+                logger.error(f"Finish event: {event}, reason: {event.response_metadata['finish_reason']}, Metadata: {metadata}")
                 pass
             else:
                 logger.debug(f"Event: {event}, Metadata: {metadata}")
@@ -89,7 +89,6 @@ class BaseAgent:
         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):
-
                 if isinstance(event.content, list):
                     thinking_content = "".join([c.get("thinking", "") for c in event.content])
                     stream_updater.update(thinking_content, type_="thinking")

{universal_mcp_agents-0.1.5 → universal_mcp_agents-0.1.6}/src/universal_mcp/agents/bigtool2/__init__.py

@@ -1,10 +1,10 @@
 from langgraph.checkpoint.base import BaseCheckpointSaver
-from universal_mcp.logger import logger
-from universal_mcp.tools.registry import ToolRegistry
 
 from universal_mcp.agents.base import BaseAgent
 from universal_mcp.agents.llm import load_chat_model
-from universal_mcp.agents.utils import initialize_ui_tools
+from universal_mcp.logger import logger
+from universal_mcp.tools.registry import ToolRegistry
+from universal_mcp.types import ToolConfig, ToolFormat
 
 from .graph import build_graph
 from .prompts import SYSTEM_PROMPT
@@ -18,6 +18,7 @@ class BigToolAgent2(BaseAgent):
         model: str,
         registry: ToolRegistry,
         memory: BaseCheckpointSaver | None = None,
+        tools: ToolConfig | None = None,
         **kwargs,
     ):
         super().__init__(name, instructions, model, memory, **kwargs)
@@ -25,7 +26,10 @@ class BigToolAgent2(BaseAgent):
         self.registry = registry
         self.llm = load_chat_model(self.model)
         self.recursion_limit = kwargs.get("recursion_limit", 10)
-        self.ui_tools = initialize_ui_tools()
+        self.tools = tools or {}
+        if "ui" not in self.tools:
+            # self.tools["ui"] = ["create_bar_chart", "create_line_chart", "create_pie_chart", "create_table", "http_get", "http_post", "http_put", "http_delete", "http_patch", "read_file"]
+            self.tools["ui"] = ["create_table"]
 
         logger.info(
             f"BigToolAgent '{self.name}' initialized with model '{self.model}'."
@@ -41,11 +45,12 @@ class BigToolAgent2(BaseAgent):
         """Build the bigtool agent graph using the existing create_agent function."""
         logger.info(f"Building graph for BigToolAgent '{self.name}'...")
         try:
+            default_tools = await self.registry.export_tools(self.tools, ToolFormat.LANGCHAIN)
             graph_builder = build_graph(
                 tool_registry=self.registry,
                 llm=self.llm,
                 system_prompt=self._build_system_message(),
-                ui_tools=self.ui_tools,
+                default_tools=default_tools,
             )
 
             compiled_graph = graph_builder.compile(checkpointer=self.memory)

{universal_mcp_agents-0.1.5 → universal_mcp_agents-0.1.6}/src/universal_mcp/agents/bigtool2/agent.py

@@ -1,5 +1,4 @@
 from universal_mcp.agentr.registry import AgentrRegistry
-
 from universal_mcp.agents.bigtool2 import BigToolAgent2
 
 

{universal_mcp_agents-0.1.5 → universal_mcp_agents-0.1.6}/src/universal_mcp/agents/bigtool2/graph.py

@@ -4,18 +4,18 @@ from typing import Literal, cast
 
 from langchain_core.language_models import BaseChatModel
 from langchain_core.messages import AIMessage, ToolMessage
-from langchain_core.tools import tool
+from langchain_core.tools import BaseTool, tool
 from langgraph.graph import StateGraph
 from langgraph.types import Command
+
+from universal_mcp.agents.bigtool2.state import State
 from universal_mcp.logger import logger
 from universal_mcp.tools.registry import ToolRegistry
 from universal_mcp.types import ToolFormat
 
-from universal_mcp.agents.bigtool2.state import State
-
 
 def build_graph(
-    tool_registry: ToolRegistry, llm: BaseChatModel, system_prompt: str, ui_tools: list
+    tool_registry: ToolRegistry, llm: BaseChatModel, system_prompt: str, default_tools: list[BaseTool]
 ):
     @tool
     async def search_tools(queries: list[str]) -> str:
@@ -36,17 +36,18 @@ def build_graph(
             ]
             app_tools = {}
             for task_query in queries:
-                tools_list = await tool_registry.search_tools(task_query, limit=40)
+                apps_list = await tool_registry.search_apps(task_query, limit=5)
+                tools_list = []
+                for app in apps_list:
+                    tools_list.extend(await tool_registry.search_tools(task_query, limit=5, app_id=app["id"]))
                 tool_candidates = [
                     f"{tool['id']}: {tool['description']}" for tool in tools_list
                 ]
                 for tool in tool_candidates:
                     app = tool.split("__")[0]
                     if app not in app_tools:
-                        if len(app_tools.keys()) >= 10:
-                            break
                         app_tools[app] = []
-                    if len(app_tools[app]) < 3:
+                    if len(app_tools[app]) < 5:
                         app_tools[app].append(tool)
             for app in app_tools:
                 app_status = "connected" if app in connected_apps else "NOT connected"
@@ -64,9 +65,19 @@ def build_graph(
 
     @tool
     async def load_tools(tool_ids: list[str]) -> list[str]:
-        """Load the tools for the given tool ids. Returns the tool ids."""
+        """Load the tools for the given tool ids. Returns the tool ids after loading them. Note that tool ids are the complete tool ids, with both the app name and the tool name separated by double underscore (__). e.g. google_mail__send_email"""
         return tool_ids
 
+    @tool
+    async def web_search(query: str) -> str:
+        """Search the web for the given query. Returns the search results."""
+        tool = await tool_registry.export_tools(
+            ["exa__search"], ToolFormat.LANGCHAIN
+        )
+        response = await tool_registry.call_tool("exa__search", {"query": query, "contents": {"summary": True}})
+        return response
+
+
     async def call_model(
         state: State,
     ) -> Command[Literal["select_tools", "call_tools"]]:
@@ -91,8 +102,17 @@ def build_graph(
 
             model = llm
 
+            tools = [search_tools, load_tools, web_search, *default_tools, *selected_tools]
+            # Remove duplicates based on tool name
+            seen_names = set()
+            unique_tools = []
+            for tool in tools:
+                if tool.name not in seen_names:
+                    seen_names.add(tool.name)
+                    unique_tools.append(tool)
+            tools = unique_tools
             model_with_tools = model.bind_tools(
-                [search_tools, load_tools, *selected_tools, *ui_tools],
+                tools,
                 tool_choice="auto",
             )
             response = cast(AIMessage, await model_with_tools.ainvoke(messages))
@@ -124,6 +144,23 @@ def build_graph(
                         },
                     )
 
+                elif tool_call["name"] == web_search.name:
+                    logger.info(f"Tool '{tool_call['name']}' is a web search tool. Proceeding to call.")
+                    web_search_result = await web_search.ainvoke(input=tool_call["args"])
+                    tool_msg = ToolMessage(
+                        f"Web search result: {web_search_result}", tool_call_id=tool_call["id"]
+                    )
+                    return Command(goto="call_model", update={"messages": [response, tool_msg]})
+
+                elif "ui_tools" in tool_call["name"]:
+                    logger.info(f"Tool '{tool_call['name']}' is a UI tool. Proceeding to call.")
+                    ui_tool_result = await ui_tools_dict[tool_call["name"]].ainvoke(input=tool_call["args"])
+                    tool_msg = ToolMessage(
+                        f"UI tool result: {ui_tool_result}", tool_call_id=tool_call["id"]
+                    )
+                    return Command(goto="call_model", update={"messages": [response, tool_msg]})
+
+
                 elif tool_call["name"] not in state["selected_tool_ids"]:
                     try:
                         await tool_registry.export_tools(

universal_mcp_agents-0.1.6/src/universal_mcp/agents/bigtool2/prompts.py

@@ -0,0 +1,15 @@
+"""Default prompts used by the agent."""
+
+SYSTEM_PROMPT = """You are a helpful AI assistant.
+
+**Core Directives:**
+1.  **Always Use Tools for Tasks:** For any user request that requires an action (e.g., sending an email, searching for information, creating an event, displaying a chart), you MUST use a tool. Do not refuse a task if a tool might exist for it.
+
+2. Check if your existing tools or knowledge can handle the user's request. If they can, use them. If they cannot, you must call the `search_tools` function to find the right tools for the user's request.You must not use the same/similar query multiple times in the list. The list should have multiple queries only if the task has clearly different sub-tasks. If you do not find any specific relevant tools, use the pre-loaded generic tools.
+
+3.  **Load Tools:** After looking at the output of `search_tools`, you MUST call the `load_tools` function to load only the tools you want to use. Provide the full tool ids, not just the app names. Use your judgement to eliminate irrelevant apps that came up just because of semantic similarity. However, sometimes, multiple apps might be relevant for the same task. Prefer connected apps over unconnected apps while breaking a tie. If more than one relevant app (or none of the relevant apps) are connected, you must ask the user to choose the app. In case the user asks you to use an app that is not connected, call the apps tools normally. The tool will return a link for connecting that you should pass on to the user.
+
+4.  **Strictly Follow the Process:** Your only job in your first turn is to analyze the user's request and answer using existing tools/knowledge or `search_tools` with a concise query describing the core task. Do not engage in conversation, or extend the conversation beyond the user's request.
+
+{instructions}
+"""

{universal_mcp_agents-0.1.5 → universal_mcp_agents-0.1.6}/src/universal_mcp/agents/bigtoolcache/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.5 → universal_mcp_agents-0.1.6}/src/universal_mcp/agents/bigtoolcache/graph.py

@@ -1,6 +1,6 @@
 import json
 from datetime import UTC, datetime
-from typing import Literal, TypedDict, cast
+from typing import Any, Literal, TypedDict, cast
 
 from langchain_anthropic import ChatAnthropic
 from langchain_core.language_models import BaseChatModel
@@ -11,12 +11,11 @@ from langgraph.runtime import Runtime
 from langgraph.types import Command
 
 from universal_mcp.agents.bigtoolcache.context import Context
+from universal_mcp.agents.bigtoolcache.prompts import TOOLS_LIST
 from universal_mcp.agents.bigtoolcache.state import State
 from universal_mcp.logger import logger
 from universal_mcp.tools.registry import ToolRegistry
 from universal_mcp.types import ToolFormat
-from universal_mcp.agents.bigtoolcache.prompts import TOOLS_LIST
-
 
 
 class ToolSelectionOutput(TypedDict):
@@ -180,16 +179,15 @@ def build_graph(tool_registry: ToolRegistry, llm: BaseChatModel):
                     content=json.dumps(tool_result),
                     name=tool_id,
                     tool_call_id=tool_call["id"],
-                )
-                recent_tool_ids.append(tool_call["name"])
-            except Exception as e:
-                logger.error(f"Error executing tool '{tool_call['name']}': {e}")
-                outputs.append(
-                    ToolMessage(
-                        content=json.dumps("Error: " + str(e)),
-                        name=tool_call["name"],
-                        tool_call_id=tool_call["id"],
-                    )
+                ))
+            recent_tool_ids.append(tool_call["name"])
+        except Exception as e:
+            logger.error(f"Error executing tool '{tool_call['name']}': {e}")
+            outputs.append(
+                ToolMessage(
+                    content=json.dumps("Error: " + str(e)),
+                    name=tool_call["name"],
+                    tool_call_id=tool_call["id"],
                 )
             )
         return Command(

{universal_mcp_agents-0.1.5 → universal_mcp_agents-0.1.6}/src/universal_mcp/agents/builder.py

@@ -14,6 +14,7 @@ from universal_mcp.types import ToolConfig
 from universal_mcp.agents.base import BaseAgent
 from universal_mcp.agents.llm import load_chat_model
 from universal_mcp.agents.shared.tool_node import build_tool_node_graph
+from universal_mcp.agents.utils import messages_to_list
 
 
 class Agent(BaseModel):
@@ -179,9 +180,13 @@ async def main():
         model="gemini/gemini-1.5-pro",
         registry=registry,
     )
-    await agent.invoke(
+    result = await agent.invoke(
         "Send a daily email to manoj@agentr.dev with daily agenda of the day",
     )
+    from rich import print
+    print(messages_to_list(result["messages"]))
+    print(result["generated_agent"])
+    print(result["tool_config"])
 
 
 if __name__ == "__main__":

{universal_mcp_agents-0.1.5 → universal_mcp_agents-0.1.6}/src/universal_mcp/agents/llm.py

@@ -8,7 +8,7 @@ from langchain_openai import AzureChatOpenAI
 
 @lru_cache(maxsize=8)
 def load_chat_model(
-    fully_specified_name: str, temperature: float = 1.0, tags: list[str] | None = None, thinking: bool = True
+    fully_specified_name: str, temperature: float = 1.0, tags: list[str] | None = None, thinking: bool = False
 ) -> BaseChatModel:
     """Load a chat model from a fully specified name.
     Args:

{universal_mcp_agents-0.1.5 → universal_mcp_agents-0.1.6}/src/universal_mcp/agents/react.py

@@ -7,7 +7,7 @@ from universal_mcp.types import ToolConfig, ToolFormat
 
 from universal_mcp.agents.base import BaseAgent
 from universal_mcp.agents.llm import load_chat_model
-from universal_mcp.agents.utils import initialize_ui_tools, messages_to_list
+from universal_mcp.agents.utils import messages_to_list
 
 DEVELOPER_PROMPT = """You are {name}.
 
@@ -38,8 +38,9 @@ class ReactAgent(BaseAgent):
     ):
         super().__init__(name, instructions, model, memory, **kwargs)
         self.llm = load_chat_model(model)
-        self.tools = tools
-        self.ui_tools = initialize_ui_tools()
+        self.tools = tools or {}
+        if "ui" not in self.tools:
+            self.tools["ui"] = ["create_bar_chart", "create_line_chart", "create_pie_chart", "create_table", "http_get", "http_post", "http_put", "http_delete", "http_patch", "read_file"]
         self.max_iterations = max_iterations
         self.registry = registry
 
@@ -48,13 +49,11 @@ class ReactAgent(BaseAgent):
         if self.tools:
             if not self.registry:
                 raise ValueError("Tools are configured but no registry is provided")
-
             tools = await self.registry.export_tools(self.tools, ToolFormat.LANGCHAIN)
             logger.debug(tools)
         else:
             tools = []
 
-        tools.extend(self.ui_tools)
 
         logger.debug(f"Initialized ReactAgent: name={self.name}, model={self.model}")
         return create_react_agent(

{universal_mcp_agents-0.1.5 → universal_mcp_agents-0.1.6}/src/universal_mcp/agents/utils.py

@@ -11,7 +11,6 @@ from rich.table import Table
 from universal_mcp.tools.manager import ToolManager
 from universal_mcp.types import ToolFormat
 
-from universal_mcp.agents.ui_tools import UIToolsApp
 
 
 class RichCLI:
@@ -138,12 +137,3 @@ Available commands:
 
 def messages_to_list(messages: list[BaseMessage]):
     return [{"type": message.type, "content": message.content} for message in messages]
-
-
-def initialize_ui_tools() -> list:
-    """
-    Initialize and return UI tools in a langchain compatible format.
-    """
-    tool_manager = ToolManager(default_format=ToolFormat.LANGCHAIN)
-    tool_manager.register_tools_from_app(UIToolsApp())
-    return tool_manager.list_tools()

universal_mcp_agents-0.1.5/src/universal_mcp/agents/ui_tools.py → universal_mcp_agents-0.1.6/src/universal_mcp/applications/ui/app.py

@@ -28,12 +28,12 @@ class ColumnDefinition(TypedDict):
     type: Literal["string", "number", "date", "boolean"] | None
 
 
-class UIToolsApp(BaseApplication):
+class UiApp(BaseApplication):
     """An application for creating UI tools"""
 
-    def __init__(self):
+    def __init__(self, **kwargs):
         """Initialize the DefaultToolsApp"""
-        super().__init__(name="ui_tools")
+        super().__init__(name="ui")
         self.markitdown = MarkItDown(enable_plugins=True)
 
     def create_bar_chart(