fix(blueprints): handle awaiting upload status

Author: jason-rl

src/runloop_api_client/resources/blueprints.py

@@ -371,7 +371,7 @@ def retrieve_blueprint() -> BlueprintView:
             )
 
         def is_done_building(blueprint: BlueprintView) -> bool:
-            return blueprint.status not in ["queued", "building", "provisioning"]
+            return blueprint.status not in ["queued", "building", "provisioning", "awaiting_upload"]
 
         blueprint = poll_until(retrieve_blueprint, is_done_building, polling_config)
 
@@ -481,7 +481,7 @@ def list(
 
           starting_after: Load the next page of data starting after the item with the given ID.
 
-          status: Filter by build status (queued, provisioning, building, failed, build_complete)
+          status: Filter by build status (queued, provisioning, building, awaiting_upload, failed, build_complete)
 
           extra_headers: Send extra headers
 
@@ -584,7 +584,7 @@ def list_public(
 
           starting_after: Load the next page of data starting after the item with the given ID.
 
-          status: Filter by build status (queued, provisioning, building, failed, build_complete)
+          status: Filter by build status (queued, provisioning, building, awaiting_upload, failed, build_complete)
 
           extra_headers: Send extra headers
 
@@ -1033,7 +1033,7 @@ async def retrieve_blueprint() -> BlueprintView:
             )
 
         def is_done_building(blueprint: BlueprintView) -> bool:
-            return blueprint.status not in ["queued", "building", "provisioning"]
+            return blueprint.status not in ["queued", "building", "provisioning", "awaiting_upload"]
 
         blueprint = await async_poll_until(retrieve_blueprint, is_done_building, polling_config)
 
@@ -1143,7 +1143,7 @@ def list(
 
           starting_after: Load the next page of data starting after the item with the given ID.
 
-          status: Filter by build status (queued, provisioning, building, failed, build_complete)
+          status: Filter by build status (queued, provisioning, building, awaiting_upload, failed, build_complete)
 
           extra_headers: Send extra headers
 
@@ -1246,7 +1246,7 @@ def list_public(
 
           starting_after: Load the next page of data starting after the item with the given ID.
 
-          status: Filter by build status (queued, provisioning, building, failed, build_complete)
+          status: Filter by build status (queued, provisioning, building, awaiting_upload, failed, build_complete)
 
           extra_headers: Send extra headers
 

src/runloop_api_client/types/blueprint_list_params.py

@@ -24,4 +24,4 @@ class BlueprintListParams(TypedDict, total=False):
     """Load the next page of data starting after the item with the given ID."""
 
     status: str
-    """Filter by build status (queued, provisioning, building, failed, build_complete)"""
+    """Filter by build status (queued, provisioning, building, awaiting_upload, failed, build_complete)"""

src/runloop_api_client/types/blueprint_list_public_params.py

@@ -24,4 +24,4 @@ class BlueprintListPublicParams(TypedDict, total=False):
     """Load the next page of data starting after the item with the given ID."""
 
     status: str
-    """Filter by build status (queued, provisioning, building, failed, build_complete)"""
+    """Filter by build status (queued, provisioning, building, awaiting_upload, failed, build_complete)"""

src/runloop_api_client/types/blueprint_view.py

@@ -63,7 +63,7 @@ class BlueprintView(BaseModel):
     state: Literal["created", "deleted"]
     """The state of the Blueprint."""
 
-    status: Literal["queued", "provisioning", "building", "failed", "build_complete"]
+    status: Literal["queued", "provisioning", "building", "awaiting_upload", "failed", "build_complete"]
     """The status of the Blueprint build."""
 
     base_blueprint_id: Optional[str] = None

tests/api_resources/test_blueprints.py

@@ -5,7 +5,9 @@
 import os
 from typing import Any, cast
 
+import httpx
 import pytest
+from respx import MockRouter
 
 from tests.utils import assert_matches_type
 from runloop_api_client import Runloop, AsyncRunloop
@@ -22,16 +24,40 @@
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
 
+def mock_register(respx_mock: MockRouter) -> None:
+    respx_mock.post("/v1/blueprints/register").mock(
+        return_value=httpx.Response(
+            200,
+            json={
+                "blueprint": {
+                    "id": "bpt_123",
+                    "name": "name",
+                    "status": "awaiting_upload",
+                    "state": "created",
+                    "create_time_ms": 0,
+                    "parameters": {"name": "name"},
+                },
+                "push_reference": "converter.runloop.ai/blueprints:bpt_123",
+            },
+        )
+    )
+
+
 class TestBlueprints:
     parametrize = pytest.mark.parametrize("client", [False, True], indirect=True, ids=["loose", "strict"])
 
     @parametrize
-    def test_method_register(self, client: Runloop) -> None:
+    @pytest.mark.respx(base_url=base_url)
+    def test_method_register(self, client: Runloop, respx_mock: MockRouter) -> None:
+        mock_register(respx_mock)
         upload = client.blueprints.register(name="name")
         assert_matches_type(BlueprintUploadView, upload, path=["response"])
+        assert upload.push_reference == "converter.runloop.ai/blueprints:bpt_123"
 
     @parametrize
-    def test_method_register_with_all_params(self, client: Runloop) -> None:
+    @pytest.mark.respx(base_url=base_url)
+    def test_method_register_with_all_params(self, client: Runloop, respx_mock: MockRouter) -> None:
+        mock_register(respx_mock)
         upload = client.blueprints.register(
             name="name",
             launch_parameters={"architecture": "x86_64"},
@@ -40,15 +66,19 @@ def test_method_register_with_all_params(self, client: Runloop) -> None:
         assert_matches_type(BlueprintUploadView, upload, path=["response"])
 
     @parametrize
-    def test_raw_response_register(self, client: Runloop) -> None:
+    @pytest.mark.respx(base_url=base_url)
+    def test_raw_response_register(self, client: Runloop, respx_mock: MockRouter) -> None:
+        mock_register(respx_mock)
         response = client.blueprints.with_raw_response.register(name="name")
         assert response.is_closed is True
         assert response.http_request.headers.get("X-Stainless-Lang") == "python"
         upload = response.parse()
         assert_matches_type(BlueprintUploadView, upload, path=["response"])
 
     @parametrize
-    def test_streaming_response_register(self, client: Runloop) -> None:
+    @pytest.mark.respx(base_url=base_url)
+    def test_streaming_response_register(self, client: Runloop, respx_mock: MockRouter) -> None:
+        mock_register(respx_mock)
         with client.blueprints.with_streaming_response.register(name="name") as response:
             assert not response.is_closed
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
@@ -492,31 +522,43 @@ class TestAsyncBlueprints:
     parametrize = pytest.mark.parametrize(
         "async_client", [False, True, {"http_client": "aiohttp"}], indirect=True, ids=["loose", "strict", "aiohttp"]
     )
+    register_parametrize = pytest.mark.parametrize(
+        "async_client", [False, True], indirect=True, ids=["loose", "strict"]
+    )
 
-    @parametrize
-    async def test_method_register(self, async_client: AsyncRunloop) -> None:
+    @register_parametrize
+    @pytest.mark.respx(base_url=base_url)
+    async def test_method_register(self, async_client: AsyncRunloop, respx_mock: MockRouter) -> None:
+        mock_register(respx_mock)
         upload = await async_client.blueprints.register(name="name")
         assert_matches_type(BlueprintUploadView, upload, path=["response"])
+        assert upload.push_reference == "converter.runloop.ai/blueprints:bpt_123"
 
-    @parametrize
-    async def test_method_register_with_all_params(self, async_client: AsyncRunloop) -> None:
+    @register_parametrize
+    @pytest.mark.respx(base_url=base_url)
+    async def test_method_register_with_all_params(self, async_client: AsyncRunloop, respx_mock: MockRouter) -> None:
+        mock_register(respx_mock)
         upload = await async_client.blueprints.register(
             name="name",
             launch_parameters={"architecture": "x86_64"},
             metadata={"source": "upload"},
         )
         assert_matches_type(BlueprintUploadView, upload, path=["response"])
 
-    @parametrize
-    async def test_raw_response_register(self, async_client: AsyncRunloop) -> None:
+    @register_parametrize
+    @pytest.mark.respx(base_url=base_url)
+    async def test_raw_response_register(self, async_client: AsyncRunloop, respx_mock: MockRouter) -> None:
+        mock_register(respx_mock)
         response = await async_client.blueprints.with_raw_response.register(name="name")
         assert response.is_closed is True
         assert response.http_request.headers.get("X-Stainless-Lang") == "python"
         upload = await response.parse()
         assert_matches_type(BlueprintUploadView, upload, path=["response"])
 
-    @parametrize
-    async def test_streaming_response_register(self, async_client: AsyncRunloop) -> None:
+    @register_parametrize
+    @pytest.mark.respx(base_url=base_url)
+    async def test_streaming_response_register(self, async_client: AsyncRunloop, respx_mock: MockRouter) -> None:
+        mock_register(respx_mock)
         async with async_client.blueprints.with_streaming_response.register(name="name") as response:
             assert not response.is_closed
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"