agentex-sdk 0.4.8__py3-none-any.whl → 0.4.9__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.4.8"  # x-release-please-version
+__version__ = "0.4.9"  # x-release-please-version

agentex/lib/adk/_modules/acp.py

@@ -1,6 +1,7 @@
 from datetime import timedelta
-from typing import Any
+from typing import Any, List
 
+from agentex.types import Event
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
@@ -58,7 +59,7 @@ class ACPModule:
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
-    ):
+    ) -> Task:
         """
         Create a new task.
 
@@ -111,7 +112,7 @@ class ACPModule:
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
-    ):
+    ) -> Event:
         """
         Send an event to a task.
 
@@ -152,8 +153,8 @@ class ACPModule:
 
     async def send_message(
         self,
-        task_id: str,
         content: TaskMessageContent,
+        task_id: str | None = None,
         agent_id: str | None = None,
         agent_name: str | None = None,
         trace_id: str | None = None,
@@ -161,7 +162,7 @@ class ACPModule:
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
-    ):
+    ) -> List[TaskMessage]:
         """
         Send a message to a task.
 
@@ -209,7 +210,7 @@ class ACPModule:
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
-    ):
+    ) -> Task:
         """
         Cancel a task.
 

agentex/lib/core/services/adk/acp/acp.py

@@ -116,6 +116,7 @@ class ACPService:
                 raise ValueError("Either agent_name or agent_id must be provided")
 
             task_messages: List[TaskMessage] = []
+            logger.info(f"json_rpc_response: {json_rpc_response}")
             if isinstance(json_rpc_response.result, list):
                 for message in json_rpc_response.result:
                     task_message = TaskMessage.model_validate(message)

agentex/lib/core/temporal/activities/__init__.py

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

agentex/lib/core/temporal/activities/adk/acp/acp_activities.py

@@ -1,5 +1,5 @@
 from enum import Enum
-from typing import Any
+from typing import Any, List
 
 from temporalio import activity
 
@@ -61,7 +61,7 @@ class ACPActivities:
         )
 
     @activity.defn(name=ACPActivityName.MESSAGE_SEND)
