From c8b6504221e2108bf737e1e74ce07cbe9974f22d Mon Sep 17 00:00:00 2001
From: MaxRobinsonTheGreat <maxdrob1@gmail.com>
Date: Sat, 1 Jun 2024 16:23:14 -0500
Subject: [PATCH] ollama no embedding by default

---
 src/agent/prompter.js | 8 ++++++--
 src/models/local.js   | 1 -
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/agent/prompter.js b/src/agent/prompter.js
index 5e6a412..5eb5761 100644
--- a/src/agent/prompter.js
+++ b/src/agent/prompter.js
@@ -51,8 +51,12 @@ export class Prompter {
             throw new Error('Unknown API:', api);
 
         let embedding = this.prompts.embedding;
-        if (embedding === undefined)
-            embedding = {api: chat.api};
+        if (embedding === undefined) {
+            if (chat.api !== 'ollama')
+                embedding = {api: chat.api};
+            else
+                embedding = {api: 'none'};
+        }
         else if (typeof embedding === 'string' || embedding instanceof String)
             embedding = {api: embedding};
 
diff --git a/src/models/local.js b/src/models/local.js
index 21b70df..18d06e0 100644
--- a/src/models/local.js
+++ b/src/models/local.js
@@ -15,7 +15,6 @@ export class Local {
         let res = null;
         try {
             console.log(`Awaiting local response... (model: ${model})`)
-            console.log('Messages:', messages);
             res = await this.send(this.chat_endpoint, {model: model, messages: messages, stream: false});
             if (res)
                 res = res['message']['content'];