universal-mcp-agents 0.1.12__py3-none-any.whl → 0.1.13__py3-none-any.whl

universal_mcp/agents/__init__.py

@@ -3,8 +3,8 @@ from typing import Literal
 from universal_mcp.agents.base import BaseAgent
 from universal_mcp.agents.bigtool import BigToolAgent
 from universal_mcp.agents.builder.builder import BuilderAgent
-from universal_mcp.agents.codeact0 import CodeActAgent as CodeActRepl
 from universal_mcp.agents.codeact import CodeActAgent as CodeActScript
+from universal_mcp.agents.codeact0 import CodeActAgent as CodeActRepl
 from universal_mcp.agents.react import ReactAgent
 from universal_mcp.agents.simple import SimpleAgent
 

universal_mcp/agents/cli.py

@@ -1,3 +1,5 @@
+import asyncio
+
 from langgraph.checkpoint.memory import MemorySaver
 from typer import Typer
 from universal_mcp.agentr.client import AgentrClient
@@ -6,8 +8,6 @@ from universal_mcp.logger import setup_logger
 
 from universal_mcp.agents import get_agent
 
-import asyncio
-
 app = Typer()
 
 

universal_mcp/agents/codeact/agent.py

@@ -48,7 +48,7 @@ class CodeActAgent(BaseAgent):
             memory=memory,
             **kwargs,
         )
-        self.model_instance = load_chat_model(model, thinking=False)
+        self.model_instance = load_chat_model(model)
         self.tools_config = tools or {}
         self.registry = registry
         self.eval_fn = eval_unsafe

universal_mcp/agents/codeact/sandbox.py

@@ -51,17 +51,13 @@ SAFE_BUILTINS = {
 }
 
 
-# (The SAFE_BUILTINS definition remains the same)
-# ...
-
-
 async def eval_unsafe(
     code: str, _locals: dict[str, Callable], timeout: int = 10
 ) -> tuple[SandboxOutput, dict[str, Any]]:
     """Executes a string of Python code in a sandboxed environment."""
     original_keys = set(_locals.keys())
     execution_context = _locals.copy()
-    execution_context["__builtins__"] = SAFE_BUILTINS
+    execution_context["__builtins__"] = __builtins__  # TODO: Use SAFE_BUILTINS instead of __builtins__
 
     stdout_capture = io.StringIO()
     output = SandboxOutput(stdout="")

universal_mcp/agents/shared/tool_node.py

@@ -1,8 +1,7 @@
-import asyncio
 from typing import Annotated, TypedDict
 
 from langchain_core.language_models import BaseChatModel
-from langchain_core.messages import AIMessage, AnyMessage, HumanMessage
+from langchain_core.messages import AIMessage, AnyMessage
 from langgraph.graph import END, StateGraph
 from langgraph.graph.message import add_messages
 from loguru import logger
@@ -226,35 +225,3 @@ def build_tool_node_graph(llm: BaseChatModel, registry: ToolRegistry) -> StateGr
     workflow.add_edge("handle_planning_failure", END)
 
     return workflow.compile()
-
-
-async def main():
-    """Main function to run the agent."""
-    from universal_mcp.agentr.registry import AgentrRegistry
-
-    from universal_mcp.agents.llm import load_chat_model
-
-    registry = AgentrRegistry()
-    llm = load_chat_model("anthropic/claude-4-sonnet-20250514")
-
-    graph = build_tool_node_graph(llm, registry)
-
-    task = "Find my latest order confirmation in Gmail, search for reviews of the main product on perplexity, and then send an email to ankit@agentr.dev telling about the reviews"
-
-    initial_state = {
-        "original_task": task,
-        "messages": [HumanMessage(content=task)],
-        "decomposition_attempts": 0,
-    }
-
-    final_state = await graph.ainvoke(initial_state)
-
-    if final_state.get("execution_plan"):
-        for step in final_state["execution_plan"]:
-            pass
-    else:
-        pass
-
-
-if __name__ == "__main__":
-    asyncio.run(main())

