@illuma-ai/agents 1.0.90 → 1.0.94

package/src/scripts/content.js

@@ -0,0 +1,111 @@
+/* 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 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(provider);
+    const run = await Run.create({
+        runId: 'test-run-id',
+        graphConfig: {
+            type: 'standard',
+            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}.`,
+        },
+        returnContent: true,
+        skipCleanup: true,
+        customHandlers,
+    });
+    const config = {
+        configurable: {
+            provider,
+            thread_id: 'conversation-num-1',
+        },
+        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,
+    };
+    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('Conversation history:');
+    process.exit(1);
+});
+testStandardStreaming().catch((err) => {
+    console.error(err);
+    console.log('Conversation history:');
+    console.dir(conversationHistory, { depth: null });
+    process.exit(1);
+});
+//# sourceMappingURL=content.js.map

package/src/scripts/content.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"content.js","sourceRoot":"","sources":["content.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;AAC9C,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,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,uDAAuD;QACvD,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,CAAC,QAAQ,CAAC,CAAC;IAEzC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,aAAa;QACpB,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,KAAK,EAAE,EAAE;YACT,YAAY,EACV,yEAAyE;YAC3E,uBAAuB,EAAE,sBAAsB,QAAQ,4BAA4B,QAAQ,GAAG;SAC/F;QACD,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,cAAc;KACf,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG;QACb,YAAY,EAAE;YACZ,QAAQ;YACR,SAAS,EAAE,oBAAoB;SAChC;QACD,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;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,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/content.ts

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

package/src/scripts/empty_input.js

@@ -0,0 +1,125 @@
+import { z } from 'zod';
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { tool } from "@langchain/core/tools";
+import { getArgs } from '@/scripts/args';
+import { Run } from '@/run';
+import { getLLMConfig } from '@/utils/llmConfig';
+import { ChatModelStreamHandler } from '@/stream';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+import { GraphEvents } from '@/common';
+const pingServerTool = tool(() => {
+    return 'server has been pinged';
+}, {
+    name: 'pingServer',
+    description: 'Ping server',
+    schema: z.object({}),
+});
+async function testPingServer() {
+    const { provider } = await getArgs();
+    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 });
+            }
+        },
+        [GraphEvents.ON_RUN_STEP]: {
+            handle: (_event, data) => {
+                console.log('====== ON_RUN_STEP ======');
+                console.dir(data, { depth: null });
+            }
+        },
+        [GraphEvents.ON_RUN_STEP_DELTA]: {
+            handle: (_event, data) => {
+                console.log('====== ON_RUN_STEP_DELTA ======');
+                console.dir(data, { depth: null });
+            }
+        },
+        [GraphEvents.ON_MESSAGE_DELTA]: {
+            handle: (_event, data) => {
+                console.log('====== ON_MESSAGE_DELTA ======');
+                console.dir(data, { depth: null });
+            }
+        },
+        [GraphEvents.TOOL_START]: {
+            handle: (_event, data, metadata) => {
+                console.log('====== TOOL_START ======');
+                console.dir(data, { depth: null });
+            }
+        },
+        // [GraphEvents.LLM_STREAM]: new LLMStreamHandler(),
+        // [GraphEvents.LLM_START]: {
+        //   handle: (_event: string, data: t.StreamEventData): void => {
+        //     console.log('====== LLM_START ======');
+        //     console.dir(data, { depth: null });
+        //   }
+        // },
+        // [GraphEvents.LLM_END]: {
+        //   handle: (_event: string, data: t.StreamEventData): void => {
+        //     console.log('====== LLM_END ======');
+        //     console.dir(data, { depth: null });
+        //   }
+        // },
+        /*
+        [GraphEvents.CHAIN_START]: {
+          handle: (_event: string, data: t.StreamEventData): void => {
+            console.log('====== CHAIN_START ======');
+            // console.dir(data, { depth: null });
+          }
+        },
+        [GraphEvents.CHAIN_END]: {
+          handle: (_event: string, data: t.StreamEventData): void => {
+            console.log('====== CHAIN_END ======');
+            // console.dir(data, { depth: null });
+          }
+        },
+        */
+        // [GraphEvents.CHAT_MODEL_START]: {
+        //   handle: (_event: string, _data: t.StreamEventData): void => {
+        //     console.log('====== CHAT_MODEL_START ======');
+        //     console.dir(_data, { depth: null });
+        //     // Intentionally left empty
+        //   }
+        // },
+    };
+    const llmConfig = getLLMConfig(provider);
+    const run = await Run.create({
+        runId: 'test-run-id',
+        graphConfig: {
+            type: 'standard',
+            llmConfig,
+            tools: [pingServerTool],
+            instructions: 'You are a helpful AI assistant.',
+        },
+        customHandlers,
+    });
+    const config = {
+        configurable: {
+            provider,
+            thread_id: 'ping-server-test',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    console.log('Pinging server test:');
+    const userMessage = "Please ping the server.";
+    const inputs = {
+        messages: [new HumanMessage(userMessage)],
+    };
+    const contentParts = await run.processStream(inputs, config);
+    const finalMessages = run.getRunMessages();
+    if (finalMessages) {
+        console.log('\nFinal messages:');
+        console.dir(finalMessages, { depth: null });
+    }
+}
+testPingServer().catch((err) => {
+    console.error(err);
+    process.exit(1);
+});
+//# sourceMappingURL=empty_input.js.map

package/src/scripts/empty_input.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"empty_input.js","sourceRoot":"","sources":["empty_input.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AACT,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACxD,OAAO,EAAE,IAAI,EAAE,MAAM,uBAAuB,CAAC;AAC7C,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAC3D,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAGvC,MAAM,cAAc,GAAG,IAAI,CACzB,GAAG,EAAE;IACH,OAAO,wBAAwB,CAAC;AAClC,CAAC,EACD;IACE,IAAI,EAAE,YAAY;IAClB,WAAW,EAAE,aAAa;IAC1B,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;CACrB,CACF,CAAC;AAEF,KAAK,UAAU,cAAc;IAC3B,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;IACrC,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,CAAC,MAAc,EAAE,IAAuB,EAAQ,EAAE;gBACxD,OAAO,CAAC,GAAG,CAAC,qCAAqC,CAAC,CAAC;gBACnD,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACrC,CAAC;SACF;QACD,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;YACzB,MAAM,EAAE,CAAC,MAAc,EAAE,IAAuB,EAAQ,EAAE;gBACxD,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;gBACzC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACrC,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;YAC/B,MAAM,EAAE,CAAC,MAAc,EAAE,IAAuB,EAAQ,EAAE;gBACxD,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;gBAC/C,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACrC,CAAC;SACF;QACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CAAC,MAAc,EAAE,IAAuB,EAAQ,EAAE;gBACxD,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;gBAC9C,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACrC,CAAC;SACF;QACD,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE;YACxB,MAAM,EAAE,CAAC,MAAc,EAAE,IAAuB,EAAE,QAAkC,EAAQ,EAAE;gBAC5F,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACrC,CAAC;SACF;QACD,oDAAoD;QACpD,6BAA6B;QAC7B,iEAAiE;QACjE,8CAA8C;QAC9C,0CAA0C;QAC1C,MAAM;QACN,KAAK;QACL,2BAA2B;QAC3B,iEAAiE;QACjE,4CAA4C;QAC5C,0CAA0C;QAC1C,MAAM;QACN,KAAK;QACL;;;;;;;;;;;;;UAaE;QACF,oCAAoC;QACpC,kEAAkE;QAClE,qDAAqD;QACrD,2CAA2C;QAC3C,kCAAkC;QAClC,MAAM;QACN,KAAK;KACN,CAAC;IAEF,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAEzC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,aAAa;QACpB,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,KAAK,EAAE,CAAC,cAAc,CAAC;YACvB,YAAY,EAAE,iCAAiC;SAChD;QACD,cAAc;KACf,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG;QACb,YAAY,EAAE;YACZ,QAAQ;YACR,SAAS,EAAE,kBAAkB;SAC9B;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;IAEpC,MAAM,WAAW,GAAG,yBAAyB,CAAC;IAC9C,MAAM,MAAM,GAAG;QACb,QAAQ,EAAE,CAAC,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;KAC1C,CAAC;IAEF,MAAM,YAAY,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7D,MAAM,aAAa,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAC3C,IAAI,aAAa,EAAE,CAAC;QAClB,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QACjC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC9C,CAAC;AACH,CAAC;AAED,cAAc,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IAC7B,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACnB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/src/scripts/handoff-test.js

@@ -0,0 +1,96 @@
+import { config } from 'dotenv';
+config();
+import { z } from 'zod';
+import { tool } from '@langchain/core/tools';
+import { ChatAnthropic } from '@langchain/anthropic';
+import { createReactAgent } from '@langchain/langgraph/prebuilt';
+import { StateGraph, MessagesAnnotation, Command, START, getCurrentTaskInput, END, } from '@langchain/langgraph';
+import { ToolMessage } from '@langchain/core/messages';
+const createHandoffTool = ({ agentName, description, }) => {
+    const toolName = `transfer_to_${agentName}`;
+    const toolDescription = description || `Ask agent '${agentName}' for help`;
+    const handoffTool = tool(async (_, config) => {
+        const toolMessage = new ToolMessage({
+            content: `Successfully transferred to ${agentName}`,
+            name: toolName,
+            tool_call_id: config.toolCall.id,
+        });
+        // inject the current agent state
+        const state = getCurrentTaskInput();
+        return new Command({
+            goto: agentName,
+            update: { messages: state.messages.concat(toolMessage) },
+            graph: Command.PARENT,
+        });
+    }, {
+        name: toolName,
+        schema: z.object({}),
+        description: toolDescription,
+    });
+    return handoffTool;
+};
+const bookHotel = tool(async (input) => {
+    return `Successfully booked a stay at ${input.hotel_name}.`;
+}, {
+    name: 'book_hotel',
+    description: 'Book a hotel',
+    schema: z.object({
+        hotel_name: z.string().describe('The name of the hotel to book'),
+    }),
+});
+const bookFlight = tool(async (input) => {
+    return `Successfully booked a flight from ${input.from_airport} to ${input.to_airport}.`;
+}, {
+    name: 'book_flight',
+    description: 'Book a flight',
+    schema: z.object({
+        from_airport: z.string().describe('The departure airport code'),
+        to_airport: z.string().describe('The arrival airport code'),
+    }),
+});
+const transferToHotelAssistant = createHandoffTool({
+    agentName: 'hotel_assistant',
+    description: 'Transfer user to the hotel-booking assistant.',
+});
+const transferToFlightAssistant = createHandoffTool({
+    agentName: 'flight_assistant',
+    description: 'Transfer user to the flight-booking assistant.',
+});
+const llm = new ChatAnthropic({
+    modelName: 'claude-haiku-4-5',
+    apiKey: process.env.ANTHROPIC_API_KEY,
+});
+const flightAssistant = createReactAgent({
+    llm,
+    tools: [bookFlight, transferToHotelAssistant],
+    prompt: 'You are a flight booking assistant',
+    name: 'flight_assistant',
+});
+const hotelAssistant = createReactAgent({
+    llm,
+    tools: [bookHotel, transferToFlightAssistant],
+    prompt: 'You are a hotel booking assistant',
+    name: 'hotel_assistant',
+});
+const multiAgentGraph = new StateGraph(MessagesAnnotation)
+    .addNode('flight_assistant', flightAssistant, {
+    ends: ['hotel_assistant', END],
+})
+    .addNode('hotel_assistant', hotelAssistant, {
+    ends: ['flight_assistant', END],
+})
+    .addEdge(START, 'flight_assistant')
+    .compile();
+const stream = await multiAgentGraph.stream({
+    messages: [
+        {
+            role: 'user',
+            content: 'book a flight from BOS to JFK and a stay at McKittrick Hotel',
+        },
+    ],
+});
+for await (const chunk of stream) {
+    console.log(chunk);
+    console.log('\n');
+}
+//# sourceMappingURL=handoff-test.js.map

package/src/scripts/handoff-test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handoff-test.js","sourceRoot":"","sources":["handoff-test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AAET,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,IAAI,EAAE,MAAM,uBAAuB,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AACjE,OAAO,EACL,UAAU,EACV,kBAAkB,EAClB,OAAO,EACP,KAAK,EACL,mBAAmB,EACnB,GAAG,GACJ,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAOvD,MAAM,iBAAiB,GAAG,CAAC,EACzB,SAAS,EACT,WAAW,GACa,EAAE,EAAE;IAC5B,MAAM,QAAQ,GAAG,eAAe,SAAS,EAAE,CAAC;IAC5C,MAAM,eAAe,GAAG,WAAW,IAAI,cAAc,SAAS,YAAY,CAAC;IAE3E,MAAM,WAAW,GAAG,IAAI,CACtB,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE;QAClB,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC;YAClC,OAAO,EAAE,+BAA+B,SAAS,EAAE;YACnD,IAAI,EAAE,QAAQ;YACd,YAAY,EAAE,MAAM,CAAC,QAAQ,CAAC,EAAE;SACjC,CAAC,CAAC;QAEH,iCAAiC;QACjC,MAAM,KAAK,GACT,mBAAmB,EAA0C,CAAC;QAChE,OAAO,IAAI,OAAO,CAAC;YACjB,IAAI,EAAE,SAAS;YACf,MAAM,EAAE,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE;YACxD,KAAK,EAAE,OAAO,CAAC,MAAM;SACtB,CAAC,CAAC;IACL,CAAC,EACD;QACE,IAAI,EAAE,QAAQ;QACd,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;QACpB,WAAW,EAAE,eAAe;KAC7B,CACF,CAAC;IAEF,OAAO,WAAW,CAAC;AACrB,CAAC,CAAC;AAEF,MAAM,SAAS,GAAG,IAAI,CACpB,KAAK,EAAE,KAA6B,EAAE,EAAE;IACtC,OAAO,iCAAiC,KAAK,CAAC,UAAU,GAAG,CAAC;AAC9D,CAAC,EACD;IACE,IAAI,EAAE,YAAY;IAClB,WAAW,EAAE,cAAc;IAC3B,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QACf,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,+BAA+B,CAAC;KACjE,CAAC;CACH,CACF,CAAC;AAEF,MAAM,UAAU,GAAG,IAAI,CACrB,KAAK,EAAE,KAAmD,EAAE,EAAE;IAC5D,OAAO,qCAAqC,KAAK,CAAC,YAAY,OAAO,KAAK,CAAC,UAAU,GAAG,CAAC;AAC3F,CAAC,EACD;IACE,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,eAAe;IAC5B,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QACf,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4BAA4B,CAAC;QAC/D,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,0BAA0B,CAAC;KAC5D,CAAC;CACH,CACF,CAAC;AAEF,MAAM,wBAAwB,GAAG,iBAAiB,CAAC;IACjD,SAAS,EAAE,iBAAiB;IAC5B,WAAW,EAAE,+CAA+C;CAC7D,CAAC,CAAC;AAEH,MAAM,yBAAyB,GAAG,iBAAiB,CAAC;IAClD,SAAS,EAAE,kBAAkB;IAC7B,WAAW,EAAE,gDAAgD;CAC9D,CAAC,CAAC;AAEH,MAAM,GAAG,GAAG,IAAI,aAAa,CAAC;IAC5B,SAAS,EAAE,kBAAkB;IAC7B,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;CACtC,CAAC,CAAC;AAEH,MAAM,eAAe,GAAG,gBAAgB,CAAC;IACvC,GAAG;IACH,KAAK,EAAE,CAAC,UAAU,EAAE,wBAAwB,CAAC;IAC7C,MAAM,EAAE,oCAAoC;IAC5C,IAAI,EAAE,kBAAkB;CACzB,CAAC,CAAC;AAEH,MAAM,cAAc,GAAG,gBAAgB,CAAC;IACtC,GAAG;IACH,KAAK,EAAE,CAAC,SAAS,EAAE,yBAAyB,CAAC;IAC7C,MAAM,EAAE,mCAAmC;IAC3C,IAAI,EAAE,iBAAiB;CACxB,CAAC,CAAC;AAEH,MAAM,eAAe,GAAG,IAAI,UAAU,CAAC,kBAAkB,CAAC;KACvD,OAAO,CAAC,kBAAkB,EAAE,eAAe,EAAE;IAC5C,IAAI,EAAE,CAAC,iBAAiB,EAAE,GAAG,CAAC;CAC/B,CAAC;KACD,OAAO,CAAC,iBAAiB,EAAE,cAAc,EAAE;IAC1C,IAAI,EAAE,CAAC,kBAAkB,EAAE,GAAG,CAAC;CAChC,CAAC;KACD,OAAO,CAAC,KAAK,EAAE,kBAAkB,CAAC;KAClC,OAAO,EAAE,CAAC;AAEb,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,MAAM,CAAC;IAC1C,QAAQ,EAAE;QACR;YACE,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,8DAA8D;SACxE;KACF;CACF,CAAC,CAAC;AAEH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;IACjC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnB,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;AACpB,CAAC"}

package/src/scripts/image.js

@@ -0,0 +1,138 @@
+// src/scripts/image.ts
+import { config } from 'dotenv';
+config();
+import { HumanMessage } from '@langchain/core/messages';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { ToolEndHandler, ModelEndHandler, createMetadataAggregator, } from '@/events';
+// @ts-expect-error — example module not in current codebase
+import { fetchRandomImageTool } from '@/tools/example';
+import { getLLMConfig } from '@/utils/llmConfig';
+import { getArgs } from '@/scripts/args';
+import { GraphEvents } from '@/common';
+import { Run } from '@/run';
+const conversationHistory = [];
+async function testCodeExecution() {
+    const { userName, location, provider, currentDate } = await getArgs();
+    const { contentParts, aggregateContent } = createContentAggregator();
+    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(provider);
+    const run = await Run.create({
+        runId: 'message-num-1',
+        graphConfig: {
+            type: 'standard',
+            llmConfig,
+            tools: [fetchRandomImageTool],
+            // tools: [fetchRandomImageURL],
+            instructions: 'You are a friendly AI assistant with internet capabilities. Always address the user by their name.',
+            additional_instructions: `The user's name is ${userName} and they are located in ${location}.`,
+        },
+        returnContent: true,
+        skipCleanup: true,
+        customHandlers,
+    });
+    const config = {
+        configurable: {
+            provider,
+            thread_id: 'conversation-num-1',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    console.log('Fetch Random Image');
+    const userMessage1 = `Hi ${userName} here. Please get me 2 random images. Describe them after you receive them.`;
+    conversationHistory.push(new HumanMessage(userMessage1));
+    let inputs = {
+        messages: conversationHistory,
+    };
+    const finalContentParts1 = await run.processStream(inputs, config);
+    const finalMessages1 = run.getRunMessages();
+    if (finalMessages1) {
+        conversationHistory.push(...finalMessages1);
+    }
+    console.log('\n\n====================\n\n');
+    console.dir(contentParts, { depth: null });
+    console.log('Test 2: Follow up with another message');
+    const userMessage2 = `thanks, you're the best!`;
+    conversationHistory.push(new HumanMessage(userMessage2));
+    inputs = {
+        messages: conversationHistory,
+    };
+    const finalContentParts2 = await run.processStream(inputs, config, {
+        keepContent: true,
+    });
+    const finalMessages2 = run.getRunMessages();
+    if (finalMessages2) {
+        conversationHistory.push(...finalMessages2);
+    }
+    console.log('\n\n====================\n\n');
+    console.dir(contentParts, { depth: null });
+    const { handleLLMEnd, collected } = createMetadataAggregator();
+    const titleResult = await run.generateTitle({
+        provider,
+        inputText: userMessage2,
+        contentParts,
+        chainOptions: {
+            callbacks: [
+                {
+                    handleLLMEnd,
+                },
+            ],
+        },
+    });
+    console.log('Generated Title:', titleResult);
+    console.log('Collected metadata:', collected);
+}
+process.on('unhandledRejection', (reason, promise) => {
+    console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+    console.log('Conversation history:');
+    console.dir(conversationHistory, { depth: null });
+    process.exit(1);
+});
+testCodeExecution().catch((err) => {
+    console.error(err);
+    console.log('Conversation history:');
+    console.dir(conversationHistory, { depth: null });
+    process.exit(1);
+});
+//# sourceMappingURL=image.js.map

package/src/scripts/image.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"image.js","sourceRoot":"","sources":["image.ts"],"names":[],"mappings":"AAAA,uBAAuB;AACvB,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AACT,OAAO,EAAE,YAAY,EAA0B,MAAM,0BAA0B,CAAC;AAGhF,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EACL,cAAc,EACd,eAAe,EACf,wBAAwB,GACzB,MAAM,UAAU,CAAC;AAClB,4DAA4D;AAC5D,OAAO,EAAE,oBAAoB,EAAuB,MAAM,iBAAiB,CAAC;AAC5E,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AACvC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAE5B,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAE9C,KAAK,UAAU,iBAAiB;IAC9B,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,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,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnC,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,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACrC,CAAC;SACF;KACF,CAAC;IAEF,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAEzC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,eAAe;QACtB,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS;YACT,KAAK,EAAE,CAAC,oBAAoB,CAAC;YAC7B,gCAAgC;YAChC,YAAY,EACV,oGAAoG;YACtG,uBAAuB,EAAE,sBAAsB,QAAQ,4BAA4B,QAAQ,GAAG;SAC/F;QACD,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,cAAc;KACf,CAAC,CAAC;IAEH,MAAM,MAAM,GAIR;QACF,YAAY,EAAE;YACZ,QAAQ;YACR,SAAS,EAAE,oBAAoB;SAChC;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IAElC,MAAM,YAAY,GAAG,MAAM,QAAQ,6EAA6E,CAAC;IAEjH,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IAEzD,IAAI,MAAM,GAAG;QACX,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;IACF,MAAM,kBAAkB,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACnE,MAAM,cAAc,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAC5C,IAAI,cAAc,EAAE,CAAC;QACnB,mBAAmB,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAC;IAC9C,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAC5C,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAE3C,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;IAEtD,MAAM,YAAY,GAAG,0BAA0B,CAAC;IAEhD,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IAEzD,MAAM,GAAG;QACP,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;IACF,MAAM,kBAAkB,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE;QACjE,WAAW,EAAE,IAAI;KAClB,CAAC,CAAC;IACH,MAAM,cAAc,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;IAC5C,IAAI,cAAc,EAAE,CAAC;QACnB,mBAAmB,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAC;IAC9C,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAC5C,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAE3C,MAAM,EAAE,YAAY,EAAE,SAAS,EAAE,GAAG,wBAAwB,EAAE,CAAC;IAC/D,MAAM,WAAW,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC;QAC1C,QAAQ;QACR,SAAS,EAAE,YAAY;QACvB,YAAY;QACZ,YAAY,EAAE;YACZ,SAAS,EAAE;gBACT;oBACE,YAAY;iBACb;aACF;SACF;KACF,CAAC,CAAC;IACH,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,WAAW,CAAC,CAAC;IAC7C,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,SAAS,CAAC,CAAC;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,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;AAEH,iBAAiB,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IAChC,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/image.ts

@@ -1,4 +1,4 @@
-// src/scripts/cli.ts
+// src/scripts/image.ts
 import { config } from 'dotenv';
 config();
 import { HumanMessage, AIMessage, BaseMessage } from '@langchain/core/messages';
@@ -10,6 +10,7 @@ import {
   ModelEndHandler,
   createMetadataAggregator,
 } from '@/events';
+// @ts-expect-error — example module not in current codebase
 import { fetchRandomImageTool, fetchRandomImageURL } from '@/tools/example';
 import { getLLMConfig } from '@/utils/llmConfig';
 import { getArgs } from '@/scripts/args';
@@ -94,6 +95,7 @@ async function testCodeExecution(): Promise<void> {
       additional_instructions: `The user's name is ${userName} and they are located in ${location}.`,
     },
     returnContent: true,
+    skipCleanup: true,
     customHandlers,
   });
 

package/src/scripts/memory.js

@@ -0,0 +1,83 @@
+// src/scripts/cli.ts
+import { config } from 'dotenv';
+config();
+import z from 'zod';
+import { tool } from '@langchain/core/tools';
+import { HumanMessage } from '@langchain/core/messages';
+import { getArgs } from '@/scripts/args';
+import { Providers } from '@/common';
+import { Run } from '@/run';
+const conversationHistory = [];
+const memoryKv = {};
+const setMemory = tool(async ({ key, value }) => {
+    if (!/^[a-z_]+$/.test(key)) {
+        throw new Error('Key must only contain lowercase letters and underscores');
+    }
+    memoryKv[key] = value;
+    return { ok: true };
+}, {
+    name: 'set_memory',
+    description: 'Saves important data about the user into memory.',
+    schema: z.object({
+        key: z
+            .string()
+            .describe('The key of the memory value. Always use lowercase and underscores, no other characters.'),
+        value: z
+            .string()
+            .describe('Value can be anything represented as a string'),
+    }),
+});
+async function testStandardStreaming() {
+    const { userName, provider } = await getArgs();
+    const run = await Run.create({
+        runId: 'memory-run',
+        graphConfig: {
+            type: 'standard',
+            llmConfig: {
+                provider: Providers.OPENAI,
+                model: 'gpt-4o-mini',
+                temperature: 0.5,
+                streaming: false,
+            },
+            tools: [setMemory],
+            instructions: 'You can use the `set_memory` tool to save important data about the user into memory. If there is nothing to note about the user specifically, respond with `nothing`.',
+            toolEnd: true,
+        },
+        returnContent: true,
+        skipCleanup: true,
+    });
+    const config = {
+        configurable: {
+            provider,
+            thread_id: 'conversation-num-1',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    console.log('Test 1: Simple message test');
+    const userMessage = `hi`;
+    conversationHistory.push(new HumanMessage(userMessage));
+    const inputs = {
+        messages: conversationHistory,
+    };
+    await run.processStream(inputs, config);
+    const finalMessages = run.getRunMessages();
+    if (finalMessages) {
+        conversationHistory.push(...finalMessages);
+        console.dir(conversationHistory, { depth: null });
+    }
+    console.log('\n\n====================\n\n');
+    console.dir(memoryKv, { 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=memory.js.map

package/src/scripts/memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory.js","sourceRoot":"","sources":["memory.ts"],"names":[],"mappings":"AAAA,qBAAqB;AACrB,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AAET,OAAO,CAAC,MAAM,KAAK,CAAC;AACpB,OAAO,EAAE,IAAI,EAAE,MAAM,uBAAuB,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAErE,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AACrC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAE5B,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAE9C,MAAM,QAAQ,GAA2B,EAAE,CAAC;AAE5C,MAAM,SAAS,GAAG,IAAI,CACpB,KAAK,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE;IACvB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CACb,yDAAyD,CAC1D,CAAC;IACJ,CAAC;IAED,QAAQ,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;IAEtB,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;AACtB,CAAC,EACD;IACE,IAAI,EAAE,YAAY;IAClB,WAAW,EAAE,kDAAkD;IAC/D,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QACf,GAAG,EAAE,CAAC;aACH,MAAM,EAAE;aACR,QAAQ,CACP,yFAAyF,CAC1F;QACH,KAAK,EAAE,CAAC;aACL,MAAM,EAAE;aACR,QAAQ,CAAC,+CAA+C,CAAC;KAC7D,CAAC;CACH,CACF,CAAC;AAEF,KAAK,UAAU,qBAAqB;IAClC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;IAE/C,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,YAAY;QACnB,WAAW,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,SAAS,EAAE;gBACT,QAAQ,EAAE,SAAS,CAAC,MAAM;gBAC1B,KAAK,EAAE,aAAa;gBACpB,WAAW,EAAE,GAAG;gBAChB,SAAS,EAAE,KAAK;aACjB;YACD,KAAK,EAAE,CAAC,SAAS,CAAC;YAClB,YAAY,EACV,uKAAuK;YACzK,OAAO,EAAE,IAAI;SACd;QACD,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;KAClB,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG;QACb,YAAY,EAAE;YACZ,QAAQ;YACR,SAAS,EAAE,oBAAoB;SAChC;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;IAE3C,MAAM,WAAW,GAAG,IAAI,CAAC;IAEzB,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC;IAExD,MAAM,MAAM,GAAG;QACb,QAAQ,EAAE,mBAAmB;KAC9B,CAAC;IACF,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACxC,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;IAED,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAC5C,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;AACzC,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"}