MrSimple07 commited on
Commit
52d42cd
·
1 Parent(s): 79d5a5c

eski holat

Browse files
Files changed (2) hide show
  1. app.py +5 -2
  2. utils.py +1 -0
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