universal-mcp-agents 0.1.15__py3-none-any.whl → 0.1.16__py3-none-any.whl

universal_mcp/agents/codeact0/llm_tool.py

@@ -27,6 +27,7 @@ def smart_print(data: Any) -> None:
     Args:
         data: Either a dictionary with string keys, or a list of such dictionaries
     """
+    print(light_copy(data)) # noqa
 
 
 def creative_writer(

universal_mcp/agents/codeact0/playbook_agent.py

@@ -25,7 +25,7 @@ from universal_mcp.agents.codeact0.state import CodeActState
 from universal_mcp.agents.codeact0.tools import create_meta_tools, enter_playbook_mode, exit_playbook_mode, get_valid_tools
 from universal_mcp.agents.codeact0.utils import inject_context, smart_truncate
 from universal_mcp.agents.llm import load_chat_model
-from universal_mcp.agents.utils import filter_retry_on, get_message_text
+from universal_mcp.agents.utils import filter_retry_on, get_message_text, convert_tool_ids_to_dict
 
 PLAYBOOK_PLANNING_PROMPT = """Now, you are tasked with creating a reusable playbook from the user's previous workflow.
 
@@ -69,6 +69,7 @@ class CodeActPlaybookAgent(BaseAgent):
         memory: BaseCheckpointSaver | None = None,
         tools: ToolConfig | None = None,
         registry: ToolRegistry | None = None,
+        playbook_registry: object | None = None,
         sandbox_timeout: int = 20,
         **kwargs,
     ):
@@ -82,33 +83,33 @@ class CodeActPlaybookAgent(BaseAgent):
         self.model_instance = load_chat_model(model, thinking=True)
         self.tools_config = tools or []
         self.registry = registry
+        self.playbook_registry = playbook_registry
         self.eval_fn = eval_unsafe
         self.sandbox_timeout = sandbox_timeout
         self.processed_tools: list[StructuredTool | Callable] = []
 
     async def _build_graph(self):
-        self.exported_tools = []
-        if self.tools_config:
-            # Convert dict format to list format if needed
-            if isinstance(self.tools_config, dict):
-                self.tools_config = [
-                    f"{provider}__{tool}" 
-                    for provider, tools in self.tools_config.items() 
-                    for tool in tools
-                ]
-            if not self.registry:
-                raise ValueError("Tools are configured but no registry is provided")
-            # Langchain tools are fine
-            self.exported_tools = await self.registry.export_tools(self.tools_config, ToolFormat.LANGCHAIN)
         meta_tools = create_meta_tools(self.registry)
-        await self.registry.export_tools(["exa__search_with_filters"], ToolFormat.LANGCHAIN)
         additional_tools = [smart_print, data_extractor, ai_classify, call_llm, meta_tools["web_search"]]
         self.additional_tools = [t if isinstance(t, StructuredTool) else create_tool(t) for t in additional_tools]
-        self.final_instructions, self.tools_context = create_default_prompt(
-            self.exported_tools, self.additional_tools, self.instructions
-        )
-
-        def call_model(state: CodeActState) -> Command[Literal["sandbox", "execute_tools"]]:
+        async def call_model(state: CodeActState) -> Command[Literal["sandbox", "execute_tools"]]:
+            self.exported_tools = []
+            if self.tools_config:
+                # Convert dict format to list format if needed
+                if isinstance(self.tools_config, dict):
+                    self.tools_config = [
+                        f"{provider}__{tool}" 
+                        for provider, tools in self.tools_config.items() 
+                        for tool in tools
+                    ]
+                if not self.registry:
+                    raise ValueError("Tools are configured but no registry is provided")
+                # Langchain tools are fine
+            self.tools_config.extend(state.get('selected_tool_ids',[]))
+            self.exported_tools = await self.registry.export_tools(self.tools_config, ToolFormat.LANGCHAIN)
+            self.final_instructions, self.tools_context = create_default_prompt(
+                self.exported_tools, self.additional_tools, self.instructions
+            )
             messages = [{"role": "system", "content": self.final_instructions}] + state["messages"]
 
             # Run the model and potentially loop for reflection
@@ -262,16 +263,16 @@ class CodeActPlaybookAgent(BaseAgent):
                 # Extract plan from response text between triple backticks
                 plan_match = re.search(r'```(.*?)```', response_text, re.DOTALL)
                 if plan_match:
