assertEquals(1, result.size());
        assertEquals("Test Doc", result.get(0).get("title"));
    }

    @Test
    public void test_searchWithQuery_emptyResults() {
        chatClient.setSearchResults(Collections.emptyList());
        chatClient.resetSearchDocumentsCalled();

        final List<Map<String, Object>> result = chatClient.testSearchWithQuery("nonexistent");