From c5528c6f932cc392cf5df2c86f94e7dc3b0554bb Mon Sep 17 00:00:00 2001 From: Akshay Kripalani <164727097+akshaykripalani@users.noreply.github.com> Date: Wed, 8 Jan 2025 23:18:51 +0530 Subject: [PATCH] fix: Google Embeddings model dimension mismatch. (#5502) * 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 --- .../langflow/components/embeddings/google_generative_ai.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/backend/base/langflow/components/embeddings/google_generative_ai.py b/src/backend/base/langflow/components/embeddings/google_generative_ai.py index edd2d63dd..8c27561ed 100644 --- a/src/backend/base/langflow/components/embeddings/google_generative_ai.py +++ b/src/backend/base/langflow/components/embeddings/google_generative_ai.py @@ -1,5 +1,4 @@ # from langflow.field_typing import Data -import numpy as np # TODO: remove ignore once the google package is published with types from google.ai.generativelanguage_v1beta.types import BatchEmbedContentsRequest @@ -46,7 +45,7 @@ class GoogleGenerativeAIEmbeddingsComponent(Component): batch_size: int = 100, task_type: str | None = None, titles: list[str] | None = None, - output_dimensionality: int | None = 1536, + output_dimensionality: int | None = 768, ) -> list[list[float]]: """Embed a list of strings. @@ -89,7 +88,7 @@ class GoogleGenerativeAIEmbeddingsComponent(Component): except Exception as e: msg = f"Error embedding content: {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 def embed_query( @@ -97,7 +96,7 @@ class GoogleGenerativeAIEmbeddingsComponent(Component): text: str, task_type: str | None = None, title: str | None = None, - output_dimensionality: int | None = 1536, + output_dimensionality: int | None = 768, ) -> list[float]: """Embed a text.