refactor: update resource function return types to include MCPResponse union

Author: msanatan

MCPForUnity/UnityMcpServer~/src/resources/active_tool.py

@@ -31,7 +31,7 @@ class ActiveToolResponse(MCPResponse):
     name="editor_active_tool",
     description="Currently active editor tool (Move, Rotate, Scale, etc.) and transform handle settings."
 )
-async def get_active_tool() -> ActiveToolResponse:
+async def get_active_tool() -> ActiveToolResponse | MCPResponse:
     """Get active editor tool information."""
     response = await async_send_command_with_retry("get_active_tool", {})
     return ActiveToolResponse(**response) if isinstance(response, dict) else response

MCPForUnity/UnityMcpServer~/src/resources/editor_state.py

@@ -26,7 +26,7 @@ class EditorStateResponse(MCPResponse):
     name="editor_state",
     description="Current editor runtime state including play mode, compilation status, active scene, and selection summary. Refresh frequently for up-to-date information."
 )
-async def get_editor_state() -> EditorStateResponse:
+async def get_editor_state() -> EditorStateResponse | MCPResponse:
     """Get current editor runtime state."""
     response = await async_send_command_with_retry("get_editor_state", {})
     return EditorStateResponse(**response) if isinstance(response, dict) else response

MCPForUnity/UnityMcpServer~/src/resources/layers.py

@@ -13,7 +13,7 @@ class LayersResponse(MCPResponse):
     name="project_layers",
     description="All layers defined in the project's TagManager with their indices (0-31). Read this before using add_layer or remove_layer tools."
 )
-async def get_layers() -> LayersResponse:
+async def get_layers() -> LayersResponse | MCPResponse:
     """Get all project layers with their indices."""
     response = await async_send_command_with_retry("get_layers", {})
     return LayersResponse(**response) if isinstance(response, dict) else response

MCPForUnity/UnityMcpServer~/src/resources/menu_items.py

@@ -15,7 +15,7 @@ class GetMenuItemsResponse(MCPResponse):
     name="menu_items",
     description="Provides a list of all menu items."
 )
-async def get_menu_items(ctx: Context) -> GetMenuItemsResponse:
+async def get_menu_items(ctx: Context) -> GetMenuItemsResponse | MCPResponse:
     """Provides a list of all menu items.
     """
     unity_instance = get_unity_instance_from_context(ctx)

MCPForUnity/UnityMcpServer~/src/resources/prefab_stage.py

@@ -23,7 +23,7 @@ class PrefabStageResponse(MCPResponse):
     name="editor_prefab_stage",
     description="Current prefab editing context if a prefab is open in isolation mode. Returns isOpen=false if no prefab is being edited."
 )
-async def get_prefab_stage() -> PrefabStageResponse:
+async def get_prefab_stage() -> PrefabStageResponse | MCPResponse:
     """Get current prefab stage information."""
     response = await async_send_command_with_retry("get_prefab_stage", {})
     return PrefabStageResponse(**response) if isinstance(response, dict) else response

MCPForUnity/UnityMcpServer~/src/resources/project_info.py

@@ -23,7 +23,7 @@ class ProjectInfoResponse(MCPResponse):
     name="project_info",
     description="Static project information including root path, Unity version, and platform. This data rarely changes."
 )
-async def get_project_info() -> ProjectInfoResponse:
+async def get_project_info() -> ProjectInfoResponse | MCPResponse:
     """Get static project configuration information."""
     response = await async_send_command_with_retry("get_project_info", {})
     return ProjectInfoResponse(**response) if isinstance(response, dict) else response

MCPForUnity/UnityMcpServer~/src/resources/selection.py

@@ -39,7 +39,7 @@ class SelectionResponse(MCPResponse):
     name="editor_selection",
     description="Detailed information about currently selected objects in the editor, including GameObjects, assets, and their properties."
 )
-async def get_selection() -> SelectionResponse:
+async def get_selection() -> SelectionResponse | MCPResponse:
     """Get detailed editor selection information."""
     response = await async_send_command_with_retry("get_selection", {})
     return SelectionResponse(**response) if isinstance(response, dict) else response

MCPForUnity/UnityMcpServer~/src/resources/tags.py

@@ -14,7 +14,7 @@ class TagsResponse(MCPResponse):
     name="project_tags",
     description="All tags defined in the project's TagManager. Read this before using add_tag or remove_tag tools."
 )
-async def get_tags() -> TagsResponse:
+async def get_tags() -> TagsResponse | MCPResponse:
     """Get all project tags."""
     response = await async_send_command_with_retry("get_tags", {})
     return TagsResponse(**response) if isinstance(response, dict) else response

MCPForUnity/UnityMcpServer~/src/resources/tests.py

@@ -21,7 +21,7 @@ class GetTestsResponse(MCPResponse):
 
 
 @mcp_for_unity_resource(uri="mcpforunity://tests", name="get_tests", description="Provides a list of all tests.")
-async def get_tests(ctx: Context) -> GetTestsResponse:
+async def get_tests(ctx: Context) -> GetTestsResponse | MCPResponse:
     """Provides a list of all tests.
     """
     unity_instance = get_unity_instance_from_context(ctx)
@@ -38,7 +38,7 @@ async def get_tests(ctx: Context) -> GetTestsResponse:
 async def get_tests_for_mode(
     ctx: Context,
     mode: Annotated[Literal["EditMode", "PlayMode"], Field(description="The mode to filter tests by.")],
-) -> GetTestsResponse:
+) -> GetTestsResponse | MCPResponse:
     """Provides a list of tests for a specific mode.
 
     Args:

MCPForUnity/UnityMcpServer~/src/resources/windows.py

@@ -31,7 +31,7 @@ class WindowsResponse(MCPResponse):
     name="editor_windows",
     description="All currently open editor windows with their titles, types, positions, and focus state."
 )
-async def get_windows() -> WindowsResponse:
+async def get_windows() -> WindowsResponse | MCPResponse:
     """Get all open editor windows."""
     response = await async_send_command_with_retry("get_windows", {})
     return WindowsResponse(**response) if isinstance(response, dict) else response