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

universal_mcp/agents/__init__.py

@@ -7,6 +7,25 @@ from universal_mcp.agents.planner import PlannerAgent
 from universal_mcp.agents.react import ReactAgent
 from universal_mcp.agents.simple import SimpleAgent
 
+
+def get_agent(agent_name: str):
+    if agent_name == "auto":
+        return AutoAgent
+    elif agent_name == "react":
+        return ReactAgent
+    elif agent_name == "simple":
+        return SimpleAgent
+    elif agent_name == "builder":
+        return BuilderAgent
+    elif agent_name == "planner":
+        return PlannerAgent
+    elif agent_name == "bigtool":
+        return BigToolAgent
+    elif agent_name == "bigtool2":
+        return BigToolAgent2
+    else:
+        raise ValueError(f"Unknown agent: {agent_name}. Possible values: auto, react, simple, builder, planner, bigtool, bigtool2")
+
 __all__ = [
     "BaseAgent",
     "ReactAgent",

universal_mcp/agents/autoagent/__init__.py

@@ -1,8 +1,8 @@
 from langgraph.checkpoint.base import BaseCheckpointSaver
+from universal_mcp.tools.registry import ToolRegistry
 
 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):

universal_mcp/agents/autoagent/__main__.py

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

universal_mcp/agents/autoagent/graph.py

@@ -6,13 +6,13 @@ 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.tools.registry import ToolRegistry
+from universal_mcp.types import ToolFormat
 
 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 = ""):

universal_mcp/agents/autoagent/studio.py

@@ -1,9 +1,10 @@
 import asyncio
 
 from universal_mcp.agentr.registry import AgentrRegistry
-from universal_mcp.agents.autoagent import build_graph
 from universal_mcp.tools import ToolManager
 
+from universal_mcp.agents.autoagent import build_graph
+
 tool_registry = AgentrRegistry()
 tool_manager = ToolManager()
 

universal_mcp/agents/base.py

@@ -2,9 +2,11 @@
 from typing import cast
 from uuid import uuid4
 
-from langchain_core.messages import AIMessageChunk
+from langchain_core.messages import AIMessage, AIMessageChunk
 from langgraph.checkpoint.base import BaseCheckpointSaver
+from langgraph.graph import StateGraph
 from langgraph.types import Command
+from universal_mcp.logger import logger
 
 from .utils import RichCLI
 
@@ -31,7 +33,7 @@ class BaseAgent:
             self._graph = await self._build_graph()
             self._initialized = True
 
-    async def _build_graph(self):
+    async def _build_graph(self) -> StateGraph:
         raise NotImplementedError("Subclasses must implement this method")
 
     async def stream(self, thread_id: str, user_input: str, metadata: dict = None):
@@ -60,32 +62,41 @@ class BaseAgent:
         ):
             # Only forward assistant token chunks that are not tool-related.
             type_ = type(event)
-            if type_ != AIMessageChunk:
-                continue
-            event = cast(AIMessageChunk, event)
-            aggregate = event if aggregate is None else aggregate + event
             tags = metadata.get("tags", []) if isinstance(metadata, dict) else []
             is_quiet = isinstance(tags, list) and ("quiet" in tags)
-
             if is_quiet:
                 continue
+            # Handle different types of messages
+            if type_ in (AIMessage, AIMessageChunk):
+                # Accumulate billing and aggregate message
+                aggregate = event if aggregate is None else aggregate + event
+            # 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.debug(f"Finish event: {event}, Metadata: {metadata}")
                 pass
             else:
-                # logger.debug(f"Event: {event}, Metadata: {metadata}")
+                logger.debug(f"Event: {event}, Metadata: {metadata}")
                 yield event
         # Send a final finished message
         # The last event would be finish
         event = cast(AIMessageChunk, event)
+        event.usage_metadata = aggregate.usage_metadata
+        logger.debug(f"Usage metadata: {event.usage_metadata}")
         yield event
 
     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):
