token统计部分

2025-03-16 23:11:32 +08:00
parent 71132315d8
commit 4bc222ba6f
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)