gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #25 from topoteretes/add_async_elements

Browse source 
Added async elements
This commit is contained in:
Vasilije 2023-10-12 17:14:08 +02:00 • committed by GitHub
commit e08712a343
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
README.md
View file

@ -1,5 +1,4 @@
# PromethAI-Memory # PromethAI-Memory
Memory management and testing for the AI Applications and RAGs 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" Examples of metadata structure and test set are in the folder "example_data"

7
level_3/scripts/create_database.py
View file

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