fixed env variables

Author: Arsalanjdev

.github/workflows/ci.yml

@@ -10,6 +10,10 @@ jobs:
   build:
     runs-on: ubuntu-latest
 
+    env:
+      HF_API_KEY: ${{ secrets.HF_API_KEY }}
+      DB_URL: ${{ secrets.DB_URL }}
+
     steps:
       - uses: actions/checkout@v3
       - name: Set Up Python
@@ -21,7 +25,6 @@ jobs:
         run: |
           pip install poetry
           poetry install
-
       - name: Run tests
         run: |
           poetry run pytest

src/rowgen/main.py

@@ -4,10 +4,8 @@
 from huggingface_hub import InferenceClient
 import sqlalchemy
 from dotenv import load_dotenv
-from utils import ENV_PATH
+from utils import API_KEY
 
-load_dotenv(dotenv_path=ENV_PATH)
-API_KEY = os.getenv("HF_API_KEY")
 print()
 
 

src/rowgen/utils.py

@@ -1,5 +1,8 @@
 from pathlib import Path
 
+from dotenv import load_dotenv
+import os
+
 
 def find_project_root(start_path: Path = None, marker_files=None) -> Path:
     """
@@ -33,13 +36,20 @@ def find_project_root(start_path: Path = None, marker_files=None) -> Path:
         current = current.parent
 
 
-def get_env_file_path() -> Path:
+def get_api_key() -> str:
     """
-    Finds and returns the path of .env file.
-    :return: Path object.
+    Returns the API key from environment, loading .env if running locally.
     """
-    root = find_project_root()
-    return f"{root}/.env"
+    # Load .env only if not already set (e.g., local dev)
+    if "HF_API_KEY" not in os.environ:
+        env_path = find_project_root() / ".env"
+        if env_path.exists():
+            load_dotenv(dotenv_path=env_path)
+
+    api_key = os.getenv("HF_API_KEY")
+    if not api_key:
+        raise RuntimeError("HF_API_KEY is not set in the environment.")
+    return api_key
 
 
-ENV_PATH = get_env_file_path()
+API_KEY = get_api_key()

tests/test_ai_api.py

@@ -4,11 +4,9 @@
 from dotenv import load_dotenv
 from huggingface_hub.errors import BadRequestError
 
-from rowgen.utils import ENV_PATH
+from rowgen.utils import API_KEY
 from huggingface_hub import InferenceClient, HfApi
 
-load_dotenv(dotenv_path=ENV_PATH)
-API_KEY = os.getenv("HF_API_KEY")
 
 DEEP_SEEK_MODEL = "deepseek-ai/DeepSeek-V3"