From fc337cf6ef13957ddf17b6ee3732cb8c83cb5d2b Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Tue, 22 Oct 2024 18:55:13 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E8=85=BE=E8=AE=AF?= =?UTF-8?q?=E4=BA=91=E5=90=91=E9=87=8F=E5=8C=96=E5=A4=B1=E8=B4=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../impl/tencent_model_provider/model/embedding.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/apps/setting/models_provider/impl/tencent_model_provider/model/embedding.py b/apps/setting/models_provider/impl/tencent_model_provider/model/embedding.py index 126e51b9d..0dcda11ca 100644 --- a/apps/setting/models_provider/impl/tencent_model_provider/model/embedding.py +++ b/apps/setting/models_provider/impl/tencent_model_provider/model/embedding.py @@ -5,8 +5,10 @@ from tencentcloud.common import credential from tencentcloud.hunyuan.v20230901.hunyuan_client import HunyuanClient from tencentcloud.hunyuan.v20230901.models import GetEmbeddingRequest +from setting.models_provider.base_model_provider import MaxKBBaseModel -class TencentEmbeddingModel(Embeddings): + +class TencentEmbeddingModel(MaxKBBaseModel, Embeddings): def embed_documents(self, texts: List[str]) -> List[List[float]]: return [self.embed_query(text) for text in texts]