4
4
from langchain_anthropic import ChatAnthropic
5
5
from langchain_aws import ChatBedrock
6
6
from langchain_community .chat_models import FakeListChatModel
7
- from langchain_google_genai import ChatGoogleGenerativeAI
7
+
8
+ # from langchain_google_genai import ChatGoogleGenerativeAI
8
9
from langchain_groq import ChatGroq
9
10
from langchain_openai import ChatOpenAI
10
11
32
33
FakeModelName .FAKE : "fake" ,
33
34
}
34
35
35
- ModelT : TypeAlias = ChatOpenAI | ChatAnthropic | ChatGoogleGenerativeAI | ChatGroq | ChatBedrock
36
+ ModelT : TypeAlias = ChatOpenAI | ChatAnthropic | ChatGroq | ChatBedrock # | ChatGoogleGenerativeAI
36
37
37
38
38
39
@cache
@@ -47,8 +48,8 @@ def get_model(model_name: AllModelEnum, /) -> ModelT:
47
48
return ChatOpenAI (model = api_model_name , temperature = 0.5 , streaming = True )
48
49
if model_name in AnthropicModelName :
49
50
return ChatAnthropic (model = api_model_name , temperature = 0.5 , streaming = True )
50
- if model_name in GoogleModelName :
51
- return ChatGoogleGenerativeAI (model = api_model_name , temperature = 0.5 , streaming = True )
51
+ # if model_name in GoogleModelName:
52
+ # return ChatGoogleGenerativeAI(model=api_model_name, temperature=0.5, streaming=True)
52
53
if model_name in GroqModelName :
53
54
if model_name == GroqModelName .LLAMA_GUARD_3_8B :
54
55
return ChatGroq (model = api_model_name , temperature = 0.0 )
0 commit comments