universal-mcp-agents 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

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/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/bigtool2/__main__.py

@@ -1,8 +1,8 @@
 import asyncio
 
 from loguru import logger
-from universal_mcp.agentr.registry import AgentrRegistry
 
+from universal_mcp.agentr.registry import AgentrRegistry
 from universal_mcp.agents.bigtool2 import BigToolAgent2
 from universal_mcp.agents.utils import messages_to_list
 

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/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_with_filters"], ToolFormat.LANGCHAIN
+        )
+        response = await tool_registry.call_tool("exa__search_with_filters", {"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/bigtool2/prompts.py

@@ -3,10 +3,13 @@
 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), you MUST use a tool. Do not answer from your own knowledge or refuse a task if a tool might exist for it.
-2.  **First Step is ALWAYS `search_tools`:** Before you can use any other tool, you MUST first call the `search_tools` function to find the right tools for the user's request. This is your mandatory first action. 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.
-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. 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.
-3.  **Strictly Follow the Process:** Your only job in your first turn is to analyze the user's request and call `search_tools` with a concise query describing the core task. Do not engage in conversation.
+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/bigtoolcache/__init__.py

@@ -1,9 +1,9 @@
 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.logger import logger
+from universal_mcp.tools.registry import ToolRegistry
 
 from .graph import build_graph
 from .prompts import SYSTEM_PROMPT

universal_mcp/agents/bigtoolcache/__main__.py

@@ -1,8 +1,8 @@
 import asyncio
 
 from loguru import logger
-from universal_mcp.agentr.registry import AgentrRegistry
 
+from universal_mcp.agentr.registry import AgentrRegistry
 from universal_mcp.agents.bigtoolcache import BigToolAgentCache
 
 

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/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/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/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/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/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/ui_tools.py → 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(

{universal_mcp_agents-0.1.5.dist-info → universal_mcp_agents-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: universal-mcp-agents
-Version: 0.1.5
+Version: 0.1.7
 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.5.dist-info → universal_mcp_agents-0.1.7.dist-info}/RECORD

@@ -1,38 +1,35 @@
 universal_mcp/agents/__init__.py,sha256=QfYDUZxIYQSqbpGt6NZ3U5tjf7SS1Y9uPzAwmaRoDrA,1186
-universal_mcp/agents/base.py,sha256=uJQnOwsggbvRpPwUkZ4QWgHBMrPVf2gJC8QbKglxlrM,6841
-universal_mcp/agents/builder.py,sha256=X3GETG465ifiJxWqk6ZxTlDvI48osHgaFbDGzc10Rsg,7640
+universal_mcp/agents/base.py,sha256=h_FDAclpFKpaMCSNhBcwIMF0DLbZtyyoy_l71UxY4Aw,6892
+universal_mcp/agents/builder.py,sha256=4RTROLljLzF4S3qrQrkl_mS3EipBBvqnvJflJ45oYCs,7850
 universal_mcp/agents/cli.py,sha256=_rJV6TxBG2amH3o8mVs4pxViaTfkBhz6n5l6xhv4Z3g,1014
 universal_mcp/agents/hil.py,sha256=XfQT8QcuDbiIpUU9N4WSbO2Tm9YNSuwRqyCTWmCWaZo,3818
-universal_mcp/agents/llm.py,sha256=hVRwjZs3MHl5_3BWedmurs2Jt1oZDfFX0Zj9F8KH7fk,1787
-universal_mcp/agents/react.py,sha256=6zwb-yPor86-bDOrrDaZl8MxyWL1ReikCkFs3noOwU4,2885
+universal_mcp/agents/llm.py,sha256=gnyCYW8Ohax8S9CXIfeI7hoBYNO0fa_1hJkIabCGoKY,1788
+universal_mcp/agents/react.py,sha256=jH42VFAB-BuPUVpaMIspBjjukYEJan-DQxtNamD1o0I,3010
 universal_mcp/agents/simple.py,sha256=Z5Ja12vJIhIHhB68WWH_5opln7FMDUiRfztKOj2Rx-U,1941
-universal_mcp/agents/tools.py,sha256=ysSwOYnpyoXA2Y1XcBFuYKKv-mUu-BYNUecpdRTK2fM,1406
-universal_mcp/agents/ui_tools.py,sha256=rNSXRHminXKvwGZISZ58qpnHQ8c6gPanXbu1icBXGFY,11224
-universal_mcp/agents/utils.py,sha256=bEQvI-dNo4sOj1te3ARmWnDSg9nFccfLiJ4baXEutoA,5323
+universal_mcp/agents/utils.py,sha256=g_v7IEKtx6CBQK-Nue_weVVie62KQLQjz7izU3kOWPQ,4988
 universal_mcp/agents/autoagent/__init__.py,sha256=fDiruZjxHhGPa6gg88as0SKHV8c3Mq56CCshZDIcLt8,836
 universal_mcp/agents/autoagent/__main__.py,sha256=S_inq-PR6KgJqcDhoXp_-PCehORF6WGputQ_hTrJOl0,654
 universal_mcp/agents/autoagent/context.py,sha256=RgjW1uCslucxYJpdmi4govd-0V1_9e6Y_kjWl3FpLrE,847
 universal_mcp/agents/autoagent/graph.py,sha256=JKdQy21w5jXyjyICTrhKS4MXHth6kHzREhR-DCU3Dws,6928
 universal_mcp/agents/autoagent/prompts.py,sha256=v-EwzZ_0XPuBNd_r8aWxmKMSQlZLTVBr0o-dmTQMN1w,892
 universal_mcp/agents/autoagent/state.py,sha256=TQeGZD99okclkoCh5oz-VYIlEsC9yLQyDpnBnm7QCN8,759
-universal_mcp/agents/autoagent/studio.py,sha256=ZxiC9C2xdOhZ4Qb5fKoAci3TOhcdKlpiTT_5C3CF_AI,639
 universal_mcp/agents/autoagent/utils.py,sha256=AFq-8scw_WlSZxDnTzxSNrOSiGYsIlqkqtQLDWf_rMU,431
 universal_mcp/agents/bigtool/__init__.py,sha256=Qw2C29q6R7TXmpzlSStwYM9Rbv9Iguj7U0p0CPy7thY,1782
 universal_mcp/agents/bigtool/__main__.py,sha256=WkPJl8r7L0CZXoFCU8w7eFEejLI8KIflkWnvLFt8jdA,660
 universal_mcp/agents/bigtool/graph.py,sha256=JAw7waUrAA7t6ZpbPcbnSu3sh6MhYfIUkJJfYc9mHCc,8528
 universal_mcp/agents/bigtool/prompts.py,sha256=pzLNaLg5ftv8B8k7McpDrSkos0N5eR6rKcw5psudBFg,1662
 universal_mcp/agents/bigtool/state.py,sha256=TQeGZD99okclkoCh5oz-VYIlEsC9yLQyDpnBnm7QCN8,759
-universal_mcp/agents/bigtool2/__init__.py,sha256=3BESPudZ_eAu3LOkNHWlabp_4Y6ioQmLXOgA5AKzKvs,2024
-universal_mcp/agents/bigtool2/__main__.py,sha256=fYWqoaJw10KsjDETbNM0yX61KXvinIm5jiwSuBcNNKY,669
-universal_mcp/agents/bigtool2/agent.py,sha256=4GIQIy2VQgdXOezmET8G7tvP_37Vv8C027bGdGXJbTI,437
-universal_mcp/agents/bigtool2/graph.py,sha256=i06Uoelw2lKgllO2REnvfDpV8T1AuAuCVJS45DIH8aY,9046
-universal_mcp/agents/bigtool2/prompts.py,sha256=UBOfJQOknNyNLaz_38CaBXsFOocbDQEKC59_mM_kpCQ,1640
+universal_mcp/agents/bigtool2/__init__.py,sha256=wkhjOeAHhIpuciLTbKZT3J2uPIJ0KFZpCDn0xX2plNs,2421
+universal_mcp/agents/bigtool2/__main__.py,sha256=SAHfoLqDEhUj3dF3vSzfetCzPGMC3UPJxBySHujSrDY,669
+universal_mcp/agents/bigtool2/agent.py,sha256=ef9IIxgJmr26eYWQdazrIA-IXHGRwT0XNyPThJR55Tk,436
+universal_mcp/agents/bigtool2/graph.py,sha256=GrFYoV2KDyUuutbiS4xrLk6V-x5LftGenZRuTSLRL30,11126
+universal_mcp/agents/bigtool2/prompts.py,sha256=rQFtZDkwU9z8d4PWdt6jpohGhyab658Xvk8hvNVBFBA,1843
 universal_mcp/agents/bigtool2/state.py,sha256=TQeGZD99okclkoCh5oz-VYIlEsC9yLQyDpnBnm7QCN8,759
-universal_mcp/agents/bigtoolcache/__init__.py,sha256=UeebKmlLdgyZfpjO1cB1ZmbYnfleKPvyi_XHegmHdPk,1844
-universal_mcp/agents/bigtoolcache/__main__.py,sha256=oldL6enfW_2UhyK1mW0UTBV_ww4PW2k_j187Hx_i8YI,622
-universal_mcp/agents/bigtoolcache/agent.py,sha256=ZRpPykErFcsV-YJCAV-3mCEkczvML9iwLcFEgEwbFfE,461
+universal_mcp/agents/bigtoolcache/__init__.py,sha256=YY7X8-XQ3AC2t_Y9MN9dZk5wTPu7iU6YS8Yhn_akgC0,1844
+universal_mcp/agents/bigtoolcache/__main__.py,sha256=HkPEQqsdnWtDzWSbYdVBBc_JhpRi82TYuaubxNMtt4w,622
+universal_mcp/agents/bigtoolcache/agent.py,sha256=GdIT7RZQ6SCIMrNUs4amZijBQk8d6YVh1fiYzLcUyO8,460
 universal_mcp/agents/bigtoolcache/context.py,sha256=ny7gd-vvVpUOYAeQbAEUT0A6Vm6Nn2qGywxTzPBzYFg,929
-universal_mcp/agents/bigtoolcache/graph.py,sha256=oFV24ap3xPwds6k5El9m28mf29Z3jzhYWc4XgNYn7pU,8694
+universal_mcp/agents/bigtoolcache/graph.py,sha256=9BrHPEl7oUPvVy_gIQ3ewwK4oz9-0yElR2G8yAsaLA8,8645
 universal_mcp/agents/bigtoolcache/prompts.py,sha256=XDU2uJWzwGwt8t3zGjOH16YIrHJCdPuLP5lyJ2llXFg,3226
 universal_mcp/agents/bigtoolcache/state.py,sha256=TQeGZD99okclkoCh5oz-VYIlEsC9yLQyDpnBnm7QCN8,759
 universal_mcp/agents/bigtoolcache/tools_all.txt,sha256=g52i00AOh9VTDsAtIAF8vhqtTHQVmzTn61k724niEA0,95408
@@ -47,6 +44,7 @@ universal_mcp/agents/planner/graph.py,sha256=qTXUVXiPWZP73GsY9mQEGmBtA-C1t06jpzg
 universal_mcp/agents/planner/prompts.py,sha256=_JoHqiAvswtqCDu90AGUHmfsu8eWE1-_yI4LLn3pqMU,657
 universal_mcp/agents/planner/state.py,sha256=qqyp-jSGsCxe1US-PRLT4-y1sITAcVE6nCMlQLnvop0,278
 universal_mcp/agents/shared/tool_node.py,sha256=FiKY0AhxKIFpzGKGnyRL-6Dzuf0KUDmiQo7IhXmE27s,8861
-universal_mcp_agents-0.1.5.dist-info/METADATA,sha256=RQnfekXpQJ0KkapTG_IQUXYPPd4tNcNGzpNaGLkFNcU,847
-universal_mcp_agents-0.1.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-universal_mcp_agents-0.1.5.dist-info/RECORD,,
+universal_mcp/applications/ui/app.py,sha256=y1DcnX1Vg-d1cBtEP0_jUOR2ypbB0X7q4YiQNAg9WdY,11223
+universal_mcp_agents-0.1.7.dist-info/METADATA,sha256=ZIkbodKsCGXwZ3iWEbJ9i5jpZ0F6xrLtlLu-jhgjYPo,798
+universal_mcp_agents-0.1.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+universal_mcp_agents-0.1.7.dist-info/RECORD,,

universal_mcp/agents/autoagent/studio.py

@@ -1,20 +0,0 @@
-import asyncio
-
-from universal_mcp.agentr.registry import AgentrRegistry
-from universal_mcp.tools import ToolManager
-
-from universal_mcp.agents.autoagent import build_graph
-
-tool_registry = AgentrRegistry()
-tool_manager = ToolManager()
-
-
-async def main():
-    instructions = """
-    You are a helpful assistant that can use tools to help the user. If a task requires multiple steps, you should perform separate different searches for different actions. Prefer completing one action before searching for another.
-    """
-    graph = await build_graph(tool_registry, instructions=instructions)
-    return graph
-
-
-graph = asyncio.run(main())

universal_mcp/agents/tools.py

@@ -1,40 +0,0 @@
-import json
-
-from langchain_mcp_adapters.client import MultiServerMCPClient
-from universal_mcp.agentr.integration import AgentrIntegration
-from universal_mcp.applications.utils import app_from_slug
-from universal_mcp.tools.adapters import ToolFormat
-from universal_mcp.tools.manager import ToolManager
-from universal_mcp.types import ToolConfig
-
-
-async def load_agentr_tools(agentr_servers: dict):
-    tool_manager = ToolManager()
-    for app_name, tool_names in agentr_servers.items():
-        app = app_from_slug(app_name)
-        integration = AgentrIntegration(name=app_name)
-        app_instance = app(integration=integration)
-        tool_manager.register_tools_from_app(
-            app_instance, tool_names=tool_names["tools"]
-        )
-    tools = tool_manager.list_tools(format=ToolFormat.LANGCHAIN)
-    return tools
-
-
-async def load_mcp_tools(mcp_servers: dict):
-    client = MultiServerMCPClient(mcp_servers)
-    tools = await client.get_tools()
-    return tools
-
-
-async def load_tools(path: str) -> ToolConfig:
-    with open(path) as f:
-        data = json.load(f)
-        config = ToolConfig.model_validate(data)
-        agentr_tools = await load_agentr_tools(
-            config.model_dump(exclude_none=True)["agentrServers"]
-        )
-        mcp_tools = await load_mcp_tools(
-            config.model_dump(exclude_none=True)["mcpServers"]
-        )
-        return agentr_tools + mcp_tools