feat: enhance telemetry and tool management with improved file handling and boolean coercion

Author: Jordonh18

Server/src/core/telemetry.py

@@ -48,8 +48,17 @@ def get_package_version() -> str:
     except Exception:
         # Fallback for development: read from pyproject.toml
         try:
-            pyproject_path = Path(__file__).parent / "pyproject.toml"
-            with open(pyproject_path, "rb") as f:
+            pyproject_path: Path | None = None
+            current = Path(__file__).resolve()
+            for parent in current.parents:
+                candidate = parent / "pyproject.toml"
+                if candidate.exists():
+                    pyproject_path = candidate
+                    break
+            if pyproject_path is None:
+                raise FileNotFoundError("pyproject.toml not found in ancestors")
+
+            with pyproject_path.open("rb") as f:
                 data = tomli.load(f)
             return data["project"]["version"]
         except Exception:

Server/src/services/custom_tool_service.py

@@ -146,6 +146,11 @@ def _register_tool(self, project_id: str, definition: ToolDefinitionModel) -> No
         self._project_tools.setdefault(project_id, {})[
             definition.name] = definition
 
+    def get_project_id_for_hash(self, project_hash: str | None) -> str | None:
+        if not project_hash:
+            return None
+        return self._hash_to_project.get(project_hash.lower())
+
     async def _poll_until_complete(
         self,
         tool_name: str,
@@ -317,8 +322,16 @@ def resolve_project_id_for_unity_instance(unity_instance: str | None) -> str | N
             hash_part = unity_instance
 
         if hash_part:
-            # Return the hash directly as the identifier for WebSocket tools
-            return hash_part.lower()
+            lowered = hash_part.lower()
+            mapped: Optional[str] = None
+            try:
+                service = CustomToolService.get_instance()
+                mapped = service.get_project_id_for_hash(lowered)
+            except RuntimeError:
+                mapped = None
+            if mapped:
+                return mapped
+            return lowered
     except Exception:
         logger.debug(
             f"Failed to resolve project id via plugin hub for {unity_instance}")

Server/src/services/tools/manage_prefabs.py

@@ -29,21 +29,38 @@ async def manage_prefabs(
     # Get active instance from session state
     # Removed session_state import
     unity_instance = get_unity_instance_from_context(ctx)
+
+    def _coerce_bool(value, default=None):
+        if value is None:
+            return default
+        if isinstance(value, bool):
+            return value
+        if isinstance(value, str):
+            v = value.strip().lower()
+            if v in ("true", "1", "yes", "on"):
+                return True
+            if v in ("false", "0", "no", "off"):
+                return False
+        return bool(value)
+
     try:
         params: dict[str, Any] = {"action": action}
 
         if prefab_path:
             params["prefabPath"] = prefab_path
         if mode:
             params["mode"] = mode
-        if save_before_close is not None:
-            params["saveBeforeClose"] = bool(save_before_close)
+        save_before_close_val = _coerce_bool(save_before_close)
+        if save_before_close_val is not None:
+            params["saveBeforeClose"] = save_before_close_val
         if target:
             params["target"] = target
-        if allow_overwrite is not None:
-            params["allowOverwrite"] = bool(allow_overwrite)
-        if search_inactive is not None:
-            params["searchInactive"] = bool(search_inactive)
+        allow_overwrite_val = _coerce_bool(allow_overwrite)
+        if allow_overwrite_val is not None:
+            params["allowOverwrite"] = allow_overwrite_val
+        search_inactive_val = _coerce_bool(search_inactive)
+        if search_inactive_val is not None:
+            params["searchInactive"] = search_inactive_val
         response = await send_with_unity_instance(async_send_command_with_retry, unity_instance, "manage_prefabs", params)
 
         if isinstance(response, dict) and response.get("success"):

Server/src/services/tools/script_apply_edits.py

@@ -600,7 +600,7 @@ def error_with_hint(message: str, expected: dict[str, Any], suggestion: dict[str
         return _with_norm(resp_struct if isinstance(resp_struct, dict) else {"success": False, "message": str(resp_struct)}, normalized_for_echo, routing="structured")
 
     # 1) read from Unity
-    read_resp = async_send_command_with_retry("manage_script", {
+    read_resp = await async_send_command_with_retry("manage_script", {
         "action": "read",
         "name": name,
         "path": path,