diff --git a/hendrik.py b/hendrik.py index 2e87d5e..d73ab46 100644 --- a/hendrik.py +++ b/hendrik.py @@ -2,7 +2,7 @@ import os, sys, threading, time, signal import config -from tools import coder, rag, carrack +from tools import coder, rag, carrack, ragroleplay from services.xmpp_client import XMPPClient from services.telegram_client import TelegramClient @@ -30,6 +30,34 @@ tools_definition = [ gadget.tools_mapping( schema = rag.schema_inspect_collection, handler = rag.inspect_collection ), gadget.tools_mapping( schema = carrack.schema_sendhttprequest, handler = carrack.sendhttprequest ), + + # Rag Roleplay Tools + gadget.tools_mapping( schema = ragroleplay.schema_users_store, handler = ragroleplay.users_store ), + gadget.tools_mapping( schema = ragroleplay.schema_users_filter, handler = ragroleplay.users_filter ), + gadget.tools_mapping( schema = ragroleplay.schema_users_update, handler = ragroleplay.users_update ), + gadget.tools_mapping( schema = ragroleplay.schema_users_delete, handler = ragroleplay.users_delete ), + gadget.tools_mapping( schema = ragroleplay.schema_memories_check, handler = ragroleplay.memories_check ), + gadget.tools_mapping( schema = ragroleplay.schema_memories_store, handler = ragroleplay.memories_store ), + gadget.tools_mapping( schema = ragroleplay.schema_memories_filter, handler = ragroleplay.memories_filter ), + gadget.tools_mapping( schema = ragroleplay.schema_memories_summarize, handler = ragroleplay.memories_summarize ), + gadget.tools_mapping( schema = ragroleplay.schema_memories_update, handler = ragroleplay.memories_update ), + gadget.tools_mapping( schema = ragroleplay.schema_memories_delete, handler = ragroleplay.memories_delete ), + gadget.tools_mapping( schema = ragroleplay.schema_worlds_store, handler = ragroleplay.worlds_store ), + gadget.tools_mapping( schema = ragroleplay.schema_worlds_filter, handler = ragroleplay.worlds_filter ), + gadget.tools_mapping( schema = ragroleplay.schema_worlds_update, handler = ragroleplay.worlds_update ), + gadget.tools_mapping( schema = ragroleplay.schema_worlds_delete, handler = ragroleplay.worlds_delete ), + gadget.tools_mapping( schema = ragroleplay.schema_objects_store, handler = ragroleplay.objects_store ), + gadget.tools_mapping( schema = ragroleplay.schema_objects_filter, handler = ragroleplay.objects_filter ), + gadget.tools_mapping( schema = ragroleplay.schema_objects_update, handler = ragroleplay.objects_update ), + gadget.tools_mapping( schema = ragroleplay.schema_objects_delete, handler = ragroleplay.objects_delete ), + gadget.tools_mapping( schema = ragroleplay.schema_outfits_store, handler = ragroleplay.outfits_store ), + gadget.tools_mapping( schema = ragroleplay.schema_outfits_filter, handler = ragroleplay.outfits_filter ), + gadget.tools_mapping( schema = ragroleplay.schema_outfits_update, handler = ragroleplay.outfits_update ), + gadget.tools_mapping( schema = ragroleplay.schema_outfits_delete, handler = ragroleplay.outfits_delete ), + gadget.tools_mapping( schema = ragroleplay.schema_todos_store, handler = ragroleplay.todos_store ), + gadget.tools_mapping( schema = ragroleplay.schema_todos_filter, handler = ragroleplay.todos_filter ), + gadget.tools_mapping( schema = ragroleplay.schema_todos_update, handler = ragroleplay.todos_update ), + gadget.tools_mapping( schema = ragroleplay.schema_todos_delete, handler = ragroleplay.todos_delete ), ]