aiqtoolkit-mem0ai 1.2.0a20250707__py3-none-any.whl → 1.2.0rc1__py3-none-any.whl

aiq/plugins/mem0ai/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 Mem0MemoryClientConfig(MemoryBaseConfig, name="mem0_memory"):
+class Mem0MemoryClientConfig(MemoryBaseConfig, RetryMixin, name="mem0_memory"):
     host: str | None = None
     organization: str | None = None
     project: str | None = None
@@ -46,4 +48,10 @@ async def mem0_memory_client(config: Mem0MemoryClientConfig, builder: Builder):
 
     memory_editor = Mem0Editor(mem0_client=mem0_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_mem0ai-1.2.0a20250707.dist-info → aiqtoolkit_mem0ai-1.2.0rc1.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: aiqtoolkit-mem0ai
-Version: 1.2.0a20250707
+Version: 1.2.0rc1
 Summary: Subpackage for Mem0 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: mem0ai~=0.1.30
 
 <!--

aiqtoolkit_mem0ai-1.2.0rc1.dist-info/RECORD

@@ -0,0 +1,10 @@
+aiq/meta/pypi.md,sha256=RdZ_PgLpmvTtrYbcnpA1LEvDofF8PFRoxVqjUKr-M5Y,1088
+aiq/plugins/mem0ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+aiq/plugins/mem0ai/mem0_editor.py,sha256=IGyUf_ACFK4VYPeLuHAK6-Q3t7Rr9UZaJB9Y4cggxow,3655
+aiq/plugins/mem0ai/memory.py,sha256=vRf0rVZtUVeuZuN3qBtXALueSoso23ElD3J_yCtHnC8,2291
+aiq/plugins/mem0ai/register.py,sha256=_ffKNKnMfkB2HzX4Nk_9EW0pwebg3GuzAE-iB-CoC3E,839
+aiqtoolkit_mem0ai-1.2.0rc1.dist-info/METADATA,sha256=G1o3hbWCl2DI36BpLQIhfGtIE5h_YE5tkDKJo0dfP0s,1416
+aiqtoolkit_mem0ai-1.2.0rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiqtoolkit_mem0ai-1.2.0rc1.dist-info/entry_points.txt,sha256=jnLDGbW6Xdx5jucbwBqkrDOTlAewgXZGk6cxqitGjvE,58
+aiqtoolkit_mem0ai-1.2.0rc1.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
+aiqtoolkit_mem0ai-1.2.0rc1.dist-info/RECORD,,

aiqtoolkit_mem0ai-1.2.0a20250707.dist-info/RECORD

@@ -1,10 +0,0 @@
-aiq/meta/pypi.md,sha256=RdZ_PgLpmvTtrYbcnpA1LEvDofF8PFRoxVqjUKr-M5Y,1088
-aiq/plugins/mem0ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aiq/plugins/mem0ai/mem0_editor.py,sha256=IGyUf_ACFK4VYPeLuHAK6-Q3t7Rr9UZaJB9Y4cggxow,3655
-aiq/plugins/mem0ai/memory.py,sha256=zNAV-Ui-JyPc4Ij-Ldsx6ptxSRV6WIVOGhdJoYJ8-Yg,1821
-aiq/plugins/mem0ai/register.py,sha256=_ffKNKnMfkB2HzX4Nk_9EW0pwebg3GuzAE-iB-CoC3E,839
-aiqtoolkit_mem0ai-1.2.0a20250707.dist-info/METADATA,sha256=oj2QpxA8t6-ouxbMswcNHiCslmued6PF7WzYKotHYRQ,1434
-aiqtoolkit_mem0ai-1.2.0a20250707.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aiqtoolkit_mem0ai-1.2.0a20250707.dist-info/entry_points.txt,sha256=jnLDGbW6Xdx5jucbwBqkrDOTlAewgXZGk6cxqitGjvE,58
-aiqtoolkit_mem0ai-1.2.0a20250707.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
-aiqtoolkit_mem0ai-1.2.0a20250707.dist-info/RECORD,,