refactor: update envs

ainize-team · Dec 20, 2024 · de99e9f · de99e9f
1 parent 1eb4857
commit de99e9f
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/.env.sample b/.env.sample
@@ -0,0 +1,4 @@
+BROKER_URI=amqp://<user_name>:<password>@<rabbitmq_container_name>:5672/<rabbitmq_vhost_name>
+FIREBASE_DATABASE_URL=<firebase_database_url>
+FIREBASE_STORAGE_BUCKET=<firebase_storage_bucket>
+FIREBASE_APP_NAME=super-resolution
diff --git a/src/configs/config.py b/src/configs/config.py
@@ -20,8 +20,8 @@ class ModelSettings(BaseSettings):
 class FirebaseSettings(BaseSettings):
     firebase_app_name: str = "super-resolution"
     cred_path: str = "/app/key/serviceAccountKey.json"
-    database_url: str
-    storage_bucket: str
+    firebase_database_url: str
+    firebase_storage_bucket: str
 
 
 celery_worker_settings = CeleryWorkerSettings()

diff --git a/src/worker.py b/src/worker.py
@@ -16,7 +16,7 @@
 firebase_admin.initialize_app(
     cred,
     {
-        "databaseURL": firebase_settings.database_url,
-        "storageBucket": firebase_settings.storage_bucket,
+        "databaseURL": firebase_settings.firebase_database_url,
+        "storageBucket": firebase_settings.firebase_storage_bucket,
     },
 )