Spaces:
Paused
Paused
Hemang Thakur
commited on
Commit
·
a6a6a62
1
Parent(s):
a0fd75c
changed working directory
Browse files
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
|