aimyapi 1.0.7 → 1.0.9

package/dist/index.cjs

@@ -80,6 +80,8 @@ function wrap(vm, value, context, beginAsyncProcess, endAsyncProcess) {
     return wrapArray(vm, value, beginAsyncProcess, endAsyncProcess);
   } else if (value === null) {
     return vm.null;
+  } else if (value instanceof Date) {
+    return vm.evalCode(`new Date(${value.getTime()})`).value;
   } else if (typeof value === "object") {
     return wrapObject(vm, value, beginAsyncProcess, endAsyncProcess);
   } else if (typeof value === "string") {
@@ -438,11 +440,12 @@ const checkCodeSchema = zod.z.object({
 const submitCodeSchema = zod.z.object({
   code: zod.z.string()
 });
-const generateCode = async function(instance, queryText, userChatHistory, createTaskPrompt, debug = false, hideLogsFromAgent = false, model = "gpt-5-mini", additionalModelOptions) {
+const generateCode = async function(instance, queryText, userChatHistory, createTaskPrompt, debug = false, hideLogsFromAgent = false, model = "gpt-5-mini", additionalModelOptions, attempt = 0) {
   let finalSubmittedCode = "";
   if (debug) {
     console.log("Generating code for query:", queryText);
   }
+  const newHistoryItems = [];
   const codingTools = [
     {
       type: "function",
@@ -487,6 +490,14 @@ const generateCode = async function(instance, queryText, userChatHistory, create
         name: "submitCode",
         // @ts-ignore
         function: async ({ code }) => {
+          const checkRes = await instance.checkCode(code);
+          if (!checkRes.success) {
+            return {
+              success: false,
+              reason: "Code check failed",
+              logs: checkRes.logs
+            };
+          }
           finalSubmittedCode = code;
           return {
             success: true
@@ -509,7 +520,7 @@ const generateCode = async function(instance, queryText, userChatHistory, create
         ...userChatHistory,
         {
           role: "user",
-          content: createTaskPrompt.replace("{{USER_QUERY}}", queryText)
+          content: attempt == 0 ? createTaskPrompt.replace("{{USER_QUERY}}", queryText) : queryText
         }
       ],
       tools: codingTools,
@@ -518,6 +529,7 @@ const generateCode = async function(instance, queryText, userChatHistory, create
       if (debug) {
         console.log("Message:", JSON.stringify(message, null, 2));
       }
+      newHistoryItems.push(message);
     }).on("functionToolCall", (functionCall) => {
       if (debug) {
         console.log("Function Call:", JSON.stringify(functionCall, null, 2));
@@ -529,15 +541,37 @@ const generateCode = async function(instance, queryText, userChatHistory, create
     });
     const finalContent = await runner.finalContent();
     if (debug) {
-      console.log("Final Content:", finalContent);
+      console.log("Generate code final response:", finalContent);
+    }
+    if (!finalSubmittedCode) {
+      console.error("No code was submitted by the agent.");
+      if (attempt < 3) {
+        return generateCode(instance, `You must call submitCode with valid code before proceeding. Attempt ${attempt + 1} of 3.`, [
+          ...userChatHistory,
+          {
+            role: "user",
+            content: attempt == 0 ? createTaskPrompt.replace("{{USER_QUERY}}", queryText) : queryText
+          },
+          ...newHistoryItems
+        ], createTaskPrompt, debug, hideLogsFromAgent, model, additionalModelOptions, attempt + 1);
+      } else {
+        return {
+          success: false,
+          comments: "No code was submitted by the agent after multiple attempts. Final comment: " + finalContent
+        };
+      }
     }
     return {
+      success: true,
       code: finalSubmittedCode,
       comments: finalContent || void 0
     };
   } catch (err) {
     console.error(err);
-    return null;
+    return {
+      success: false,
+      comments: `Error generating code: ${err}`
+    };
   }
 };
 async function createWithAPI(options) {

package/dist/index.d.cts

@@ -40,12 +40,13 @@ interface AIMyAPIInstance {
     processRequest: (userQuery: string, context?: object) => Promise<GenerateCodeResult | null>;
 }
 interface GenerateCodeResult {
+    success: boolean;
     code?: string;
     comments?: string;
 }
 declare function zodParseJSON<T>(schema: ZodSchema<T>): (input: string) => T;
 declare function createBasePrompt(apiFilePath: string, apiGlobalName: string, documentationPath?: string): string;
-declare const generateCode: (instance: AIMyAPIInstance, queryText: string, userChatHistory: ChatCompletionMessageParam[], createTaskPrompt: string, debug?: boolean, hideLogsFromAgent?: boolean, model?: string, additionalModelOptions?: object) => Promise<GenerateCodeResult | null>;
+declare const generateCode: (instance: AIMyAPIInstance, queryText: string, userChatHistory: ChatCompletionMessageParam[], createTaskPrompt: string, debug?: boolean, hideLogsFromAgent?: boolean, model?: string, additionalModelOptions?: object, attempt?: number) => Promise<GenerateCodeResult | null>;
 interface AIMyAPIModuleExports {
     createWithAPI: (options: AIMyAPIOptions) => Promise<AIMyAPIInstance>;
     createBasePrompt: (apiFilePath: string, documentationPath: string) => string;

package/dist/index.d.mts

@@ -40,12 +40,13 @@ interface AIMyAPIInstance {
     processRequest: (userQuery: string, context?: object) => Promise<GenerateCodeResult | null>;
 }
 interface GenerateCodeResult {
+    success: boolean;
     code?: string;
     comments?: string;
 }
 declare function zodParseJSON<T>(schema: ZodSchema<T>): (input: string) => T;
 declare function createBasePrompt(apiFilePath: string, apiGlobalName: string, documentationPath?: string): string;
-declare const generateCode: (instance: AIMyAPIInstance, queryText: string, userChatHistory: ChatCompletionMessageParam[], createTaskPrompt: string, debug?: boolean, hideLogsFromAgent?: boolean, model?: string, additionalModelOptions?: object) => Promise<GenerateCodeResult | null>;
+declare const generateCode: (instance: AIMyAPIInstance, queryText: string, userChatHistory: ChatCompletionMessageParam[], createTaskPrompt: string, debug?: boolean, hideLogsFromAgent?: boolean, model?: string, additionalModelOptions?: object, attempt?: number) => Promise<GenerateCodeResult | null>;
 interface AIMyAPIModuleExports {
     createWithAPI: (options: AIMyAPIOptions) => Promise<AIMyAPIInstance>;
     createBasePrompt: (apiFilePath: string, documentationPath: string) => string;

package/dist/index.d.ts

@@ -40,12 +40,13 @@ interface AIMyAPIInstance {
     processRequest: (userQuery: string, context?: object) => Promise<GenerateCodeResult | null>;
 }
 interface GenerateCodeResult {
+    success: boolean;
     code?: string;
     comments?: string;
 }
 declare function zodParseJSON<T>(schema: ZodSchema<T>): (input: string) => T;
 declare function createBasePrompt(apiFilePath: string, apiGlobalName: string, documentationPath?: string): string;
-declare const generateCode: (instance: AIMyAPIInstance, queryText: string, userChatHistory: ChatCompletionMessageParam[], createTaskPrompt: string, debug?: boolean, hideLogsFromAgent?: boolean, model?: string, additionalModelOptions?: object) => Promise<GenerateCodeResult | null>;
+declare const generateCode: (instance: AIMyAPIInstance, queryText: string, userChatHistory: ChatCompletionMessageParam[], createTaskPrompt: string, debug?: boolean, hideLogsFromAgent?: boolean, model?: string, additionalModelOptions?: object, attempt?: number) => Promise<GenerateCodeResult | null>;
 interface AIMyAPIModuleExports {
     createWithAPI: (options: AIMyAPIOptions) => Promise<AIMyAPIInstance>;
     createBasePrompt: (apiFilePath: string, documentationPath: string) => string;

package/dist/index.js

@@ -80,6 +80,8 @@ function wrap(vm, value, context, beginAsyncProcess, endAsyncProcess) {
     return wrapArray(vm, value, beginAsyncProcess, endAsyncProcess);
   } else if (value === null) {
     return vm.null;
+  } else if (value instanceof Date) {
+    return vm.evalCode(`new Date(${value.getTime()})`).value;
   } else if (typeof value === "object") {
     return wrapObject(vm, value, beginAsyncProcess, endAsyncProcess);
   } else if (typeof value === "string") {
@@ -438,11 +440,12 @@ const checkCodeSchema = zod.z.object({
 const submitCodeSchema = zod.z.object({
   code: zod.z.string()
 });
-const generateCode = async function(instance, queryText, userChatHistory, createTaskPrompt, debug = false, hideLogsFromAgent = false, model = "gpt-5-mini", additionalModelOptions) {
+const generateCode = async function(instance, queryText, userChatHistory, createTaskPrompt, debug = false, hideLogsFromAgent = false, model = "gpt-5-mini", additionalModelOptions, attempt = 0) {
   let finalSubmittedCode = "";
   if (debug) {
     console.log("Generating code for query:", queryText);
   }
+  const newHistoryItems = [];
   const codingTools = [
     {
       type: "function",
@@ -487,6 +490,14 @@ const generateCode = async function(instance, queryText, userChatHistory, create
         name: "submitCode",
         // @ts-ignore
         function: async ({ code }) => {
+          const checkRes = await instance.checkCode(code);
+          if (!checkRes.success) {
+            return {
+              success: false,
+              reason: "Code check failed",
+              logs: checkRes.logs
+            };
+          }
           finalSubmittedCode = code;
           return {
             success: true
@@ -509,7 +520,7 @@ const generateCode = async function(instance, queryText, userChatHistory, create
         ...userChatHistory,
         {
           role: "user",
-          content: createTaskPrompt.replace("{{USER_QUERY}}", queryText)
+          content: attempt == 0 ? createTaskPrompt.replace("{{USER_QUERY}}", queryText) : queryText
         }
       ],
       tools: codingTools,
@@ -518,6 +529,7 @@ const generateCode = async function(instance, queryText, userChatHistory, create
       if (debug) {
         console.log("Message:", JSON.stringify(message, null, 2));
       }
+      newHistoryItems.push(message);
     }).on("functionToolCall", (functionCall) => {
       if (debug) {
         console.log("Function Call:", JSON.stringify(functionCall, null, 2));
@@ -529,15 +541,37 @@ const generateCode = async function(instance, queryText, userChatHistory, create
     });
     const finalContent = await runner.finalContent();
     if (debug) {
-      console.log("Final Content:", finalContent);
+      console.log("Generate code final response:", finalContent);
+    }
+    if (!finalSubmittedCode) {
+      console.error("No code was submitted by the agent.");
+      if (attempt < 3) {
+        return generateCode(instance, `You must call submitCode with valid code before proceeding. Attempt ${attempt + 1} of 3.`, [
+          ...userChatHistory,
+          {
+            role: "user",
+            content: attempt == 0 ? createTaskPrompt.replace("{{USER_QUERY}}", queryText) : queryText
+          },
+          ...newHistoryItems
+        ], createTaskPrompt, debug, hideLogsFromAgent, model, additionalModelOptions, attempt + 1);
+      } else {
+        return {
+          success: false,
+          comments: "No code was submitted by the agent after multiple attempts. Final comment: " + finalContent
+        };
+      }
     }
     return {
+      success: true,
       code: finalSubmittedCode,
       comments: finalContent || void 0
     };
   } catch (err) {
     console.error(err);
-    return null;
+    return {
+      success: false,
+      comments: `Error generating code: ${err}`
+    };
   }
 };
 async function createWithAPI(options) {

package/dist/index.mjs

@@ -58,6 +58,8 @@ function wrap(vm, value, context, beginAsyncProcess, endAsyncProcess) {
     return wrapArray(vm, value, beginAsyncProcess, endAsyncProcess);
   } else if (value === null) {
     return vm.null;
+  } else if (value instanceof Date) {
+    return vm.evalCode(`new Date(${value.getTime()})`).value;
   } else if (typeof value === "object") {
     return wrapObject(vm, value, beginAsyncProcess, endAsyncProcess);
   } else if (typeof value === "string") {
@@ -416,11 +418,12 @@ const checkCodeSchema = z.object({
 const submitCodeSchema = z.object({
   code: z.string()
 });
-const generateCode = async function(instance, queryText, userChatHistory, createTaskPrompt, debug = false, hideLogsFromAgent = false, model = "gpt-5-mini", additionalModelOptions) {
+const generateCode = async function(instance, queryText, userChatHistory, createTaskPrompt, debug = false, hideLogsFromAgent = false, model = "gpt-5-mini", additionalModelOptions, attempt = 0) {
   let finalSubmittedCode = "";
   if (debug) {
     console.log("Generating code for query:", queryText);
   }
+  const newHistoryItems = [];
   const codingTools = [
     {
       type: "function",
@@ -465,6 +468,14 @@ const generateCode = async function(instance, queryText, userChatHistory, create
         name: "submitCode",
         // @ts-ignore
         function: async ({ code }) => {
+          const checkRes = await instance.checkCode(code);
+          if (!checkRes.success) {
+            return {
+              success: false,
+              reason: "Code check failed",
+              logs: checkRes.logs
+            };
+          }
           finalSubmittedCode = code;
           return {
             success: true
@@ -487,7 +498,7 @@ const generateCode = async function(instance, queryText, userChatHistory, create
         ...userChatHistory,
         {
           role: "user",
-          content: createTaskPrompt.replace("{{USER_QUERY}}", queryText)
+          content: attempt == 0 ? createTaskPrompt.replace("{{USER_QUERY}}", queryText) : queryText
         }
       ],
       tools: codingTools,
@@ -496,6 +507,7 @@ const generateCode = async function(instance, queryText, userChatHistory, create
       if (debug) {
         console.log("Message:", JSON.stringify(message, null, 2));
       }
+      newHistoryItems.push(message);
     }).on("functionToolCall", (functionCall) => {
       if (debug) {
         console.log("Function Call:", JSON.stringify(functionCall, null, 2));
@@ -507,15 +519,37 @@ const generateCode = async function(instance, queryText, userChatHistory, create
     });
     const finalContent = await runner.finalContent();
     if (debug) {
-      console.log("Final Content:", finalContent);
+      console.log("Generate code final response:", finalContent);
+    }
+    if (!finalSubmittedCode) {
+      console.error("No code was submitted by the agent.");
+      if (attempt < 3) {
+        return generateCode(instance, `You must call submitCode with valid code before proceeding. Attempt ${attempt + 1} of 3.`, [
+          ...userChatHistory,
+          {
+            role: "user",
+            content: attempt == 0 ? createTaskPrompt.replace("{{USER_QUERY}}", queryText) : queryText
+          },
+          ...newHistoryItems
+        ], createTaskPrompt, debug, hideLogsFromAgent, model, additionalModelOptions, attempt + 1);
+      } else {
+        return {
+          success: false,
+          comments: "No code was submitted by the agent after multiple attempts. Final comment: " + finalContent
+        };
+      }
     }
     return {
+      success: true,
       code: finalSubmittedCode,
       comments: finalContent || void 0
     };
   } catch (err) {
     console.error(err);
-    return null;
+    return {
+      success: false,
+      comments: `Error generating code: ${err}`
+    };
   }
 };
 async function createWithAPI(options) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aimyapi",
-  "version": "1.0.7",
+  "version": "1.0.9",
   "description": "AI My API! Typescript sandbox for AI agents to interact with your API in a safe way.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/prompts/create-task-prompt.md

@@ -4,23 +4,30 @@ Most recent user query:
 
 ---
 
-Based on the above conversation, write a typescript script that performs the next operation requested by the user. 
+Based on the prior conversation, write a typescript script that performs the operation requested by the user and call the `submitCode` tool with the script. 
 
-Here's the typescript API to use for your script.
+## API documentation
+
+You can make use of this global API object:
 
 `api.ts`:
 ```typescript
 {{API}}
 ```
 
+## Instructions
+
 Use the global api object to make calls to the API.
+
 You may use top level `await`. 
 
-If needed, gather any information you need to complete the request. 
-You may use `runCode` to run any non-destructive API calls to understand what you need to do. Nothing will be returned unless you `console.log` it.
-In your code, `console.log` and `console.error` can be used to understand the execution or indicate that an error happened, but debug logs should be removed in the final version.
-You can use the `checkCode` tool to confirm that the code passes the typescript compilation and doesn't have errors
-You can call `submitCode` to submit the final version of the code.
+If needed, you may use `runCode` to run any non-destructive API calls to gather data for your script. Nothing will be visible to you unless you `console.log` it.
+
+You may use the `checkCode` tool to confirm that your code passes typescript compilation and doesn't have errors before submitting.
+
+In your code, `console.log` and `console.error` can be used to understand the execution or indicate that an error happened, but debug logs should be removed in the final version (errors can stay).
+
+When finished, you must call `submitCode` to submit the final version of the code, then respond with "Successfully submitted code" on success. On failure, correct any errors and submit again.
 
 ---
 

package/src/index.ts

@@ -11,6 +11,7 @@ import { QuickJSWASMModule, getQuickJS } from "quickjs-emscripten";
 import { AutoParseableTool } from 'openai/lib/parser.js';
 import { ZodSchema, z } from 'zod';
 import { zodToJsonSchema } from 'zod-to-json-schema';
+import { check } from 'zod/v4';
 
 const openai = new OpenAI( { baseURL: process.env.OPENAI_API_BASE_URL || undefined, apiKey: process.env.OPENAI_API_KEY || undefined } );
 
@@ -40,6 +41,7 @@ export interface AIMyAPIInstance {
 }
 
 export interface GenerateCodeResult {
+    success: boolean;
     code?: string;
     comments?: string;
 }
@@ -89,12 +91,15 @@ const submitCodeSchema = z.object({
     code: z.string(),
 });
 
-export const generateCode = async function(instance:AIMyAPIInstance, queryText:string, userChatHistory:ChatCompletionMessageParam[], createTaskPrompt:string, debug:boolean = false, hideLogsFromAgent:boolean = false, model="gpt-5-mini", additionalModelOptions?: object): Promise<GenerateCodeResult | null> {
+// Basic harness for writing code for the sandbox.
+export const generateCode = async function(instance:AIMyAPIInstance, queryText:string, userChatHistory:ChatCompletionMessageParam[], createTaskPrompt:string, debug:boolean = false, hideLogsFromAgent:boolean = false, model="gpt-5-mini", additionalModelOptions?: object, attempt:number = 0): Promise<GenerateCodeResult | null> {
     let finalSubmittedCode = "";
 
     if(debug) {
         console.log("Generating code for query:", queryText);
     }
+
+    const newHistoryItems = [];
     // checkcode
     const codingTools: AutoParseableTool<any, true>[] = [
         {
@@ -143,7 +148,16 @@ export const generateCode = async function(instance:AIMyAPIInstance, queryText:s
             function: {
                 name: "submitCode",
                 // @ts-ignore
-                function: async ({code}: {code: string}) => {   
+                function: async ({code}: {code: string}) => {
+                    const checkRes = await instance.checkCode(code);
+                    if(!checkRes.success) {
+                        return {
+                            success: false,
+                            reason: "Code check failed",
+                            logs: checkRes.logs,
+                        }
+                    }
+
                     finalSubmittedCode = code;
                     return {
                         success: true,
@@ -169,7 +183,7 @@ export const generateCode = async function(instance:AIMyAPIInstance, queryText:s
             ...userChatHistory,
             {
                 role: 'user',
-                content: createTaskPrompt.replace("{{USER_QUERY}}", queryText)
+                content: attempt == 0 ? createTaskPrompt.replace("{{USER_QUERY}}", queryText) : queryText,
             },
 
           ],
@@ -180,8 +194,9 @@ export const generateCode = async function(instance:AIMyAPIInstance, queryText:s
         .on('message', async (message: ChatCompletionMessageParam) => {
             if(debug) {
                 console.log("Message:", JSON.stringify(message, null, 2));
-
             }
+
+            newHistoryItems.push(message);
         })
         .on('functionToolCall', (functionCall: OpenAI.Chat.Completions.ChatCompletionMessageFunctionToolCall.Function) => {
             if(debug) {
@@ -199,17 +214,41 @@ export const generateCode = async function(instance:AIMyAPIInstance, queryText:s
       const finalContent = await runner.finalContent();
 
       if(debug) {
-        console.log("Final Content:", finalContent);
+        console.log("Generate code final response:", finalContent);
+      }
+
+      if(!finalSubmittedCode) {
+        console.error("No code was submitted by the agent.");
+        if(attempt < 3) {
+            return generateCode(instance, `You must call submitCode with valid code before proceeding. Attempt ${attempt + 1} of 3.`, [
+                ...userChatHistory, 
+                {
+                    role: 'user',
+                    content: attempt == 0 ? createTaskPrompt.replace("{{USER_QUERY}}", queryText) : queryText,
+                },
+                ...newHistoryItems
+            ], createTaskPrompt, debug, hideLogsFromAgent, model, additionalModelOptions, attempt + 1);
+        } else {
+            return {
+                success: false,
+                comments: "No code was submitted by the agent after multiple attempts. Final comment: " + finalContent,
+            }
+        }
+
       }
       
       return {
+        success: true,
         code: finalSubmittedCode,
         comments: finalContent || undefined,
       }
 
     } catch(err) {
         console.error(err);
-        return null;
+        return {
+            success: false,
+            comments: `Error generating code: ${err}`,
+        };
     }
 };
 

package/src/sandbox-wrappers.ts

@@ -61,6 +61,8 @@ export function wrap(vm, value: any, context: object,  beginAsyncProcess: () =>
   } else if (value === null) {
     //console.log("Wrapped null")
     return vm.null;
+  } else if (value instanceof Date) {
+    return vm.evalCode(`new Date(${value.getTime()})`).value;    
   } else if (typeof value === "object") {
     //console.log("Wrapped object")
     return wrapObject(vm, value, beginAsyncProcess, endAsyncProcess);