diff --git a/__init__.py b/__init__.py
index 36ea3d8..c4dd3f5 100644
--- a/__init__.py
+++ b/__init__.py
@@ -427,7 +427,7 @@ def register(ctx) -> None:
     # Register tools manually since we can't register as memory provider
     def make_handler(tool_name: str):
         """Create a handler closure for a specific tool."""
-        return lambda args: provider.handle_tool_call(tool_name, args)
+        return lambda args, **kwargs: provider.handle_tool_call(tool_name, args)
 
     for schema in [PROFILE_SCHEMA, SEARCH_SCHEMA, CONCLUDE_SCHEMA]:
         ctx.register_tool(