4c95b7bde3
Co-authored-by: Starlette <starlette@Starlettes-MacBook-Pro.local> Co-authored-by: cpacker <packercharles@gmail.com>
20 lines
668 B
Python
20 lines
668 B
Python
import os
|
|
from memgpt.migrate import migrate_all_agents, migrate_all_sources
|
|
from memgpt import MemGPT
|
|
from memgpt.config import MemGPTConfig
|
|
from .utils import wipe_config
|
|
|
|
|
|
def test_migrate_0211():
|
|
wipe_config()
|
|
|
|
data_dir = "tests/data/memgpt-0.2.11"
|
|
# os.environ["MEMGPT_CONFIG_PATH"] = os.path.join(data_dir, "config")
|
|
# print(f"MEMGPT_CONFIG_PATH={os.environ['MEMGPT_CONFIG_PATH']}")
|
|
res = migrate_all_agents(data_dir)
|
|
assert res["failed_migrations"] == 0, f"Failed migrations: {res}"
|
|
res = migrate_all_sources(data_dir)
|
|
assert res["failed_migrations"] == 0, f"Failed migrations: {res}"
|
|
|
|
# TODO: assert everything is in the DB
|