quash-mcp 0.2.4__py3-none-any.whl → 0.2.6__py3-none-any.whl

quash_mcp/tools/execute_v3.py

@@ -7,12 +7,27 @@ 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 functions
+try:
+    from mahoraga.tools import Tools, describe_tools
+    from mahoraga.tools.adb import AdbTools as MahoragaAdbTools
+    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
+    MahoragaAdbTools = None
+    DEFAULT = None
+    async_to_sync = None
+
 
 async def execute_v3(
     task: str,
@@ -114,11 +129,46 @@ 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 and MahoragaAdbTools:
+        try:
+            # Create a mahoraga AdbTools instance for tool execution
+            # This instance has all the tool methods like swipe, start_app, etc.
+            mahoraga_tools = MahoragaAdbTools(
+                serial=state.device_serial,
+                use_tcp=True,
+                remote_tcp_port=8080
+            )
+
+            # Get all tool functions from mahoraga AdbTools instance
+            tool_list = describe_tools(mahoraga_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}")
+            import traceback
+            log_progress(f"Traceback: {traceback.format_exc()}")
+
     executor_locals = {}
 
     try:

{quash_mcp-0.2.4.dist-info → quash_mcp-0.2.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quash-mcp
-Version: 0.2.4
+Version: 0.2.6
 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.4.dist-info → quash_mcp-0.2.6.dist-info}/RECORD

@@ -14,10 +14,10 @@ quash_mcp/tools/configure.py,sha256=cv4RTolu6qae-XzyACSJUDrALfd0gYC-XE5s66_zfNk,
 quash_mcp/tools/connect.py,sha256=Kc7RGRUgtd2sR_bv6U4CB4kWSaLfsDc5kBo9u4FEjzs,4799
 quash_mcp/tools/execute.py,sha256=kR3VzIl31Lek-js4Hgxs-S_ls4YwKnbqkt79KFbvFuM,909
 quash_mcp/tools/execute_v2_backup.py,sha256=waWnaD0dEVcOJgRBbqZo3HnxME1s6YUOn8aRbm4R3X4,6081
-quash_mcp/tools/execute_v3.py,sha256=z8aNTyA9bdfc3Ub4LBd1HWaUWyTUEwsWc557USlHI9M,11089
+quash_mcp/tools/execute_v3.py,sha256=Pn506rAED6u6NNDq_M2pRKfSRhPIqhvDJwn_uYQCt5k,13329
 quash_mcp/tools/runsuite.py,sha256=gohLk9FpN8v7F0a69fspqOqUexTcslpYf3qU-iIZZ3s,7220
 quash_mcp/tools/usage.py,sha256=g76A6FO36fThoyRFG7q92QmS3Kh1pIKOrhYOzUdIubA,1155
-quash_mcp-0.2.4.dist-info/METADATA,sha256=Hhm57GjXV_rz9KOhTelfjZB3q41mtXGi3YKhcmzeUo4,8129
-quash_mcp-0.2.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-quash_mcp-0.2.4.dist-info/entry_points.txt,sha256=9sbDxrx0ApGDVRS-IE3mQgSao3DwKnnV_k-_ipFn9QI,52
-quash_mcp-0.2.4.dist-info/RECORD,,
+quash_mcp-0.2.6.dist-info/METADATA,sha256=UqoMmv6xb1hu_k_YLdpol8wh23k6eUoCfHaWrKbWARI,8129
+quash_mcp-0.2.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+quash_mcp-0.2.6.dist-info/entry_points.txt,sha256=9sbDxrx0ApGDVRS-IE3mQgSao3DwKnnV_k-_ipFn9QI,52
+quash_mcp-0.2.6.dist-info/RECORD,,