Merge pull request #3176 from M0ch0/staging Support Gemini-2.0-flash-exp

c4c9f43a60c867571805de0d4185224306d4a762

Cohee <18619528+Cohee1207@users.noreply.github.com>

Signed
Parents: 0090d685598f13393dbd3ce1eb79df3c86675c81 67cb89f634c19653a1ffcc52dbf47ac399b0e74c
5 files changed, +12 -3Showing whitespace changes
public/index.html+2 -1
@@ -3014,8 +3014,9 @@
30143014 <option value="chat-bison-001">PaLM 2 Chat (Legacy)</option>
30153015 </optgroup>
30163016 <optgroup label="Subversions">
30173017 <option value="gemini-exp2.0-1121flash-exp">Gemini Experimental2.0 2024-11-21Flash Experiment</option>
30183018 <option value="gemini-exp-1114">Gemini Experimental 2024-11-14</option>
3019+ <option value="gemini-exp-1121">Gemini Experimental 2024-11-21</option>
30193020 <option value="gemini-exp-1206">Gemini Experimental 2024-12-06</option>
30203021 <option value="gemini-1.5-pro-exp-0801">Gemini 1.5 Pro Experiment 2024-08-01</option>
30213022 <option value="gemini-1.5-pro-exp-0827">Gemini 1.5 Pro Experiment 2024-08-27</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',