diff --git a/cognitive_architecture/config.py b/cognitive_architecture/config.py index d5cc58184..a113fd5ad 100644 --- a/cognitive_architecture/config.py +++ b/cognitive_architecture/config.py @@ -64,20 +64,18 @@ class Config: graph_database_password: str = os.getenv("GRAPH_DB_PW") weaviate_url: str = os.getenv("WEAVIATE_URL") 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 ( os.getenv("ENV") == "prod" or os.getenv("ENV") == "dev" or os.getenv("AWS_ENV") == "dev" or os.getenv("AWS_ENV") == "prd" ): - postgres_host: str = os.getenv("POSTGRES_PROD_HOST") - elif os.getenv("ENV") == "docker": - postgres_host: str = os.getenv("POSTGRES_HOST_DOCKER") - elif os.getenv("ENV") == "local": - postgres_host: str = os.getenv("POSTGRES_HOST_LOCAL") + db_host: str = os.getenv("POSTGRES_PROD_HOST") + db_user: str = os.getenv("POSTGRES_USER") + db_password: str = os.getenv("POSTGRES_PASSWORD") + db_name: str = os.getenv("POSTGRES_DB") + # Client ID anon_clientid: Optional[str] = field(default_factory=lambda: uuid.uuid4().hex)