From 6df17a334c77c557ca35bd0589f06f224d66167f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=96=B9=E7=A8=8B?= Date: Fri, 3 Jan 2025 19:19:17 +0800 Subject: [PATCH] fix: Update the API call address for the text_embedding model (#12342) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: 方程 --- .../gitee_ai/text_embedding/text_embedding.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/api/core/model_runtime/model_providers/gitee_ai/text_embedding/text_embedding.py b/api/core/model_runtime/model_providers/gitee_ai/text_embedding/text_embedding.py index a1fa89c5b..97d5ecc24 100644 --- a/api/core/model_runtime/model_providers/gitee_ai/text_embedding/text_embedding.py +++ b/api/core/model_runtime/model_providers/gitee_ai/text_embedding/text_embedding.py @@ -24,8 +24,5 @@ class GiteeAIEmbeddingModel(OAICompatEmbeddingModel): super().validate_credentials(model, credentials) @staticmethod - def _add_custom_parameters(credentials: dict, model: Optional[str]) -> None: - if model is None: - model = "bge-m3" - - credentials["endpoint_url"] = f"https://ai.gitee.com/api/serverless/{model}/v1/" + def _add_custom_parameters(credentials: dict, model: str) -> None: + credentials["endpoint_url"] = "https://ai.gitee.com/v1"