-                    self.plan = plan_match.group(1).strip()
+                    plan = plan_match.group(1).strip()
                 else:
-                    self.plan = response_text.strip()
-                return Command(update={"messages": [response], "playbook_mode": "confirming"})
+                    plan = response_text.strip()
+                return Command(update={"messages": [response], "playbook_mode": "confirming", "plan": plan})
                 
 
             elif playbook_mode == "confirming":
                 confirmation_instructions = self.instructions + PLAYBOOK_CONFIRMING_PROMPT
                 messages = [{"role": "system", "content": confirmation_instructions}] + state["messages"]
-                response = self.model_instance.invoke(messages)
+                response = self.model_instance.invoke(messages, stream=False)
                 response = get_message_text(response)
                 if "true" in response.lower():
                     return Command(goto="playbook", update={"playbook_mode": "generating"})
@@ -296,24 +297,34 @@ class CodeActPlaybookAgent(BaseAgent):
                 else:
                     function_name = "generated_playbook"
 
+                # Save or update an Agent using the helper registry
+                saved_note = ""
                 try:
-                    current_path = Path(__file__).resolve()
-                    repo_root = None
-                    for ancestor in current_path.parents:
-                        if ancestor.name == "src":
-                            repo_root = ancestor.parent
-                            break
-                    if repo_root is None:
-                        repo_root = current_path.parents[-1]
-
-                    playbooks_dir = repo_root / "playbooks"
-                    playbooks_dir.mkdir(parents=True, exist_ok=True)
-
-                    file_path = playbooks_dir / f"{function_name}.py"
-                    file_path.write_text(func_code, encoding="utf-8")
-                    saved_note = f"Playbook function saved to: {file_path} ```{func_code}```"
+                    if not self.playbook_registry:
+                        raise ValueError("Playbook registry is not configured")
+
+                    # Build instructions payload embedding the plan and function code
+                    instructions_payload = {
+                        "playbookPlan": state["plan"],
+                        "playbookScript": {
+                            "name": function_name,
+                            "code": func_code,
+                        },
+                    }
+
+                    # Convert tool ids list to dict
+                    tool_dict = convert_tool_ids_to_dict(state["selected_tool_ids"])
+
+                    res = self.playbook_registry.create_agent(
+                        name=function_name,
+                        description=f"Generated playbook: {function_name}",
+                        instructions=instructions_payload,
+                        tools=tool_dict,
+                        visibility="private",
+                    )
+                    saved_note = f"Successfully created your playbook! Check it out here: [View Playbook](https://wingmen.info/agents/{res.id})"
                 except Exception as e:
-                    saved_note = f"Failed to save playbook function {function_name}: {e}"
+                    saved_note = f"Failed to save generated playbook as Agent '{function_name}': {e}"
 
                 # Mock tool call for exit_playbook_mode (for testing/demonstration)
                 mock_exit_tool_call = {
@@ -322,7 +333,7 @@ class CodeActPlaybookAgent(BaseAgent):
                     "id": "mock_exit_playbook_123"
                 }
                 mock_assistant_message = AIMessage(
-                    content="",  # Can be empty or a brief message
+                    content=saved_note, 
                     tool_calls=[mock_exit_tool_call]
                 )
 

universal_mcp/agents/codeact0/state.py

@@ -34,3 +34,5 @@ class CodeActState(AgentState):
     """State for the playbook agent."""
     selected_tool_ids: Annotated[list[str], _enqueue]
     """Queue for tools exported from registry"""
+    plan: str | None
+    """Plan for the playbook agent."""

universal_mcp/agents/codeact0/tools.py

@@ -4,6 +4,7 @@ from typing import Any
 
 from langchain_core.tools import tool
 from universal_mcp.tools.registry import ToolRegistry
+from universal_mcp.types import ToolFormat
 
 MAX_LENGHT=100
 
@@ -30,8 +31,8 @@ def create_meta_tools(tool_registry: ToolRegistry) -> dict[str, Any]:
             connections = await tool_registry.list_connected_apps()
             connected_apps = {connection["app_id"] for connection in connections}
 
-            # Use defaultdict to avoid key existence checks
-            app_tools = defaultdict(list)
+            app_tools = defaultdict(set)
+            MAX_LENGTH = 20
 
             # Process all queries concurrently
             search_tasks = []
