agentex-sdk 0.2.5__tar.gz → 0.2.6__tar.gz

agentex_sdk-0.2.6/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "0.2.6"
+}

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/CHANGELOG.md

@@ -1,5 +1,14 @@
 # Changelog
 
+## 0.2.6 (2025-08-01)
+
+Full Changelog: [v0.2.5...v0.2.6](https://github.com/scaleapi/agentex-python/compare/v0.2.5...v0.2.6)
+
+### Features
+
+* **api:** add query params to tasks.list ([d4902d5](https://github.com/scaleapi/agentex-python/commit/d4902d52caf82e2f57d1bbf19527cdc1448ed397))
+* **client:** support file upload requests ([e004b30](https://github.com/scaleapi/agentex-python/commit/e004b304c22286151330c2200bcb85046a7ac111))
+
 ## 0.2.5 (2025-07-30)
 
 Full Changelog: [v0.2.4...v0.2.5](https://github.com/scaleapi/agentex-python/compare/v0.2.4...v0.2.5)

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentex-sdk
-Version: 0.2.5
+Version: 0.2.6
 Summary: The official Python library for the agentex API
 Project-URL: Homepage, https://github.com/scaleapi/agentex-python
 Project-URL: Repository, https://github.com/scaleapi/agentex-python

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/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>

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentex-sdk"
-version = "0.2.5"
+version = "0.2.6"
 description = "The official Python library for the agentex API"
 dynamic = ["readme"]
 license = "Apache-2.0"

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/_base_client.py

@@ -532,7 +532,10 @@ class BaseClient(Generic[_HttpxClientT, _DefaultStreamT]):
         is_body_allowed = options.method.lower() != "get"
 
         if is_body_allowed:
-            kwargs["json"] = json_data if is_given(json_data) else None
+            if isinstance(json_data, bytes):
+                kwargs["content"] = json_data
+            else:
+                kwargs["json"] = json_data if is_given(json_data) else None
             kwargs["files"] = files
         else:
             headers.pop("Content-Type", None)

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/_files.py

@@ -69,12 +69,12 @@ def _transform_file(file: FileTypes) -> HttpxFileTypes:
         return file
 
     if is_tuple_t(file):
-        return (file[0], _read_file_content(file[1]), *file[2:])
+        return (file[0], read_file_content(file[1]), *file[2:])
 
     raise TypeError(f"Expected file types input to be a FileContent type or to be a tuple")
 
 
-def _read_file_content(file: FileContent) -> HttpxFileContent:
+def read_file_content(file: FileContent) -> HttpxFileContent:
     if isinstance(file, os.PathLike):
         return pathlib.Path(file).read_bytes()
     return file
@@ -111,12 +111,12 @@ async def _async_transform_file(file: FileTypes) -> HttpxFileTypes:
         return file
 
     if is_tuple_t(file):
-        return (file[0], await _async_read_file_content(file[1]), *file[2:])
+        return (file[0], await async_read_file_content(file[1]), *file[2:])
 
     raise TypeError(f"Expected file types input to be a FileContent type or to be a tuple")
 
 
-async def _async_read_file_content(file: FileContent) -> HttpxFileContent:
+async def async_read_file_content(file: FileContent) -> HttpxFileContent:
     if isinstance(file, os.PathLike):
         return await anyio.Path(file).read_bytes()
 

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "agentex"
-__version__ = "0.2.5"  # x-release-please-version
+__version__ = "0.2.6"  # x-release-please-version

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/adk/_modules/acp.py

@@ -4,7 +4,7 @@ from typing import Any
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
-from agentex.lib.adk.utils._modules.client import get_async_agentex_client
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from agentex.lib.core.services.adk.acp.acp import ACPService
 from agentex.lib.core.temporal.activities.activity_helpers import ActivityHelpers
 from agentex.lib.core.temporal.activities.adk.acp.acp_activities import (
@@ -41,7 +41,7 @@ class ACPModule:
             acp_activities (Optional[ACPActivities]): Optional pre-configured ACP activities. If None, will be auto-initialized.
         """
         if acp_service is None:
-            agentex_client = get_async_agentex_client()
+            agentex_client = create_async_agentex_client()
             tracer = AsyncTracer(agentex_client)
             self._acp_service = ACPService(agentex_client=agentex_client, tracer=tracer)
         else:

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/adk/_modules/agent_task_tracker.py

@@ -3,7 +3,7 @@ from datetime import timedelta
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
-from agentex.lib.adk.utils._modules.client import get_async_agentex_client
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from agentex.lib.core.services.adk.agent_task_tracker import AgentTaskTrackerService
 from agentex.lib.core.temporal.activities.activity_helpers import ActivityHelpers
 from agentex.lib.core.temporal.activities.adk.agent_task_tracker_activities import (
@@ -34,7 +34,7 @@ class AgentTaskTrackerModule:
         agent_task_tracker_service: AgentTaskTrackerService | None = None,
     ):
         if agent_task_tracker_service is None:
-            agentex_client = get_async_agentex_client()
+            agentex_client = create_async_agentex_client()
             tracer = AsyncTracer(agentex_client)
             self._agent_task_tracker_service = AgentTaskTrackerService(
                 agentex_client=agentex_client, tracer=tracer

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/adk/_modules/agents.py

@@ -1,7 +1,7 @@
 from datetime import timedelta
 from typing import Optional
 
-from agentex.lib.adk.utils._modules.client import get_async_agentex_client
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from agentex.lib.core.temporal.activities.adk.agents_activities import AgentsActivityName, GetAgentParams
 from temporalio.common import RetryPolicy
 
@@ -29,7 +29,7 @@ class AgentsModule:
         agents_service: Optional[AgentsService] = None,
     ):
         if agents_service is None:
-            agentex_client = get_async_agentex_client()
+            agentex_client = create_async_agentex_client()
             tracer = AsyncTracer(agentex_client)
             self._agents_service = AgentsService(agentex_client=agentex_client, tracer=tracer)
         else:

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/adk/_modules/events.py

@@ -3,7 +3,7 @@ from datetime import timedelta
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
-from agentex.lib.adk.utils._modules.client import get_async_agentex_client
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from agentex.lib.core.services.adk.events import EventsService
 from agentex.lib.core.temporal.activities.activity_helpers import ActivityHelpers
 from agentex.lib.core.temporal.activities.adk.events_activities import (
@@ -33,7 +33,7 @@ class EventsModule:
         events_service: EventsService | None = None,
     ):
         if events_service is None:
-            agentex_client = get_async_agentex_client()
+            agentex_client = create_async_agentex_client()
             tracer = AsyncTracer(agentex_client)
             self._events_service = EventsService(
                 agentex_client=agentex_client, tracer=tracer

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/adk/_modules/messages.py

@@ -3,7 +3,7 @@ from datetime import timedelta
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
-from agentex.lib.adk.utils._modules.client import get_async_agentex_client
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from agentex.lib.core.adapters.streams.adapter_redis import RedisStreamRepository
 from agentex.lib.core.services.adk.messages import MessagesService
 from agentex.lib.core.services.adk.streaming import StreamingService
@@ -38,7 +38,7 @@ class MessagesModule:
         messages_service: MessagesService | None = None,
     ):
         if messages_service is None:
-            agentex_client = get_async_agentex_client()
+            agentex_client = create_async_agentex_client()
             stream_repository = RedisStreamRepository()
             streaming_service = StreamingService(
                 agentex_client=agentex_client,

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/adk/_modules/state.py

@@ -5,7 +5,7 @@ from pydantic import BaseModel
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
-from agentex.lib.adk.utils._modules.client import get_async_agentex_client
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from agentex.lib.core.services.adk.state import StateService
 from agentex.lib.core.temporal.activities.activity_helpers import ActivityHelpers
 from agentex.lib.core.temporal.activities.adk.state_activities import (
@@ -37,7 +37,7 @@ class StateModule:
         state_service: StateService | None = None,
     ):
         if state_service is None:
-            agentex_client = get_async_agentex_client()
+            agentex_client = create_async_agentex_client()
             tracer = AsyncTracer(agentex_client)
             self._state_service = StateService(
                 agentex_client=agentex_client, tracer=tracer

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/adk/_modules/streaming.py

@@ -1,7 +1,7 @@
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
-from agentex.lib.adk.utils._modules.client import get_async_agentex_client
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from agentex.lib.core.adapters.streams.adapter_redis import RedisStreamRepository
 from agentex.lib.core.services.adk.streaming import (
     StreamingService,
@@ -35,7 +35,7 @@ class StreamingModule:
         """
         if streaming_service is None:
             stream_repository = RedisStreamRepository()
-            agentex_client = get_async_agentex_client()
+            agentex_client = create_async_agentex_client()
             self._streaming_service = StreamingService(
                 agentex_client=agentex_client,
                 stream_repository=stream_repository,

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/adk/_modules/tasks.py

@@ -3,7 +3,7 @@ from datetime import timedelta
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
-from agentex.lib.adk.utils._modules.client import get_async_agentex_client
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from agentex.lib.core.services.adk.tasks import TasksService
 from agentex.lib.core.temporal.activities.activity_helpers import ActivityHelpers
 from agentex.lib.core.temporal.activities.adk.tasks_activities import (
@@ -32,7 +32,7 @@ class TasksModule:
         tasks_service: TasksService | None = None,
     ):
         if tasks_service is None:
-            agentex_client = get_async_agentex_client()
+            agentex_client = create_async_agentex_client()
             tracer = AsyncTracer(agentex_client)
             self._tasks_service = TasksService(
                 agentex_client=agentex_client, tracer=tracer

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/adk/_modules/tracing.py

@@ -6,7 +6,7 @@ from typing import Any
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
-from agentex.lib.adk.utils._modules.client import get_async_agentex_client
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from agentex.lib.core.services.adk.tracing import TracingService
 from agentex.lib.core.temporal.activities.activity_helpers import ActivityHelpers
 from agentex.lib.core.temporal.activities.adk.tracing_activities import (
@@ -39,7 +39,7 @@ class TracingModule:
             tracing_activities (Optional[TracingActivities]): Optional pre-configured tracing activities. If None, will be auto-initialized.
         """
         if tracing_service is None:
-            agentex_client = get_async_agentex_client()
+            agentex_client = create_async_agentex_client()
             tracer = AsyncTracer(agentex_client)
             self._tracing_service = TracingService(tracer=tracer)
         else:

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/adk/providers/_modules/litellm.py

@@ -1,6 +1,7 @@
 from collections.abc import AsyncGenerator
 from datetime import timedelta
 
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
@@ -39,7 +40,7 @@ class LiteLLMModule:
     ):
         if litellm_service is None:
             # Create default service
-            agentex_client = AsyncAgentex()
+            agentex_client = create_async_agentex_client()
             stream_repository = RedisStreamRepository()
             streaming_service = StreamingService(
                 agentex_client=agentex_client,

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/adk/providers/_modules/openai.py

@@ -1,6 +1,7 @@
 from datetime import timedelta
 from typing import Any, Literal
 
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from agents import Agent, RunResult, RunResultStreaming
 from agents.agent import StopAtTools, ToolsToFinalOutputFunction
 from agents.agent_output import AgentOutputSchemaBase
@@ -46,7 +47,7 @@ class OpenAIModule:
     ):
         if openai_service is None:
             # Create default service
-            agentex_client = AsyncAgentex()
+            agentex_client = create_async_agentex_client()
             stream_repository = RedisStreamRepository()
             streaming_service = StreamingService(
                 agentex_client=agentex_client,

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/adk/providers/_modules/sgp.py

@@ -1,5 +1,6 @@
 from datetime import timedelta
 
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from scale_gp import SGPClient, SGPClientError
 from temporalio.common import RetryPolicy
 
@@ -33,7 +34,7 @@ class SGPModule:
         if sgp_service is None:
             try:
                 sgp_client = SGPClient()
-                agentex_client = AsyncAgentex()
+                agentex_client = create_async_agentex_client()
                 tracer = AsyncTracer(agentex_client)
                 self._sgp_service = SGPService(sgp_client=sgp_client, tracer=tracer)
             except SGPClientError:

agentex_sdk-0.2.6/src/agentex/lib/adk/utils/_modules/client.py

@@ -0,0 +1,28 @@
+import httpx
+
+from agentex import AsyncAgentex
+from agentex.lib.environment_variables import EnvironmentVariables
+from agentex.lib.utils.logging import make_logger
+
+logger = make_logger(__name__)
+
+
+class EnvAuth(httpx.Auth):
+    def __init__(self, header_name="x-agent-identity"):
+        self.header_name = header_name
+
+    def auth_flow(self, request):
+        # This gets called for every request
+        env_vars = EnvironmentVariables.refresh()
+        if env_vars:
+            agent_id = env_vars.AGENT_ID
+            if agent_id:
+                request.headers[self.header_name] = agent_id
+                logger.info(f"Adding header {self.header_name}:{agent_id}")
+        yield request
+
+
+def create_async_agentex_client(**kwargs) -> AsyncAgentex:
+    client = AsyncAgentex(**kwargs)
+    client._client.auth = EnvAuth()
+    return client

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/adk/utils/_modules/templating.py

@@ -1,6 +1,7 @@
 from datetime import timedelta
 from typing import Any
 
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
@@ -39,7 +40,7 @@ class TemplatingModule:
             templating_service (Optional[TemplatingService]): Optional pre-configured templating service. If None, will be auto-initialized.
         """
         if templating_service is None:
-            agentex_client = AsyncAgentex()
+            agentex_client = create_async_agentex_client()
             tracer = AsyncTracer(agentex_client)
             self._templating_service = TemplatingService(tracer=tracer)
         else:

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/cli/handlers/deploy_handlers.py

@@ -22,7 +22,7 @@ logger = make_logger(__name__)
 console = Console()
 
 TEMPORAL_WORKER_KEY = "temporal-worker"
-AGENTEX_AGENTS_HELM_CHART_VERSION = "0.1.2-v2-beta"
+AGENTEX_AGENTS_HELM_CHART_VERSION = "0.1.4-v1-beta"
 
 
 class InputDeployOverrides(BaseModel):

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/core/temporal/activities/__init__.py

@@ -1,3 +1,4 @@
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from scale_gp import SGPClient, SGPClientError
 
 from agentex import AsyncAgentex
@@ -58,7 +59,7 @@ def get_all_activities(sgp_client=None):
 
     llm_gateway = LiteLLMGateway()
     stream_repository = RedisStreamRepository()
-    agentex_client = AsyncAgentex()
+    agentex_client = create_async_agentex_client()
     tracer = AsyncTracer(agentex_client)
 
     # Services

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/core/tracing/processors/agentex_tracing_processor.py

@@ -1,6 +1,7 @@
 from typing import Any, Dict, override
 
 from agentex import Agentex, AsyncAgentex
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from agentex.lib.core.tracing.processors.tracing_processor_interface import (
     AsyncTracingProcessor,
     SyncTracingProcessor,
@@ -65,7 +66,7 @@ class AgentexSyncTracingProcessor(SyncTracingProcessor):
 
 class AgentexAsyncTracingProcessor(AsyncTracingProcessor):
     def __init__(self, config: AgentexTracingProcessorConfig):
-        self.client = AsyncAgentex()
+        self.client = create_async_agentex_client()
 
     @override
     async def on_span_start(self, span: Span) -> None:

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/environment_variables.py

@@ -4,12 +4,15 @@ import os
 from enum import Enum
 from pathlib import Path
 
+from agentex.lib.utils.logging import make_logger
 from dotenv import load_dotenv
 
 from agentex.lib.utils.model_utils import BaseModel
 
 PROJECT_ROOT = Path(__file__).resolve().parents[2]
 
+logger = make_logger(__name__)
+
 
 class EnvVarKeys(str, Enum):
     ENVIRONMENT = "ENVIRONMENT"
@@ -37,7 +40,7 @@ class Environment(str, Enum):
     PROD = "production"
 
 
-refreshed_environment_variables = None
+refreshed_environment_variables: "EnvironmentVariables" | None = None
 
 
 class EnvironmentVariables(BaseModel):
@@ -64,6 +67,7 @@ class EnvironmentVariables(BaseModel):
         if refreshed_environment_variables is not None:
             return refreshed_environment_variables
 
+        logger.info("Refreshing environment variables")
         if os.environ.get(EnvVarKeys.ENVIRONMENT) == Environment.DEV:
             # Load global .env file first
             global_env_path = PROJECT_ROOT / ".env"

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/sdk/fastacp/base/base_acp_server.py

@@ -9,7 +9,7 @@ from typing import Any
 
 import httpx
 import uvicorn
-from agentex.lib.adk.utils._modules.client import get_async_agentex_client
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from fastapi import FastAPI, Request
 from fastapi.responses import StreamingResponse
 from pydantic import TypeAdapter, ValidationError
@@ -397,9 +397,10 @@ class BaseACPServer(FastAPI):
 
                         os.environ["AGENT_ID"] = agent_id
                         os.environ["AGENT_NAME"] = agent_name
-                        refreshed_environment_variables.AGENT_ID = agent_id
-                        refreshed_environment_variables.AGENT_NAME = agent_name
-                        get_async_agentex_client()  # refresh cache
+                        env_vars.AGENT_ID = agent_id
+                        env_vars.AGENT_NAME = agent_name
+                        global refreshed_environment_variables
+                        refreshed_environment_variables = env_vars
                         logger.info(
                             f"Successfully registered agent '{env_vars.AGENT_NAME}' with Agentex server with acp_url: {full_acp_url}. Registration data: {registration_data}"
                         )

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/lib/sdk/fastacp/impl/agentic_base_acp.py

@@ -1,4 +1,6 @@
 from typing import Any
+
+from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from typing_extensions import override
 from agentex import AsyncAgentex
 from agentex.lib.sdk.fastacp.base.base_acp_server import BaseACPServer
@@ -24,7 +26,7 @@ class AgenticBaseACP(BaseACPServer):
     def __init__(self):
         super().__init__()
         self._setup_handlers()
-        self._agentex_client = AsyncAgentex()
+        self._agentex_client = create_async_agentex_client()
 
     @classmethod
     @override

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/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,6 +82,8 @@ class TasksResource(SyncAPIResource):
     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,
@@ -85,11 +91,32 @@ class TasksResource(SyncAPIResource):
         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,6 +347,8 @@ class AsyncTasksResource(AsyncAPIResource):
     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,
@@ -327,11 +356,32 @@ class AsyncTasksResource(AsyncAPIResource):
         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,
         )

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/src/agentex/types/__init__.py

@@ -19,6 +19,7 @@ from .message_author import MessageAuthor as MessageAuthor
 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

agentex_sdk-0.2.6/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]

{agentex_sdk-0.2.5 → agentex_sdk-0.2.6}/tests/api_resources/test_tasks.py

@@ -66,6 +66,15 @@ class TestTasks:
         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 @@ class TestAsyncTasks:
         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: