chore(internal): add --fix argument to lint script

Author: stainless-app[bot]

scripts/lint

@@ -4,6 +4,12 @@ set -e
 
 cd "$(dirname "$0")/.."
 
+echo "==> Running pyright"
+uv run pyright
+
+echo "==> Running mypy"
+uv run mypy .
+
 echo "==> Running lints"
 uv run ruff check . "$@"
 

src/runloop_api_client/_base_client.py

@@ -1303,21 +1303,21 @@ def __init__(self, **kwargs: Any) -> None:
 
 
 try:
-    import httpx_aiohttp
+    import httpx_aiohttp  # type: ignore[import-not-found]
 except ImportError:
 
     class _DefaultAioHttpClient(httpx.AsyncClient):
         def __init__(self, **_kwargs: Any) -> None:
             raise RuntimeError("To use the aiohttp client you must have installed the package with the `aiohttp` extra")
 else:
 
-    class _DefaultAioHttpClient(httpx_aiohttp.HttpxAiohttpClient):  # type: ignore
+    class _DefaultAioHttpClient(httpx_aiohttp.HttpxAiohttpClient):  # type: ignore[misc, no-redef]
         def __init__(self, **kwargs: Any) -> None:
             kwargs.setdefault("timeout", DEFAULT_TIMEOUT)
             kwargs.setdefault("limits", DEFAULT_CONNECTION_LIMITS)
             kwargs.setdefault("follow_redirects", True)
 
-            super().__init__(**kwargs)
+            super().__init__(**kwargs)  # type: ignore[unknown-call]
 
 
 if TYPE_CHECKING:

src/runloop_api_client/sdk/_types.py

@@ -75,7 +75,7 @@ class PollingRequestOptions(BaseRequestOptions, total=False):
     """Configuration for polling behavior"""
 
 
-class LongPollingRequestOptions(LongRequestOptions, PollingRequestOptions):
+class LongPollingRequestOptions(LongRequestOptions, PollingRequestOptions):  # type: ignore[misc]
     pass
 
 

src/runloop_api_client/sdk/agent.py

@@ -64,9 +64,7 @@ def get_info(
         :return: Agent details
         :rtype: AgentView
         """
-        if self._agent_view is None:
-            self._agent_view = self._client.agents.retrieve(
-                self._id,
-                **options,
-            )
-        return self._agent_view
+        return self._client.agents.retrieve(
+            self._id,
+            **options,
+        )

src/runloop_api_client/sdk/async_agent.py

@@ -68,9 +68,3 @@ 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/devbox.py

@@ -446,10 +446,8 @@ def exec(
             execution.execution_id,
             **filter_params(params, ExecuteStreamingCallbacks),
         )
-        final = execution
-        if execution.status == "completed":
-            final: DevboxAsyncExecutionDetailView = execution
-        else:
+        final: DevboxAsyncExecutionDetailView = execution
+        if execution.status != "completed":
             final = client.devboxes.executions.await_completed(
                 execution.execution_id,
                 devbox_id=devbox.id,

tests/sdk/conftest.py

@@ -1,4 +1,5 @@
 """Shared fixtures and utilities for SDK tests."""
+# pyright: reportUnknownVariableType=false
 
 from __future__ import annotations
 

tests/sdk/test_ops.py

@@ -769,6 +769,17 @@ def test_list(self, mock_client: Mock) -> None:
         page = SimpleNamespace(agents=[agent_view_1, agent_view_2, agent_view_3])
         mock_client.agents.list.return_value = page
 
+        # Mock retrieve to return the corresponding agent_view when called
+        def mock_retrieve(agent_id: str, **_kwargs: object) -> MockAgentView:
+            agent_views = {
+                "agent_001": agent_view_1,
+                "agent_002": agent_view_2,
+                "agent_003": agent_view_3,
+            }
+            return agent_views[agent_id]
+
+        mock_client.agents.retrieve.side_effect = mock_retrieve
+
         client = AgentOps(mock_client)
         agents = client.list(
             limit=10,
@@ -784,32 +795,32 @@ def test_list(self, mock_client: Mock) -> None:
         assert agents[1].id == "agent_002"
         assert agents[2].id == "agent_003"
 
-        # Test that get_info() retrieves the cached AgentView for the first agent
+        # Test that get_info() retrieves the AgentView for the first agent
         info = agents[0].get_info()
         assert info.id == "agent_001"
         assert info.name == "first-agent"
         assert info.create_time_ms == 1234567890000
         assert info.is_public is False
         assert info.source is None
 
-        # Test that get_info() retrieves the cached AgentView for the second agent
+        # Test that get_info() retrieves the AgentView for the second agent
         info = agents[1].get_info()
         assert info.id == "agent_002"
         assert info.name == "second-agent"
         assert info.create_time_ms == 1234567891000
         assert info.is_public is True
         assert info.source == {"type": "git", "git": {"repository": "https://github.com/example/repo"}}
 
-        # Test that get_info() retrieves the cached AgentView for the third agent
+        # Test that get_info() retrieves the AgentView for the third agent
         info = agents[2].get_info()
         assert info.id == "agent_003"
         assert info.name == "third-agent"
         assert info.create_time_ms == 1234567892000
         assert info.is_public is False
         assert info.source == {"type": "npm", "npm": {"package_name": "example-package"}}
 
-        # Verify that agents.retrieve was NOT called (because we're using cached data)
-        mock_client.agents.retrieve.assert_not_called()
+        # Verify that agents.retrieve was called for each agent
+        assert mock_client.agents.retrieve.call_count == 3
 
         mock_client.agents.list.assert_called_once()