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

fix: 修复腾讯云向量化失败

Browse Source
This commit is contained in:
shaohuzhang1 2024-10-22 18:55:13 +08:00 committed by shaohuzhang1
parent aeef9c7714
commit fc337cf6ef
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/setting/models_provider/impl/tencent_model_provider/model/embedding.py
View File

@ -5,8 +5,10 @@ from tencentcloud.common import credential
from tencentcloud.hunyuan.v20230901.hunyuan_client import HunyuanClient from tencentcloud.hunyuan.v20230901.hunyuan_client import HunyuanClient
from tencentcloud.hunyuan.v20230901.models import GetEmbeddingRequest 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]]: def embed_documents(self, texts: List[str]) -> List[List[float]]:
return [self.embed_query(text) for text in texts] return [self.embed_query(text) for text in texts]