Skip to content

Commit 3b3f98c

Browse files
committed
(test) remove Google Generative AI install for 3.13 support
1 parent ce3a7ec commit 3b3f98c

File tree

3 files changed

+6
-223
lines changed

3 files changed

+6
-223
lines changed

pyproject.toml

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ dependencies = [
2323
"langchain-community ~=0.3.7",
2424
"langchain-openai ~=0.2.9",
2525
"langchain-anthropic ~= 0.3.0",
26-
"langchain-google-genai ~=2.0.5",
26+
# "langchain-google-genai ~=2.0.5", # blocking 3.13 upgrade, see https://github.com/google-gemini/generative-ai-python/issues/518
2727
"langchain-groq ~=0.2.1",
2828
"langchain-aws ~=0.2.7",
2929
"langgraph ~=0.2.53",
@@ -40,7 +40,6 @@ dependencies = [
4040

4141
# package requirements for python 3.13 support
4242
"grpcio >=1.68.0",
43-
"protobuf >=5.28.3",
4443
"pyarrow >=18.1.0",
4544
"tiktoken >=0.8.0",
4645
]

src/core/llm.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,8 @@
44
from langchain_anthropic import ChatAnthropic
55
from langchain_aws import ChatBedrock
66
from langchain_community.chat_models import FakeListChatModel
7-
from langchain_google_genai import ChatGoogleGenerativeAI
7+
8+
# from langchain_google_genai import ChatGoogleGenerativeAI
89
from langchain_groq import ChatGroq
910
from langchain_openai import ChatOpenAI
1011

@@ -32,7 +33,7 @@
3233
FakeModelName.FAKE: "fake",
3334
}
3435

35-
ModelT: TypeAlias = ChatOpenAI | ChatAnthropic | ChatGoogleGenerativeAI | ChatGroq | ChatBedrock
36+
ModelT: TypeAlias = ChatOpenAI | ChatAnthropic | ChatGroq | ChatBedrock # | ChatGoogleGenerativeAI
3637

3738

3839
@cache
@@ -47,8 +48,8 @@ def get_model(model_name: AllModelEnum, /) -> ModelT:
4748
return ChatOpenAI(model=api_model_name, temperature=0.5, streaming=True)
4849
if model_name in AnthropicModelName:
4950
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)
5253
if model_name in GroqModelName:
5354
if model_name == GroqModelName.LLAMA_GUARD_3_8B:
5455
return ChatGroq(model=api_model_name, temperature=0.0)

uv.lock

Lines changed: 0 additions & 217 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)