agentex-sdk 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

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.1"  # x-release-please-version
+__version__ = "0.2.2"  # x-release-please-version

agentex/lib/adk/_modules/acp.py

@@ -4,6 +4,7 @@ from typing import Any
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
+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 (
@@ -40,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 = AsyncAgentex()
+            agentex_client = create_async_agentex_client()
             tracer = AsyncTracer(agentex_client)
             self._acp_service = ACPService(agentex_client=agentex_client, tracer=tracer)
         else:

agentex/lib/adk/_modules/agent_task_tracker.py

@@ -3,6 +3,7 @@ from datetime import timedelta
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
+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 (
@@ -33,7 +34,7 @@ class AgentTaskTrackerModule:
         agent_task_tracker_service: AgentTaskTrackerService | None = None,
     ):
         if agent_task_tracker_service is None:
-            agentex_client = AsyncAgentex()
+            agentex_client = create_async_agentex_client()
             tracer = AsyncTracer(agentex_client)
             self._agent_task_tracker_service = AgentTaskTrackerService(
                 agentex_client=agentex_client, tracer=tracer

agentex/lib/adk/_modules/agents.py

@@ -1,6 +1,7 @@
 from datetime import timedelta
 from typing import Optional
 
+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
 
@@ -28,7 +29,7 @@ class AgentsModule:
         agents_service: Optional[AgentsService] = None,
     ):
         if agents_service is None:
-            agentex_client = AsyncAgentex()
+            agentex_client = create_async_agentex_client()
             tracer = AsyncTracer(agentex_client)
             self._agents_service = AgentsService(agentex_client=agentex_client, tracer=tracer)
         else:

agentex/lib/adk/_modules/events.py

@@ -3,6 +3,7 @@ from datetime import timedelta
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
+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 (
@@ -32,7 +33,7 @@ class EventsModule:
         events_service: EventsService | None = None,
     ):
         if events_service is None:
-            agentex_client = AsyncAgentex()
+            agentex_client = create_async_agentex_client()
             tracer = AsyncTracer(agentex_client)
             self._events_service = EventsService(
                 agentex_client=agentex_client, tracer=tracer

agentex/lib/adk/_modules/messages.py

@@ -3,6 +3,7 @@ from datetime import timedelta
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
+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
@@ -37,7 +38,7 @@ class MessagesModule:
         messages_service: MessagesService | None = None,
     ):
         if messages_service is None:
-            agentex_client = AsyncAgentex()
+            agentex_client = create_async_agentex_client()
             stream_repository = RedisStreamRepository()
             streaming_service = StreamingService(
                 agentex_client=agentex_client,

agentex/lib/adk/_modules/state.py

@@ -5,6 +5,7 @@ from pydantic import BaseModel
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
+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 (
@@ -36,7 +37,7 @@ class StateModule:
         state_service: StateService | None = None,
     ):
         if state_service is None:
-            agentex_client = AsyncAgentex()
+            agentex_client = create_async_agentex_client()
             tracer = AsyncTracer(agentex_client)
             self._state_service = StateService(
                 agentex_client=agentex_client, tracer=tracer

agentex/lib/adk/_modules/streaming.py

@@ -1,6 +1,7 @@
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
+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,
@@ -34,7 +35,7 @@ class StreamingModule:
         """
         if streaming_service is None:
             stream_repository = RedisStreamRepository()
-            agentex_client = AsyncAgentex()
+            agentex_client = create_async_agentex_client()
             self._streaming_service = StreamingService(
                 agentex_client=agentex_client,
                 stream_repository=stream_repository,

agentex/lib/adk/_modules/tasks.py

@@ -3,6 +3,7 @@ from datetime import timedelta
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
+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 (
@@ -31,7 +32,7 @@ class TasksModule:
         tasks_service: TasksService | None = None,
     ):
         if tasks_service is None:
-            agentex_client = AsyncAgentex()
+            agentex_client = create_async_agentex_client()
             tracer = AsyncTracer(agentex_client)
             self._tasks_service = TasksService(
                 agentex_client=agentex_client, tracer=tracer

agentex/lib/adk/_modules/tracing.py

@@ -6,6 +6,7 @@ from typing import Any
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
+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 (
@@ -38,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 = AsyncAgentex()
+            agentex_client = create_async_agentex_client()
             tracer = AsyncTracer(agentex_client)
             self._tracing_service = TracingService(tracer=tracer)
         else:

agentex/lib/adk/utils/_modules/client.py

@@ -0,0 +1,12 @@
+import os
+
+from agentex import AsyncAgentex
+from agentex.lib.environment_variables import refreshed_environment_variables
+
+
+def create_async_agentex_client(**kwargs):
+    agent_id = refreshed_environment_variables.AGENT_ID
+    default_headers = {
+        "x-agent-identity": agent_id
+    }
+    return AsyncAgentex(default_headers=default_headers, **kwargs)

agentex/lib/cli/templates/default/dev.ipynb.j2

@@ -30,24 +30,17 @@
    "outputs": [],
    "source": [
     "# (REQUIRED) Create a new task. For Agentic agents, you must create a task for messages to be associated with.\n",
-    "\n",
-    "from typing import cast\n",
     "import uuid\n",
     "\n",
-    "from agentex.types import Task\n",
-    "\n",
-    "TASK_ID = str(uuid.uuid4())[:8]\n",
-    "\n",
-    "rpc_response = client.agents.rpc_by_name(\n",
+    "rpc_response = client.agents.create_task(\n",
     "    agent_name=AGENT_NAME,\n",
-    "    method=\"task/create\",\n",
     "    params={\n",
-    "        \"name\": f\"{TASK_ID}-task\",\n",
+    "        \"name\": f\"{str(uuid.uuid4())[:8]}-task\",\n",
     "        \"params\": {}\n",
     "    }\n",
     ")\n",
     "\n",
-    "task = cast(Task, rpc_response.result)\n",
+    "task = rpc_response.result\n",
     "print(task)"
    ]
   },
@@ -58,9 +51,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Test non streaming response\n",
-    "from typing import cast\n",
-    "from agentex.types import Event\n",
+    "# Send an event to the agent\n",
     "\n",
     "# The response is expected to be a list of TaskMessage objects, which is a union of the following types:\n",
     "# - TextContent: A message with just text content   \n",
@@ -70,16 +61,15 @@
     "\n",
     "# When processing the message/send response, if you are expecting more than TextContent, such as DataContent, ToolRequestContent, or ToolResponseContent, you can process them as well\n",
     "\n",
-    "rpc_response = client.agents.rpc_by_name(\n",
+    "rpc_response = client.agents.send_event(\n",
     "    agent_name=AGENT_NAME,\n",
-    "    method=\"event/send\",\n",
     "    params={\n",
     "        \"content\": {\"type\": \"text\", \"author\": \"user\", \"content\": \"Hello what can you do?\"},\n",
     "        \"task_id\": task.id,\n",
     "    }\n",
     ")\n",
     "\n",
-    "event = cast(Event, rpc_response.result)\n",
+    "event = rpc_response.result\n",
     "print(event)"
    ]
   },
@@ -90,6 +80,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "# Subscribe to the async task messages produced by the agent\n",
     "from agentex.lib.utils.dev_tools import subscribe_to_async_task_messages\n",
     "\n",
     "task_messages = subscribe_to_async_task_messages(\n",
@@ -101,6 +92,14 @@
     "    timeout=5,\n",
     ")"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "4864e354",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {

agentex/lib/cli/templates/sync/dev.ipynb.j2

@@ -56,8 +56,7 @@
    "outputs": [],
    "source": [
     "# Test non streaming response\n",
-    "from typing import List, cast\n",
-    "from agentex.types import TaskMessage, TextContent\n",
+    "from agentex.types import TextContent\n",
     "\n",
     "# The response is expected to be a list of TaskMessage objects, which is a union of the following types:\n",
     "# - TextContent: A message with just text content   \n",
@@ -67,29 +66,23 @@
     "\n",
     "# When processing the message/send response, if you are expecting more than TextContent, such as DataContent, ToolRequestContent, or ToolResponseContent, you can process them as well\n",
     "\n",
-    "rpc_response = client.agents.rpc_by_name(\n",
+    "rpc_response = client.agents.send_message(\n",
     "    agent_name=AGENT_NAME,\n",
-    "    method=\"message/send\",\n",
     "    params={\n",
     "        \"content\": {\"type\": \"text\", \"author\": \"user\", \"content\": \"Hello what can you do?\"},\n",
     "        \"stream\": False\n",
     "    }\n",
     ")\n",
     "\n",
-    "# # Extract and print just the text content from the response\n",
-    "# # The response is expected to be a dict with a \"result\" key containing a list of message dicts\n",
-    "if rpc_response and rpc_response.result:\n",
+    "if not rpc_response or not rpc_response.result:\n",
+    "    raise ValueError(\"No result in response\")\n",
     "\n",
-    "    # We know that the result of the message/send when stream is set to False will be a list of TaskMessage objects\n",
-    "    task_message_list = cast(List[TaskMessage], rpc_response.result)\n",
-    "    for task_message in rpc_response.result:\n",
-    "        if isinstance(task_message, TaskMessage):\n",
-    "            content = task_message.content\n",
-    "            if isinstance(content, TextContent):\n",
-    "                text = content.content\n",
-    "                print(text)\n",
-    "        else:\n",
-    "            print(f\"Found non-text {type(task_message)} object in response.\")\n"
+    "# Extract and print just the text content from the response\n",
+    "for task_message in rpc_response.result:\n",
+    "    content = task_message.content\n",
+    "    if isinstance(content, TextContent):\n",
+    "        text = content.content\n",
+    "        print(text)\n"
    ]
   },
   {
@@ -100,11 +93,8 @@
    "outputs": [],
    "source": [
     "# Test streaming response\n",
-    "import json\n",
-    "from agentex.types import AgentRpcResponse\n",
-    "from agentex.types.agent_rpc_result import StreamTaskMessageDelta, StreamTaskMessageFull\n",
+    "from agentex.types.task_message_update import StreamTaskMessageDelta, StreamTaskMessageFull\n",
     "from agentex.types.text_delta import TextDelta\n",
-    "from agentex.types.task_message_update import TaskMessageUpdate\n",
     "\n",
     "\n",
     "# The result object of message/send will be a TaskMessageUpdate which is a union of the following types:\n",
@@ -120,38 +110,34 @@
     "# Whenn processing StreamTaskMessageDelta, if you are expecting more than TextDeltas, such as DataDelta, ToolRequestDelta, or ToolResponseDelta, you can process them as well\n",
     "# Whenn processing StreamTaskMessageFull, if you are expecting more than TextContent, such as DataContent, ToolRequestContent, or ToolResponseContent, you can process them as well\n",
     "\n",
-    "with client.agents.with_streaming_response.rpc_by_name(\n",
+    "for agent_rpc_response_chunk in client.agents.send_message_stream(\n",
     "    agent_name=AGENT_NAME,\n",
-    "    method=\"message/send\",\n",
     "    params={\n",
     "        \"content\": {\"type\": \"text\", \"author\": \"user\", \"content\": \"Hello what can you do?\"},\n",
     "        \"stream\": True\n",
     "    }\n",
-    ") as response:\n",
-    "    for agent_rpc_response_str in response.iter_text():\n",
-    "        chunk_rpc_response = AgentRpcResponse.model_validate(json.loads(agent_rpc_response_str))\n",
-    "        # We know that the result of the message/send when stream is set to True will be a TaskMessageUpdate\n",
-    "        task_message_update = cast(TaskMessageUpdate, chunk_rpc_response.result)\n",
-    "\n",
-    "        # Print oly the text deltas as they arrive or any full messages\n",
-    "        if isinstance(task_message_update, StreamTaskMessageDelta):\n",
-    "            delta = task_message_update.delta\n",
-    "            if isinstance(delta, TextDelta):\n",
-    "                print(delta.text_delta, end=\"\", flush=True)\n",
-    "            else:\n",
-    "                print(f\"Found non-text {type(task_message)} object in streaming message.\")\n",
-    "        elif isinstance(task_message_update, StreamTaskMessageFull):\n",
-    "            content = task_message_update.content\n",
-    "            if isinstance(content, TextContent):\n",
-    "                print(content.content)\n",
-    "            else:\n",
-    "                print(f\"Found non-text {type(task_message)} object in full message.\")\n"
+    "):\n",
+    "    # We know that the result of the message/send when stream is set to True will be a TaskMessageUpdate\n",
+    "    task_message_update = agent_rpc_response_chunk.result\n",
+    "    # Print oly the text deltas as they arrive or any full messages\n",
+    "    if isinstance(task_message_update, StreamTaskMessageDelta):\n",
+    "        delta = task_message_update.delta\n",
+    "        if isinstance(delta, TextDelta):\n",
+    "            print(delta.text_delta, end=\"\", flush=True)\n",
+    "        else:\n",
+    "            print(f\"Found non-text {type(task_message)} object in streaming message.\")\n",
+    "    elif isinstance(task_message_update, StreamTaskMessageFull):\n",
+    "        content = task_message_update.content\n",
+    "        if isinstance(content, TextContent):\n",
+    "            print(content.content)\n",
+    "        else:\n",
+    "            print(f\"Found non-text {type(task_message)} object in full message.\")\n"
    ]
   },
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "4ffb663c",
+   "id": "c5e7e042",
    "metadata": {},
    "outputs": [],
    "source": []

agentex/lib/cli/templates/temporal/dev.ipynb.j2

@@ -30,24 +30,17 @@
    "outputs": [],
    "source": [
     "# (REQUIRED) Create a new task. For Agentic agents, you must create a task for messages to be associated with.\n",
-    "\n",
-    "from typing import cast\n",
     "import uuid\n",
     "\n",
-    "from agentex.types import Task\n",
-    "\n",
-    "TASK_ID = str(uuid.uuid4())[:8]\n",
-    "\n",
-    "rpc_response = client.agents.rpc_by_name(\n",
+    "rpc_response = client.agents.create_task(\n",
     "    agent_name=AGENT_NAME,\n",
-    "    method=\"task/create\",\n",
     "    params={\n",
-    "        \"name\": f\"{TASK_ID}-task\",\n",
+    "        \"name\": f\"{str(uuid.uuid4())[:8]}-task\",\n",
     "        \"params\": {}\n",
     "    }\n",
     ")\n",
     "\n",
-    "task = cast(Task, rpc_response.result)\n",
+    "task = rpc_response.result\n",
     "print(task)"
    ]
   },
@@ -58,9 +51,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Test non streaming response\n",
-    "from typing import cast\n",
-    "from agentex.types import Event\n",
+    "# Send an event to the agent\n",
     "\n",
     "# The response is expected to be a list of TaskMessage objects, which is a union of the following types:\n",
     "# - TextContent: A message with just text content   \n",
@@ -70,16 +61,15 @@
     "\n",
     "# When processing the message/send response, if you are expecting more than TextContent, such as DataContent, ToolRequestContent, or ToolResponseContent, you can process them as well\n",
     "\n",
-    "rpc_response = client.agents.rpc_by_name(\n",
+    "rpc_response = client.agents.send_event(\n",
     "    agent_name=AGENT_NAME,\n",
-    "    method=\"event/send\",\n",
     "    params={\n",
     "        \"content\": {\"type\": \"text\", \"author\": \"user\", \"content\": \"Hello what can you do?\"},\n",
     "        \"task_id\": task.id,\n",
     "    }\n",
     ")\n",
     "\n",
-    "event = cast(Event, rpc_response.result)\n",
+    "event = rpc_response.result\n",
     "print(event)"
    ]
   },
@@ -90,6 +80,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "# Subscribe to the async task messages produced by the agent\n",
     "from agentex.lib.utils.dev_tools import subscribe_to_async_task_messages\n",
     "\n",
     "task_messages = subscribe_to_async_task_messages(\n",
@@ -101,6 +92,14 @@
     "    timeout=5,\n",
     ")"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "4864e354",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {

agentex/lib/sdk/fastacp/base/base_acp_server.py

@@ -2,6 +2,7 @@ import asyncio
 import base64
 import inspect
 import json
+import os
 from collections.abc import AsyncGenerator, Awaitable, Callable
 from contextlib import asynccontextmanager
 from typing import Any
@@ -13,7 +14,7 @@ from fastapi.responses import StreamingResponse
 from pydantic import TypeAdapter, ValidationError
 
 # from agentex.lib.sdk.fastacp.types import BaseACPConfig
-from agentex.lib.environment_variables import EnvironmentVariables
+from agentex.lib.environment_variables import EnvironmentVariables, refreshed_environment_variables
 from agentex.lib.types.acp import (
     PARAMS_MODEL_BY_METHOD,
     RPC_SYNC_METHODS,
@@ -390,8 +391,16 @@ class BaseACPServer(FastAPI):
                         registration_url, json=registration_data, timeout=30.0
                     )
                     if response.status_code == 200:
+                        agent = response.json()
+                        agent_id, agent_name = agent["id"], agent["name"]
+
+                        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
+                        
                         logger.info(
-                            f"Successfully registered agent '{env_vars.AGENT_NAME}' with Agentex server with acp_url: {full_acp_url}. Registration data: {registration_data}"
+                            f"Successfully registered agent '{agent_name}' with Agentex server with acp_url: {full_acp_url}. Registration data: {registration_data}"
                         )
                         return  # Success, exit the retry loop
                     else:

agentex/resources/agents.py

@@ -2,7 +2,8 @@
 
 from __future__ import annotations
 
-from typing import Union, Optional
+import json
+from typing import AsyncGenerator, Generator, Union, Optional
 from typing_extensions import Literal
 
 import httpx
@@ -20,7 +21,7 @@ from .._response import (
 )
 from ..types.agent import Agent
 from .._base_client import make_request_options
-from ..types.agent_rpc_response import AgentRpcResponse
+from ..types.agent_rpc_response import AgentRpcResponse, CancelTaskResponse, CreateTaskResponse, SendEventResponse, SendMessageResponse, SendMessageStreamResponse
 from ..types.agent_list_response import AgentListResponse
 
 __all__ = ["AgentsResource", "AsyncAgentsResource"]
@@ -310,6 +311,260 @@ class AgentsResource(SyncAPIResource):
             ),
             cast_to=AgentRpcResponse,
         )
+    
+    def create_task(
+      self,
+      agent_id: str | None = None,
+      agent_name: str | None = None,
+      *,
+      params: agent_rpc_params.ParamsCreateTaskRequest,
+      id: Union[int, str, None] | NotGiven = NOT_GIVEN,
+      jsonrpc: Literal["2.0"] | 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,
+    ) -> CreateTaskResponse:
+      if agent_id is not None and agent_name is not None:
+        raise ValueError("Either agent_id or agent_name must be provided, but not both")
+      
+      if agent_id is not None:
+        raw_agent_rpc_response = self.rpc(
+          agent_id=agent_id,
+          method="task/create",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      elif agent_name is not None:
+        raw_agent_rpc_response = self.rpc_by_name(
+          agent_name=agent_name,
+          method="task/create",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      else:
+        raise ValueError("Either agent_id or agent_name must be provided")
+      
+      return CreateTaskResponse.model_validate(raw_agent_rpc_response, from_attributes=True)
+    
+    def cancel_task(
+      self,
+      agent_id: str | None = None,
+      agent_name: str | None = None,
+      *,
+      params: agent_rpc_params.ParamsCancelTaskRequest,
+      id: Union[int, str, None] | NotGiven = NOT_GIVEN,
+      jsonrpc: Literal["2.0"] | 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,
+    ) -> CancelTaskResponse:
+      if agent_id is not None and agent_name is not None:
+        raise ValueError("Either agent_id or agent_name must be provided, but not both")
+      
+      if agent_id is not None:
+        raw_agent_rpc_response = self.rpc(
+          agent_id=agent_id,
+          method="task/cancel",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      elif agent_name is not None:
+        raw_agent_rpc_response = self.rpc_by_name(
+          agent_name=agent_name,
+          method="task/cancel",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      else:
+        raise ValueError("Either agent_id or agent_name must be provided")
+      
+      return CancelTaskResponse.model_validate(raw_agent_rpc_response, from_attributes=True)
+
+    def send_message(
+      self,
+      agent_id: str | None = None,
+      agent_name: str | None = None,
+      *,
+      params: agent_rpc_params.ParamsSendMessageRequest,
+      id: Union[int, str, None] | NotGiven = NOT_GIVEN,
+      jsonrpc: Literal["2.0"] | 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,
+    ) -> SendMessageResponse:
+      if agent_id is not None and agent_name is not None:
+        raise ValueError("Either agent_id or agent_name must be provided, but not both")
+      
+      if "stream" in params and params["stream"] == True:
+        raise ValueError("If stream is set to True, use send_message_stream() instead")
+      else:
+        if agent_id is not None:
+          raw_agent_rpc_response = self.rpc(
+            agent_id=agent_id,
+            method="message/send",
+            params=params,
+            id=id,
+            jsonrpc=jsonrpc,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+          )
+        elif agent_name is not None:
+          raw_agent_rpc_response = self.rpc_by_name(
+            agent_name=agent_name,
+            method="message/send",
+            params=params,
+            id=id,
+            jsonrpc=jsonrpc,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+          )
+        else:
+          raise ValueError("Either agent_id or agent_name must be provided")
+        
+        return SendMessageResponse.model_validate(raw_agent_rpc_response, from_attributes=True)
+    
+    def send_message_stream(
+      self,
+      agent_id: str | None = None,
+      agent_name: str | None = None,
+      *,
+      params: agent_rpc_params.ParamsSendMessageRequest,
+      id: Union[int, str, None] | NotGiven = NOT_GIVEN,
+      jsonrpc: Literal["2.0"] | 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,
+    ) -> Generator[SendMessageStreamResponse, None, None]:
+      if agent_id is not None and agent_name is not None:
+        raise ValueError("Either agent_id or agent_name must be provided, but not both")
+
+      if "stream" in params and params["stream"] == False:
+        raise ValueError("If stream is set to False, use send_message() instead")
+      
+      params["stream"] = True
+      
+      if agent_id is not None:
+        raw_agent_rpc_response = self.with_streaming_response.rpc(
+          agent_id=agent_id,
+          method="message/send",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      elif agent_name is not None:
+        raw_agent_rpc_response = self.with_streaming_response.rpc_by_name(
+          agent_name=agent_name,
+          method="message/send",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      else:
+        raise ValueError("Either agent_id or agent_name must be provided")
+      
+      with raw_agent_rpc_response as response:
+        for agent_rpc_response_str in response.iter_text():
+          if agent_rpc_response_str.strip():  # Only process non-empty lines
+            try:
+              chunk_rpc_response = SendMessageStreamResponse.model_validate(
+                json.loads(agent_rpc_response_str), 
+                from_attributes=True
+              )
+              yield chunk_rpc_response
+            except json.JSONDecodeError:
+              # Skip invalid JSON lines
+              continue
+    
+    def send_event(
+      self,
+      agent_id: str | None = None,
+      agent_name: str | None = None,
+      *,
+      params: agent_rpc_params.ParamsSendEventRequest,
+      id: Union[int, str, None] | NotGiven = NOT_GIVEN,
+      jsonrpc: Literal["2.0"] | 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,
+    ) -> SendEventResponse:
+      if agent_id is not None and agent_name is not None:
+        raise ValueError("Either agent_id or agent_name must be provided, but not both")
+      
+      if agent_id is not None:
+        raw_agent_rpc_response = self.rpc(
+          agent_id=agent_id,
+          method="event/send",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      elif agent_name is not None:
+        raw_agent_rpc_response = self.rpc_by_name(
+          agent_name=agent_name,
+          method="event/send",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      else:
+        raise ValueError("Either agent_id or agent_name must be provided")
+      
+      return SendEventResponse.model_validate(raw_agent_rpc_response, from_attributes=True)
 
 
 class AsyncAgentsResource(AsyncAPIResource):
@@ -596,7 +851,260 @@ class AsyncAgentsResource(AsyncAPIResource):
             ),
             cast_to=AgentRpcResponse,
         )
-
+    
+    async def create_task(
+      self,
+      agent_id: str | None = None,
+      agent_name: str | None = None,
+      *,
+      params: agent_rpc_params.ParamsCreateTaskRequest,
+      id: Union[int, str, None] | NotGiven = NOT_GIVEN,
+      jsonrpc: Literal["2.0"] | 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,
+    ) -> CreateTaskResponse:
+      if agent_id is not None and agent_name is not None:
+        raise ValueError("Either agent_id or agent_name must be provided, but not both")
+      
+      if agent_id is not None:
+        raw_agent_rpc_response = await self.rpc(
+          agent_id=agent_id,
+          method="task/create",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      elif agent_name is not None:
+        raw_agent_rpc_response = await self.rpc_by_name(
+          agent_name=agent_name,
+          method="task/create",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      else:
+        raise ValueError("Either agent_id or agent_name must be provided")
+      
+      return CreateTaskResponse.model_validate(raw_agent_rpc_response, from_attributes=True)
+    
+    async def cancel_task(
+      self,
+      agent_id: str | None = None,
+      agent_name: str | None = None,
+      *,
+      params: agent_rpc_params.ParamsCancelTaskRequest,
+      id: Union[int, str, None] | NotGiven = NOT_GIVEN,
+      jsonrpc: Literal["2.0"] | 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,
+    ) -> CancelTaskResponse:
+      if agent_id is not None and agent_name is not None:
+        raise ValueError("Either agent_id or agent_name must be provided, but not both")
+      
+      if agent_id is not None:
+        raw_agent_rpc_response = await self.rpc(
+          agent_id=agent_id,
+          method="task/cancel",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      elif agent_name is not None:
+        raw_agent_rpc_response = await self.rpc_by_name(
+          agent_name=agent_name,
+          method="task/cancel",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      else:
+        raise ValueError("Either agent_id or agent_name must be provided")
+      
+      return CancelTaskResponse.model_validate(raw_agent_rpc_response, from_attributes=True)
+
+    async def send_message(
+      self,
+      agent_id: str | None = None,
+      agent_name: str | None = None,
+      *,
+      params: agent_rpc_params.ParamsSendMessageRequest,
+      id: Union[int, str, None] | NotGiven = NOT_GIVEN,
+      jsonrpc: Literal["2.0"] | 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,
+    ) -> SendMessageResponse:
+      if agent_id is not None and agent_name is not None:
+        raise ValueError("Either agent_id or agent_name must be provided, but not both")
+      
+      if "stream" in params and params["stream"] == True:
+        raise ValueError("If stream is set to True, use send_message_stream() instead")
+      else:
+        if agent_id is not None:
+          raw_agent_rpc_response = await self.rpc(
+            agent_id=agent_id,
+            method="message/send",
+            params=params,
+            id=id,
+            jsonrpc=jsonrpc,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+          )
+        elif agent_name is not None:
+          raw_agent_rpc_response = await self.rpc_by_name(
+            agent_name=agent_name,
+            method="message/send",
+            params=params,
+            id=id,
+            jsonrpc=jsonrpc,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+          )
+        else:
+          raise ValueError("Either agent_id or agent_name must be provided")
+        
+        return SendMessageResponse.model_validate(raw_agent_rpc_response, from_attributes=True)
+    
+    async def send_message_stream(
+      self,
+      agent_id: str | None = None,
+      agent_name: str | None = None,
+      *,
+      params: agent_rpc_params.ParamsSendMessageRequest,
+      id: Union[int, str, None] | NotGiven = NOT_GIVEN,
+      jsonrpc: Literal["2.0"] | 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,
+    ) -> AsyncGenerator[SendMessageStreamResponse, None]:
+      if agent_id is not None and agent_name is not None:
+        raise ValueError("Either agent_id or agent_name must be provided, but not both")
+      
+      if "stream" in params and params["stream"] == False:
+        raise ValueError("If stream is set to False, use send_message() instead")
+      
+      params["stream"] = True
+      
+      if agent_id is not None:
+        raw_agent_rpc_response = self.with_streaming_response.rpc(
+          agent_id=agent_id,
+          method="message/send",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      elif agent_name is not None:
+        raw_agent_rpc_response = self.with_streaming_response.rpc_by_name(
+          agent_name=agent_name,
+          method="message/send",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      else:
+        raise ValueError("Either agent_id or agent_name must be provided")
+      
+      async with raw_agent_rpc_response as response:
+        async for agent_rpc_response_str in response.iter_text():
+          if agent_rpc_response_str.strip():  # Only process non-empty lines
+            try:
+              chunk_rpc_response = SendMessageStreamResponse.model_validate(
+                json.loads(agent_rpc_response_str), 
+                from_attributes=True
+              )
+              yield chunk_rpc_response
+            except json.JSONDecodeError:
+              # Skip invalid JSON lines
+              continue
+    
+    async def send_event(
+      self,
+      agent_id: str | None = None,
+      agent_name: str | None = None,
+      *,
+      params: agent_rpc_params.ParamsSendEventRequest,
+      id: Union[int, str, None] | NotGiven = NOT_GIVEN,
+      jsonrpc: Literal["2.0"] | 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,
+    ) -> SendEventResponse:
+      if agent_id is not None and agent_name is not None:
+        raise ValueError("Either agent_id or agent_name must be provided, but not both")
+      
+      if agent_id is not None:
+        raw_agent_rpc_response = await self.rpc(
+          agent_id=agent_id,
+          method="event/send",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      elif agent_name is not None:
+        raw_agent_rpc_response = await self.rpc_by_name(
+          agent_name=agent_name,
+          method="event/send",
+          params=params,
+          id=id,
+          jsonrpc=jsonrpc,
+          extra_headers=extra_headers,
+          extra_query=extra_query,
+          extra_body=extra_body,
+          timeout=timeout,
+        )
+      else:
+        raise ValueError("Either agent_id or agent_name must be provided")
+      
+      return SendEventResponse.model_validate(raw_agent_rpc_response, from_attributes=True)
 
 class AgentsResourceWithRawResponse:
     def __init__(self, agents: AgentsResource) -> None:

agentex/resources/tasks.py

@@ -206,7 +206,7 @@ class TasksResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> Stream[object]:
         """
-        Stream events for a task by its unique ID.
+        Stream message updates for a task by its unique ID.
 
         Args:
           extra_headers: Send extra headers
@@ -241,7 +241,7 @@ class TasksResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> Stream[object]:
         """
-        Stream events for a task by its unique name.
+        Stream message updates for a task by its unique name.
 
         Args:
           extra_headers: Send extra headers
@@ -448,7 +448,7 @@ class AsyncTasksResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> AsyncStream[object]:
         """
-        Stream events for a task by its unique ID.
+        Stream message updates for a task by its unique ID.
 
         Args:
           extra_headers: Send extra headers
@@ -483,7 +483,7 @@ class AsyncTasksResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> AsyncStream[object]:
         """
-        Stream events for a task by its unique name.
+        Stream message updates for a task by its unique name.
 
         Args:
           extra_headers: Send extra headers

agentex/types/agent_rpc_response.py

@@ -5,16 +5,44 @@ from typing_extensions import Literal
 
 from .._models import BaseModel
 from .agent_rpc_result import AgentRpcResult
+from .event import Event
+from .task import Task
+from .task_message import TaskMessage
+from .task_message_update import TaskMessageUpdate
 
 __all__ = ["AgentRpcResponse"]
 
 
-class AgentRpcResponse(BaseModel):
+class BaseAgentRpcResponse(BaseModel):
+    id: Union[int, str, None] = None
+    error: Optional[object] = None
+    jsonrpc: Optional[Literal["2.0"]] = None
+
+
+class AgentRpcResponse(BaseAgentRpcResponse):
     result: Optional[AgentRpcResult] = None
     """The result of the agent RPC request"""
 
-    id: Union[int, str, None] = None
 
-    error: Optional[object] = None
+class CreateTaskResponse(BaseAgentRpcResponse):
+    result: Task
+    """The result of the task creation"""
 
-    jsonrpc: Optional[Literal["2.0"]] = None
+
+class CancelTaskResponse(BaseAgentRpcResponse):
+    result: Task
+    """The result of the task cancellation"""
+
+
+class SendMessageResponse(BaseAgentRpcResponse):
+    result: list[TaskMessage]
+    """The result of the message sending"""
+
+class SendMessageStreamResponse(BaseAgentRpcResponse):
+    result: TaskMessageUpdate
+    """The result of the message sending"""
+
+
+class SendEventResponse(BaseAgentRpcResponse):
+    result: Event
+    """The result of the event sending"""

{agentex_sdk-0.2.1.dist-info → agentex_sdk-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentex-sdk
-Version: 0.2.1
+Version: 0.2.2
 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.1.dist-info → agentex_sdk-0.2.2.dist-info}/RECORD

@@ -11,7 +11,7 @@ agentex/_resource.py,sha256=S1t7wmR5WUvoDIhZjo_x-E7uoTJBynJ3d8tPJMQYdjw,1106
 agentex/_response.py,sha256=Tb9zazsnemO2rTxWtBjAD5WBqlhli5ZaXGbiKgdu5DE,28794
 agentex/_streaming.py,sha256=FNGJExRCF-vTRUZHFKUfoAWFhDGOB3XbioVCF37Jr7E,10104
 agentex/_types.py,sha256=KyKYySGIfHPod2hho1fPxssk5NuVn8C4MeMTtA-lg80,6198
-agentex/_version.py,sha256=FS15FaZ9bBn3jU7AoPIcXle_LlEcQ7SKHwog2bYuba0,159
+agentex/_version.py,sha256=9l4awzDo0H4hmmCOlgavow683f4hzG__e6Tv4BF93Lw,159
 agentex/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/_utils/__init__.py,sha256=PNZ_QJuzZEgyYXqkO1HVhGkj5IU9bglVUcw7H-Knjzw,2062
 agentex/_utils/_logs.py,sha256=LUjFPc3fweSChBUmjhQD8uYmwQAmFMNDuVFKfjYBQfM,777
@@ -29,15 +29,15 @@ agentex/lib/environment_variables.py,sha256=FbJ-tm_thJs5Fv1q4-oY4CED4DNlwCeu3v4x
 agentex/lib/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/adk/__init__.py,sha256=-PpVfEvYr_HD7TnxUWU8RCW2OnxfwpPxTW97dKTnqvI,1082
 agentex/lib/adk/_modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentex/lib/adk/_modules/acp.py,sha256=0cNsHYCMvMA6GfWpPg4SSbBTFLP8PPx_SXZIoizswUU,8934
-agentex/lib/adk/_modules/agent_task_tracker.py,sha256=B6Rwh-_rHzCJrTVWcteVT9l1acSj9WJtQ6NhLgJ9now,6929
-agentex/lib/adk/_modules/agents.py,sha256=Rhv2hf2eADfsmSj3Y2CBgoGLfsavgyv14_6cMK_qUJY,2695
-agentex/lib/adk/_modules/events.py,sha256=FxJHmENDBzYIcXtWuuDp86tLxbffLGju3qbVih2an_s,5169
-agentex/lib/adk/_modules/messages.py,sha256=vYJ4i-69aOYXf-9phN_fMEtqwvp5HL9cW1Kf2vpRxEs,10543
-agentex/lib/adk/_modules/state.py,sha256=F4svy-QPVKs0xfb6lz9eMKpb6ibbdUK2thwv94wuL4Y,10637
-agentex/lib/adk/_modules/streaming.py,sha256=XpLvIz-MA_SDOw-sIuPS8hFhN-jaGcOxN-LEd1F9pDg,3002
-agentex/lib/adk/_modules/tasks.py,sha256=ch-EFsVdc9bkU2tQMm30JVZUlP48ai7pKIOhUkPPABw,4127
-agentex/lib/adk/_modules/tracing.py,sha256=sNpG7WNMdWpscjUMGFFthJ1gqxm0_gOkf2yrMulcliM,7251
+agentex/lib/adk/_modules/acp.py,sha256=0D9HLwBO3SX4eEsx1BY9G90jyvTABmaxdOzANg1rmzQ,9027
+agentex/lib/adk/_modules/agent_task_tracker.py,sha256=fXMDs1zgAC6n8EmWTU5_trs79EGVBCEPxd5eBOZmdE0,7022
+agentex/lib/adk/_modules/agents.py,sha256=VlT_PXnyb3Knh-nx1rfflD9N6gjp5O0_6Z_QL5SCThU,2788
+agentex/lib/adk/_modules/events.py,sha256=RHQZ4ircC60zdm1chRzo3cdt7LNqDoJfPLNJXEz4BH0,5262
+agentex/lib/adk/_modules/messages.py,sha256=1LxexTBif7HXltN3b1u6A5upXuXuq6ZkFAdIrbSv2zo,10636
+agentex/lib/adk/_modules/state.py,sha256=z2Zh1JB3qJo-4BsVABnFZIeDWWKQDbcOw6itl0hU_E4,10730
+agentex/lib/adk/_modules/streaming.py,sha256=lbRQBmySsx63wKiHeR5ueNW_6DJckCNStQ0wQynC4r4,3095
+agentex/lib/adk/_modules/tasks.py,sha256=q3IBdnwVXrv9Ht7z4jkm8RWOpzU9n3VKgrX3FTNXSZ8,4220
+agentex/lib/adk/_modules/tracing.py,sha256=A_kH332f_DFVfMftPHGEg3oi4RFcwDXohg7xtvWUuTY,7344
 agentex/lib/adk/providers/__init__.py,sha256=KPWC8AYsl8lPgpFoRXlGwzozb-peKLAzV_DcTWXBsz4,306
 agentex/lib/adk/providers/_modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/adk/providers/_modules/litellm.py,sha256=YUHZ_m0kL9wEG5pX9x4BeyioiDjOa2anCWk6uo6-vqQ,9393
@@ -45,6 +45,7 @@ agentex/lib/adk/providers/_modules/openai.py,sha256=h7G3y4Gd2wF3XuYEi_aQXORxZhtH
 agentex/lib/adk/providers/_modules/sgp.py,sha256=Me5TeU7OPjPBVeEw7a-DQnYcJ3fbGi3liYbuW5CGiLE,3131
 agentex/lib/adk/utils/__init__.py,sha256=7f6ayV0_fqyw5cwzVANNcZWGJZ-vrrYtZ0qi7KKBRFs,130
 agentex/lib/adk/utils/_modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+agentex/lib/adk/utils/_modules/client.py,sha256=ujttuam5Kd0CRcbsI1BLrQ7dhcAP68xaXyWXx34imR0,357
 agentex/lib/adk/utils/_modules/templating.py,sha256=YPm2bY_iBv9jWf0lncRGUpBZyq_QHBCRlM-XbHjauRc,3502
 agentex/lib/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/cli/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -64,7 +65,7 @@ agentex/lib/cli/templates/default/.dockerignore.j2,sha256=hweGFxw5eDZYsb5EnRHpv2
 agentex/lib/cli/templates/default/Dockerfile-uv.j2,sha256=tGJo_C4vwHYikV4QhGFtSiG6K7Nt4UDdJ71Gob_uTho,1109
 agentex/lib/cli/templates/default/Dockerfile.j2,sha256=T7ouO3glVBo7U2I1TPQvgARyxXGL194cvM5hfbq0e8U,1119
 agentex/lib/cli/templates/default/README.md.j2,sha256=4Sn_cY_TMfN_c5h-I_L3TgTrNKUvhzQvKGDZjS3n2Gw,6361
-agentex/lib/cli/templates/default/dev.ipynb.j2,sha256=m0gu1WIGqVz5QUOl7AI6eWKXrhX_jIZOJ5HcuC8oK8s,3494
+agentex/lib/cli/templates/default/dev.ipynb.j2,sha256=8xY82gVfQ0mhzlEZzI4kLqIXCF19YgimLnpEirDMX8I,3395
 agentex/lib/cli/templates/default/manifest.yaml.j2,sha256=n-XK6LqSyBVY6a7KOfIlOQx4LbMNEVzzHIh6yEiqAhs,3752
 agentex/lib/cli/templates/default/pyproject.toml.j2,sha256=9cpTISM7rOoICWejV5GYMEwPn8RUmB6-E7csM1pmSFo,528
 agentex/lib/cli/templates/default/requirements.txt.j2,sha256=iTmO-z8qFkUa1jTctFCs0WYuq7Sqi6VNQAwATakh2fQ,94
@@ -76,7 +77,7 @@ agentex/lib/cli/templates/sync/.dockerignore.j2,sha256=hweGFxw5eDZYsb5EnRHpv27o9
 agentex/lib/cli/templates/sync/Dockerfile-uv.j2,sha256=tGJo_C4vwHYikV4QhGFtSiG6K7Nt4UDdJ71Gob_uTho,1109
 agentex/lib/cli/templates/sync/Dockerfile.j2,sha256=T7ouO3glVBo7U2I1TPQvgARyxXGL194cvM5hfbq0e8U,1119
 agentex/lib/cli/templates/sync/README.md.j2,sha256=_S7Ngl4qOUQHPFldLXDBvuIWPFU2-WcuxGmr5EXLX6k,8816
-agentex/lib/cli/templates/sync/dev.ipynb.j2,sha256=1kYOPA1xvhyyBUiEndh-iBtTRKrxY5h-MWwRo_Z-cRc,7042
+agentex/lib/cli/templates/sync/dev.ipynb.j2,sha256=Z42iRveuI_k5LcJqWX-3H1glPtNTkxg_MKVe1lwuJos,6055
 agentex/lib/cli/templates/sync/manifest.yaml.j2,sha256=V497KXzvA76sHrgIJ5zRJptpIH8sGbSXZaIsEyp5NZ4,3747
 agentex/lib/cli/templates/sync/pyproject.toml.j2,sha256=9cpTISM7rOoICWejV5GYMEwPn8RUmB6-E7csM1pmSFo,528
 agentex/lib/cli/templates/sync/requirements.txt.j2,sha256=iTmO-z8qFkUa1jTctFCs0WYuq7Sqi6VNQAwATakh2fQ,94
@@ -86,7 +87,7 @@ agentex/lib/cli/templates/temporal/.dockerignore.j2,sha256=hweGFxw5eDZYsb5EnRHpv
 agentex/lib/cli/templates/temporal/Dockerfile-uv.j2,sha256=bnvx-zba5DFjl7UC-TYt-Zi_UDJucjlNRCdkSqHgBiQ,1450
 agentex/lib/cli/templates/temporal/Dockerfile.j2,sha256=pcszlprNTqKMpYEtA4XYlc3vWjq1b0IMQDrN1GxF7yI,1461
 agentex/lib/cli/templates/temporal/README.md.j2,sha256=d6BQAZBf7lCKROXnpDu-Ml1Rq0OM8cCFEOpmStoPWfc,10132
-agentex/lib/cli/templates/temporal/dev.ipynb.j2,sha256=m0gu1WIGqVz5QUOl7AI6eWKXrhX_jIZOJ5HcuC8oK8s,3494
+agentex/lib/cli/templates/temporal/dev.ipynb.j2,sha256=8xY82gVfQ0mhzlEZzI4kLqIXCF19YgimLnpEirDMX8I,3395
 agentex/lib/cli/templates/temporal/manifest.yaml.j2,sha256=S-hJlpVCnT62r66otqHa3iROOn1w_yYvehOOxvLQotY,4483
 agentex/lib/cli/templates/temporal/pyproject.toml.j2,sha256=hBZgFBkgdQ4x7lfmmUXuUL10ER11ojaDp_swnrg26Vo,546
 agentex/lib/cli/templates/temporal/requirements.txt.j2,sha256=iTmO-z8qFkUa1jTctFCs0WYuq7Sqi6VNQAwATakh2fQ,94
@@ -176,7 +177,7 @@ agentex/lib/sdk/config/local_development_config.py,sha256=b1AZsOVo1RoHKbk8Nm5nC8
 agentex/lib/sdk/config/project_config.py,sha256=CGH_r9KbnSFMj2CnBkZnfg41L2o0TeVNz6MwBDKPT_U,3642
 agentex/lib/sdk/fastacp/__init__.py,sha256=UvAdexdnfb4z0F4a2sfXROFyh9EjH89kf3AxHPybzCM,75
 agentex/lib/sdk/fastacp/fastacp.py,sha256=K4D7a9EiJfCgWp2hrE_TbpZBaF4Uc46MfndZK3F9mA0,3061
-agentex/lib/sdk/fastacp/base/base_acp_server.py,sha256=gsbBb8-zNu991vTh6_f5KjR_NC8Knoyy1YTsPAPN2KQ,17066
+agentex/lib/sdk/fastacp/base/base_acp_server.py,sha256=60xo9GGe4vfb2Sj2Oi_BzjrnBVTjhBYnmhKyNHUc2zg,17524
 agentex/lib/sdk/fastacp/impl/agentic_base_acp.py,sha256=fWLX9_mxmX502EBXoymUz1Tu1vEppshS48-aOSpuvlM,2600
 agentex/lib/sdk/fastacp/impl/sync_acp.py,sha256=8FEDfBxI31NoVLX9nyckQ8QwA0UV4svC3fhGKgXBIwk,3862
 agentex/lib/sdk/fastacp/impl/temporal_acp.py,sha256=ZLqjzBBVrXJCXD2bFlrcDkFvpsXZp3thC2rTwZ6xNaY,3737
@@ -220,11 +221,11 @@ agentex/lib/utils/temporal.py,sha256=sXo8OPMMXiyrF7OSBCJBuN_ufyQOD2bLOXgDbVZoyds
 agentex/lib/utils/dev_tools/__init__.py,sha256=oaHxw6ymfhNql-kzXHv3NWVHuqD4fHumasNXJG7kHTU,261
 agentex/lib/utils/dev_tools/async_messages.py,sha256=-alUK1KFltcRb6xtRtIJIRJW9Sf1FwDRgaNPhOn-luw,18105
 agentex/resources/__init__.py,sha256=74rMqWBzQ2dSrKQqsrd7-jskPws0O_ogkFltvZO3HoU,3265
-agentex/resources/agents.py,sha256=fLXbcSKVB3nvPyhvOMil140Tg3Iqfk3MCQnHKswsEOo,26621
+agentex/resources/agents.py,sha256=ibt5SXgcpwIsRJ9BFptfQyqapHZCKQsKGpcKnWcX0vI,46605
 agentex/resources/events.py,sha256=Zc9JhUm3bq2VFnBAolC0M7KZernzj1AjZ_vj0ibP4GY,10412
 agentex/resources/spans.py,sha256=wmcUs4XbXIF5rPeyU_f39c2RTbTLnkuh2LYogZEBD6s,20936
 agentex/resources/states.py,sha256=O31A8--n7n0rHsng2e1oCUAzLNjQIxDUk7rq0IXfgGM,19262
-agentex/resources/tasks.py,sha256=Nw-pb0ZFSGGMir4t9R_zSLsV8SsdCdnmTubcFDQN2hY,23292
+agentex/resources/tasks.py,sha256=NSVPPS2sZGzMTHxuKAvFeV_5EqsM54t1uLOMzqpZayo,23328
 agentex/resources/tracker.py,sha256=YxSeiloMwIqrS9nY7SlHclauRA7142qrKw34xgwqYbA,14103
 agentex/resources/messages/__init__.py,sha256=_J1eusFtr_k6zrAntJSuqx6LWEUBSTrV1OZZh7MaDPE,1015
 agentex/resources/messages/batch.py,sha256=pegCmnjK_J0jek5ChX1pKpq5RmCucTYLbK69H6qGVS4,9629
@@ -236,7 +237,7 @@ agentex/types/agent_list_params.py,sha256=81IWnRZ2rLfHH7GB6VkXShYjb44DO0guG1znJc
 agentex/types/agent_list_response.py,sha256=g0b9Cw7j2yk14veyJORpF3me8iW7g7pr2USpXGokoFI,254
 agentex/types/agent_rpc_by_name_params.py,sha256=G6xkjrZKPmJvhwqgc68tAnzmb4wYh9k0zlcm9CsLQR0,2024
 agentex/types/agent_rpc_params.py,sha256=qkwICON3v6AIE8J2oN1483K_jmGueBYGkVg2W9EUJK0,2012
-agentex/types/agent_rpc_response.py,sha256=xqdq6apfdaPTT5f8x1Q5z19UETI5AjVMzEIIvAHEZw8,517
+agentex/types/agent_rpc_response.py,sha256=dYvtwXzznZZmpJzX8e5pAW3Cm5m8qXE5-QN6Eb1cZNI,1288
 agentex/types/agent_rpc_result.py,sha256=Sx_DKdM9vCZs7uvWo6H0L9PxU6Je0IYw04T4BwOvVYU,2265
 agentex/types/agent_task_tracker.py,sha256=JK1kmQ7LIx1eWC-XaU2pJcIvdtQCmEn21dsJTxglAYA,803
 agentex/types/data_content.py,sha256=BxJrovgQUCvPP2ac3KTQdErbdDpkZSIbLZhJ4nEDyZE,786
@@ -287,8 +288,8 @@ agentex/types/messages/batch_create_params.py,sha256=trUV75ntEVnxYhd1FIWub6dHBaN
 agentex/types/messages/batch_create_response.py,sha256=yaSLTw2MWWpHF23BGO1Xy9js38_7EIqHAuiLha8VSfc,278
 agentex/types/messages/batch_update_params.py,sha256=Ug5CThbD49a8j4qucg04OdmVrp_gApgpw0hg6NLsR3g,433
 agentex/types/messages/batch_update_response.py,sha256=TbSBe6SuPzjXXWSj-nRjT1JHGBooTshHQQDa1AixQA8,278
-agentex_sdk-0.2.1.dist-info/METADATA,sha256=61uoi5QRrVlFQnrM66ChrdTULQz6U_q0HwEJq07CPv8,14110
-agentex_sdk-0.2.1.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-agentex_sdk-0.2.1.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
-agentex_sdk-0.2.1.dist-info/licenses/LICENSE,sha256=Q1AOx2FtRcMlyMgQJ9eVN2WKPq2mQ33lnB4tvWxabLA,11337
-agentex_sdk-0.2.1.dist-info/RECORD,,
+agentex_sdk-0.2.2.dist-info/METADATA,sha256=WHWvOaMDaUS5QYUJ3PiQaU1uH2zvw7PP2IzpoxoyHYg,14110
+agentex_sdk-0.2.2.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+agentex_sdk-0.2.2.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
+agentex_sdk-0.2.2.dist-info/licenses/LICENSE,sha256=Q1AOx2FtRcMlyMgQJ9eVN2WKPq2mQ33lnB4tvWxabLA,11337
+agentex_sdk-0.2.2.dist-info/RECORD,,