Spaces:
Sleeping
Sleeping
Commit ·
52d42cd
1
Parent(s): 79d5a5c
eski holat
Browse files
app.py
CHANGED
|
@@ -321,9 +321,12 @@ def main_switch_model(model_name):
|
|
| 321 |
|
| 322 |
def main():
|
| 323 |
global query_engine, chunks_df, reranker, vector_index, current_model
|
| 324 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 325 |
log_message("Запуск AIEXP - AI Expert для нормативной документации")
|
| 326 |
-
|
| 327 |
query_engine, chunks_df, reranker, vector_index, chunk_info = initialize_system(
|
| 328 |
repo_id=HF_REPO_ID,
|
| 329 |
hf_token=HF_TOKEN,
|
|
|
|
| 321 |
|
| 322 |
def main():
|
| 323 |
global query_engine, chunks_df, reranker, vector_index, current_model
|
| 324 |
+
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY", "")
|
| 325 |
+
if GOOGLE_API_KEY:
|
| 326 |
+
log_message("Использование Google API для модели генерации текста")
|
| 327 |
+
else:
|
| 328 |
+
log_message("Google API ключ не найден, использование локальной модели")
|
| 329 |
log_message("Запуск AIEXP - AI Expert для нормативной документации")
|
|
|
|
| 330 |
query_engine, chunks_df, reranker, vector_index, chunk_info = initialize_system(
|
| 331 |
repo_id=HF_REPO_ID,
|
| 332 |
hf_token=HF_TOKEN,
|
utils.py
CHANGED
|
@@ -6,6 +6,7 @@ import os
|
|
| 6 |
|
| 7 |
api_key = os.getenv('GOOGLE_API_KEY') # or however you're loading it
|
| 8 |
if not api_key:
|
|
|
|
| 9 |
raise ValueError("GOOGLE_API_KEY not found in environment")
|
| 10 |
|
| 11 |
|
|
|
|
| 6 |
|
| 7 |
api_key = os.getenv('GOOGLE_API_KEY') # or however you're loading it
|
| 8 |
if not api_key:
|
| 9 |
+
log_message("GOOGLE_API_KEY not found in environment")
|
| 10 |
raise ValueError("GOOGLE_API_KEY not found in environment")
|
| 11 |
|
| 12 |
|