MemGPT/tests/clear_postgres_db.py
Sarah Wooders 85faf5f474
chore: migrate package name to letta (#1775)
Co-authored-by: Charles Packer <packercharles@gmail.com>
Co-authored-by: Shubham Naik <shubham.naik10@gmail.com>
Co-authored-by: Shubham Naik <shub@memgpt.ai>
2024-09-23 09:15:18 -07:00

20 lines
348 B
Python

import os
from sqlalchemy import MetaData, create_engine
def main():
uri = os.environ.get(
"MEMGPT_PGURI",
"postgresql+pg8000://letta:letta@localhost:8888/letta",
)
engine = create_engine(uri)
meta = MetaData()
meta.reflect(bind=engine)
meta.drop_all(bind=engine)
if __name__ == "__main__":
main()