fix: Remove the hard-coded database URI in the test helper to use an env var (#865)

This commit is contained in:
RT
2024-01-21 01:56:46 -05:00
committed by GitHub
parent 45edc624d8
commit 1f7f5384fb

View File

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