dolphinium commited on
Commit
7cbc718
Β·
1 Parent(s): bba916d

fix gemini model

Browse files
Files changed (1) hide show
  1. connections.py +1 -1
connections.py CHANGED
@@ -46,7 +46,7 @@ def initialize_connections():
46
  raise ValueError("GEMINI_API_KEY is missing.")
47
 
48
  genai.configure(api_key=config.GEMINI_API_KEY)
49
- llm_model = genai.GenerativeModel('gemini-1.5-flash', generation_config=genai.types.GenerationConfig(temperature=0))
50
  print(f"βœ… LLM Model '{llm_model.model_name}' initialized.")
51
 
52
  print("βœ… System Initialized Successfully.")
 
46
  raise ValueError("GEMINI_API_KEY is missing.")
47
 
48
  genai.configure(api_key=config.GEMINI_API_KEY)
49
+ llm_model = genai.GenerativeModel('gemini-2.5-flash', generation_config=genai.types.GenerationConfig(temperature=0))
50
  print(f"βœ… LLM Model '{llm_model.model_name}' initialized.")
51
 
52
  print("βœ… System Initialized Successfully.")