in sessions/next24/books-genai-vertex-springai/src/main/java/services/utility/PromptUtility.java [62:78]
public static Map<Integer, String> getSortedPagesBasedonPageNumber(List<Map<String, Object>> responseDoc) {
// Check for an empty topics list
Map<Integer, String> sortByPageNumber = new TreeMap<>(); // TreeMap to automatically sort by key
int i =0;
for(Map<String, Object> page: responseDoc) {
sortByPageNumber.put(Integer.parseInt(page.get("page_number").toString()), (String) page.get("page"));
i++;
//Get top 5 results
if(i==5) break;
}
String context = "";
for(String page: sortByPageNumber.values()) {
context += page + "\n";
}
return sortByPageNumber;
}