freestyle-sandboxes 0.0.35 → 0.0.37

package/dist/mastra/index.mjs

@@ -1,10 +1,28 @@
-import { createTool } from '@mastra/core';
 import { z } from 'zod';
-import { executeCodeDescription, executeCodeSchema } from '../ai/index.mjs';
+import { e as executeCodeDescription, a as executeCodeSchema } from '../index-DCF70Xbq.mjs';
 import { FreestyleSandboxes } from '../index.mjs';
-import 'ai';
 import '@hey-api/client-fetch';
 
+var Tool = class {
+  id;
+  description;
+  inputSchema;
+  outputSchema;
+  execute;
+  mastra;
+  constructor(opts) {
+    this.id = opts.id;
+    this.description = opts.description;
+    this.inputSchema = opts.inputSchema;
+    this.outputSchema = opts.outputSchema;
+    this.execute = opts.execute;
+    this.mastra = opts.mastra;
+  }
+};
+function createTool(opts) {
+  return new Tool(opts);
+}
+
 const executeTool = (config) => {
   const description = executeCodeDescription(
     Object.keys(config.envVars ?? {}).join(", "),

package/dist/utils/index.cjs

@@ -2,7 +2,7 @@
 
 var glob = require('glob');
 var fs = require('fs/promises');
-var fsSync = require('fs');
+var require$$0 = require('fs');
 var path = require('path');
 
 function _interopNamespaceDefault(e) {
@@ -23,7 +23,7 @@ function _interopNamespaceDefault(e) {
 }
 
 var fs__namespace = /*#__PURE__*/_interopNamespaceDefault(fs);
-var fsSync__namespace = /*#__PURE__*/_interopNamespaceDefault(fsSync);
+var require$$0__namespace = /*#__PURE__*/_interopNamespaceDefault(require$$0);
 var path__namespace = /*#__PURE__*/_interopNamespaceDefault(path);
 
 const prepareDirForDeployment = async (directory) => {
@@ -64,7 +64,7 @@ const prepareDirForDeploymentSync = (directory) => {
   for (const relativePath of patterns) {
     try {
       const filePath = path__namespace.join(directory, relativePath);
-      const content = fsSync__namespace.readFileSync(filePath, "base64");
+      const content = require$$0__namespace.readFileSync(filePath, "base64");
       files[relativePath] = {
         content,
         encoding: "base64"

package/dist/utils/index.mjs

@@ -1,6 +1,6 @@
 import { glob, globSync } from 'glob';
 import * as fs from 'fs/promises';
-import * as fsSync from 'fs';
+import * as require$$0 from 'fs';
 import * as path from 'path';
 
 const prepareDirForDeployment = async (directory) => {
@@ -41,7 +41,7 @@ const prepareDirForDeploymentSync = (directory) => {
   for (const relativePath of patterns) {
     try {
       const filePath = path.join(directory, relativePath);
-      const content = fsSync.readFileSync(filePath, "base64");
+      const content = require$$0.readFileSync(filePath, "base64");
       files[relativePath] = {
         content,
         encoding: "base64"

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "freestyle-sandboxes",
-  "version": "0.0.35",
+  "version": "0.0.37",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",
@@ -67,16 +67,16 @@
   "description": "",
   "devDependencies": {
     "@hey-api/openapi-ts": "^0.60.1",
-    "pkgroll": "^2.6.0"
-  },
-  "dependencies": {
-    "@hey-api/client-fetch": "^0.5.7",
+    "pkgroll": "^2.6.0",
     "@langchain/core": "^0.3.38",
     "@langchain/langgraph": "^0.2.44",
     "@mastra/core": "^0.6.0",
     "ai": "^4.0.25",
+    "humanlayer": "^0.7.0"
+  },
+  "dependencies": {
+    "@hey-api/client-fetch": "^0.5.7",
     "glob": "^11.0.1",
-    "humanlayer": "^0.7.0",
     "openai": "^4.77.3",
     "openapi": "^1.0.1",
     "zod": "^3.24.1"

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);