sivan22 commited on
Commit
f3a8fa4
·
verified ·
1 Parent(s): fffe65f

Update llm_providers.py

Browse files
Files changed (1) hide show
  1. llm_providers.py +3 -1
llm_providers.py CHANGED
@@ -42,7 +42,9 @@ class LLMProvider:
42
 
43
  # Google Gemini
44
  if google_key := os.getenv('GOOGLE_API_KEY') or self.api_keys.get('Gemimi'):
45
- self.providers['Gemini'] = ChatGoogleGenerativeAI(api_key=google_key)
 
 
46
 
47
  # Anthropicsel
48
  if anthropic_key := os.getenv('ANTHROPIC_API_KEY') or self.api_keys.get('Claude'):
 
42
 
43
  # Google Gemini
44
  if google_key := os.getenv('GOOGLE_API_KEY') or self.api_keys.get('Gemimi'):
45
+ self.providers['Gemini'] = ChatGoogleGenerativeAI(
46
+ model="gemini-2.0-flash-exp",
47
+ api_key=google_key)
48
 
49
  # Anthropicsel
50
  if anthropic_key := os.getenv('ANTHROPIC_API_KEY') or self.api_keys.get('Claude'):