From 2619d3743e7c0fe6e77b02d365e844e63b6e9006 Mon Sep 17 00:00:00 2001 From: yinjiaqi Date: Mon, 20 Jan 2025 18:31:47 +0800 Subject: [PATCH] update --- .../appbuilder/console/knowledgebase/Knowledgebase.java | 9 ++++++++- .../model/knowledgebase/QueryKnowledgeBaseRequest.java | 2 +- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/java/src/main/java/com/baidubce/appbuilder/console/knowledgebase/Knowledgebase.java b/java/src/main/java/com/baidubce/appbuilder/console/knowledgebase/Knowledgebase.java index 13fd4959..8a6c6457 100644 --- a/java/src/main/java/com/baidubce/appbuilder/console/knowledgebase/Knowledgebase.java +++ b/java/src/main/java/com/baidubce/appbuilder/console/knowledgebase/Knowledgebase.java @@ -722,6 +722,9 @@ public ChunksDescribeResponse describeChunks(String documentId, String marker, I public QueryKnowledgeBaseResponse queryKnowledgeBase(QueryKnowledgeBaseRequest request) throws IOException, AppBuilderServerException { + if (request.rank_score_threshold == null) { + request.rank_score_threshold = 0.4f; + } String url = AppBuilderConfig.QUERY_KNOWLEDGEBASE_URL; String jsonBody = JsonUtils.serialize(request); @@ -734,10 +737,14 @@ public QueryKnowledgeBaseResponse queryKnowledgeBase(QueryKnowledgeBaseRequest r return respBody; } - public QueryKnowledgeBaseResponse queryKnowledgeBase(String query, String type, float rank_score_threshold, Integer top, Integer skip, + public QueryKnowledgeBaseResponse queryKnowledgeBase(String query, String type, Float rank_score_threshold, Integer top, Integer skip, String[] knowledgebaseIDs, QueryKnowledgeBaseRequest.MetadataFilters filters, QueryKnowledgeBaseRequest.QueryPipelineConfig pipelineConfig) throws IOException, AppBuilderServerException { + if (rank_score_threshold == null) { + rank_score_threshold = 0.4f; + } + String url = AppBuilderConfig.QUERY_KNOWLEDGEBASE_URL; QueryKnowledgeBaseRequest request = new QueryKnowledgeBaseRequest(query, type, rank_score_threshold,top, skip, knowledgebaseIDs, filters, pipelineConfig); String jsonBody = JsonUtils.serialize(request); diff --git a/java/src/main/java/com/baidubce/appbuilder/model/knowledgebase/QueryKnowledgeBaseRequest.java b/java/src/main/java/com/baidubce/appbuilder/model/knowledgebase/QueryKnowledgeBaseRequest.java index 66a52aba..d3f7d570 100644 --- a/java/src/main/java/com/baidubce/appbuilder/model/knowledgebase/QueryKnowledgeBaseRequest.java +++ b/java/src/main/java/com/baidubce/appbuilder/model/knowledgebase/QueryKnowledgeBaseRequest.java @@ -5,7 +5,7 @@ public class QueryKnowledgeBaseRequest { private String query; private String type; - private float rank_score_threshold = 0.4f; + private Float rank_score_threshold; private Integer top; private Integer skip; private String[] knowledgebase_ids;