aiqtoolkit-zep-cloud 1.2.0a20250707__py3-none-any.whl → 1.2.0rc1__py3-none-any.whl

aiq/plugins/zep_cloud/memory.py

@@ -16,9 +16,11 @@
 from aiq.builder.builder import Builder
 from aiq.cli.register_workflow import register_memory
 from aiq.data_models.memory import MemoryBaseConfig
+from aiq.data_models.retry_mixin import RetryMixin
+from aiq.utils.exception_handlers.automatic_retries import patch_with_retry
 
 
-class ZepMemoryClientConfig(MemoryBaseConfig, name="zep_memory"):
+class ZepMemoryClientConfig(MemoryBaseConfig, RetryMixin, name="zep_memory"):
     base_url: str | None = None
     timeout: float | None = None
     follow_redirects: bool | None = None
@@ -43,4 +45,10 @@ async def zep_memory_client(config: ZepMemoryClientConfig, builder: Builder):
                           follow_redirects=config.follow_redirects)
     memory_editor = ZepEditor(zep_client)
 
+    if isinstance(config, RetryMixin):
+        memory_editor = patch_with_retry(memory_editor,
+                                         retries=config.num_retries,
+                                         retry_codes=config.retry_on_status_codes,
+                                         retry_on_messages=config.retry_on_errors)
+
     yield memory_editor

{aiqtoolkit_zep_cloud-1.2.0a20250707.dist-info → aiqtoolkit_zep_cloud-1.2.0rc1.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: aiqtoolkit-zep-cloud
-Version: 1.2.0a20250707
+Version: 1.2.0rc1
 Summary: Subpackage for Zep memory integration in AIQtoolkit
 Keywords: ai,agents,memory
 Classifier: Programming Language :: Python
 Requires-Python: <3.13,>=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: aiqtoolkit==v1.2.0a20250707
+Requires-Dist: aiqtoolkit~=1.2
 Requires-Dist: zep-cloud~=2.2.0
 
 <!--

aiqtoolkit_zep_cloud-1.2.0rc1.dist-info/RECORD

@@ -0,0 +1,10 @@
+aiq/meta/pypi.md,sha256=myK30Tyjwzm0wp4Clp-bkXffhqWIU-hh28urPrudb2A,1087
+aiq/plugins/zep_cloud/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+aiq/plugins/zep_cloud/memory.py,sha256=QCRFPIp09yamIfDIU0X18an20AaEFNLQ2At6Wrd3XQE,2198
+aiq/plugins/zep_cloud/register.py,sha256=_ffKNKnMfkB2HzX4Nk_9EW0pwebg3GuzAE-iB-CoC3E,839
+aiq/plugins/zep_cloud/zep_editor.py,sha256=n8CzRcN6FupQ4gJT6hB7o6-cGnAL3OP3mI2IiUfdvOA,3778
+aiqtoolkit_zep_cloud-1.2.0rc1.dist-info/METADATA,sha256=RGiUIY3vno_sbJscVihAqXUBwDHO_IkvwdvLmS9qV9A,1419
+aiqtoolkit_zep_cloud-1.2.0rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiqtoolkit_zep_cloud-1.2.0rc1.dist-info/entry_points.txt,sha256=4YncaCW2G71ZTzahBnUuvTEhIbz9adjbPXC2waBqZ0Q,64
+aiqtoolkit_zep_cloud-1.2.0rc1.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
+aiqtoolkit_zep_cloud-1.2.0rc1.dist-info/RECORD,,

aiqtoolkit_zep_cloud-1.2.0a20250707.dist-info/RECORD

@@ -1,10 +0,0 @@
-aiq/meta/pypi.md,sha256=myK30Tyjwzm0wp4Clp-bkXffhqWIU-hh28urPrudb2A,1087
-aiq/plugins/zep_cloud/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aiq/plugins/zep_cloud/memory.py,sha256=nCdjuqc88lbHgOzryX03lXsZvT26CLtIhaWjPyzYYBo,1728
-aiq/plugins/zep_cloud/register.py,sha256=_ffKNKnMfkB2HzX4Nk_9EW0pwebg3GuzAE-iB-CoC3E,839
-aiq/plugins/zep_cloud/zep_editor.py,sha256=n8CzRcN6FupQ4gJT6hB7o6-cGnAL3OP3mI2IiUfdvOA,3778
-aiqtoolkit_zep_cloud-1.2.0a20250707.dist-info/METADATA,sha256=-uDSVO8K3pbiGiiUFEVG8b4aE0lx9NTjvp4HXsKwM5s,1437
-aiqtoolkit_zep_cloud-1.2.0a20250707.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aiqtoolkit_zep_cloud-1.2.0a20250707.dist-info/entry_points.txt,sha256=4YncaCW2G71ZTzahBnUuvTEhIbz9adjbPXC2waBqZ0Q,64
-aiqtoolkit_zep_cloud-1.2.0a20250707.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
-aiqtoolkit_zep_cloud-1.2.0a20250707.dist-info/RECORD,,