20 lines
571 B
Python
20 lines
571 B
Python
|
"""知识库搜索工具"""
|
|||
|
|
from langchain_core.tools import tool
|
|||
|
|
|
|||
|
|
|
|||
|
|
@tool
|
|||
|
|
def search_knowledge(query: str) -> str:
|
|||
|
|
"""搜索知识库（模拟）"""
|
|||
|
|
kb = {
|
|||
|
|
"黄庄三号": "黄庄三号是AI助手，严肃认真听话聪明",
|
|||
|
|
"LangGraph": "LangGraph是Agent框架，支持状态图、循环、持久化",
|
|||
|
|
"MCP": "MCP是Model Context Protocol，AI工具互操作标准协议",
|
|||
|
|
}
|
|||
|
|
for key, val in kb.items():
|
|||
|
|
if key in query:
|
|||
|
|
return val
|
|||
|
|
return f"未找到关于'{query}'的信息"
|
|||
|
|
|
|||
|
|
|
|||
|
|
TOOLS = [search_knowledge]
|