diff --git a/surfsense_backend/app/tasks/chat/stream_new_chat.py b/surfsense_backend/app/tasks/chat/stream_new_chat.py index 0e566b418..a319554a5 100644 --- a/surfsense_backend/app/tasks/chat/stream_new_chat.py +++ b/surfsense_backend/app/tasks/chat/stream_new_chat.py @@ -792,10 +792,16 @@ async def _stream_agent_events( f"Report generation failed: {error_msg}", "error", ) - elif tool_name in ("create_notion_page", "update_notion_page", "delete_notion_page"): + 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}, + tool_output + if isinstance(tool_output, dict) + else {"result": tool_output}, ) else: yield streaming_service.format_tool_output_available(