@nestbox-ai/cli 1.0.63 → 1.0.64

package/dist/agents/reportGenerator/index.js

@@ -0,0 +1,210 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runReportComposerAgent = runReportComposerAgent;
+const sdk_1 = __importDefault(require("@anthropic-ai/sdk"));
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const js_yaml_1 = __importDefault(require("js-yaml"));
+const ajv_1 = __importDefault(require("ajv"));
+// ─── Constants ────────────────────────────────────────────────────────────────
+const AGENTS_DIR = __dirname;
+const DEFAULT_MODEL = 'claude-sonnet-4-6';
+const DEFAULT_MAX_ITERATIONS = 5;
+// ─── Helpers ─────────────────────────────────────────────────────────────────
+function readLocalFile(filename) {
+    return fs_1.default.readFileSync(path_1.default.join(AGENTS_DIR, filename), 'utf8');
+}
+function validateYaml(content, schemaContent) {
+    var _a;
+    // Replace ${ENV_VAR} and ${ENV_VAR:-default} references with a plain string
+    // so YAML parses cleanly and AJV treats them as valid string values.
+    const preprocessed = content.replace(/\$\{[^}]+\}/g, 'env-var-placeholder');
+    let data;
+    try {
+        data = js_yaml_1.default.load(preprocessed);
+    }
+    catch (e) {
+        return { valid: false, errors: [`YAML parse error: ${e.message}`] };
+    }
+    let schema;
+    try {
+        schema = js_yaml_1.default.load(schemaContent);
+    }
+    catch (e) {
+        return { valid: false, errors: [`Schema parse error: ${e.message}`] };
+    }
+    const ajv = new ajv_1.default({ strict: false, allErrors: true });
+    // Suppress "unknown format 'uri' ignored" console warnings from AJV
+    ajv.addFormat('uri', () => true);
+    const validate = ajv.compile(schema);
+    const ok = validate(data);
+    if (ok)
+        return { valid: true, errors: [] };
+    const errors = ((_a = validate.errors) !== null && _a !== void 0 ? _a : []).map((err) => {
+        const loc = err.instancePath || '(root)';
+        return `[${loc}] ${err.message}`;
+    });
+    return { valid: false, errors };
+}
+// ─── System Prompt ────────────────────────────────────────────────────────────
+function buildSystemPrompt() {
+    const systemPrompt = readLocalFile('SYSTEM_PROMPT.md');
+    const configGuide = readLocalFile('REPORT_CONFIG_GUIDE.md');
+    const example1 = readLocalFile('annual_report_10k.yaml');
+    const example2 = readLocalFile('vc_portfolio_monitoring.yaml');
+    return [
+        systemPrompt,
+        '---',
+        configGuide,
+        '---',
+        '# Example 1: Annual Report / 10-K Analysis\n\n```yaml\n' + example1 + '\n```',
+        '---',
+        '# Example 2: VC Portfolio Monitoring\n\n```yaml\n' + example2 + '\n```',
+    ].join('\n\n');
+}
+// ─── Tool Definitions ─────────────────────────────────────────────────────────
+const TOOLS = [
+    {
+        name: 'write_and_validate_report',
+        description: 'Write the report.yaml content and validate it against the schema. Returns "VALID" on success or a list of validation errors to fix.',
+        input_schema: {
+            type: 'object',
+            properties: {
+                yaml_content: {
+                    type: 'string',
+                    description: 'The complete YAML content for report.yaml',
+                },
+            },
+            required: ['yaml_content'],
+        },
+    },
+    {
+        name: 'finish',
+        description: 'Signal that the report.yaml is complete and valid. Call this only after write_and_validate_report has returned "VALID".',
+        input_schema: {
+            type: 'object',
+            properties: {
+                summary: {
+                    type: 'string',
+                    description: 'Brief summary of what was generated and why key choices were made',
+                },
+            },
+            required: ['summary'],
+        },
+    },
+];
+// ─── Agent ────────────────────────────────────────────────────────────────────
+function runReportComposerAgent(options) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const { instructions, anthropicApiKey, model = DEFAULT_MODEL, maxIterations = DEFAULT_MAX_ITERATIONS, onProgress = () => { }, } = options;
+        const reportSchema = readLocalFile('report_config.schema.yaml');
+        const client = new sdk_1.default({ apiKey: anthropicApiKey });
+        // Mutable state updated by tool calls
+        let latestReport = '';
+        let reportValid = false;
+        let finished = false;
+        let iteration = 0;
+        // Execute a tool call and return the result string
+        function executeTool(name, input) {
+            var _a;
+            if (name === 'write_and_validate_report') {
+                const content = (_a = input.yaml_content) !== null && _a !== void 0 ? _a : '';
+                latestReport = content;
+                const result = validateYaml(content, reportSchema);
+                reportValid = result.valid;
+                if (result.valid)
+                    return 'VALID';
+                return `VALIDATION ERRORS — fix all of these before calling again:\n${result.errors.map((e) => `  • ${e}`).join('\n')}`;
+            }
+            if (name === 'finish') {
+                if (!reportValid) {
+                    return 'Cannot finish: the report has not passed schema validation yet. Call write_and_validate_report first and ensure it returns "VALID".';
+                }
+                finished = true;
+                return 'Done.';
+            }
+            return `Unknown tool: ${name}`;
+        }
+        // Build the system prompt once and mark it for caching.
+        // On iteration 1 the prompt is written to the cache (normal price).
+        // On iterations 2+ the large system prompt is read from cache at
+        // 10% of the normal input token price.
+        const systemPromptText = buildSystemPrompt();
+        const systemPrompt = [
+            {
+                type: 'text',
+                text: systemPromptText,
+                cache_control: { type: 'ephemeral' },
+            },
+        ];
+        const messages = [
+            {
+                role: 'user',
+                content: `Here are the instructions for the report you need to configure:\n\n${instructions}\n\nGenerate the report.yaml file now. Use the tools to write and validate it.`,
+            },
+        ];
+        onProgress('Starting agent...');
+        while (iteration < maxIterations && !finished) {
+            iteration++;
+            onProgress(`Iteration ${iteration}/${maxIterations} — calling Claude...`);
+            const response = yield client.messages.create({
+                model,
+                max_tokens: 8096,
+                system: systemPrompt,
+                tools: TOOLS,
+                // Force Claude to call a tool every turn — prevents it from
+                // replying with plain text and exiting the loop prematurely.
+                tool_choice: { type: 'any' },
+                messages,
+            });
+            // Append assistant turn
+            messages.push({ role: 'assistant', content: response.content });
+            if (response.stop_reason !== 'tool_use') {
+                onProgress(`Agent stopped unexpectedly: ${response.stop_reason}`);
+                break;
+            }
+            // Process all tool calls and collect results
+            const toolResults = [];
+            for (const block of response.content) {
+                if (block.type !== 'tool_use')
+                    continue;
+                onProgress(`  → tool: ${block.name}`);
+                const result = executeTool(block.name, block.input);
+                onProgress(`    ${result.startsWith('VALID') ? '✓ valid' : result.split('\n')[0]}`);
+                toolResults.push({
+                    type: 'tool_result',
+                    tool_use_id: block.id,
+                    content: result,
+                });
+                if (finished)
+                    break;
+            }
+            if (toolResults.length > 0) {
+                messages.push({ role: 'user', content: toolResults });
+            }
+            if (finished)
+                break;
+        }
+        if (iteration >= maxIterations && !finished) {
+            onProgress(`Warning: reached max iterations (${maxIterations}) without finishing.`);
+        }
+        return {
+            reportYaml: latestReport,
+            iterations: iteration,
+            reportValid,
+        };
+    });
+}
+//# sourceMappingURL=index.js.map

