Merge pull request #441 from Maple127667/main-fix

token统计部分
2025-03-17 17:23:18 +08:00
parent 6611d9b46d 4bc222ba6f
commit 4bf6f1e204
7 changed files with 43 additions and 13 deletions
--- a/src/plugins/chat/utils.py
+++ b/src/plugins/chat/utils.py
@@ -54,7 +54,7 @@ def is_mentioned_bot_in_message(message: MessageRecv) -> bool:

 async def get_embedding(text):
    """获取文本的embedding向量"""
-    llm = LLM_request(model=global_config.embedding)
+    llm = LLM_request(model=global_config.embedding,request_type = 'embedding')
    # return llm.get_embedding_sync(text)
    return await llm.get_embedding(text)