@saltcorn/large-language-model 0.7.6 → 0.7.8

package/generate.js

@@ -9,7 +9,7 @@ const {
 } = require("@google-cloud/aiplatform");
 const { google } = require("googleapis");
 const Plugin = require("@saltcorn/data/models/plugin");
-
+const path = require("path");
 const { features, getState } = require("@saltcorn/data/db/state");
 let ollamaMod;
 if (features.esm_plugins) ollamaMod = require("ollama");
@@ -104,17 +104,15 @@ const getCompletion = async (config, opts) => {
         opts
       );
     case "Local Ollama":
-      if (!ollamaMod) throw new Error("Not implemented for this backend");
-
-      const { Ollama } = ollamaMod;
-
-      const ollama = new Ollama();
-      const olres = await ollama.generate({
-        model: opts?.model || config.model,
-        ...opts,
-      });
-      //console.log("the response ", olres);
-      return olres.response;
+      return await getCompletionOpenAICompatible(
+        {
+          chatCompleteEndpoint: config.ollama_host
+            ? path.join(config.ollama_host, "v1/chat/completions")
+            : "http://localhost:11434/v1/chat/completions",
+          model: opts?.model || config.model,
+        },
+        opts
+      );
     case "Local llama.cpp":
       //TODO only check if unsafe plugins not allowed
       const isRoot = db.getTenantSchema() === db.connectObj.default_schema;

package/package.json

@@ -1,13 +1,13 @@
 {
   "name": "@saltcorn/large-language-model",
-  "version": "0.7.6",
+  "version": "0.7.8",
   "description": "Large language models and functionality for Saltcorn",
   "main": "index.js",
   "dependencies": {
     "@saltcorn/data": "^0.9.0",
     "node-fetch": "2.6.9",
     "underscore": "1.13.6",
-    "ollama": "0.5.0",
+    "ollama": "0.5.15",
     "@google-cloud/vertexai": "^1.9.3",
     "@google-cloud/aiplatform": "^3.34.0",
     "googleapis": "^144.0.0"