aiqtoolkit-zep-cloud 1.2.0a20250706__py3-none-any.whl → 1.2.0a20250730__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of aiqtoolkit-zep-cloud might be problematic. Click here for more details.
- aiq/plugins/zep_cloud/memory.py +9 -1
- {aiqtoolkit_zep_cloud-1.2.0a20250706.dist-info → aiqtoolkit_zep_cloud-1.2.0a20250730.dist-info}/METADATA +2 -2
- {aiqtoolkit_zep_cloud-1.2.0a20250706.dist-info → aiqtoolkit_zep_cloud-1.2.0a20250730.dist-info}/RECORD +6 -6
- {aiqtoolkit_zep_cloud-1.2.0a20250706.dist-info → aiqtoolkit_zep_cloud-1.2.0a20250730.dist-info}/WHEEL +0 -0
- {aiqtoolkit_zep_cloud-1.2.0a20250706.dist-info → aiqtoolkit_zep_cloud-1.2.0a20250730.dist-info}/entry_points.txt +0 -0
- {aiqtoolkit_zep_cloud-1.2.0a20250706.dist-info → aiqtoolkit_zep_cloud-1.2.0a20250730.dist-info}/top_level.txt +0 -0
aiq/plugins/zep_cloud/memory.py
CHANGED
|
@@ -16,9 +16,11 @@
|
|
|
16
16
|
from aiq.builder.builder import Builder
|
|
17
17
|
from aiq.cli.register_workflow import register_memory
|
|
18
18
|
from aiq.data_models.memory import MemoryBaseConfig
|
|
19
|
+
from aiq.data_models.retry_mixin import RetryMixin
|
|
20
|
+
from aiq.utils.exception_handlers.automatic_retries import patch_with_retry
|
|
19
21
|
|
|
20
22
|
|
|
21
|
-
class ZepMemoryClientConfig(MemoryBaseConfig, name="zep_memory"):
|
|
23
|
+
class ZepMemoryClientConfig(MemoryBaseConfig, RetryMixin, name="zep_memory"):
|
|
22
24
|
base_url: str | None = None
|
|
23
25
|
timeout: float | None = None
|
|
24
26
|
follow_redirects: bool | None = None
|
|
@@ -43,4 +45,10 @@ async def zep_memory_client(config: ZepMemoryClientConfig, builder: Builder):
|
|
|
43
45
|
follow_redirects=config.follow_redirects)
|
|
44
46
|
memory_editor = ZepEditor(zep_client)
|
|
45
47
|
|
|
48
|
+
if isinstance(config, RetryMixin):
|
|
49
|
+
memory_editor = patch_with_retry(memory_editor,
|
|
50
|
+
retries=config.num_retries,
|
|
51
|
+
retry_codes=config.retry_on_status_codes,
|
|
52
|
+
retry_on_messages=config.retry_on_errors)
|
|
53
|
+
|
|
46
54
|
yield memory_editor
|
|
@@ -1,12 +1,12 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: aiqtoolkit-zep-cloud
|
|
3
|
-
Version: 1.2.
|
|
3
|
+
Version: 1.2.0a20250730
|
|
4
4
|
Summary: Subpackage for Zep memory integration in AIQtoolkit
|
|
5
5
|
Keywords: ai,agents,memory
|
|
6
6
|
Classifier: Programming Language :: Python
|
|
7
7
|
Requires-Python: <3.13,>=3.11
|
|
8
8
|
Description-Content-Type: text/markdown
|
|
9
|
-
Requires-Dist: aiqtoolkit==v1.2.
|
|
9
|
+
Requires-Dist: aiqtoolkit==v1.2.0a20250730
|
|
10
10
|
Requires-Dist: zep-cloud~=2.2.0
|
|
11
11
|
|
|
12
12
|
<!--
|
|
@@ -1,10 +1,10 @@
|
|
|
1
1
|
aiq/meta/pypi.md,sha256=myK30Tyjwzm0wp4Clp-bkXffhqWIU-hh28urPrudb2A,1087
|
|
2
2
|
aiq/plugins/zep_cloud/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
3
|
-
aiq/plugins/zep_cloud/memory.py,sha256=
|
|
3
|
+
aiq/plugins/zep_cloud/memory.py,sha256=QCRFPIp09yamIfDIU0X18an20AaEFNLQ2At6Wrd3XQE,2198
|
|
4
4
|
aiq/plugins/zep_cloud/register.py,sha256=_ffKNKnMfkB2HzX4Nk_9EW0pwebg3GuzAE-iB-CoC3E,839
|
|
5
5
|
aiq/plugins/zep_cloud/zep_editor.py,sha256=n8CzRcN6FupQ4gJT6hB7o6-cGnAL3OP3mI2IiUfdvOA,3778
|
|
6
|
-
aiqtoolkit_zep_cloud-1.2.
|
|
7
|
-
aiqtoolkit_zep_cloud-1.2.
|
|
8
|
-
aiqtoolkit_zep_cloud-1.2.
|
|
9
|
-
aiqtoolkit_zep_cloud-1.2.
|
|
10
|
-
aiqtoolkit_zep_cloud-1.2.
|
|
6
|
+
aiqtoolkit_zep_cloud-1.2.0a20250730.dist-info/METADATA,sha256=wN3WfkATmXsNzbEDReAwiMOOsNr2pn3itFjqSm9hWWw,1437
|
|
7
|
+
aiqtoolkit_zep_cloud-1.2.0a20250730.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
8
|
+
aiqtoolkit_zep_cloud-1.2.0a20250730.dist-info/entry_points.txt,sha256=4YncaCW2G71ZTzahBnUuvTEhIbz9adjbPXC2waBqZ0Q,64
|
|
9
|
+
aiqtoolkit_zep_cloud-1.2.0a20250730.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
|
|
10
|
+
aiqtoolkit_zep_cloud-1.2.0a20250730.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|