@saltcorn/large-language-model 0.8.6 → 0.8.8

package/generate.js

@@ -165,7 +165,16 @@ const getCompletion = async (config, opts) => {
 
 const getCompletionOpenAICompatible = async (
   { chatCompleteEndpoint, bearer, apiKey, model, responses_api, temperature },
-  { systemPrompt, prompt, debugResult, chat = [], api_key, endpoint, ...rest }
+  {
+    systemPrompt,
+    prompt,
+    debugResult,
+    debugCollector,
+    chat = [],
+    api_key,
+    endpoint,
+    ...rest
+  }
 ) => {
   const headers = {
     "Content-Type": "application/json",
@@ -281,6 +290,8 @@ const getCompletionOpenAICompatible = async (
         body
       )} to ${chatCompleteEndpoint} headers ${JSON.stringify(headers)}`
     );
+  if (debugCollector) debugCollector.request = body;
+
   const rawResponse = await fetch(chatCompleteEndpoint, {
     method: "POST",
     headers,
@@ -291,6 +302,8 @@ const getCompletionOpenAICompatible = async (
   if (debugResult)
     console.log("OpenAI response", JSON.stringify(results, null, 2));
   else getState().log(6, `OpenAI response ${JSON.stringify(results)}`);
+  if (debugCollector) debugCollector.response = results;
+
   if (results.error) throw new Error(`OpenAI error: ${results.error.message}`);
   if (responses_api) {
     const textOutput = results.output
@@ -341,6 +354,7 @@ const getImageGenOpenAICompatible = async (
     prompt,
     model,
     debugResult,
+    debugCollector,
     size,
     quality,
     n,
@@ -367,6 +381,7 @@ const getImageGenOpenAICompatible = async (
   if (response_format) body.response_format = response_format;
   if (n) body.n = n;
   if (debugResult) console.log("OpenAI image request", imageEndpoint, body);
+  if (debugCollector) debugCollector.request = body;
 
   const rawResponse = await fetch(imageEndpoint, {
     method: "POST",
@@ -374,6 +389,7 @@ const getImageGenOpenAICompatible = async (
     body: JSON.stringify(body),
   });
   const results = await rawResponse.json();
+  if (debugCollector) debugCollector.response = results;
   if (debugResult) console.log("OpenAI image response", results);
   if (results.error) throw new Error(`OpenAI error: ${results.error.message}`);
   return results?.data?.[0];

package/index.js

@@ -601,6 +601,13 @@ module.exports = {
           label: r.role,
         }));
         const commonFields = [
+          {
+            name: "filename",
+            label: "File name",
+            type: "String",
+            sublabel:
+              "Name of the generated file. Interpolations <code>{{ }}</code> available",
+          },
           {
             label: "Minimum role to access",
             name: "min_role",
@@ -683,6 +690,7 @@ module.exports = {
           answer_field,
           min_role,
           model,
+          filename,
         },
       }) => {
         let prompt;
@@ -696,6 +704,10 @@ module.exports = {
             "llm_generate prompt formula"
           );
         else prompt = row[prompt_field];
+
+        const use_filename = filename
+          ? interpolate(filename, row, user, "llm_generate_image file name")
+          : "generated.png";
         const opts = { debugResult: true }; // response_format: "b64_json" };
 
         if (model) opts.model = model;
@@ -712,7 +724,7 @@ module.exports = {
         } else if (ans.b64_json) {
           const imgContents = Buffer.from(ans.b64_json, "base64");
           const file = await File.from_contents(
-            "generated.png",
+            use_filename,
             "image/png",
             imgContents,
             user?.id,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@saltcorn/large-language-model",
-  "version": "0.8.6",
+  "version": "0.8.8",
   "description": "Large language models and functionality for Saltcorn",
   "main": "index.js",
   "dependencies": {