@saltcorn/large-language-model 0.2.0 → 0.2.1

package/generate.js

@@ -1,4 +1,4 @@
-const axios = require("axios");
+const fetch = require("node-fetch");
 const util = require("util");
 const exec = util.promisify(require("child_process").exec);
 const db = require("@saltcorn/data/db");
@@ -52,12 +52,10 @@ const getCompletionOpenAICompatible = async (
 ) => {
   const headers = {
     "Content-Type": "application/json",
+    Accept: "application/json",
   };
   if (bearer) headers.Authorization = "Bearer " + bearer;
-  const client = axios.create({
-    headers,
-  });
-  const params = {
+  const body = {
     //prompt: "How are you?",
     model,
     messages: [
@@ -69,9 +67,15 @@ const getCompletionOpenAICompatible = async (
     ],
     temperature: temperature || 0.7,
   };
+  const rawResponse = await fetch(chatCompleteEndpoint, {
+    method: "POST",
+    headers,
+    body: JSON.stringify(body),
+  });
+  const results = await rawResponse.json();
+  console.log(results);
 
-  const results = await client.post(chatCompleteEndpoint, params);
-  return results?.data?.choices?.[0]?.message?.content;
+  return results?.choices?.[0]?.message?.content;
 };
 
 module.exports = { getCompletion };

package/package.json

@@ -1,11 +1,11 @@
 {
   "name": "@saltcorn/large-language-model",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Large language models and functionality for Saltcorn",
   "main": "index.js",
   "dependencies": {
     "@saltcorn/data": "^0.9.0",
-    "axios": "0.16.2",
+    "node-fetch": "2.6.9",
     "underscore": "1.13.6"
   },
   "author": "Tom Nielsen",