package/dist/agents/reportGenerator/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/agents/reportGenerator/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;AAsIA,wDA2HC;AAjQD,4DAA0C;AAC1C,4CAAoB;AACpB,gDAAwB;AACxB,sDAA2B;AAC3B,8CAAsB;AA4BtB,iFAAiF;AAEjF,MAAM,UAAU,GAAG,SAAS,CAAC;AAC7B,MAAM,aAAa,GAAG,mBAAmB,CAAC;AAC1C,MAAM,sBAAsB,GAAG,CAAC,CAAC;AAEjC,gFAAgF;AAEhF,SAAS,aAAa,CAAC,QAAgB;IACrC,OAAO,YAAE,CAAC,YAAY,CAAC,cAAI,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,CAAC;AAClE,CAAC;AAED,SAAS,YAAY,CAAC,OAAe,EAAE,aAAqB;;IAC1D,4EAA4E;IAC5E,qEAAqE;IACrE,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,cAAc,EAAE,qBAAqB,CAAC,CAAC;IAE5E,IAAI,IAAa,CAAC;IAClB,IAAI,CAAC;QACH,IAAI,GAAG,iBAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACjC,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,qBAAqB,CAAC,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;IACtE,CAAC;IAED,IAAI,MAAc,CAAC;IACnB,IAAI,CAAC;QACH,MAAM,GAAG,iBAAI,CAAC,IAAI,CAAC,aAAa,CAAW,CAAC;IAC9C,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,uBAAuB,CAAC,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;IACxE,CAAC;IAED,MAAM,GAAG,GAAG,IAAI,aAAG,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACxD,oEAAoE;IACpE,GAAG,CAAC,SAAS,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;IACjC,MAAM,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACrC,MAAM,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;IAE1B,IAAI,EAAE;QAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;IAE3C,MAAM,MAAM,GAAG,CAAC,MAAA,QAAQ,CAAC,MAAM,mCAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QACjD,MAAM,GAAG,GAAG,GAAG,CAAC,YAAY,IAAI,QAAQ,CAAC;QACzC,OAAO,IAAI,GAAG,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;IACH,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;AAClC,CAAC;AAED,iFAAiF;AAEjF,SAAS,iBAAiB;IACxB,MAAM,YAAY,GAAI,aAAa,CAAC,kBAAkB,CAAC,CAAC;IACxD,MAAM,WAAW,GAAK,aAAa,CAAC,wBAAwB,CAAC,CAAC;IAC9D,MAAM,QAAQ,GAAQ,aAAa,CAAC,wBAAwB,CAAC,CAAC;IAC9D,MAAM,QAAQ,GAAQ,aAAa,CAAC,8BAA8B,CAAC,CAAC;IAEpE,OAAO;QACL,YAAY;QACZ,KAAK;QACL,WAAW;QACX,KAAK;QACL,yDAAyD,GAAG,QAAQ,GAAG,OAAO;QAC9E,KAAK;QACL,mDAAmD,GAAG,QAAQ,GAAG,OAAO;KACzE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AACjB,CAAC;AAED,iFAAiF;AAEjF,MAAM,KAAK,GAAqB;IAC9B;QACE,IAAI,EAAE,2BAA2B;QACjC,WAAW,EACT,qIAAqI;QACvI,YAAY,EAAE;YACZ,IAAI,EAAE,QAAiB;YACvB,UAAU,EAAE;gBACV,YAAY,EAAE;oBACZ,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,2CAA2C;iBACzD;aACF;YACD,QAAQ,EAAE,CAAC,cAAc,CAAC;SAC3B;KACF;IACD;QACE,IAAI,EAAE,QAAQ;QACd,WAAW,EACT,yHAAyH;QAC3H,YAAY,EAAE;YACZ,IAAI,EAAE,QAAiB;YACvB,UAAU,EAAE;gBACV,OAAO,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,mEAAmE;iBACjF;aACF;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;KACF;CACF,CAAC;AAEF,iFAAiF;AAEjF,SAAsB,sBAAsB,CAC1C,OAAmC;;QAEnC,MAAM,EACJ,YAAY,EACZ,eAAe,EACf,KAAK,GAAG,aAAa,EACrB,aAAa,GAAG,sBAAsB,EACtC,UAAU,GAAG,GAAG,EAAE,GAAE,CAAC,GACtB,GAAG,OAAO,CAAC;QAEZ,MAAM,YAAY,GAAG,aAAa,CAAC,2BAA2B,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,IAAI,aAAS,CAAC,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC,CAAC;QAE1D,sCAAsC;QACtC,IAAI,YAAY,GAAG,EAAE,CAAC;QACtB,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,mDAAmD;QACnD,SAAS,WAAW,CAAC,IAAY,EAAE,KAA6B;;YAC9D,IAAI,IAAI,KAAK,2BAA2B,EAAE,CAAC;gBACzC,MAAM,OAAO,GAAG,MAAA,KAAK,CAAC,YAAY,mCAAI,EAAE,CAAC;gBACzC,YAAY,GAAG,OAAO,CAAC;gBACvB,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;gBACnD,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC;gBAC3B,IAAI,MAAM,CAAC,KAAK;oBAAE,OAAO,OAAO,CAAC;gBACjC,OAAO,+DAA+D,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC1H,CAAC;YAED,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtB,IAAI,CAAC,WAAW,EAAE,CAAC;oBACjB,OAAO,qIAAqI,CAAC;gBAC/I,CAAC;gBACD,QAAQ,GAAG,IAAI,CAAC;gBAChB,OAAO,OAAO,CAAC;YACjB,CAAC;YAED,OAAO,iBAAiB,IAAI,EAAE,CAAC;QACjC,CAAC;QAED,wDAAwD;QACxD,oEAAoE;QACpE,iEAAiE;QACjE,uCAAuC;QACvC,MAAM,gBAAgB,GAAG,iBAAiB,EAAE,CAAC;QAC7C,MAAM,YAAY,GAA+B;YAC/C;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,gBAAgB;gBACtB,aAAa,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;aACrC;SACF,CAAC;QAEF,MAAM,QAAQ,GAA6B;YACzC;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,sEAAsE,YAAY,gFAAgF;aAC5K;SACF,CAAC;QAEF,UAAU,CAAC,mBAAmB,CAAC,CAAC;QAEhC,OAAO,SAAS,GAAG,aAAa,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC9C,SAAS,EAAE,CAAC;YACZ,UAAU,CAAC,aAAa,SAAS,IAAI,aAAa,sBAAsB,CAAC,CAAC;YAE1E,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAC5C,KAAK;gBACL,UAAU,EAAE,IAAI;gBAChB,MAAM,EAAE,YAAY;gBACpB,KAAK,EAAE,KAAK;gBACZ,4DAA4D;gBAC5D,6DAA6D;gBAC7D,WAAW,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE;gBAC5B,QAAQ;aACT,CAAC,CAAC;YAEH,wBAAwB;YACxB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC;YAEhE,IAAI,QAAQ,CAAC,WAAW,KAAK,UAAU,EAAE,CAAC;gBACxC,UAAU,CAAC,+BAA+B,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;gBAClE,MAAM;YACR,CAAC;YAED,6CAA6C;YAC7C,MAAM,WAAW,GAAqC,EAAE,CAAC;YAEzD,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACrC,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU;oBAAE,SAAS;gBAExC,UAAU,CAAC,aAAa,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;gBACtC,MAAM,MAAM,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,KAA+B,CAAC,CAAC;gBAC9E,UAAU,CAAC,OAAO,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBAEpF,WAAW,CAAC,IAAI,CAAC;oBACf,IAAI,EAAE,aAAa;oBACnB,WAAW,EAAE,KAAK,CAAC,EAAE;oBACrB,OAAO,EAAE,MAAM;iBAChB,CAAC,CAAC;gBAEH,IAAI,QAAQ;oBAAE,MAAM;YACtB,CAAC;YAED,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;YACxD,CAAC;YAED,IAAI,QAAQ;gBAAE,MAAM;QACtB,CAAC;QAED,IAAI,SAAS,IAAI,aAAa,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC5C,UAAU,CAAC,oCAAoC,aAAa,sBAAsB,CAAC,CAAC;QACtF,CAAC;QAED,OAAO;YACL,UAAU,EAAE,YAAY;YACxB,UAAU,EAAE,SAAS;YACrB,WAAW;SACZ,CAAC;IACJ,CAAC;CAAA"}