Fix HuggingFace model error. (#3870)

### What problem does this PR solve?

#3865

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu
2024-12-05 13:28:42 +08:00
committed by GitHub
parent a88a1848ff
commit 593ffc4067
3 changed files with 5 additions and 4 deletions

View File

@ -745,7 +745,7 @@ class HuggingFaceEmbed(Base):
if not model_name:
raise ValueError("Model name cannot be None")
self.key = key
self.model_name = model_name
self.model_name = model_name.split("___")[0]
self.base_url = base_url or "http://127.0.0.1:8080"
def encode(self, texts: list):
@ -775,6 +775,7 @@ class HuggingFaceEmbed(Base):
else:
raise Exception(f"Error: {response.status_code} - {response.text}")
class VolcEngineEmbed(OpenAIEmbed):
def __init__(self, key, model_name, base_url="https://ark.cn-beijing.volces.com/api/v3"):
if not base_url: