@illuma-ai/agents 1.0.90 → 1.0.94

package/src/scripts/test_code_api.js

@@ -0,0 +1,263 @@
+// src/scripts/test_code_api.ts
+/**
+ * Direct test of the Code API to verify session file persistence.
+ * This bypasses the LLM and tests the API directly.
+ *
+ * Run with: npx ts-node -r dotenv/config src/scripts/test_code_api.ts
+ */
+import { config } from 'dotenv';
+config();
+import fetch from 'node-fetch';
+import { HttpsProxyAgent } from 'https-proxy-agent';
+const API_KEY = process.env.CODE_EXECUTOR_API_KEY ?? '';
+const BASE_URL = process.env.ILLUMA_CODE_BASEURL ?? 'https://api.illuma.ai/v1';
+const PROXY = process.env.PROXY;
+if (!API_KEY) {
+    console.error('CODE_EXECUTOR_API_KEY not set');
+    process.exit(1);
+}
+async function makeRequest(endpoint, body) {
+    const fetchOptions = {
+        method: body ? 'POST' : 'GET',
+        headers: {
+            'Content-Type': 'application/json',
+            'User-Agent': 'Illuma/1.0',
+            'X-API-Key': API_KEY,
+        },
+    };
+    if (body) {
+        fetchOptions.body = JSON.stringify(body);
+    }
+    if (PROXY) {
+        fetchOptions.agent = new HttpsProxyAgent(PROXY);
+    }
+    console.log(`\n>>> ${body ? 'POST' : 'GET'} ${endpoint}`);
+    if (body) {
+        console.log('Body:', JSON.stringify(body, null, 2));
+    }
+    const response = await fetch(endpoint, fetchOptions);
+    const result = await response.json();
+    console.log(`<<< Response (${response.status}):`);
+    console.log(JSON.stringify(result, null, 2));
+    if (!response.ok) {
+        throw new Error(`HTTP ${response.status}: ${JSON.stringify(result)}`);
+    }
+    return result;
+}
+async function testCodeAPI() {
+    console.log('='.repeat(60));
+    console.log('TEST 1: Create a file');
+    console.log('='.repeat(60));
+    const createCode = `
+import json
+
+config = {
+    "app_name": "TestApp",
+    "version": "1.0.0",
+    "debug": True
+}
+
+with open("/mnt/data/test_config.json", "w") as f:
+    json.dump(config, f, indent=2)
+
+with open("/mnt/data/test_config.json", "r") as f:
+    print(f.read())
+`;
+    const result1 = (await makeRequest(`${BASE_URL}/exec`, {
+        lang: 'py',
+        code: createCode,
+    }));
+    const sessionId = result1.session_id;
+    const files = result1.files ?? [];
+    console.log('\n--- Result Summary ---');
+    console.log('session_id:', sessionId);
+    console.log('files:', files);
+    console.log('stdout:', result1.stdout);
+    console.log('stderr:', result1.stderr);
+    if (!sessionId || files.length === 0) {
+        console.error('\n❌ No session_id or files returned!');
+        return;
+    }
+    // Check if files now include session_id (new API feature)
+    const hasSessionIdInFiles = files.some((f) => f.session_id != null);
+    console.log('\n✅ Files include session_id:', hasSessionIdInFiles);
+    console.log('\n' + '='.repeat(60));
+    console.log('TEST 2: Fetch files IMMEDIATELY (no delay - testing race condition fix)');
+    console.log('='.repeat(60));
+    const filesResult = (await makeRequest(`${BASE_URL}/files/${sessionId}?detail=full`));
+    console.log('\n--- Files in session (detail=full) ---');
+    for (const file of filesResult) {
+        console.log('File:', file.name);
+        console.log('  metadata:', file.metadata);
+    }
+    if (filesResult.length === 0) {
+        console.log('\n⚠️  Files endpoint returned empty - race condition may still exist');
+    }
+    else {
+        console.log('\n✅ Files available immediately!');
+    }
+    // Test new normalized detail level
+    console.log('\n' + '='.repeat(60));
+    console.log('TEST 2b: Fetch files with detail=normalized');
+    console.log('='.repeat(60));
+    const normalizedResult = (await makeRequest(`${BASE_URL}/files/${sessionId}?detail=normalized`));
+    console.log('\n--- Files in session (detail=normalized) ---');
+    console.log(JSON.stringify(normalizedResult, null, 2));
+    console.log('\n' + '='.repeat(60));
+    console.log('TEST 3: Read file IMMEDIATELY using files from original response');
+    console.log('='.repeat(60));
+    // Use files directly - if API returns session_id, use that; otherwise add it
+    const fileReferences = files.map((file) => ({
+        session_id: file.session_id ?? sessionId,
+        id: file.id,
+        name: file.name,
+    }));
+    console.log('\nFile references we will send:', JSON.stringify(fileReferences, null, 2));
+    const readCode = `
+import json
+
+with open("/mnt/data/test_config.json", "r") as f:
+    config = json.load(f)
+    print("Read config:")
+    print(json.dumps(config, indent=2))
+    print("Version:", config.get("version"))
+`;
+    const result2 = (await makeRequest(`${BASE_URL}/exec`, {
+        lang: 'py',
+        code: readCode,
+        files: fileReferences,
+    }));
+    console.log('\n--- Result Summary ---');
+    console.log('stdout:', result2.stdout);
+    console.log('stderr:', result2.stderr);
+    if (result2.stderr && result2.stderr.includes('FileNotFoundError')) {
+        console.log('\n❌ File not found! The file reference format might be wrong.');
+        // Try alternative format - just session_id
+        console.log('\n' + '='.repeat(60));
+        console.log('TEST 4: Try with just session_id in request');
+        console.log('='.repeat(60));
+        const result3 = (await makeRequest(`${BASE_URL}/exec`, {
+            lang: 'py',
+            code: readCode,
+            session_id: sessionId,
+        }));
+        console.log('\n--- Result Summary ---');
+        console.log('stdout:', result3.stdout);
+        console.log('stderr:', result3.stderr);
+    }
+    else {
+        console.log('\n✅ File read successfully!');
+    }
+    // ============================================================
+    // TEST 4: MODIFY the file (same filename) - tests editable files
+    // ============================================================
+    console.log('\n' + '='.repeat(60));
+    console.log('TEST 4: MODIFY file in-place (testing editable files feature)');
+    console.log('='.repeat(60));
+    const modifyCode = `
+import json
+
+# Read the existing file
+with open("/mnt/data/test_config.json", "r") as f:
+    config = json.load(f)
+
+print("Original config:")
+print(json.dumps(config, indent=2))
+
+# Modify the config
+config["version"] = "2.0.0"
+config["modified"] = True
+
+# Write BACK to the SAME filename (should work now!)
+with open("/mnt/data/test_config.json", "w") as f:
+    json.dump(config, f, indent=2)
+
+# Verify the write
+with open("/mnt/data/test_config.json", "r") as f:
+    updated = json.load(f)
+
+print("\\nUpdated config:")
+print(json.dumps(updated, indent=2))
+`;
+    const result3 = (await makeRequest(`${BASE_URL}/exec`, {
+        lang: 'py',
+        code: modifyCode,
+        files: fileReferences,
+    }));
+    console.log('\n--- Result Summary ---');
+    console.log('stdout:', result3.stdout);
+    console.log('stderr:', result3.stderr);
+    console.log('files:', JSON.stringify(result3.files, null, 2));
+    if (result3.stderr && result3.stderr.includes('Permission denied')) {
+        console.log('\n❌ Permission denied - files are still read-only!');
+    }
+    else if (result3.stderr && result3.stderr.includes('Error')) {
+        console.log('\n❌ Error modifying file:', result3.stderr);
+    }
+    else {
+        console.log('\n✅ File modified successfully!');
+        // Check for modified_from lineage
+        const modifiedFile = result3.files?.find((f) => f.name === 'test_config.json');
+        if (modifiedFile) {
+            console.log('\n--- Modified File Details ---');
+            console.log('  id:', modifiedFile.id);
+            console.log('  name:', modifiedFile.name);
+            console.log('  session_id:', modifiedFile.session_id);
+            if (modifiedFile.modified_from) {
+                console.log('  modified_from:', JSON.stringify(modifiedFile.modified_from));
+                console.log('\n✅ Lineage tracking working! File shows it was modified from previous session.');
+            }
+            else {
+                console.log('\n⚠️  No modified_from field - lineage tracking not present');
+            }
+        }
+        else {
+            console.log('\n⚠️  Modified file not found in response files array');
+        }
+    }
+    // ============================================================
+    // TEST 5: Verify modification persists in next execution
+    // ============================================================
+    console.log('\n' + '='.repeat(60));
+    console.log('TEST 5: Verify modified file can be read in subsequent execution');
+    console.log('='.repeat(60));
+    // Use the new file references from the modify response
+    const newFileRefs = (result3.files ?? []).map((file) => ({
+        session_id: file.session_id ?? result3.session_id,
+        id: file.id,
+        name: file.name,
+    }));
+    if (newFileRefs.length === 0) {
+        console.log('\n⚠️  No files returned from modification, skipping verification');
+    }
+    else {
+        console.log('\nUsing new file references:', JSON.stringify(newFileRefs, null, 2));
+        const verifyCode = `
+import json
+
+with open("/mnt/data/test_config.json", "r") as f:
+    config = json.load(f)
+
+print("Verified config:")
+print(json.dumps(config, indent=2))
+
+if config.get("version") == "2.0.0" and config.get("modified") == True:
+    print("\\n✅ Modification persisted correctly!")
+else:
+    print("\\n❌ Modification did NOT persist!")
+`;
+        const result4 = (await makeRequest(`${BASE_URL}/exec`, {
+            lang: 'py',
+            code: verifyCode,
+            files: newFileRefs,
+        }));
+        console.log('\n--- Result Summary ---');
+        console.log('stdout:', result4.stdout);
+        console.log('stderr:', result4.stderr);
+    }
+}
+testCodeAPI().catch((err) => {
+    console.error('Error:', err);
+    process.exit(1);
+});
+//# sourceMappingURL=test_code_api.js.map

