Hemang Thakur commited on
Commit
a6a6a62
·
1 Parent(s): a0fd75c

changed working directory

Browse files
Files changed (1) hide show
  1. main.py +3 -1
main.py CHANGED
@@ -17,6 +17,8 @@ from google.api_core.exceptions import ResourceExhausted
17
  logger = logging.getLogger()
18
  logger.setLevel(logging.INFO)
19
 
 
 
20
  CONTEXT_LENGTH = 128000
21
  BUFFER = 10000
22
  MAX_TOKENS_ALLOWED = CONTEXT_LENGTH - BUFFER
@@ -35,7 +37,7 @@ def format_error_sse(event_type: str, data: str) -> str:
35
 
36
  # Initialize the components
37
  def initialize_components():
38
- load_dotenv(override=True)
39
 
40
  from src.search.search_engine import SearchEngine
41
  from src.query_processing.query_processor import QueryProcessor
 
17
  logger = logging.getLogger()
18
  logger.setLevel(logging.INFO)
19
 
20
+ ENV_FILE_PATH = os.getenv("WRITABLE_DIR", "/tmp") + "/.env"
21
+
22
  CONTEXT_LENGTH = 128000
23
  BUFFER = 10000
24
  MAX_TOKENS_ALLOWED = CONTEXT_LENGTH - BUFFER
 
37
 
38
  # Initialize the components
39
  def initialize_components():
40
+ load_dotenv(ENV_FILE_PATH, override=True)
41
 
42
  from src.search.search_engine import SearchEngine
43
  from src.query_processing.query_processor import QueryProcessor