@@ -123,8 +123,10 @@ async def test_hybrid_node_search_with_limit():
123123 assert mock_fulltext_search .call_count == 1
124124 assert mock_similarity_search .call_count == 1
125125 # Verify that the limit was passed to the search functions
126- mock_fulltext_search .assert_called_with (mock_driver , 'Test' , ['1' ], 2 )
127- mock_similarity_search .assert_called_with (mock_driver , [0.1 , 0.2 , 0.3 ], ['1' ], 2 )
126+ mock_fulltext_search .assert_called_with (mock_driver , SearchFilters (), 'Test' , ['1' ], 2 )
127+ mock_similarity_search .assert_called_with (
128+ mock_driver , SearchFilters (), [0.1 , 0.2 , 0.3 ], ['1' ], 2
129+ )
128130
129131
130132@pytest .mark .asyncio
@@ -160,5 +162,7 @@ async def test_hybrid_node_search_with_limit_and_duplicates():
160162 assert set (node .name for node in results ) == {'Alice' , 'Bob' , 'Charlie' }
161163 assert mock_fulltext_search .call_count == 1
162164 assert mock_similarity_search .call_count == 1
163- mock_fulltext_search .assert_called_with (mock_driver , 'Test' , ['1' ], 4 )
164- mock_similarity_search .assert_called_with (mock_driver , [0.1 , 0.2 , 0.3 ], ['1' ], 4 )
165+ mock_fulltext_search .assert_called_with (mock_driver , SearchFilters (), 'Test' , ['1' ], 4 )
166+ mock_similarity_search .assert_called_with (
167+ mock_driver , SearchFilters (), [0.1 , 0.2 , 0.3 ], ['1' ], 4
168+ )
0 commit comments