quash-mcp 0.2.3__tar.gz → 0.2.5__tar.gz

{quash_mcp-0.2.3 → quash_mcp-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quash-mcp
-Version: 0.2.3
+Version: 0.2.5
 Summary: Model Context Protocol server for Quash - AI-powered mobile automation agent
 Project-URL: Homepage, https://quashbugs.com
 Project-URL: Repository, https://github.com/quash/quash-mcp

{quash_mcp-0.2.3 → quash_mcp-0.2.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "quash-mcp"
-version = "0.2.3"
+version = "0.2.5"
 description = "Model Context Protocol server for Quash - AI-powered mobile automation agent"
 readme = "README.md"
 license = {text = "MIT"}

{quash_mcp-0.2.3 → quash_mcp-0.2.5}/quash_mcp/backend_client.py

@@ -245,19 +245,20 @@ class BackendClient:
             # Convert to JSON string
             data_json = json.dumps(data_dict)
 
-            # Prepare files dict
-            files = {
-                "data": ("data.json", data_json, "application/json")
-            }
+            # Prepare form data (data field as string)
+            form_data = {"data": data_json}
 
-            # Add screenshot if provided
+            # Prepare files dict (only screenshot if provided)
+            files = {}
             if screenshot_bytes:
                 files["screenshot"] = ("screenshot.png", screenshot_bytes, "image/png")
 
             async with httpx.AsyncClient(timeout=self.timeout) as client:
+                # Send both form data and files (multipart/form-data)
                 response = await client.post(
                     f"{self.base_url}/api/agent/step",
-                    files=files
+                    data=form_data,
+                    files=files if files else None
                 )
 
                 if response.status_code == 200:

{quash_mcp-0.2.3 → quash_mcp-0.2.5}/quash_mcp/tools/execute_v3.py

@@ -7,12 +7,25 @@ This hybrid approach keeps proprietary code private while allowing local device
 
 import time
 import uuid
+import asyncio
 from typing import Dict, Any, Callable, Optional
 from ..state import get_state
 from ..backend_client import get_backend_client
 from ..device.state_capture import get_device_state
 from ..device.adb_tools import AdbTools
 
+# Import mahoraga components for tool descriptions
+try:
+    from mahoraga.tools import Tools, describe_tools
+    from mahoraga.agent.context.personas import DEFAULT
+    from mahoraga.agent.utils.async_utils import async_to_sync
+except ImportError as e:
+    print(f"Warning: Could not import mahoraga components: {e}")
+    Tools = None
+    describe_tools = None
+    DEFAULT = None
+    async_to_sync = None
+
 
 async def execute_v3(
     task: str,
@@ -114,11 +127,36 @@ async def execute_v3(
     # Initialize local ADB tools for code execution
     adb_tools = AdbTools(serial=state.device_serial, use_tcp=True)
 
-    # Code executor namespace
+    # Code executor namespace - add tool functions so generated code can call them
     executor_globals = {
         "__builtins__": __builtins__,
         "adb_tools": adb_tools
     }
+
+    # Add tool functions to executor namespace (like start_app, swipe, etc.)
+    if describe_tools and DEFAULT:
+        try:
+            # Get all tool functions from AdbTools instance
+            tool_list = describe_tools(adb_tools, exclude_tools=None)
+
+            # Filter by allowed tools from DEFAULT persona
+            allowed_tool_names = DEFAULT.allowed_tools if hasattr(DEFAULT, 'allowed_tools') else []
+            filtered_tools = {name: func for name, func in tool_list.items() if name in allowed_tool_names}
+
+            # Add each tool function to executor globals
+            for tool_name, tool_function in filtered_tools.items():
+                # Convert async functions to sync if needed
+                if asyncio.iscoroutinefunction(tool_function):
+                    if async_to_sync:
+                        tool_function = async_to_sync(tool_function)
+
+                # Add to globals so code can call it directly
+                executor_globals[tool_name] = tool_function
+
+            log_progress(f"🔧 Loaded {len(filtered_tools)} tool functions: {list(filtered_tools.keys())}")
+        except Exception as e:
+            log_progress(f"⚠️ Warning: Could not load tool functions: {e}")
+
     executor_locals = {}
 
     try: