BruceSun
/
MaiBot
mirror of https://github.com/Mai-with-u/MaiBot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: 修复不应用 embedding 模型配置的现象

pull/75/head
Rikki 2025-03-07 05:50:59 +08:00
parent dac57cf154
commit b69c9ac7f7
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/plugins/models/utils_model.py
View File

@ -317,12 +317,11 @@ class LLM_request:
logger.error("达到最大重试次数，请求仍然失败") logger.error("达到最大重试次数，请求仍然失败")
raise RuntimeError("达到最大重试次数，API请求仍然失败") raise RuntimeError("达到最大重试次数，API请求仍然失败")
def get_embedding_sync(self, text: str, model: str = "BAAI/bge-m3") -> Union[list, None]: def get_embedding_sync(self, text: str) -> Union[list, None]:
"""同步方法：获取文本的embedding向量 """同步方法：获取文本的embedding向量
Args: Args:
text: 需要获取embedding的文本 text: 需要获取embedding的文本
model: 使用的模型名称，默认为"BAAI/bge-m3"
Returns: Returns:
list: embedding向量，如果失败则返回None list: embedding向量，如果失败则返回None
@ -333,7 +332,7 @@ class LLM_request:
} }
data = { data = {
"model": model, "model": self.model_name,
"input": text, "input": text,
"encoding_format": "float" "encoding_format": "float"
} }