@cogitator-ai/self-modifying 1.0.0 → 2.0.0

package/src/tool-generation/tool-sandbox.ts

@@ -176,18 +176,16 @@ export class ToolSandbox {
   }
 
   private createContext(): SandboxContext {
-    const sandbox = this;
-
     return {
       console: {
         log: (...args: unknown[]) => {
-          sandbox.logs.push(`[LOG] ${args.map(String).join(' ')}`);
+          this.logs.push(`[LOG] ${args.map(String).join(' ')}`);
         },
         warn: (...args: unknown[]) => {
-          sandbox.logs.push(`[WARN] ${args.map(String).join(' ')}`);
+          this.logs.push(`[WARN] ${args.map(String).join(' ')}`);
         },
         error: (...args: unknown[]) => {
-          sandbox.logs.push(`[ERROR] ${args.map(String).join(' ')}`);
+          this.logs.push(`[ERROR] ${args.map(String).join(' ')}`);
         },
       },
       Math,
@@ -236,6 +234,7 @@ export class ToolSandbox {
       }, timeout);
 
       try {
+        // eslint-disable-next-line @typescript-eslint/no-implied-eval
         const factory = new Function(wrappedCode);
         const executor = factory();
 
@@ -244,13 +243,13 @@ export class ToolSandbox {
             clearTimeout(timer);
             resolve(result);
           })
-          .catch((error) => {
+          .catch((error: unknown) => {
             clearTimeout(timer);
-            reject(error);
+            reject(error instanceof Error ? error : new Error(String(error)));
           });
       } catch (error) {
         clearTimeout(timer);
-        reject(error);
+        reject(error instanceof Error ? error : new Error(String(error)));
       }
     });
   }
@@ -261,6 +260,7 @@ export class ToolSandbox {
     startTime: number
   ): Promise<ToolSandboxResult> {
     try {
+      // eslint-disable-next-line @typescript-eslint/no-implied-eval
       const factory = new Function(`
         "use strict";
         ${tool.implementation}

package/src/tool-generation/tool-validator.ts

@@ -56,9 +56,9 @@ const STATIC_VALIDATION_RULES: ValidationRule[] = [
     name: 'No prototype pollution',
     severity: 'error',
     check: (code) => {
-      if (/__proto__/.test(code)) return 'Uses __proto__';
+      if (code.includes('__proto__')) return 'Uses __proto__';
       if (/\.prototype\s*=/.test(code)) return 'Modifies prototype';
-      if (/Object\.setPrototypeOf/.test(code)) return 'Uses setPrototypeOf';
+      if (code.includes('Object.setPrototypeOf')) return 'Uses setPrototypeOf';
       return null;
     },
   },
@@ -67,10 +67,10 @@ const STATIC_VALIDATION_RULES: ValidationRule[] = [
     name: 'No obvious infinite loops',
     severity: 'warning',
     check: (code) => {
-      if (/while\s*\(\s*true\s*\)/.test(code) && !/break/.test(code)) {
+      if (/while\s*\(\s*true\s*\)/.test(code) && !code.includes('break')) {
         return 'Contains while(true) without break';
       }
-      if (/for\s*\(\s*;\s*;\s*\)/.test(code) && !/break/.test(code)) {
+      if (/for\s*\(\s*;\s*;\s*\)/.test(code) && !code.includes('break')) {
         return 'Contains for(;;) without break';
       }
       return null;
@@ -107,7 +107,7 @@ const STATIC_VALIDATION_RULES: ValidationRule[] = [
     name: 'No shell command execution',
     severity: 'error',
     check: (code) => {
-      if (/child_process/.test(code)) return 'Uses child_process';
+      if (code.includes('child_process')) return 'Uses child_process';
       if (/\bexec\s*\(/.test(code)) return 'Uses exec()';
       if (/\bspawn\s*\(/.test(code)) return 'Uses spawn()';
       if (/\bexecSync\s*\(/.test(code)) return 'Uses execSync()';