From 1c93697ed2ebe3d65011866bbbf352c5ce0a9caa Mon Sep 17 00:00:00 2001
From: Sarah Wooders <sarahwooders@gmail.com>
Date: Fri, 19 Jan 2024 15:31:40 -0800
Subject: [PATCH] feat: Helper script for clearing out test database for
 postgres (#863)

---
 tests/clear_postgres_db.py | 8 ++++++++
 1 file changed, 8 insertions(+)
 create mode 100644 tests/clear_postgres_db.py

diff --git a/tests/clear_postgres_db.py b/tests/clear_postgres_db.py
new file mode 100644
index 00000000..60618363
--- /dev/null
+++ b/tests/clear_postgres_db.py
@@ -0,0 +1,8 @@
+from sqlalchemy import create_engine, MetaData
+
+engine = create_engine("postgresql+pg8000://memgpt:memgpt@localhost:8888/memgpt")
+
+meta = MetaData()
+meta.reflect(bind=engine)
+
+meta.drop_all(bind=engine)