Spaces:
Sleeping
Sleeping
File size: 944 Bytes
8c43ca8 a7672c3 8c43ca8 66ba458 dd53ab9 a7672c3 8c43ca8 a7672c3 85ff578 66ba458 85ff578 dd53ab9 85ff578 a7672c3 8c43ca8 a7672c3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | """
Re-exports real tools (memory, scheduler, summarizer) as LangChain tools.
Replaces the previous mock implementations.
"""
# Import modules so register_tool decorators fire and populate base.TOOLS
from . import memory as _memory_mod # noqa: F401
from . import scheduler as _scheduler_mod # noqa: F401
from . import summarizer as _summarizer_mod # noqa: F401
from . import chart as _chart_mod # noqa: F401
from . import rag as _rag_mod # noqa: F401
from .base import TOOLS as _REGISTRY, get_langchain_tools
_ALLOWED = {
# Facilitator
"summarize_chat",
"summarize_chart",
# Scheduler
"get_schedule", "add_event", "update_event", "delete_event",
"add_reminder", "get_reminders",
# Memory
"save_memory", "get_memories",
# Web
"read_link",
# Knowledge base (PDF RAG)
"rag_search",
}
TOOLS = [t for t in get_langchain_tools() if t.name in _ALLOWED]
TOOL_MAP = {t.name: t for t in TOOLS}
|