From ea4922e8f0f9d2f36be0ea96eef743e3ec0f9c01 Mon Sep 17 00:00:00 2001
From: MaxRobinsonTheGreat <maxdrob1@gmail.com>
Date: Sat, 1 Jun 2024 16:05:59 -0500
Subject: [PATCH] fixed coder, gemini

---
 src/agent/coder.js   | 2 +-
 src/models/gemini.js | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/agent/coder.js b/src/agent/coder.js
index 47a7ae8..d408b6c 100644
--- a/src/agent/coder.js
+++ b/src/agent/coder.js
@@ -1,5 +1,5 @@
 import { writeFile, readFile, mkdirSync } from 'fs';
-
+import settings from '../../settings.js';
 
 export class Coder {
     constructor(agent) {
diff --git a/src/models/gemini.js b/src/models/gemini.js
index 7d652d5..ee4dfa4 100644
--- a/src/models/gemini.js
+++ b/src/models/gemini.js
@@ -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);