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

Update llm_providers.py

Browse files
Files changed (1) hide show
  1. 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'] = ChatGemini(api_key=google_key)
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'):