Do not include document text in response discussion langchain4j#974

One16-KS · May 6, 2024 · 8d6a66a · 8d6a66a
1 parent f3142b3
commit 8d6a66a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/langchain4j-jina-ai/src/main/java/dev/langchain4j/model/jinaAi/rerank/JinaScoringModel.java b/langchain4j-jina-ai/src/main/java/dev/langchain4j/model/jinaAi/rerank/JinaScoringModel.java
@@ -68,6 +68,7 @@ public Response<List<Double>> scoreAll(List<TextSegment> segments, String query)
                 .documents(segments.stream()
                         .map(TextSegment::text)
                         .collect(toList()))
+                .returnDocuments(false)  //decreasing response size, do not include text in response
                 .build();
 
         RerankResponse response = withRetry(() -> client.rerank(request), maxRetries);

diff --git a/langchain4j-jina-ai/src/main/java/dev/langchain4j/model/jinaAi/rerank/RerankRequest.java b/langchain4j-jina-ai/src/main/java/dev/langchain4j/model/jinaAi/rerank/RerankRequest.java
@@ -16,4 +16,5 @@ class RerankRequest {
     private String model;
     private String query;
     private List<String> documents;
+    private Boolean returnDocuments;
 }