Spaces:
Sleeping
Sleeping
Update llm_providers.py
Browse files- 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('
|
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('
|
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('
|
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",
|