Skip to content

feat(langchain): Add support to google_genai provider in init_embeddings #34388

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Christophe-Lamarche-Cossette wants to merge 10 commits into
base: master
Choose a base branch
from
+39 −24
Open

feat(langchain): Add support to google_genai provider in init_embeddings #34388

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
d918e1d
Add Google GenAI as provider for embeddings factory
Christophe-Lamarche-Cossette Dec 16, 2025
166d854
Test: Add/Refactor unit test for embedding/base.py.
Christophe-Lamarche-Cossette Dec 16, 2025
7baf378
Fix: Whitespace
Christophe-Lamarche-Cossette Dec 16, 2025
a81c647
Fix: missing type annotation
Christophe-Lamarche-Cossette Dec 16, 2025
ff65b4d
Fix: Wrong order of expected output
Christophe-Lamarche-Cossette Dec 16, 2025
0113606
Fix: Import order
Christophe-Lamarche-Cossette Dec 16, 2025
171dbd6
Merge branch 'master' into cossette-christophe-lamarche/embeddings-fa…
Christophe-Lamarche-Cossette Dec 16, 2025
aee630e
Feature: Add modifications to langchain_v1
Christophe-Lamarche-Cossette Dec 16, 2025
8da2e13
Fix: Formatting error
Christophe-Lamarche-Cossette Dec 16, 2025
87a4ca1
Fix: Too many linejump
Christophe-Lamarche-Cossette Dec 16, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions libs/langchain/langchain_classic/embeddings/base.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@
"azure_openai": "langchain_openai",
"bedrock": "langchain_aws",
"cohere": "langchain_cohere",
"google_genai": "langchain_google_genai",
Copy link
Member

@mdrxy mdrxy Dec 16, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I would also expect this to be applied to libs/langchain_v1/langchain/embeddings/base.py

Copy link
Author

@Christophe-Lamarche-Cossette Christophe-Lamarche-Cossette Dec 16, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Good catch. The changes are applied to the langchain_v1 lib as well.

"google_vertexai": "langchain_google_vertexai",
"huggingface": "langchain_huggingface",
"mistralai": "langchain_mistralai",
Expand Down Expand Up @@ -155,6 +156,7 @@ def init_embeddings(
- `azure_openai` -> [`langchain-openai`](https://docs.langchain.com/oss/python/integrations/providers/openai)
- `bedrock` -> [`langchain-aws`](https://docs.langchain.com/oss/python/integrations/providers/aws)
- `cohere` -> [`langchain-cohere`](https://docs.langchain.com/oss/python/integrations/providers/cohere)
- `google_genai` -> [`langchain-google-genai`](https://docs.langchain.com/oss/python/integrations/providers/google)
- `google_vertexai` -> [`langchain-google-vertexai`](https://docs.langchain.com/oss/python/integrations/providers/google)
- `huggingface` -> [`langchain-huggingface`](https://docs.langchain.com/oss/python/integrations/providers/huggingface)
- `mistraiai` -> [`langchain-mistralai`](https://docs.langchain.com/oss/python/integrations/providers/mistralai)
Expand Down Expand Up @@ -207,6 +209,10 @@ def init_embeddings(
from langchain_openai import AzureOpenAIEmbeddings

return AzureOpenAIEmbeddings(model=model_name, **kwargs)
if provider == "google_genai":
from langchain_google_genai import GoogleGenerativeAIEmbeddings

return GoogleGenerativeAIEmbeddings(model=model_name, **kwargs)
if provider == "google_vertexai":
from langchain_google_vertexai import VertexAIEmbeddings

Expand Down
27 changes: 15 additions & 12 deletions libs/langchain/tests/unit_tests/embeddings/test_base.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,19 +9,22 @@
)


def test_parse_model_string() -> None:
@pytest.mark.parametrize(
("model_string", "expected_provider", "expected_model"),
[
("openai:text-embedding-3-small", "openai", "text-embedding-3-small"),
("bedrock:amazon.titan-embed-text-v1", "bedrock", "amazon.titan-embed-text-v1"),
("huggingface:BAAI/bge-base-en:v1.5", "huggingface", "BAAI/bge-base-en:v1.5"),
("google_genai:gemini-embedding-001", "google_genai", "gemini-embedding-001"),
],
)
def test_parse_model_string(
model_string: str, expected_provider: str, expected_model: str
) -> None:
"""Test parsing model strings into provider and model components."""
assert _parse_model_string("openai:text-embedding-3-small") == (
"openai",
"text-embedding-3-small",
)
assert _parse_model_string("bedrock:amazon.titan-embed-text-v1") == (
"bedrock",
"amazon.titan-embed-text-v1",
)
assert _parse_model_string("huggingface:BAAI/bge-base-en:v1.5") == (
"huggingface",
"BAAI/bge-base-en:v1.5",
assert _parse_model_string(model_string) == (
expected_provider,
expected_model,
)


Expand Down
5 changes: 5 additions & 0 deletions libs/langchain_v1/langchain/embeddings/base.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@
"azure_openai": "langchain_openai",
"bedrock": "langchain_aws",
"cohere": "langchain_cohere",
"google_genai": "langchain_google_genai",
"google_vertexai": "langchain_google_vertexai",
"huggingface": "langchain_huggingface",
"mistralai": "langchain_mistralai",
Expand Down Expand Up @@ -207,6 +208,10 @@ def init_embeddings(
from langchain_openai import AzureOpenAIEmbeddings

return AzureOpenAIEmbeddings(model=model_name, **kwargs)
if provider == "google_genai":
from langchain_google_genai import GoogleGenerativeAIEmbeddings

return GoogleGenerativeAIEmbeddings(model=model_name, **kwargs)
if provider == "google_vertexai":
from langchain_google_vertexai import VertexAIEmbeddings

Expand Down
25 changes: 13 additions & 12 deletions libs/langchain_v1/tests/unit_tests/embeddings/test_base.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,19 +9,20 @@
)


def test_parse_model_string() -> None:
@pytest.mark.parametrize(
("model_string", "expected_provider", "expected_model"),
[
("openai:text-embedding-3-small", "openai", "text-embedding-3-small"),
("bedrock:amazon.titan-embed-text-v1", "bedrock", "amazon.titan-embed-text-v1"),
("huggingface:BAAI/bge-base-en:v1.5", "huggingface", "BAAI/bge-base-en:v1.5"),
("google_genai:gemini-embedding-001", "google_genai", "gemini-embedding-001"),
],
)
def test_parse_model_string(model_string: str, expected_provider: str, expected_model: str) -> None:
"""Test parsing model strings into provider and model components."""
assert _parse_model_string("openai:text-embedding-3-small") == (
"openai",
"text-embedding-3-small",
)
assert _parse_model_string("bedrock:amazon.titan-embed-text-v1") == (
"bedrock",
"amazon.titan-embed-text-v1",
)
assert _parse_model_string("huggingface:BAAI/bge-base-en:v1.5") == (
"huggingface",
"BAAI/bge-base-en:v1.5",
assert _parse_model_string(model_string) == (
expected_provider,
expected_model,
)


Expand Down