Skip to content

Commit 283551c

Browse files
Fix some variable loading issues
1 parent 2fe0cd6 commit 283551c

File tree

5 files changed

+7
-4
lines changed

5 files changed

+7
-4
lines changed

extract_ccdrs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@
4949
cleanup_local_files,
5050
)
5151

52-
load_dotenv(override=True)
52+
load_dotenv(os.getenv("ENVIRONMENT", ".env"))
5353

5454

5555
def identify_new_publications(

load/db.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,11 @@
22
from sqlmodel import create_engine
33
import requests
44
import difflib
5+
from dotenv import load_dotenv
56
from pathlib import Path
67

8+
load_dotenv(os.getenv("ENVIRONMENT", ".env"))
9+
710

811
# Database connection setup
912
def get_database_url():

load/schema.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from sqlalchemy.orm import Mapped
1010

1111
# Load environment variables
12-
load_dotenv(os.getenv("ENVIRONMENT", ".env"), override=True)
12+
load_dotenv(os.getenv("ENVIRONMENT", ".env"))
1313

1414

1515
# Enums for document and node types

load/upload_pdfs_to_aws_s3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
from load.schema import Document
3434

3535
# Load environment variables
36-
load_dotenv()
36+
load_dotenv(os.getenv("ENVIRONMENT", ".env"))
3737

3838
# Set up logging
3939
logging.basicConfig(

load/upload_pdfs_to_openai.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
logger = logging.getLogger(__name__)
2525

2626
# Load environment variables from .env file
27-
load_dotenv(override=True)
27+
load_dotenv(os.getenv("ENVIRONMENT", ".env"))
2828

2929

3030
# Helper function to get or create a vector store

0 commit comments

Comments
 (0)