@saltcorn/agents 0.4.6 → 0.4.8

package/agent-view.js

@@ -93,7 +93,6 @@ const configuration_workflow = (req) =>
                 name: "stream",
                 label: "Stream response",
                 type: "Bool",
-                sublabel: "Requires dynamic update (Event settings)",
               },
               {
                 name: "placeholder",
@@ -159,6 +158,45 @@ const uploadForm = (viewname, req) =>
     span({ class: "ms-2 filename-label" })
   );
 
+const realTimeCollabScript = (viewname) => {
+  const view = View.findOne({ name: viewname });
+  return script(
+    domReady(`
+  const callback = () => {
+    const collabCfg = {
+      events: {
+        ['${view.getRealTimeEventName(
+          "STREAM_CHUNK"
+        )}' + \`?page_load_tag=\${_sc_pageloadtag}\`]: async (data) => {
+          $('form.agent-view div.next_response_scratch').append(
+            data.content
+          );
+        }
+      }
+    };
+    let retries = 0
+    function init_it() {
+      if(window.io) init_collab_room('${viewname}', collabCfg);
+      else setTimeout(init_it, retries * 100);
+      retries+=1;
+    }
+    init_it();
+  };
+
+  if (ensure_script_loaded.length >= 2) {
+    ensure_script_loaded("/static_assets/${
+      db.connectObj.version_tag
+    }/socket.io.min.js", callback);
+  }
+  else {
+    ensure_script_loaded("/static_assets/${
+    db.connectObj.version_tag
+      }/socket.io.min.js");
+    callback();
+  }`)
+  );
+};
+
 const run = async (
   table_id,
   viewname,
@@ -388,7 +426,8 @@ const run = async (
       skill_form_widgets,
       explainer && small({ class: "explainer" }, i(explainer))
     ),
-    stream && div({ class: "next_response_scratch" })
+    stream &&
+      realTimeCollabScript(viewname) + div({ class: "next_response_scratch" })
   );
 
   const prev_runs_side_bar = div(
@@ -692,7 +731,7 @@ const interact = async (table_id, viewname, config, body, { req, res }) => {
     action.name,
     [],
     triggering_row,
-    config.stream
+    config
   );
 };
 

package/common.js

@@ -1,6 +1,7 @@
 const { getState } = require("@saltcorn/data/db/state");
 const { div, span } = require("@saltcorn/markup/tags");
 const Trigger = require("@saltcorn/data/models/trigger");
+const View = require("@saltcorn/data/models/view");
 const { interpolate } = require("@saltcorn/data/utils");
 const db = require("@saltcorn/data/db");
 
@@ -163,11 +164,13 @@ const addToContext = async (run, newCtx) => {
 };
 
 const wrapSegment = (html, who) =>
-  '<div class="interaction-segment"><span class="badge bg-secondary">' +
-  who +
-  "</span>" +
-  html +
-  "</div>";
+  who === null
+    ? html
+    : '<div class="interaction-segment"><span class="badge bg-secondary">' +
+      who +
+      "</span>" +
+      html +
+      "</div>";
 
 const wrapCard = (title, ...inners) =>
   span({ class: "badge bg-info ms-1" }, title) +
