Merged PR 130: Added null pointer tracking for Rag
Added null pointer tracking for Rag
This commit is contained in:
@@ -290,7 +290,10 @@ public class AdvancedQueryRagSolver extends StepSolver {
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
docs = docs.stream().collect(Collectors.toMap(Document::getId, d -> d, (d1, d2) -> d1)).values().stream().collect(Collectors.toList());
|
docs = docs.stream()
|
||||||
|
.filter(d -> d.getMetadata().get("KsDocumentId") != null)
|
||||||
|
.filter(d -> d.getMetadata().get("KsDocumentIndex") != null)
|
||||||
|
.collect(Collectors.toMap(Document::getId, d -> d, (d1, d2) -> d1)).values().stream().collect(Collectors.toList());
|
||||||
|
|
||||||
//Sort by KsDocumentId and KsDocumentIndex
|
//Sort by KsDocumentId and KsDocumentIndex
|
||||||
docs.sort((d1, d2) -> {
|
docs.sort((d1, d2) -> {
|
||||||
|
|||||||
Reference in New Issue
Block a user