feat: Allow custom model and params with huggingface_base_url (#3574)

This commit is contained in:
Vedant Thakkar
2025-10-14 17:33:18 +05:30
committed by GitHub
parent ce8a285003
commit ea22e8d9cd
2 changed files with 35 additions and 1 deletions
+4 -1
View File
@@ -18,6 +18,7 @@ class HuggingFaceEmbedding(EmbeddingBase):
if config.huggingface_base_url:
self.client = OpenAI(base_url=config.huggingface_base_url)
self.config.model = self.config.model or "tei"
else:
self.config.model = self.config.model or "multi-qa-MiniLM-L6-cos-v1"
@@ -36,6 +37,8 @@ class HuggingFaceEmbedding(EmbeddingBase):
list: The embedding vector.
"""
if self.config.huggingface_base_url:
return self.client.embeddings.create(input=text, model="tei").data[0].embedding
return self.client.embeddings.create(
input=text, model=self.config.model, **self.config.model_kwargs
).data[0].embedding
else:
return self.model.encode(text, convert_to_numpy=True).tolist()