jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: Google Embeddings model dimension mismatch. (#5502)

Browse source 
* Fix Google Embeddings model dimension mismatch.

* edit model size

* [autofix.ci] apply automated fixes

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: Eric Hare <ericrhare@gmail.com>
This commit is contained in:
Akshay Kripalani 2025-01-08 23:18:51 +05:30 • committed by GitHub
commit c5528c6f93
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/base/langflow/components/embeddings/google_generative_ai.py
View file

@ -1,5 +1,4 @@
# from langflow.field_typing import Data # from langflow.field_typing import Data
import numpy as np
# TODO: remove ignore once the google package is published with types # TODO: remove ignore once the google package is published with types
from google.ai.generativelanguage_v1beta.types import BatchEmbedContentsRequest from google.ai.generativelanguage_v1beta.types import BatchEmbedContentsRequest
@ -46,7 +45,7 @@ class GoogleGenerativeAIEmbeddingsComponent(Component):
batch_size: int = 100, batch_size: int = 100,
task_type: str | None = None, task_type: str | None = None,
titles: list[str] | None = None, titles: list[str] | None = None,
output_dimensionality: int | None = 1536, output_dimensionality: int | None = 768,
) -> list[list[float]]: ) -> list[list[float]]:
"""Embed a list of strings. """Embed a list of strings.
@ -89,7 +88,7 @@ class GoogleGenerativeAIEmbeddingsComponent(Component):
except Exception as e: except Exception as e:
msg = f"Error embedding content: {e}" msg = f"Error embedding content: {e}"
raise GoogleGenerativeAIError(msg) from e raise GoogleGenerativeAIError(msg) from e
embeddings.extend([list(np.pad(e.values, (0, 768), "constant")) for e in result.embeddings]) embeddings.extend([list(e.values) for e in result.embeddings])
return embeddings return embeddings
def embed_query( def embed_query(
@ -97,7 +96,7 @@ class GoogleGenerativeAIEmbeddingsComponent(Component):
text: str, text: str,
task_type: str | None = None, task_type: str | None = None,
title: str | None = None, title: str | None = None,
output_dimensionality: int | None = 1536, output_dimensionality: int | None = 768,
) -> list[float]: ) -> list[float]:
"""Embed a text. """Embed a text.