in bigquery-antipattern-recognition/src/main/java/com/google/zetasql/toolkit/antipattern/rewriter/gemini/GeminiRewriter.java [115:131]
private static String getSqlFromRes(String res) {
JsonParser parser = new JsonParser();
String[] resList = res.split("data: ");
String optimizedSQL = "";
String currText = "";
for(String currRes: resList){
if(currRes.length()==0){continue;}
JsonObject object = (JsonObject) parser.parse(currRes);
currText = object.getAsJsonArray("candidates")
.get(0).getAsJsonObject().getAsJsonObject("content")
.getAsJsonArray("parts").get(0).getAsJsonObject().get("text")
.getAsString();
optimizedSQL += currText;
}
return optimizedSQL;
}