From f2a1babb7512d6546a792022f94b0aa3b5709e41 Mon Sep 17 00:00:00 2001
From: XI <xman.india@gmail.com>
Date: Thu, 4 Jun 2026 08:48:10 +0100
Subject: [PATCH] fix: set up mock engine extraction attrs in
 test_register_handlers_creates_working_handler

---
 api/tests/test_custom_tools.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/api/tests/test_custom_tools.py b/api/tests/test_custom_tools.py
index 0b1f5a77..186deb44 100644
--- a/api/tests/test_custom_tools.py
+++ b/api/tests/test_custom_tools.py
@@ -1201,6 +1201,9 @@ class TestCustomToolManagerUnit:
             mock_engine
         )
         mock_engine.llm = mock_llm
+        mock_engine._perform_variable_extraction_if_needed = AsyncMock()
+        mock_engine._current_node = None
+        mock_engine._gathered_context = {}
 
         manager = CustomToolManager(mock_engine)