@@ -40,20 +41,24 @@ def create_meta_tools(tool_registry: ToolRegistry) -> dict[str, Any]:
 
             query_results = await asyncio.gather(*search_tasks)
 
-            # Aggregate results with limit per app
+            # Aggregate results with limit per app and automatic deduplication
             for tools_list in query_results:
                 for tool in tools_list:
                     app = tool["id"].split("__")[0]
-                    if len(app_tools[app]) < MAX_LENGHT:
+                    tool_id = tool["id"]
+                    
+                    # Check if within limit and add to set (automatically deduplicates)
+                    if len(app_tools[app]) < MAX_LENGTH:
                         cleaned_desc = tool["description"].split("Context:")[0].strip()
-                        app_tools[app].append(f"{tool['id']}: {cleaned_desc}")
+                        app_tools[app].add(f"{tool_id}: {cleaned_desc}")
 
             # Build result string efficiently
             result_parts = []
             for app, tools in app_tools.items():
                 app_status = "connected" if app in connected_apps else "NOT connected"
                 result_parts.append(f"Tools from {app} (status: {app_status} by user):")
-                for tool in tools:
+                # Convert set to sorted list for consistent output
+                for tool in sorted(tools):
                     result_parts.append(f" - {tool}")
                 result_parts.append("")  # Empty line between apps
 
