Added Gemini temperature

pull/215/head
M66B 1 year ago
parent 497ce2fda6
commit 4442619c73

@ -2790,8 +2790,9 @@ public class FragmentCompose extends FragmentBase {
SharedPreferences prefs = PreferenceManager.getDefaultSharedPreferences(context); SharedPreferences prefs = PreferenceManager.getDefaultSharedPreferences(context);
String model = prefs.getString("gemini_model", "gemini-pro"); String model = prefs.getString("gemini_model", "gemini-pro");
float temperature = prefs.getFloat("gemini_temperature", 0.5f);
return Gemini.generate(context, model, new String[]{Gemini.truncateParagraphs(body)}); return Gemini.generate(context, model, new String[]{Gemini.truncateParagraphs(body)}, temperature);
} }
@Override @Override

@ -106,9 +106,10 @@ public class FragmentDialogSummarize extends FragmentDialogBase {
return sb.toString(); return sb.toString();
} else if (Gemini.isAvailable(context)) { } else if (Gemini.isAvailable(context)) {
String model = prefs.getString("gemini_model", "gemini-pro"); 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 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); return TextUtils.join("\n", result);
} }

@ -90,6 +90,8 @@ public class FragmentOptionsIntegrations extends FragmentBase implements SharedP
private EditText etGemini; private EditText etGemini;
private TextInputLayout tilGemini; private TextInputLayout tilGemini;
private EditText etGeminiModel; private EditText etGeminiModel;
private TextView tvGeminiTemperature;
private SeekBar sbGeminiTemperature;
private EditText etGeminiSummarize; private EditText etGeminiSummarize;
private ImageButton ibGemini; private ImageButton ibGemini;
@ -106,7 +108,7 @@ public class FragmentOptionsIntegrations extends FragmentBase implements SharedP
"vt_enabled", "vt_apikey", "vt_enabled", "vt_apikey",
"send_enabled", "send_host", "send_dlimit", "send_tlimit", "send_enabled", "send_host", "send_dlimit", "send_tlimit",
"openai_enabled", "openai_uri", "openai_apikey", "openai_model", "openai_temperature", "openai_moderation", "openai_summarize", "openai_enabled", "openai_uri", "openai_apikey", "openai_model", "openai_temperature", "openai_moderation", "openai_summarize",
"gemini_enabled", "gemini_uri", "gemini_apikey", "gemini_model", "gemini_summarize" "gemini_enabled", "gemini_uri", "gemini_apikey", "gemini_model", "gemini_temperature", "gemini_summarize"
)); ));
@Override @Override
@ -162,6 +164,8 @@ public class FragmentOptionsIntegrations extends FragmentBase implements SharedP
etGemini = view.findViewById(R.id.etGemini); etGemini = view.findViewById(R.id.etGemini);
tilGemini = view.findViewById(R.id.tilGemini); tilGemini = view.findViewById(R.id.tilGemini);
etGeminiModel = view.findViewById(R.id.etGeminiModel); etGeminiModel = view.findViewById(R.id.etGeminiModel);
tvGeminiTemperature = view.findViewById(R.id.tvGeminiTemperature);
sbGeminiTemperature = view.findViewById(R.id.sbGeminiTemperature);
etGeminiSummarize = view.findViewById(R.id.etGeminiSummarize); etGeminiSummarize = view.findViewById(R.id.etGeminiSummarize);
ibGemini = view.findViewById(R.id.ibGemini); ibGemini = view.findViewById(R.id.ibGemini);
@ -556,6 +560,7 @@ public class FragmentOptionsIntegrations extends FragmentBase implements SharedP
public void onCheckedChanged(CompoundButton compoundButton, boolean checked) { public void onCheckedChanged(CompoundButton compoundButton, boolean checked) {
prefs.edit().putBoolean("gemini_enabled", checked).apply(); prefs.edit().putBoolean("gemini_enabled", checked).apply();
etGeminiModel.setEnabled(checked); etGeminiModel.setEnabled(checked);
sbGeminiTemperature.setEnabled(checked);
etGeminiSummarize.setEnabled(checked); etGeminiSummarize.setEnabled(checked);
if (checked) if (checked)
swOpenAi.setChecked(false); swOpenAi.setChecked(false);
@ -634,6 +639,24 @@ public class FragmentOptionsIntegrations extends FragmentBase implements SharedP
} }
}); });
sbGeminiTemperature.setOnSeekBarChangeListener(new SeekBar.OnSeekBarChangeListener() {
@Override
public void onProgressChanged(SeekBar seekBar, int progress, boolean fromUser) {
float temp = progress / 20f;
prefs.edit().putFloat("gemini_temperature", temp).apply();
}
@Override
public void onStartTrackingTouch(SeekBar seekBar) {
// Do nothing
}
@Override
public void onStopTrackingTouch(SeekBar seekBar) {
// Do nothing
}
});
etGeminiSummarize.setHint(Gemini.SUMMARY_PROMPT); etGeminiSummarize.setHint(Gemini.SUMMARY_PROMPT);
etGeminiSummarize.addTextChangedListener(new TextWatcher() { etGeminiSummarize.addTextChangedListener(new TextWatcher() {
@Override @Override
@ -780,6 +803,12 @@ public class FragmentOptionsIntegrations extends FragmentBase implements SharedP
tilGemini.getEditText().setText(prefs.getString("gemini_apikey", null)); tilGemini.getEditText().setText(prefs.getString("gemini_apikey", null));
etGeminiModel.setText(prefs.getString("gemini_model", null)); etGeminiModel.setText(prefs.getString("gemini_model", null));
etGeminiModel.setEnabled(swGemini.isChecked()); etGeminiModel.setEnabled(swGemini.isChecked());
temperature = prefs.getFloat("gemini_temperature", 0.5f);
tvGeminiTemperature.setText(getString(R.string.title_advanced_openai_temperature, NF.format(temperature)));
sbGeminiTemperature.setProgress(Math.round(temperature * 20));
sbGeminiTemperature.setEnabled(swGemini.isChecked());
etGeminiSummarize.setText(prefs.getString("gemini_summarize", null)); etGeminiSummarize.setText(prefs.getString("gemini_summarize", null));
etGeminiSummarize.setEnabled(swGemini.isChecked()); etGeminiSummarize.setEnabled(swGemini.isChecked());
} catch (Throwable ex) { } catch (Throwable ex) {

@ -40,6 +40,8 @@ import java.util.Objects;
public class Gemini { public class Gemini {
// https://ai.google.dev/models/gemini // https://ai.google.dev/models/gemini
static final String MODEL = "model";
static final String USER = "user";
static final String SUMMARY_PROMPT = "Summarize the following text:"; static final String SUMMARY_PROMPT = "Summarize the following text:";
private static final int MAX_GEMINI_LEN = 4000; // characters private static final int MAX_GEMINI_LEN = 4000; // characters
@ -57,7 +59,7 @@ public class Gemini {
(!TextUtils.isEmpty(apikey) || !Objects.equals(getUri(context), BuildConfig.GEMINI_ENDPOINT))); (!TextUtils.isEmpty(apikey) || !Objects.equals(getUri(context), BuildConfig.GEMINI_ENDPOINT)));
} }
static String[] generate(Context context, String model, String[] texts) throws JSONException, IOException { static String[] generate(Context context, String model, String[] texts, float temperature) throws JSONException, IOException {
JSONArray jpart = new JSONArray(); JSONArray jpart = new JSONArray();
for (String text : texts) { for (String text : texts) {
JSONObject jtext = new JSONObject(); JSONObject jtext = new JSONObject();
@ -67,10 +69,17 @@ public class Gemini {
JSONObject jcontent0 = new JSONObject(); JSONObject jcontent0 = new JSONObject();
jcontent0.put("parts", jpart); jcontent0.put("parts", jpart);
jcontent0.put("role", USER);
JSONArray jcontents = new JSONArray(); JSONArray jcontents = new JSONArray();
jcontents.put(jcontent0); jcontents.put(jcontent0);
// https://ai.google.dev/api/python/google/generativeai/GenerationConfig
JSONObject jconfig = new JSONObject();
jconfig.put("temperature", temperature);
JSONObject jrequest = new JSONObject(); JSONObject jrequest = new JSONObject();
jrequest.put("contents", jcontents); jrequest.put("contents", jcontents);
jrequest.put("generationConfig", jconfig);
String path = "models/" + Uri.encode(model) + ":generateContent"; String path = "models/" + Uri.encode(model) + ":generateContent";

@ -750,6 +750,31 @@
app:layout_constraintStart_toStartOf="parent" app:layout_constraintStart_toStartOf="parent"
app:layout_constraintTop_toBottomOf="@id/tvGeminiModel" /> app:layout_constraintTop_toBottomOf="@id/tvGeminiModel" />
<TextView
android:id="@+id/tvGeminiTemperature"
android:layout_width="0dp"
android:layout_height="wrap_content"
android:layout_marginStart="12dp"
android:layout_marginTop="12dp"
android:layout_marginEnd="48dp"
android:text="@string/title_advanced_openai_temperature"
android:textAppearance="@style/TextAppearance.AppCompat.Small"
app:layout_constraintEnd_toEndOf="parent"
app:layout_constraintStart_toStartOf="parent"
app:layout_constraintTop_toBottomOf="@id/etGeminiModel" />
<SeekBar
android:id="@+id/sbGeminiTemperature"
android:layout_width="0dp"
android:layout_height="wrap_content"
android:layout_marginStart="12dp"
android:max="20"
android:min="0"
android:progress="10"
app:layout_constraintEnd_toEndOf="parent"
app:layout_constraintStart_toStartOf="parent"
app:layout_constraintTop_toBottomOf="@id/tvGeminiTemperature" />
<TextView <TextView
android:id="@+id/tvGeminiSummarize" android:id="@+id/tvGeminiSummarize"
android:layout_width="wrap_content" android:layout_width="wrap_content"
@ -759,7 +784,7 @@
android:text="@string/title_advanced_summarize_prompt" android:text="@string/title_advanced_summarize_prompt"
android:textAppearance="@style/TextAppearance.AppCompat.Small" android:textAppearance="@style/TextAppearance.AppCompat.Small"
app:layout_constraintStart_toStartOf="parent" app:layout_constraintStart_toStartOf="parent"
app:layout_constraintTop_toBottomOf="@id/etGeminiModel" /> app:layout_constraintTop_toBottomOf="@id/sbGeminiTemperature" />
<EditText <EditText
android:id="@+id/etGeminiSummarize" android:id="@+id/etGeminiSummarize"

Loading…
Cancel
Save