sivan22 commited on
Commit
0a2cb3b
·
verified ·
1 Parent(s): 196447c

Update llm_providers.py

Browse files
Files changed (1) hide show
  1. llm_providers.py +3 -3
llm_providers.py CHANGED
@@ -40,18 +40,18 @@ class LLMProvider:
40
  def _setup_providers(self):
41
 
42
  # Google Gemini
43
- if google_key := os.getenv('GOOGLE_API_KEY') or self.api_keys.get('google'):
44
  self.providers['Gemini'] = ChatGemini(api_key=google_key)
45
 
46
  # Anthropicsel
47
- if anthropic_key := os.getenv('ANTHROPIC_API_KEY') or self.api_keys.get('anthropic'):
48
  self.providers['Claude'] = ChatAnthropic(
49
  api_key=anthropic_key,
50
  model_name="claude-3-5-sonnet-20241022",
51
  )
52
 
53
  # OpenAI
54
- if openai_key := os.getenv('OPENAI_API_KEY') or self.api_keys.get('openai'):
55
  self.providers['ChatGPT'] = ChatOpenAI(
56
  api_key=openai_key,
57
  model_name="gpt-4o-2024-11-20",
 
40
  def _setup_providers(self):
41
 
42
  # Google Gemini
43
+ if google_key := os.getenv('GOOGLE_API_KEY') or self.api_keys.get('Gemimi'):
44
  self.providers['Gemini'] = ChatGemini(api_key=google_key)
45
 
46
  # Anthropicsel
47
+ if anthropic_key := os.getenv('ANTHROPIC_API_KEY') or self.api_keys.get('Claude'):
48
  self.providers['Claude'] = ChatAnthropic(
49
  api_key=anthropic_key,
50
  model_name="claude-3-5-sonnet-20241022",
51
  )
52
 
53
  # OpenAI
54
+ if openai_key := os.getenv('OPENAI_API_KEY') or self.api_keys.get('ChatGPT'):
55
  self.providers['ChatGPT'] = ChatOpenAI(
56
  api_key=openai_key,
57
  model_name="gpt-4o-2024-11-20",