@@ -116,6 +121,7 @@ def create_meta_tools(tool_registry: ToolRegistry) -> dict[str, Any]:
         Example:
             results = await web_search(query="python programming")
         """
+        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}}
         )

universal_mcp/agents/codeact0/utils.py

@@ -6,7 +6,7 @@ from typing import Any
 
 from langchain_core.messages import BaseMessage
 
-MAX_CHARS = 300
+MAX_CHARS = 700
 
 
 def light_copy(data):

universal_mcp/agents/utils.py

@@ -214,3 +214,21 @@ def filter_retry_on(exc: Exception) -> bool:
 
     # Default: do not retry unknown exceptions
     return False
+
+
+def convert_tool_ids_to_dict(tool_ids: list[str]) -> dict[str, list[str]]:
+    """Convert list of tool ids like 'provider__tool' into a provider->tools dict.
+
+    Any ids without the expected delimiter are ignored.
+    """
+    provider_to_tools: dict[str, list[str]] = {}
+    for tool_id in tool_ids or []:
+        if "__" not in tool_id:
+            continue
+        provider, tool = tool_id.split("__", 1)
+        if not provider or not tool:
+            continue
+        if provider not in provider_to_tools:
+            provider_to_tools[provider] = []
+        provider_to_tools[provider].append(tool)
+    return provider_to_tools

{universal_mcp_agents-0.1.15.dist-info → universal_mcp_agents-0.1.16.dist-info}/METADATA

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

@@ -5,7 +5,7 @@ universal_mcp/agents/hil.py,sha256=_5PCK6q0goGm8qylJq44aSp2MadP-yCPvhOJYKqWLMo,3
 universal_mcp/agents/llm.py,sha256=hVRwjZs3MHl5_3BWedmurs2Jt1oZDfFX0Zj9F8KH7fk,1787
 universal_mcp/agents/react.py,sha256=8XQvJ0HLVgc-K0qn9Ml48WGcgUGuIKtL67HatlT6Da0,3334
 universal_mcp/agents/simple.py,sha256=NSATg5TWzsRNS7V3LFiDG28WSOCIwCdcC1g7NRwg2nM,2095
-universal_mcp/agents/utils.py,sha256=lGeDam3Efcrxv6dve31LQonBtwkSpNmBuoNXap_TqaQ,7128
+universal_mcp/agents/utils.py,sha256=P6W9k6XAOBp6tdjC2VTP4tE0B2M4-b1EDmr-ylJ47Pw,7765
 universal_mcp/agents/bigtool/__init__.py,sha256=mZG8dsaCVyKlm82otxtiTA225GIFLUCUUYPEIPF24uw,2299
 universal_mcp/agents/bigtool/__main__.py,sha256=0i-fbd2yQ90qa8n2nM3luqoJVN9Reh5HZXR5oK7SAck,445
 universal_mcp/agents/bigtool/agent.py,sha256=mtCDNN8WjE2hjJjooDqusmbferKBHeJMHrhXUPUWaVc,252
@@ -32,19 +32,19 @@ universal_mcp/agents/codeact0/__main__.py,sha256=V2wLWW9ym3rtiSvPEs-N0Mki7G5dYHz
 universal_mcp/agents/codeact0/agent.py,sha256=9BInAQr3csES-XHSscmeJlYJ3-wQUHPvLOf-6wFILUU,6695
 universal_mcp/agents/codeact0/config.py,sha256=H-1woj_nhSDwf15F63WYn723y4qlRefXzGxuH81uYF0,2215
 universal_mcp/agents/codeact0/langgraph_agent.py,sha256=ehjMV_Z1118pCFWB_Sa5H7XnUp0udsbUHjfjXjhIQM8,435
-universal_mcp/agents/codeact0/llm_tool.py,sha256=I7QIlgZZbzBdxHuNUYODA28Z7xqWgYz5v5qWSWqB0rE,13781
-universal_mcp/agents/codeact0/playbook_agent.py,sha256=oj8zP-c8rObzjAlS-lkP-pl3xleIr0fCJ_ENtn-C_OM,17435
+universal_mcp/agents/codeact0/llm_tool.py,sha256=_eT0yreFMRbHheWo-gGcRJ4Co6jMRJZXJ1oVe6f_ZyQ,13816
+universal_mcp/agents/codeact0/playbook_agent.py,sha256=YNuWQ_xK9KlOtb4kkHkxLl3CEJfmM3sUf8dwaSPyYcg,18089
 universal_mcp/agents/codeact0/prompts.py,sha256=j8HxA3Rp-EZsms9qMBcRmFrUjeySrG1IWjqrNFXZZn8,9457
 universal_mcp/agents/codeact0/sandbox.py,sha256=zMgHrWnQYkSkJb2MzfXvT3euCc4hvqzBE_EbX2_iLxA,3142
-universal_mcp/agents/codeact0/state.py,sha256=Qcr1whST3J8v7w0opnKSfOG9u5gRtxAzPs2NFhaAhJE,1199
-universal_mcp/agents/codeact0/tools.py,sha256=emfBLA3eChQ5B1wirOWf5RWHMy3OIRDQYnN5h5OPqFk,7401
-universal_mcp/agents/codeact0/utils.py,sha256=s1SfVrC1_UePxYSIL9zt-cG0xhwzFPuViyLl2Xj-45c,15943
+universal_mcp/agents/codeact0/state.py,sha256=Y-Rzn_S7--aXH18KPvyhqDqOOB-miu1lsAmLgmMlaAg,1259
+universal_mcp/agents/codeact0/tools.py,sha256=Adh9Go87JnIk5U_rtBOHglFRPowk2JmRK9bumR4zJoo,7751
+universal_mcp/agents/codeact0/utils.py,sha256=ACncBI9-JJ4Qe_4VHWnSziY3P2nE6Vb9pj71NPeVZ0o,15943
 universal_mcp/agents/shared/__main__.py,sha256=XxH5qGDpgFWfq7fwQfgKULXGiUgeTp_YKfcxftuVZq8,1452
 universal_mcp/agents/shared/prompts.py,sha256=yjP3zbbuKi87qCj21qwTTicz8TqtkKgnyGSeEjMu3ho,3761
 universal_mcp/agents/shared/tool_node.py,sha256=DC9F-Ri28Pam0u3sXWNODVgmj9PtAEUb5qP1qOoGgfs,9169
 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=c7OkZsO2fRtndgAzAQbKu-1xXRuRp9Kjgml57YD2NR4,9459
-universal_mcp_agents-0.1.15.dist-info/METADATA,sha256=8FNfgWbbFHVStTqIwzHZzohMEvvD27IuPvIUWRjea80,878
-universal_mcp_agents-0.1.15.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-universal_mcp_agents-0.1.15.dist-info/RECORD,,
+universal_mcp_agents-0.1.16.dist-info/METADATA,sha256=uIRCPa0Yr0Q1vmRnEuq3omcFLxX8fxE_mWy1WT7biIc,878
+universal_mcp_agents-0.1.16.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+universal_mcp_agents-0.1.16.dist-info/RECORD,,