{universal_mcp_agents-0.1.12.dist-info → universal_mcp_agents-0.1.13.dist-info}/METADATA

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

{universal_mcp_agents-0.1.12.dist-info → universal_mcp_agents-0.1.13.dist-info}/RECORD

@@ -1,6 +1,6 @@
-universal_mcp/agents/__init__.py,sha256=xQ0Yoln3f0MQrqTQhAZ0im7Tp1DawEJ04YPp20Nw1DQ,1244
+universal_mcp/agents/__init__.py,sha256=kM4mC6Pf6lmaaZF1volo7VtKgA8FDyzb1sNenpB7Ulk,1244
 universal_mcp/agents/base.py,sha256=le5vy02eXN15u8ntHWCu6Y-LgOmE_DA8tWmPOmLohAk,6908
-universal_mcp/agents/cli.py,sha256=Yv9g8VHOyK__V7me4UcWsWIXiviPz7F1vqMYH1jkCms,1016
+universal_mcp/agents/cli.py,sha256=AG9e4iSX3GazT537573YrYT1wSaZYOr42rrYQ7xP3YA,1016
 universal_mcp/agents/hil.py,sha256=_5PCK6q0goGm8qylJq44aSp2MadP-yCPvhOJYKqWLMo,3808
 universal_mcp/agents/llm.py,sha256=hVRwjZs3MHl5_3BWedmurs2Jt1oZDfFX0Zj9F8KH7fk,1787
 universal_mcp/agents/react.py,sha256=8XQvJ0HLVgc-K0qn9Ml48WGcgUGuIKtL67HatlT6Da0,3334
@@ -20,18 +20,16 @@ universal_mcp/agents/builder/prompts.py,sha256=RFzDBdGxF5BsAQL09BVas9RjAwt_Q2ZAC
 universal_mcp/agents/builder/state.py,sha256=7DeWllxfN-yD6cd9wJ3KIgjO8TctkJvVjAbZT8W_zqk,922
 universal_mcp/agents/codeact/__init__.py,sha256=rLE8gvOo5H4YSr71DRq76b3RV3uuotxuAy_VnBVaVwk,60
 universal_mcp/agents/codeact/__main__.py,sha256=W2cHXRwH1dZG3ETIkMwUqA_d62K3IctHP-FDZWDjxdw,1067
-universal_mcp/agents/codeact/agent.py,sha256=RRMeXVgn3Ow2gxhDKszx9hSUYGDTk6yqqDVyNIOu5kI,9778
+universal_mcp/agents/codeact/agent.py,sha256=sKZWokTHcuL68Y6SNyaaHe6_XkWxaIq36TrNmPJfQto,9762
 universal_mcp/agents/codeact/models.py,sha256=2fdAcF5bxWDpljjEwDEdPBflTMShSPwwncHrphRjsYg,222
 universal_mcp/agents/codeact/prompts.py,sha256=EMI-imnd0Ps0Bd2FOvSqgiicvvtFFu0MF9s93PiC_3k,4493
-universal_mcp/agents/codeact/sandbox.py,sha256=gEAmZ08PovnFmCKP7uIQaKXHFLOyrnuSmoh7E_SHhjQ,2307
+universal_mcp/agents/codeact/sandbox.py,sha256=NjN6ISj8psFtHf8V0w24ChJdUMUWkq7OrlbHdzm4wBc,2299
 universal_mcp/agents/codeact/state.py,sha256=WTPfpxDlGRnlr5tZuXMg_KU7GS7TZbnrIKslOvZLbQI,565
 universal_mcp/agents/codeact/utils.py,sha256=JUbT_HYGS_D1BzmzoVpORIe7SGur1KgJguTZ_1tZ4JY,1918
 universal_mcp/agents/codeact0/__init__.py,sha256=rLE8gvOo5H4YSr71DRq76b3RV3uuotxuAy_VnBVaVwk,60
 universal_mcp/agents/codeact0/__main__.py,sha256=V2wLWW9ym3rtiSvPEs-N0Mki7G5dYHzV5dAsAoF-ygQ,1148
 universal_mcp/agents/codeact0/agent.py,sha256=e1v_XVRqblKrAgrkyiPz7bs4atgDRTVHPnZQUjnT_4o,6495
 universal_mcp/agents/codeact0/config.py,sha256=H-1woj_nhSDwf15F63WYn723y4qlRefXzGxuH81uYF0,2215