@@ -202,8 +205,9 @@ const process_interaction = async (
   agent_label = "Copilot",
   prevResponses = [],
   triggering_row = {},
-  stream = false
+  agentsViewCfg = { stream: false }
 ) => {
+  const { stream, viewname } = agentsViewCfg;
   const sysState = getState();
   const complArgs = await getCompletionArguments(
     config,
@@ -217,21 +221,19 @@ const process_interaction = async (
   const debugMode = is_debug_mode(config, req.user);
   const debugCollector = {};
   if (debugMode) complArgs.debugCollector = debugCollector;
-  if (stream && sysState.getConfig("enable_dynamic_updates") && req.user) {
+  if (stream && viewname) {
+    const view = View.findOne({ name: viewname });
     complArgs.streamCallback = (response) => {
       const content =
-        response.choices[0].content || response.choices[0].delta?.content;
-      if (content)
-        sysState?.emitDynamicUpdate(
-          db.getTenantSchema(),
-          {
-            eval_js: `$('form.agent-view div.next_response_scratch').append(${JSON.stringify(
-              content
-            )})`,
-            page_load_tag: req.body.page_load_tag,
-          },
-          [req.user.id]
-        );
+        typeof response === "string"
+          ? response
+          : response.choices[0].content || response.choices[0].delta?.content;
+      if (content) {
+        const pageLoadTag = req.body.page_load_tag;
+        view.emitRealTimeEvent(`STREAM_CHUNK?page_load_tag=${pageLoadTag}`, {
+          content,
+        });
+      }
     };
   }
   const answer = await sysState.functions.llm_generate.run("", complArgs);
@@ -272,61 +274,77 @@ const process_interaction = async (
       }
     }
     if (answer.content && !answer.tool_calls)
-      responses.push(wrapSegment(md.render(answer.content), agent_label));
+      responses.push(
+        req.disable_markdown_render
+          ? answer
+          : wrapSegment(md.render(answer.content), agent_label)
+      );
   }
-  await addToContext(run, {
-    interactions:
-      typeof answer === "object" && answer.tool_calls
-        ? [
-            {
-              role: "assistant",
-              tool_calls: answer.tool_calls,
-              content: answer.content,
-            },
-          ]
-        : [{ role: "assistant", content: answer }],
-  });
+  if (answer.ai_sdk)
+    await addToContext(run, {
+      interactions: answer.messages,
+    });
+  else
+    await addToContext(run, {
+      interactions:
+        typeof answer === "object" && answer.tool_calls
+          ? [
+              {
+                role: "assistant",
+                tool_calls: answer.tool_calls,
+                content: answer.content,
+              },
+            ]
+          : [{ role: "assistant", content: answer }],
+    });
   if (
     answer &&
     typeof answer === "object" &&
     (answer.tool_calls || answer.mcp_calls)
   ) {
     if (answer.content)
-      responses.push(wrapSegment(md.render(answer.content), agent_label));
+      responses.push(
+        req.disable_markdown_render
+          ? answer
+          : wrapSegment(md.render(answer.content), agent_label)
+      );
     //const actions = [];
     let hasResult = false;
     if ((answer.mcp_calls || []).length && !answer.content) hasResult = true;
     for (const tool_call of answer.tool_calls || []) {
-      console.log("call function", tool_call.function?.name);
+      console.log(
+        "call function",
+        tool_call.toolName || tool_call.function?.name
+      );
 
       await addToContext(run, {
-        funcalls: { [tool_call.id]: tool_call.function },
+        funcalls: {
+          [tool_call.id || tool_call.toolCallId]: answer.ai_sdk
+            ? tool_call
+            : tool_call.function,
+        },
       });
 
-      const tool = find_tool(tool_call.function?.name, config);
+      const tool = find_tool(
+        tool_call.toolName || tool_call.function?.name,
+        config
+      );
 
       if (tool) {
-        if (
-          stream &&
-          sysState.getConfig("enable_dynamic_updates") &&
-          req.user
-        ) {
+        if (stream && viewname) {
           let content =
             "Using skill: " + tool.skill.skill_label ||
             tool.skill.constructor.skill_name;
-          sysState?.emitDynamicUpdate(
-            db.getTenantSchema(),
-            {
-              eval_js: `$('form.agent-view div.next_response_scratch').append(${JSON.stringify(
-                content
-              )})`,
-              page_load_tag: req.body.page_load_tag,
-            },
-            [req.user.id]
-          );
+          const view = View.findOne({ name: viewname });
+          const pageLoadTag = req.body.page_load_tag;
+          view.emitRealTimeEvent(`STREAM_CHUNK?page_load_tag=${pageLoadTag}`, {
+            content,
+          });
         }
         if (tool.tool.renderToolCall) {
-          const row = JSON.parse(tool_call.function.arguments);
+          const row = answer.ai_sdk
+            ? tool_call.input
+            : JSON.parse(tool_call.function.arguments);
           const rendered = await tool.tool.renderToolCall(row, {
             req,
           });
@@ -343,7 +361,9 @@ const process_interaction = async (
         }
         hasResult = true;
         const result = await tool.tool.process(
-          JSON.parse(tool_call.function.arguments),
+          answer.ai_sdk
+            ? tool_call.input
+            : JSON.parse(tool_call.function.arguments),
           { req }
         );
         if (
@@ -367,20 +387,46 @@ const process_interaction = async (
           }
           hasResult = true;
         }
-        await addToContext(run, {
-          interactions: [
-            {
-              role: "tool",
-              tool_call_id: tool_call.id,
-              call_id: tool_call.call_id,
-              name: tool_call.function.name,
-              content:
-                result && typeof result !== "string"
-                  ? JSON.stringify(result)
-                  : result || "Action run",
-            },
-          ],
-        });
+        if (answer.ai_sdk)
+          await addToContext(run, {
+            interactions: [
+              {
+                role: "tool",
+                content: [
+                  {
+                    type: "tool-result",
+                    toolCallId: tool_call.toolCallId,
+                    toolName: tool_call.toolName,
+                    output:
+                      !result || typeof result === "string"
+                        ? {
+                            type: "text",
+                            value: result || "Action run",
+                          }
+                        : {
+                            type: "json",
+                            value: JSON.parse(JSON.stringify(result)),
+                          },
+                  },
+                ],
+              },
+            ],
+          });
+        else
+          await addToContext(run, {
+            interactions: [
+              {
+                role: "tool",
+                tool_call_id: tool_call.toolCallId || tool_call.id,
+                call_id: tool_call.call_id,
+                name: tool_call.toolName || tool_call.function.name,
+                content:
+                  result && typeof result !== "string"
+                    ? JSON.stringify(result)
+                    : result || "Action run",
+              },
+            ],
+          });
       }
     }
     if (hasResult)
@@ -391,10 +437,14 @@ const process_interaction = async (
         agent_label,
         [...prevResponses, ...responses],
         triggering_row,
-        stream
+        agentsViewCfg
       );
   } else if (typeof answer === "string")
-    responses.push(wrapSegment(md.render(answer), agent_label));
+    responses.push(
+      req.disable_markdown_render
+        ? answer
+        : wrapSegment(md.render(answer), agent_label)
+    );
 
   return {
     json: {

package/index.js

@@ -1,7 +1,7 @@
 const Workflow = require("@saltcorn/data/models/workflow");
 const Form = require("@saltcorn/data/models/form");
 const FieldRepeat = require("@saltcorn/data/models/fieldrepeat");
-const { features } = require("@saltcorn/data/db/state");
+const Trigger = require("@saltcorn/data/models/trigger");
 const {
   get_skills,
   getCompletionArguments,
@@ -101,6 +101,40 @@ module.exports = {
       },
     },
   },
+  functions: {
+    agent_generate: {
+      run: async (agent_name, prompt, opts = {}) => {
+        const action = await Trigger.findOne({ name: agent_name });
+        const run = await WorkflowRun.create({
+          status: "Running",
+          started_by: opts.user?.id,
+          trigger_id: action.id,
+          context: {
+            implemented_fcall_ids: [],
+            interactions: [{ role: "user", content: prompt }],
+            funcalls: {},
+          },
+        });
+        const result = await process_interaction(
+          run,
+          action.configuration,
+          {
+            user: opts?.user,
+            body: {},
+            disable_markdown: opts?.disable_markdown_render
+          },
+          null
+        );
+        return result.json.response;
+      },
+      isAsync: true,
+      description: "Run an agent on a prompt",
+      arguments: [
+        { name: "agent_name", type: "String" },
+        { name: "prompt", type: "String" },
+      ],
+    },
+  },
 };
 
 /* 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@saltcorn/agents",
-  "version": "0.4.6",
+  "version": "0.4.8",
   "description": "AI agents for Saltcorn",
   "main": "index.js",
   "dependencies": {