diff --git a/surfsense_backend/app/tasks/chat/stream_new_chat.py b/surfsense_backend/app/tasks/chat/stream_new_chat.py index 2cd05e80c..204bbb54a 100644 --- a/surfsense_backend/app/tasks/chat/stream_new_chat.py +++ b/surfsense_backend/app/tasks/chat/stream_new_chat.py @@ -693,6 +693,11 @@ async def _stream_agent_events( yield streaming_service.format_terminal_info( "Knowledge base search completed", "success" ) + elif tool_name in ("create_notion_page", "update_notion_page", "delete_notion_page"): + yield streaming_service.format_tool_output_available( + tool_call_id, + tool_output if isinstance(tool_output, dict) else {"result": tool_output}, + ) else: yield streaming_service.format_tool_output_available( tool_call_id,