MemGPT/tests/clear_postgres_db.py

20 lines
359 B
Python

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