Support Gemini 2.0 Flash-exp

a64c8ade9d53a610abc03d19d0585abc5462e500

M0cho <77959408+M0ch0@users.noreply.github.com>

Parents: d9e0df0884fe9cd1f96ef5077829ee16d7ba0ace
5 files changed, +11 -2Ignore whitespace
public/index.html+1 -0
@@ -3014,6 +3014,7 @@
30143014 <option value="chat-bison-001">PaLM 2 Chat (Legacy)</option>
30153015 </optgroup>
30163016 <optgroup label="Subversions">
3017+ <option value="gemini-2.0-flash-exp">Gemini 2.0 Flash Experiment</option>
30173018 <option value="gemini-exp-1121">Gemini Experimental 2024-11-21</option>
30183019 <option value="gemini-exp-1114">Gemini Experimental 2024-11-14</option>
30193020 <option value="gemini-exp-1206">Gemini Experimental 2024-12-06</option>
public/scripts/extensions/caption/settings.html+1 -0
@@ -53,6 +53,7 @@
5353 <option data-type="anthropic" value="claude-3-opus-20240229">claude-3-opus-20240229</option>
5454 <option data-type="anthropic" value="claude-3-sonnet-20240229">claude-3-sonnet-20240229</option>
5555 <option data-type="anthropic" value="claude-3-haiku-20240307">claude-3-haiku-20240307</option>
56+ <option data-type="google" value="gemini-2.0-flash-exp">gemini-2.0-flash-exp</option>
5657 <option data-type="google" value="gemini-1.5-flash">gemini-1.5-flash</option>
5758 <option data-type="google" value="gemini-1.5-flash-latest">gemini-1.5-flash-latest</option>
5859 <option data-type="google" value="gemini-1.5-flash-001">gemini-1.5-flash-001</option>
public/scripts/openai.js+1 -1
@@ -4081,7 +4081,7 @@ async function onModelChange() {
40814081 $('#openai_max_context').attr('max', max_32k);
40824082 } else if (value.includes('gemini-1.5-pro') || value.includes('gemini-exp-1206')) {
40834083 $('#openai_max_context').attr('max', max_2mil);
40844084 } else if (value.includes('gemini-1.5-flash') || value.includes('gemini-2.0-flash-exp')) {
40854085 $('#openai_max_context').attr('max', max_1mil);
40864086 } else if (value.includes('gemini-1.0-pro-vision') || value === 'gemini-pro-vision') {
40874087 $('#openai_max_context').attr('max', max_16k);
src/endpoints/backends/chat-completions.js+7 -1
@@ -280,7 +280,13 @@ async function sendMakerSuiteRequest(request, response) {
280280 delete generationConfig.stopSequences;
281281 }
282282
283- const should_use_system_prompt = (model.includes('gemini-1.5-flash') || model.includes('gemini-1.5-pro') || model.startsWith('gemini-exp')) && request.body.use_makersuite_sysprompt;
283+ const should_use_system_prompt = (
284+ model.includes('gemini-2.0-flash-exp') ||
285+ model.includes('gemini-1.5-flash') ||
286+ model.includes('gemini-1.5-pro') ||
287+ model.startsWith('gemini-exp')
288+ ) && request.body.use_makersuite_sysprompt;
289+
284290 const prompt = convertGooglePrompt(request.body.messages, model, should_use_system_prompt, request.body.char_name, request.body.user_name);
285291 let body = {
286292 contents: prompt.contents,
src/prompt-converters.js+1 -0
@@ -337,6 +337,7 @@ export function convertGooglePrompt(messages, model, useSysPrompt = false, charN
337337 const PNG_PIXEL = 'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mNkYAAAAAYAAjCB0C8AAAAASUVORK5CYII=';
338338
339339 const visionSupportedModels = [
340+ 'gemini-2.0-flash-exp',
340341 'gemini-1.5-flash',
341342 'gemini-1.5-flash-latest',
342343 'gemini-1.5-flash-001',