Update embeddings and retrievers components

src/backend/langflow/components/embeddings/AmazonBedrockEmbeddings.py

@@ -1,7 +1,9 @@
 from typing import Optional
 
-from langchain.embeddings import BedrockEmbeddings
 from langchain.embeddings.base import Embeddings
+from langchain_community.embeddings import BedrockEmbeddings
+
+
 from langflow import CustomComponent
 
 

src/backend/langflow/components/embeddings/VertexAIEmbeddings.py

@@ -1,5 +1,5 @@
 from langflow import CustomComponent
-from langchain.embeddings import VertexAIEmbeddings
+from langchain_community.embeddings import VertexAIEmbeddings
 from typing import Optional, List
 
 
@@ -9,17 +9,45 @@ class VertexAIEmbeddingsComponent(CustomComponent):
 
     def build_config(self):
         return {
-            "credentials": {"display_name": "Credentials", "value": "", "file_types": [".json"], "field_type": "file"},
+            "credentials": {
-            "instance": {"display_name": "instance", "advanced": True, "field_type": "dict"},
+                "display_name": "Credentials",
-            "location": {"display_name": "Location", "value": "us-central1", "advanced": True},
+                "value": "",
+                "file_types": [".json"],
+                "field_type": "file",
+            },
+            "instance": {
+                "display_name": "instance",
+                "advanced": True,
+                "field_type": "dict",
+            },
+            "location": {
+                "display_name": "Location",
+                "value": "us-central1",
+                "advanced": True,
+            },
             "max_output_tokens": {"display_name": "Max Output Tokens", "value": 128},
-            "max_retries": {"display_name": "Max Retries", "value": 6, "advanced": True},
+            "max_retries": {
-            "model_name": {"display_name": "Model Name", "value": "textembedding-gecko"},
+                "display_name": "Max Retries",
+                "value": 6,
+                "advanced": True,
+            },
+            "model_name": {
+                "display_name": "Model Name",
+                "value": "textembedding-gecko",
+            },
             "n": {"display_name": "N", "value": 1, "advanced": True},
             "project": {"display_name": "Project", "advanced": True},
-            "request_parallelism": {"display_name": "Request Parallelism", "value": 5, "advanced": True},
+            "request_parallelism": {
+                "display_name": "Request Parallelism",
+                "value": 5,
+                "advanced": True,
+            },
             "stop": {"display_name": "Stop", "advanced": True},
-            "streaming": {"display_name": "Streaming", "value": False, "advanced": True},
+            "streaming": {
+                "display_name": "Streaming",
+                "value": False,
+                "advanced": True,
+            },
             "temperature": {"display_name": "Temperature", "value": 0.0},
             "top_k": {"display_name": "Top K", "value": 40, "advanced": True},
             "top_p": {"display_name": "Top P", "value": 0.95, "advanced": True},

src/backend/langflow/components/model_specs/AmazonBedrockSpecs.py

@@ -1,7 +1,9 @@
 from typing import Optional
 
 from langchain.llms.base import BaseLLM
-from langchain.llms.bedrock import Bedrock
+from langchain_community.llms.bedrock import Bedrock
+
+
 from langflow import CustomComponent
 
 

src/backend/langflow/components/models/HuggingFaceModel.py

@@ -1,7 +1,10 @@
 from typing import Optional
-from langflow import CustomComponent
+
-from langchain.llms.huggingface_endpoint import HuggingFaceEndpoint
 from langchain_community.chat_models.huggingface import ChatHuggingFace
+from langchain_community.llms.huggingface_endpoint import HuggingFaceEndpoint
+
+from langflow import CustomComponent
+
 from langflow.field_typing import Text
 
 

src/backend/langflow/components/retrievers/AmazonKendra.py

@@ -1,7 +1,9 @@
 from typing import Optional
-from langflow import CustomComponent
+
-from langchain.retrievers import AmazonKendraRetriever
 from langchain.schema import BaseRetriever
+from langchain_community.retrievers import AmazonKendraRetriever
+
+from langflow import CustomComponent
 
 
 class AmazonKendraRetrieverComponent(CustomComponent):

src/backend/langflow/components/retrievers/MetalRetriever.py

@@ -1,9 +1,11 @@
 from typing import Optional
-from langflow import CustomComponent
+
-from langchain.retrievers import MetalRetriever
 from langchain.schema import BaseRetriever
+from langchain_community.retrievers import MetalRetriever
 from metal_sdk.metal import Metal  # type: ignore
 
+from langflow import CustomComponent
+
 
 class MetalRetrieverComponent(CustomComponent):
     display_name: str = "Metal Retriever"