Added Gemini temperature

This commit is contained in:
M66B
2024-05-09 08:49:48 +02:00
parent 497ce2fda6
commit 4442619c73
5 changed files with 70 additions and 5 deletions

View File

@@ -106,9 +106,10 @@ public class FragmentDialogSummarize extends FragmentDialogBase {
return sb.toString();
} else if (Gemini.isAvailable(context)) {
String model = prefs.getString("gemini_model", "gemini-pro");
float temperature = prefs.getFloat("gemini_temperature", 0.5f);
String prompt = prefs.getString("gemini_summarize", Gemini.SUMMARY_PROMPT);
String[] result = Gemini.generate(context, model, new String[]{prompt, text});
String[] result = Gemini.generate(context, model, new String[]{prompt, text}, temperature);
return TextUtils.join("\n", result);
}