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

fix: improved the huggingface emdeddings component, to handle local inference and serverless inference (#6292)

Browse source 
Update huggingface_inference_api.py
This commit is contained in:
Edwin Jose 2025-02-12 11:14:42 -05:00 • committed by GitHub
commit 553e3a0b12
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/backend/base/langflow/components/embeddings/huggingface_inference_api.py
View file

@ -2,6 +2,8 @@ from urllib.parse import urlparse
import requests
from langchain_community.embeddings.huggingface import HuggingFaceInferenceAPIEmbeddings
# Next update: use langchain_huggingface
from pydantic import SecretStr
from tenacity import retry, stop_after_attempt, wait_fixed
@ -21,7 +23,7 @@ class HuggingFaceInferenceAPIEmbeddingsComponent(LCEmbeddingsModel):
SecretStrInput(
name="api_key",
display_name="API Key",
advanced=True,
advanced=False,
info="Required for non-local inference endpoints. Local inference does not require an API Key.",
),
MessageTextInput(
@ -83,11 +85,14 @@ class HuggingFaceInferenceAPIEmbeddingsComponent(LCEmbeddingsModel):
def build_embeddings(self) -> Embeddings:
api_url = self.get_api_url()
is_local_url = api_url.startswith(("http://localhost", "http://127.0.0.1"))
is_local_url = (
api_url.startswith(("http://localhost", "http://127.0.0.1", "http://0.0.0.0", "http://docker"))
or "huggingface.co" not in api_url.lower()
)
if not self.api_key and is_local_url:
self.validate_inference_endpoint(api_url)
api_key = SecretStr("DummyAPIKeyForLocalDeployment")
api_key = SecretStr("APIKeyForLocalDeployment")
elif not self.api_key:
msg = "API Key is required for non-local inference endpoints"
raise ValueError(msg)