@langchain/langgraph 0.2.72 → 0.2.74

package/dist/pregel/messages.test.cjs

@@ -1,27 +1,27 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-const globals_1 = require("@jest/globals");
+const vitest_1 = require("vitest");
 const messages_1 = require("@langchain/core/messages");
 const outputs_1 = require("@langchain/core/outputs");
 const messages_js_1 = require("./messages.cjs");
 const constants_js_1 = require("../constants.cjs");
-(0, globals_1.describe)("StreamMessagesHandler", () => {
-    (0, globals_1.describe)("constructor", () => {
-        (0, globals_1.it)("should properly initialize the handler", () => {
-            const streamFn = globals_1.jest.fn();
+(0, vitest_1.describe)("StreamMessagesHandler", () => {
+    (0, vitest_1.describe)("constructor", () => {
+        (0, vitest_1.it)("should properly initialize the handler", () => {
+            const streamFn = vitest_1.vi.fn();
             const handler = new messages_js_1.StreamMessagesHandler(streamFn);
-            (0, globals_1.expect)(handler.name).toBe("StreamMessagesHandler");
-            (0, globals_1.expect)(handler.streamFn).toBe(streamFn);
-            (0, globals_1.expect)(handler.metadatas).toEqual({});
-            (0, globals_1.expect)(handler.seen).toEqual({});
-            (0, globals_1.expect)(handler.emittedChatModelRunIds).toEqual({});
-            (0, globals_1.expect)(handler.stableMessageIdMap).toEqual({});
-            (0, globals_1.expect)(handler.lc_prefer_streaming).toBe(true);
+            (0, vitest_1.expect)(handler.name).toBe("StreamMessagesHandler");
+            (0, vitest_1.expect)(handler.streamFn).toBe(streamFn);
+            (0, vitest_1.expect)(handler.metadatas).toEqual({});
+            (0, vitest_1.expect)(handler.seen).toEqual({});
+            (0, vitest_1.expect)(handler.emittedChatModelRunIds).toEqual({});
+            (0, vitest_1.expect)(handler.stableMessageIdMap).toEqual({});
+            (0, vitest_1.expect)(handler.lc_prefer_streaming).toBe(true);
         });
     });
-    (0, globals_1.describe)("_emit", () => {
-        (0, globals_1.it)("should emit a message with metadata", () => {
-            const streamFn = globals_1.jest.fn();
+    (0, vitest_1.describe)("_emit", () => {
+        (0, vitest_1.it)("should emit a message with metadata", () => {
+            const streamFn = vitest_1.vi.fn();
             const handler = new messages_js_1.StreamMessagesHandler(streamFn);
             const meta = [
                 ["ns1", "ns2"],
@@ -30,7 +30,7 @@ const constants_js_1 = require("../constants.cjs");
             const message = new messages_1.AIMessage({ content: "Hello world" });
             const runId = "run-123";
             handler._emit(meta, message, runId);
-            (0, globals_1.expect)(streamFn).toHaveBeenCalledWith([
+            (0, vitest_1.expect)(streamFn).toHaveBeenCalledWith([
                 ["ns1", "ns2"],
                 "messages",
                 [message, { name: "test", tags: [] }],
@@ -38,10 +38,10 @@ const constants_js_1 = require("../constants.cjs");
             // Should store the message in seen if it has an ID
             message.id = "msg-123";
             handler._emit(meta, message, runId);
-            (0, globals_1.expect)(handler.seen["msg-123"]).toBe(message);
+            (0, vitest_1.expect)(handler.seen["msg-123"]).toBe(message);
         });
-        (0, globals_1.it)("should deduplicate messages when dedupe=true and message has been seen", () => {
-            const streamFn = globals_1.jest.fn();
+        (0, vitest_1.it)("should deduplicate messages when dedupe=true and message has been seen", () => {
+            const streamFn = vitest_1.vi.fn();
             const handler = new messages_js_1.StreamMessagesHandler(streamFn);
             const meta = [
                 ["ns1"],
@@ -51,14 +51,14 @@ const constants_js_1 = require("../constants.cjs");
             const runId = "run-123";
             // First emit should work
             handler._emit(meta, message, runId);
-            (0, globals_1.expect)(streamFn).toHaveBeenCalledTimes(1);
+            (0, vitest_1.expect)(streamFn).toHaveBeenCalledTimes(1);
             // Second emit with same ID and dedupe=true should be ignored
             streamFn.mockClear();
             handler._emit(meta, message, runId, true);
-            (0, globals_1.expect)(streamFn).not.toHaveBeenCalled();
+            (0, vitest_1.expect)(streamFn).not.toHaveBeenCalled();
         });
-        (0, globals_1.it)("should assign proper ID to tool messages", () => {
-            const streamFn = globals_1.jest.fn();
+        (0, vitest_1.it)("should assign proper ID to tool messages", () => {
+            const streamFn = vitest_1.vi.fn();
             const handler = new messages_js_1.StreamMessagesHandler(streamFn);
             const meta = [
                 ["ns1"],
@@ -71,11 +71,11 @@ const constants_js_1 = require("../constants.cjs");
             const runId = "run-456";
             handler._emit(meta, toolMessage, runId);
             // Should assign an ID based on the tool call ID
-            (0, globals_1.expect)(toolMessage.id).toBe(`run-${runId}-tool-tc-123`);
-            (0, globals_1.expect)(streamFn).toHaveBeenCalled();
+            (0, vitest_1.expect)(toolMessage.id).toBe(`run-${runId}-tool-tc-123`);
+            (0, vitest_1.expect)(streamFn).toHaveBeenCalled();
         });
-        (0, globals_1.it)("should maintain stable message IDs for the same run", () => {
-            const streamFn = globals_1.jest.fn();
+        (0, vitest_1.it)("should maintain stable message IDs for the same run", () => {
+            const streamFn = vitest_1.vi.fn();
             const handler = new messages_js_1.StreamMessagesHandler(streamFn);
             const meta = [
                 ["ns1"],
@@ -89,13 +89,13 @@ const constants_js_1 = require("../constants.cjs");
             // Second message with no ID should get the same stable ID
             const message2 = new messages_1.AIMessage({ content: "Second chunk" });
             handler._emit(meta, message2, runId);
-            (0, globals_1.expect)(message2.id).toBe(stableId);
-            (0, globals_1.expect)(handler.stableMessageIdMap[runId]).toBe(stableId);
+            (0, vitest_1.expect)(message2.id).toBe(stableId);
+            (0, vitest_1.expect)(handler.stableMessageIdMap[runId]).toBe(stableId);
         });
     });
-    (0, globals_1.describe)("handleChatModelStart", () => {
-        (0, globals_1.it)("should store metadata when provided", () => {
-            const handler = new messages_js_1.StreamMessagesHandler(globals_1.jest.fn());
+    (0, vitest_1.describe)("handleChatModelStart", () => {
+        (0, vitest_1.it)("should store metadata when provided", () => {
+            const handler = new messages_js_1.StreamMessagesHandler(vitest_1.vi.fn());
             const runId = "run-123";
             const metadata = {
                 langgraph_checkpoint_ns: "ns1|ns2",
@@ -109,13 +109,13 @@ const constants_js_1 = require("../constants.cjs");
             metadata, // metadata
             "ModelName" // name
             );
-            (0, globals_1.expect)(handler.metadatas[runId]).toEqual([
+            (0, vitest_1.expect)(handler.metadatas[runId]).toEqual([
                 ["ns1", "ns2"],
                 { tags: [], name: "ModelName", ...metadata },
             ]);
         });
-        (0, globals_1.it)("should not store metadata when TAG_NOSTREAM is present", () => {
-            const handler = new messages_js_1.StreamMessagesHandler(globals_1.jest.fn());
+        (0, vitest_1.it)("should not store metadata when TAG_NOSTREAM is present", () => {
+            const handler = new messages_js_1.StreamMessagesHandler(vitest_1.vi.fn());
             const runId = "run-123";
             const metadata = {
                 langgraph_checkpoint_ns: "ns1|ns2",
@@ -123,32 +123,32 @@ const constants_js_1 = require("../constants.cjs");
             handler.handleChatModelStart({}, [], runId, undefined, {}, [constants_js_1.TAG_NOSTREAM], // nostream tag
             metadata, "ModelName");
             // Should not store metadata due to TAG_NOSTREAM
-            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+            (0, vitest_1.expect)(handler.metadatas[runId]).toBeUndefined();
         });
     });
-    (0, globals_1.describe)("handleLLMNewToken", () => {
-        (0, globals_1.it)("should emit message chunk when metadata exists", () => {
-            const streamFn = globals_1.jest.fn();
+    (0, vitest_1.describe)("handleLLMNewToken", () => {
+        (0, vitest_1.it)("should emit message chunk when metadata exists", () => {
+            const streamFn = vitest_1.vi.fn();
             const handler = new messages_js_1.StreamMessagesHandler(streamFn);
             const runId = "run-123";
             handler.metadatas[runId] = [["ns1", "ns2"], { name: "test" }];
             // Spy on _emit
-            const emitSpy = globals_1.jest.spyOn(handler, "_emit");
+            const emitSpy = vitest_1.vi.spyOn(handler, "_emit");
             handler.handleLLMNewToken("token", { prompt: 0, completion: 0 }, // idx
             runId);
             // Should mark run as emitted
-            (0, globals_1.expect)(handler.emittedChatModelRunIds[runId]).toBe(true);
+            (0, vitest_1.expect)(handler.emittedChatModelRunIds[runId]).toBe(true);
             // Should emit AIMessageChunk when no chunk is provided
-            (0, globals_1.expect)(emitSpy).toHaveBeenCalledWith(handler.metadatas[runId], globals_1.expect.any(messages_1.AIMessageChunk), runId);
-            (0, globals_1.expect)(emitSpy.mock.calls[0][1].content).toBe("token");
+            (0, vitest_1.expect)(emitSpy).toHaveBeenCalledWith(handler.metadatas[runId], vitest_1.expect.any(messages_1.AIMessageChunk), runId);
+            (0, vitest_1.expect)(emitSpy.mock.calls[0][1].content).toBe("token");
         });
-        (0, globals_1.it)("should emit provided chunk when available", () => {
-            const streamFn = globals_1.jest.fn();
+        (0, vitest_1.it)("should emit provided chunk when available", () => {
+            const streamFn = vitest_1.vi.fn();
             const handler = new messages_js_1.StreamMessagesHandler(streamFn);
             const runId = "run-123";
             handler.metadatas[runId] = [["ns1"], { name: "test" }];
             // Spy on _emit
-            const emitSpy = globals_1.jest.spyOn(handler, "_emit");
+            const emitSpy = vitest_1.vi.spyOn(handler, "_emit");
             // Create a chunk
             const chunk = new outputs_1.ChatGenerationChunk({
                 message: new messages_1.AIMessageChunk({ content: "chunk content" }),
@@ -157,49 +157,49 @@ const constants_js_1 = require("../constants.cjs");
             handler.handleLLMNewToken("token", { prompt: 0, completion: 0 }, runId, undefined, undefined, { chunk } // provide the chunk
             );
             // Should emit the chunk's message
-            (0, globals_1.expect)(emitSpy).toHaveBeenCalledWith(handler.metadatas[runId], chunk.message, runId);
+            (0, vitest_1.expect)(emitSpy).toHaveBeenCalledWith(handler.metadatas[runId], chunk.message, runId);
         });
-        (0, globals_1.it)("should not emit when metadata is missing", () => {
-            const streamFn = globals_1.jest.fn();
+        (0, vitest_1.it)("should not emit when metadata is missing", () => {
+            const streamFn = vitest_1.vi.fn();
             const handler = new messages_js_1.StreamMessagesHandler(streamFn);
             const runId = "run-123";
             // No metadata for this runId
             // Spy on _emit
-            const emitSpy = globals_1.jest.spyOn(handler, "_emit");
+            const emitSpy = vitest_1.vi.spyOn(handler, "_emit");
             handler.handleLLMNewToken("token", { prompt: 0, completion: 0 }, runId);
             // Should mark run as emitted
-            (0, globals_1.expect)(handler.emittedChatModelRunIds[runId]).toBe(true);
+            (0, vitest_1.expect)(handler.emittedChatModelRunIds[runId]).toBe(true);
             // But should not call _emit
-            (0, globals_1.expect)(emitSpy).not.toHaveBeenCalled();
+            (0, vitest_1.expect)(emitSpy).not.toHaveBeenCalled();
         });
     });
-    (0, globals_1.describe)("handleLLMEnd", () => {
-        (0, globals_1.it)("should emit message from non-streaming run", () => {
-            const streamFn = globals_1.jest.fn();
+    (0, vitest_1.describe)("handleLLMEnd", () => {
+        (0, vitest_1.it)("should emit message from non-streaming run", () => {
+            const streamFn = vitest_1.vi.fn();
             const handler = new messages_js_1.StreamMessagesHandler(streamFn);
             const runId = "run-123";
             handler.metadatas[runId] = [["ns1"], { name: "test" }];
             // Not marked as emitted yet
             // Mock _emit directly instead of spying
-            handler._emit = globals_1.jest.fn();
+            handler._emit = vitest_1.vi.fn();
             const message = new messages_1.AIMessage({ content: "final result" });
             handler.handleLLMEnd({
                 generations: [[{ text: "test output", message }]],
             }, runId);
             // Should emit the message with dedupe=true
-            (0, globals_1.expect)(handler._emit).toHaveBeenCalledWith(globals_1.expect.anything(), globals_1.expect.objectContaining({ content: "final result" }), runId, true);
+            (0, vitest_1.expect)(handler._emit).toHaveBeenCalledWith(vitest_1.expect.anything(), vitest_1.expect.objectContaining({ content: "final result" }), runId, true);
             // Should clean up
-            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+            (0, vitest_1.expect)(handler.metadatas[runId]).toBeUndefined();
         });
-        (0, globals_1.it)("should not emit for streaming runs that already emitted", () => {
-            const streamFn = globals_1.jest.fn();
+        (0, vitest_1.it)("should not emit for streaming runs that already emitted", () => {
+            const streamFn = vitest_1.vi.fn();
             const handler = new messages_js_1.StreamMessagesHandler(streamFn);
             const runId = "run-123";
             handler.metadatas[runId] = [["ns1"], { name: "test" }];
             // Mark as already emitted
             handler.emittedChatModelRunIds[runId] = true;
             // Mock _emit directly
-            handler._emit = globals_1.jest.fn();
+            handler._emit = vitest_1.vi.fn();
             handler.handleLLMEnd({
                 generations: [
                     [
@@ -211,24 +211,24 @@ const constants_js_1 = require("../constants.cjs");
                 ],
             }, runId);
             // Should not emit anything
-            (0, globals_1.expect)(handler._emit).not.toHaveBeenCalled();
+            (0, vitest_1.expect)(handler._emit).not.toHaveBeenCalled();
             // Should clean up metadata
-            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+            (0, vitest_1.expect)(handler.metadatas[runId]).toBeUndefined();
         });
     });
-    (0, globals_1.describe)("handleLLMError", () => {
-        (0, globals_1.it)("should clean up metadata on error", () => {
-            const handler = new messages_js_1.StreamMessagesHandler(globals_1.jest.fn());
+    (0, vitest_1.describe)("handleLLMError", () => {
+        (0, vitest_1.it)("should clean up metadata on error", () => {
+            const handler = new messages_js_1.StreamMessagesHandler(vitest_1.vi.fn());
             const runId = "run-123";
             handler.metadatas[runId] = [["ns1"], { name: "test" }];
             handler.handleLLMError(new Error("Test error"), runId);
             // Should clean up metadata
-            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+            (0, vitest_1.expect)(handler.metadatas[runId]).toBeUndefined();
         });
     });
-    (0, globals_1.describe)("handleChainStart", () => {
-        (0, globals_1.it)("should store metadata for matching node name", () => {
-            const handler = new messages_js_1.StreamMessagesHandler(globals_1.jest.fn());
+    (0, vitest_1.describe)("handleChainStart", () => {
+        (0, vitest_1.it)("should store metadata for matching node name", () => {
+            const handler = new messages_js_1.StreamMessagesHandler(vitest_1.vi.fn());
             const runId = "chain-123";
             const metadata = {
                 langgraph_checkpoint_ns: "ns1|ns2",
@@ -236,13 +236,13 @@ const constants_js_1 = require("../constants.cjs");
             };
             handler.handleChainStart({}, {}, runId, undefined, [], metadata, undefined, "NodeName" // Name matches langgraph_node
             );
-            (0, globals_1.expect)(handler.metadatas[runId]).toEqual([
+            (0, vitest_1.expect)(handler.metadatas[runId]).toEqual([
                 ["ns1", "ns2"],
                 { tags: [], name: "NodeName", ...metadata },
             ]);
         });
-        (0, globals_1.it)("should not store metadata when node name doesn't match", () => {
-            const handler = new messages_js_1.StreamMessagesHandler(globals_1.jest.fn());
+        (0, vitest_1.it)("should not store metadata when node name doesn't match", () => {
+            const handler = new messages_js_1.StreamMessagesHandler(vitest_1.vi.fn());
             const runId = "chain-123";
             const metadata = {
                 langgraph_checkpoint_ns: "ns1|ns2",
@@ -250,10 +250,10 @@ const constants_js_1 = require("../constants.cjs");
             };
             handler.handleChainStart({}, {}, runId, undefined, [], metadata, undefined, "DifferentName" // Different from langgraph_node
             );
-            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+            (0, vitest_1.expect)(handler.metadatas[runId]).toBeUndefined();
         });
-        (0, globals_1.it)("should not store metadata when TAG_HIDDEN is present", () => {
-            const handler = new messages_js_1.StreamMessagesHandler(globals_1.jest.fn());
+        (0, vitest_1.it)("should not store metadata when TAG_HIDDEN is present", () => {
+            const handler = new messages_js_1.StreamMessagesHandler(vitest_1.vi.fn());
             const runId = "chain-123";
             const metadata = {
                 langgraph_checkpoint_ns: "ns1|ns2",
@@ -261,52 +261,52 @@ const constants_js_1 = require("../constants.cjs");
             };
             handler.handleChainStart({}, {}, runId, undefined, [constants_js_1.TAG_HIDDEN], // Hidden tag
             metadata, undefined, "NodeName");
-            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+            (0, vitest_1.expect)(handler.metadatas[runId]).toBeUndefined();
         });
     });
-    (0, globals_1.describe)("handleChainEnd", () => {
-        (0, globals_1.it)("should emit a single message output", () => {
-            const streamFn = globals_1.jest.fn();
+    (0, vitest_1.describe)("handleChainEnd", () => {
+        (0, vitest_1.it)("should emit a single message output", () => {
+            const streamFn = vitest_1.vi.fn();
             const handler = new messages_js_1.StreamMessagesHandler(streamFn);
             const runId = "chain-123";
             handler.metadatas[runId] = [["ns1"], { name: "test" }];
             // Mock _emit directly
-            handler._emit = globals_1.jest.fn();
+            handler._emit = vitest_1.vi.fn();
             const message = new messages_1.AIMessage({ content: "chain result" });
             handler.handleChainEnd(message, runId);
             // Should emit the message with dedupe=true
-            (0, globals_1.expect)(handler._emit).toHaveBeenCalledWith(globals_1.expect.anything(), globals_1.expect.objectContaining({ content: "chain result" }), runId, true);
+            (0, vitest_1.expect)(handler._emit).toHaveBeenCalledWith(vitest_1.expect.anything(), vitest_1.expect.objectContaining({ content: "chain result" }), runId, true);
             // Should clean up
-            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+            (0, vitest_1.expect)(handler.metadatas[runId]).toBeUndefined();
         });
-        (0, globals_1.it)("should emit messages from an array output", () => {
-            const streamFn = globals_1.jest.fn();
+        (0, vitest_1.it)("should emit messages from an array output", () => {
+            const streamFn = vitest_1.vi.fn();
             const handler = new messages_js_1.StreamMessagesHandler(streamFn);
             const runId = "chain-123";
             handler.metadatas[runId] = [["ns1"], { name: "test" }];
             // Mock _emit directly
-            handler._emit = globals_1.jest.fn();
+            handler._emit = vitest_1.vi.fn();
             const message1 = new messages_1.AIMessage({ content: "result 1" });
             const message2 = new messages_1.AIMessage({ content: "result 2" });
             const notAMessage = "not a message";
             handler.handleChainEnd([message1, message2, notAMessage], runId);
             // Should emit both messages
-            (0, globals_1.expect)(handler._emit).toHaveBeenCalledTimes(2);
+            (0, vitest_1.expect)(handler._emit).toHaveBeenCalledTimes(2);
             // Verify calls in a way that's less brittle
             const callArgs = handler._emit.mock.calls;
             const emittedContents = callArgs.map((args) => args[1].content);
-            (0, globals_1.expect)(emittedContents).toContain("result 1");
-            (0, globals_1.expect)(emittedContents).toContain("result 2");
+            (0, vitest_1.expect)(emittedContents).toContain("result 1");
+            (0, vitest_1.expect)(emittedContents).toContain("result 2");
             // Should clean up
-            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+            (0, vitest_1.expect)(handler.metadatas[runId]).toBeUndefined();
         });
-        (0, globals_1.it)("should emit messages from object output properties", () => {
-            const streamFn = globals_1.jest.fn();
+        (0, vitest_1.it)("should emit messages from object output properties", () => {
+            const streamFn = vitest_1.vi.fn();
             const handler = new messages_js_1.StreamMessagesHandler(streamFn);
             const runId = "chain-123";
             handler.metadatas[runId] = [["ns1"], { name: "test" }];
             // Mock _emit directly
-            handler._emit = globals_1.jest.fn();
+            handler._emit = vitest_1.vi.fn();
             const message = new messages_1.AIMessage({ content: "direct result" });
             const arrayMessage = new messages_1.AIMessage({ content: "array result" });
             handler.handleChainEnd({
@@ -315,36 +315,36 @@ const constants_js_1 = require("../constants.cjs");
                 otherProp: "something else",
             }, runId);
             // Should emit both messages
-            (0, globals_1.expect)(handler._emit).toHaveBeenCalledTimes(2);
+            (0, vitest_1.expect)(handler._emit).toHaveBeenCalledTimes(2);
             // Verify calls in a way that's less brittle
             const callArgs = handler._emit.mock.calls;
             const emittedContents = callArgs.map((args) => args[1].content);
-            (0, globals_1.expect)(emittedContents).toContain("direct result");
-            (0, globals_1.expect)(emittedContents).toContain("array result");
+            (0, vitest_1.expect)(emittedContents).toContain("direct result");
+            (0, vitest_1.expect)(emittedContents).toContain("array result");
             // Should clean up
-            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+            (0, vitest_1.expect)(handler.metadatas[runId]).toBeUndefined();
         });
-        (0, globals_1.it)("should do nothing when metadata is missing", () => {
-            const streamFn = globals_1.jest.fn();
+        (0, vitest_1.it)("should do nothing when metadata is missing", () => {
+            const streamFn = vitest_1.vi.fn();
             const handler = new messages_js_1.StreamMessagesHandler(streamFn);
             const runId = "chain-123";
             // No metadata for this runId
             // Spy on _emit
-            const emitSpy = globals_1.jest.spyOn(handler, "_emit");
+            const emitSpy = vitest_1.vi.spyOn(handler, "_emit");
             const message = new messages_1.AIMessage({ content: "result" });
             handler.handleChainEnd(message, runId);
             // Should not emit anything
-            (0, globals_1.expect)(emitSpy).not.toHaveBeenCalled();
+            (0, vitest_1.expect)(emitSpy).not.toHaveBeenCalled();
         });
     });
-    (0, globals_1.describe)("handleChainError", () => {
-        (0, globals_1.it)("should clean up metadata on error", () => {
-            const handler = new messages_js_1.StreamMessagesHandler(globals_1.jest.fn());
+    (0, vitest_1.describe)("handleChainError", () => {
+        (0, vitest_1.it)("should clean up metadata on error", () => {
+            const handler = new messages_js_1.StreamMessagesHandler(vitest_1.vi.fn());
             const runId = "chain-123";
             handler.metadatas[runId] = [["ns1"], { name: "test" }];
             handler.handleChainError(new Error("Test error"), runId);
             // Should clean up metadata
-            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+            (0, vitest_1.expect)(handler.metadatas[runId]).toBeUndefined();
         });
     });
 });