@agentforge/testing 0.16.3 → 0.16.5

package/dist/index.cjs

@@ -1634,14 +1634,18 @@ var StateBuilder = class {
     Object.assign(this.state, fields);
     return this;
   }
+  ensureMessages() {
+    if (!this.state.messages) {
+      const messages = [];
+      this.state.messages = messages;
+    }
+    return this.state.messages;
+  }
   /**
    * Add a message to the messages array
    */
   addMessage(message) {
-    if (!this.state.messages) {
-      this.state.messages = [];
-    }
-    this.state.messages.push(message);
+    this.ensureMessages().push(message);
     return this;
   }
   /**
@@ -1687,7 +1691,7 @@ function createStateBuilder() {
   return new StateBuilder();
 }
 function createConversationState(messages) {
-  const builder = createStateBuilder();
+  const builder = createStateBuilder().set("messages", []);
   messages.forEach((msg, index) => {
     if (index % 2 === 0) {
       builder.addHumanMessage(msg);
@@ -1699,21 +1703,21 @@ function createConversationState(messages) {
 }
 function createReActState(config2 = {}) {
   return {
-    messages: config2.messages || [],
-    thoughts: config2.thoughts || [],
-    toolCalls: config2.toolCalls || [],
-    toolResults: config2.toolResults || [],
-    scratchpad: config2.scratchpad || [],
-    iterations: config2.iterations || 0,
-    maxIterations: config2.maxIterations || 10
+    messages: config2.messages ?? [],
+    thoughts: config2.thoughts ?? [],
+    toolCalls: config2.toolCalls ?? [],
+    toolResults: config2.toolResults ?? [],
+    scratchpad: config2.scratchpad ?? [],
+    iterations: config2.iterations ?? 0,
+    maxIterations: config2.maxIterations ?? 10
   };
 }
 function createPlanningState(config2 = {}) {
   return {
-    messages: config2.messages || [],
-    plan: config2.plan || [],
-    currentStep: config2.currentStep || 0,
-    results: config2.results || {}
+    messages: config2.messages ?? [],
+    plan: config2.plan ?? [],
+    currentStep: config2.currentStep ?? 0,
+    results: config2.results ?? {}
   };
 }
 
@@ -18344,13 +18348,25 @@ function getImporter(name) {
 __toESM(require_dist());
 
 // src/helpers/assertions.ts
+function isMessageLike(value) {
+  if (typeof value !== "object" || value === null) {
+    return false;
+  }
+  const candidate = value;
+  return "content" in candidate && typeof candidate._getType === "function";
+}
 function assertIsMessage(value, type3) {
   globalExpect(value).toBeDefined();
-  globalExpect(value).toHaveProperty("content");
-  if (type3 === "human") {
-    globalExpect(value).toBeInstanceOf(messages.HumanMessage);
-  } else if (type3 === "ai") {
-    globalExpect(value).toBeInstanceOf(messages.AIMessage);
+  globalExpect(value).not.toBeNull();
+  globalExpect(typeof value).toBe("object");
+  const samePackageMessage = value instanceof messages.BaseMessage;
+  const structuralMessage = isMessageLike(value);
+  globalExpect(samePackageMessage || structuralMessage).toBe(true);
+  const message = value;
+  const messageType = message._getType();
+  globalExpect(typeof messageType).toBe("string");
+  if (type3) {
+    globalExpect(messageType).toBe(type3);
   }
 }
 function assertMessageContains(messages, content) {
@@ -18366,7 +18382,7 @@ function assertLastMessageContains(messages, content) {
 }
 function assertStateHasFields(state, fields) {
   fields.forEach((field) => {
-    globalExpect(state).toHaveProperty(field);
+    globalExpect(state).toHaveProperty(typeof field === "number" ? [field] : field);
   });
 }
 function assertToolCalled(toolCalls, toolName, args) {