@agentforge/testing 0.16.19 → 0.16.21

package/dist/index.cjs

@@ -5,6 +5,7 @@ var messages = require('@langchain/core/messages');
 var zod = require('zod');
 var core = require('@agentforge/core');
 var tty = require('tty');
+var util = require('util');
 
 var __create = Object.create;
 var __defProp = Object.defineProperty;
@@ -18750,31 +18751,38 @@ var ConversationSimulator = class {
 function createConversationSimulator(agent, config2) {
   return new ConversationSimulator(agent, config2);
 }
-
-// src/runners/snapshot-testing.ts
-function normalizeValue(value, config2) {
-  if (value === null || value === void 0) {
-    return value;
+var ROOT_SNAPSHOT_DIFF_KEY = "$root";
+function createSnapshotObject() {
+  return /* @__PURE__ */ Object.create(null);
+}
+function isPlainSnapshotObject(value) {
+  if (typeof value !== "object" || value === null || Array.isArray(value)) {
+    return false;
   }
-  if (config2.normalizer) {
-    return config2.normalizer(value);
+  const prototype = Object.getPrototypeOf(value);
+  return prototype === Object.prototype || prototype === null;
+}
+function normalizeValue(value, config2) {
+  const valueToNormalize = config2.normalizer ? config2.normalizer(value) : value;
+  if (valueToNormalize === null || valueToNormalize === void 0) {
+    return valueToNormalize;
   }
-  if (config2.normalizeTimestamps && value instanceof Date) {
+  if (config2.normalizeTimestamps && valueToNormalize instanceof Date) {
     return "[TIMESTAMP]";
   }
-  if (config2.normalizeTimestamps && typeof value === "string") {
-    if (/^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}/.test(value)) {
+  if (config2.normalizeTimestamps && typeof valueToNormalize === "string") {
+    if (/^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}/.test(valueToNormalize)) {
       return "[TIMESTAMP]";
     }
   }
-  if (config2.normalizeIds && typeof value === "string") {
-    if (/^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i.test(value)) {
+  if (config2.normalizeIds && typeof valueToNormalize === "string") {
+    if (/^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i.test(valueToNormalize)) {
       return "[UUID]";
     }
   }
-  if (typeof value === "object" && !Array.isArray(value)) {
-    const normalized = {};
-    for (const [key, val] of Object.entries(value)) {
+  if (isPlainSnapshotObject(valueToNormalize)) {
+    const normalized = createSnapshotObject();
+    for (const [key, val] of Object.entries(valueToNormalize)) {
       if (config2.excludeFields?.includes(key)) {
         continue;
       }
@@ -18785,10 +18793,10 @@ function normalizeValue(value, config2) {
     }
     return normalized;
   }
-  if (Array.isArray(value)) {
-    return value.map((item) => normalizeValue(item, config2));
+  if (Array.isArray(valueToNormalize)) {
+    return valueToNormalize.map((item) => normalizeValue(item, config2));
   }
-  return value;
+  return valueToNormalize;
 }
 function createSnapshot(state, config2 = {}) {
   return normalizeValue(state, {
@@ -18804,35 +18812,45 @@ function assertMatchesSnapshot(state, config2) {
 function createMessageSnapshot(messages, config2) {
   return messages.map((msg) => ({
     type: msg._getType(),
-    content: msg.content
+    content: normalizeValue(msg.content, {
+      normalizeTimestamps: true,
+      normalizeIds: true,
+      ...config2
+    })
   }));
 }
 function assertMessagesMatchSnapshot(messages, config2) {
-  const snapshot = createMessageSnapshot(messages);
+  const snapshot = createMessageSnapshot(messages, config2);
   globalExpect(snapshot).toMatchSnapshot();
 }
 function compareStates(state1, state2, config2) {
   const snapshot1 = createSnapshot(state1, config2);
   const snapshot2 = createSnapshot(state2, config2);
-  return JSON.stringify(snapshot1) === JSON.stringify(snapshot2);
+  return util.isDeepStrictEqual(snapshot1, snapshot2);
 }
 function createStateDiff(state1, state2, config2) {
   const snapshot1 = createSnapshot(state1, config2);
   const snapshot2 = createSnapshot(state2, config2);
   const diff2 = {
-    added: {},
-    removed: {},
-    changed: {}
+    added: createSnapshotObject(),
+    removed: createSnapshotObject(),
+    changed: /* @__PURE__ */ Object.create(null)
   };
+  if (!isPlainSnapshotObject(snapshot1) || !isPlainSnapshotObject(snapshot2)) {
+    if (!util.isDeepStrictEqual(snapshot1, snapshot2)) {
+      diff2.changed[ROOT_SNAPSHOT_DIFF_KEY] = { from: snapshot1, to: snapshot2 };
+    }
+    return diff2;
+  }
   for (const [key, value] of Object.entries(snapshot2)) {
-    if (!(key in snapshot1)) {
+    if (!Object.hasOwn(snapshot1, key)) {
       diff2.added[key] = value;
-    } else if (JSON.stringify(snapshot1[key]) !== JSON.stringify(value)) {
+    } else if (!util.isDeepStrictEqual(snapshot1[key], value)) {
       diff2.changed[key] = { from: snapshot1[key], to: value };
     }
   }
   for (const key of Object.keys(snapshot1)) {
-    if (!(key in snapshot2)) {
+    if (!Object.hasOwn(snapshot2, key)) {
       diff2.removed[key] = snapshot1[key];
     }
   }
@@ -19121,6 +19139,7 @@ chai/index.js:
 exports.AgentTestRunner = AgentTestRunner;
 exports.ConversationSimulator = ConversationSimulator;
 exports.MockLLM = MockLLM;
+exports.ROOT_SNAPSHOT_DIFF_KEY = ROOT_SNAPSHOT_DIFF_KEY;
 exports.StateBuilder = StateBuilder;
 exports.assertAlternatingMessages = assertAlternatingMessages;
 exports.assertCompletesWithin = assertCompletesWithin;