feat(api): add query params to tasks.list

allows listing tasks with filters on agent ID and name

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 34
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp%2Fagentex-sdk-733bcb26b2f53abaac835267d630274ca18eba7726350b65fa2a74cc490908c1.yml
-openapi_spec_hash: f9a141a82a0b19bde31d6ea48ae4d6d6
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp%2Fagentex-sdk-02d466207a009c07fd85804f8917889d1d5503dc600ff6d7fd78ef29cf9e93a1.yml
+openapi_spec_hash: e40b923e3cd395998d53a4d5830ab7e4
 config_hash: f6ec6016df1ff072b5b60cdf7b438361

api.md

@@ -47,7 +47,7 @@ from agentex.types import Task, TaskListResponse
 Methods:
 
 - <code title="get /tasks/{task_id}">client.tasks.<a href="./src/agentex/resources/tasks.py">retrieve</a>(task_id) -> <a href="./src/agentex/types/task.py">Task</a></code>
-- <code title="get /tasks">client.tasks.<a href="./src/agentex/resources/tasks.py">list</a>() -> <a href="./src/agentex/types/task_list_response.py">TaskListResponse</a></code>
+- <code title="get /tasks">client.tasks.<a href="./src/agentex/resources/tasks.py">list</a>(\*\*<a href="src/agentex/types/task_list_params.py">params</a>) -> <a href="./src/agentex/types/task_list_response.py">TaskListResponse</a></code>
 - <code title="delete /tasks/{task_id}">client.tasks.<a href="./src/agentex/resources/tasks.py">delete</a>(task_id) -> <a href="./src/agentex/types/shared/delete_response.py">DeleteResponse</a></code>
 - <code title="delete /tasks/name/{task_name}">client.tasks.<a href="./src/agentex/resources/tasks.py">delete_by_name</a>(task_name) -> <a href="./src/agentex/types/shared/delete_response.py">DeleteResponse</a></code>
 - <code title="get /tasks/name/{task_name}">client.tasks.<a href="./src/agentex/resources/tasks.py">retrieve_by_name</a>(task_name) -> <a href="./src/agentex/types/task.py">Task</a></code>

src/agentex/resources/tasks.py

@@ -2,9 +2,13 @@
 
 from __future__ import annotations
 
+from typing import Optional
+
 import httpx
 
+from ..types import task_list_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
 from .._response import (
@@ -78,18 +82,41 @@ def retrieve(
     def list(
         self,
         *,
+        agent_id: Optional[str] | NotGiven = NOT_GIVEN,
+        agent_name: Optional[str] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> TaskListResponse:
-        """List all tasks."""
+        """
+        List all tasks.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
         return self._get(
             "/tasks",
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "agent_id": agent_id,
+                        "agent_name": agent_name,
+                    },
+                    task_list_params.TaskListParams,
+                ),
             ),
             cast_to=TaskListResponse,
         )
@@ -320,18 +347,41 @@ async def retrieve(
     async def list(
         self,
         *,
+        agent_id: Optional[str] | NotGiven = NOT_GIVEN,
+        agent_name: Optional[str] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> TaskListResponse:
-        """List all tasks."""
+        """
+        List all tasks.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
         return await self._get(
             "/tasks",
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {
+                        "agent_id": agent_id,
+                        "agent_name": agent_name,
+                    },
+                    task_list_params.TaskListParams,
+                ),
             ),
             cast_to=TaskListResponse,
         )

src/agentex/types/__init__.py

@@ -19,6 +19,7 @@
 from .agent_rpc_params import AgentRpcParams as AgentRpcParams
 from .agent_rpc_result import AgentRpcResult as AgentRpcResult
 from .span_list_params import SpanListParams as SpanListParams
+from .task_list_params import TaskListParams as TaskListParams
 from .agent_list_params import AgentListParams as AgentListParams
 from .event_list_params import EventListParams as EventListParams
 from .state_list_params import StateListParams as StateListParams

src/agentex/types/task_list_params.py

@@ -0,0 +1,14 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import TypedDict
+
+__all__ = ["TaskListParams"]
+
+
+class TaskListParams(TypedDict, total=False):
+    agent_id: Optional[str]
+
+    agent_name: Optional[str]

tests/api_resources/test_tasks.py

@@ -66,6 +66,15 @@ def test_method_list(self, client: Agentex) -> None:
         task = client.tasks.list()
         assert_matches_type(TaskListResponse, task, path=["response"])
 
+    @pytest.mark.skip()
+    @parametrize
+    def test_method_list_with_all_params(self, client: Agentex) -> None:
+        task = client.tasks.list(
+            agent_id="agent_id",
+            agent_name="agent_name",
+        )
+        assert_matches_type(TaskListResponse, task, path=["response"])
+
     @pytest.mark.skip()
     @parametrize
     def test_raw_response_list(self, client: Agentex) -> None:
@@ -350,6 +359,15 @@ async def test_method_list(self, async_client: AsyncAgentex) -> None:
         task = await async_client.tasks.list()
         assert_matches_type(TaskListResponse, task, path=["response"])
 
+    @pytest.mark.skip()
+    @parametrize
+    async def test_method_list_with_all_params(self, async_client: AsyncAgentex) -> None:
+        task = await async_client.tasks.list(
+            agent_id="agent_id",
+            agent_name="agent_name",
+        )
+        assert_matches_type(TaskListResponse, task, path=["response"])
+
     @pytest.mark.skip()
     @parametrize
     async def test_raw_response_list(self, async_client: AsyncAgentex) -> None: