Skip to content

Commit 3f41518

Browse files
committed
Add import export memories
1 parent 78ca116 commit 3f41518

File tree

2 files changed

+24
-1
lines changed

2 files changed

+24
-1
lines changed

agixtsdk/__init__.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -853,3 +853,26 @@ def get_agent_memories(
853853
return response.json()["memories"]
854854
except Exception as e:
855855
return self.handle_error(e)
856+
857+
def export_agent_memories(self, agent_name: str) -> List[Dict[str, Any]]:
858+
try:
859+
response = requests.get(
860+
headers=self.headers,
861+
url=f"{self.base_uri}/api/agent/{agent_name}/memory/export",
862+
)
863+
return response.json()["memories"]
864+
except Exception as e:
865+
return self.handle_error(e)
866+
867+
def import_agent_memories(
868+
self, agent_name: str, memories: List[Dict[str, Any]]
869+
) -> str:
870+
try:
871+
response = requests.post(
872+
headers=self.headers,
873+
url=f"{self.base_uri}/api/agent/{agent_name}/memory/import",
874+
json={"memories": memories},
875+
)
876+
return response.json()["message"]
877+
except Exception as e:
878+
return self.handle_error(e)

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
setup(
1010
name="agixtsdk",
11-
version="0.0.28",
11+
version="0.0.29",
1212
description="The AGiXT SDK for Python.",
1313
long_description=long_description,
1414
long_description_content_type="text/markdown",

0 commit comments

Comments
 (0)