diff --git a/surfsense_backend/app/agents/new_chat/chat_deepagent.py b/surfsense_backend/app/agents/new_chat/chat_deepagent.py index d58a0fadb..525b18e7c 100644 --- a/surfsense_backend/app/agents/new_chat/chat_deepagent.py +++ b/surfsense_backend/app/agents/new_chat/chat_deepagent.py @@ -273,6 +273,7 @@ async def create_surfsense_deep_agent( system_prompt=system_prompt, context_schema=SurfSenseContextSchema, checkpointer=checkpointer, + interrupt_on={"create_notion_page": True}, ) return agent diff --git a/surfsense_backend/app/agents/new_chat/tools/registry.py b/surfsense_backend/app/agents/new_chat/tools/registry.py index 8092a6104..3716a19b6 100644 --- a/surfsense_backend/app/agents/new_chat/tools/registry.py +++ b/surfsense_backend/app/agents/new_chat/tools/registry.py @@ -45,6 +45,7 @@ from langchain_core.tools import BaseTool from app.db import ChatVisibility +from .create_notion_page import create_create_notion_page_tool from .display_image import create_display_image_tool from .generate_image import create_generate_image_tool from .knowledge_base import create_search_knowledge_base_tool @@ -202,13 +203,12 @@ BUILTIN_TOOLS: list[ToolDefinition] = [ # ========================================================================= # ADD YOUR CUSTOM TOOLS BELOW # ========================================================================= - # Example: - # ToolDefinition( - # name="my_custom_tool", - # description="What my tool does", - # factory=lambda deps: create_my_custom_tool(...), - # requires=["search_space_id"], - # ), + ToolDefinition( + name="create_notion_page", + description="Create a new page in the user's Notion workspace", + factory=lambda deps: create_create_notion_page_tool(), + requires=[], + ), ]