universal-mcp 0.1.24rc13__py3-none-any.whl → 0.1.24rc17__py3-none-any.whl

universal_mcp/agentr/registry.py

@@ -3,7 +3,7 @@ from typing import Any
 from loguru import logger
 
 from universal_mcp.agentr.client import AgentrClient
-from universal_mcp.applications import app_from_slug
+from universal_mcp.applications.utils import app_from_slug
 from universal_mcp.tools.manager import ToolManager, _get_app_and_tool_name
 from universal_mcp.tools.registry import ToolRegistry
 from universal_mcp.types import ToolConfig, ToolFormat
@@ -133,7 +133,7 @@ class AgentrRegistry(ToolRegistry):
         try:
             # Clear tools from tool manager before loading new tools
             self.tool_manager.clear_tools()
-            if isinstance(tools, ToolConfig):
+            if isinstance(tools, dict):
                 logger.info("Loading tools from tool config")
                 self._load_tools_from_tool_config(tools, self.tool_manager)
             else:
@@ -176,8 +176,8 @@ class AgentrRegistry(ToolRegistry):
             tool_config: The tool configuration containing app names and tools
             tool_manager: The tool manager to register tools with
         """
-        for app_name, tool_data in tool_config.agentrServers.items():
-            self._load_tools(app_name, tool_data.tools, tool_manager)
+        for app_name, tool_names in tool_config.items():
+            self._load_tools(app_name, tool_names, tool_manager)
 
     async def call_tool(self, tool_name: str, tool_args: dict[str, Any]) -> dict[str, Any]:
         """Call a tool with the given name and arguments."""

universal_mcp/applications/application.py

@@ -9,7 +9,6 @@ from gql.transport.requests import RequestsHTTPTransport
 from graphql import DocumentNode
 from loguru import logger
 
-from universal_mcp.analytics import analytics
 from universal_mcp.integrations.integration import Integration
 
 
@@ -37,7 +36,6 @@ class BaseApplication(ABC):
         """
         self.name = name
         logger.debug(f"Initializing Application '{name}' with kwargs: {kwargs}")
-        analytics.track_app_loaded(name)  # Track app loading
 
     @abstractmethod
     def list_tools(self) -> list[Callable]:

universal_mcp/applications/utils.py

@@ -0,0 +1,52 @@
+import importlib
+
+from loguru import logger
+
+from universal_mcp.applications.application import BaseApplication
+
+# --- Default Name Generators ---
+
+
+def get_default_package_name(slug: str) -> str:
+    """
+    Convert a repository slug to a package name.
+    """
+    slug = slug.strip().lower().replace("-", "_")
+    package_name = f"universal_mcp.applications.{slug}"
+    return package_name
+
+
+def get_default_module_path(slug: str) -> str:
+    """
+    Convert a repository slug to a module path.
+    """
+    package_name = get_default_package_name(slug)
+    module_path = f"{package_name}.app"
+    return module_path
+
+
+def get_default_class_name(slug: str) -> str:
+    """
+    Convert a repository slug to a class name.
+    """
+    slug = slug.strip().lower()
+    parts = slug.replace("-", "_").split("_")
+    class_name = "".join(part.capitalize() for part in parts) + "App"
+    return class_name
+
+
+# --- Application Loaders ---
+
+
+def app_from_slug(slug: str) -> type[BaseApplication]:
+    """Loads an application from a slug."""
+    return load_app_from_package(slug)
+
+
+def load_app_from_package(slug: str) -> type[BaseApplication]:
+    """Loads an application from a pip-installable package."""
+    logger.debug(f"Loading '{slug}' as a package.")
+    module_path_str = get_default_module_path(slug)
+    class_name_str = get_default_class_name(slug)
+    module = importlib.import_module(module_path_str)
+    return getattr(module, class_name_str)

universal_mcp/servers/server.py

@@ -5,7 +5,8 @@ from loguru import logger
 from mcp.server.fastmcp import FastMCP
 from mcp.types import TextContent
 
-from universal_mcp.applications import BaseApplication, app_from_config
+from universal_mcp.applications.application import BaseApplication
+from universal_mcp.applications.utils import app_from_slug
 from universal_mcp.config import ServerConfig
 from universal_mcp.exceptions import ConfigurationError, ToolError
 from universal_mcp.integrations.integration import ApiKeyIntegration, OAuthIntegration
