freestyle-sandboxes 0.0.36 → 0.0.37

package/dist/langgraph/index.cjs

@@ -2,7 +2,7 @@
 
 var index$1 = require('../index.cjs');
 var zod = require('zod');
-var index = require('../index-CEEa9WHp.cjs');
+var index = require('../index-BBXyg0JQ.cjs');
 require('@hey-api/client-fetch');
 
 var REGEX = /^(?:[0-9a-f]{8}-[0-9a-f]{4}-[1-8][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}|00000000-0000-0000-0000-000000000000|ffffffff-ffff-ffff-ffff-ffffffffffff)$/i;

package/dist/langgraph/index.mjs

@@ -1,6 +1,6 @@
 import { FreestyleSandboxes } from '../index.mjs';
 import { z } from 'zod';
-import { g as getDefaultExportFromCjs, z as zodToJsonSchema, a as executeCodeSchema, e as executeCodeDescription } from '../index-BQHqnjZK.mjs';
+import { g as getDefaultExportFromCjs, z as zodToJsonSchema, a as executeCodeSchema, e as executeCodeDescription } from '../index-DCF70Xbq.mjs';
 import '@hey-api/client-fetch';
 
 var REGEX = /^(?:[0-9a-f]{8}-[0-9a-f]{4}-[1-8][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}|00000000-0000-0000-0000-000000000000|ffffffff-ffff-ffff-ffff-ffffffffffff)$/i;

package/dist/mastra/index.cjs

@@ -1,7 +1,7 @@
 'use strict';
 
 var zod = require('zod');
-var index = require('../index-CEEa9WHp.cjs');
+var index = require('../index-BBXyg0JQ.cjs');
 var index$1 = require('../index.cjs');
 require('@hey-api/client-fetch');
 

package/dist/mastra/index.mjs

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { e as executeCodeDescription, a as executeCodeSchema } from '../index-BQHqnjZK.mjs';
+import { e as executeCodeDescription, a as executeCodeSchema } from '../index-DCF70Xbq.mjs';
 import { FreestyleSandboxes } from '../index.mjs';
 import '@hey-api/client-fetch';
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "freestyle-sandboxes",
-  "version": "0.0.36",
+  "version": "0.0.37",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",

package/src/ai/index.ts

@@ -1,7 +1,12 @@
-import { FreestyleExecuteScriptParamsConfiguration } from "../../openapi";
+import {
+  FreestyleExecuteScriptParamsConfiguration,
+  FreestyleExecuteScriptResultSuccess,
+  HandleExecuteScriptError,
+} from "../../openapi";
 import { FreestyleSandboxes } from "..";
 import { tool } from "ai";
 import { z } from "zod";
+import { P } from "../../dist/types.gen-BuhQ5LpB";
 
 export const executeCodeSchema = z.object({
   script: z.string().describe(`
@@ -40,11 +45,19 @@ export const executeCodeDescription = (envVars: string, nodeModules: string) =>
  *
  * @param config - Configuration for the tool
  * @param config.apiKey - The API key to use
- *
+ * @param {Function} [config.onResult] - Optional callback function to handle the result.
+ * @param {boolean} [config.truncateOutput=false] - Whether to truncate the result to 1000 characters and truncate individual logs to the first 250 characters; useful to prevent long outputs from filling the context window.
  */
 export const executeTool = (
   config: FreestyleExecuteScriptParamsConfiguration & {
     apiKey: string;
+    onResult?: (_v: {
+      input: {
+        script: string;
+      } & Record<string, unknown>;
+      result: FreestyleExecuteScriptResultSuccess | HandleExecuteScriptError;
+    }) => void | Promise<void>;
+    truncateOutput?: boolean;
   }
 ) => {
   const api = new FreestyleSandboxes({
@@ -56,9 +69,24 @@ export const executeTool = (
   return tool({
     description: executeCodeDescription(envVars, nodeModules),
     parameters: executeCodeSchema,
-    execute: async ({ script }) => {
+    execute: async ({ script, ...otherParams }) => {
       try {
         const res = await api.executeScript(script, config);
+        if (config.onResult) {
+          await config.onResult({
+            result: res,
+            input: {
+              script,
+              ...otherParams,
+            },
+          });
+        }
+        if (config.truncateOutput) {
+          if ("output" in res) {
+            res.result = JSON.stringify(res.result).slice(0, 1000);
+            res.logs = res.logs.slice(0, 1000);
+          }
+        }
         return res;
       } catch (e) {
         console.log("ERROR: ", e.message);