31 lines
706 B
Python
31 lines
706 B
Python
"""
|
|
记忆检索工具模块
|
|
提供统一的工具注册和管理系统
|
|
"""
|
|
|
|
from .tool_registry import (
|
|
MemoryRetrievalTool,
|
|
MemoryRetrievalToolRegistry,
|
|
register_memory_retrieval_tool,
|
|
get_tool_registry,
|
|
)
|
|
|
|
# 导入所有工具的注册函数
|
|
from .query_jargon import register_tool as register_query_jargon
|
|
from .query_chat_history import register_tool as register_query_chat_history
|
|
|
|
|
|
def init_all_tools():
|
|
"""初始化并注册所有记忆检索工具"""
|
|
register_query_jargon()
|
|
register_query_chat_history()
|
|
|
|
|
|
__all__ = [
|
|
"MemoryRetrievalTool",
|
|
"MemoryRetrievalToolRegistry",
|
|
"register_memory_retrieval_tool",
|
|
"get_tool_registry",
|
|
"init_all_tools",
|
|
]
|