fix: preserve result body in async FAILURE responses

Author: dm36

launch/model_endpoint.py

@@ -286,11 +286,12 @@ def get(self, timeout: Optional[float] = None) -> EndpointResponse:
                         status_code=async_response.get("status_code", None),
                     )
                 elif status == "FAILURE":
+                    result_body = async_response.get("result") or {}
                     return EndpointResponse(
                         client=self.client,
                         status=status,
-                        result_url=None,
-                        result=None,
+                        result_url=result_body.get("result_url", None),
+                        result=result_body.get("result", None),
                         traceback=async_response.get("traceback", None),
                         status_code=async_response.get("status_code", None),
                     )

tests/test_model_endpoint.py

@@ -75,3 +75,44 @@ def test_status_returns_updated_value(requests_mock):  # noqa: F811
         response=HTTPResponse(body=json.dumps(resp), status=200)
     )
     assert endpoint.status() == "SUCCESS"
+
+
+def test_endpoint_response_future_failure_preserves_result_body():
+    """FAILURE responses should expose the result body so callers can extract error details."""
+    mock_client = MagicMock()
+    mock_client._get_async_endpoint_response.return_value = {
+        "status": "FAILURE",
+        "result": {"result": '{"error": "Out of memory"}', "result_url": None},
+        "traceback": None,
+        "status_code": 500,
+    }
+
+    from launch.model_endpoint import EndpointResponseFuture
+
+    future = EndpointResponseFuture(mock_client, "test-endpoint", "task-123")
+    response = future.get()
+
+    assert response.status == "FAILURE"
+    assert response.status_code == 500
+    assert response.traceback is None
+    assert response.result == '{"error": "Out of memory"}'
+
+
+def test_endpoint_response_future_failure_no_result_body():
+    """FAILURE responses with no result body should still work (e.g. pod crash with no response)."""
+    mock_client = MagicMock()
+    mock_client._get_async_endpoint_response.return_value = {
+        "status": "FAILURE",
+        "result": None,
+        "traceback": None,
+        "status_code": 500,
+    }
+
+    from launch.model_endpoint import EndpointResponseFuture
+
+    future = EndpointResponseFuture(mock_client, "test-endpoint", "task-456")
+    response = future.get()
+
+    assert response.status == "FAILURE"
+    assert response.result is None
+    assert response.traceback is None