-                stream_updater.update(event.content)
+
+                if isinstance(event.content, list):
+                    thinking_content = "".join([c.get("thinking", "") for c in event.content])
+                    stream_updater.update(thinking_content, type_="thinking")
+                    content = "".join([c.get("text", "") for c in event.content])
+                    stream_updater.update(content, type_="text")
+                else:
+                    stream_updater.update(event.content, type_="text")
 
     async def invoke(
         self, user_input: str, thread_id: str = str(uuid4()), metadata: dict = None
@@ -106,11 +117,12 @@ class BaseAgent:
             "metadata": run_metadata,
         }
 
-        return await self._graph.ainvoke(
+        result = await self._graph.ainvoke(
             {"messages": [{"role": "user", "content": user_input}]},
             config=run_config,
             context={"system_prompt": self.instructions, "model": self.model},
         )
+        return result
 
     async def run_interactive(self, thread_id: str = str(uuid4())):
         """Main application loop"""
@@ -145,7 +157,7 @@ class BaseAgent:
                             f"Agent is {self.name}. {self.instructions}"
                         )
                         continue
-                    elif command == "exit" or command == "quit" or command == "q":
+                    elif command in {"exit", "quit", "q"}:
                         self.cli.display_info("Goodbye! 👋")
                         break
                     elif command == "reset":
@@ -170,6 +182,6 @@ class BaseAgent:
                 break
             except Exception as e:
                 import traceback
-
                 traceback.print_exc()
                 self.cli.display_error(f"An error occurred: {str(e)}")
+                break

universal_mcp/agents/bigtool/__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
@@ -19,18 +19,20 @@ class BigToolAgent(BaseAgent):
         memory: BaseCheckpointSaver | None = None,
         **kwargs,
     ):
-        # Combine the base system prompt with agent-specific instructions
-        full_instructions = f"{SYSTEM_PROMPT}\n\n**User Instructions:**\n{instructions}"
-        super().__init__(name, full_instructions, model, memory, **kwargs)
-
+        super().__init__(name, instructions, model, memory, **kwargs)
         self.registry = registry
         self.llm = load_chat_model(self.model)
-        self.tool_selection_llm = load_chat_model("gemini/gemini-2.0-flash-001")
 
         logger.info(
             f"BigToolAgent '{self.name}' initialized with model '{self.model}'."
         )
 
+    def _build_system_message(self):
+        return SYSTEM_PROMPT.format(
+            name=self.name,
+            instructions=self.instructions,
+        )
+
     async def _build_graph(self):
         """Build the bigtool agent graph using the existing create_agent function."""
         logger.info(f"Building graph for BigToolAgent '{self.name}'...")
@@ -38,7 +40,7 @@ class BigToolAgent(BaseAgent):
             graph_builder = build_graph(
                 tool_registry=self.registry,
                 llm=self.llm,
-                tool_selection_llm=self.tool_selection_llm,
+                system_prompt=self._build_system_message(),
             )
 
             compiled_graph = graph_builder.compile(checkpointer=self.memory)

universal_mcp/agents/bigtool/__main__.py

@@ -1,9 +1,10 @@
 import asyncio
 
 from loguru import logger
-
 from universal_mcp.agentr.registry import AgentrRegistry
+
 from universal_mcp.agents.bigtool import BigToolAgent
+from universal_mcp.agents.utils import messages_to_list
 
 
 async def main():
@@ -13,12 +14,10 @@ async def main():
         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)
-
+    await agent.ainit()
+    output = await agent.invoke(
+        user_input="Send an email to manoj@agentr.dev")
+    logger.info(messages_to_list(output["messages"]))
 
 if __name__ == "__main__":
     asyncio.run(main())

universal_mcp/agents/bigtool/graph.py

@@ -1,28 +1,24 @@
 import json
-from datetime import UTC, datetime
 from typing import Literal, TypedDict, cast
 
-from langchain_anthropic import ChatAnthropic
 from langchain_core.language_models import BaseChatModel
 from langchain_core.messages import AIMessage, ToolMessage
 from langchain_core.tools import tool
 from langgraph.graph import StateGraph
-from langgraph.runtime import Runtime
 from langgraph.types import Command
