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

agentex/lib/adk/_modules/acp.py

@@ -23,7 +23,7 @@ from agentex.types.task_message_content import TaskMessageContent
 
 logger = make_logger(__name__)
 
-DEFAULT_RETRY_POLICY = RetryPolicy(maximum_attempts=1)
+DEFAULT_RETRY_POLICY = RetryPolicy(maximum_attempts=0)
 
 
 class ACPModule:

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

@@ -20,10 +20,6 @@ class ActivityHelpers:
         heartbeat_timeout: timedelta | None = None,
         retry_policy: RetryPolicy | None = None,
     ) -> Any:
-        if start_to_close_timeout is None:
-            start_to_close_timeout = timedelta(seconds=10)
-        if retry_policy is None:
-            retry_policy = RetryPolicy(maximum_attempts=0)
 
         response = await workflow.execute_activity(
             activity=activity_name,

agentex/lib/core/temporal/activities/adk/providers/openai_activities.py

@@ -1,10 +1,14 @@
 # Standard library imports
+import base64
 from collections.abc import Callable
 from contextlib import AsyncExitStack, asynccontextmanager
 from enum import Enum
-from typing import Any, Literal
+from typing import Any, Literal, Optional, override
 
-from agents import RunResult, RunResultStreaming
+from pydantic import Field, PrivateAttr
+
+import cloudpickle
+from agents import RunContextWrapper, RunResult, RunResultStreaming
 from agents.mcp import MCPServerStdio, MCPServerStdioParams
 from agents.model_settings import ModelSettings as OAIModelSettings
 from agents.tool import FunctionTool as OAIFunctionTool
@@ -41,12 +45,92 @@ class FunctionTool(BaseModelWithTraceParams):
     name: str
     description: str
     params_json_schema: dict[str, Any]
-    on_invoke_tool: Callable[[dict[str, Any]], Any]
+
     strict_json_schema: bool = True
     is_enabled: bool = True
 
+    _on_invoke_tool: Callable[[RunContextWrapper, str], Any] = PrivateAttr()
+    on_invoke_tool_serialized: str = Field(
+        default="",
+        description=(
+            "Normally will be set automatically during initialization and"
+            " doesn't need to be passed. "
+            "Instead, pass `on_invoke_tool` to the constructor. "
+            "See the __init__ method for details."
+        ),
+    )
+
+    def __init__(
+        self,
+        *,
+        on_invoke_tool: Optional[Callable[[RunContextWrapper, str], Any]] = None,
+        **data,
+    ):
+        """
+        Initialize a FunctionTool with hacks to support serialization of the
+         on_invoke_tool callable arg. This is required to facilitate over-the-wire
+         communication of this object to/from temporal services/workers.
+
+        Args:
+            on_invoke_tool: The callable to invoke when the tool is called.
+            **data: Additional data to initialize the FunctionTool.
+        """
+        super().__init__(**data)
+        if not on_invoke_tool:
+            if not self.on_invoke_tool_serialized:
+                raise ValueError(
+                    "One of `on_invoke_tool` or `on_invoke_tool_serialized` should be set"
+                )
+            else:
+                on_invoke_tool = self._deserialize_callable(
+                    self.on_invoke_tool_serialized
+                )
+        else:
+            self.on_invoke_tool_serialized = self._serialize_callable(on_invoke_tool)
+
+        self._on_invoke_tool = on_invoke_tool
+
+    @classmethod
+    def _deserialize_callable(
+        cls, serialized: str
+    ) -> Callable[[RunContextWrapper, str], Any]:
+        encoded = serialized.encode()
+        serialized_bytes = base64.b64decode(encoded)
+        return cloudpickle.loads(serialized_bytes)
+
+    @classmethod
+    def _serialize_callable(cls, func: Callable) -> str:
+        serialized_bytes = cloudpickle.dumps(func)
+        encoded = base64.b64encode(serialized_bytes)
+        return encoded.decode()
+
+    @property
+    def on_invoke_tool(self) -> Callable[[RunContextWrapper, str], Any]:
+        if self._on_invoke_tool is None and self.on_invoke_tool_serialized:
+            self._on_invoke_tool = self._deserialize_callable(
+                self.on_invoke_tool_serialized
+            )
+        return self._on_invoke_tool
+    
+    @on_invoke_tool.setter
+    def on_invoke_tool(self, value: Callable[[RunContextWrapper, str], Any]):
+        self.on_invoke_tool_serialized = self._serialize_callable(value)
+        self._on_invoke_tool = value
+
     def to_oai_function_tool(self) -> OAIFunctionTool:
-        return OAIFunctionTool(**self.model_dump(exclude=["trace_id", "parent_span_id"]))
+        """Convert to OpenAI function tool, excluding serialization fields."""
+        # Create a dictionary with only the fields OAIFunctionTool expects
+        data = self.model_dump(
+            exclude={
+                "trace_id",
+                "parent_span_id",
+                "_on_invoke_tool",
+                "on_invoke_tool_serialized",
+            }
+        )
+        # Add the callable for OAI tool since properties are not serialized
+        data["on_invoke_tool"] = self.on_invoke_tool
+        return OAIFunctionTool(**data)
 
 
 class ModelSettings(BaseModelWithTraceParams):
@@ -68,7 +152,9 @@ class ModelSettings(BaseModelWithTraceParams):
     extra_args: dict[str, Any] | None = None
 
     def to_oai_model_settings(self) -> OAIModelSettings:
-        return OAIModelSettings(**self.model_dump(exclude=["trace_id", "parent_span_id"]))
+        return OAIModelSettings(
+            **self.model_dump(exclude=["trace_id", "parent_span_id"])
+        )
 
 
 class RunAgentParams(BaseModelWithTraceParams):

agentex/lib/utils/model_utils.py

@@ -34,11 +34,31 @@ class BaseModel(PydanticBaseModel):
 
 def recursive_model_dump(obj: Any) -> Any:
     if isinstance(obj, PydanticBaseModel):
-        # Serialize BaseModel to dict
-        return obj.model_dump(mode="json")
+        # Get the model data as dict and recursively process each field
+        # This allows us to handle non-serializable objects like functions
+        try:
+            return obj.model_dump(mode="json")
+        except Exception:
+            # If model_dump fails (e.g., due to functions), manually process
+            model_dict = {}
+            for field_name in obj.__class__.model_fields:
+                field_value = getattr(obj, field_name)
+                model_dict[field_name] = recursive_model_dump(field_value)
+            return model_dict
     elif isinstance(obj, datetime):
         # Serialize datetime to ISO format string
         return obj.isoformat()
+    elif callable(obj):
+        # Serialize functions and other callable objects
+        if hasattr(obj, "__name__"):
+            func_name = obj.__name__
+        else:
+            func_name = str(obj)
+
+        if hasattr(obj, "__module__"):
+            return f"<function {obj.__module__}.{func_name}>"
+        else:
+            return f"<function {func_name}>"
     elif isinstance(obj, Mapping):
         # Recursively serialize dictionary values
         return {k: recursive_model_dump(v) for k, v in obj.items()}

agentex/types/__init__.py

@@ -11,6 +11,7 @@ from .shared import DeleteResponse as DeleteResponse
 from .acp_type import AcpType as AcpType
 from .data_delta import DataDelta as DataDelta
 from .text_delta import TextDelta as TextDelta
+from .text_format import TextFormat as TextFormat
 from .data_content import DataContent as DataContent
 from .task_message import TaskMessage as TaskMessage
 from .text_content import TextContent as TextContent

agentex/types/text_content.py

@@ -4,6 +4,7 @@ from typing import List, Optional
 from typing_extensions import Literal
 
 from .._models import BaseModel
+from .text_format import TextFormat
 from .message_style import MessageStyle
 from .message_author import MessageAuthor
 
@@ -37,7 +38,7 @@ class TextContent(BaseModel):
     attachments: Optional[List[Attachment]] = None
     """Optional list of file attachments with structured metadata."""
 
-    format: Literal["markdown", "plain", "code"] = "plain"
+    format: TextFormat = "plain"
     """The format of the message.
 
     This is used by the client to determine how to display the message.

agentex/types/text_content_param.py

@@ -5,6 +5,7 @@ from __future__ import annotations
 from typing import Iterable, Optional
 from typing_extensions import Literal, Required, TypedDict
 
+from .text_format import TextFormat
 from .message_style import MessageStyle
 from .message_author import MessageAuthor
 
@@ -38,7 +39,7 @@ class TextContentParam(TypedDict, total=False):
     attachments: Optional[Iterable[Attachment]]
     """Optional list of file attachments with structured metadata."""
 
-    format: Literal["markdown", "plain", "code"]
+    format: TextFormat
     """The format of the message.
 
     This is used by the client to determine how to display the message.

agentex/types/text_format.py

@@ -0,0 +1,7 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing_extensions import Literal, TypeAlias
+
+__all__ = ["TextFormat"]
+
+TextFormat: TypeAlias = Literal["markdown", "plain", "code"]

{agentex_sdk-0.4.4.dist-info → agentex_sdk-0.4.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentex-sdk
-Version: 0.4.4
+Version: 0.4.5
 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
@@ -21,6 +21,7 @@ Classifier: Typing :: Typed
 Requires-Python: <4,>=3.12
 Requires-Dist: aiohttp<4,>=3.10.10
 Requires-Dist: anyio<5,>=3.5.0
+Requires-Dist: cloudpickle>=3.1.1
 Requires-Dist: distro<2,>=1.7.0
 Requires-Dist: fastapi<0.116,>=0.115.0
 Requires-Dist: httpx<0.28,>=0.27.2
@@ -32,7 +33,7 @@ Requires-Dist: kubernetes<29.0.0,>=25.0.0
 Requires-Dist: litellm<2,>=1.66.0
 Requires-Dist: mcp[cli]>=1.4.1
 Requires-Dist: openai-agents==0.2.7
-Requires-Dist: openai>=1.99.9
+Requires-Dist: openai==1.99.9
 Requires-Dist: pydantic<3,>=2.0.0
 Requires-Dist: pytest-asyncio>=1.0.0
 Requires-Dist: pytest>=8.4.0

{agentex_sdk-0.4.4.dist-info → agentex_sdk-0.4.5.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=5yXU0s9SIFUQBbJSqTSYIrCR8f9aMgnkR1KwqcIpRgU,159
+agentex/_version.py,sha256=Z94_mPwwCwrKDFOOptIOf2dWIIEp31IIhpHBidxw9vU,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=PFGIDIa0RdkkJVr3pxqh2QHrNRPBwGsf-ulN
 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=0D9HLwBO3SX4eEsx1BY9G90jyvTABmaxdOzANg1rmzQ,9027
+agentex/lib/adk/_modules/acp.py,sha256=zGR4NmmAZM777TFbj2hWX5pES3xFKzblSiJx8hVlRJI,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
@@ -137,7 +137,7 @@ agentex/lib/core/services/adk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQ
 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/activity_helpers.py,sha256=QCiEFwNSMtpJuZKiAV_UolTtlf84NZk_lG3EEQmQT0s,1208
+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
 agentex/lib/core/temporal/activities/adk/agents_activities.py,sha256=q3PGgOLXvcEOs1bJaTT9TuEZIVhdmzGNNnDiOr7Wh5M,1004
@@ -151,7 +151,7 @@ agentex/lib/core/temporal/activities/adk/acp/__init__.py,sha256=47DEQpj8HBSa-_TI
 agentex/lib/core/temporal/activities/adk/acp/acp_activities.py,sha256=JJgb-UODz3y2gOM5Yf9zQp1MlsptpFUCmz8etLKYijs,2681
 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=ck_G9sSFZ74zmH_yi9-M07JvjjQFzMAkBZxhunUCP2c,8001
+agentex/lib/core/temporal/activities/adk/providers/openai_activities.py,sha256=SLypk8i8hB98hPPcj8rWQwb_Sr4NXoMcfFUV7-dsauU,11096
 agentex/lib/core/temporal/activities/adk/providers/sgp_activities.py,sha256=C8n97mEhc3yFDgdejERDob9g5STfWgGoY-BmkQBH6Lg,1266
 agentex/lib/core/temporal/activities/adk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/core/temporal/activities/adk/utils/templating_activities.py,sha256=dX6wpcTKY_qWsm3uJV-XU5fQLqwWAhnjO83k8EK0N7o,1189
@@ -220,7 +220,7 @@ agentex/lib/utils/iterables.py,sha256=cD49hj98mtY0QPaS0ZA2kguNeGzRU4UfbZfwyASWdz
 agentex/lib/utils/json_schema.py,sha256=nSHbi6LC-oWXHP6sMLCBychA7B0R2DItRIJNCCEzRsY,741
 agentex/lib/utils/logging.py,sha256=RychRZ4I5JlhBRtkP6KGMZBEcRVNN8mpBjv03aV2S0E,798
 agentex/lib/utils/mcp.py,sha256=lYQPwKAOH6Gf2I_TeovhEG9YUijUPcN0rENNdT0Vk6c,505
-agentex/lib/utils/model_utils.py,sha256=DC1CHm_X7ZZ2skBdbY7sEepEmgQlOoohswRD8I92I20,1709
+agentex/lib/utils/model_utils.py,sha256=bkcB1I0DMRAtFQpznsXgFGKZGrT7NGJ4tIwFM4dtUXQ,2547
 agentex/lib/utils/parsing.py,sha256=2T-B4nJSupo2RLf9AkpV6VQHqDnF97ZBgO6zvGCauCo,369
 agentex/lib/utils/regex.py,sha256=Y3VcehCznAqa59D4WTwK_ki722oudHBlFqBk0I930zo,212
 agentex/lib/utils/registration.py,sha256=FAL40KSMRCltHZcDtioGeRNgfB19Twqu6Srp4x7UMSU,3781
@@ -237,7 +237,7 @@ agentex/resources/tracker.py,sha256=YxSeiloMwIqrS9nY7SlHclauRA7142qrKw34xgwqYbA,
 agentex/resources/messages/__init__.py,sha256=_J1eusFtr_k6zrAntJSuqx6LWEUBSTrV1OZZh7MaDPE,1015
 agentex/resources/messages/batch.py,sha256=pegCmnjK_J0jek5ChX1pKpq5RmCucTYLbK69H6qGVS4,9629
 agentex/resources/messages/messages.py,sha256=fuFmmGNOjRJVzmYHcfP2trg0yii0n9MPPCpt7F8fDs4,17930
-agentex/types/__init__.py,sha256=_tsP_hk7b8V3b2wCIw1jjdIsedgmtlI5Jd0ijFtk6NA,3834
+agentex/types/__init__.py,sha256=lEuWxr1ggBzZAUGYrSmIhcSSZ7z5UQSIfNtmUrELC2s,3884
 agentex/types/acp_type.py,sha256=Fj-4SzmM6m95ck_ZXtNbcWggHiD9F49bxBLPbl1fxe4,208
 agentex/types/agent.py,sha256=t6CUsKSaK6Gs_X9g0YxSi5G3Hh21FrGVGnAD6MytVBk,981
 agentex/types/agent_list_params.py,sha256=81IWnRZ2rLfHH7GB6VkXShYjb44DO0guG1znJcV3tuI,316
@@ -281,9 +281,10 @@ agentex/types/task_message_content.py,sha256=3itFPfk9V3AHggZFpHKHkkzostFpyZ8O4vM
 agentex/types/task_message_content_param.py,sha256=IZxdUJWkUynteRQa-y743fCyum54IR6nqr0uybfUTO4,675
 agentex/types/task_message_delta.py,sha256=3i1Qw2CcZYxXhofbCJnQDO0QrIIQsLX6P-UAynvc4fs,716
 agentex/types/task_message_update.py,sha256=gevnU4UW67qTzN_OGT-S3E5zbS9KQNehJZF3XhiirtU,2247
-agentex/types/text_content.py,sha256=i6ksdQXIyWosXAm-O_26m_eZA-4x26dgoNdzWnhue9k,1347
-agentex/types/text_content_param.py,sha256=5xtCBUJj6xKGlmpNuWVnBG3BBFQS3HyrKjq6KfpfVz0,1439
+agentex/types/text_content.py,sha256=-rNhHh0L_b3dxnG49Baf47JRNnDGW0xhglq8xkaGDdE,1357
+agentex/types/text_content_param.py,sha256=A2Xuxclh44Ih-C9etF5gl_VBNF-Ld36ZzI61SNCPdUE,1449
 agentex/types/text_delta.py,sha256=lUTceOXVsh-O4QOzZnkpB3I9d6V1Z0co0JwPZsXnXjQ,322
+agentex/types/text_format.py,sha256=b73CgswRSlvdLzHi_0QQCiZaGRyifkgXN8Yihrhshgs,224
 agentex/types/tool_request_content.py,sha256=8swTXUME4x2_CaQuAwA1hNYjjsmFHEEWT4qodIAQXqc,975
 agentex/types/tool_request_content_param.py,sha256=st06ZsV9qigwPBprhoKyzGOUeV40q27dwOWbWxKxmHk,1025
 agentex/types/tool_request_delta.py,sha256=AUi9LoPwxyOx_JGOTOvSND-9oulsJk2y3Q7RlCgiqGI,387
@@ -300,8 +301,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.4.dist-info/METADATA,sha256=bP_6Dg7xzW4Xy82BSq1t5bGKSnYZOjwagJ1L2PBtAkg,15059
-agentex_sdk-0.4.4.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-agentex_sdk-0.4.4.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
-agentex_sdk-0.4.4.dist-info/licenses/LICENSE,sha256=Q1AOx2FtRcMlyMgQJ9eVN2WKPq2mQ33lnB4tvWxabLA,11337
-agentex_sdk-0.4.4.dist-info/RECORD,,
+agentex_sdk-0.4.5.dist-info/METADATA,sha256=vjmBM8ry7PTbX8W5WC9T8xrVuFisBxqOhksx7NEolCg,15093
+agentex_sdk-0.4.5.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+agentex_sdk-0.4.5.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
+agentex_sdk-0.4.5.dist-info/licenses/LICENSE,sha256=Q1AOx2FtRcMlyMgQJ9eVN2WKPq2mQ33lnB4tvWxabLA,11337
+agentex_sdk-0.4.5.dist-info/RECORD,,