diff --git a/app/src/main/java/eu/faircode/email/FragmentCompose.java b/app/src/main/java/eu/faircode/email/FragmentCompose.java index 98e32c625f..688a294c0a 100644 --- a/app/src/main/java/eu/faircode/email/FragmentCompose.java +++ b/app/src/main/java/eu/faircode/email/FragmentCompose.java @@ -2685,7 +2685,8 @@ public class FragmentCompose extends FragmentBase { List result = new ArrayList<>(); if (inreplyto.size() > 0 && inreplyto.get(0).content) { - String role = (MessageHelper.equalEmail(draft.from, inreplyto.get(0).from) ? "assistant" : "user"); + String role = (MessageHelper.equalEmail(draft.from, inreplyto.get(0).from) + ? OpenAI.ASSISTANT : OpenAI.USER); Document parsed = JsoupEx.parse(inreplyto.get(0).getFile(context)); Document document = HtmlHelper.sanitizeView(context, parsed, false); Spanned spanned = HtmlHelper.fromDocument(context, document, null, null); @@ -2693,7 +2694,7 @@ public class FragmentCompose extends FragmentBase { } if (!TextUtils.isEmpty(body)) - result.add(new OpenAI.Message("assistant", OpenAI.truncateParagraphs(body))); + result.add(new OpenAI.Message(OpenAI.ASSISTANT, OpenAI.truncateParagraphs(body))); if (result.size() == 0) return null; diff --git a/app/src/main/java/eu/faircode/email/FragmentDialogSummarize.java b/app/src/main/java/eu/faircode/email/FragmentDialogSummarize.java index 566feee11b..257d85c657 100644 --- a/app/src/main/java/eu/faircode/email/FragmentDialogSummarize.java +++ b/app/src/main/java/eu/faircode/email/FragmentDialogSummarize.java @@ -118,8 +118,8 @@ public class FragmentDialogSummarize extends FragmentDialogBase { float temperature = prefs.getFloat("openai_temperature", 0.5f); List result = new ArrayList<>(); - result.add(new OpenAI.Message("assistant", OpenAI.SUMMARY_PROMPT)); - result.add(new OpenAI.Message("user", text)); + result.add(new OpenAI.Message(OpenAI.ASSISTANT, OpenAI.SUMMARY_PROMPT)); + result.add(new OpenAI.Message(OpenAI.USER, text)); OpenAI.Message[] completions = OpenAI.completeChat(context, model, result.toArray(new OpenAI.Message[0]), temperature, 1); StringBuilder sb = new StringBuilder(); diff --git a/app/src/main/java/eu/faircode/email/OpenAI.java b/app/src/main/java/eu/faircode/email/OpenAI.java index 282b18e4c8..26a0e5b905 100644 --- a/app/src/main/java/eu/faircode/email/OpenAI.java +++ b/app/src/main/java/eu/faircode/email/OpenAI.java @@ -43,6 +43,8 @@ import java.util.Objects; import javax.net.ssl.HttpsURLConnection; public class OpenAI { + static final String ASSISTANT = "assistant"; + static final String USER = "user"; static final String SUMMARY_PROMPT = "Summarize the following text:"; private static final int MAX_OPENAI_LEN = 1000; // characters