-    async def message_send(self, params: MessageSendParams) -> TaskMessage:
+    async def message_send(self, params: MessageSendParams) -> List[TaskMessage]:
         return await self._acp_service.message_send(
             agent_id=params.agent_id,
             agent_name=params.agent_name,

{agentex_sdk-0.4.8.dist-info → agentex_sdk-0.4.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentex-sdk
-Version: 0.4.8
+Version: 0.4.9
 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.4.8.dist-info → agentex_sdk-0.4.9.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=B50Ua8_D0D-KO2Gx_YEx60xydpxFLCIU8BwZh9UZhbY,159
+agentex/_version.py,sha256=uP-Bz46kvosoPnuC2x255N_8qwRibejXkKPv4P-Gg9Q,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,7 +29,7 @@ agentex/lib/environment_variables.py,sha256=6iWpCEURbahnLxZIjvJGES-zs7euUI8bxVXF
 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=zGR4NmmAZM777TFbj2hWX5pES3xFKzblSiJx8hVlRJI,9027
+agentex/lib/adk/_modules/acp.py,sha256=kC8VzbpLoCm9INGOS_6JmzdXVU5w8TFg6dkAekHVoFw,9125
 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
@@ -128,7 +128,7 @@ agentex/lib/core/services/adk/streaming.py,sha256=keZyy7CixM2_3tf2THv-X4gGOOSAk0
 agentex/lib/core/services/adk/tasks.py,sha256=XEX0u4skq81Fn-gjTzSE_-Iv_mKySZHchpotxAsnJwg,2618
 agentex/lib/core/services/adk/tracing.py,sha256=w4mnIoq5m5iwlkDN_NOUjhi6hUpD958cYn1QR6hFgB8,1156
 agentex/lib/core/services/adk/acp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentex/lib/core/services/adk/acp/acp.py,sha256=4KeRPaka8GqnF2PQiNigcijgvQ_rf8Si0EoOdQNqjN4,7926
+agentex/lib/core/services/adk/acp/acp.py,sha256=s5dUYTFTIlkvD0x5g5FYVt9Z9Zt2KEHsWya30dRpO0w,7993
 agentex/lib/core/services/adk/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/core/services/adk/providers/litellm.py,sha256=EKzus_xohNW-85V5hwvd1WqUd3ebv2wc9vDIWO2t1Mw,10044
 agentex/lib/core/services/adk/providers/openai.py,sha256=Ad7iFQDg1iz8KjswO16lzjpzEJCz5FtyXV-fB6qMPLk,39535
@@ -136,7 +136,7 @@ agentex/lib/core/services/adk/providers/sgp.py,sha256=9gm-sPNQ_OSTaBzL0onerKhokP
 agentex/lib/core/services/adk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/core/services/adk/utils/templating.py,sha256=eaXSFq31Y9p5pRD6J6SL4QdTFtxy81dilbF2XXc2JYQ,1889
 agentex/lib/core/temporal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentex/lib/core/temporal/activities/__init__.py,sha256=-XmQcWIMdk7xdpOKdg4rtuIlE0K2UABcIHlzUi8e_os,7634
+agentex/lib/core/temporal/activities/__init__.py,sha256=BWFKs8eFogUHvUyRwGNTpi3QA3KX78XegOzI4cCbZrs,7697
 agentex/lib/core/temporal/activities/activity_helpers.py,sha256=vvQv0rCyrIbnm2gFSQRgCIwIUlKf0jwmI15XEDTNz8A,1014
 agentex/lib/core/temporal/activities/adk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/core/temporal/activities/adk/agent_task_tracker_activities.py,sha256=cqND3YrRybE9Li4fw-oLfTyKzcI9M0lGBhoCq37vThc,2668
@@ -148,7 +148,7 @@ agentex/lib/core/temporal/activities/adk/streaming_activities.py,sha256=PEXeFU9m
 agentex/lib/core/temporal/activities/adk/tasks_activities.py,sha256=hQjdD-TfSzxsiW9eiJ5ZTvY0Ieu0tE950_Zk2PUVc9g,1442
 agentex/lib/core/temporal/activities/adk/tracing_activities.py,sha256=RF8nSBA9x6Ui7_MD9He5ljlH_xZduIhx_JUTEyKdCxA,1523
 agentex/lib/core/temporal/activities/adk/acp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentex/lib/core/temporal/activities/adk/acp/acp_activities.py,sha256=JJgb-UODz3y2gOM5Yf9zQp1MlsptpFUCmz8etLKYijs,2681
+agentex/lib/core/temporal/activities/adk/acp/acp_activities.py,sha256=yWgL99R3rYF2o_A38B92IsJPSdCWFN0DT6DtNEGY9YA,2693
 agentex/lib/core/temporal/activities/adk/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/core/temporal/activities/adk/providers/litellm_activities.py,sha256=Ix92XZIQIbhf-_fOWkUZDxZ1UUCOF0kOwxbcRYBzBQA,2583
 agentex/lib/core/temporal/activities/adk/providers/openai_activities.py,sha256=SLypk8i8hB98hPPcj8rWQwb_Sr4NXoMcfFUV7-dsauU,11096
@@ -300,8 +300,8 @@ agentex/types/messages/batch_update_params.py,sha256=Ug5CThbD49a8j4qucg04OdmVrp_
 agentex/types/messages/batch_update_response.py,sha256=TbSBe6SuPzjXXWSj-nRjT1JHGBooTshHQQDa1AixQA8,278
 agentex/types/shared/__init__.py,sha256=IKs-Qn5Yja0kFh1G1kDqYZo43qrOu1hSoxlPdN-85dI,149
 agentex/types/shared/delete_response.py,sha256=8qH3zvQXaOHYQSHyXi7UQxdR4miTzR7V9K4zXVsiUyk,215
-agentex_sdk-0.4.8.dist-info/METADATA,sha256=_ZN2ZmRBA_8REs6q2Q8gsidbWdcI2kqDPPvoDgqkdZQ,15094
-agentex_sdk-0.4.8.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-agentex_sdk-0.4.8.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
-agentex_sdk-0.4.8.dist-info/licenses/LICENSE,sha256=Q1AOx2FtRcMlyMgQJ9eVN2WKPq2mQ33lnB4tvWxabLA,11337
-agentex_sdk-0.4.8.dist-info/RECORD,,
+agentex_sdk-0.4.9.dist-info/METADATA,sha256=Z4iJELn7KSCCX8g9yo-rvJskP_2euQ-2CW6VXvOFjIE,15094
+agentex_sdk-0.4.9.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+agentex_sdk-0.4.9.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
+agentex_sdk-0.4.9.dist-info/licenses/LICENSE,sha256=Q1AOx2FtRcMlyMgQJ9eVN2WKPq2mQ33lnB4tvWxabLA,11337
+agentex_sdk-0.4.9.dist-info/RECORD,,