diff --git a/agentscope-extensions/agentscope-extensions-rag-dify/src/main/java/io/agentscope/core/rag/integration/dify/DifyRAGClient.java b/agentscope-extensions/agentscope-extensions-rag-dify/src/main/java/io/agentscope/core/rag/integration/dify/DifyRAGClient.java index 7e352a9b8..c7d86a61d 100644 --- a/agentscope-extensions/agentscope-extensions-rag-dify/src/main/java/io/agentscope/core/rag/integration/dify/DifyRAGClient.java +++ b/agentscope-extensions/agentscope-extensions-rag-dify/src/main/java/io/agentscope/core/rag/integration/dify/DifyRAGClient.java @@ -158,19 +158,19 @@ public Mono retrieve(String query, Integer limit) { retrievalModel.put("reranking_enable", true); if (config.getRerankConfig() != null) { - Map rerankingMode = new HashMap<>(); + Map rerankingModel = new HashMap<>(); if (config.getRerankConfig().getProviderName() != null) { - rerankingMode.put( + rerankingModel.put( "reranking_provider_name", config.getRerankConfig().getProviderName()); } if (config.getRerankConfig().getModelName() != null) { - rerankingMode.put( + rerankingModel.put( "reranking_model_name", config.getRerankConfig().getModelName()); } - if (!rerankingMode.isEmpty()) { - retrievalModel.put("reranking_mode", rerankingMode); + if (!rerankingModel.isEmpty()) { + retrievalModel.put("reranking_model", rerankingModel); } } } else { diff --git a/agentscope-extensions/agentscope-extensions-rag-dify/src/test/java/io/agentscope/core/rag/integration/dify/DifyRAGClientTest.java b/agentscope-extensions/agentscope-extensions-rag-dify/src/test/java/io/agentscope/core/rag/integration/dify/DifyRAGClientTest.java index f35463b20..2bca9bc81 100644 --- a/agentscope-extensions/agentscope-extensions-rag-dify/src/test/java/io/agentscope/core/rag/integration/dify/DifyRAGClientTest.java +++ b/agentscope-extensions/agentscope-extensions-rag-dify/src/test/java/io/agentscope/core/rag/integration/dify/DifyRAGClientTest.java @@ -246,11 +246,11 @@ void testIncludeRerankingConfig() throws Exception { assertEquals(true, retrievalModel.get("reranking_enable")); @SuppressWarnings("unchecked") - Map rerankingMode = - (Map) retrievalModel.get("reranking_mode"); - assertNotNull(rerankingMode); - assertEquals("cohere", rerankingMode.get("reranking_provider_name")); - assertEquals("rerank-english-v2.0", rerankingMode.get("reranking_model_name")); + Map rerankingModel = + (Map) retrievalModel.get("reranking_model"); + assertNotNull(rerankingModel); + assertEquals("cohere", rerankingModel.get("reranking_provider_name")); + assertEquals("rerank-english-v2.0", rerankingModel.get("reranking_model_name")); } @Test