40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
"""LPMM 知识库搜索插件 — 新 SDK 版本
|
||
|
||
提供 LLM 可调用的知识库搜索工具。
|
||
"""
|
||
|
||
from maibot_sdk import MaiBotPlugin, Tool
|
||
from maibot_sdk.types import ToolParameterInfo, ToolParamType
|
||
|
||
|
||
class KnowledgePlugin(MaiBotPlugin):
|
||
"""LPMM 知识库插件"""
|
||
|
||
@Tool(
|
||
"lpmm_search_knowledge",
|
||
description="从知识库中搜索相关信息，如果你需要知识，就使用这个工具",
|
||
parameters=[
|
||
ToolParameterInfo(name="query", param_type=ToolParamType.STRING, description="搜索查询关键词", required=True),
|
||
ToolParameterInfo(name="limit", param_type=ToolParamType.INTEGER, description="希望返回的相关知识条数，默认5", required=False, default=5),
|
||
],
|
||
)
|
||
async def handle_lpmm_search_knowledge(self, query: str = "", limit: int = 5, **kwargs):
|
||
"""执行知识库搜索"""
|
||
if not query:
|
||
return {"type": "info", "id": "", "content": "未提供搜索关键词"}
|
||
|
||
try:
|
||
limit_value = max(1, int(limit))
|
||
except (TypeError, ValueError):
|
||
limit_value = 5
|
||
|
||
result = await self.ctx.call_capability("knowledge.search", query=query, limit=limit_value)
|
||
if result and result.get("success"):
|
||
content = result.get("content", f"你不太了解有关{query}的知识")
|
||
return {"type": "lpmm_knowledge", "id": query, "content": content}
|
||
return {"type": "info", "id": query, "content": f"知识库搜索失败: {result}"}
|
||
|
||
|
||
def create_plugin():
|
||
return KnowledgePlugin()
|