universal-mcp-agents 0.1.22__tar.gz → 0.1.23rc1__tar.gz

{universal_mcp_agents-0.1.22 → universal_mcp_agents-0.1.23rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: universal-mcp-agents
-Version: 0.1.22
+Version: 0.1.23rc1
 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.22 → universal_mcp_agents-0.1.23rc1}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "universal-mcp-agents"
-version = "0.1.22"
+version = "0.1.23-rc1"
 description = "Add your description here"
 readme = "README.md"
 authors = [

{universal_mcp_agents-0.1.22 → universal_mcp_agents-0.1.23rc1}/src/universal_mcp/agents/codeact0/agent.py

@@ -39,7 +39,6 @@ class CodeActPlaybookAgent(BaseAgent):
         instructions: str,
         model: str,
         memory: BaseCheckpointSaver | None = None,
-        tools: ToolConfig | None = None,
         registry: ToolRegistry | None = None,
         playbook_registry: object | None = None,
         sandbox_timeout: int = 20,
@@ -54,10 +53,10 @@ class CodeActPlaybookAgent(BaseAgent):
         )
         self.model_instance = load_chat_model(model)
         self.playbook_model_instance = load_chat_model("azure/gpt-4.1")
-        self.tools_config = tools or {}
         self.registry = registry
         self.playbook_registry = playbook_registry
         self.playbook = playbook_registry.get_agent() if playbook_registry else None
