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

Fixes to database manager

Browse source
This commit is contained in:
Vasilije 2024-02-20 10:41:50 +01:00
parent a904fa9d29
commit 86555460d2
1 changed files with 6 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
cognitive_architecture/config.py
View file

@ -64,20 +64,18 @@ class Config:
graph_database_password: str = os.getenv("GRAPH_DB_PW") graph_database_password: str = os.getenv("GRAPH_DB_PW")
weaviate_url: str = os.getenv("WEAVIATE_URL") weaviate_url: str = os.getenv("WEAVIATE_URL")
weaviate_api_key: str = os.getenv("WEAVIATE_API_KEY") weaviate_api_key: str = os.getenv("WEAVIATE_API_KEY")
postgres_user: str = os.getenv("POSTGRES_USER")
postgres_password: str = os.getenv("POSTGRES_PASSWORD")
postgres_db: str = os.getenv("POSTGRES_DB")
if ( if (
os.getenv("ENV") == "prod" os.getenv("ENV") == "prod"
or os.getenv("ENV") == "dev" or os.getenv("ENV") == "dev"
or os.getenv("AWS_ENV") == "dev" or os.getenv("AWS_ENV") == "dev"
or os.getenv("AWS_ENV") == "prd" or os.getenv("AWS_ENV") == "prd"
): ):
postgres_host: str = os.getenv("POSTGRES_PROD_HOST") db_host: str = os.getenv("POSTGRES_PROD_HOST")
elif os.getenv("ENV") == "docker": db_user: str = os.getenv("POSTGRES_USER")
postgres_host: str = os.getenv("POSTGRES_HOST_DOCKER") db_password: str = os.getenv("POSTGRES_PASSWORD")
elif os.getenv("ENV") == "local": db_name: str = os.getenv("POSTGRES_DB")
postgres_host: str = os.getenv("POSTGRES_HOST_LOCAL")
# Client ID # Client ID
anon_clientid: Optional[str] = field(default_factory=lambda: uuid.uuid4().hex) anon_clientid: Optional[str] = field(default_factory=lambda: uuid.uuid4().hex)