Mikhail Putilovskij
b52fdc4670
- add save/load/reset/context handlers and matrix interception flows - persist current session and last token usage in prototype state
62 lines
2.7 KiB
Python
62 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
from adapter.matrix.handlers.chat import (
|
|
handle_list_chats,
|
|
make_handle_archive,
|
|
make_handle_new_chat,
|
|
make_handle_rename,
|
|
)
|
|
from adapter.matrix.handlers.confirm import make_handle_cancel, make_handle_confirm
|
|
from adapter.matrix.handlers.context_commands import (
|
|
make_handle_context,
|
|
make_handle_load,
|
|
make_handle_reset,
|
|
make_handle_save,
|
|
)
|
|
from adapter.matrix.handlers.settings import (
|
|
handle_help,
|
|
handle_settings,
|
|
handle_settings_connectors,
|
|
handle_settings_plan,
|
|
handle_settings_safety,
|
|
handle_settings_skills,
|
|
handle_settings_soul,
|
|
handle_settings_status,
|
|
handle_settings_whoami,
|
|
handle_toggle_skill,
|
|
)
|
|
from core.handler import EventDispatcher
|
|
from core.protocol import IncomingCallback, IncomingCommand
|
|
|
|
|
|
def register_matrix_handlers(
|
|
dispatcher: EventDispatcher,
|
|
client=None,
|
|
store=None,
|
|
agent_api=None,
|
|
prototype_state=None,
|
|
agent_base_url: str = "http://127.0.0.1:8000",
|
|
) -> None:
|
|
dispatcher.register(IncomingCommand, "new", make_handle_new_chat(client, store))
|
|
dispatcher.register(IncomingCommand, "chats", handle_list_chats)
|
|
dispatcher.register(IncomingCommand, "rename", make_handle_rename(client, store))
|
|
dispatcher.register(IncomingCommand, "archive", make_handle_archive(client, store))
|
|
dispatcher.register(IncomingCommand, "help", handle_help)
|
|
dispatcher.register(IncomingCommand, "settings", handle_settings)
|
|
dispatcher.register(IncomingCommand, "settings_skills", handle_settings_skills)
|
|
dispatcher.register(IncomingCommand, "settings_connectors", handle_settings_connectors)
|
|
dispatcher.register(IncomingCommand, "settings_soul", handle_settings_soul)
|
|
dispatcher.register(IncomingCommand, "settings_safety", handle_settings_safety)
|
|
dispatcher.register(IncomingCommand, "settings_plan", handle_settings_plan)
|
|
dispatcher.register(IncomingCommand, "settings_status", handle_settings_status)
|
|
dispatcher.register(IncomingCommand, "settings_whoami", handle_settings_whoami)
|
|
|
|
dispatcher.register(IncomingCallback, "confirm", make_handle_confirm(store))
|
|
dispatcher.register(IncomingCallback, "cancel", make_handle_cancel(store))
|
|
dispatcher.register(IncomingCallback, "toggle_skill", handle_toggle_skill)
|
|
|
|
if agent_api is not None and prototype_state is not None:
|
|
dispatcher.register(IncomingCommand, "save", make_handle_save(agent_api, store, prototype_state))
|
|
dispatcher.register(IncomingCommand, "load", make_handle_load(store, prototype_state))
|
|
dispatcher.register(IncomingCommand, "reset", make_handle_reset(store, agent_base_url))
|
|
dispatcher.register(IncomingCommand, "context", make_handle_context(store, prototype_state))
|