@agentforge/testing 0.16.21 → 0.16.23

package/README.md

@@ -6,11 +6,11 @@
 [![TypeScript](https://img.shields.io/badge/TypeScript-5.9-blue)](https://www.typescriptlang.org/)
 [![License](https://img.shields.io/badge/license-MIT-green)](../../LICENSE)
 
-## 🎉 Status: Production Ready & Published
+## Status: Production Ready & Published
 
 **Complete testing toolkit** | **Full TypeScript support** | **Comprehensive documentation**
 
-## 📦 Installation
+## Installation
 
 ```bash
 npm install --save-dev @agentforge/testing
@@ -20,15 +20,15 @@ pnpm add -D @agentforge/testing
 yarn add -D @agentforge/testing
 ```
 
-## ✨ Features
+## Features
 
-- 🎭 **Mock Factories** - Create mock LLMs, tools, and states for testing
-- 🔧 **Test Helpers** - Assertion helpers and state builders
-- 📦 **Fixtures** - Pre-built sample agents, tools, and conversations
-- 🏃 **Test Runners** - Agent test runner and conversation simulator
-- 📸 **Snapshot Testing** - State and message snapshot utilities
-- ✅ **Full TypeScript** - Complete type safety and inference
-- 🧪 **Vitest Integration** - Works seamlessly with Vitest
+- **Mock Factories** - Create mock LLMs, tools, and states for testing
+- **Test Helpers** - Assertion helpers and state builders
+- **Fixtures** - Pre-built sample agents, tools, and conversations
+- **Test Runners** - Agent test runner and conversation simulator
+- **Snapshot Testing** - State and message snapshot utilities
+- **Full TypeScript** - Complete type safety and inference
+- **Vitest Integration** - Works seamlessly with Vitest
 
 ## Quick Start
 
@@ -368,21 +368,21 @@ describe('ReAct Agent Integration Tests', () => {
 - `calculatorTool` - Calculator tool
 - `searchTool` - Search tool
 
-## 📖 Documentation
+## Documentation
 
-- 📚 **[Full Documentation](https://tvscoundrel.github.io/agentforge/)**
-- 🚀 **[Quick Start](https://tvscoundrel.github.io/agentforge/guide/quick-start)**
-- 🧪 **[Testing API Reference](https://tvscoundrel.github.io/agentforge/api/testing)**
-- 💡 **[Testing Tutorial](https://tvscoundrel.github.io/agentforge/tutorials/testing)**
+- **[Full Documentation](https://tvscoundrel.github.io/agentforge/)**
+- **[Quick Start](https://tvscoundrel.github.io/agentforge/guide/quick-start)**
+- **[Testing API Reference](https://tvscoundrel.github.io/agentforge/api/testing)**
+- **[Testing Tutorial](https://tvscoundrel.github.io/agentforge/tutorials/testing)**
 
-## 🔗 Links
+## Links
 
 - [GitHub Repository](https://github.com/TVScoundrel/agentforge)
 - [npm Package](https://www.npmjs.com/package/@agentforge/testing)
 - [Changelog](https://tvscoundrel.github.io/agentforge/changelog.html) - See what's new before upgrading
 - [Report Issues](https://github.com/TVScoundrel/agentforge/issues)
 
-## 📚 Related Packages
+## Related Packages
 
 - [@agentforge/core](https://www.npmjs.com/package/@agentforge/core) - Core abstractions
 - [@agentforge/patterns](https://www.npmjs.com/package/@agentforge/patterns) - Agent patterns

package/dist/index.cjs

@@ -12808,14 +12808,14 @@ function withTimeout(fn2, timeout, isHook = false) {
   if (timeout <= 0 || timeout === Number.POSITIVE_INFINITY) {
     return fn2;
   }
-  const { setTimeout: setTimeout2, clearTimeout } = getSafeTimers();
+  const { setTimeout: setTimeout2, clearTimeout: clearTimeout2 } = getSafeTimers();
   return function runWithTimeout(...args) {
     return Promise.race([
       fn2(...args),
       new Promise((resolve4, reject) => {
         var _a;
         const timer = setTimeout2(() => {
-          clearTimeout(timer);
+          clearTimeout2(timer);
           reject(new Error(makeTimeoutMsg(isHook, timeout)));
         }, timeout);
         (_a = timer.unref) == null ? void 0 : _a.call(timer);
@@ -15673,9 +15673,9 @@ function createExpectPoll(expect2) {
           const promise = () => new Promise((resolve4, reject) => {
             let intervalId;
             let lastError;
-            const { setTimeout: setTimeout2, clearTimeout } = getSafeTimers();
+            const { setTimeout: setTimeout2, clearTimeout: clearTimeout2 } = getSafeTimers();
             const timeoutId = setTimeout2(() => {
-              clearTimeout(intervalId);
+              clearTimeout2(intervalId);
               reject(
                 copyStackTrace$1(
                   new Error(`Matcher did not succeed in ${timeout}ms`, {
@@ -15691,8 +15691,8 @@ function createExpectPoll(expect2) {
                 const obj = await fn2();
                 utils_exports.flag(assertion, "object", obj);
                 resolve4(await assertionFunction.call(assertion, ...args));
-                clearTimeout(intervalId);
-                clearTimeout(timeoutId);
+                clearTimeout2(intervalId);
+                clearTimeout2(timeoutId);
               } catch (err) {
                 lastError = err;
                 intervalId = setTimeout2(check, interval);
@@ -17283,7 +17283,7 @@ To automatically clean-up native timers, use \`shouldClearNativeTimers\`.`
           });
         };
       }
-      clock.clearTimeout = function clearTimeout(timerId) {
+      clock.clearTimeout = function clearTimeout2(timerId) {
         return clearTimer(clock, timerId, "Timeout");
       };
       clock.nextTick = function nextTick(func) {
@@ -17926,7 +17926,7 @@ function copyStackTrace(target, source) {
   return target;
 }
 function waitFor(callback, options = {}) {
-  const { setTimeout: setTimeout2, setInterval, clearTimeout, clearInterval } = getSafeTimers();
+  const { setTimeout: setTimeout2, setInterval, clearTimeout: clearTimeout2, clearInterval } = getSafeTimers();
   const { interval = 50, timeout = 1e3 } = typeof options === "number" ? { timeout: options } : options;
   const STACK_TRACE_ERROR = new Error("STACK_TRACE_ERROR");
   return new Promise((resolve4, reject) => {
@@ -17936,7 +17936,7 @@ function waitFor(callback, options = {}) {
     let intervalId;
     const onResolve = (result) => {
       if (timeoutId) {
-        clearTimeout(timeoutId);
+        clearTimeout2(timeoutId);
       }
       if (intervalId) {
         clearInterval(intervalId);
@@ -17994,7 +17994,7 @@ function waitFor(callback, options = {}) {
   });
 }
 function waitUntil(callback, options = {}) {
-  const { setTimeout: setTimeout2, setInterval, clearTimeout, clearInterval } = getSafeTimers();
+  const { setTimeout: setTimeout2, setInterval, clearTimeout: clearTimeout2, clearInterval } = getSafeTimers();
   const { interval = 50, timeout = 1e3 } = typeof options === "number" ? { timeout: options } : options;
   const STACK_TRACE_ERROR = new Error("STACK_TRACE_ERROR");
   return new Promise((resolve4, reject) => {
@@ -18018,7 +18018,7 @@ function waitUntil(callback, options = {}) {
         return;
       }
       if (timeoutId) {
-        clearTimeout(timeoutId);
+        clearTimeout2(timeoutId);
       }
       if (intervalId) {
         clearInterval(intervalId);
@@ -18604,20 +18604,21 @@ var AgentTestRunner = class {
     let messages = [];
     let passed = true;
     let error;
+    let timeoutId;
     try {
-      const timeout = this.config.timeout || 3e4;
+      const timeout = this.config.timeout ?? 3e4;
       const timeoutPromise = new Promise((_, reject) => {
-        setTimeout(() => reject(new Error("Agent test timeout")), timeout);
+        timeoutId = setTimeout(() => reject(new Error("Agent test timeout")), timeout);
       });
       const runPromise = (async () => {
         if (this.config.captureSteps) {
           const result = await this.agent.invoke(input);
           finalState = result;
-          messages = result.messages || [];
+          messages = extractMessages(result);
         } else {
           const result = await this.agent.invoke(input);
           finalState = result;
-          messages = result.messages || [];
+          messages = extractMessages(result);
         }
         if (this.config.validateState && this.config.stateValidator) {
           const isValid = await this.config.stateValidator(finalState);
@@ -18626,7 +18627,13 @@ var AgentTestRunner = class {
           }
         }
       })();
-      await Promise.race([runPromise, timeoutPromise]);
+      try {
+        await Promise.race([runPromise, timeoutPromise]);
+      } finally {
+        if (timeoutId !== void 0) {
+          clearTimeout(timeoutId);
+        }
+      }
     } catch (err) {
       passed = false;
       error = err;
@@ -18651,6 +18658,13 @@ var AgentTestRunner = class {
 function createAgentTestRunner(agent, config2) {
   return new AgentTestRunner(agent, config2);
 }
+function extractMessages(state) {
+  if (typeof state !== "object" || state === null) {
+    return [];
+  }
+  const { messages } = state;
+  return Array.isArray(messages) ? messages : [];
+}
 var ConversationSimulator = class {
   constructor(agent, config2 = {}) {
     this.agent = agent;
@@ -18679,7 +18693,7 @@ var ConversationSimulator = class {
           console.log(`User: ${input}`);
         }
         const result = await this.agent.invoke({ messages: messages$1 });
-        const aiMessage = result.messages[result.messages.length - 1];
+        const aiMessage = extractLatestMessage(result);
         messages$1.push(aiMessage);
         if (this.config.verbose) {
           console.log(`AI: ${aiMessage.content}`);
@@ -18728,7 +18742,7 @@ var ConversationSimulator = class {
         const userMessage = new messages.HumanMessage(input);
         messages$1.push(userMessage);
         const result = await this.agent.invoke({ messages: messages$1 });
-        const aiMessage = result.messages[result.messages.length - 1];
+        const aiMessage = extractLatestMessage(result);
         messages$1.push(aiMessage);
         turns++;
       }
@@ -18751,6 +18765,14 @@ var ConversationSimulator = class {
 function createConversationSimulator(agent, config2) {
   return new ConversationSimulator(agent, config2);
 }
+function extractLatestMessage(state) {
+  const messages = extractMessages(state);
+  const latestMessage = messages[messages.length - 1];
+  if (!latestMessage) {
+    throw new Error("Agent response did not include any messages");
+  }
+  return latestMessage;
+}
 var ROOT_SNAPSHOT_DIFF_KEY = "$root";
 function createSnapshotObject() {
   return /* @__PURE__ */ Object.create(null);