@agentforge/testing 0.16.22 → 0.16.23

package/dist/index.d.cts

@@ -650,6 +650,9 @@ interface ConversationResult {
      */
     error?: Error;
 }
+interface ConversationSimulatorInput {
+    messages: BaseMessage[];
+}
 /**
  * Conversation simulator for testing multi-turn interactions
  *
@@ -670,10 +673,10 @@ interface ConversationResult {
  * expect(result.completed).toBe(true);
  * ```
  */
-declare class ConversationSimulator {
+declare class ConversationSimulator<TState = unknown> {
     private agent;
     private config;
-    constructor(agent: any, config?: ConversationSimulatorConfig);
+    constructor(agent: AgentTestAgent<ConversationSimulatorInput, TState>, config?: ConversationSimulatorConfig);
     /**
      * Simulate a conversation with predefined user inputs
      */
@@ -686,7 +689,7 @@ declare class ConversationSimulator {
 /**
  * Create a conversation simulator
  */
-declare function createConversationSimulator(agent: any, config?: ConversationSimulatorConfig): ConversationSimulator;
+declare function createConversationSimulator<TState = unknown>(agent: AgentTestAgent<ConversationSimulatorInput, TState>, config?: ConversationSimulatorConfig): ConversationSimulator<TState>;
 
 type SnapshotObject = Record<string, unknown>;
 interface SnapshotDiff {

package/dist/index.d.ts

@@ -650,6 +650,9 @@ interface ConversationResult {
      */
     error?: Error;
 }
+interface ConversationSimulatorInput {
+    messages: BaseMessage[];
+}
 /**
  * Conversation simulator for testing multi-turn interactions
  *
@@ -670,10 +673,10 @@ interface ConversationResult {
  * expect(result.completed).toBe(true);
  * ```
  */
-declare class ConversationSimulator {
+declare class ConversationSimulator<TState = unknown> {
     private agent;
     private config;
-    constructor(agent: any, config?: ConversationSimulatorConfig);
+    constructor(agent: AgentTestAgent<ConversationSimulatorInput, TState>, config?: ConversationSimulatorConfig);
     /**
      * Simulate a conversation with predefined user inputs
      */
@@ -686,7 +689,7 @@ declare class ConversationSimulator {
 /**
  * Create a conversation simulator
  */
-declare function createConversationSimulator(agent: any, config?: ConversationSimulatorConfig): ConversationSimulator;
+declare function createConversationSimulator<TState = unknown>(agent: AgentTestAgent<ConversationSimulatorInput, TState>, config?: ConversationSimulatorConfig): ConversationSimulator<TState>;
 
 type SnapshotObject = Record<string, unknown>;
 interface SnapshotDiff {

package/dist/index.js

@@ -18691,7 +18691,7 @@ var ConversationSimulator = class {
           console.log(`User: ${input}`);
         }
         const result = await this.agent.invoke({ messages });
-        const aiMessage = result.messages[result.messages.length - 1];
+        const aiMessage = extractLatestMessage(result);
         messages.push(aiMessage);
         if (this.config.verbose) {
           console.log(`AI: ${aiMessage.content}`);
@@ -18740,7 +18740,7 @@ var ConversationSimulator = class {
         const userMessage = new HumanMessage(input);
         messages.push(userMessage);
         const result = await this.agent.invoke({ messages });
-        const aiMessage = result.messages[result.messages.length - 1];
+        const aiMessage = extractLatestMessage(result);
         messages.push(aiMessage);
         turns++;
       }
@@ -18763,6 +18763,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);