mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-06-13 08:15:21 +02:00
fix: set up mock engine extraction attrs in test_register_handlers_creates_working_handler
This commit is contained in:
parent
3a4d2cee21
commit
f2a1babb75
1 changed files with 3 additions and 0 deletions
|
|
@ -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)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue