diff --git a/embedding/manager.py b/embedding/manager.py index e81b679..19577a7 100644 --- a/embedding/manager.py +++ b/embedding/manager.py @@ -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(