Spaces:
Running
Running
Update llm_providers.py
Browse files- llm_providers.py +2 -1
llm_providers.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
from langchain_anthropic import ChatAnthropic
|
2 |
from langchain_openai import ChatOpenAI
|
3 |
from langchain_ollama import ChatOllama
|
|
|
4 |
from langchain_core.language_models.base import BaseLanguageModel
|
5 |
from langchain_core.messages import BaseMessage, HumanMessage, AIMessage
|
6 |
from typing import Optional, Dict, List, Any, Sequence
|
@@ -41,7 +42,7 @@ class LLMProvider:
|
|
41 |
|
42 |
# Google Gemini
|
43 |
if google_key := os.getenv('GOOGLE_API_KEY') or self.api_keys.get('Gemimi'):
|
44 |
-
self.providers['Gemini'] =
|
45 |
|
46 |
# Anthropicsel
|
47 |
if anthropic_key := os.getenv('ANTHROPIC_API_KEY') or self.api_keys.get('Claude'):
|
|
|
1 |
from langchain_anthropic import ChatAnthropic
|
2 |
from langchain_openai import ChatOpenAI
|
3 |
from langchain_ollama import ChatOllama
|
4 |
+
from langchain_google_genai import ChatGoogleGenerativeAI
|
5 |
from langchain_core.language_models.base import BaseLanguageModel
|
6 |
from langchain_core.messages import BaseMessage, HumanMessage, AIMessage
|
7 |
from typing import Optional, Dict, List, Any, Sequence
|
|
|
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'):
|