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

agentex/lib/adk/providers/_modules/openai.py

@@ -392,7 +392,7 @@ class OpenAIModule:
 
     @deprecated(
         "Use the OpenAI Agents SDK integration with Temporal instead. "
-        "See examples in tutorials/10_agentic/10_temporal/ for migration guidance."
+        "See examples in tutorials/10_async/10_temporal/ for migration guidance."
     )
     async def run_agent_streamed_auto_send(
         self,
@@ -426,7 +426,7 @@ class OpenAIModule:
 
         .. deprecated::
             Use the OpenAI Agents SDK integration with Temporal instead.
-            See examples in tutorials/10_agentic/10_temporal/ for migration guidance.
+            See examples in tutorials/10_async/10_temporal/ for migration guidance.
 
         Args:
             task_id: The ID of the task to run the agent for.

agentex/lib/core/temporal/plugins/openai_agents/hooks/hooks.py

@@ -55,9 +55,8 @@ class TemporalStreamingHooks(RunHooks):
     Power users can ignore this class and subclass agents.RunHooks directly for full control.
 
     Note:
-        Tool arguments are not available in hooks due to OpenAI SDK architecture.
-        The SDK's hook signature doesn't include tool arguments - they're only passed
-        to the actual tool function. This is why arguments={} in ToolRequestContent.
+        Tool arguments are extracted from the ToolContext's tool_arguments field,
+        which contains a JSON string of the arguments passed to the tool.
 
     Attributes:
         task_id: The AgentEx task ID for routing streamed events
@@ -108,21 +107,30 @@ class TemporalStreamingHooks(RunHooks):
     async def on_tool_start(self, context: RunContextWrapper, agent: Agent, tool: Tool) -> None:  # noqa: ARG002
         """Stream tool request when a tool starts execution.
 
-        Extracts the tool_call_id from the context and streams a ToolRequestContent
-        message to the UI showing that the tool is about to execute.
-
-        Note: Tool arguments are not available in the hook context due to OpenAI SDK
-        design. The hook signature doesn't include tool arguments - they're passed
-        directly to the tool function instead. We send an empty dict as a placeholder.
+        Extracts the tool_call_id and tool_arguments from the context and streams a
+        ToolRequestContent message to the UI showing that the tool is about to execute.
 
         Args:
-            context: The run context wrapper (will be a ToolContext with tool_call_id)
+            context: The run context wrapper (will be a ToolContext with tool_call_id and tool_arguments)
             agent: The agent executing the tool
             tool: The tool being executed
         """
+        import json
+
         tool_context = context if isinstance(context, ToolContext) else None
         tool_call_id = tool_context.tool_call_id if tool_context else f"call_{id(tool)}"
 
+        # Extract tool arguments from context
+        tool_arguments = {}
+        if tool_context and hasattr(tool_context, 'tool_arguments'):
+            try:
+                # tool_arguments is a JSON string, parse it
+                tool_arguments = json.loads(tool_context.tool_arguments)
+            except (json.JSONDecodeError, TypeError):
+                # If parsing fails, log and use empty dict
+                logger.warning(f"Failed to parse tool arguments: {tool_context.tool_arguments}")
+                tool_arguments = {}
+
         await workflow.execute_activity_method(
             stream_lifecycle_content,
             args=[
@@ -131,7 +139,7 @@ class TemporalStreamingHooks(RunHooks):
                     author="agent",
                     tool_call_id=tool_call_id,
                     name=tool.name,
-                    arguments={},  # Not available in hook context - SDK limitation
+                    arguments=tool_arguments,  # Now properly extracted from context
                 ),
             ],
             start_to_close_timeout=self.timeout,

agentex/lib/core/temporal/plugins/openai_agents/models/temporal_streaming_model.py

@@ -102,10 +102,22 @@ def _serialize_item(item: Any) -> dict[str, Any]:
 class TemporalStreamingModel(Model):
     """Custom model implementation with streaming support."""
 
-    def __init__(self, model_name: str = "gpt-4o", _use_responses_api: bool = True):
-        """Initialize the streaming model with OpenAI client and model name."""
-        # Match the default behavior with no retries (Temporal handles retries)
-        self.client = AsyncOpenAI(max_retries=0)
+    def __init__(
+        self,
+        model_name: str = "gpt-4o",
+        _use_responses_api: bool = True,
+        openai_client: Optional[AsyncOpenAI] = None,
+    ):
+        """Initialize the streaming model with OpenAI client and model name.
+
+        Args:
+            model_name: The name of the OpenAI model to use (default: "gpt-4o")
+            _use_responses_api: Internal flag for responses API (deprecated, always True)
+            openai_client: Optional custom AsyncOpenAI client. If not provided, a default
+                          client with max_retries=0 will be created (since Temporal handles retries)
+        """
+        # Use provided client or create default (Temporal handles retries)
+        self.client = openai_client if openai_client is not None else AsyncOpenAI(max_retries=0)
         self.model_name = model_name
         # Always use Responses API for all models
         self.use_responses_api = True
@@ -114,7 +126,7 @@ class TemporalStreamingModel(Model):
         agentex_client = create_async_agentex_client()
         self.tracer = AsyncTracer(agentex_client)
 
-        logger.info(f"[TemporalStreamingModel] Initialized model={self.model_name}, use_responses_api={self.use_responses_api}, tracer=initialized")
+        logger.info(f"[TemporalStreamingModel] Initialized model={self.model_name}, use_responses_api={self.use_responses_api}, custom_client={openai_client is not None}, tracer=initialized")
 
     def _non_null_or_not_given(self, value: Any) -> Any:
         """Convert None to NOT_GIVEN sentinel, matching OpenAI SDK pattern."""

agentex/lib/core/temporal/plugins/openai_agents/models/temporal_tracing_model.py

@@ -23,6 +23,7 @@ from agents import (
     TResponseInputItem,
     AgentOutputSchemaBase,
 )
+from openai import AsyncOpenAI
 from openai.types.responses import ResponsePromptParam
 from agents.models.openai_responses import OpenAIResponsesModel
 from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
@@ -86,17 +87,25 @@ class TemporalTracingModelProvider(OpenAIProvider):
     the context interceptor enabled.
     """
 
-    def __init__(self, *args, **kwargs):
+    def __init__(self, openai_client: Optional[AsyncOpenAI] = None, **kwargs):
         """Initialize the tracing model provider.
 
-        Accepts all the same arguments as OpenAIProvider.
+        Args:
+            openai_client: Optional custom AsyncOpenAI client. If provided, this client
+                          will be used for all model calls. If not provided, OpenAIProvider
+                          will create a default client.
+            **kwargs: All other arguments are passed to OpenAIProvider.
         """
-        super().__init__(*args, **kwargs)
+        # Pass openai_client to parent if provided
+        if openai_client is not None:
+            super().__init__(openai_client=openai_client, **kwargs)
+        else:
+            super().__init__(**kwargs)
 
         # Initialize tracer for all models
         agentex_client = create_async_agentex_client()
         self._tracer = AsyncTracer(agentex_client)
-        logger.info("[TemporalTracingModelProvider] Initialized with AgentEx tracer")
+        logger.info(f"[TemporalTracingModelProvider] Initialized with AgentEx tracer, custom_client={openai_client is not None}")
 
     @override
     def get_model(self, model_name: Optional[str]) -> Model:

{agentex_sdk-0.6.2.dist-info → agentex_sdk-0.6.3.dist-info}/METADATA

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

{agentex_sdk-0.6.2.dist-info → agentex_sdk-0.6.3.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=p-m2didLkbw_VBZsP4QqeIPc2haAdGZmB0BOU3gUM2A,10153
 agentex/_types.py,sha256=F6X63N7bOstytAtVqJ9Yl7T_JbR9Od2MJfZ_iK5DqOY,7237
-agentex/_version.py,sha256=e3De3d0qz0jkC87-IeW5SrqByX5rqRyZ0CQ6lTUa5EM,159
+agentex/_version.py,sha256=c2fQYF_Uula9lejrW5sHf4Mfcdn6IMrnCDZlp3EDgno,159
 agentex/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/_utils/__init__.py,sha256=7fch0GT9zpNnErbciSpUNa-SjTxxjY6kxHxKMOM4AGs,2305
 agentex/_utils/_compat.py,sha256=D8gtAvjJQrDWt9upS0XaG9Rr5l1QhiAx_I_1utT_tt0,1195
@@ -43,7 +43,7 @@ agentex/lib/adk/_modules/tracing.py,sha256=LZvItZg2ALZZMvJasrsGEBgss4wnXGMnL_lw0
 agentex/lib/adk/providers/__init__.py,sha256=bOS-D_lXV3QXRtGKrUvsYb2ZAcZG51ZAtHdfiHgYN-M,306
 agentex/lib/adk/providers/_modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/adk/providers/_modules/litellm.py,sha256=6eOdEpd1g9ZljQHXtIokjGymlFpnaw2g8ow11hldovo,9470
-agentex/lib/adk/providers/_modules/openai.py,sha256=aJqt6jfX80xcviXryrppd4D83Vi-C8gPspdUQujOWdU,23683
+agentex/lib/adk/providers/_modules/openai.py,sha256=rguTE4uGLodqr8zx5GRRtSkcWC9llNA1kUFKF0yhdK0,23679
 agentex/lib/adk/providers/_modules/sgp.py,sha256=x64axb0oVmVh5W8hwpnMMPqxad2HySf2DYHPxRNwjck,3208
 agentex/lib/adk/providers/_modules/sync_provider.py,sha256=9RQ_gMICiUN97bvnbVnVypiKoalCeiUiqonn3nKAjwU,26324
 agentex/lib/adk/utils/__init__.py,sha256=7f6ayV0_fqyw5cwzVANNcZWGJZ-vrrYtZ0qi7KKBRFs,130
@@ -167,12 +167,12 @@ agentex/lib/core/temporal/plugins/openai_agents/README.md,sha256=FzvW3xM7dCvdxq2
 agentex/lib/core/temporal/plugins/openai_agents/__init__.py,sha256=ZLHVA3LAkPDvRQlSLnJlPs0sDzqy6sUtwlapoG02Hyw,3174
 agentex/lib/core/temporal/plugins/openai_agents/hooks/__init__.py,sha256=soOuozGd7H9uWSwwjA60psUGtRR4xQ6e4wfoQbr73Fo,483
 agentex/lib/core/temporal/plugins/openai_agents/hooks/activities.py,sha256=APpAOZQ90kbhaNpJ6pAeZ2mxxcPx37AchCuLJzeGzB0,3088
-agentex/lib/core/temporal/plugins/openai_agents/hooks/hooks.py,sha256=iATUCspNLweCGuG8WGd29wK8_SiZPlor5P3G9p5nvH0,7853
+agentex/lib/core/temporal/plugins/openai_agents/hooks/hooks.py,sha256=qbB6RLPlveEIIfziXZBRMkaPJQMQBPsovIX_yM4QuWU,8100
 agentex/lib/core/temporal/plugins/openai_agents/interceptors/__init__.py,sha256=hrj6lRPi9nb_HAohRK4oPnaji69QQ6brj-Wu2q0mU0s,521
 agentex/lib/core/temporal/plugins/openai_agents/interceptors/context_interceptor.py,sha256=sBLJonJJ5Ke1BJIlzbqtGeO5p8NIbvftbEYQbjgeZCE,7256
 agentex/lib/core/temporal/plugins/openai_agents/models/__init__.py,sha256=FeTt91JkSfYLlCTdrVFpjcQ0asbQyCd6Rl5efqZkslo,791
-agentex/lib/core/temporal/plugins/openai_agents/models/temporal_streaming_model.py,sha256=GpEYqXdf_fYEQVti-hxGtdsQ2XnG5AOg8aFvGH00z3c,41755
-agentex/lib/core/temporal/plugins/openai_agents/models/temporal_tracing_model.py,sha256=BrBX4rClOq3g_V2_UlKJVRtX8E_hu8dYiQiX-js-pto,16913
+agentex/lib/core/temporal/plugins/openai_agents/models/temporal_streaming_model.py,sha256=ZODPIUuvSZO5eYhByMv2n87SOKYjAh9JunkJm8ECRPQ,42310
+agentex/lib/core/temporal/plugins/openai_agents/models/temporal_tracing_model.py,sha256=BiuIhSvyNfocwMYQtxOoqgMpyJsMHLkyXzYPYnw4ChA,17458
 agentex/lib/core/temporal/plugins/openai_agents/tests/__init__.py,sha256=suEVJuonfBoVZ3IqdO0UMn0hkFFzDqRoso0VEOit-KQ,80
 agentex/lib/core/temporal/plugins/openai_agents/tests/conftest.py,sha256=oMI_3dVn6DoiLgCjRVUeQE_Z2Gz3tGTwPxTQ1krjKSE,7692
 agentex/lib/core/temporal/plugins/openai_agents/tests/test_streaming_model.py,sha256=w8rkQbn3j_f9GZSXHl5j8FMDPaBQ3wED-BFzxQaeBIc,32909
@@ -330,8 +330,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.6.2.dist-info/METADATA,sha256=BmFDFE0X8Rhv-8mMdZBIDPOR0Bt3vrvA9mHndpVFUuY,15375
-agentex_sdk-0.6.2.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-agentex_sdk-0.6.2.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
-agentex_sdk-0.6.2.dist-info/licenses/LICENSE,sha256=Q1AOx2FtRcMlyMgQJ9eVN2WKPq2mQ33lnB4tvWxabLA,11337
-agentex_sdk-0.6.2.dist-info/RECORD,,
+agentex_sdk-0.6.3.dist-info/METADATA,sha256=FPfO3b3DUDmtZvVDGhsammLJgD_PlhwLG-zQqo7DzKI,15375
+agentex_sdk-0.6.3.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+agentex_sdk-0.6.3.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
+agentex_sdk-0.6.3.dist-info/licenses/LICENSE,sha256=Q1AOx2FtRcMlyMgQJ9eVN2WKPq2mQ33lnB4tvWxabLA,11337
+agentex_sdk-0.6.3.dist-info/RECORD,,