@@ -44,8 +45,8 @@ def load_from_local_config(config: ServerConfig, tool_manager: ToolManager) -> N
                     integration = OAuthIntegration(config.name, store=store, **app_config.integration.credentials)
                 else:
                     raise ValueError(f"Unsupported integration type: {app_config.integration.type}")
-            app = app_from_config(app_config)(integration=integration)
-            tool_manager.register_tools_from_app(app, app_config.actions)
+            app = app_from_slug(app_config.name)(integration=integration)
+            tool_manager.register_tools_from_app(app, tool_names=app_config.actions)
             logger.info(f"Loaded app: {app_config.name}")
         except Exception as e:
             logger.error(f"Failed to load app {app_config.name}: {e}", exc_info=True)

universal_mcp/tools/manager.py

@@ -3,7 +3,6 @@ from typing import Any
 
 from loguru import logger
 
-from universal_mcp.analytics import analytics
 from universal_mcp.applications.application import BaseApplication
 from universal_mcp.exceptions import ToolNotFoundError
 from universal_mcp.tools.adapters import (
@@ -298,8 +297,6 @@ class ToolManager:
             raise ToolNotFoundError(f"Unknown tool: {name}")
         try:
             result = await tool.run(arguments, context)
-            analytics.track_tool_called(name, app_name, "success")
             return result
         except Exception as e:
-            analytics.track_tool_called(name, app_name, "error", str(e))
             raise e

universal_mcp/types.py

@@ -1,7 +1,4 @@
 from enum import Enum
-from typing import Literal
-
-from pydantic import BaseModel
 
 # Constants
 DEFAULT_IMPORTANT_TAG = "important"
@@ -18,21 +15,4 @@ class ToolFormat(str, Enum):
     OPENAI = "openai"
 
 
-class AgentrConnection(BaseModel):
-    tools: list[str]
-
-
-class MCPConnection(BaseModel):
-    transport: Literal["stdio", "sse", "streamable-http"]
-    command: str | None = None
-    args: list[str] | None = None
-    url: str | None = None
-    headers: dict[str, str] | None = None
-
-
-class AgentrToolConfig(BaseModel):
-    agentrServers: dict[str, AgentrConnection] | None = None
-
-
-class ToolConfig(AgentrToolConfig):
-    mcpServers: dict[str, MCPConnection] | None = None
+ToolConfig = dict[str, list[str]]

universal_mcp/utils/prompts.py

@@ -229,7 +229,6 @@ from loguru import logger
 
 import httpx
 
-from universal_mcp.analytics import analytics
 from universal_mcp.integrations import Integration
 
 class BaseApplication(ABC):
@@ -256,7 +255,6 @@ class BaseApplication(ABC):
         """
         self.name = name
         logger.debug(f"Initializing Application '{name}' with kwargs: {kwargs}")
-        analytics.track_app_loaded(name)  # Track app loading
 
     @abstractmethod
     def list_tools(self) -> list[Callable]:

universal_mcp/utils/testing.py

@@ -8,7 +8,7 @@ from pydantic import BaseModel, SecretStr
 
 from universal_mcp.agentr import AgentrIntegration
 from universal_mcp.agentr.client import AgentrClient
-from universal_mcp.applications import APIApplication, BaseApplication
+from universal_mcp.applications.application import APIApplication, BaseApplication
 from universal_mcp.tools import Tool, ToolManager
 from universal_mcp.types import ToolFormat
 

{universal_mcp-0.1.24rc13.dist-info → universal_mcp-0.1.24rc17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: universal-mcp
-Version: 0.1.24rc13
+Version: 0.1.24rc17
 Summary: Universal MCP acts as a middle ware for your API applications. It can store your credentials, authorize, enable disable apps on the fly and much more.
 Author-email: Manoj Bajaj <manojbajaj95@gmail.com>
 License: MIT
@@ -32,6 +32,7 @@ Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: rich>=14.0.0
 Requires-Dist: streamlit>=1.46.1
 Requires-Dist: typer>=0.15.2
+Requires-Dist: universal-mcp-applications>=0.1.2
 Provides-Extra: dev
 Requires-Dist: litellm>=1.30.7; extra == 'dev'
 Requires-Dist: pre-commit>=4.2.0; extra == 'dev'

universal_mcp-0.1.24rc17.dist-info/RECORD

@@ -0,0 +1,54 @@
+universal_mcp/cli.py,sha256=pPnIWLhSrLV9ukI8YAg2znehCR3VovhEkmh8XkRT3MU,2505
+universal_mcp/config.py,sha256=lOlDAgQMT7f6VymmsuCP9sYLlxGKj0hDF3hFcJ2nzS4,8135
+universal_mcp/exceptions.py,sha256=Uen8UFgLHGlSwXgRUyF-nhqTwdiBuL3okgBVRV2AgtA,2150
+universal_mcp/logger.py,sha256=VmH_83efpErLEDTJqz55Dp0dioTXfGvMBLZUx5smOLc,2116
+universal_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+universal_mcp/types.py,sha256=DAzYyS7Zqdi38SpnhdOm08_JeN2CJiC0hDIpLLIhlSw,316
+universal_mcp/agentr/README.md,sha256=t15pVgkCwZM5wzgLgrf0Zv6hVL7dPmKXvAeTf8CiXPQ,6641
+universal_mcp/agentr/__init__.py,sha256=fv1ZnOCduIUiJ9oN4e6Ya_hA2oWQvcEuDU3Ek1vEufI,180
+universal_mcp/agentr/client.py,sha256=TQgwrNc7dEMXuprELf0Q-fdYdrH92Ppd7PUDZoD-KcY,7429
+universal_mcp/agentr/integration.py,sha256=V5GjqocqS02tRoI8MeV9PL6m-BzejwBzgJhOHo4MxAE,4212
+universal_mcp/agentr/registry.py,sha256=6KFTv45p_crqRqpdcwj1ksaF41PKXzgYIACRxsLaj3o,6909
+universal_mcp/agentr/server.py,sha256=bIPmHMiKKwnUYnxmfZVRh1thcn7Rytm_-bNiXTfANzc,2098
+universal_mcp/applications/application.py,sha256=do45GC0jnNepZHL8US2yDNq9s0ZnE6bII4Xbw90GRTc,23727
+universal_mcp/applications/utils.py,sha256=8Pp9lZU6IPt9z9BnuJ-vpv-NGuzryt1c4e4-ShDd2XI,1450
+universal_mcp/applications/sample/app.py,sha256=E0JwaWD7qytwawb_iWc1pBnJ-Te7MMtab4MxOOebLdc,8972
+universal_mcp/client/oauth.py,sha256=O00zOUfQxINaruFU2zt-64DIR1_mAqrY8ykLQo-teJU,8679
+universal_mcp/client/token_store.py,sha256=6VAzjzJG49wYvmEDqksFvb-fVqdjHIKWv7yYyh_AuF8,3912
+universal_mcp/client/transport.py,sha256=qqtb0ky6yvLBxsaA9-oFU0v9MwfcQb4eomq-O2fmwtQ,11850
+universal_mcp/integrations/__init__.py,sha256=tfzLyPEPly5tfIcT8K6-oKCr_MEFGxOROHy_NeVy0KM,200
+universal_mcp/integrations/integration.py,sha256=H-hOoDHqk78A4Fi_TGN7OOFS7PDfqXK_nedH8iSz-6A,16459
+universal_mcp/servers/__init__.py,sha256=speBb_E94UJa4A6Fv8RHFeoJ7cR-q2bCMtKV7R21P5w,142
+universal_mcp/servers/server.py,sha256=bJTG86X_kn0R8lWIYvCZw-cp-rVAOKd12-E5WOEsz8Y,5950
+universal_mcp/stores/__init__.py,sha256=quvuwhZnpiSLuojf0NfmBx2xpaCulv3fbKtKaSCEmuM,603
+universal_mcp/stores/store.py,sha256=yWbEGZb53z3fpVyqGWbes63z1CtIzC_IuM49OXy__UY,10137
+universal_mcp/tools/__init__.py,sha256=jC8hsqfTdtn32yU57AVFUXiU3ZmUOCfCERSCaNEIH7E,395
+universal_mcp/tools/adapters.py,sha256=YJ2oqgc8JgmtsdRRtvO-PO0Q0bKqTJ4Y3J6yxlESoTo,3947
+universal_mcp/tools/docstring_parser.py,sha256=efEOE-ME7G5Jbbzpn7pN2xNuyu2M5zfZ1Tqu1lRB0Gk,8392
+universal_mcp/tools/func_metadata.py,sha256=F4jd--hoZWKPBbZihVtluYKUsIdXdq4a0VWRgMl5k-Q,10838
+universal_mcp/tools/manager.py,sha256=T8iSk7Q6s3MFaQcwUapPvcQ6_l7_g4Xt2xXk55lPJ1w,10321
+universal_mcp/tools/registry.py,sha256=LD0J_bPsd8PRTObyvXglqTW1jfZX98m7KBdyP8Yn7wA,2585
+universal_mcp/tools/tools.py,sha256=Lk-wUO3rfhwdxaRANtC7lQr5fXi7nclf0oHzxNAb79Q,4927
+universal_mcp/utils/__init__.py,sha256=8wi4PGWu-SrFjNJ8U7fr2iFJ1ktqlDmSKj1xYd7KSDc,41
+universal_mcp/utils/installation.py,sha256=PU_GfHPqzkumKk-xG4L9CkBzSmABxmchwblZkx-zY-I,7204
+universal_mcp/utils/prompts.py,sha256=yaZ8lj5GooA8bbT42lcWR2r35HAnOalCvyWS1I2ltGM,27439
+universal_mcp/utils/singleton.py,sha256=RoOiKxBOAhp0TK1QaMDYi-8GjRcy2Vh-bAOuIAcYan0,775
+universal_mcp/utils/testing.py,sha256=PJYB-QVlBuDkRW-luHkJapf4T1oaNjxt8gvQoTS7cCA,8021
+universal_mcp/utils/openapi/__inti__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+universal_mcp/utils/openapi/api_generator.py,sha256=sygUE8Sropq3Pz09N1kSwGelgnTCAz_lOGspTaYCWEs,6985
+universal_mcp/utils/openapi/api_splitter.py,sha256=io7fV-E8hUIR4NxFlakqydbgrQF6aBAnZHPMlpxw-wc,20967
+universal_mcp/utils/openapi/cli.py,sha256=az5ObS74R-MmDCOZ1PHTJVKZJrHnsBOAweOUa7A-GqA,25918
+universal_mcp/utils/openapi/docgen.py,sha256=DNmwlhg_-TRrHa74epyErMTRjV2nutfCQ7seb_Rq5hE,21366
+universal_mcp/utils/openapi/filters.py,sha256=96FajO5nLbvjNPy2A1HvSS9jqpzMDHd4q_QTP-DIsPI,3842
+universal_mcp/utils/openapi/openapi.py,sha256=mcmahLJrR-CS-r2RDaYOgMV1Eq7ZuCHMk7_Qrijadh4,62613
+universal_mcp/utils/openapi/postprocessor.py,sha256=j_OZ5u2VCb44rfvnJZdKcxQRyGgrEVMr8ue9oN0Ed7A,12235
+universal_mcp/utils/openapi/preprocessor.py,sha256=r4n0WQI__OzPL8FTza7jxiM4EYeZwa-3tvEJaJYZC44,63081
+universal_mcp/utils/openapi/readme.py,sha256=R2Jp7DUXYNsXPDV6eFTkLiy7MXbSULUj1vHh4O_nB4c,2974
+universal_mcp/utils/openapi/test_generator.py,sha256=vucBh9klWmQOUA740TFwfM9ry2nkwKWQiNRcsiZ9HbY,12229
+universal_mcp/utils/templates/README.md.j2,sha256=Mrm181YX-o_-WEfKs01Bi2RJy43rBiq2j6fTtbWgbTA,401
+universal_mcp/utils/templates/api_client.py.j2,sha256=972Im7LNUAq3yZTfwDcgivnb-b8u6_JLKWXwoIwXXXQ,908
+universal_mcp-0.1.24rc17.dist-info/METADATA,sha256=Z_twD5L03occBIYlR1zfsz-hvETI6HRvDqaNv4vuD5Y,3304
+universal_mcp-0.1.24rc17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+universal_mcp-0.1.24rc17.dist-info/entry_points.txt,sha256=QlBrVKmA2jIM0q-C-3TQMNJTTWOsOFQvgedBq2rZTS8,56
+universal_mcp-0.1.24rc17.dist-info/licenses/LICENSE,sha256=NweDZVPslBAZFzlgByF158b85GR0f5_tLQgq1NS48To,1063
+universal_mcp-0.1.24rc17.dist-info/RECORD,,

universal_mcp/agents/__init__.py

@@ -1,9 +0,0 @@
-from universal_mcp.agents.autoagent import AutoAgent
-from universal_mcp.agents.base import BaseAgent
-from universal_mcp.agents.bigtool import BigToolAgent
-from universal_mcp.agents.builder import BuilderAgent
-from universal_mcp.agents.planner import PlannerAgent
-from universal_mcp.agents.react import ReactAgent
-from universal_mcp.agents.simple import SimpleAgent
-
-__all__ = ["BaseAgent", "ReactAgent", "SimpleAgent", "AutoAgent", "BigToolAgent", "PlannerAgent", "BuilderAgent"]

universal_mcp/agents/autoagent/__init__.py

@@ -1,30 +0,0 @@
-from langgraph.checkpoint.base import BaseCheckpointSaver
-
-from universal_mcp.agents.autoagent.graph import build_graph
-from universal_mcp.agents.base import BaseAgent
-from universal_mcp.tools.registry import ToolRegistry
-
-
-class AutoAgent(BaseAgent):
-    def __init__(
-        self,
-        name: str,
-        instructions: str,
-        model: str,
-        memory: BaseCheckpointSaver | None = None,
-        registry: ToolRegistry | None = None,
-        **kwargs,
-    ):
-        super().__init__(name, instructions, model, memory, **kwargs)
-        self.tool_registry = registry
-
-    async def _build_graph(self):
-        builder = await build_graph(self.tool_registry, self.instructions)
-        return builder.compile(checkpointer=self.memory)
-
-    @property
-    def graph(self):
-        return self._graph
-
-
-__all__ = ["AutoAgent"]

universal_mcp/agents/autoagent/__main__.py

@@ -1,25 +0,0 @@
-import asyncio
-
-from loguru import logger
-
-from universal_mcp.agentr.registry import AgentrRegistry
-from universal_mcp.agents.autoagent import AutoAgent
-
-
-async def main():
-    agent = AutoAgent(
-        name="autoagent",
-        instructions="You are a helpful assistant that can use tools to help the user.",
-        model="azure/gpt-4.1",
-        registry=AgentrRegistry(),
-    )
-    async for event in agent.stream(
-        user_input="Send an email to manoj@agentr.dev",
-        thread_id="test123",
-    ):
-        logger.info(event.content)
-    # from loguru import logger; logger.debug(result)
-
-
-if __name__ == "__main__":
-    asyncio.run(main())

universal_mcp/agents/autoagent/context.py

@@ -1,26 +0,0 @@
-from dataclasses import dataclass, field
-from typing import Annotated
-
-from universal_mcp.agents.autoagent.prompts import SYSTEM_PROMPT
-
-
-@dataclass(kw_only=True)
-class Context:
-    """The context for the agent."""
-
-    system_prompt: str = field(
-        default=SYSTEM_PROMPT,
-        metadata={
-            "description": "The system prompt to use for the agent's interactions. "
-            "This prompt sets the context and behavior for the agent."
-        },
-    )
-
-    model: Annotated[str, {"__template_metadata__": {"kind": "llm"}}] = field(
-        default="anthropic/claude-4-sonnet-20250514",
-        # default="vertex/gemini-2.5-flash",
-        metadata={
-            "description": "The name of the language model to use for the agent's main interactions. "
-            "Should be in the form: provider/model-name."
-        },
-    )

universal_mcp/agents/autoagent/graph.py

@@ -1,151 +0,0 @@
-import json
-from datetime import UTC, datetime
-from typing import cast
-
-from langchain_core.messages import AIMessage, ToolMessage
-from langchain_core.tools import tool
-from langgraph.graph import END, START, StateGraph
-from langgraph.runtime import Runtime
-
-from universal_mcp.agents.autoagent.context import Context
-from universal_mcp.agents.autoagent.prompts import SYSTEM_PROMPT
-from universal_mcp.agents.autoagent.state import State
-from universal_mcp.agents.llm import load_chat_model
-from universal_mcp.tools.registry import ToolRegistry
-from universal_mcp.types import ToolFormat
-
-
-async def build_graph(tool_registry: ToolRegistry, instructions: str = ""):
-    @tool()
-    async def search_tools(query: str, app_ids: list[str] | None = None) -> list[str]:
-        """Retrieve tools using a search query and a list of app ids. Use multiple times if you require tools for different queries."""
-        tools_list = []
-        if app_ids is not None:
-            for app_id in app_ids:
-                tools_list.extend(await tool_registry.search_tools(query, limit=10, app_id=app_id))
-        else:
-            tools_list = await tool_registry.search_tools(query, limit=10)
-        tools_list = [f"{tool['id']}: {tool['description']}" for tool in tools_list]
-        return tools_list
-
-    @tool()
-    async def ask_user(question: str) -> str:
-        """Ask the user a question. Use this tool to ask the user for any missing information for performing a task, or when you have multiple apps to choose from for performing a task."""
-        full_question = question
-        return f"ASKING_USER: {full_question}"
-
-    @tool()
-    async def load_tools(tools: list[str]) -> list[str]:
-        """Choose the tools you want to use by passing their tool ids.  Loads the tools for the chosen tools and returns the tool ids."""
-        return tools
-
-    async def call_model(
-        state: State,
-        runtime: Runtime[Context],
-    ):
-        system_prompt = SYSTEM_PROMPT
-        app_ids = await tool_registry.list_all_apps()
-        connections = await tool_registry.list_connected_apps()
-        connection_ids = set([connection["app_id"] for connection in connections])
-        connected_apps = [app["id"] for app in app_ids if app["id"] in connection_ids]
-        unconnected_apps = [app["id"] for app in app_ids if app["id"] not in connection_ids]
-        app_id_descriptions = "These are the apps connected to the user's account:\n" + "\n".join(
-            [f"{app}" for app in connected_apps]
-        )
-        if unconnected_apps:
-            app_id_descriptions += "\n\nOther (not connected) apps: " + "\n".join(
-                [f"{app}" for app in unconnected_apps]
-            )
-
-        system_prompt = system_prompt.format(system_time=datetime.now(tz=UTC).isoformat(), app_ids=app_id_descriptions)
-
-        messages = [{"role": "system", "content": system_prompt + "\n" + instructions}, *state["messages"]]
-        model = load_chat_model(runtime.context.model)
-        loaded_tools = await tool_registry.export_tools(tools=state["selected_tool_ids"], format=ToolFormat.LANGCHAIN)
-        model_with_tools = model.bind_tools([search_tools, ask_user, load_tools, *loaded_tools], tool_choice="auto")
-        response_raw = model_with_tools.invoke(messages)
-        response = cast(AIMessage, response_raw)
-        return {"messages": [response]}
-
-    # Define the conditional edge that determines whether to continue or not
-    def should_continue(state: State):
-        messages = state["messages"]
-        last_message = messages[-1]
-        # If there is no function call, then we finish
-        if not last_message.tool_calls:
-            return END
-        else:
-            return "tools"
-
-    def tool_router(state: State):
-        last_message = state["messages"][-1]
-        if isinstance(last_message, ToolMessage) and last_message.name == ask_user.name:
-            return END
-        else:
-            return "agent"
-
-    async def tool_node(state: State):
-        outputs = []
-        tool_ids = state["selected_tool_ids"]
-        for tool_call in state["messages"][-1].tool_calls:
-            if tool_call["name"] == ask_user.name:
-                outputs.append(
-                    ToolMessage(
-                        content=json.dumps(
-                            "The user has been asked the question, and the run will wait for the user's response."
-                        ),
-                        name=tool_call["name"],
-                        tool_call_id=tool_call["id"],
-                    )
-                )
-            elif tool_call["name"] == search_tools.name:
-                tools = await search_tools.ainvoke(tool_call["args"])
-                outputs.append(
-                    ToolMessage(
-                        content=json.dumps(tools) + "\n\nUse the load_tools tool to load the tools you want to use.",
-                        name=tool_call["name"],
-                        tool_call_id=tool_call["id"],
-                    )
-                )
-
-            elif tool_call["name"] == load_tools.name:
-                tool_ids = await load_tools.ainvoke(tool_call["args"])
-
-                outputs.append(
-                    ToolMessage(
-                        content=json.dumps(tool_ids),
-                        name=tool_call["name"],
-                        tool_call_id=tool_call["id"],
-                    )
-                )
-            else:
-                await tool_registry.export_tools([tool_call["name"]], ToolFormat.LANGCHAIN)
-                try:
-                    tool_result = await tool_registry.call_tool(tool_call["name"], tool_call["args"])
-                    outputs.append(
-                        ToolMessage(
-                            content=json.dumps(tool_result),
-                            name=tool_call["name"],
-                            tool_call_id=tool_call["id"],
-                        )
-                    )
-                except Exception as e:
-                    outputs.append(
-                        ToolMessage(
-                            content=json.dumps("Error: " + str(e)),
-                            name=tool_call["name"],
-                            tool_call_id=tool_call["id"],
-                        )
-                    )
-        return {"messages": outputs, "selected_tool_ids": tool_ids}
-
-    builder = StateGraph(State, context_schema=Context)
-
-    builder.add_node("agent", call_model)
-    builder.add_node("tools", tool_node)
-
-    builder.add_edge(START, "agent")
-    builder.add_conditional_edges("agent", should_continue)
-    builder.add_conditional_edges("tools", tool_router)
-
-    return builder

universal_mcp/agents/autoagent/prompts.py

@@ -1,9 +0,0 @@
-"""Default prompts used by the agent."""
-
-SYSTEM_PROMPT = """You are a helpful AI assistant. When you lack tools for any task you should use the `search_tools` function to unlock relevant tools. Whenever you need to ask the user for any information, or choose between multiple different applications, you can ask the user using the `ask_user` function.
-
-System time: {system_time}
-These are the list of apps available to you:
-{app_ids}
-Note that when multiple apps seem relevant for a task, you MUST ask the user to choose the app. 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. You will be provided a link for connection that you should pass on to the user.
-"""

universal_mcp/agents/autoagent/state.py

@@ -1,27 +0,0 @@
-from typing import Annotated
-
-from langgraph.prebuilt.chat_agent_executor import AgentState
-
-
-def _enqueue(left: list, right: list) -> list:
-    """Treat left as a FIFO queue, append new items from right (preserve order),
-    keep items unique, and cap total size to 20 (drop oldest items)."""
-    max_size = 30
-    preferred_size = 20
-    if len(right) > preferred_size:
-        preferred_size = min(max_size, len(right))
-    queue = list(left or [])
-
-    for item in right[:preferred_size] or []:
-        if item in queue:
-            queue.remove(item)
-        queue.append(item)
-
-    if len(queue) > preferred_size:
-        queue = queue[-preferred_size:]
-
-    return queue
-
-
-class State(AgentState):
-    selected_tool_ids: Annotated[list[str], _enqueue]

universal_mcp/agents/autoagent/studio.py

@@ -1,25 +0,0 @@
-import asyncio
-
-from universal_mcp.agentr.registry import AgentrRegistry
-from universal_mcp.agents.autoagent import build_graph
-from universal_mcp.tools import ToolManager
-
-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/autoagent/utils.py

@@ -1,13 +0,0 @@
-from langchain_core.messages import BaseMessage
-
-
-def get_message_text(msg: BaseMessage) -> str:
-    """Get the text content of a message."""
-    content = msg.content
-    if isinstance(content, str):
-        return content
-    elif isinstance(content, dict):
-        return content.get("text", "")
-    else:
-        txts = [c if isinstance(c, str) else (c.get("text") or "") for c in content]
-        return "".join(txts).strip()