-universal_mcp/agents/codeact0/langgraph_graph.py,sha256=mP3JrZ1I-Cqhk6E2DMRHUdLFevpyyad-Y3qhERn45JA,550
-universal_mcp/agents/codeact0/legacy_codeact.py,sha256=Lj4oH-AaP1zg6ifBkUxzWJcwMs9X5SepGzzZ6S5QMSg,4096
 universal_mcp/agents/codeact0/llm_tool.py,sha256=GEG8L2crRPyO5la_wlHohuqI9m6xH9KXgQxcP5fCFuU,13814
 universal_mcp/agents/codeact0/prompts.py,sha256=GTUwnxyyUVdeAoAwo4ODGOlEtZStMkoP_lJEFRbDglY,7324
 universal_mcp/agents/codeact0/sandbox.py,sha256=tkrhQoV7sAIT5rtd5kpNYEgDz4y82WSHKpMGE6bqthE,3108
@@ -56,10 +54,10 @@ universal_mcp/agents/planner/graph.py,sha256=70hhIoEZOcYojpiyVSCedgYpnmxVP7aqdn8
 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/prompts.py,sha256=VOsXSUEwBXPaAuxJTUF6bgDGr41u6uctUNQSMRt_OJc,6414
-universal_mcp/agents/shared/tool_node.py,sha256=Ua_wzMt4YgIx4zLp3_ZCow-28qORwrZ2FvKqLPt3RlI,10415
+universal_mcp/agents/shared/tool_node.py,sha256=KPF_nONfF12BmUTb9stHnxveqL7enGBi7wBGLrOjMsQ,9489
 universal_mcp/applications/llm/__init__.py,sha256=xnpxq4Wl_pevvwtSUtEwcty8_d61ywO1V2YnEXyCREY,46
 universal_mcp/applications/llm/app.py,sha256=iNLU6z2LRZc01GfSKvV0vNzT1LhKAjq_UrSJEmjthjw,6032
 universal_mcp/applications/ui/app.py,sha256=56h9GvkHiELyAVOZvi0YjiowSymlqkJ5GW4O7LmGIPs,9459
-universal_mcp_agents-0.1.12.dist-info/METADATA,sha256=hqXAJNr5hrlsdZHsQwHed6ELV0b0h4q-l8Ar1IpsoEI,878
-universal_mcp_agents-0.1.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-universal_mcp_agents-0.1.12.dist-info/RECORD,,
+universal_mcp_agents-0.1.13.dist-info/METADATA,sha256=wiruytjCRQN0GSX6yMToFwnIlmqWJPZbdbUOHXpUXJY,878
+universal_mcp_agents-0.1.13.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+universal_mcp_agents-0.1.13.dist-info/RECORD,,

universal_mcp/agents/codeact0/langgraph_graph.py

@@ -1,17 +0,0 @@
-from langgraph.checkpoint.memory import MemorySaver
-from universal_mcp.agentr.registry import AgentrRegistry
-
-from universal_mcp.agents.codeact0.agent import CodeActAgent
-
-
-async def agent():
-    memory = MemorySaver()
-    agent_object = await CodeActAgent(
-        name="CodeAct Agent",
-        instructions="Be very concise in your answers.",
-        model="anthropic:claude-4-sonnet-20250514",
-        tools={"google_mail": ["list_messages"]},
-        registry=AgentrRegistry(),
-        memory=memory,
-    )._build_graph()
-    return agent_object

universal_mcp/agents/codeact0/legacy_codeact.py

