Merge pull request #25 from topoteretes/add_async_elements

Added async elements

README.md

@@ -1,5 +1,4 @@
 # PromethAI-Memory
-
 Memory management and testing for the AI Applications and RAGs
 
 
@@ -148,4 +147,3 @@ After that, you can run the RAG test manager.
 ```
 
 Examples of metadata structure and test set are in the folder "example_data"
-

level_3/scripts/create_database.py

@@ -47,19 +47,20 @@ def create_database(username, password, host, db_name):
     engine.dispose()
 
 
-def create_tables():
+def create_tables(engine):
     Base.metadata.create_all(bind=engine)
 
-
 if __name__ == "__main__":
     username = os.getenv('POSTGRES_USER')
     password = os.getenv('POSTGRES_PASSWORD')
     database_name = os.getenv('POSTGRES_DB')
     host = os.getenv('POSTGRES_HOST')
 
+    engine = create_admin_engine(username, password, host, database_name)
+
     if not database_exists(username, password, host, database_name):
         print(f"Database {database_name} does not exist. Creating...")
         create_database(username, password, host, database_name)
         print(f"Database {database_name} created successfully.")
 
-    create_tables()
+    create_tables(engine)