assertEquals("original query", result);
    }

    @Test
    public void test_regenerateQuery_exception_returnsFailedQuery() {
        client.setChatResponse(null); // causes LlmException

        final String result = client.regenerateQuery("test question", "original query", "no_results", Collections.emptyList());

                    logger.info("[RAG] Primary search returned 0 results, regenerating query. originalQuery={}", query);
                    final String newQuery = llmClientManager.regenerateQuery(userMessage, query, "no_results", history);
                    if (StringUtil.isNotBlank(newQuery) && !newQuery.equals(query)) {
                        logger.info("[RAG] Regenerated query. newQuery={}", newQuery);

        request.addSystemMessage(resolvedPrompt);

        addHistoryWithBudget(request, history, getHistoryMaxChars());
        request.addUserMessage(wrapUserInput(userMessage));
        applyPromptTypeParams(request, "noresults");
        request.setStream(true);

        streamChatWithConcurrencyControl(request, callback);
    }

    @Override