fixed snapshot list, cleaned up getattr

Author: sid-rl

src/runloop_api_client/sdk/_async.py

@@ -217,7 +217,7 @@ async def list(
             extra_body=extra_body,
             timeout=timeout,
         )
-        return [AsyncDevbox(self._client, item.id) for item in getattr(page, "devboxes", [])]
+        return [AsyncDevbox(self._client, item.id) for item in page.devboxes]
 
 
 class AsyncSnapshotClient:
@@ -250,7 +250,7 @@ async def list(
             extra_body=extra_body,
             timeout=timeout,
         )
-        return [AsyncSnapshot(self._client, item.id) for item in getattr(page, "disk_snapshots", [])]
+        return [AsyncSnapshot(self._client, item.id) for item in page.snapshots]
 
     def from_id(self, snapshot_id: str) -> AsyncSnapshot:
         return AsyncSnapshot(self._client, snapshot_id)
@@ -329,7 +329,7 @@ async def list(
             extra_body=extra_body,
             timeout=timeout,
         )
-        return [AsyncBlueprint(self._client, item.id) for item in getattr(page, "blueprints", [])]
+        return [AsyncBlueprint(self._client, item.id) for item in page.blueprints]
 
 
 class AsyncStorageObjectClient:
@@ -378,7 +378,7 @@ async def list(
             extra_body=extra_body,
             timeout=timeout,
         )
-        return [AsyncStorageObject(self._client, item.id, upload_url=None) for item in getattr(page, "objects", [])]
+        return [AsyncStorageObject(self._client, item.id, upload_url=item.upload_url) for item in page.objects]
 
     async def upload_from_file(
         self,

src/runloop_api_client/sdk/_sync.py

@@ -224,7 +224,7 @@ def list(
             extra_body=extra_body,
             timeout=timeout,
         )
-        return [Devbox(self._client, item.id) for item in getattr(page, "devboxes", [])]
+        return [Devbox(self._client, item.id) for item in page.devboxes]
 
 
 class SnapshotClient:
@@ -257,7 +257,7 @@ def list(
             extra_body=extra_body,
             timeout=timeout,
         )
-        return [Snapshot(self._client, item.id) for item in getattr(page, "disk_snapshots", [])]
+        return [Snapshot(self._client, item.id) for item in page.snapshots]
 
     def from_id(self, snapshot_id: str) -> Snapshot:
         return Snapshot(self._client, snapshot_id)
@@ -336,7 +336,7 @@ def list(
             extra_body=extra_body,
             timeout=timeout,
         )
-        return [Blueprint(self._client, item.id) for item in getattr(page, "blueprints", [])]
+        return [Blueprint(self._client, item.id) for item in page.blueprints]
 
 
 class StorageObjectClient:
@@ -385,7 +385,7 @@ def list(
             extra_body=extra_body,
             timeout=timeout,
         )
-        return [StorageObject(self._client, item.id, upload_url=None) for item in getattr(page, "objects", [])]
+        return [StorageObject(self._client, item.id, upload_url=item.upload_url) for item in page.objects]
 
     def upload_from_file(
         self,

src/runloop_api_client/sdk/async_devbox.py

@@ -298,7 +298,7 @@ async def _stream_worker(
             stream = await stream_factory()
             async with stream:
                 async for chunk in stream:
-                    text = getattr(chunk, "output", "")
+                    text = chunk.output
                     for callback in callbacks:
                         try:
                             callback(text)

src/runloop_api_client/sdk/devbox.py

@@ -293,7 +293,7 @@ def worker() -> None:
                     for chunk in stream:
                         if stop_event.is_set():
                             break
-                        text = getattr(chunk, "output", "")
+                        text = chunk.output
                         for callback in callbacks:
                             try:
                                 callback(text)