From 2e00829b1e7ebae50da09192dc8bf8110a2deca5 Mon Sep 17 00:00:00 2001 From: yihong Date: Mon, 25 Nov 2024 19:50:23 +0800 Subject: [PATCH] fix: drop useless and wrong code for zhipu embedding (#11069) Signed-off-by: yihong0618 --- .../zhipuai/text_embedding/text_embedding.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/api/core/model_runtime/model_providers/zhipuai/text_embedding/text_embedding.py b/api/core/model_runtime/model_providers/zhipuai/text_embedding/text_embedding.py index f629b62fd5..2428284ba9 100644 --- a/api/core/model_runtime/model_providers/zhipuai/text_embedding/text_embedding.py +++ b/api/core/model_runtime/model_providers/zhipuai/text_embedding/text_embedding.py @@ -105,17 +105,6 @@ class ZhipuAITextEmbeddingModel(_CommonZhipuaiAI, TextEmbeddingModel): return [list(map(float, e)) for e in embeddings], embedding_used_tokens - def embed_query(self, text: str) -> list[float]: - """Call out to ZhipuAI's embedding endpoint. - - Args: - text: The text to embed. - - Returns: - Embeddings for the text. - """ - return self.embed_documents([text])[0] - def _calc_response_usage(self, model: str, credentials: dict, tokens: int) -> EmbeddingUsage: """ Calculate response usage