illuma-agents 1.0.16 → 1.0.18

package/src/messages/tools.ts

@@ -1,7 +1,7 @@
 // src/messages/toolDiscovery.ts
 import { AIMessageChunk, ToolMessage } from '@langchain/core/messages';
 import type { BaseMessage } from '@langchain/core/messages';
-import { Constants } from '@/common';
+import { Constants, MessageTypes } from '@/common';
 import { findLastIndex } from './core';
 
 type ToolSearchArtifact = {
@@ -20,7 +20,9 @@ type ToolSearchArtifact = {
  */
 export function extractToolDiscoveries(messages: BaseMessage[]): string[] {
   const lastMessage = messages[messages.length - 1];
-  if (!(lastMessage instanceof ToolMessage)) return [];
+  // Use getType() instead of instanceof to avoid module mismatch issues
+  if (lastMessage.getType() !== MessageTypes.TOOL) return [];
+  const lastToolMessage = lastMessage as ToolMessage;
 
   // Find the latest AIMessage with tool_calls that this tool message belongs to
   const latestAIParentIndex = findLastIndex(
@@ -28,7 +30,7 @@ export function extractToolDiscoveries(messages: BaseMessage[]): string[] {
     (msg) =>
       (msg instanceof AIMessageChunk &&
         (msg.tool_calls?.length ?? 0) > 0 &&
-        msg.tool_calls?.some((tc) => tc.id === lastMessage.tool_call_id)) ??
+        msg.tool_calls?.some((tc) => tc.id === lastToolMessage.tool_call_id)) ??
       false
   );
 
@@ -42,13 +44,15 @@ export function extractToolDiscoveries(messages: BaseMessage[]): string[] {
   const discoveredNames: string[] = [];
   for (let i = latestAIParentIndex + 1; i < messages.length; i++) {
     const msg = messages[i];
-    if (!(msg instanceof ToolMessage)) continue;
-    if (msg.name !== Constants.TOOL_SEARCH_REGEX) continue;
-    if (!toolCallIds.has(msg.tool_call_id)) continue;
+    // Use getType() instead of instanceof to avoid module mismatch issues
+    if (msg.getType() !== MessageTypes.TOOL) continue;
+    const toolMsg = msg as ToolMessage;
+    if (toolMsg.name !== Constants.TOOL_SEARCH_REGEX) continue;
+    if (!toolCallIds.has(toolMsg.tool_call_id)) continue;
 
     // This is a tool search result from the current turn
-    if (typeof msg.artifact === 'object' && msg.artifact != null) {
-      const artifact = msg.artifact as ToolSearchArtifact;
+    if (typeof toolMsg.artifact === 'object' && toolMsg.artifact != null) {
+      const artifact = toolMsg.artifact as ToolSearchArtifact;
       if (artifact.tool_references && artifact.tool_references.length > 0) {
         for (const ref of artifact.tool_references) {
           discoveredNames.push(ref.tool_name);
@@ -66,7 +70,9 @@ export function extractToolDiscoveries(messages: BaseMessage[]): string[] {
  */
 export function hasToolSearchInCurrentTurn(messages: BaseMessage[]): boolean {
   const lastMessage = messages[messages.length - 1];
-  if (!(lastMessage instanceof ToolMessage)) return false;
+  // Use getType() instead of instanceof to avoid module mismatch issues
+  if (lastMessage.getType() !== MessageTypes.TOOL) return false;
+  const lastToolMessage = lastMessage as ToolMessage;
 
   // Find the latest AIMessage with tool_calls
   const latestAIParentIndex = findLastIndex(
@@ -74,7 +80,7 @@ export function hasToolSearchInCurrentTurn(messages: BaseMessage[]): boolean {
     (msg) =>
       (msg instanceof AIMessageChunk &&
         (msg.tool_calls?.length ?? 0) > 0 &&
-        msg.tool_calls?.some((tc) => tc.id === lastMessage.tool_call_id)) ??
+        msg.tool_calls?.some((tc) => tc.id === lastToolMessage.tool_call_id)) ??
       false
   );
 
@@ -84,12 +90,13 @@ export function hasToolSearchInCurrentTurn(messages: BaseMessage[]): boolean {
   const toolCallIds = new Set(aiMessage.tool_calls?.map((tc) => tc.id) ?? []);
 
   // Check if any tool search results exist after the AI message
+  // Use getType() instead of instanceof to avoid module mismatch issues
   for (let i = latestAIParentIndex + 1; i < messages.length; i++) {
     const msg = messages[i];
     if (
-      msg instanceof ToolMessage &&
-      msg.name === Constants.TOOL_SEARCH_REGEX &&
-      toolCallIds.has(msg.tool_call_id)
+      msg.getType() === MessageTypes.TOOL &&
+      (msg as ToolMessage).name === Constants.TOOL_SEARCH_REGEX &&
+      toolCallIds.has((msg as ToolMessage).tool_call_id)
     ) {
       return true;
     }

package/src/scripts/simple.ts

@@ -145,7 +145,7 @@ async function testStandardStreaming(): Promise<void> {
   });
 
   const config = {
-    runId: 'test-simple-script',
+    runId: uuidv4(),
     configurable: {
       user_id: 'user-123',
       thread_id: 'conversation-num-1',