persist additional state in the Agent and AsyncAgent objects

Author: jrvb-rl

src/runloop_api_client/sdk/agent.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+from typing import Optional
 from typing_extensions import Unpack, override
 
 from ._types import (
@@ -18,6 +19,7 @@ def __init__(
         self,
         client: Runloop,
         agent_id: str,
+        agent_view: Optional[AgentView] = None,
     ) -> None:
         """Initialize the wrapper.
 
@@ -28,6 +30,7 @@ def __init__(
         """
         self._client = client
         self._id = agent_id
+        self._agent_view = agent_view
 
     @override
     def __repr__(self) -> str:
@@ -52,7 +55,9 @@ def get_info(
         :return: Agent details
         :rtype: AgentView
         """
-        return self._client.agents.retrieve(
-            self._id,
-            **options,
-        )
+        if self._agent_view is None:
+            self._agent_view = self._client.agents.retrieve(
+                self._id,
+                **options,
+            )
+        return self._agent_view

src/runloop_api_client/sdk/async_.py

@@ -159,7 +159,7 @@ async def list(
         page = await self._client.devboxes.list(
             **params,
         )
-        return [AsyncDevbox(self._client, item.id) for item in page.devboxes]
+        return [AsyncDevbox(self._client, item.id, item) for item in page.devboxes]
 
 
 class AsyncSnapshotOps:

src/runloop_api_client/sdk/async_agent.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+from typing import Optional
 from typing_extensions import Unpack, override
 
 from ._types import (
@@ -18,6 +19,7 @@ def __init__(
         self,
         client: AsyncRunloop,
         agent_id: str,
+        agent_view: Optional[AgentView] = None,
     ) -> None:
         """Initialize the wrapper.
 
@@ -28,6 +30,7 @@ def __init__(
         """
         self._client = client
         self._id = agent_id
+        self._agent_view = agent_view
 
     @override
     def __repr__(self) -> str:
@@ -56,3 +59,9 @@ async def get_info(
             self._id,
             **options,
         )
+        if self._agent_view is None:
+            self._agent_view = self._client.agents.retrieve(
+                self._id,
+                **options,
+            )
+        return self._agent_view

src/runloop_api_client/sdk/sync.py

@@ -599,7 +599,7 @@ def list(
         page = self._client.agents.list(
             **params,
         )
-        return [Agent(self._client, item.id) for item in page.agents]
+        return [Agent(self._client, item.id, item) for item in page.agents]
 
 
 class RunloopSDK: