Fix Gemini captioning

df7c2226f6a50691d8a6d6670aa22820270f5ba3

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

Parents: 18cc17142f996f844a849d70c638df43bb25737b
3 files changed, +3 -2Showing whitespace changes
public/scripts/extensions/caption/settings.html+1 -0
@@ -65,6 +65,7 @@
6565 <option data-type="anthropic" value="claude-3-opus-20240229">claude-3-opus-20240229</option>
6666 <option data-type="anthropic" value="claude-3-sonnet-20240229">claude-3-sonnet-20240229</option>
6767 <option data-type="anthropic" value="claude-3-haiku-20240307">claude-3-haiku-20240307</option>
68+ <option data-type="google" value="gemini-2.5-pro-preview-03-25">gemini-2.5-pro-preview-03-25</option>
6869 <option data-type="google" value="gemini-2.5-pro-exp-03-25">gemini-2.5-pro-exp-03-25</option>
6970 <option data-type="google" value="gemini-2.0-pro-exp">gemini-2.0-pro-exp</option>
7071 <option data-type="google" value="gemini-2.0-pro-exp-02-05">gemini-2.0-pro-exp-02-05</option>
src/endpoints/google.js+1 -2
@@ -16,7 +16,7 @@ router.post('/caption-image', async (request, response) => {
1616 const base64Data = request.body.image.split(',')[1];
1717 const apiKey = request.body.reverse_proxy ? request.body.proxy_password : readSecret(request.user.directories, SECRET_KEYS.MAKERSUITE);
1818 const apiUrl = new URL(request.body.reverse_proxy || API_MAKERSUITE);
1919 const model = request.body.model || 'gemini-pro2.0-visionflash';
2020 const url = `${apiUrl.origin}/v1beta/models/${model}:generateContent?key=${apiKey}`;
2121 const body = {
2222 contents: [{
@@ -30,7 +30,6 @@ router.post('/caption-image', async (request, response) => {
3030 }],
3131 }],
3232 safetySettings: GEMINI_SAFETY,
33- generationConfig: { maxOutputTokens: 1000 },
3433 };
3534
3635 console.debug('Multimodal captioning request', model, body);
src/prompt-converters.js+1 -0
@@ -360,6 +360,7 @@ export function convertCohereMessages(messages, names) {
360360 */
361361export function convertGooglePrompt(messages, model, useSysPrompt, names) {
362362 const visionSupportedModels = [
363+ 'gemini-2.5-pro-preview-03-25',
363364 'gemini-2.5-pro-exp-03-25',
364365 'gemini-2.0-pro-exp',
365366 'gemini-2.0-pro-exp-02-05',