PyPI - agentex-sdk - Versions diffs - 0.2.1__py3-none-any.whl → 0.2.3__py3-none-any.whl - Mend

agentex-sdk 0.2.1py3-none-any.whl → 0.2.3py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (12) hide show

agentex/_version.py CHANGED Viewed

@@ -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.3"  # x-release-please-version

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

@@ -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 CHANGED Viewed

@@ -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 CHANGED Viewed

@@ -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/resources/agents.py CHANGED Viewed

@@ -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 CHANGED Viewed

@@ -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 CHANGED Viewed

@@ -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.3.dist-info}/METADATA RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentex-sdk
-Version: 0.2.1
+Version: 0.2.3
 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.3.dist-info}/RECORD RENAMED Viewed

@@ -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=gaveMH33nXDCGMmyPCNNYeNapFs-32mlnSES-b86nfA,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
@@ -64,7 +64,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 +76,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 +86,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
@@ -220,11 +220,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 +236,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 +287,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.3.dist-info/METADATA,sha256=hoWAam811pyQl421I9RIHGvrZc3O8lVUWaTZbd-y4UQ,14110
+agentex_sdk-0.2.3.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+agentex_sdk-0.2.3.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
+agentex_sdk-0.2.3.dist-info/licenses/LICENSE,sha256=Q1AOx2FtRcMlyMgQJ9eVN2WKPq2mQ33lnB4tvWxabLA,11337
+agentex_sdk-0.2.3.dist-info/RECORD,,

{agentex_sdk-0.2.1.dist-info → agentex_sdk-0.2.3.dist-info}/WHEEL RENAMED Viewed

File without changes

{agentex_sdk-0.2.1.dist-info → agentex_sdk-0.2.3.dist-info}/entry_points.txt RENAMED Viewed

File without changes

{agentex_sdk-0.2.1.dist-info → agentex_sdk-0.2.3.dist-info}/licenses/LICENSE RENAMED Viewed

File without changes

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

agentex-sdk 0.2.1py3-none-any.whl → 0.2.3py3-none-any.whl