zhuchaowe/qwen_agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #34 from sparticleinc/fix/local-embedding-dimensions

Browse Source 
Skip dimensions for local embedding model
This commit is contained in:
csh28 2026-05-28 10:40:04 +08:00 committed by GitHub
commit 0260663aa9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
embedding/manager.py
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(