package/src/scripts/test_code_api.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test_code_api.js","sourceRoot":"","sources":["test_code_api.ts"],"names":[],"mappings":"AAAA,+BAA+B;AAC/B;;;;;GAKG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AAET,OAAO,KAAsB,MAAM,YAAY,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEpD,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,IAAI,EAAE,CAAC;AACxD,MAAM,QAAQ,GACZ,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,0BAA0B,CAAC;AAChE,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC;AAEhC,IAAI,CAAC,OAAO,EAAE,CAAC;IACb,OAAO,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC;IAC/C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC;AAyBD,KAAK,UAAU,WAAW,CACxB,QAAgB,EAChB,IAA8B;IAE9B,MAAM,YAAY,GAAgB;QAChC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK;QAC7B,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,YAAY,EAAE,YAAY;YAC1B,WAAW,EAAE,OAAO;SACrB;KACF,CAAC;IAEF,IAAI,IAAI,EAAE,CAAC;QACT,YAAY,CAAC,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,IAAI,KAAK,EAAE,CAAC;QACV,YAAY,CAAC,KAAK,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;IAClD,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,IAAI,QAAQ,EAAE,CAAC,CAAC;IAC1D,IAAI,IAAI,EAAE,CAAC;QACT,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACtD,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;IACrD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAErC,OAAO,CAAC,GAAG,CAAC,iBAAiB,QAAQ,CAAC,MAAM,IAAI,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IAE7C,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,QAAQ,QAAQ,CAAC,MAAM,KAAK,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACxE,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,KAAK,UAAU,WAAW;IACxB,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5B,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,MAAM,UAAU,GAAG;;;;;;;;;;;;;;CAcpB,CAAC;IAEA,MAAM,OAAO,GAAG,CAAC,MAAM,WAAW,CAAC,GAAG,QAAQ,OAAO,EAAE;QACrD,IAAI,EAAE,IAAI;QACV,IAAI,EAAE,UAAU;KACjB,CAAC,CAAe,CAAC;IAElB,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC;IACrC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;IAElC,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;IACxC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;IACtC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAC7B,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IACvC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IAEvC,IAAI,CAAC,SAAS,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrC,OAAO,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;QACtD,OAAO;IACT,CAAC;IAED,0DAA0D;IAC1D,MAAM,mBAAmB,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,IAAI,IAAI,CAAC,CAAC;IACpE,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE,mBAAmB,CAAC,CAAC;IAElE,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CACT,yEAAyE,CAC1E,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,MAAM,WAAW,GAAG,CAAC,MAAM,WAAW,CACpC,GAAG,QAAQ,UAAU,SAAS,cAAc,CAC7C,CAAe,CAAC;IAEjB,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;IACxD,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;QAC/B,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QAChC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC5C,CAAC;IAED,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,CAAC,GAAG,CACT,sEAAsE,CACvE,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;IAClD,CAAC;IAED,mCAAmC;IACnC,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;IAC3D,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,MAAM,gBAAgB,GAAG,CAAC,MAAM,WAAW,CACzC,GAAG,QAAQ,UAAU,SAAS,oBAAoB,CACnD,CAAc,CAAC;IAEhB,OAAO,CAAC,GAAG,CAAC,gDAAgD,CAAC,CAAC;IAC9D,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IAEvD,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CACT,kEAAkE,CACnE,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,6EAA6E;IAC7E,MAAM,cAAc,GAAc,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACrD,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,SAAS;QACxC,EAAE,EAAE,IAAI,CAAC,EAAE;QACX,IAAI,EAAE,IAAI,CAAC,IAAI;KAChB,CAAC,CAAC,CAAC;IAEJ,OAAO,CAAC,GAAG,CACT,iCAAiC,EACjC,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,CACxC,CAAC;IAEF,MAAM,QAAQ,GAAG;;;;;;;;CAQlB,CAAC;IAEA,MAAM,OAAO,GAAG,CAAC,MAAM,WAAW,CAAC,GAAG,QAAQ,OAAO,EAAE;QACrD,IAAI,EAAE,IAAI;QACV,IAAI,EAAE,QAAQ;QACd,KAAK,EAAE,cAAc;KACtB,CAAC,CAAe,CAAC;IAElB,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;IACxC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IACvC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IAEvC,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;QACnE,OAAO,CAAC,GAAG,CACT,+DAA+D,CAChE,CAAC;QAEF,2CAA2C;QAC3C,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;QAC3D,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAE5B,MAAM,OAAO,GAAG,CAAC,MAAM,WAAW,CAAC,GAAG,QAAQ,OAAO,EAAE;YACrD,IAAI,EAAE,IAAI;YACV,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,SAAS;SACtB,CAAC,CAAe,CAAC;QAElB,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;QACxC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACvC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IACzC,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;IAC7C,CAAC;IAED,+DAA+D;IAC/D,iEAAiE;IACjE,+DAA+D;IAC/D,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CAAC,+DAA+D,CAAC,CAAC;IAC7E,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,MAAM,UAAU,GAAG;;;;;;;;;;;;;;;;;;;;;;;;CAwBpB,CAAC;IAEA,MAAM,OAAO,GAAG,CAAC,MAAM,WAAW,CAAC,GAAG,QAAQ,OAAO,EAAE;QACrD,IAAI,EAAE,IAAI;QACV,IAAI,EAAE,UAAU;QAChB,KAAK,EAAE,cAAc;KACtB,CAAC,CAAe,CAAC;IAElB,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;IACxC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IACvC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IACvC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IAE9D,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;QACnE,OAAO,CAAC,GAAG,CAAC,oDAAoD,CAAC,CAAC;IACpE,CAAC;SAAM,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QAC9D,OAAO,CAAC,GAAG,CAAC,2BAA2B,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IAC3D,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;QAE/C,kCAAkC;QAClC,MAAM,YAAY,GAAG,OAAO,CAAC,KAAK,EAAE,IAAI,CACtC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,kBAAkB,CACrC,CAAC;QACF,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;YAC/C,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,YAAY,CAAC,EAAE,CAAC,CAAC;YACtC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,YAAY,CAAC,UAAU,CAAC,CAAC;YACtD,IAAI,YAAY,CAAC,aAAa,EAAE,CAAC;gBAC/B,OAAO,CAAC,GAAG,CACT,kBAAkB,EAClB,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,aAAa,CAAC,CAC3C,CAAC;gBACF,OAAO,CAAC,GAAG,CACT,iFAAiF,CAClF,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CACT,6DAA6D,CAC9D,CAAC;YACJ,CAAC;QACH,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,uDAAuD,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAED,+DAA+D;IAC/D,yDAAyD;IACzD,+DAA+D;IAC/D,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CACT,kEAAkE,CACnE,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,uDAAuD;IACvD,MAAM,WAAW,GAAc,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAClE,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU;QACjD,EAAE,EAAE,IAAI,CAAC,EAAE;QACX,IAAI,EAAE,IAAI,CAAC,IAAI;KAChB,CAAC,CAAC,CAAC;IAEJ,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,CAAC,GAAG,CACT,kEAAkE,CACnE,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CACT,8BAA8B,EAC9B,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC,CACrC,CAAC;QAEF,MAAM,UAAU,GAAG;;;;;;;;;;;;;CAatB,CAAC;QAEE,MAAM,OAAO,GAAG,CAAC,MAAM,WAAW,CAAC,GAAG,QAAQ,OAAO,EAAE;YACrD,IAAI,EAAE,IAAI;YACV,IAAI,EAAE,UAAU;YAChB,KAAK,EAAE,WAAW;SACnB,CAAC,CAAe,CAAC;QAElB,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;QACxC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACvC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IACzC,CAAC;AACH,CAAC;AAED,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IAC1B,OAAO,CAAC,KAAK,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IAC7B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/src/scripts/thinking-bedrock.js

