diff --git a/libs/vertexai/langchain_google_vertexai/vectorstores/vectorstores.py b/libs/vertexai/langchain_google_vertexai/vectorstores/vectorstores.py index 31df7954e..08e9c6f60 100644 --- a/libs/vertexai/langchain_google_vertexai/vectorstores/vectorstores.py +++ b/libs/vertexai/langchain_google_vertexai/vectorstores/vectorstores.py @@ -202,11 +202,11 @@ def _get_default_embeddings(cls) -> Embeddings: ) # TODO: Change to vertexai embbedingss - from langchain_community import ( - embeddings, # type: ignore[import-not-found, unused-ignore] + from langchain_community.embeddings import ( + TensorflowHubEmbeddings, # type: ignore[import-not-found, unused-ignore] ) - return embeddings.TensorflowHubEmbeddings() + return TensorflowHubEmbeddings() def _generate_unique_ids(self, number: int) -> List[str]: """Generates a list of unique ids of length `number`