jesusvilela commited on
Commit
2b1cebd
·
verified ·
1 Parent(s): 98cbcc7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,7 +40,7 @@ except ImportError: WHISPER_AVAILABLE = False; print("WARNING: OpenAI Whisper no
40
 
41
  # --- google-genai SDK (Unified SDK) ---
42
  from google import genai as google_genai_sdk # Alias for clarity
43
- from genai.types import HarmCategory, HarmBlockThreshold # ***** CORRECTED IMPORT *****
44
  # For FileState enum later
45
  from google.ai import generativelanguage as glm
46
  # --- End google-genai SDK ---
@@ -409,7 +409,7 @@ def initialize_agent_and_tools(force_reinit=False):
409
  model=GEMINI_MODEL_NAME,
410
  google_api_key=GOOGLE_API_KEY,
411
  temperature=0.0,
412
- safety_settings=llm_safety_settings_corrected_final, # USE THE DICTIONARY WITH INT VALUES FOR BOTH
413
  timeout=120,
414
  convert_system_message_to_human=True
415
  )
 
40
 
41
  # --- google-genai SDK (Unified SDK) ---
42
  from google import genai as google_genai_sdk # Alias for clarity
43
+ from google.genai.types import HarmCategory, HarmBlockThreshold # ***** CORRECTED IMPORT *****
44
  # For FileState enum later
45
  from google.ai import generativelanguage as glm
46
  # --- End google-genai SDK ---
 
409
  model=GEMINI_MODEL_NAME,
410
  google_api_key=GOOGLE_API_KEY,
411
  temperature=0.0,
412
+ #safety_settings=llm_safety_settings_corrected_final, # USE THE DICTIONARY WITH INT VALUES FOR BOTH
413
  timeout=120,
414
  convert_system_message_to_human=True
415
  )