@@ -1,104 +0,0 @@
-import contextlib
-import inspect
-import io
-import queue
-import re
-import socket
-import threading
-import types
-from typing import Any
-
-import yaml
-from langchain.chat_models import init_chat_model
-from langchain_anthropic import ChatAnthropic
-from langchain_core.runnables import RunnableConfig
-from langchain_google_vertexai.model_garden import ChatAnthropicVertex
-from universal_mcp.agentr import Agentr
-from universal_mcp.types import ToolFormat
-
-from universal_mcp.agents.codeact0 import create_codeact
-from universal_mcp.agents.codeact0.config import ContextSchema
-from universal_mcp.agents.codeact0.utils import derive_context
-
-
-def eval(code: str, _locals: dict[str, Any], add_context: dict[str, Any]) -> tuple[str, dict[str, Any], dict[str, Any]]:
-    # print(_locals)
-    EXCLUDE_TYPES = (
-        types.ModuleType,  # modules
-        type(re.match("", "")),
-        type(threading.Lock()),  # instead of threading.Lock
-        type(threading.RLock()),  # reentrant lock
-        threading.Event,  # events
-        threading.Condition,  # condition vars
-        threading.Semaphore,  # semaphores
-        queue.Queue,  # thread-safe queues
-        socket.socket,  # network sockets
-        io.IOBase,  # file handles (and StringIO/BytesIO)
-    )
-    try:
-        with contextlib.redirect_stdout(io.StringIO()) as f:
-            # Execute the code in the provided locals context
-            # Using exec to allow dynamic code execution
-            # This is a simplified version; in production, consider security implications
-            exec(code, _locals, _locals)
-        result = f.getvalue()
-        if not result:
-            result = "<code ran, no output printed to stdout>"
-    except Exception as e:
-        result = f"Error during execution: {repr(e)}"
-
-    # Return all variables in locals except __builtins__ and unpicklable objects (including tools)
-    all_vars = {}
-    for key, value in _locals.items():
-        if key == "__builtins__":
-            continue
-
-        # Skip coroutines, async generators, and coroutine functions
-        if inspect.iscoroutine(value) or inspect.iscoroutinefunction(value):
-            continue
-        if inspect.isasyncgen(value) or inspect.isasyncgenfunction(value):
-            continue
-
-        # Skip "obviously unpicklable" types
-        if isinstance(value, EXCLUDE_TYPES):
-            continue
-
-        # Keep if it's not a callable OR if it has no __name__ attribute
-        if not callable(value) or not hasattr(value, "__name__"):
-            all_vars[key] = value
-
-    new_add_context = derive_context(code, add_context)
-    return result, all_vars, new_add_context
-
-
-async def agent(config: RunnableConfig):
-    cfg = ContextSchema(**config.get("configurable", {}))
-
-    if cfg.json_prompt_name and cfg.json_prompt_name.strip():
-        with open(f"usecases/{cfg.json_prompt_name}.yaml", encoding="utf-8") as f:
-            content = f.read()
-            data = yaml.safe_load(content)
-            if cfg.base_prompt and cfg.base_prompt.strip():
-                pass
-            else:
-                cfg.base_prompt = data["base_prompt"]
-            cfg.tool_names = data["tools"]
-    agentr = Agentr()
-    agentr.load_tools(cfg.tool_names)
-    tools = []  # can add custom tools here like get_weather, get_simple_weather, etc.
-
-    tools_agentr = agentr.list_tools(format=ToolFormat.NATIVE)
-    tools.extend(tools_agentr)
-
-    if cfg.model_provider == "google_anthropic_vertex":
-        # For Google Anthropic Vertex, we need to use the specific model initialization due to location
-        model = ChatAnthropicVertex(model=cfg.model, temperature=0.2, location="asia-east1")
-    elif cfg.model == "claude-4-sonnet-20250514":
-        model = ChatAnthropic(
-            model=cfg.model, temperature=1, thinking={"type": "enabled", "budget_tokens": 2048}, max_tokens=4096
-        )  # pyright: ignore[reportCallIssue]
-    else:
-        model = init_chat_model(model=cfg.model, model_provider=cfg.model_provider, temperature=0.2)
-
-    code_act = create_codeact(model, cfg.base_prompt, tools, eval)
-    return code_act.compile()