mirror of https://github.com/Mai-with-u/MaiBot.git
93 lines
3.4 KiB
Python
93 lines
3.4 KiB
Python
"""
|
||
根据关键词在jargon库中查询 - 工具实现
|
||
"""
|
||
|
||
from src.common.logger import get_logger
|
||
from src.jargon.jargon_miner import search_jargon
|
||
from .tool_registry import register_memory_retrieval_tool
|
||
|
||
logger = get_logger("memory_retrieval_tools")
|
||
|
||
|
||
async def query_jargon(
|
||
keyword: str,
|
||
chat_id: str,
|
||
fuzzy: bool = False,
|
||
search_all: bool = False
|
||
) -> str:
|
||
"""根据关键词在jargon库中查询
|
||
|
||
Args:
|
||
keyword: 关键词（黑话/俚语/缩写）
|
||
chat_id: 聊天ID
|
||
fuzzy: 是否使用模糊搜索，默认False（精确匹配）
|
||
search_all: 是否搜索全库（不限chat_id），默认False（仅搜索当前会话或global）
|
||
|
||
Returns:
|
||
str: 查询结果
|
||
"""
|
||
try:
|
||
content = str(keyword).strip()
|
||
if not content:
|
||
return "关键词为空"
|
||
|
||
# 根据参数执行搜索
|
||
search_chat_id = None if search_all else chat_id
|
||
results = search_jargon(
|
||
keyword=content,
|
||
chat_id=search_chat_id,
|
||
limit=1,
|
||
case_sensitive=False,
|
||
fuzzy=fuzzy
|
||
)
|
||
|
||
if results:
|
||
result = results[0]
|
||
translation = result.get("translation", "").strip()
|
||
meaning = result.get("meaning", "").strip()
|
||
search_type = "模糊搜索" if fuzzy else "精确匹配"
|
||
search_scope = "全库" if search_all else "当前会话或全局"
|
||
output = f"“{content}可能为黑话或者网络简写，翻译为：{translation}，含义为：{meaning}”"
|
||
logger.info(f"在jargon库中找到匹配（{search_scope}，{search_type}）: {content}")
|
||
return output
|
||
|
||
# 未命中
|
||
search_type = "模糊搜索" if fuzzy else "精确匹配"
|
||
search_scope = "全库" if search_all else "当前会话或全局"
|
||
logger.info(f"在jargon库中未找到匹配（{search_scope}，{search_type}）: {content}")
|
||
return f"未在jargon库中找到'{content}'的解释"
|
||
|
||
except Exception as e:
|
||
logger.error(f"查询jargon失败: {e}")
|
||
return f"查询失败: {str(e)}"
|
||
|
||
|
||
def register_tool():
|
||
"""注册工具"""
|
||
register_memory_retrieval_tool(
|
||
name="query_jargon",
|
||
description="根据关键词在jargon库中查询黑话/俚语/缩写的含义。支持大小写不敏感搜索和模糊搜索。默认优先搜索当前会话或全局jargon，可以设置为搜索全库。",
|
||
parameters=[
|
||
{
|
||
"name": "keyword",
|
||
"type": "string",
|
||
"description": "关键词（黑话/俚语/缩写），支持模糊搜索",
|
||
"required": True
|
||
},
|
||
{
|
||
"name": "fuzzy",
|
||
"type": "boolean",
|
||
"description": "是否使用模糊搜索（部分匹配），默认False（精确匹配）。当精确匹配找不到时，可以尝试使用模糊搜索。",
|
||
"required": False
|
||
},
|
||
{
|
||
"name": "search_all",
|
||
"type": "boolean",
|
||
"description": "是否搜索全库（不限chat_id），默认False（仅搜索当前会话或global的jargon）。当在当前会话中找不到时，可以尝试搜索全库。",
|
||
"required": False
|
||
}
|
||
],
|
||
execute_func=query_jargon
|
||
)
|
||
|