Spaces:
Runtime error
Runtime error
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",
|