-
-from universal_mcp.agents.bigtool.context import Context
-from universal_mcp.agents.bigtool.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.bigtool.state import State
+
 from .prompts import SELECT_TOOL_PROMPT
 
 
 def build_graph(
     tool_registry: ToolRegistry,
     llm: BaseChatModel,
-    tool_selection_llm: BaseChatModel,
+    system_prompt: str,
 ):
     @tool
     async def retrieve_tools(task_query: str) -> list[str]:
@@ -40,7 +36,7 @@ def build_graph(
             class ToolSelectionOutput(TypedDict):
                 tool_names: list[str]
 
-            model = tool_selection_llm
+            model = llm
             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])
@@ -76,16 +72,14 @@ def build_graph(
             logger.error(f"Error retrieving tools: {e}")
             return []
 
+
     async def call_model(
-        state: State, runtime: Runtime[Context]
+        state: State
     ) -> Command[Literal["select_tools", "call_tools"]]:
         logger.info("Calling model...")
         try:
-            system_message = runtime.context.system_prompt.format(
-                system_time=datetime.now(tz=UTC).isoformat()
-            )
             messages = [
-                {"role": "system", "content": system_message},
+                {"role": "system", "content": system_prompt},
                 *state["messages"],
             ]
 
@@ -98,18 +92,15 @@ def build_graph(
             else:
                 selected_tools = []
 
-            model = llm
-            if isinstance(model, ChatAnthropic):
-                model_with_tools = model.bind_tools(
-                    [retrieve_tools, *selected_tools],
-                    tool_choice="auto",
-                    cache_control={"type": "ephemeral"},
-                )
-            else:
-                model_with_tools = model.bind_tools(
-                    [retrieve_tools, *selected_tools], tool_choice="auto"
-                )
-            response = cast(AIMessage, await model_with_tools.ainvoke(messages))
+            model_with_tools = llm.bind_tools(
+                [retrieve_tools, *selected_tools], tool_choice="auto"
+            )
+
+
+            response = await model_with_tools.ainvoke(messages)
+            cast(AIMessage, response)
+            logger.debug(f"Response: {response}")
+
 
             if response.tool_calls:
                 logger.info(
@@ -151,7 +142,7 @@ def build_graph(
             raise
 
     async def select_tools(
-        state: State, runtime: Runtime[Context]
+        state: State
     ) -> Command[Literal["call_model"]]:
         logger.info("Selecting tools...")
         try:
@@ -210,7 +201,7 @@ def build_graph(
             update={"messages": outputs, "selected_tool_ids": recent_tool_ids},
         )
 
-    builder = StateGraph(State, context_schema=Context)
+    builder = StateGraph(State)
 
     builder.add_node(call_model)
     builder.add_node(select_tools)

universal_mcp/agents/bigtool/prompts.py

@@ -1,15 +1,15 @@
 """Default prompts used by the agent."""
 
-SYSTEM_PROMPT = """You are a helpful AI assistant.
+SYSTEM_PROMPT = """You are {name}, 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 `retrieve_tools`:** Before you can use any other tool, you MUST first call the `retrieve_tools` function to find the right tool for the user's request. This is your mandatory first action.
 3.  **Strictly Follow the Process:** Your only job in your first turn is to analyze the user's request and call `retrieve_tools` with a concise query describing the core task. Do not engage in conversation.
 
-System time: {system_time}
-
 When multiple tools are available for the same task, you must ask the user.
+
+{instructions}
 """
 
 SELECT_TOOL_PROMPT = """You are an AI assistant that helps the user perform tasks using various apps (each app has multiple tools).

universal_mcp/agents/bigtool2/__init__.py

@@ -1,9 +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.logger import logger
-from universal_mcp.tools.registry import ToolRegistry
+from universal_mcp.agents.utils import initialize_ui_tools
 
 from .graph import build_graph
 from .prompts import SYSTEM_PROMPT
@@ -19,18 +20,23 @@ class BigToolAgent2(BaseAgent):
         memory: BaseCheckpointSaver | None = None,
         **kwargs,
     ):
-        # Combine the base system prompt with agent-specific instructions
-        full_instructions = f"{SYSTEM_PROMPT}\n\n**User Instructions:**\n{instructions}"
-        super().__init__(name, full_instructions, model, memory, **kwargs)
+        super().__init__(name, instructions, model, memory, **kwargs)
 
         self.registry = registry
         self.llm = load_chat_model(self.model)
         self.recursion_limit = kwargs.get("recursion_limit", 10)
+        self.ui_tools = initialize_ui_tools()
 
         logger.info(
             f"BigToolAgent '{self.name}' initialized with model '{self.model}'."
         )
 
+    def _build_system_message(self):
+        return SYSTEM_PROMPT.format(
+            name=self.name,
+            instructions=f"**User Instructions:**\n{self.instructions}",
+        )
+
     async def _build_graph(self):
         """Build the bigtool agent graph using the existing create_agent function."""
         logger.info(f"Building graph for BigToolAgent '{self.name}'...")
@@ -38,6 +44,8 @@ class BigToolAgent2(BaseAgent):
             graph_builder = build_graph(
                 tool_registry=self.registry,
                 llm=self.llm,
+                system_prompt=self._build_system_message(),
+                ui_tools=self.ui_tools,
             )
 
             compiled_graph = graph_builder.compile(checkpointer=self.memory)

universal_mcp/agents/bigtool2/__main__.py

@@ -1,9 +1,10 @@
 import asyncio
 
 from loguru import logger
-
 from universal_mcp.agentr.registry import AgentrRegistry
+
 from universal_mcp.agents.bigtool2 import BigToolAgent2
+from universal_mcp.agents.utils import messages_to_list
 
 
 async def main():
@@ -13,11 +14,11 @@ async def main():
         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)
+    await agent.ainit()
+    output = await agent.invoke(
+        user_input="Send an email to manoj@agentr.dev"
+    )
+    logger.info(messages_to_list(output["messages"]))
 
 
 if __name__ == "__main__":

universal_mcp/agents/bigtool2/agent.py

@@ -1,6 +1,7 @@
-from universal_mcp.agents.bigtool2 import BigToolAgent2
 from universal_mcp.agentr.registry import AgentrRegistry
 
+from universal_mcp.agents.bigtool2 import BigToolAgent2
+
 
 async def agent():
     agent_object = await BigToolAgent2(

universal_mcp/agents/bigtool2/graph.py

@@ -1,23 +1,22 @@
 import json
 from datetime import UTC, datetime
-from typing import Literal, TypedDict, cast
+from typing import Literal, cast
 
-from langchain_anthropic import ChatAnthropic
 from langchain_core.language_models import BaseChatModel
 from langchain_core.messages import AIMessage, ToolMessage
 from langchain_core.tools import tool
 from langgraph.graph import StateGraph
-from langgraph.runtime import Runtime
 from langgraph.types import Command
-
-from universal_mcp.agents.bigtool2.context import Context
-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):
+def build_graph(
+    tool_registry: ToolRegistry, llm: BaseChatModel, system_prompt: str, ui_tools: list
+):
     @tool
     async def search_tools(queries: list[str]) -> str:
         """Search tools for a given list of queries
@@ -32,7 +31,7 @@ def build_graph(tool_registry: ToolRegistry, llm: BaseChatModel):
             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_tools = {}
@@ -69,11 +68,11 @@ def build_graph(tool_registry: ToolRegistry, llm: BaseChatModel):
         return tool_ids
 
     async def call_model(
-        state: State, runtime: Runtime[Context]
+        state: State,
     ) -> Command[Literal["select_tools", "call_tools"]]:
         logger.info("Calling model...")
         try:
-            system_message = runtime.context.system_prompt.format(
+            system_message = system_prompt.format(
                 system_time=datetime.now(tz=UTC).isoformat()
             )
             messages = [
@@ -93,7 +92,8 @@ def build_graph(tool_registry: ToolRegistry, llm: BaseChatModel):
             model = llm
 
             model_with_tools = model.bind_tools(
-                [search_tools, load_tools, *selected_tools], tool_choice="auto"
+                [search_tools, load_tools, *selected_tools, *ui_tools],
+                tool_choice="auto",
             )
             response = cast(AIMessage, await model_with_tools.ainvoke(messages))
 
@@ -112,7 +112,7 @@ def build_graph(tool_registry: ToolRegistry, llm: BaseChatModel):
                 elif tool_call["name"] == load_tools.name:
                     logger.info("Model requested to load tools.")
                     tool_msg = ToolMessage(
-                        f"Loaded tools.", tool_call_id=tool_call["id"]
+                        "Loaded tools.", tool_call_id=tool_call["id"]
                     )
                     selected_tool_ids = tool_call["args"]["tool_ids"]
                     logger.info(f"Loaded tools: {selected_tool_ids}")
@@ -151,9 +151,7 @@ def build_graph(tool_registry: ToolRegistry, llm: BaseChatModel):
             logger.error(f"Error in call_model: {e}")
             raise
 
-    async def select_tools(
-        state: State, runtime: Runtime[Context]
-    ) -> Command[Literal["call_model"]]:
+    async def select_tools(state: State) -> Command[Literal["call_model"]]:
         logger.info("Selecting tools...")
         try:
             tool_call = state["messages"][-1].tool_calls[0]
@@ -204,7 +202,7 @@ def build_graph(tool_registry: ToolRegistry, llm: BaseChatModel):
             update={"messages": outputs, "selected_tool_ids": recent_tool_ids},
         )
 
-    builder = StateGraph(State, context_schema=Context)
+    builder = StateGraph(State)
 
     builder.add_node(call_model)
     builder.add_node(select_tools)

universal_mcp/agents/bigtool2/prompts.py

@@ -8,5 +8,5 @@ SYSTEM_PROMPT = """You are a helpful AI assistant.
 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.
 
-System time: {system_time}
+{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.agents.bigtoolcache import BigToolAgentCache
 
 

universal_mcp/agents/bigtoolcache/agent.py

@@ -1,10 +1,11 @@
-from universal_mcp.agents.bigtoolcache import BigToolAgentCache
 from universal_mcp.agentr.registry import AgentrRegistry
 
+from universal_mcp.agents.bigtoolcache import BigToolAgentCache
+
 
 async def agent():
     agent_object = await BigToolAgentCache(
-        name="BigTool Agent 2",
+        name="BigTool Agent Cache version",
         instructions="You are a helpful assistant that can use various tools to complete tasks.",
         model="anthropic/claude-4-sonnet-20250514",
         registry=AgentrRegistry(),

universal_mcp/agents/bigtoolcache/graph.py

@@ -1,6 +1,6 @@
 import json
 from datetime import UTC, datetime
-from typing import Any, Literal, TypedDict, cast
+from typing import Literal, TypedDict, cast
 
 from langchain_anthropic import ChatAnthropic
 from langchain_core.language_models import BaseChatModel
@@ -18,6 +18,7 @@ from universal_mcp.types import ToolFormat
 from universal_mcp.agents.bigtoolcache.prompts import TOOLS_LIST
 
 
+
 class ToolSelectionOutput(TypedDict):
     connected_tool_ids: list[str]
     unconnected_tool_ids: list[str]
@@ -180,15 +181,15 @@ def build_graph(tool_registry: ToolRegistry, llm: BaseChatModel):
                     name=tool_id,
                     tool_call_id=tool_call["id"],
                 )
-            )
-            recent_tool_ids.append(tool_id)
-        except Exception as e:
-            logger.error(f"Error executing tool '{tool_id}': {e}")
-            outputs.append(
-                ToolMessage(
-                    content=json.dumps("Error: " + str(e)),
-                    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"],
+                    )
                 )
             )
         return Command(

universal_mcp/agents/bigtoolcache/prompts.py

@@ -1,6 +1,5 @@
 """Default prompts used by the agent."""
 
-import os
 from pathlib import Path
 
 
@@ -14,7 +13,7 @@ def load_tools_from_file():
         tools_file = current_dir / "tools_all.txt"
 
     if tools_file.exists():
-        with open(tools_file, "r", encoding="utf-8") as f:
+        with open(tools_file, encoding="utf-8") as f:
             return f.read()
     else:
         return "No tools file found. Please run tool_retrieve.py to generate the tools list."