+        self.tools_config = self.playbook.tools if self.playbook else {}
         self.eval_fn = eval_unsafe
         self.sandbox_timeout = sandbox_timeout
         self.default_tools = {
@@ -169,7 +168,7 @@ class CodeActPlaybookAgent(BaseAgent):
                 self.tools_config.extend(new_tool_ids)
                 self.exported_tools = await self.registry.export_tools(new_tool_ids, ToolFormat.LANGCHAIN)
                 self.final_instructions, self.tools_context = create_default_prompt(
-                    self.exported_tools, self.additional_tools, self.instructions, await get_connected_apps_string(self.registry)
+                    self.exported_tools, self.additional_tools, self.instructions, await get_connected_apps_string(self.registry), self.playbook
                 )
             if ask_user:
                 tool_messages.append(AIMessage(content=ai_msg))
@@ -226,23 +225,47 @@ class CodeActPlaybookAgent(BaseAgent):
 
                 t = user_text.lower()
                 if t == "yes, this is great":
-                    # Generate playbook metadata (name and description) before moving to generation
-                    meta_id = str(uuid.uuid4())
-                    writer({
-                        "type": "custom",
-                        id: meta_id,
-                        "name": "metadata",
-                        "data": {"update": bool(self.playbook)}
-                    })
-                    meta_instructions = self.instructions + PLAYBOOK_META_PROMPT
-                    messages = [{"role": "system", "content": meta_instructions}] + state["messages"]
-
-                    model_with_structured_output = self.playbook_model_instance.with_structured_output(PlaybookMeta)
-                    meta_response = model_with_structured_output.invoke(messages)
-                    meta = cast(PlaybookMeta, meta_response)
-
-                    writer({"type": "custom", id: meta_id, "name": "metadata", "data": {"name": meta.name, "description": meta.description}})
-                    return Command(goto="playbook", update={"playbook_mode": "generating", "playbook_name": meta.name, "playbook_description": meta.description})
+                    self.meta_id = str(uuid.uuid4())
+                    name, description = None, None
+                    if self.playbook:
+                        # Update flow: use existing name/description and do not re-generate
+                        name = getattr(self.playbook, "name", None)
+                        description = getattr(self.playbook, "description", None)
+                        writer({
+                            "type": "custom",
+                            id: self.meta_id,
+                            "name": "generating",
+                            "data": {
+                                "update": True,
+                                "name": name,
+                                "description": description,
+                            }
+                        })
+                    else:
+                        writer({
+                            "type": "custom",
+                            id: self.meta_id,
+                            "name": "generating",
+                            "data": {"update": False}
+                        })
+
+                        meta_instructions = self.instructions + PLAYBOOK_META_PROMPT
+                        messages = [{"role": "system", "content": meta_instructions}] + state["messages"]
+
+                        model_with_structured_output = self.playbook_model_instance.with_structured_output(PlaybookMeta)
+                        meta_response = model_with_structured_output.invoke(messages)
+                        meta = cast(PlaybookMeta, meta_response)
+                        name, description = meta.name, meta.description
+
+                        # Emit intermediary UI update with created name/description
+                        writer({
+                            "type": "custom",
+                            id: self.meta_id,
+                            "name": "generating",
+                            "data": {"update": False, "name": name, "description": description}
+                        })
+
+                    return Command(goto="playbook", update={"playbook_mode": "generating", "playbook_name": name, "playbook_description": description})
                 if t == "i would like to modify the plan":
                     prompt_ai = AIMessage(content="What would you like to change about the plan? Let me know and I'll update the plan accordingly.", additional_kwargs={"stream": "true"})
                     return Command(update={"playbook_mode": "planning", "messages": [prompt_ai]})
@@ -253,13 +276,6 @@ class CodeActPlaybookAgent(BaseAgent):
                 return Command(goto="call_model", update={"playbook_mode": "inactive"})
 
             elif playbook_mode == "generating":
-                generate_id = str(uuid.uuid4())
-                writer({
-                    "type": "custom",
-                    id: generate_id,
-                    "name": "generating",
-                    "data": {"update": bool(self.playbook)}
-                })
                 generating_instructions = self.instructions + PLAYBOOK_GENERATING_PROMPT
                 messages = [{"role": "system", "content": generating_instructions}] + state["messages"]
                 
@@ -304,11 +320,25 @@ class CodeActPlaybookAgent(BaseAgent):
 
                 writer({
                     "type": "custom",
-                    id: generate_id,
+                    id: self.meta_id,
                     "name": "generating",
-                    "data": {"id": str(res.id), "update": bool(self.playbook)}
+                    "data": {
+                        "id": str(res.id),
+                        "update": bool(self.playbook),
+                        "name": final_name,
+                        "description": final_description,
+                    }
                 })
-                mock_assistant_message = AIMessage(content=json.dumps(response.dict()), additional_kwargs={"type": "generating", "id": str(res.id), "update": bool(self.playbook)})
+                mock_assistant_message = AIMessage(
+                    content=json.dumps(response.dict()),
+                    additional_kwargs={
+                        "type": "generating",
+                        "id": str(res.id),
+                        "update": bool(self.playbook),
+                        "name": final_name,
+                        "description": final_description,
+                    },
+                )
 
                 return Command(
                     update={"messages": [mock_assistant_message], "playbook_mode": "normal"}
@@ -320,7 +350,7 @@ class CodeActPlaybookAgent(BaseAgent):
             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, await get_connected_apps_string(self.registry)
+                self.exported_tools, self.additional_tools, self.instructions, await get_connected_apps_string(self.registry), self.playbook
             )
             if state.get("playbook_mode") in ["planning", "confirming", "generating"]:
                 return "playbook"

{universal_mcp_agents-0.1.22 → universal_mcp_agents-0.1.23rc1}/src/universal_mcp/agents/codeact0/tools.py

@@ -14,12 +14,6 @@ def enter_playbook_mode():
     """Call this function to enter playbook mode. Playbook mode is when the user wants to store a repeated task as a script with some inputs for the future."""
     return
 
-
-def exit_playbook_mode():
-    """Call this function to exit playbook mode. Playbook mode is when the user wants to store a repeated task as a script with some inputs for the future."""
-    return
-
-
 def create_meta_tools(tool_registry: AgentrRegistry) -> dict[str, Any]:
     """Create the meta tools for searching and loading tools"""
 

{universal_mcp_agents-0.1.22 → universal_mcp_agents-0.1.23rc1}/uv.lock

@@ -4976,7 +4976,7 @@ wheels = [
 
 [[package]]
 name = "universal-mcp-agents"
-version = "0.1.21"
+version = "0.1.22"
 source = { editable = "." }
 dependencies = [
     { name = "langchain-anthropic" },

universal_mcp_agents-0.1.22/bech.py

@@ -1,38 +0,0 @@
-from universal_mcp.agentr.registry import AgentrRegistry
-from universal_mcp.agents import get_agent
-from langgraph.checkpoint.memory import MemorySaver
-from universal_mcp.agents.utils import messages_to_list
-import time
-from loguru import logger
-
-
-async def main():
-    start_time = time.time()
-    memory = MemorySaver()
-    logger.info(f"Checkpointer: Time consumed: {time.time() - start_time}")
-    agent_cls = get_agent("codeact-repl")
-    logger.info(f"Get class: Time consumed: {time.time() - start_time}")
-    registry = AgentrRegistry()
-    logger.info(f"Init Registry: Time consumed: {time.time() - start_time}")
-    agent = agent_cls(
-        name="CodeAct Agent",
-        instructions="Be very concise in your answers.",
-        model="anthropic:claude-4-sonnet-20250514",
-        tools={},
-        registry=registry,
-        memory=memory,
-    )
-    logger.info(f"Create agent: Time consumed: {time.time() - start_time}")
-    print("Init agent...")
-    await agent.ainit()
-    logger.info(f"Init Agent: Time consumed: {time.time() - start_time}")
-    print("Starting agent...")
-    async for e in agent.stream(user_input="hi"):
-        logger.info(f"First token: Time consumed: {time.time() - start_time}")
-        print(e)
-
-
-if __name__ == "__main__":
-    import asyncio
-
-    asyncio.run(main())