@illuma-ai/agents 1.0.89 → 1.0.93

package/src/scripts/abort.js

@@ -0,0 +1,127 @@
+/* eslint-disable no-console */
+// src/scripts/cli.ts
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { ToolEndHandler } from '@/events';
+import { getArgs } from '@/scripts/args';
+import { Run } from '@/run';
+import { GraphEvents } from '@/common';
+import { getLLMConfig } from '@/utils/llmConfig';
+const conversationHistory = [];
+async function testStandardStreaming() {
+    const { userName, location, provider, currentDate } = await getArgs();
+    const { contentParts, aggregateContent } = createContentAggregator();
+    const controller = new AbortController();
+    const customHandlers = {
+        [GraphEvents.TOOL_END]: new ToolEndHandler(),
+        [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) => {
+                console.log('====== ON_RUN_STEP ======');
+                console.dir(data, { depth: null });
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP_DELTA]: {
+            handle: (event, data) => {
+                console.log('====== ON_RUN_STEP_DELTA ======');
+                console.dir(data, { depth: null });
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_MESSAGE_DELTA]: {
+            handle: (event, data) => {
+                console.log('====== ON_MESSAGE_DELTA ======');
+                console.dir(data, { depth: null });
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.TOOL_START]: {
+            handle: (_event, data, metadata) => {
+                console.log('====== TOOL_START ======');
+            },
+        },
+    };
+    const llmConfig = getLLMConfig(provider);
+    const signal = controller.signal;
+    const run = await Run.create({
+        runId: 'test-run-id',
+        graphConfig: {
+            type: 'standard',
+            signal,
+            llmConfig,
+            tools: [],
+            instructions: 'You are a friendly AI assistant. Always address the user by their name.',
+            additional_instructions: `The user's name is ${userName} and they are located in ${location}.`,
+        },
+        customHandlers,
+    });
+    const config = {
+        configurable: {
+            provider,
+            thread_id: 'conversation-num-1',
+        },
+        signal,
+        streamMode: 'values',
+        version: 'v2',
+    };
+    console.log('Test 1: Weather query (content parts test)');
+    const userMessage = `
+  Make a search for the weather in ${location} today, which is ${currentDate}.
+  Make sure to always refer to me by name, which is ${userName}.
+  After giving me a thorough summary, tell me a joke about the weather forecast we went over.
+  `;
+    conversationHistory.push(new HumanMessage(userMessage));
+    const inputs = {
+        messages: conversationHistory,
+    };
+    // Set a timeout to abort the operation after 5 seconds
+    setTimeout(() => {
+        controller.abort();
+        console.log('Operation aborted');
+        console.log('Current content parts:');
+        console.dir(contentParts, { depth: null });
+    }, 4000);
+    try {
+        const finalContentParts = await run.processStream(inputs, config);
+        const finalMessages = run.getRunMessages();
+        if (finalMessages) {
+            conversationHistory.push(...finalMessages);
+            console.dir(conversationHistory, { depth: null });
+        }
+        console.dir(finalContentParts, { depth: null });
+    }
+    catch (error) {
+        if (error?.name === 'AbortError') {
+            console.log('Operation was aborted');
+        }
+        else {
+            console.error('An error occurred:', error);
+        }
+    }
+    console.log('\n\n====================\n\n');
+    console.dir(contentParts, { depth: null });
+}
+process.on('unhandledRejection', (reason, promise) => {
+    console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+    console.log('Conversation history:');
+    process.exit(1);
+});
+testStandardStreaming().catch((err) => {
+    console.error(err);
+    console.log('Conversation history:');
+    console.dir(conversationHistory, { depth: null });
+    process.exit(1);
+});
+//# sourceMappingURL=abort.js.map

package/src/scripts/abort.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"abort.js","sourceRoot":"","sources":["abort.ts"],"names":[],"mappings":"AAAA,+BAA+B;AAC/B,qBAAqB;AACrB,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AACT,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAErE,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAE1C,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,WAAW,EAAY,MAAM,UAAU,CAAC;AACjD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjD,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAE9C,KAAK,UAAU,qBAAqB;IAClC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;IACtE,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IACrE,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IAEzC,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,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,CACN,KAA8B,EAC9B,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;gBACzC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAAiB,EAAE,CAAC,CAAC;YACvD,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;YAC/B,MAAM,EAAE,CACN,KAAoC,EACpC,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;gBAC/C,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;QACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CACN,KAAmC,EACnC,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;gBAC9C,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;QACD,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE;YACxB,MAAM,EAAE,CACN,MAAc,EACd,IAAuB,EACvB,QAAkC,EAC5B,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;YAC1C,CAAC;SACF;KACF,CAAC;IAEF,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IACzC,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;IACjC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,aAAa;QACpB,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,MAAM;YACN,SAAS;YACT,KAAK,EAAE,EAAE;YACT,YAAY,EACV,yEAAyE;YAC3E,uBAAuB,EAAE,sBAAsB,QAAQ,4BAA4B,QAAQ,GAAG;SAC/F;QACD,cAAc;KACf,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG;QACb,YAAY,EAAE;YACZ,QAAQ;YACR,SAAS,EAAE,oBAAoB;SAChC;QACD,MAAM;QACN,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;IAE1D,MAAM,WAAW,GAAG;qCACe,QAAQ,oBAAoB,WAAW;sDACtB,QAAQ;;GAE3D,CAAC;IAEF,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC;IAExD,MAAM,MAAM,GAAG;QACb,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;IAEF,uDAAuD;IACvD,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,CAAC,KAAK,EAAE,CAAC;QACnB,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QACjC,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;QACtC,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC7C,CAAC,EAAE,IAAI,CAAC,CAAC;IAET,IAAI,CAAC;QACH,MAAM,iBAAiB,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAClE,MAAM,aAAa,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;QAC3C,IAAI,aAAa,EAAE,CAAC;YAClB,mBAAmB,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;YAC3C,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACpD,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAClD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAK,KAAe,EAAE,IAAI,KAAK,YAAY,EAAE,CAAC;YAC5C,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;QACvC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,KAAK,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAC5C,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;AAC7C,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,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,qBAAqB,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACpC,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,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/src/scripts/ant_web_search.js

@@ -0,0 +1,130 @@
+/* eslint-disable no-console */
+// src/scripts/cli.ts
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+import { getArgs } from '@/scripts/args';
+import { Run } from '@/run';
+import { GraphEvents, Providers } from '@/common';
+import { getLLMConfig } from '@/utils/llmConfig';
+const conversationHistory = [];
+let _contentParts = [];
+async function testStandardStreaming() {
+    const { userName, location, currentDate } = await getArgs();
+    const { contentParts, aggregateContent } = createContentAggregator();
+    _contentParts = contentParts;
+    const customHandlers = {
+        [GraphEvents.TOOL_END]: new ToolEndHandler(),
+        [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
+        [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
+        [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+            handle: (event, data) => {
+                console.log('====== ON_RUN_STEP_COMPLETED ======');
+                // console.dir(data, { depth: null });
+                aggregateContent({
+                    event,
+                    data: data,
+                });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP]: {
+            handle: (event, data) => {
+                console.log('====== ON_RUN_STEP ======');
+                console.dir(data, { depth: null });
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP_DELTA]: {
+            handle: (event, data) => {
+                console.log('====== ON_RUN_STEP_DELTA ======');
+                console.dir(data, { depth: null });
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_MESSAGE_DELTA]: {
+            handle: (event, data) => {
+                console.log('====== ON_MESSAGE_DELTA ======');
+                console.dir(data, { depth: null });
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.TOOL_START]: {
+            handle: (_event, data, metadata) => {
+                console.log('====== TOOL_START ======');
+                // console.dir(data, { depth: null });
+            },
+        },
+    };
+    const llmConfig = getLLMConfig(Providers.ANTHROPIC);
+    llmConfig.model = 'claude-haiku-4-5';
+    const run = await Run.create({
+        runId: 'test-run-id',
+        graphConfig: {
+            type: 'standard',
+            llmConfig,
+            tools: [
+                {
+                    type: 'web_search_20250305',
+                    name: 'web_search',
+                    max_uses: 5,
+                },
+            ],
+            instructions: 'You are a friendly AI assistant.',
+            // additional_instructions: `Always address the user by their name. The user's name is ${userName} and they are located in ${location}.`,
+        },
+        returnContent: true,
+        skipCleanup: true,
+        customHandlers,
+    });
+    const config = {
+        configurable: {
+            provider: Providers.ANTHROPIC,
+            thread_id: 'conversation-num-1',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    console.log('Test 1: Search query (search tool test)');
+    // const userMessage = `
+    // Make a search for the weather in ${location} today, which is ${currentDate}.
+    // Before making the search, please let me know what you're about to do, then immediately start searching without hesitation.
+    // Make sure to always refer to me by name, which is ${userName}.
+    // After giving me a thorough summary, tell me a joke about the weather forecast we went over.
+    // `;
+    // const userMessage = 'Are massage guns good?';
+    // const userMessage = 'What is functional programming?';
+    const userMessage = "Get me today's trending news.";
+    // const userMessage = "search recent italy earthquake volcano activity";
+    // const userMessage =
+    //   "use 'Trump' as the exact search query and tell me what you find.";
+    conversationHistory.push(new HumanMessage(userMessage));
+    const inputs = {
+        messages: conversationHistory,
+    };
+    const finalContentParts = await run.processStream(inputs, config);
+    const finalMessages = run.getRunMessages();
+    if (finalMessages) {
+        conversationHistory.push(...finalMessages);
+        console.dir(conversationHistory, { depth: null });
+    }
+    // console.dir(finalContentParts, { depth: null });
+    console.log('\n\n====================\n\n');
+    // console.dir(contentParts, { depth: null });
+}
+process.on('unhandledRejection', (reason, promise) => {
+    console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+    console.log('Content Parts:');
+    console.dir(_contentParts, { depth: null });
+    process.exit(1);
+});
+testStandardStreaming().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=ant_web_search.js.map

package/src/scripts/ant_web_search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ant_web_search.js","sourceRoot":"","sources":["ant_web_search.ts"],"names":[],"mappings":"AAAA,+BAA+B;AAC/B,qBAAqB;AACrB,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AACT,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAErE,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAE3D,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,WAAW,EAAY,SAAS,EAAE,MAAM,UAAU,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjD,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAC9C,IAAI,aAAa,GAA4C,EAAE,CAAC;AAChE,KAAK,UAAU,qBAAqB;IAClC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;IAC5D,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IACrE,aAAa,GAAG,YAAY,CAAC;IAC7B,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,EAAE;QACnD,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,sCAAsC;gBACtC,gBAAgB,CAAC;oBACf,KAAK;oBACL,IAAI,EAAE,IAA6C;iBACpD,CAAC,CAAC;YACL,CAAC;SACF;QACD,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;YACzB,MAAM,EAAE,CACN,KAA8B,EAC9B,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;gBACzC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAAiB,EAAE,CAAC,CAAC;YACvD,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;YAC/B,MAAM,EAAE,CACN,KAAoC,EACpC,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;gBAC/C,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;QACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CACN,KAAmC,EACnC,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;gBAC9C,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;QACD,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE;YACxB,MAAM,EAAE,CACN,MAAc,EACd,IAAuB,EACvB,QAAkC,EAC5B,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;gBACxC,sCAAsC;YACxC,CAAC;SACF;KACF,CAAC;IAEF,MAAM,SAAS,GAAG,YAAY,CAC5B,SAAS,CAAC,SAAS,CAC4B,CAAC;IAClD,SAAS,CAAC,KAAK,GAAG,kBAAkB,CAAC;IAErC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,aAAa;QACpB,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,qBAAqB;oBAC3B,IAAI,EAAE,YAAY;oBAClB,QAAQ,EAAE,CAAC;iBACZ;aACF;YACD,YAAY,EAAE,kCAAkC;YAChD,yIAAyI;SAC1I;QACD,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,cAAc;KACf,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG;QACb,YAAY,EAAE;YACZ,QAAQ,EAAE,SAAS,CAAC,SAAS;YAC7B,SAAS,EAAE,oBAAoB;SAChC;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC;IAEvD,wBAAwB;IACxB,+EAA+E;IAC/E,6HAA6H;IAC7H,iEAAiE;IACjE,8FAA8F;IAC9F,KAAK;IACL,gDAAgD;IAChD,yDAAyD;IACzD,MAAM,WAAW,GAAG,+BAA+B,CAAC;IACpD,yEAAyE;IACzE,sBAAsB;IACtB,wEAAwE;IAExE,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC;IAExD,MAAM,MAAM,GAAG;QACb,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;IACF,MAAM,iBAAiB,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAClE,MAAM,aAAa,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAC3C,IAAI,aAAa,EAAE,CAAC;QAClB,mBAAmB,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;QAC3C,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC;IACD,mDAAmD;IACnD,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAC5C,8CAA8C;AAChD,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,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,qBAAqB,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACpC,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/ant_web_search.ts

@@ -98,6 +98,7 @@ async function testStandardStreaming(): Promise<void> {
       // additional_instructions: `Always address the user by their name. The user's name is ${userName} and they are located in ${location}.`,
     },
     returnContent: true,
+    skipCleanup: true,
     customHandlers,
   });
 

package/src/scripts/ant_web_search_edge_case.js

@@ -0,0 +1,133 @@
+/* eslint-disable no-console */
+// src/scripts/cli.ts
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+import { Calculator } from '@/tools/Calculator';
+import { getArgs } from '@/scripts/args';
+import { Run } from '@/run';
+import { GraphEvents, Providers } from '@/common';
+import { getLLMConfig } from '@/utils/llmConfig';
+const conversationHistory = [];
+let _contentParts = [];
+async function testStandardStreaming() {
+    const { userName, location, currentDate } = await getArgs();
+    const { contentParts, aggregateContent } = createContentAggregator();
+    _contentParts = contentParts;
+    const customHandlers = {
+        [GraphEvents.TOOL_END]: new ToolEndHandler(),
+        [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
+        [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
+        [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+            handle: (event, data) => {
+                console.log('====== ON_RUN_STEP_COMPLETED ======');
+                // console.dir(data, { depth: null });
+                aggregateContent({
+                    event,
+                    data: data,
+                });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP]: {
+            handle: (event, data) => {
+                console.log('====== ON_RUN_STEP ======');
+                console.dir(data, { depth: null });
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP_DELTA]: {
+            handle: (event, data) => {
+                console.log('====== ON_RUN_STEP_DELTA ======');
+                console.dir(data, { depth: null });
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_MESSAGE_DELTA]: {
+            handle: (event, data) => {
+                // console.log('====== ON_MESSAGE_DELTA ======');
+                // console.dir(data, { depth: null });
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.TOOL_START]: {
+            handle: (_event, data, metadata) => {
+                console.log('====== TOOL_START ======');
+                // console.dir(data, { depth: null });
+            },
+        },
+    };
+    const llmConfig = getLLMConfig(Providers.ANTHROPIC);
+    llmConfig.model = 'claude-haiku-4-5';
+    const run = await Run.create({
+        runId: 'test-run-id',
+        graphConfig: {
+            type: 'standard',
+            llmConfig,
+            tools: [
+                {
+                    type: 'web_search_20250305',
+                    name: 'web_search',
+                    max_uses: 5,
+                },
+                new Calculator(),
+            ],
+            instructions: 'You are a friendly AI assistant.',
+            // additional_instructions: `Always address the user by their name. The user's name is ${userName} and they are located in ${location}.`,
+        },
+        returnContent: true,
+        skipCleanup: true,
+        customHandlers,
+    });
+    const config = {
+        configurable: {
+            provider: Providers.ANTHROPIC,
+            thread_id: 'conversation-num-1',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    console.log('Test 1: Web search + calculator (simultaneous tool test)');
+    // const userMessage = `
+    // Make a search for the weather in ${location} today, which is ${currentDate}.
+    // Before making the search, please let me know what you're about to do, then immediately start searching without hesitation.
+    // Make sure to always refer to me by name, which is ${userName}.
+    // After giving me a thorough summary, tell me a joke about the weather forecast we went over.
+    // `;
+    // const userMessage = 'Are massage guns good?';
+    // const userMessage = 'What is functional programming?';
+    // const userMessage = "Get me today's trending news.";
+    // const userMessage = "search recent italy earthquake volcano activity";
+    // const userMessage =
+    //   "use 'Trump' as the exact search query and tell me what you find.";
+    const userMessage = 'Can you search the web for the current population of Tokyo, and also calculate what 15% of that population would be? Do both at the same time.';
+    conversationHistory.push(new HumanMessage(userMessage));
+    const inputs = {
+        messages: conversationHistory,
+    };
+    const finalContentParts = await run.processStream(inputs, config);
+    const finalMessages = run.getRunMessages();
+    if (finalMessages) {
+        conversationHistory.push(...finalMessages);
+        console.dir(conversationHistory, { depth: null });
+    }
+    // console.dir(finalContentParts, { depth: null });
+    console.log('\n\n====================\n\n');
+    // console.dir(contentParts, { depth: null });
+}
+process.on('unhandledRejection', (reason, promise) => {
+    console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+    console.log('Content Parts:');
+    console.dir(_contentParts, { depth: null });
+    process.exit(1);
+});
+testStandardStreaming().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=ant_web_search_edge_case.js.map

package/src/scripts/ant_web_search_edge_case.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ant_web_search_edge_case.js","sourceRoot":"","sources":["ant_web_search_edge_case.ts"],"names":[],"mappings":"AAAA,+BAA+B;AAC/B,qBAAqB;AACrB,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AACT,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAErE,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAC3D,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAEhD,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,WAAW,EAAY,SAAS,EAAE,MAAM,UAAU,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjD,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAC9C,IAAI,aAAa,GAA4C,EAAE,CAAC;AAChE,KAAK,UAAU,qBAAqB;IAClC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;IAC5D,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IACrE,aAAa,GAAG,YAAY,CAAC;IAC7B,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,EAAE;QACnD,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,sCAAsC;gBACtC,gBAAgB,CAAC;oBACf,KAAK;oBACL,IAAI,EAAE,IAA6C;iBACpD,CAAC,CAAC;YACL,CAAC;SACF;QACD,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;YACzB,MAAM,EAAE,CACN,KAA8B,EAC9B,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;gBACzC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAAiB,EAAE,CAAC,CAAC;YACvD,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;YAC/B,MAAM,EAAE,CACN,KAAoC,EACpC,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;gBAC/C,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;QACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CACN,KAAmC,EACnC,IAAuB,EACjB,EAAE;gBACR,iDAAiD;gBACjD,sCAAsC;gBACtC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;QACD,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE;YACxB,MAAM,EAAE,CACN,MAAc,EACd,IAAuB,EACvB,QAAkC,EAC5B,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;gBACxC,sCAAsC;YACxC,CAAC;SACF;KACF,CAAC;IAEF,MAAM,SAAS,GAAG,YAAY,CAC5B,SAAS,CAAC,SAAS,CAC4B,CAAC;IAClD,SAAS,CAAC,KAAK,GAAG,kBAAkB,CAAC;IAErC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,aAAa;QACpB,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,qBAAqB;oBAC3B,IAAI,EAAE,YAAY;oBAClB,QAAQ,EAAE,CAAC;iBACZ;gBACD,IAAI,UAAU,EAAE;aACjB;YACD,YAAY,EAAE,kCAAkC;YAChD,yIAAyI;SAC1I;QACD,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,cAAc;KACf,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG;QACb,YAAY,EAAE;YACZ,QAAQ,EAAE,SAAS,CAAC,SAAS;YAC7B,SAAS,EAAE,oBAAoB;SAChC;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,0DAA0D,CAAC,CAAC;IAExE,wBAAwB;IACxB,+EAA+E;IAC/E,6HAA6H;IAC7H,iEAAiE;IACjE,8FAA8F;IAC9F,KAAK;IACL,gDAAgD;IAChD,yDAAyD;IACzD,uDAAuD;IACvD,yEAAyE;IACzE,sBAAsB;IACtB,wEAAwE;IACxE,MAAM,WAAW,GACf,gJAAgJ,CAAC;IAEnJ,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC;IAExD,MAAM,MAAM,GAAG;QACb,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;IACF,MAAM,iBAAiB,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAClE,MAAM,aAAa,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAC3C,IAAI,aAAa,EAAE,CAAC;QAClB,mBAAmB,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;QAC3C,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC;IACD,mDAAmD;IACnD,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAC5C,8CAA8C;AAChD,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,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,qBAAqB,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACpC,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/ant_web_search_edge_case.ts

@@ -100,6 +100,7 @@ async function testStandardStreaming(): Promise<void> {
       // additional_instructions: `Always address the user by their name. The user's name is ${userName} and they are located in ${location}.`,
     },
     returnContent: true,
+    skipCleanup: true,
     customHandlers,
   });
 

package/src/scripts/ant_web_search_error_edge_case.js

@@ -0,0 +1,119 @@
+/* eslint-disable no-console */
+// src/scripts/ant_web_search_error_edge_case.ts
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+import { getArgs } from '@/scripts/args';
+import { Run } from '@/run';
+import { GraphEvents, Providers } from '@/common';
+import { getLLMConfig } from '@/utils/llmConfig';
+const conversationHistory = [];
+let _contentParts = [];
+async function testStandardStreaming() {
+    const { userName, location, currentDate } = await getArgs();
+    const { contentParts, aggregateContent } = createContentAggregator();
+    _contentParts = contentParts;
+    const customHandlers = {
+        [GraphEvents.TOOL_END]: new ToolEndHandler(),
+        [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
+        [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
+        [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+            handle: (event, data) => {
+                console.log('====== ON_RUN_STEP_COMPLETED ======');
+                // console.dir(data, { depth: null });
+                aggregateContent({
+                    event,
+                    data: data,
+                });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP]: {
+            handle: (event, data) => {
+                console.log('====== ON_RUN_STEP ======');
+                console.dir(data, { depth: null });
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_RUN_STEP_DELTA]: {
+            handle: (event, data) => {
+                console.log('====== ON_RUN_STEP_DELTA ======');
+                console.dir(data, { depth: null });
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.ON_MESSAGE_DELTA]: {
+            handle: (event, data) => {
+                // console.log('====== ON_MESSAGE_DELTA ======');
+                // console.dir(data, { depth: null });
+                aggregateContent({ event, data: data });
+            },
+        },
+        [GraphEvents.TOOL_START]: {
+            handle: (_event, data, metadata) => {
+                console.log('====== TOOL_START ======');
+                // console.dir(data, { depth: null });
+            },
+        },
+    };
+    const llmConfig = getLLMConfig(Providers.ANTHROPIC);
+    llmConfig.model = 'claude-haiku-4-5';
+    const run = await Run.create({
+        runId: 'test-run-id',
+        graphConfig: {
+            type: 'standard',
+            llmConfig,
+            tools: [
+                {
+                    type: 'web_search_20250305',
+                    name: 'web_search',
+                    max_uses: 5,
+                },
+            ],
+            instructions: 'You are a helpful AI research assistant.',
+        },
+        returnContent: true,
+        skipCleanup: true,
+        customHandlers,
+    });
+    const config = {
+        configurable: {
+            provider: Providers.ANTHROPIC,
+            thread_id: 'conversation-num-1',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    console.log('Test: Web search with multiple searches (error edge case test)');
+    // This prompt should trigger multiple web searches which may result in errors
+    const userMessage = 'Do a deep deep research on CoreWeave. I need you to perform multiple searches before you generate the answer. The basis of our research should be to investigate if this is a solid long term investment.';
+    conversationHistory.push(new HumanMessage(userMessage));
+    const inputs = {
+        messages: conversationHistory,
+    };
+    const finalContentParts = await run.processStream(inputs, config);
+    const finalMessages = run.getRunMessages();
+    if (finalMessages) {
+        conversationHistory.push(...finalMessages);
+        console.dir(conversationHistory, { depth: null });
+    }
+    // console.dir(finalContentParts, { depth: null });
+    console.log('\n\n====================\n\n');
+    // console.dir(contentParts, { depth: null });
+}
+process.on('unhandledRejection', (reason, promise) => {
+    console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+    console.log('Content Parts:');
+    console.dir(_contentParts, { depth: null });
+    process.exit(1);
+});
+testStandardStreaming().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=ant_web_search_error_edge_case.js.map

package/src/scripts/ant_web_search_error_edge_case.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ant_web_search_error_edge_case.js","sourceRoot":"","sources":["ant_web_search_error_edge_case.ts"],"names":[],"mappings":"AAAA,+BAA+B;AAC/B,gDAAgD;AAChD,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AACT,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAErE,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAE3D,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjD,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAC9C,IAAI,aAAa,GAA4C,EAAE,CAAC;AAChE,KAAK,UAAU,qBAAqB;IAClC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;IAC5D,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IACrE,aAAa,GAAG,YAAY,CAAC;IAC7B,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,EAAE;QACnD,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,sCAAsC;gBACtC,gBAAgB,CAAC;oBACf,KAAK;oBACL,IAAI,EAAE,IAA6C;iBACpD,CAAC,CAAC;YACL,CAAC;SACF;QACD,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;YACzB,MAAM,EAAE,CACN,KAA8B,EAC9B,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;gBACzC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAAiB,EAAE,CAAC,CAAC;YACvD,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;YAC/B,MAAM,EAAE,CACN,KAAoC,EACpC,IAAuB,EACjB,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;gBAC/C,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;QACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CACN,KAAmC,EACnC,IAAuB,EACjB,EAAE;gBACR,iDAAiD;gBACjD,sCAAsC;gBACtC,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;YACjE,CAAC;SACF;QACD,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE;YACxB,MAAM,EAAE,CACN,MAAc,EACd,IAAuB,EACvB,QAAkC,EAC5B,EAAE;gBACR,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;gBACxC,sCAAsC;YACxC,CAAC;SACF;KACF,CAAC;IAEF,MAAM,SAAS,GAAG,YAAY,CAC5B,SAAS,CAAC,SAAS,CAC4B,CAAC;IAClD,SAAS,CAAC,KAAK,GAAG,kBAAkB,CAAC;IAErC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,aAAa;QACpB,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,qBAAqB;oBAC3B,IAAI,EAAE,YAAY;oBAClB,QAAQ,EAAE,CAAC;iBACZ;aACF;YACD,YAAY,EAAE,0CAA0C;SACzD;QACD,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,cAAc;KACf,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG;QACb,YAAY,EAAE;YACZ,QAAQ,EAAE,SAAS,CAAC,SAAS;YAC7B,SAAS,EAAE,oBAAoB;SAChC;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,gEAAgE,CAAC,CAAC;IAE9E,8EAA8E;IAC9E,MAAM,WAAW,GACf,2MAA2M,CAAC;IAE9M,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC;IAExD,MAAM,MAAM,GAAG;QACb,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;IACF,MAAM,iBAAiB,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAClE,MAAM,aAAa,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAC3C,IAAI,aAAa,EAAE,CAAC;QAClB,mBAAmB,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;QAC3C,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC;IACD,mDAAmD;IACnD,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAC5C,8CAA8C;AAChD,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,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,qBAAqB,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACpC,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/ant_web_search_error_edge_case.ts

@@ -97,6 +97,7 @@ async function testStandardStreaming(): Promise<void> {
       instructions: 'You are a helpful AI research assistant.',
     },
     returnContent: true,
+    skipCleanup: true,
     customHandlers,
   });
 

package/src/scripts/args.js

@@ -0,0 +1,41 @@
+// src/scripts/args.ts
+import yargs from 'yargs';
+import { hideBin } from 'yargs/helpers';
+import { llmConfigs } from '@/utils/llmConfig';
+import { Providers } from '@/common';
+export async function getArgs() {
+    const argv = yargs(hideBin(process.argv))
+        .option('name', {
+        alias: 'n',
+        type: 'string',
+        description: 'User name',
+        default: 'Jo',
+    })
+        .option('location', {
+        alias: 'l',
+        type: 'string',
+        description: 'User location',
+        default: 'New York',
+    })
+        .option('provider', {
+        alias: 'p',
+        type: 'string',
+        description: 'LLM provider',
+        choices: Object.keys(llmConfigs),
+        default: Providers.OPENAI,
+    })
+        .help()
+        .alias('help', 'h').argv;
+    const args = await argv;
+    const userName = args.name;
+    const location = args.location;
+    const provider = args.provider;
+    const currentDate = new Date().toLocaleString();
+    return {
+        userName,
+        location,
+        provider,
+        currentDate,
+    };
+}
+//# sourceMappingURL=args.js.map

package/src/scripts/args.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"args.js","sourceRoot":"","sources":["args.ts"],"names":[],"mappings":"AAAA,sBAAsB;AACtB,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACxC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAErC,MAAM,CAAC,KAAK,UAAU,OAAO;IAM3B,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;SACtC,MAAM,CAAC,MAAM,EAAE;QACd,KAAK,EAAE,GAAG;QACV,IAAI,EAAE,QAAQ;QACd,WAAW,EAAE,WAAW;QACxB,OAAO,EAAE,IAAI;KACd,CAAC;SACD,MAAM,CAAC,UAAU,EAAE;QAClB,KAAK,EAAE,GAAG;QACV,IAAI,EAAE,QAAQ;QACd,WAAW,EAAE,eAAe;QAC5B,OAAO,EAAE,UAAU;KACpB,CAAC;SACD,MAAM,CAAC,UAAU,EAAE;QAClB,KAAK,EAAE,GAAG;QACV,IAAI,EAAE,QAAQ;QACd,WAAW,EAAE,cAAc;QAC3B,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC;QAChC,OAAO,EAAE,SAAS,CAAC,MAAM;KAC1B,CAAC;SACD,IAAI,EAAE;SACN,KAAK,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC;IAE3B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC;IACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAc,CAAC;IACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAkB,CAAC;IACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAqB,CAAC;IAC5C,MAAM,WAAW,GAAG,IAAI,IAAI,EAAE,CAAC,cAAc,EAAE,CAAC;IAEhD,OAAO;QACL,QAAQ;QACR,QAAQ;QACR,QAAQ;QACR,WAAW;KACZ,CAAC;AACJ,CAAC"}