mirrors/mindcraft
1
0
Fork 0
mirror of https://github.com/kolbytn/mindcraft.git synced 2025-09-10 12:02:59 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update gemini.js

Browse source 
Fixed some logging
This commit is contained in:
Sweaterdog 2025-06-07 17:02:21 -07:00 • committed by GitHub
parent f6b276b3cf
commit 6ae7b82a53
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 17 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
src/models/gemini.js
View file

@ -80,7 +80,21 @@ export class Gemini {
if (typeof text === 'string') {
text = text.replace(/<thinking>/g, '<think>').replace(/<\/thinking>/g, '</think>');
}
log(JSON.stringify(originalTurnsForLog), text);
if (imageData) { // If imageData was part of this sendRequest call
let visionPromptText = ""; // Attempt to find the text prompt associated with the image
// `contents` is the array sent to the model
if (contents.length > 0) {
const lastUserTurnParts = contents[contents.length -1].parts;
if (Array.isArray(lastUserTurnParts)) {
const textPart = lastUserTurnParts.find(part => part.text);
if (textPart) visionPromptText = textPart.text;
}
}
logVision(originalTurnsForLog, imageData, text, visionPromptText);
} else {
log(JSON.stringify(originalTurnsForLog), text);
}
return text;
}
@ -102,7 +116,7 @@ export class Gemini {
const text = response.text();
console.log('Received.');
if (imageBuffer && text) {
logVision(turns, imageBuffer, text, prompt);
logVision([{role: 'system', content: systemMessage}, ...turns], imageBuffer, text, prompt);
}
if (!text.includes(stop_seq)) return text;
const idx = text.indexOf(stop_seq);
@ -118,6 +132,7 @@ export class Gemini {
if (typeof res === 'string') {
res = res.replace(/<thinking>/g, '<think>').replace(/<\/thinking>/g, '</think>');
}
// For error cases in vision, still use regular log since there's no image to save
log(JSON.stringify(loggedTurnsForError), res);
}
return res;