From 1163515e7af856e880208e37cc5c971605e00e36 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 21:43:27 -0300 Subject: [PATCH] refactor: Remove unused code and update HuggingFaceInferenceAPIEmbeddingsComponent --- .../HuggingFaceInferenceAPIEmbeddings.py | 16 ++-------------- 1 file changed, 2 insertions(+), 14 deletions(-) diff --git a/src/backend/base/langflow/components/embeddings/HuggingFaceInferenceAPIEmbeddings.py b/src/backend/base/langflow/components/embeddings/HuggingFaceInferenceAPIEmbeddings.py index 6fef75ef1..cd80f7b47 100644 --- a/src/backend/base/langflow/components/embeddings/HuggingFaceInferenceAPIEmbeddings.py +++ b/src/backend/base/langflow/components/embeddings/HuggingFaceInferenceAPIEmbeddings.py @@ -3,7 +3,7 @@ from pydantic.v1.types import SecretStr from langflow.base.models.model import LCModelComponent from langflow.field_typing import Embeddings -from langflow.io import BoolInput, DictInput, Output, SecretStrInput, TextInput +from langflow.io import Output, SecretStrInput, TextInput class HuggingFaceInferenceAPIEmbeddingsComponent(LCModelComponent): @@ -16,10 +16,6 @@ class HuggingFaceInferenceAPIEmbeddingsComponent(LCModelComponent): SecretStrInput(name="api_key", display_name="API Key", advanced=True), TextInput(name="api_url", display_name="API URL", advanced=True, value="http://localhost:8080"), TextInput(name="model_name", display_name="Model Name", value="BAAI/bge-large-en-v1.5"), - TextInput(name="cache_folder", display_name="Cache Folder", advanced=True), - DictInput(name="encode_kwargs", display_name="Encode Kwargs", advanced=True), - DictInput(name="model_kwargs", display_name="Model Kwargs", advanced=True), - BoolInput(name="multi_process", display_name="Multi Process", advanced=True), ] outputs = [ @@ -32,12 +28,4 @@ class HuggingFaceInferenceAPIEmbeddingsComponent(LCModelComponent): api_key = SecretStr(self.api_key) - return HuggingFaceInferenceAPIEmbeddings( - api_key=api_key, - api_url=self.api_url, - model_name=self.model_name, - cache_folder=self.cache_folder, - encode_kwargs=self.encode_kwargs, - model_kwargs=self.model_kwargs, - multi_process=self.multi_process, - ) + return HuggingFaceInferenceAPIEmbeddings(api_key=api_key, api_url=self.api_url, model_name=self.model_name)