freestyle-sandboxes 0.0.13 → 0.0.14

package/dist/ai/index.cjs

@@ -5,6 +5,29 @@ var ai = require('ai');
 var zod = require('zod');
 require('@hey-api/client-fetch');
 
+const executeCodeSchema = zod.z.object({
+  script: zod.z.string().describe(`
+        The JavaScript or TypeScript script to execute, must be in the format of:
+
+        import { someModule } from "someModule";
+        export default () => {
+           ... your code here ...
+           return output;
+        }
+
+        or for async functions:
+
+        import { someModule } from "someModule";
+
+        export default async () => {
+            ... your code here ...
+            return output;
+        }
+        `)
+});
+const executeCodeDescription = (envVars, nodeModules) => `Execute a JavaScript or TypeScript script.
+${envVars.length > 0 ? `You can use the following environment variables: ${envVars}` : ""}
+${nodeModules.length > 0 ? `You can use the following node modules: ${nodeModules}` : "You cannot use any node modules."}`;
 const executeTool = (config) => {
   const api = new index.FreestyleSandboxes({
     ...config
@@ -12,29 +35,8 @@ const executeTool = (config) => {
   const envVars = Object.keys(config.envVars ?? {}).join(", ");
   const nodeModules = Object.keys(config.nodeModules ?? {}).join(", ");
   return ai.tool({
-    description: `Execute a JavaScript or TypeScript script.
-${envVars.length > 0 ? `You can use the following environment variables: ${envVars}` : ""}
-${nodeModules.length > 0 ? `You can use the following node modules: ${nodeModules}` : "You cannot use any node modules."}`,
-    parameters: zod.z.object({
-      script: zod.z.string().describe(`
-            The JavaScript or TypeScript script to execute, must be in the format of:
-            
-            import { someModule } from "someModule";
-            export default () => {
-               ... your code here ...
-               return output;
-            }
-
-            or for async functions:
-
-            import { someModule } from "someModule";
-
-            export default async () => {
-                ... your code here ...
-                return output;
-            }
-            `)
-    }),
+    description: executeCodeDescription(envVars, nodeModules),
+    parameters: executeCodeSchema,
     execute: async ({ script }) => {
       try {
         const res = await api.executeScript(script, config);
@@ -92,4 +94,6 @@ Error: ${e.message}`;
 };
 
 exports.deployWebTool = deployWebTool;
+exports.executeCodeDescription = executeCodeDescription;
+exports.executeCodeSchema = executeCodeSchema;
 exports.executeTool = executeTool;

package/dist/ai/index.d.cts

@@ -2,6 +2,14 @@ import * as ai from 'ai';
 import { F as FreestyleExecuteScriptParamsConfiguration, a as FreestyleExecureScriptResultSuccess, c as FreestyleDeployWebSuccessResponse } from '../types.gen-BtK6PMQy.js';
 import { z } from 'zod';
 
+declare const executeCodeSchema: z.ZodObject<{
+    script: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    script?: string;
+}, {
+    script?: string;
+}>;
+declare const executeCodeDescription: (envVars: string, nodeModules: string) => string;
 /**
  * Execute a JavaScript or TypeScript script
  *
@@ -42,4 +50,4 @@ declare const deployWebTool: (config: FreestyleExecuteScriptParamsConfiguration
     }, options: ai.ToolExecutionOptions) => PromiseLike<string | FreestyleDeployWebSuccessResponse>;
 };
 
-export { deployWebTool, executeTool };
+export { deployWebTool, executeCodeDescription, executeCodeSchema, executeTool };

package/dist/ai/index.d.mts

@@ -2,6 +2,14 @@ import * as ai from 'ai';
 import { F as FreestyleExecuteScriptParamsConfiguration, a as FreestyleExecureScriptResultSuccess, c as FreestyleDeployWebSuccessResponse } from '../types.gen-BtK6PMQy.js';
 import { z } from 'zod';
 
+declare const executeCodeSchema: z.ZodObject<{
+    script: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    script?: string;
+}, {
+    script?: string;
+}>;
+declare const executeCodeDescription: (envVars: string, nodeModules: string) => string;
 /**
  * Execute a JavaScript or TypeScript script
  *
@@ -42,4 +50,4 @@ declare const deployWebTool: (config: FreestyleExecuteScriptParamsConfiguration
     }, options: ai.ToolExecutionOptions) => PromiseLike<string | FreestyleDeployWebSuccessResponse>;
 };
 
-export { deployWebTool, executeTool };
+export { deployWebTool, executeCodeDescription, executeCodeSchema, executeTool };

package/dist/ai/index.mjs

@@ -3,6 +3,29 @@ import { tool } from 'ai';
 import { z } from 'zod';
 import '@hey-api/client-fetch';
 
+const executeCodeSchema = z.object({
+  script: z.string().describe(`
+        The JavaScript or TypeScript script to execute, must be in the format of:
+
+        import { someModule } from "someModule";
+        export default () => {
+           ... your code here ...
+           return output;
+        }
+
+        or for async functions:
+
+        import { someModule } from "someModule";
+
+        export default async () => {
+            ... your code here ...
+            return output;
+        }
+        `)
+});
+const executeCodeDescription = (envVars, nodeModules) => `Execute a JavaScript or TypeScript script.
+${envVars.length > 0 ? `You can use the following environment variables: ${envVars}` : ""}
+${nodeModules.length > 0 ? `You can use the following node modules: ${nodeModules}` : "You cannot use any node modules."}`;
 const executeTool = (config) => {
   const api = new FreestyleSandboxes({
     ...config
@@ -10,29 +33,8 @@ const executeTool = (config) => {
   const envVars = Object.keys(config.envVars ?? {}).join(", ");
   const nodeModules = Object.keys(config.nodeModules ?? {}).join(", ");
   return tool({
-    description: `Execute a JavaScript or TypeScript script.
-${envVars.length > 0 ? `You can use the following environment variables: ${envVars}` : ""}
-${nodeModules.length > 0 ? `You can use the following node modules: ${nodeModules}` : "You cannot use any node modules."}`,
-    parameters: z.object({
-      script: z.string().describe(`
-            The JavaScript or TypeScript script to execute, must be in the format of:
-            
-            import { someModule } from "someModule";
-            export default () => {
-               ... your code here ...
-               return output;
-            }
-
-            or for async functions:
-
-            import { someModule } from "someModule";
-
-            export default async () => {
-                ... your code here ...
-                return output;
-            }
-            `)
-    }),
+    description: executeCodeDescription(envVars, nodeModules),
+    parameters: executeCodeSchema,
     execute: async ({ script }) => {
       try {
         const res = await api.executeScript(script, config);
@@ -89,4 +91,4 @@ Error: ${e.message}`;
   });
 };
 
-export { deployWebTool, executeTool };
+export { deployWebTool, executeCodeDescription, executeCodeSchema, executeTool };

package/dist/mastra/index.cjs

@@ -0,0 +1,37 @@
+'use strict';
+
+var core = require('@mastra/core');
+var zod = require('zod');
+var index = require('../ai/index.cjs');
+var index$1 = require('../index.cjs');
+require('ai');
+require('@hey-api/client-fetch');
+
+const executeTool = (config) => {
+  const description = index.executeCodeDescription(
+    Object.keys(config.envVars ?? {}).join(", "),
+    Object.keys(config.nodeModules ?? {}).join(", ")
+  );
+  const client = new index$1.FreestyleSandboxes({
+    apiKey: config.apiKey
+  });
+  return core.createTool({
+    id: "Execute a TypeScript or JavaScript Script",
+    description,
+    inputSchema: index.executeCodeSchema,
+    execute: async ({ context: { script } }) => {
+      return await client.executeScript(script, config);
+    },
+    outputSchema: zod.z.object({
+      logs: zod.z.array(
+        zod.z.object({
+          message: zod.z.string(),
+          type: zod.z.string()
+        })
+      ),
+      result: zod.z.unknown()
+    })
+  });
+};
+
+exports.executeTool = executeTool;

package/dist/mastra/index.d.cts

@@ -0,0 +1,45 @@
+import * as _mastra_core from '@mastra/core';
+import { z } from 'zod';
+import { F as FreestyleExecuteScriptParamsConfiguration } from '../types.gen-BtK6PMQy.js';
+
+declare const executeTool: (config: FreestyleExecuteScriptParamsConfiguration & {
+    apiKey: string;
+}) => _mastra_core.Tool<"Execute a TypeScript or JavaScript Script", z.ZodObject<{
+    script: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    script?: string;
+}, {
+    script?: string;
+}>, z.ZodObject<{
+    logs: z.ZodArray<z.ZodObject<{
+        message: z.ZodString;
+        type: z.ZodString;
+    }, "strip", z.ZodTypeAny, {
+        message?: string;
+        type?: string;
+    }, {
+        message?: string;
+        type?: string;
+    }>, "many">;
+    result: z.ZodUnknown;
+}, "strip", z.ZodTypeAny, {
+    logs?: {
+        message?: string;
+        type?: string;
+    }[];
+    result?: unknown;
+}, {
+    logs?: {
+        message?: string;
+        type?: string;
+    }[];
+    result?: unknown;
+}>, _mastra_core.ToolExecutionContext<z.ZodObject<{
+    script: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    script?: string;
+}, {
+    script?: string;
+}>, _mastra_core.WorkflowContext<any>>>;
+
+export { executeTool };

package/dist/mastra/index.d.mts

@@ -0,0 +1,45 @@
+import * as _mastra_core from '@mastra/core';
+import { z } from 'zod';
+import { F as FreestyleExecuteScriptParamsConfiguration } from '../types.gen-BtK6PMQy.js';
+
+declare const executeTool: (config: FreestyleExecuteScriptParamsConfiguration & {
+    apiKey: string;
+}) => _mastra_core.Tool<"Execute a TypeScript or JavaScript Script", z.ZodObject<{
+    script: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    script?: string;
+}, {
+    script?: string;
+}>, z.ZodObject<{
+    logs: z.ZodArray<z.ZodObject<{
+        message: z.ZodString;
+        type: z.ZodString;
+    }, "strip", z.ZodTypeAny, {
+        message?: string;
+        type?: string;
+    }, {
+        message?: string;
+        type?: string;
+    }>, "many">;
+    result: z.ZodUnknown;
+}, "strip", z.ZodTypeAny, {
+    logs?: {
+        message?: string;
+        type?: string;
+    }[];
+    result?: unknown;
+}, {
+    logs?: {
+        message?: string;
+        type?: string;
+    }[];
+    result?: unknown;
+}>, _mastra_core.ToolExecutionContext<z.ZodObject<{
+    script: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    script?: string;
+}, {
+    script?: string;
+}>, _mastra_core.WorkflowContext<any>>>;
+
+export { executeTool };

package/dist/mastra/index.mjs

@@ -0,0 +1,35 @@
+import { createTool } from '@mastra/core';
+import { z } from 'zod';
+import { executeCodeDescription, executeCodeSchema } from '../ai/index.mjs';
+import { FreestyleSandboxes } from '../index.mjs';
+import 'ai';
+import '@hey-api/client-fetch';
+
+const executeTool = (config) => {
+  const description = executeCodeDescription(
+    Object.keys(config.envVars ?? {}).join(", "),
+    Object.keys(config.nodeModules ?? {}).join(", ")
+  );
+  const client = new FreestyleSandboxes({
+    apiKey: config.apiKey
+  });
+  return createTool({
+    id: "Execute a TypeScript or JavaScript Script",
+    description,
+    inputSchema: executeCodeSchema,
+    execute: async ({ context: { script } }) => {
+      return await client.executeScript(script, config);
+    },
+    outputSchema: z.object({
+      logs: z.array(
+        z.object({
+          message: z.string(),
+          type: z.string()
+        })
+      ),
+      result: z.unknown()
+    })
+  });
+};
+
+export { executeTool };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "freestyle-sandboxes",
-  "version": "0.0.13",
+  "version": "0.0.14",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",
@@ -25,6 +25,16 @@
         "types": "./dist/ai/index.d.mts",
         "default": "./dist/ai/index.mjs"
       }
+    },
+    "./mastra": {
+      "require": {
+        "types": "./dist/mastra/index.d.cts",
+        "default": "./dist/mastra/index.cjs"
+      },
+      "import": {
+        "types": "./dist/mastra/index.d.mts",
+        "default": "./dist/mastra/index.mjs"
+      }
     }
   },
   "scripts": {
@@ -41,6 +51,7 @@
   },
   "dependencies": {
     "@hey-api/client-fetch": "^0.5.7",
+    "@mastra/core": "^0.1.27-alpha.66",
     "ai": "^4.0.25",
     "humanlayer": "^0.7.0",
     "openai": "^4.77.3",

package/src/ai/index.ts

@@ -3,6 +3,37 @@ import { FreestyleSandboxes } from "..";
 import { tool } from "ai";
 import { z } from "zod";
 
+export const executeCodeSchema = z.object({
+  script: z.string().describe(`
+        The JavaScript or TypeScript script to execute, must be in the format of:
+
+        import { someModule } from "someModule";
+        export default () => {
+           ... your code here ...
+           return output;
+        }
+
+        or for async functions:
+
+        import { someModule } from "someModule";
+
+        export default async () => {
+            ... your code here ...
+            return output;
+        }
+        `),
+});
+
+export const executeCodeDescription = (envVars: string, nodeModules: string) =>
+  `Execute a JavaScript or TypeScript script.\n${
+    envVars.length > 0
+      ? `You can use the following environment variables: ${envVars}`
+      : ""
+  }\n${
+    nodeModules.length > 0
+      ? `You can use the following node modules: ${nodeModules}`
+      : "You cannot use any node modules."
+  }`;
 /**
  * Execute a JavaScript or TypeScript script
  *
@@ -23,35 +54,8 @@ export const executeTool = (
   const envVars = Object.keys(config.envVars ?? {}).join(", ");
   const nodeModules = Object.keys(config.nodeModules ?? {}).join(", ");
   return tool({
-    description: `Execute a JavaScript or TypeScript script.\n${
-      envVars.length > 0
-        ? `You can use the following environment variables: ${envVars}`
-        : ""
-    }\n${
-      nodeModules.length > 0
-        ? `You can use the following node modules: ${nodeModules}`
-        : "You cannot use any node modules."
-    }`,
-    parameters: z.object({
-      script: z.string().describe(`
-            The JavaScript or TypeScript script to execute, must be in the format of:
-            
-            import { someModule } from "someModule";
-            export default () => {
-               ... your code here ...
-               return output;
-            }
-
-            or for async functions:
-
-            import { someModule } from "someModule";
-
-            export default async () => {
-                ... your code here ...
-                return output;
-            }
-            `),
-    }),
+    description: executeCodeDescription(envVars, nodeModules),
+    parameters: executeCodeSchema,
     execute: async ({ script }) => {
       try {
         const res = await api.executeScript(script, config);

package/src/mastra/index.ts

@@ -0,0 +1,38 @@
+import { createTool } from "@mastra/core";
+import { z } from "zod";
+import { executeCodeDescription, executeCodeSchema } from "../ai";
+import { FreestyleExecuteScriptParamsConfiguration } from "../../openapi";
+import { FreestyleSandboxes } from "..";
+
+export const executeTool = (
+  config: FreestyleExecuteScriptParamsConfiguration & {
+    apiKey: string;
+  }
+) => {
+  const description = executeCodeDescription(
+    Object.keys(config.envVars ?? {}).join(", "),
+    Object.keys(config.nodeModules ?? {}).join(", ")
+  );
+
+  const client = new FreestyleSandboxes({
+    apiKey: config.apiKey,
+  });
+
+  return createTool({
+    id: "Execute a TypeScript or JavaScript Script",
+    description,
+    inputSchema: executeCodeSchema,
+    execute: async ({ context: { script } }) => {
+      return await client.executeScript(script, config);
+    },
+    outputSchema: z.object({
+      logs: z.array(
+        z.object({
+          message: z.string(),
+          type: z.string(),
+        })
+      ),
+      result: z.unknown(),
+    }),
+  });
+};