@cogitator-ai/self-modifying 17.0.17 → 18.0.0

package/dist/tool-generation/prompts.js

@@ -5,9 +5,27 @@ Rules:
 - Plain JavaScript only, no TypeScript
 - No require/import/eval/process/fetch
 - Must have an async function called "execute"
+- Always propagate errors with throw, never silently swallow them
 
 Example response:
-{"name":"add","description":"Add two numbers","implementation":"async function execute(params) { try { return params.a + params.b; } catch(e) { return 0; } }","parameters":{"type":"object","properties":{"a":{"type":"number"},"b":{"type":"number"}},"required":["a","b"]},"reasoning":"Simple addition"}`;
+{"name":"add","description":"Add two numbers","implementation":"async function execute(params) { if (typeof params.a !== 'number' || typeof params.b !== 'number') { throw new Error('Parameters a and b must be numbers'); } return params.a + params.b; }","parameters":{"type":"object","properties":{"a":{"type":"number"},"b":{"type":"number"}},"required":["a","b"]},"reasoning":"Simple addition"}`;
+const TOOL_NAME_REGEX = /^[a-z_][a-z0-9_]*$/;
+function sanitizeForPrompt(value) {
+    return value
+        .replace(/\\/g, '\\\\')
+        .replace(/"/g, '\\"')
+        .replace(/\n/g, '\\n')
+        .replace(/\r/g, '\\r')
+        .replace(/\t/g, '\\t');
+}
+function sanitizeToolName(name) {
+    const cleaned = name
+        .toLowerCase()
+        .replace(/[^a-z0-9_]/g, '_')
+        .replace(/^[0-9]/, '_$&')
+        .slice(0, 64);
+    return TOOL_NAME_REGEX.test(cleaned) ? cleaned : `generated_tool_${Date.now()}`;
+}
 export function buildGapAnalysisPrompt(userIntent, availableTools, failedAttempts) {
     const toolList = availableTools.map((t) => `- ${t.name}: ${t.description}`).join('\n');
     const failureContext = failedAttempts?.length
@@ -40,13 +58,39 @@ Respond with a JSON object:
   "alternativeApproach": "If gaps exist but can still proceed, explain how"
 }`;
 }
-export function buildToolGenerationPrompt(gap, _existingTools, _constraints) {
-    return `Create a tool named "${gap.suggestedToolName}".
-Task: ${gap.description}
-Formula/Logic: ${gap.requiredCapability}
-
+export function buildToolGenerationPrompt(gap, existingTools, constraints) {
+    const toolName = sanitizeToolName(gap.suggestedToolName);
+    const description = sanitizeForPrompt(gap.description);
+    const capability = sanitizeForPrompt(gap.requiredCapability);
+    const existingToolsSection = existingTools.length > 0
+        ? `\nEXISTING TOOLS (avoid duplication, consider composing with these):\n${existingTools.map((t) => `- ${t.name}: ${t.description}`).join('\n')}\n`
+        : '';
+    const constraintsSection = [];
+    if (constraints?.maxLines) {
+        constraintsSection.push(`- Maximum ${constraints.maxLines} lines of code`);
+    }
+    if (constraints?.securityLevel) {
+        constraintsSection.push(`- Security level: ${constraints.securityLevel}`);
+    }
+    if (constraints?.allowedModules && constraints.allowedModules.length > 0) {
+        constraintsSection.push(`- Only these modules may be referenced: ${constraints.allowedModules.join(', ')}`);
+    }
+    if (constraints?.securityLevel === 'strict') {
+        constraintsSection.push('- No eval, Function constructor, process, global, globalThis, fetch, setTimeout, setInterval', '- No bracket notation access to global objects', '- No prototype chain manipulation');
+    }
+    const constraintsText = constraintsSection.length > 0 ? `\nCONSTRAINTS:\n${constraintsSection.join('\n')}\n` : '';
+    const parametersJson = constraints?.parameters
+        ? JSON.stringify(constraints.parameters)
+        : '{"type":"object","properties":{/* DEFINE PARAMS */},"required":[/* REQUIRED PARAMS */]}';
+    const parametersText = constraints?.parameters
+        ? `\nThe tool MUST accept exactly this parameters JSON Schema: ${parametersJson}\n`
+        : '';
+    return `Create a tool named "${toolName}".
+Task: ${description}
+Formula/Logic: ${capability}
+${existingToolsSection}${constraintsText}${parametersText}
 Respond with ONLY this JSON (no other text):
-{"name":"${gap.suggestedToolName}","description":"${gap.description}","implementation":"async function execute(params) { try { /* YOUR CODE HERE */ } catch(e) { throw e; } }","parameters":{"type":"object","properties":{/* DEFINE PARAMS */},"required":[/* REQUIRED PARAMS */]},"reasoning":"explanation"}`;
+{"name":"${toolName}","description":"${description}","implementation":"async function execute(params) { /* YOUR CODE HERE — throw on invalid input */ }","parameters":${parametersJson},"reasoning":"explanation"}`;
 }
 export function buildToolValidationPrompt(tool, testCases) {
     return `Validate the following generated tool for correctness, safety, and edge cases.
@@ -56,7 +100,7 @@ Name: ${tool.name}
 Description: ${tool.description}
 
 IMPLEMENTATION:
-\`\`\`typescript
+\`\`\`javascript
 ${tool.implementation}
 \`\`\`
 
@@ -95,7 +139,7 @@ Name: ${tool.name}
 Description: ${tool.description}
 
 CURRENT IMPLEMENTATION:
-\`\`\`typescript
+\`\`\`javascript
 ${tool.implementation}
 \`\`\`
 
@@ -130,11 +174,13 @@ export function parseGapAnalysisResponse(response) {
                     id: String(g.id || `gap_${idx}`),
                     description: String(g.description || ''),
                     requiredCapability: String(g.requiredCapability || ''),
-                    suggestedToolName: String(g.suggestedToolName || `generated_tool_${idx}`),
+                    suggestedToolName: sanitizeToolName(String(g.suggestedToolName || `generated_tool_${idx}`)),
                     complexity: (['simple', 'moderate', 'complex'].includes(String(g.complexity))
                         ? g.complexity
                         : 'moderate'),
-                    confidence: typeof g.confidence === 'number' ? g.confidence : 0.5,
+                    confidence: typeof g.confidence === 'number' && Number.isFinite(g.confidence)
+                        ? Math.min(1, Math.max(0, g.confidence))
+                        : 0.5,
                     reasoning: String(g.reasoning || ''),
                 }))
                 : [],
@@ -159,30 +205,62 @@ export function parseToolGenerationResponse(response) {
         }
         catch { }
     }
-    const fnRegex = /(?:async\s+)?function\s+execute\s*\([^)]*\)\s*\{[\s\S]*?\n\}/;
+    const fnRegex = /(?:async\s+)?function\s+execute\s*\([^)]*\)\s*\{[\s\S]*\}/;
     const fnMatch = fnRegex.exec(response);
     if (fnMatch) {
+        let impl = fnMatch[0];
+        let braceDepth = 0;
+        let endIdx = impl.length;
+        const firstBrace = impl.indexOf('{');
+        for (let i = firstBrace; i < impl.length; i++) {
+            if (impl[i] === '{')
+                braceDepth++;
+            if (impl[i] === '}')
+                braceDepth--;
+            if (braceDepth === 0) {
+                endIdx = i + 1;
+                break;
+            }
+        }
+        impl = impl.slice(0, endIdx);
         const nameRegex = /["']?name["']?\s*[:=]\s*["']([^"']+)["']/;
         const nameMatch = nameRegex.exec(response);
         return buildToolFromParsed({
             name: nameMatch?.[1] ?? 'generated_tool',
             description: '',
-            implementation: fnMatch[0],
+            implementation: impl,
             parameters: { type: 'object', properties: {} },
         });
     }
     return null;
 }
+function normalizeParameters(value) {
+    if (!value || typeof value !== 'object' || Array.isArray(value)) {
+        return { type: 'object', properties: {} };
+    }
+    const schema = value;
+    const properties = schema.properties && typeof schema.properties === 'object' && !Array.isArray(schema.properties)
+        ? schema.properties
+        : {};
+    const required = Array.isArray(schema.required)
+        ? schema.required.filter((key) => typeof key === 'string' && Object.hasOwn(properties, key))
+        : [];
+    const normalized = { ...schema, type: 'object', properties };
+    if (required.length > 0) {
+        normalized.required = required;
+    }
+    else {
+        delete normalized.required;
+    }
+    return normalized;
+}
 function buildToolFromParsed(parsed) {
     return {
         id: `gen_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`,
-        name: String(parsed.name),
+        name: sanitizeToolName(String(parsed.name)),
         description: String(parsed.description || ''),
         implementation: String(parsed.implementation),
-        parameters: parsed.parameters || {
-            type: 'object',
-            properties: {},
-        },
+        parameters: normalizeParameters(parsed.parameters),
         createdAt: new Date(),
         version: 1,
         status: 'pending_validation',

package/dist/tool-generation/prompts.js.map

@@ -1 +1 @@
-{"version":3,"file":"prompts.js","sourceRoot":"","sources":["../../src/tool-generation/prompts.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAEvC,MAAM,CAAC,MAAM,6BAA6B,GAAG;;;;;;;;6SAQgQ,CAAC;AAE9S,MAAM,UAAU,sBAAsB,CACpC,UAAkB,EAClB,cAA4D,EAC5D,cAAyB;IAEzB,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEvF,MAAM,cAAc,GAAG,cAAc,EAAE,MAAM;QAC3C,CAAC,CAAC,kCAAkC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;QACpF,CAAC,CAAC,EAAE,CAAC;IAEP,OAAO;;;EAGP,UAAU;;;EAGV,QAAQ;EACR,cAAc;;;;;;;;;;;;;;;;;;EAkBd,CAAC;AACH,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,GAAkB,EAClB,cAA4D,EAC5D,YAIC;IAED,OAAO,wBAAwB,GAAG,CAAC,iBAAiB;QAC9C,GAAG,CAAC,WAAW;iBACN,GAAG,CAAC,kBAAkB;;;WAG5B,GAAG,CAAC,iBAAiB,oBAAoB,GAAG,CAAC,WAAW,4OAA4O,CAAC;AAChT,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,IAAmB,EACnB,SAA8D;IAE9D,OAAO;;;QAGD,IAAI,CAAC,IAAI;eACF,IAAI,CAAC,WAAW;;;;EAI7B,IAAI,CAAC,cAAc;;;;EAInB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;;;EAGxC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,KAAK,CAAC,gBAAgB,EAAE,CAAC,gBAAgB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;;;;;;;;;;;;EActH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,IAAmB,EACnB,gBAAsC,EACtC,SAAiB;IAEjB,MAAM,MAAM,GAAG;QACb,GAAG,gBAAgB,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,EAAE,CAAC;QAChE,GAAG,gBAAgB,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;QAC1D,GAAG,gBAAgB,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;KAC7D,CAAC;IAEF,OAAO;;aAEI,SAAS;;;QAGd,IAAI,CAAC,IAAI;eACF,IAAI,CAAC,WAAW;;;;EAI7B,IAAI,CAAC,cAAc;;;;EAInB,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;EAGrD,gBAAgB,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;;;;aAMhE,IAAI,CAAC,IAAI;;;;;EAKpB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,QAAgB;IAMvD,MAAM,IAAI,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;IACnC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACvD,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAChC,OAAO;YACL,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC;YAC9B,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC;gBAC9B,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAA0B,EAAE,GAAW,EAAE,EAAE,CAAC,CAAC;oBAC5D,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,OAAO,GAAG,EAAE,CAAC;oBAChC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC,WAAW,IAAI,EAAE,CAAC;oBACxC,kBAAkB,EAAE,MAAM,CAAC,CAAC,CAAC,kBAAkB,IAAI,EAAE,CAAC;oBACtD,iBAAiB,EAAE,MAAM,CAAC,CAAC,CAAC,iBAAiB,IAAI,kBAAkB,GAAG,EAAE,CAAC;oBACzE,UAAU,EAAE,CAAC,CAAC,QAAQ,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;wBAC3E,CAAC,CAAC,CAAC,CAAC,UAAU;wBACd,CAAC,CAAC,UAAU,CAAsC;oBACpD,UAAU,EAAE,OAAO,CAAC,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG;oBACjE,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,SAAS,IAAI,EAAE,CAAC;iBACrC,CAAC,CAAC;gBACL,CAAC,CAAC,EAAE;YACN,UAAU,EAAE,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC;YACtC,mBAAmB,EAAE,MAAM,CAAC,mBAAmB;gBAC7C,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC;gBACpC,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACvD,CAAC;AACH,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,QAAgB;IAC1D,MAAM,IAAI,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;IACnC,IAAI,IAAI,EAAE,CAAC;QACT,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAChC,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;gBACzC,OAAO,mBAAmB,CAAC,MAAM,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;IACZ,CAAC;IAED,MAAM,OAAO,GAAG,8DAA8D,CAAC;IAC/E,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACvC,IAAI,OAAO,EAAE,CAAC;QACZ,MAAM,SAAS,GAAG,0CAA0C,CAAC;QAC7D,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3C,OAAO,mBAAmB,CAAC;YACzB,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,IAAI,gBAAgB;YACxC,WAAW,EAAE,EAAE;YACf,cAAc,EAAE,OAAO,CAAC,CAAC,CAAC;YAC1B,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;SAC/C,CAAC,CAAC;IACL,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,mBAAmB,CAAC,MAA+B;IAC1D,OAAO;QACL,EAAE,EAAE,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;QACjE,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC;QACzB,WAAW,EAAE,MAAM,CAAC,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;QAC7C,cAAc,EAAE,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC;QAC7C,UAAU,EAAG,MAAM,CAAC,UAA0C,IAAI;YAChE,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;SACf;QACD,SAAS,EAAE,IAAI,IAAI,EAAE;QACrB,OAAO,EAAE,CAAC;QACV,MAAM,EAAE,oBAAoB;QAC5B,QAAQ,EAAE;YACR,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS;SACnE;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,QAAgB;IACtD,MAAM,IAAI,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;IACnC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEhC,OAAO;YACL,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC;YAChC,cAAc,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE;YAC7F,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE;YACpF,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE;YAC9E,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE;YACpF,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC;gBAC5C,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,EAA2B,EAAE,EAAE,CAAC,CAAC;oBACvD,KAAK,EAAE,EAAE,CAAC,KAAK;oBACf,MAAM,EAAE,EAAE,CAAC,MAAM;oBACjB,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,SAAS,IAAI,EAAE,CAAC,MAAM,CAAC;oBAC1C,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS;iBAC/C,CAAC,CAAC;gBACL,CAAC,CAAC,EAAE;YACN,YAAY,EAAE,OAAO,MAAM,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;SAChF,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC"}
+{"version":3,"file":"prompts.js","sourceRoot":"","sources":["../../src/tool-generation/prompts.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAEvC,MAAM,CAAC,MAAM,6BAA6B,GAAG;;;;;;;;;2YAS8V,CAAC;AAE5Y,MAAM,eAAe,GAAG,oBAAoB,CAAC;AAE7C,SAAS,iBAAiB,CAAC,KAAa;IACtC,OAAO,KAAK;SACT,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC;SACtB,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC;SACpB,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC;SACrB,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC;SACrB,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;AAC3B,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAY;IACpC,MAAM,OAAO,GAAG,IAAI;SACjB,WAAW,EAAE;SACb,OAAO,CAAC,aAAa,EAAE,GAAG,CAAC;SAC3B,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC;SACxB,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAChB,OAAO,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,kBAAkB,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;AAClF,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,UAAkB,EAClB,cAA4D,EAC5D,cAAyB;IAEzB,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEvF,MAAM,cAAc,GAAG,cAAc,EAAE,MAAM;QAC3C,CAAC,CAAC,kCAAkC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;QACpF,CAAC,CAAC,EAAE,CAAC;IAEP,OAAO;;;EAGP,UAAU;;;EAGV,QAAQ;EACR,cAAc;;;;;;;;;;;;;;;;;;EAkBd,CAAC;AACH,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,GAAkB,EAClB,aAA2D,EAC3D,WAKC;IAED,MAAM,QAAQ,GAAG,gBAAgB,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;IACzD,MAAM,WAAW,GAAG,iBAAiB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACvD,MAAM,UAAU,GAAG,iBAAiB,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;IAE7D,MAAM,oBAAoB,GACxB,aAAa,CAAC,MAAM,GAAG,CAAC;QACtB,CAAC,CAAC,yEAAyE,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI;QACnJ,CAAC,CAAC,EAAE,CAAC;IAET,MAAM,kBAAkB,GAAa,EAAE,CAAC;IACxC,IAAI,WAAW,EAAE,QAAQ,EAAE,CAAC;QAC1B,kBAAkB,CAAC,IAAI,CAAC,aAAa,WAAW,CAAC,QAAQ,gBAAgB,CAAC,CAAC;IAC7E,CAAC;IACD,IAAI,WAAW,EAAE,aAAa,EAAE,CAAC;QAC/B,kBAAkB,CAAC,IAAI,CAAC,qBAAqB,WAAW,CAAC,aAAa,EAAE,CAAC,CAAC;IAC5E,CAAC;IACD,IAAI,WAAW,EAAE,cAAc,IAAI,WAAW,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzE,kBAAkB,CAAC,IAAI,CACrB,2CAA2C,WAAW,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACnF,CAAC;IACJ,CAAC;IACD,IAAI,WAAW,EAAE,aAAa,KAAK,QAAQ,EAAE,CAAC;QAC5C,kBAAkB,CAAC,IAAI,CACrB,8FAA8F,EAC9F,gDAAgD,EAChD,mCAAmC,CACpC,CAAC;IACJ,CAAC;IAED,MAAM,eAAe,GACnB,kBAAkB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,mBAAmB,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;IAE5F,MAAM,cAAc,GAAG,WAAW,EAAE,UAAU;QAC5C,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,UAAU,CAAC;QACxC,CAAC,CAAC,yFAAyF,CAAC;IAC9F,MAAM,cAAc,GAAG,WAAW,EAAE,UAAU;QAC5C,CAAC,CAAC,+DAA+D,cAAc,IAAI;QACnF,CAAC,CAAC,EAAE,CAAC;IAEP,OAAO,wBAAwB,QAAQ;QACjC,WAAW;iBACF,UAAU;EACzB,oBAAoB,GAAG,eAAe,GAAG,cAAc;;WAE9C,QAAQ,oBAAoB,WAAW,sHAAsH,cAAc,6BAA6B,CAAC;AACpN,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,IAAmB,EACnB,SAA8D;IAE9D,OAAO;;;QAGD,IAAI,CAAC,IAAI;eACF,IAAI,CAAC,WAAW;;;;EAI7B,IAAI,CAAC,cAAc;;;;EAInB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;;;EAGxC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,KAAK,CAAC,gBAAgB,EAAE,CAAC,gBAAgB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;;;;;;;;;;;;EActH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,IAAmB,EACnB,gBAAsC,EACtC,SAAiB;IAEjB,MAAM,MAAM,GAAG;QACb,GAAG,gBAAgB,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,EAAE,CAAC;QAChE,GAAG,gBAAgB,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;QAC1D,GAAG,gBAAgB,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;KAC7D,CAAC;IAEF,OAAO;;aAEI,SAAS;;;QAGd,IAAI,CAAC,IAAI;eACF,IAAI,CAAC,WAAW;;;;EAI7B,IAAI,CAAC,cAAc;;;;EAInB,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;EAGrD,gBAAgB,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;;;;aAMhE,IAAI,CAAC,IAAI;;;;;EAKpB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,QAAgB;IAMvD,MAAM,IAAI,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;IACnC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACvD,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAChC,OAAO;YACL,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC;YAC9B,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC;gBAC9B,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAA0B,EAAE,GAAW,EAAE,EAAE,CAAC,CAAC;oBAC5D,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,OAAO,GAAG,EAAE,CAAC;oBAChC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC,WAAW,IAAI,EAAE,CAAC;oBACxC,kBAAkB,EAAE,MAAM,CAAC,CAAC,CAAC,kBAAkB,IAAI,EAAE,CAAC;oBACtD,iBAAiB,EAAE,gBAAgB,CACjC,MAAM,CAAC,CAAC,CAAC,iBAAiB,IAAI,kBAAkB,GAAG,EAAE,CAAC,CACvD;oBACD,UAAU,EAAE,CAAC,CAAC,QAAQ,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;wBAC3E,CAAC,CAAC,CAAC,CAAC,UAAU;wBACd,CAAC,CAAC,UAAU,CAAsC;oBACpD,UAAU,EACR,OAAO,CAAC,CAAC,UAAU,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC;wBAC/D,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,CAAC;wBACxC,CAAC,CAAC,GAAG;oBACT,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,SAAS,IAAI,EAAE,CAAC;iBACrC,CAAC,CAAC;gBACL,CAAC,CAAC,EAAE;YACN,UAAU,EAAE,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC;YACtC,mBAAmB,EAAE,MAAM,CAAC,mBAAmB;gBAC7C,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC;gBACpC,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACvD,CAAC;AACH,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,QAAgB;IAC1D,MAAM,IAAI,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;IACnC,IAAI,IAAI,EAAE,CAAC;QACT,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAChC,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;gBACzC,OAAO,mBAAmB,CAAC,MAAM,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;IACZ,CAAC;IAED,MAAM,OAAO,GAAG,2DAA2D,CAAC;IAC5E,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACvC,IAAI,OAAO,EAAE,CAAC;QACZ,IAAI,IAAI,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACtB,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QACzB,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACrC,KAAK,IAAI,CAAC,GAAG,UAAU,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9C,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,GAAG;gBAAE,UAAU,EAAE,CAAC;YAClC,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,GAAG;gBAAE,UAAU,EAAE,CAAC;YAClC,IAAI,UAAU,KAAK,CAAC,EAAE,CAAC;gBACrB,MAAM,GAAG,CAAC,GAAG,CAAC,CAAC;gBACf,MAAM;YACR,CAAC;QACH,CAAC;QACD,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;QAE7B,MAAM,SAAS,GAAG,0CAA0C,CAAC;QAC7D,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3C,OAAO,mBAAmB,CAAC;YACzB,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,IAAI,gBAAgB;YACxC,WAAW,EAAE,EAAE;YACf,cAAc,EAAE,IAAI;YACpB,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;SAC/C,CAAC,CAAC;IACL,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAc;IACzC,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAChE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;IAC5C,CAAC;IAED,MAAM,MAAM,GAAG,KAAgC,CAAC;IAChD,MAAM,UAAU,GACd,MAAM,CAAC,UAAU,IAAI,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC;QAC7F,CAAC,CAAE,MAAM,CAAC,UAAsC;QAChD,CAAC,CAAC,EAAE,CAAC;IACT,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC;QAC7C,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CACpB,CAAC,GAAG,EAAiB,EAAE,CAAC,OAAO,GAAG,KAAK,QAAQ,IAAI,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,GAAG,CAAC,CAClF;QACH,CAAC,CAAC,EAAE,CAAC;IAEP,MAAM,UAAU,GAA4B,EAAE,GAAG,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC;IACtF,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,UAAU,CAAC,QAAQ,GAAG,QAAQ,CAAC;IACjC,CAAC;SAAM,CAAC;QACN,OAAO,UAAU,CAAC,QAAQ,CAAC;IAC7B,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,mBAAmB,CAAC,MAA+B;IAC1D,OAAO;QACL,EAAE,EAAE,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;QACjE,IAAI,EAAE,gBAAgB,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC3C,WAAW,EAAE,MAAM,CAAC,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC;QAC7C,cAAc,EAAE,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC;QAC7C,UAAU,EAAE,mBAAmB,CAAC,MAAM,CAAC,UAAU,CAAC;QAClD,SAAS,EAAE,IAAI,IAAI,EAAE;QACrB,OAAO,EAAE,CAAC;QACV,MAAM,EAAE,oBAAoB;QAC5B,QAAQ,EAAE;YACR,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS;SACnE;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,QAAgB;IACtD,MAAM,IAAI,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;IACnC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEhC,OAAO;YACL,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC;YAChC,cAAc,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE;YAC7F,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE;YACpF,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE;YAC9E,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE;YACpF,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC;gBAC5C,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,EAA2B,EAAE,EAAE,CAAC,CAAC;oBACvD,KAAK,EAAE,EAAE,CAAC,KAAK;oBACf,MAAM,EAAE,EAAE,CAAC,MAAM;oBACjB,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,SAAS,IAAI,EAAE,CAAC,MAAM,CAAC;oBAC1C,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS;iBAC/C,CAAC,CAAC;gBACL,CAAC,CAAC,EAAE;YACN,YAAY,EAAE,OAAO,MAAM,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;SAChF,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/tool-generation/tool-generator.d.ts

@@ -1,9 +1,13 @@
 import type { Tool, LLMBackend, CapabilityGap, GeneratedTool, ToolValidationResult, ToolSelfGenerationConfig } from '@cogitator-ai/types';
+import { type SandboxTestCase } from './tool-sandbox.js';
 export interface ToolGeneratorOptions {
     llm: LLMBackend;
     config: ToolSelfGenerationConfig;
     model?: string;
 }
+export interface GenerateOptions {
+    parameters?: Record<string, unknown>;
+}
 export interface GenerationResult {
     tool: GeneratedTool | null;
     validationResult: ToolValidationResult | null;
@@ -18,14 +22,10 @@ export declare class ToolGenerator {
     private readonly validator;
     private readonly sandbox;
     constructor(options: ToolGeneratorOptions);
-    generate(gap: CapabilityGap, existingTools: Tool[], testCases?: Array<{
-        input: unknown;
-        expectedOutput?: unknown;
-    }>): Promise<GenerationResult>;
-    generateQuick(description: string, name: string, _parameters: Record<string, unknown>): Promise<GeneratedTool | null>;
+    generate(gap: CapabilityGap, existingTools: Tool[], testCases?: SandboxTestCase[], options?: GenerateOptions): Promise<GenerationResult>;
+    generateQuick(description: string, name: string, parameters: Record<string, unknown>): Promise<GeneratedTool | null>;
     private generateInitial;
     private improve;
     createExecutableTool(generated: GeneratedTool): Tool;
-    private callLLM;
 }
 //# sourceMappingURL=tool-generator.d.ts.map

package/dist/tool-generation/tool-generator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tool-generator.d.ts","sourceRoot":"","sources":["../../src/tool-generation/tool-generator.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,IAAI,EACJ,UAAU,EACV,aAAa,EACb,aAAa,EACb,oBAAoB,EACpB,wBAAwB,EACzB,MAAM,qBAAqB,CAAC;AAW7B,MAAM,WAAW,oBAAoB;IACnC,GAAG,EAAE,UAAU,CAAC;IAChB,MAAM,EAAE,wBAAwB,CAAC;IACjC,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,aAAa,GAAG,IAAI,CAAC;IAC3B,gBAAgB,EAAE,oBAAoB,GAAG,IAAI,CAAC;IAC9C,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAa;IACjC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA2B;IAClD,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAgB;IAC1C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAc;gBAE1B,OAAO,EAAE,oBAAoB;IAYnC,QAAQ,CACZ,GAAG,EAAE,aAAa,EAClB,aAAa,EAAE,IAAI,EAAE,EACrB,SAAS,CAAC,EAAE,KAAK,CAAC;QAAE,KAAK,EAAE,OAAO,CAAC;QAAC,cAAc,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC,GAC9D,OAAO,CAAC,gBAAgB,CAAC;IAgEtB,aAAa,CACjB,WAAW,EAAE,MAAM,EACnB,IAAI,EAAE,MAAM,EACZ,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACnC,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC;YAelB,eAAe;YAoCf,OAAO;IA+BrB,oBAAoB,CAAC,SAAS,EAAE,aAAa,GAAG,IAAI;YA4BtC,OAAO;CAStB"}
+{"version":3,"file":"tool-generator.d.ts","sourceRoot":"","sources":["../../src/tool-generation/tool-generator.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,IAAI,EAEJ,UAAU,EACV,aAAa,EACb,aAAa,EACb,oBAAoB,EACpB,wBAAwB,EACzB,MAAM,qBAAqB,CAAC;AAG7B,OAAO,EAAe,KAAK,eAAe,EAAE,MAAM,gBAAgB,CAAC;AASnE,MAAM,WAAW,oBAAoB;IACnC,GAAG,EAAE,UAAU,CAAC;IAChB,MAAM,EAAE,wBAAwB,CAAC;IACjC,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,eAAe;IAC9B,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,aAAa,GAAG,IAAI,CAAC;IAC3B,gBAAgB,EAAE,oBAAoB,GAAG,IAAI,CAAC;IAC9C,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAa;IACjC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA2B;IAClD,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAgB;IAC1C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAc;gBAE1B,OAAO,EAAE,oBAAoB;IAYnC,QAAQ,CACZ,GAAG,EAAE,aAAa,EAClB,aAAa,EAAE,IAAI,EAAE,EACrB,SAAS,CAAC,EAAE,eAAe,EAAE,EAC7B,OAAO,GAAE,eAAoB,GAC5B,OAAO,CAAC,gBAAgB,CAAC;IAwEtB,aAAa,CACjB,WAAW,EAAE,MAAM,EACnB,IAAI,EAAE,MAAM,EACZ,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAClC,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC;YAuBlB,eAAe;YAuCf,OAAO;IAkCrB,oBAAoB,CAAC,SAAS,EAAE,aAAa,GAAG,IAAI;CAiCrD"}

package/dist/tool-generation/tool-generator.js

@@ -1,6 +1,7 @@
 import { z } from 'zod';
 import { ToolValidator } from './tool-validator.js';
 import { ToolSandbox } from './tool-sandbox.js';
+import { llmChat } from '../utils/llm-helper.js';
 import { TOOL_GENERATION_SYSTEM_PROMPT, buildToolGenerationPrompt, buildToolImprovementPrompt, parseToolGenerationResponse, } from './prompts.js';
 export class ToolGenerator {
     llm;
@@ -19,7 +20,7 @@ export class ToolGenerator {
         });
         this.sandbox = new ToolSandbox(options.config.sandboxConfig);
     }
-    async generate(gap, existingTools, testCases) {
+    async generate(gap, existingTools, testCases, options = {}) {
         let currentTool = null;
         let validationResult = null;
         let iterations = 0;
@@ -28,14 +29,21 @@ export class ToolGenerator {
             iterations++;
             try {
                 if (currentTool === null) {
-                    currentTool = await this.generateInitial(gap, existingTools);
+                    currentTool = await this.generateInitial(gap, existingTools, options.parameters);
                 }
                 else if (validationResult) {
-                    currentTool = await this.improve(currentTool, validationResult, iterations);
+                    const improved = await this.improve(currentTool, validationResult, iterations);
+                    if (!improved) {
+                        continue;
+                    }
+                    currentTool = improved;
                 }
                 if (!currentTool) {
                     continue;
                 }
+                if (options.parameters) {
+                    currentTool.parameters = options.parameters;
+                }
                 validationResult = await this.validator.validate(currentTool, testCases);
                 if (validationResult.isValid) {
                     currentTool.status = 'validated';
@@ -75,7 +83,7 @@ export class ToolGenerator {
             error: `Failed to generate valid tool after ${maxIterations} iterations`,
         };
     }
-    async generateQuick(description, name, _parameters) {
+    async generateQuick(description, name, parameters) {
         const gap = {
             id: `quick_${Date.now()}`,
             description,
@@ -85,10 +93,12 @@ export class ToolGenerator {
             confidence: 1,
             reasoning: 'User-requested quick generation',
         };
-        const result = await this.generate(gap, []);
-        return result.tool;
+        const result = await this.generate(gap, [], undefined, Object.keys(parameters).length > 0
+            ? { parameters: { type: 'object', properties: parameters } }
+            : {});
+        return result.success ? result.tool : null;
     }
-    async generateInitial(gap, existingTools) {
+    async generateInitial(gap, existingTools, parameters) {
         const toolSummaries = existingTools.map((t) => ({
             name: t.name,
             description: t.description,
@@ -97,12 +107,13 @@ export class ToolGenerator {
             maxLines: 100,
             securityLevel: 'strict',
             allowedModules: this.config.sandboxConfig?.allowedModules,
+            parameters,
         });
-        const response = await this.callLLM([
+        const content = await llmChat(this.llm, [
             { role: 'system', content: TOOL_GENERATION_SYSTEM_PROMPT },
             { role: 'user', content: prompt },
-        ], 0.4);
-        const tool = parseToolGenerationResponse(response.content);
+        ], { model: this.model, temperature: 0.4 });
+        const tool = parseToolGenerationResponse(content);
         if (tool) {
             tool.metadata = {
                 ...tool.metadata,
@@ -114,13 +125,15 @@ export class ToolGenerator {
     }
     async improve(tool, validationResult, iteration) {
         const prompt = buildToolImprovementPrompt(tool, validationResult, iteration);
-        const response = await this.callLLM([
+        const content = await llmChat(this.llm, [
             { role: 'system', content: TOOL_GENERATION_SYSTEM_PROMPT },
             { role: 'user', content: prompt },
-        ], 0.3);
-        const improved = parseToolGenerationResponse(response.content);
+        ], { model: this.model, temperature: 0.3 });
+        const improved = parseToolGenerationResponse(content);
         if (improved) {
             improved.id = tool.id;
+            improved.name = tool.name;
+            improved.createdAt = tool.createdAt;
             improved.version = tool.version + 1;
             improved.metadata = {
                 ...tool.metadata,
@@ -134,8 +147,14 @@ export class ToolGenerator {
     createExecutableTool(generated) {
         const sandbox = this.sandbox;
         const tool = generated;
+        const paramSchema = buildZodSchema(generated.parameters);
+        const jsonSchema = toToolJsonSchema(generated.parameters);
         const execute = async (params) => {
-            const result = await sandbox.execute(tool, params);
+            const parsed = paramSchema.safeParse(params);
+            if (!parsed.success) {
+                throw new Error(`Invalid parameters: ${parsed.error.issues.map((i) => `${i.path.join('.')}: ${i.message}`).join('; ')}`);
+            }
+            const result = await sandbox.execute(tool, parsed.data);
             if (!result.success) {
                 throw new Error(result.error ?? 'Tool execution failed');
             }
@@ -144,23 +163,69 @@ export class ToolGenerator {
         return {
             name: generated.name,
             description: generated.description,
-            parameters: z.record(z.string(), z.unknown()),
+            parameters: paramSchema,
             execute,
             toJSON: () => ({
                 name: generated.name,
                 description: generated.description,
-                parameters: {
-                    type: 'object',
-                    properties: generated.parameters,
-                },
+                parameters: jsonSchema,
             }),
         };
     }
-    async callLLM(messages, temperature) {
-        if (this.llm.complete) {
-            return this.llm.complete({ messages, temperature });
+}
+function toToolJsonSchema(parameters) {
+    const properties = parameters.type === 'object' &&
+        parameters.properties &&
+        typeof parameters.properties === 'object' &&
+        !Array.isArray(parameters.properties)
+        ? parameters.properties
+        : {};
+    const required = Array.isArray(parameters.required)
+        ? parameters.required.filter((key) => typeof key === 'string')
+        : [];
+    return required.length > 0
+        ? { type: 'object', properties, required }
+        : { type: 'object', properties };
+}
+function buildZodSchema(parameters) {
+    if (parameters.type !== 'object' ||
+        !parameters.properties ||
+        typeof parameters.properties !== 'object') {
+        return z.record(z.string(), z.unknown());
+    }
+    const properties = parameters.properties;
+    const required = Array.isArray(parameters.required)
+        ? new Set(parameters.required)
+        : new Set();
+    const shape = {};
+    for (const [key, propSchema] of Object.entries(properties)) {
+        let fieldSchema = jsonTypeToZod(propSchema.type);
+        if (!required.has(key)) {
+            fieldSchema = fieldSchema.optional();
         }
-        return this.llm.chat({ model: this.model, messages, temperature });
+        shape[key] = fieldSchema;
+    }
+    if (Object.keys(shape).length === 0) {
+        return z.record(z.string(), z.unknown());
+    }
+    return z.object(shape).passthrough();
+}
+function jsonTypeToZod(type) {
+    switch (type) {
+        case 'string':
+            return z.string();
+        case 'number':
+            return z.number();
+        case 'integer':
+            return z.number().int();
+        case 'boolean':
+            return z.boolean();
+        case 'array':
+            return z.array(z.unknown());
+        case 'object':
+            return z.record(z.string(), z.unknown());
+        default:
+            return z.unknown();
     }
 }
 //# sourceMappingURL=tool-generator.js.map

package/dist/tool-generation/tool-generator.js.map

@@ -1 +1 @@
-{"version":3,"file":"tool-generator.js","sourceRoot":"","sources":["../../src/tool-generation/tool-generator.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,CAAC,EAAgB,MAAM,KAAK,CAAC;AACtC,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACjD,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EACL,6BAA6B,EAC7B,yBAAyB,EACzB,0BAA0B,EAC1B,2BAA2B,GAC5B,MAAM,WAAW,CAAC;AAgBnB,MAAM,OAAO,aAAa;IACP,GAAG,CAAa;IAChB,MAAM,CAA2B;IACjC,KAAK,CAAS;IACd,SAAS,CAAgB;IACzB,OAAO,CAAc;IAEtC,YAAY,OAA6B;QACvC,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,SAAS,CAAC;QACxC,IAAI,CAAC,SAAS,GAAG,IAAI,aAAa,CAAC;YACjC,GAAG,EAAE,OAAO,CAAC,GAAG;YAChB,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,KAAK,EAAE,OAAO,CAAC,KAAK;SACrB,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,GAAG,IAAI,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,GAAkB,EAClB,aAAqB,EACrB,SAA+D;QAE/D,IAAI,WAAW,GAAyB,IAAI,CAAC;QAC7C,IAAI,gBAAgB,GAAgC,IAAI,CAAC;QACzD,IAAI,UAAU,GAAG,CAAC,CAAC;QAEnB,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,IAAI,CAAC,CAAC;QAE5D,OAAO,UAAU,GAAG,aAAa,EAAE,CAAC;YAClC,UAAU,EAAE,CAAC;YAEb,IAAI,CAAC;gBACH,IAAI,WAAW,KAAK,IAAI,EAAE,CAAC;oBACzB,WAAW,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;gBAC/D,CAAC;qBAAM,IAAI,gBAAgB,EAAE,CAAC;oBAC5B,WAAW,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,gBAAgB,EAAE,UAAU,CAAC,CAAC;gBAC9E,CAAC;gBAED,IAAI,CAAC,WAAW,EAAE,CAAC;oBACjB,SAAS;gBACX,CAAC;gBAED,gBAAgB,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;gBAEzE,IAAI,gBAAgB,CAAC,OAAO,EAAE,CAAC;oBAC7B,WAAW,CAAC,MAAM,GAAG,WAAW,CAAC;oBACjC,WAAW,CAAC,eAAe,GAAG,gBAAgB,CAAC,YAAY,CAAC;oBAE5D,OAAO;wBACL,IAAI,EAAE,WAAW;wBACjB,gBAAgB;wBAChB,UAAU;wBACV,OAAO,EAAE,IAAI;qBACd,CAAC;gBACJ,CAAC;gBAED,IAAI,gBAAgB,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,IAAI,UAAU,IAAI,CAAC,EAAE,CAAC;oBAClE,OAAO;wBACL,IAAI,EAAE,WAAW;wBACjB,gBAAgB;wBAChB,UAAU;wBACV,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,iCAAiC,UAAU,gBAAgB,gBAAgB,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;qBAC/G,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO;oBACL,IAAI,EAAE,WAAW;oBACjB,gBAAgB;oBAChB,UAAU;oBACV,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC9D,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO;YACL,IAAI,EAAE,WAAW;YACjB,gBAAgB;YAChB,UAAU;YACV,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,uCAAuC,aAAa,aAAa;SACzE,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,aAAa,CACjB,WAAmB,EACnB,IAAY,EACZ,WAAoC;QAEpC,MAAM,GAAG,GAAkB;YACzB,EAAE,EAAE,SAAS,IAAI,CAAC,GAAG,EAAE,EAAE;YACzB,WAAW;YACX,kBAAkB,EAAE,WAAW;YAC/B,iBAAiB,EAAE,IAAI;YACvB,UAAU,EAAE,QAAQ;YACpB,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,iCAAiC;SAC7C,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAC5C,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,GAAkB,EAClB,aAAqB;QAErB,MAAM,aAAa,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC9C,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,WAAW,EAAE,CAAC,CAAC,WAAW;SAC3B,CAAC,CAAC,CAAC;QAEJ,MAAM,MAAM,GAAG,yBAAyB,CAAC,GAAG,EAAE,aAAa,EAAE;YAC3D,QAAQ,EAAE,GAAG;YACb,aAAa,EAAE,QAAQ;YACvB,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,cAAc;SAC1D,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CACjC;YACE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,6BAA6B,EAAE;YAC1D,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE;SAClC,EACD,GAAG,CACJ,CAAC;QAEF,MAAM,IAAI,GAAG,2BAA2B,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAE3D,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,QAAQ,GAAG;gBACd,GAAG,IAAI,CAAC,QAAQ;gBAChB,KAAK,EAAE,GAAG,CAAC,EAAE;gBACb,UAAU,EAAE,GAAG,CAAC,UAAU;aAC3B,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,OAAO,CACnB,IAAmB,EACnB,gBAAsC,EACtC,SAAiB;QAEjB,MAAM,MAAM,GAAG,0BAA0B,CAAC,IAAI,EAAE,gBAAgB,EAAE,SAAS,CAAC,CAAC;QAE7E,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CACjC;YACE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,6BAA6B,EAAE;YAC1D,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE;SAClC,EACD,GAAG,CACJ,CAAC;QAEF,MAAM,QAAQ,GAAG,2BAA2B,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAE/D,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,CAAC,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC;YACtB,QAAQ,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC;YACpC,QAAQ,CAAC,QAAQ,GAAG;gBAClB,GAAG,IAAI,CAAC,QAAQ;gBAChB,GAAG,QAAQ,CAAC,QAAQ;gBACpB,eAAe,EAAE,IAAI,CAAC,OAAO;gBAC7B,oBAAoB,EAAE,SAAS;aAChC,CAAC;QACJ,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,oBAAoB,CAAC,SAAwB;QAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,MAAM,IAAI,GAAG,SAAS,CAAC;QAEvB,MAAM,OAAO,GAAG,KAAK,EAAE,MAAe,EAAoB,EAAE;YAC1D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACnD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpB,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,IAAI,uBAAuB,CAAC,CAAC;YAC3D,CAAC;YACD,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,CAAC,CAAC;QAEF,OAAO;YACL,IAAI,EAAE,SAAS,CAAC,IAAI;YACpB,WAAW,EAAE,SAAS,CAAC,WAAW;YAClC,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAqB;YACjE,OAAO;YACP,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;gBACb,IAAI,EAAE,SAAS,CAAC,IAAI;gBACpB,WAAW,EAAE,SAAS,CAAC,WAAW;gBAClC,UAAU,EAAE;oBACV,IAAI,EAAE,QAAiB;oBACvB,UAAU,EAAE,SAAS,CAAC,UAAqC;iBAC5D;aACF,CAAC;SACH,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,OAAO,CACnB,QAA2E,EAC3E,WAAmB;QAEnB,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;QACtD,CAAC;QACD,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;IACrE,CAAC;CACF"}
+{"version":3,"file":"tool-generator.js","sourceRoot":"","sources":["../../src/tool-generation/tool-generator.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,CAAC,EAAgB,MAAM,KAAK,CAAC;AACtC,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACjD,OAAO,EAAE,WAAW,EAAwB,MAAM,gBAAgB,CAAC;AACnE,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EACL,6BAA6B,EAC7B,yBAAyB,EACzB,0BAA0B,EAC1B,2BAA2B,GAC5B,MAAM,WAAW,CAAC;AAoBnB,MAAM,OAAO,aAAa;IACP,GAAG,CAAa;IAChB,MAAM,CAA2B;IACjC,KAAK,CAAS;IACd,SAAS,CAAgB;IACzB,OAAO,CAAc;IAEtC,YAAY,OAA6B;QACvC,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,SAAS,CAAC;QACxC,IAAI,CAAC,SAAS,GAAG,IAAI,aAAa,CAAC;YACjC,GAAG,EAAE,OAAO,CAAC,GAAG;YAChB,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,KAAK,EAAE,OAAO,CAAC,KAAK;SACrB,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,GAAG,IAAI,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,GAAkB,EAClB,aAAqB,EACrB,SAA6B,EAC7B,UAA2B,EAAE;QAE7B,IAAI,WAAW,GAAyB,IAAI,CAAC;QAC7C,IAAI,gBAAgB,GAAgC,IAAI,CAAC;QACzD,IAAI,UAAU,GAAG,CAAC,CAAC;QAEnB,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,IAAI,CAAC,CAAC;QAE5D,OAAO,UAAU,GAAG,aAAa,EAAE,CAAC;YAClC,UAAU,EAAE,CAAC;YAEb,IAAI,CAAC;gBACH,IAAI,WAAW,KAAK,IAAI,EAAE,CAAC;oBACzB,WAAW,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,aAAa,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;gBACnF,CAAC;qBAAM,IAAI,gBAAgB,EAAE,CAAC;oBAC5B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,gBAAgB,EAAE,UAAU,CAAC,CAAC;oBAC/E,IAAI,CAAC,QAAQ,EAAE,CAAC;wBACd,SAAS;oBACX,CAAC;oBACD,WAAW,GAAG,QAAQ,CAAC;gBACzB,CAAC;gBAED,IAAI,CAAC,WAAW,EAAE,CAAC;oBACjB,SAAS;gBACX,CAAC;gBAED,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;oBACvB,WAAW,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;gBAC9C,CAAC;gBAED,gBAAgB,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;gBAEzE,IAAI,gBAAgB,CAAC,OAAO,EAAE,CAAC;oBAC7B,WAAW,CAAC,MAAM,GAAG,WAAW,CAAC;oBACjC,WAAW,CAAC,eAAe,GAAG,gBAAgB,CAAC,YAAY,CAAC;oBAE5D,OAAO;wBACL,IAAI,EAAE,WAAW;wBACjB,gBAAgB;wBAChB,UAAU;wBACV,OAAO,EAAE,IAAI;qBACd,CAAC;gBACJ,CAAC;gBAED,IAAI,gBAAgB,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,IAAI,UAAU,IAAI,CAAC,EAAE,CAAC;oBAClE,OAAO;wBACL,IAAI,EAAE,WAAW;wBACjB,gBAAgB;wBAChB,UAAU;wBACV,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,iCAAiC,UAAU,gBAAgB,gBAAgB,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;qBAC/G,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO;oBACL,IAAI,EAAE,WAAW;oBACjB,gBAAgB;oBAChB,UAAU;oBACV,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC9D,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO;YACL,IAAI,EAAE,WAAW;YACjB,gBAAgB;YAChB,UAAU;YACV,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,uCAAuC,aAAa,aAAa;SACzE,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,aAAa,CACjB,WAAmB,EACnB,IAAY,EACZ,UAAmC;QAEnC,MAAM,GAAG,GAAkB;YACzB,EAAE,EAAE,SAAS,IAAI,CAAC,GAAG,EAAE,EAAE;YACzB,WAAW;YACX,kBAAkB,EAAE,WAAW;YAC/B,iBAAiB,EAAE,IAAI;YACvB,UAAU,EAAE,QAAQ;YACpB,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,iCAAiC;SAC7C,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAChC,GAAG,EACH,EAAE,EACF,SAAS,EACT,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC;YAChC,CAAC,CAAC,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE;YAC5D,CAAC,CAAC,EAAE,CACP,CAAC;QAEF,OAAO,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;IAC7C,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,GAAkB,EAClB,aAAqB,EACrB,UAAoC;QAEpC,MAAM,aAAa,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC9C,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,WAAW,EAAE,CAAC,CAAC,WAAW;SAC3B,CAAC,CAAC,CAAC;QAEJ,MAAM,MAAM,GAAG,yBAAyB,CAAC,GAAG,EAAE,aAAa,EAAE;YAC3D,QAAQ,EAAE,GAAG;YACb,aAAa,EAAE,QAAQ;YACvB,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,cAAc;YACzD,UAAU;SACX,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,OAAO,CAC3B,IAAI,CAAC,GAAG,EACR;YACE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,6BAA6B,EAAE;YAC1D,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE;SAClC,EACD,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,WAAW,EAAE,GAAG,EAAE,CACxC,CAAC;QAEF,MAAM,IAAI,GAAG,2BAA2B,CAAC,OAAO,CAAC,CAAC;QAElD,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,QAAQ,GAAG;gBACd,GAAG,IAAI,CAAC,QAAQ;gBAChB,KAAK,EAAE,GAAG,CAAC,EAAE;gBACb,UAAU,EAAE,GAAG,CAAC,UAAU;aAC3B,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,OAAO,CACnB,IAAmB,EACnB,gBAAsC,EACtC,SAAiB;QAEjB,MAAM,MAAM,GAAG,0BAA0B,CAAC,IAAI,EAAE,gBAAgB,EAAE,SAAS,CAAC,CAAC;QAE7E,MAAM,OAAO,GAAG,MAAM,OAAO,CAC3B,IAAI,CAAC,GAAG,EACR;YACE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,6BAA6B,EAAE;YAC1D,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE;SAClC,EACD,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,WAAW,EAAE,GAAG,EAAE,CACxC,CAAC;QAEF,MAAM,QAAQ,GAAG,2BAA2B,CAAC,OAAO,CAAC,CAAC;QAEtD,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,CAAC,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC;YACtB,QAAQ,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;YAC1B,QAAQ,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;YACpC,QAAQ,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC;YACpC,QAAQ,CAAC,QAAQ,GAAG;gBAClB,GAAG,IAAI,CAAC,QAAQ;gBAChB,GAAG,QAAQ,CAAC,QAAQ;gBACpB,eAAe,EAAE,IAAI,CAAC,OAAO;gBAC7B,oBAAoB,EAAE,SAAS;aAChC,CAAC;QACJ,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,oBAAoB,CAAC,SAAwB;QAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,MAAM,IAAI,GAAG,SAAS,CAAC;QACvB,MAAM,WAAW,GAAG,cAAc,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACzD,MAAM,UAAU,GAAG,gBAAgB,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAE1D,MAAM,OAAO,GAAG,KAAK,EAAE,MAAe,EAAoB,EAAE;YAC1D,MAAM,MAAM,GAAG,WAAW,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAC7C,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpB,MAAM,IAAI,KAAK,CACb,uBAAuB,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACxG,CAAC;YACJ,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YACxD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpB,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,IAAI,uBAAuB,CAAC,CAAC;YAC3D,CAAC;YACD,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,CAAC,CAAC;QAEF,OAAO;YACL,IAAI,EAAE,SAAS,CAAC,IAAI;YACpB,WAAW,EAAE,SAAS,CAAC,WAAW;YAClC,UAAU,EAAE,WAA+B;YAC3C,OAAO;YACP,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;gBACb,IAAI,EAAE,SAAS,CAAC,IAAI;gBACpB,WAAW,EAAE,SAAS,CAAC,WAAW;gBAClC,UAAU,EAAE,UAAU;aACvB,CAAC;SACH,CAAC;IACJ,CAAC;CACF;AAED,SAAS,gBAAgB,CAAC,UAAmC;IAC3D,MAAM,UAAU,GACd,UAAU,CAAC,IAAI,KAAK,QAAQ;QAC5B,UAAU,CAAC,UAAU;QACrB,OAAO,UAAU,CAAC,UAAU,KAAK,QAAQ;QACzC,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC;QACnC,CAAC,CAAE,UAAU,CAAC,UAAsC;QACpD,CAAC,CAAC,EAAE,CAAC;IACT,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;QACjD,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAiB,EAAE,CAAC,OAAO,GAAG,KAAK,QAAQ,CAAC;QAC7E,CAAC,CAAC,EAAE,CAAC;IAEP,OAAO,QAAQ,CAAC,MAAM,GAAG,CAAC;QACxB,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE;QAC1C,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC;AACrC,CAAC;AAED,SAAS,cAAc,CAAC,UAAmC;IACzD,IACE,UAAU,CAAC,IAAI,KAAK,QAAQ;QAC5B,CAAC,UAAU,CAAC,UAAU;QACtB,OAAO,UAAU,CAAC,UAAU,KAAK,QAAQ,EACzC,CAAC;QACD,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,UAAU,GAAG,UAAU,CAAC,UAAqD,CAAC;IACpF,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;QACjD,CAAC,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,QAAoB,CAAC;QAC1C,CAAC,CAAC,IAAI,GAAG,EAAU,CAAC;IAEtB,MAAM,KAAK,GAAqC,EAAE,CAAC;IAEnD,KAAK,MAAM,CAAC,GAAG,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QAC3D,IAAI,WAAW,GAAqB,aAAa,CAAC,UAAU,CAAC,IAA0B,CAAC,CAAC;QAEzF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YACvB,WAAW,GAAG,WAAW,CAAC,QAAQ,EAAsB,CAAC;QAC3D,CAAC;QAED,KAAK,CAAC,GAAG,CAAC,GAAG,WAAW,CAAC;IAC3B,CAAC;IAED,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACpC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAsB,CAAC;AAC3D,CAAC;AAED,SAAS,aAAa,CAAC,IAAa;IAClC,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,QAAQ;YACX,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC;QACpB,KAAK,QAAQ;YACX,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC;QACpB,KAAK,SAAS;YACZ,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC;QAC1B,KAAK,SAAS;YACZ,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;QACrB,KAAK,OAAO;YACV,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QAC9B,KAAK,QAAQ;YACX,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QAC3C;YACE,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;IACvB,CAAC;AACH,CAAC"}

package/dist/tool-generation/tool-sandbox.d.ts

@@ -1,15 +1,16 @@
 import type { ToolSandboxConfig, ToolSandboxResult, GeneratedTool } from '@cogitator-ai/types';
 export declare const DEFAULT_SANDBOX_CONFIG: ToolSandboxConfig;
+export interface SandboxTestCase {
+    input: unknown;
+    expectedOutput?: unknown;
+    shouldThrow?: boolean;
+    allowThrow?: boolean;
+}
 export declare class ToolSandbox {
     private readonly config;
-    private readonly logs;
     constructor(config?: Partial<ToolSandboxConfig>);
     execute(tool: GeneratedTool, params: unknown): Promise<ToolSandboxResult>;
-    testWithCases(tool: GeneratedTool, testCases: Array<{
-        input: unknown;
-        expectedOutput?: unknown;
-        shouldThrow?: boolean;
-    }>): Promise<{
+    testWithCases(tool: GeneratedTool, testCases: SandboxTestCase[]): Promise<{
         passed: number;
         failed: number;
         results: Array<{
@@ -20,12 +21,10 @@ export declare class ToolSandbox {
             executionTime: number;
         }>;
     }>;
+    private run;
     private validateImplementation;
-    private createContext;
-    private wrapImplementation;
-    private executeWithTimeout;
+    private executeInWorker;
     private executeUnsandboxed;
-    private estimateMemoryUsage;
-    private deepEqual;
 }
+export declare function deepEqual(a: unknown, b: unknown): boolean;
 //# sourceMappingURL=tool-sandbox.d.ts.map

package/dist/tool-generation/tool-sandbox.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tool-sandbox.d.ts","sourceRoot":"","sources":["../../src/tool-generation/tool-sandbox.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAE/F,eAAO,MAAM,sBAAsB,EAAE,iBAMpC,CAAC;AAyBF,qBAAa,WAAW;IACtB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAoB;IAC3C,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAgB;gBAEzB,MAAM,GAAE,OAAO,CAAC,iBAAiB,CAAM;IAI7C,OAAO,CAAC,IAAI,EAAE,aAAa,EAAE,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAuCzE,aAAa,CACjB,IAAI,EAAE,aAAa,EACnB,SAAS,EAAE,KAAK,CAAC;QAAE,KAAK,EAAE,OAAO,CAAC;QAAC,cAAc,CAAC,EAAE,OAAO,CAAC;QAAC,WAAW,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC,GACpF,OAAO,CAAC;QACT,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,EAAE,KAAK,CAAC;YACb,KAAK,EAAE,OAAO,CAAC;YACf,MAAM,CAAC,EAAE,OAAO,CAAC;YACjB,KAAK,CAAC,EAAE,MAAM,CAAC;YACf,MAAM,EAAE,OAAO,CAAC;YAChB,aAAa,EAAE,MAAM,CAAC;SACvB,CAAC,CAAC;KACJ,CAAC;IAqCF,OAAO,CAAC,sBAAsB;IAgD9B,OAAO,CAAC,aAAa;IA+BrB,OAAO,CAAC,kBAAkB;YAgBZ,kBAAkB;YAgClB,kBAAkB;IA2ChC,OAAO,CAAC,mBAAmB;IAS3B,OAAO,CAAC,SAAS;CAsBlB"}
+{"version":3,"file":"tool-sandbox.d.ts","sourceRoot":"","sources":["../../src/tool-generation/tool-sandbox.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAE/F,eAAO,MAAM,sBAAsB,EAAE,iBAMpC,CAAC;AAkIF,MAAM,WAAW,eAAe;IAC9B,KAAK,EAAE,OAAO,CAAC;IACf,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,UAAU,CAAC,EAAE,OAAO,CAAC;CACtB;AAED,qBAAa,WAAW;IACtB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAoB;gBAE/B,MAAM,GAAE,OAAO,CAAC,iBAAiB,CAAM;IAI7C,OAAO,CAAC,IAAI,EAAE,aAAa,EAAE,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAKzE,aAAa,CACjB,IAAI,EAAE,aAAa,EACnB,SAAS,EAAE,eAAe,EAAE,GAC3B,OAAO,CAAC;QACT,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,EAAE,KAAK,CAAC;YACb,KAAK,EAAE,OAAO,CAAC;YACf,MAAM,CAAC,EAAE,OAAO,CAAC;YACjB,KAAK,CAAC,EAAE,MAAM,CAAC;YACf,MAAM,EAAE,OAAO,CAAC;YAChB,aAAa,EAAE,MAAM,CAAC;SACvB,CAAC,CAAC;KACJ,CAAC;YAuCY,GAAG;IAgCjB,OAAO,CAAC,sBAAsB;IAwE9B,OAAO,CAAC,eAAe;YAkFT,kBAAkB;CAmDjC;AA4BD,wBAAgB,SAAS,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,GAAG,OAAO,CAEzD"}