fixed coder, gemini

This commit is contained in:
MaxRobinsonTheGreat 2024-06-01 16:05:59 -05:00
parent 666b45a194
commit ea4922e8f0
2 changed files with 3 additions and 1 deletions

View file

@ -1,5 +1,5 @@
import { writeFile, readFile, mkdirSync } from 'fs';
import settings from '../../settings.js';
export class Coder {
constructor(agent) {

View file

@ -25,9 +25,11 @@ export class Gemini {
const stop_seq = '***';
const prompt = toSinglePrompt(turns, systemMessage, stop_seq, 'model');
console.log('Awaiting Google API response...');
const result = await model.generateContent(prompt);
const response = await result.response;
const text = response.text();
console.log('Received.');
if (!text.includes(stop_seq)) return text;
const idx = text.indexOf(stop_seq);
return text.slice(0, idx);