universal-mcp-agents 0.1.22__py3-none-any.whl → 0.1.23rc1__py3-none-any.whl

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/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.dist-info → universal_mcp_agents-0.1.23rc1.dist-info}/METADATA

@@ -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.dist-info → universal_mcp_agents-0.1.23rc1.dist-info}/RECORD

@@ -22,14 +22,14 @@ universal_mcp/agents/builder/prompts.py,sha256=8Xs6uzTUHguDRngVMLak3lkXFkk2VV_uQ
 universal_mcp/agents/builder/state.py,sha256=7DeWllxfN-yD6cd9wJ3KIgjO8TctkJvVjAbZT8W_zqk,922
 universal_mcp/agents/codeact0/__init__.py,sha256=8-fvUo1Sm6dURGI-lW-X3Kd78LqySYbb5NMkNJ4NDwg,76
 universal_mcp/agents/codeact0/__main__.py,sha256=_7qSz97YnRgYJTESkALS5_eBIGHiMjA5rhr3IAeBvVo,896
-universal_mcp/agents/codeact0/agent.py,sha256=Cxdh9eJvrIfUISdFQKXnYrGc6PePrIGusSvWiALmAJo,16634
+universal_mcp/agents/codeact0/agent.py,sha256=--cAjQ05VntXga2ofpnL9oFXAk6ANZ2SftwtLkQgy00,17844
 universal_mcp/agents/codeact0/config.py,sha256=H-1woj_nhSDwf15F63WYn723y4qlRefXzGxuH81uYF0,2215
 universal_mcp/agents/codeact0/langgraph_agent.py,sha256=8nz2wq-LexImx-l1y9_f81fK72IQetnCeljwgnduNGY,420
 universal_mcp/agents/codeact0/llm_tool.py,sha256=-pAz04OrbZ_dJ2ueysT1qZd02DrbLY4EbU0tiuF_UNU,798
 universal_mcp/agents/codeact0/prompts.py,sha256=Gk6WTx2X7IOlbC3wYtGwKWDoeUtgpkMeiyfhj6LEq2I,11221
 universal_mcp/agents/codeact0/sandbox.py,sha256=Xw4tbUV_6haYIZZvteJi6lIYsW6ni_3DCRCOkslTKgM,4459
 universal_mcp/agents/codeact0/state.py,sha256=co3BZBuMIt1FP2qzgsbsLkyKbddCG1ieKyAw9TAskSU,1944
-universal_mcp/agents/codeact0/tools.py,sha256=WRJKNIf6_cP_enElboEyVd-aHF-kk6aq8Bg0biZijR4,13363
+universal_mcp/agents/codeact0/tools.py,sha256=1EBStJQQQCuNeWOqt0VP-XFiT0fE4oCR9nvUiwzRhe4,13164
 universal_mcp/agents/codeact0/utils.py,sha256=PgisAxmqYIzimc4lFA1nV-R7gxkICIrtO1q0FQZ-UoY,17580
 universal_mcp/agents/shared/__main__.py,sha256=XxH5qGDpgFWfq7fwQfgKULXGiUgeTp_YKfcxftuVZq8,1452
 universal_mcp/agents/shared/prompts.py,sha256=yjP3zbbuKi87qCj21qwTTicz8TqtkKgnyGSeEjMu3ho,3761
@@ -39,6 +39,6 @@ universal_mcp/applications/filesystem/app.py,sha256=0TRjjm8YnslVRSmfkXI7qQOAlqWl
 universal_mcp/applications/llm/__init__.py,sha256=_XGRxN3O1--ZS5joAsPf8IlI9Qa6negsJrwJ5VJXno0,46
 universal_mcp/applications/llm/app.py,sha256=g9mK-luOLUshZzBGyQZMOHBeCSXmh2kCKir40YnsGUo,12727
 universal_mcp/applications/ui/app.py,sha256=c7OkZsO2fRtndgAzAQbKu-1xXRuRp9Kjgml57YD2NR4,9459
-universal_mcp_agents-0.1.22.dist-info/METADATA,sha256=6f34bvvybyI6et8rvKIu-4BtFJHGmt0YZmaZutJBXuw,878
-universal_mcp_agents-0.1.22.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-universal_mcp_agents-0.1.22.dist-info/RECORD,,
+universal_mcp_agents-0.1.23rc1.dist-info/METADATA,sha256=Fa-ACWH1oawoVDBAbp2iSSDGSCRovD2rWP9e9lUhcYE,881
+universal_mcp_agents-0.1.23rc1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+universal_mcp_agents-0.1.23rc1.dist-info/RECORD,,