load config files from .env file if it exists

climatepolicyradar · Feb 8, 2024 · 2f10bf8 · 2f10bf8
1 parent 91b2e4e
commit 2f10bf8
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 1 deletion.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,6 +18,7 @@ aws-error-utils = "^2.7.0"
 moto = "^4.1.11"
 torch = "2.0.0"
 cpr-data-access = {git = "https://github.com/climatepolicyradar/data-access.git", tag = "0.4.0"}
+python-dotenv = "^1.0.1"
 
 [tool.poetry.dev-dependencies]
 black = "^22.1.0"

diff --git a/src/config.py b/src/config.py
@@ -3,6 +3,9 @@
 import os
 from typing import Set
 import re
+from dotenv import load_dotenv, find_dotenv
+
+load_dotenv(find_dotenv())
 
 SBERT_MODEL: str = os.getenv("SBERT_MODEL", "msmarco-distilbert-dot-v5")
 INDEX_ENCODER_CACHE_FOLDER: str = os.getenv("INDEX_ENCODER_CACHE_FOLDER", "/models")