dolphinium
commited on
Commit
Β·
7cbc718
1
Parent(s):
bba916d
fix gemini model
Browse files- 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-
|
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.")
|