Merge pull request #38 from sparticleinc/hotfix/staging-local-embedding-dimensions

Hotfix: skip dimensions for local embedding model
This commit is contained in:
csh28 2026-05-28 10:52:50 +08:00 committed by GitHub
commit 6ff9555e91
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -64,7 +64,7 @@ class GlobalModelManager:
"model": self.external_model_name,
"input": texts,
}
if self.external_dimensions and self.external_model_name != "text-embedding-ada-002":
if self.external_dimensions and self.external_model_name not in ("text-embedding-ada-002", "local-embedding"):
payload["dimensions"] = self.external_dimensions
response = requests.post(