agentex-sdk 0.9.0__py3-none-any.whl → 0.9.2__py3-none-any.whl

agentex/_base_client.py

@@ -86,6 +86,7 @@ from ._exceptions import (
     APIConnectionError,
     APIResponseValidationError,
 )
+from ._utils._json import openapi_dumps
 
 log: logging.Logger = logging.getLogger(__name__)
 
@@ -554,8 +555,10 @@ class BaseClient(Generic[_HttpxClientT, _DefaultStreamT]):
                 kwargs["content"] = options.content
             elif isinstance(json_data, bytes):
                 kwargs["content"] = json_data
-            else:
-                kwargs["json"] = json_data if is_given(json_data) else None
+            elif not files:
+                # Don't set content when JSON is sent as multipart/form-data,
+                # since httpx's content param overrides other body arguments
+                kwargs["content"] = openapi_dumps(json_data) if is_given(json_data) and json_data is not None else None
             kwargs["files"] = files
         else:
             headers.pop("Content-Type", None)

agentex/_compat.py

@@ -139,6 +139,7 @@ def model_dump(
     exclude_defaults: bool = False,
     warnings: bool = True,
     mode: Literal["json", "python"] = "python",
+    by_alias: bool | None = None,
 ) -> dict[str, Any]:
     if (not PYDANTIC_V1) or hasattr(model, "model_dump"):
         return model.model_dump(
@@ -148,13 +149,12 @@ def model_dump(
             exclude_defaults=exclude_defaults,
             # warnings are not supported in Pydantic v1
             warnings=True if PYDANTIC_V1 else warnings,
+            by_alias=by_alias,
         )
     return cast(
         "dict[str, Any]",
         model.dict(  # pyright: ignore[reportDeprecated, reportUnnecessaryCast]
-            exclude=exclude,
-            exclude_unset=exclude_unset,
-            exclude_defaults=exclude_defaults,
+            exclude=exclude, exclude_unset=exclude_unset, exclude_defaults=exclude_defaults, by_alias=bool(by_alias)
         ),
     )
 

agentex/_utils/_json.py

@@ -0,0 +1,35 @@
+import json
+from typing import Any
+from datetime import datetime
+from typing_extensions import override
+
+import pydantic
+
+from .._compat import model_dump
+
+
+def openapi_dumps(obj: Any) -> bytes:
+    """
+    Serialize an object to UTF-8 encoded JSON bytes.
+
+    Extends the standard json.dumps with support for additional types
+    commonly used in the SDK, such as `datetime`, `pydantic.BaseModel`, etc.
+    """
+    return json.dumps(
+        obj,
+        cls=_CustomEncoder,
+        # Uses the same defaults as httpx's JSON serialization
+        ensure_ascii=False,
+        separators=(",", ":"),
+        allow_nan=False,
+    ).encode()
+
+
+class _CustomEncoder(json.JSONEncoder):
+    @override
+    def default(self, o: Any) -> Any:
+        if isinstance(o, datetime):
+            return o.isoformat()
+        if isinstance(o, pydantic.BaseModel):
+            return model_dump(o, exclude_unset=True, mode="json", by_alias=True)
+        return super().default(o)

agentex/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "agentex"
-__version__ = "0.9.0"  # x-release-please-version
+__version__ = "0.9.2"  # x-release-please-version

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

@@ -26,7 +26,13 @@ from agentex.lib.core.temporal.activities.adk.providers.litellm_activities impor
 logger = make_logger(__name__)
 
 # Default retry policy for all LiteLLM operations
-DEFAULT_RETRY_POLICY = RetryPolicy(maximum_attempts=1)
+# Retries with exponential backoff: 1s, 2s, 4s, ... up to 30s between attempts
+DEFAULT_RETRY_POLICY = RetryPolicy(
+    maximum_attempts=3,
+    initial_interval=timedelta(seconds=1),
+    backoff_coefficient=2.0,
+    maximum_interval=timedelta(seconds=30),
+)
 
 
 class LiteLLMModule:

agentex/lib/types/llm_messages.py

@@ -58,6 +58,7 @@ class LLMConfig(BaseModel):
     parallel_tool_calls: bool | None = None
     logprobs: bool | None = None
     top_logprobs: int | None = None
+    num_retries: int | None = 3
 
 
 class ContentPartText(BaseModel):

{agentex_sdk-0.9.0.dist-info → agentex_sdk-0.9.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentex-sdk
-Version: 0.9.0
+Version: 0.9.2
 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
@@ -47,7 +47,7 @@ Requires-Dist: pyyaml<7,>=6.0.2
 Requires-Dist: questionary<3,>=2.0.1
 Requires-Dist: redis<6,>=5.2.0
 Requires-Dist: rich<14,>=13.9.2
-Requires-Dist: scale-gp-beta==0.1.0a20
+Requires-Dist: scale-gp-beta>=0.1.0a20
 Requires-Dist: scale-gp>=0.1.0a59
 Requires-Dist: sniffio
 Requires-Dist: temporalio<2,>=1.18.2

{agentex_sdk-0.9.0.dist-info → agentex_sdk-0.9.2.dist-info}/RECORD

@@ -1,7 +1,7 @@
 agentex/__init__.py,sha256=TvS8DtvGAnubcoUjYIsuCpBzpsdpxBaJCS76s-l-PRo,2712
-agentex/_base_client.py,sha256=gb-HyiWzYErOgODj-HypzcOTPQnxZxCuW2wZD2wixrY,73410
+agentex/_base_client.py,sha256=M7ZqaZpxBrCwUhnZAfGUHn42kRjqsG1RYYoeLeDbDs4,73659
 agentex/_client.py,sha256=39ydvoIqEPVYkM891ZMhCMKrbXR59QUxuKTtrTJGgMM,28482
-agentex/_compat.py,sha256=DQBVORjFb33zch24jzkhM14msvnzY7mmSmgDLaVFUM8,6562
+agentex/_compat.py,sha256=teO44AYozpv2wFRrUi7brcZfGPpFSERQZ4fcdX6zVvs,6627
 agentex/_constants.py,sha256=oGldMuFz7eZtwD8_6rJUippKhZB5fGSA7ffbCDGourA,466
 agentex/_exceptions.py,sha256=B09aFjWFRSShb9BFJd-MNDblsGDyGk3w-vItYmjg_AI,3222
 agentex/_files.py,sha256=KnEzGi_O756MvKyJ4fOCW_u3JhOeWPQ4RsmDvqihDQU,3545
@@ -11,11 +11,12 @@ agentex/_resource.py,sha256=S1t7wmR5WUvoDIhZjo_x-E7uoTJBynJ3d8tPJMQYdjw,1106
 agentex/_response.py,sha256=Tb9zazsnemO2rTxWtBjAD5WBqlhli5ZaXGbiKgdu5DE,28794
 agentex/_streaming.py,sha256=aOvLOte7kaclPGm-D0iNdM3uRcWrZ-T2B8t9BDNmpuA,10225
 agentex/_types.py,sha256=fa7vkH4XTd2n7GdGxf8-i_2kMDtXgwO3X_usN1GyknI,7595
-agentex/_version.py,sha256=QknnjghXHjI-VZe2lpTxhOcs4xErxRJqPkosahapkpw,159
+agentex/_version.py,sha256=OawrYfHrb4YHUdja00eu4HXsP48H_fuwTx6l7EDpZ6I,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
 agentex/_utils/_datetime_parse.py,sha256=bABTs0Bc6rabdFvnIwXjEhWL15TcRgWZ_6XGTqN8xUk,4204
+agentex/_utils/_json.py,sha256=bl95uuIWwgSfXX-gP1trK_lDAPwJujYfJ05Cxo2SEC4,962
 agentex/_utils/_logs.py,sha256=LUjFPc3fweSChBUmjhQD8uYmwQAmFMNDuVFKfjYBQfM,777
 agentex/_utils/_proxy.py,sha256=aglnj2yBTDyGX9Akk2crZHrl10oqRmceUy2Zp008XEs,1975
 agentex/_utils/_reflection.py,sha256=ZmGkIgT_PuwedyNBrrKGbxoWtkpytJNU1uU4QHnmEMU,1364
@@ -42,7 +43,7 @@ agentex/lib/adk/_modules/tasks.py,sha256=aWrOrRvMYfPjFyI6Wt6-5STTyQsTaVZUFZ0XdWC
 agentex/lib/adk/_modules/tracing.py,sha256=LZvItZg2ALZZMvJasrsGEBgss4wnXGMnL_lw0l15z04,7505
 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/litellm.py,sha256=zWy6VuG8pG4qR4k-QARmNAPmEFEnhx67Clv8IaReLVU,9672
 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=YNiApD2NQO0wQXkGAtQwz72dx9HPxgm4UZ_R-Vs-yds,31369
@@ -260,7 +261,7 @@ agentex/lib/types/credentials.py,sha256=k6gtiIqSsKuS8vgucgQAHzb8oUlFdTj4CLtU69sk
 agentex/lib/types/fastacp.py,sha256=WlFfChfd3CKovZHcVsVXkFEw58SLwvZRnf6iU_humaU,2239
 agentex/lib/types/files.py,sha256=sNxiuR_oEo7Z0YMqjQShErvS3txWyobrZzc4QMMM61U,320
 agentex/lib/types/json_rpc.py,sha256=jxr8Jb4M3Z4F58gx6aRI6thsWb_66wmZtJhKNazE4Q8,1104
-agentex/lib/types/llm_messages.py,sha256=fpGMuDbWQ-81isIIMYJCFwLCdtVaXwUb7McxamEwVOM,10704
+agentex/lib/types/llm_messages.py,sha256=fDX1QeqMpESjOXMSL1pDIM4GmxYou6XbGL41EkmugZM,10736
 agentex/lib/types/tracing.py,sha256=6B8tCODAd4EAUoZXXyLYZ7KT6EHcdT4IEbo--aVZWlk,803
 agentex/lib/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/utils/completions.py,sha256=TS0eYK_3yuF3fv4DiOIKLGVRsp0-AfN4RenCPSa_Wsk,4678
@@ -362,8 +363,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.9.0.dist-info/METADATA,sha256=fyyMd_afMH7xhLdmQ7mLlYCvFU6SiqmMrIn4dVFe9eg,15553
-agentex_sdk-0.9.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-agentex_sdk-0.9.0.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
-agentex_sdk-0.9.0.dist-info/licenses/LICENSE,sha256=Eejl902yry8m7Bl6gRYRUObLifIrC61CmV369C6-dqQ,11337
-agentex_sdk-0.9.0.dist-info/RECORD,,
+agentex_sdk-0.9.2.dist-info/METADATA,sha256=LXmG1t5m2W0izEKj3NLWJ_mJweirnxwpO3CKTPUhPWI,15553
+agentex_sdk-0.9.2.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+agentex_sdk-0.9.2.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
+agentex_sdk-0.9.2.dist-info/licenses/LICENSE,sha256=Eejl902yry8m7Bl6gRYRUObLifIrC61CmV369C6-dqQ,11337
+agentex_sdk-0.9.2.dist-info/RECORD,,