update

baidubce · Jan 20, 2025 · 2619d37 · 2619d37
1 parent 73ef777
commit 2619d37
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 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
@@ -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/.../src/main/java/com/baidubce/appbuilder/model/knowledgebase/QueryKnowledgeBaseRequest.java b/.../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;