@@ -0,0 +1,128 @@
+// src/scripts/thinking-bedrock.ts
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { createCodeExecutionTool } from '@/tools/CodeExecutor';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+import { GraphEvents, Providers } from '@/common';
+import { getLLMConfig } from '@/utils/llmConfig';
+import { getArgs } from '@/scripts/args';
+import { Run } from '@/run';
+const conversationHistory = [];
+let _contentParts = [];
+const collectedUsage = [];
+async function testBedrockThinking() {
+    const { userName } = await getArgs();
+    const instructions = `You are a helpful AI assistant for ${userName}. When answering questions, be thorough in your reasoning.`;
+    const { contentParts, aggregateContent } = createContentAggregator();
+    _contentParts = contentParts;
+    // Set up event handlers
+    const customHandlers = {
+        [GraphEvents.TOOL_END]: new ToolEndHandler(),
+        [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(collectedUsage),
+        [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
+        [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+            handle: (event, data) => {
+                console.log('====== ON_RUN_STEP_COMPLETED ======');
+                aggregateContent({
+                    event,
+                    data: data,
+                });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP]: {
+            handle: (event, data) => {
+                aggregateContent({ event, data });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP_DELTA]: {
+            handle: (event, data) => {
+                aggregateContent({ event, data });
+            },
+        },
+        [GraphEvents.ON_MESSAGE_DELTA]: {
+            handle: (event, data) => {
+                aggregateContent({ event, data });
+            },
+        },
+        [GraphEvents.ON_REASONING_DELTA]: {
+            handle: (event, data) => {
+                aggregateContent({ event, data });
+            },
+        },
+    };
+    const baseLlmConfig = getLLMConfig(Providers.BEDROCK);
+    // Enable thinking with token budget for Bedrock
+    const llmConfig = {
+        ...baseLlmConfig,
+        model: 'us.anthropic.claude-3-7-sonnet-20250219-v1:0',
+        maxTokens: 5000,
+        additionalModelRequestFields: {
+            thinking: { type: 'enabled', budget_tokens: 2000 },
+        },
+    };
+    const run = await Run.create({
+        runId: 'test-bedrock-thinking-id',
+        graphConfig: {
+            instructions,
+            type: 'standard',
+            tools: [createCodeExecutionTool()],
+            llmConfig,
+        },
+        returnContent: true,
+        skipCleanup: true,
+        customHandlers: customHandlers,
+    });
+    const config = {
+        configurable: {
+            thread_id: 'bedrock-thinking-test-thread',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    // Test 1: Regular thinking mode
+    console.log('\n\nTest 1: Bedrock Regular thinking mode');
+    const userMessage1 = `Please print 'hello world' in python`;
+    conversationHistory.push(new HumanMessage(userMessage1));
+    console.log('Running first query with Bedrock thinking enabled...');
+    const firstInputs = { messages: [...conversationHistory] };
+    await run.processStream(firstInputs, config);
+    // Extract and display thinking blocks
+    const finalMessages = run.getRunMessages();
+    console.log('\n\nFinal messages after Test 1:');
+    console.dir(finalMessages, { depth: null });
+    // Test 2: Try multi-turn conversation
+    console.log('\n\nTest 2: Multi-turn conversation with Bedrock thinking enabled');
+    const userMessage2 = `Given your previous analysis, what would be the most significant technical challenges in making this transition?`;
+    conversationHistory.push(new HumanMessage(userMessage2));
+    console.log('Running second query with Bedrock thinking enabled...');
+    const secondInputs = { messages: [...conversationHistory] };
+    await run.processStream(secondInputs, config);
+    // Display thinking blocks for second response
+    const finalMessages2 = run.getRunMessages();
+    console.log('\n\nBedrock thinking feature test completed!');
+    console.dir(finalMessages2, { depth: null });
+    console.log('\n\nContent parts:');
+    console.dir(_contentParts, { depth: null });
+}
+process.on('unhandledRejection', (reason, promise) => {
+    console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+    console.log('Conversation history:');
+    console.dir(conversationHistory, { depth: null });
+    console.log('Content parts:');
+    console.dir(_contentParts, { depth: null });
+    process.exit(1);
+});
+process.on('uncaughtException', (err) => {
+    console.error('Uncaught Exception:', err);
+});
+testBedrockThinking().catch((err) => {
+    console.error(err);
+    console.log('Conversation history:');
+    console.dir(conversationHistory, { depth: null });
+    console.log('Content parts:');
+    console.dir(_contentParts, { depth: null });
+    process.exit(1);
+});
+//# sourceMappingURL=thinking-bedrock.js.map

package/src/scripts/thinking-bedrock.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"thinking-bedrock.js","sourceRoot":"","sources":["thinking-bedrock.ts"],"names":[],"mappings":"AAAA,kCAAkC;AAClC,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AACT,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAGrE,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAC3D,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAE5B,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAC9C,IAAI,aAAa,GAA8B,EAAE,CAAC;AAClD,MAAM,cAAc,GAAoB,EAAE,CAAC;AAE3C,KAAK,UAAU,mBAAmB;IAChC,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;IACrC,MAAM,YAAY,GAAG,sCAAsC,QAAQ,4DAA4D,CAAC;IAChI,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IACrE,aAAa,GAAG,YAAyC,CAAC;IAE1D,wBAAwB;IACxB,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,CAAC,cAAc,CAAC;QACjE,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,sBAAsB,EAAE;QAC7D,CAAC,WAAW,CAAC,qBAAqB,CAAC,EAAE;YACnC,MAAM,EAAE,CACN,KAAwC,EACxC,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,qCAAqC,CAAC,CAAC;gBACnD,gBAAgB,CAAC;oBACf,KAAK;oBACL,IAAI,EAAE,IAA6C;iBACpD,CAAC,CAAC;YACL,CAAC;SACF;QACD,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;YACzB,MAAM,EAAE,CAAC,KAA8B,EAAE,IAAe,EAAE,EAAE;gBAC1D,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACpC,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;YAC/B,MAAM,EAAE,CACN,KAAoC,EACpC,IAAyB,EACzB,EAAE;gBACF,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACpC,CAAC;SACF;QACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CACN,KAAmC,EACnC,IAAyB,EACzB,EAAE;gBACF,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACpC,CAAC;SACF;QACD,CAAC,WAAW,CAAC,kBAAkB,CAAC,EAAE;YAChC,MAAM,EAAE,CACN,KAAqC,EACrC,IAA2B,EAC3B,EAAE;gBACF,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACpC,CAAC;SACF;KACF,CAAC;IAEF,MAAM,aAAa,GAAG,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAEtD,gDAAgD;IAChD,MAAM,SAAS,GAAG;QAChB,GAAG,aAAa;QAChB,KAAK,EAAE,8CAA8C;QACrD,SAAS,EAAE,IAAI;QACf,4BAA4B,EAAE;YAC5B,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,aAAa,EAAE,IAAI,EAAE;SACnD;KACF,CAAC;IAEF,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,0BAA0B;QACjC,WAAW,EAAE;YACX,YAAY;YACZ,IAAI,EAAE,UAAU;YAChB,KAAK,EAAE,CAAC,uBAAuB,EAAE,CAAC;YAClC,SAAS;SACV;QACD,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,cAAc,EAAE,cAA+C;KAChE,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG;QACb,YAAY,EAAE;YACZ,SAAS,EAAE,8BAA8B;SAC1C;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,gCAAgC;IAChC,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;IACzD,MAAM,YAAY,GAAG,sCAAsC,CAAC;IAC5D,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IAEzD,OAAO,CAAC,GAAG,CAAC,sDAAsD,CAAC,CAAC;IACpE,MAAM,WAAW,GAAG,EAAE,QAAQ,EAAE,CAAC,GAAG,mBAAmB,CAAC,EAAE,CAAC;IAC3D,MAAM,GAAG,CAAC,aAAa,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;IAE7C,sCAAsC;IACtC,MAAM,aAAa,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAC3C,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;IAChD,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAE5C,sCAAsC;IACtC,OAAO,CAAC,GAAG,CACT,mEAAmE,CACpE,CAAC;IACF,MAAM,YAAY,GAAG,kHAAkH,CAAC;IACxI,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IAEzD,OAAO,CAAC,GAAG,CAAC,uDAAuD,CAAC,CAAC;IACrE,MAAM,YAAY,GAAG,EAAE,QAAQ,EAAE,CAAC,GAAG,mBAAmB,CAAC,EAAE,CAAC;IAC5D,MAAM,GAAG,CAAC,aAAa,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;IAE9C,8CAA8C;IAC9C,MAAM,cAAc,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAC5C,OAAO,CAAC,GAAG,CAAC,8CAA8C,CAAC,CAAC;IAC5D,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAE7C,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IAClC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;AAC9C,CAAC;AAED,OAAO,CAAC,EAAE,CAAC,oBAAoB,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;IACnD,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;IACrE,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC9B,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,OAAO,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC,GAAG,EAAE,EAAE;IACtC,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC;AAC5C,CAAC,CAAC,CAAC;AAEH,mBAAmB,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IAClC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACnB,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC9B,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/src/scripts/thinking-bedrock.ts

@@ -91,6 +91,7 @@ async function testBedrockThinking(): Promise<void> {
       llmConfig,
     },
     returnContent: true,
+    skipCleanup: true,
     customHandlers: customHandlers as t.RunConfig['customHandlers'],
   });
 

package/src/scripts/thinking-vertexai.js

@@ -0,0 +1,130 @@
+// src/scripts/thinking-vertexai.ts
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+import { GraphEvents, Providers } from '@/common';
+import { getLLMConfig } from '@/utils/llmConfig';
+import { getArgs } from '@/scripts/args';
+import { Run } from '@/run';
+const conversationHistory = [];
+let _contentParts = [];
+const collectedUsage = [];
+async function testVertexAIThinking() {
+    const { userName } = await getArgs();
+    const instructions = `You are a helpful AI assistant for ${userName}. When answering questions, be thorough in your reasoning.`;
+    const { contentParts, aggregateContent } = createContentAggregator();
+    _contentParts = contentParts;
+    // Set up event handlers
+    const customHandlers = {
+        [GraphEvents.TOOL_END]: new ToolEndHandler(),
+        [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(collectedUsage),
+        [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
+        [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+            handle: (event, data) => {
+                console.log('====== ON_RUN_STEP_COMPLETED ======');
+                aggregateContent({
+                    event,
+                    data: data,
+                });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP]: {
+            handle: (event, data) => {
+                aggregateContent({ event, data });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP_DELTA]: {
+            handle: (event, data) => {
+                aggregateContent({ event, data });
+            },
+        },
+        [GraphEvents.ON_MESSAGE_DELTA]: {
+            handle: (event, data) => {
+                aggregateContent({ event, data });
+            },
+        },
+        [GraphEvents.ON_REASONING_DELTA]: {
+            handle: (event, data) => {
+                console.log('[ON_REASONING_DELTA]', JSON.stringify(data.delta.content?.[0]).slice(0, 100));
+                aggregateContent({ event, data });
+            },
+        },
+    };
+    const baseLlmConfig = getLLMConfig(Providers.VERTEXAI);
+    const llmConfig = {
+        ...baseLlmConfig,
+        model: 'gemini-3-flash-preview',
+        location: 'global',
+        streaming: true,
+        streamUsage: true,
+        thinkingConfig: {
+            thinkingLevel: 'HIGH',
+            includeThoughts: true,
+        },
+    };
+    const run = await Run.create({
+        runId: 'test-vertexai-thinking-id',
+        graphConfig: {
+            instructions,
+            type: 'standard',
+            llmConfig,
+        },
+        returnContent: true,
+        skipCleanup: true,
+        customHandlers: customHandlers,
+    });
+    const streamConfig = {
+        configurable: {
+            thread_id: 'vertexai-thinking-test-thread',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    // Test 1: Regular thinking mode
+    console.log('\n\nTest 1: Vertex AI thinking mode with thinkingLevel=HIGH');
+    const userMessage1 = 'How many r\'s are in the word "strawberry"? Think carefully.';
+    conversationHistory.push(new HumanMessage(userMessage1));
+    console.log('Running first query with Vertex AI thinking enabled...');
+    const firstInputs = { messages: [...conversationHistory] };
+    await run.processStream(firstInputs, streamConfig);
+    // Extract and display results
+    const finalMessages = run.getRunMessages();
+    console.log('\n\nFinal messages after Test 1:');
+    console.dir(finalMessages, { depth: null });
+    // Test 2: Multi-turn conversation
+    console.log('\n\nTest 2: Multi-turn conversation with Vertex AI thinking enabled');
+    const userMessage2 = 'Now count the number of letters in "Mississippi". Explain step by step.';
+    conversationHistory.push(new HumanMessage(userMessage2));
+    console.log('Running second query with Vertex AI thinking enabled...');
+    const secondInputs = { messages: [...conversationHistory] };
+    await run.processStream(secondInputs, streamConfig);
+    const finalMessages2 = run.getRunMessages();
+    console.log('\n\nVertex AI thinking feature test completed!');
+    console.dir(finalMessages2, { depth: null });
+    console.log('\n\nContent parts:');
+    console.dir(_contentParts, { depth: null });
+    console.log('\n\nCollected usage:');
+    console.dir(collectedUsage, { depth: null });
+}
+process.on('unhandledRejection', (reason, promise) => {
+    console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+    console.log('Conversation history:');
+    console.dir(conversationHistory, { depth: null });
+    console.log('Content parts:');
+    console.dir(_contentParts, { depth: null });
+    process.exit(1);
+});
+process.on('uncaughtException', (err) => {
+    console.error('Uncaught Exception:', err);
+});
+testVertexAIThinking().catch((err) => {
+    console.error(err);
+    console.log('Conversation history:');
+    console.dir(conversationHistory, { depth: null });
+    console.log('Content parts:');
+    console.dir(_contentParts, { depth: null });
+    process.exit(1);
+});
+//# sourceMappingURL=thinking-vertexai.js.map

package/src/scripts/thinking-vertexai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"thinking-vertexai.js","sourceRoot":"","sources":["thinking-vertexai.ts"],"names":[],"mappings":"AAAA,mCAAmC;AACnC,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AACT,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAGrE,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAC3D,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAE5B,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAC9C,IAAI,aAAa,GAA8B,EAAE,CAAC;AAClD,MAAM,cAAc,GAAoB,EAAE,CAAC;AAE3C,KAAK,UAAU,oBAAoB;IACjC,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;IACrC,MAAM,YAAY,GAAG,sCAAsC,QAAQ,4DAA4D,CAAC;IAChI,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IACrE,aAAa,GAAG,YAAyC,CAAC;IAE1D,wBAAwB;IACxB,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,CAAC,cAAc,CAAC;QACjE,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,sBAAsB,EAAE;QAC7D,CAAC,WAAW,CAAC,qBAAqB,CAAC,EAAE;YACnC,MAAM,EAAE,CACN,KAAwC,EACxC,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,qCAAqC,CAAC,CAAC;gBACnD,gBAAgB,CAAC;oBACf,KAAK;oBACL,IAAI,EAAE,IAA6C;iBACpD,CAAC,CAAC;YACL,CAAC;SACF;QACD,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;YACzB,MAAM,EAAE,CAAC,KAA8B,EAAE,IAAe,EAAE,EAAE;gBAC1D,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACpC,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;YAC/B,MAAM,EAAE,CACN,KAAoC,EACpC,IAAyB,EACzB,EAAE;gBACF,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACpC,CAAC;SACF;QACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CACN,KAAmC,EACnC,IAAyB,EACzB,EAAE;gBACF,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACpC,CAAC;SACF;QACD,CAAC,WAAW,CAAC,kBAAkB,CAAC,EAAE;YAChC,MAAM,EAAE,CACN,KAAqC,EACrC,IAA2B,EAC3B,EAAE;gBACF,OAAO,CAAC,GAAG,CACT,sBAAsB,EACtB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CACtD,CAAC;gBACF,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACpC,CAAC;SACF;KACF,CAAC;IAEF,MAAM,aAAa,GAAG,YAAY,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IAEvD,MAAM,SAAS,GAAG;QAChB,GAAG,aAAa;QAChB,KAAK,EAAE,wBAAwB;QAC/B,QAAQ,EAAE,QAAQ;QAClB,SAAS,EAAE,IAAI;QACf,WAAW,EAAE,IAAI;QACjB,cAAc,EAAE;YACd,aAAa,EAAE,MAAM;YACrB,eAAe,EAAE,IAAI;SACtB;KACF,CAAC;IAEF,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,2BAA2B;QAClC,WAAW,EAAE;YACX,YAAY;YACZ,IAAI,EAAE,UAAU;YAChB,SAAS;SACV;QACD,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,cAAc,EAAE,cAA+C;KAChE,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG;QACnB,YAAY,EAAE;YACZ,SAAS,EAAE,+BAA+B;SAC3C;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,gCAAgC;IAChC,OAAO,CAAC,GAAG,CAAC,6DAA6D,CAAC,CAAC;IAC3E,MAAM,YAAY,GAChB,8DAA8D,CAAC;IACjE,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IAEzD,OAAO,CAAC,GAAG,CAAC,wDAAwD,CAAC,CAAC;IACtE,MAAM,WAAW,GAAG,EAAE,QAAQ,EAAE,CAAC,GAAG,mBAAmB,CAAC,EAAE,CAAC;IAC3D,MAAM,GAAG,CAAC,aAAa,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;IAEnD,8BAA8B;IAC9B,MAAM,aAAa,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAC3C,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;IAChD,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAE5C,kCAAkC;IAClC,OAAO,CAAC,GAAG,CACT,qEAAqE,CACtE,CAAC;IACF,MAAM,YAAY,GAChB,yEAAyE,CAAC;IAC5E,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IAEzD,OAAO,CAAC,GAAG,CAAC,yDAAyD,CAAC,CAAC;IACvE,MAAM,YAAY,GAAG,EAAE,QAAQ,EAAE,CAAC,GAAG,mBAAmB,CAAC,EAAE,CAAC;IAC5D,MAAM,GAAG,CAAC,aAAa,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;IAEpD,MAAM,cAAc,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAC5C,OAAO,CAAC,GAAG,CAAC,gDAAgD,CAAC,CAAC;IAC9D,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAE7C,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IAClC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAE5C,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;IACpC,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;AAC/C,CAAC;AAED,OAAO,CAAC,EAAE,CAAC,oBAAoB,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;IACnD,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;IACrE,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC9B,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,OAAO,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC,GAAG,EAAE,EAAE;IACtC,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC;AAC5C,CAAC,CAAC,CAAC;AAEH,oBAAoB,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACnC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACnB,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC9B,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}