From 9628939990d9130f87effe70209531af20a18b91 Mon Sep 17 00:00:00 2001 From: lifl00001 <565644596@qq.com> Date: Mon, 9 Feb 2026 19:46:52 +0800 Subject: [PATCH] fix(dify): correct reranking_model JSON key in DifyRAG retrieval request Fixed incorrect JSON key 'reranking_mode' to 'reranking_model' in DifyRAG retrieval request payload. Also renamed variable from 'rerankingMode' to 'rerankingModel' for consistency. Changes: - DifyRAGClient.java: Changed 'reranking_mode' to 'reranking_model' - DifyRAGClientTest.java: Updated test to use correct key name Reference: Dify API expects 'reranking_model' for reranking configuration Co-Authored-By: Claude Code --- .../core/rag/integration/dify/DifyRAGClient.java | 10 +++++----- .../core/rag/integration/dify/DifyRAGClientTest.java | 10 +++++----- 2 files changed, 10 insertions(+), 10 deletions(-) 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