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

refactor: Remove unused code and update HuggingFaceInferenceAPIEmbeddingsComponent

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 21:43:27 -03:00
commit 1163515e7a
1 changed files with 2 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/backend/base/langflow/components/embeddings/HuggingFaceInferenceAPIEmbeddings.py
View file

@ -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)