@agentforge/testing 0.16.4 → 0.16.5

package/dist/index.d.cts

@@ -150,19 +150,54 @@ declare function createDelayedTool(name?: string, delay?: number): _agentforge_c
  */
 declare function createCalculatorTool(): _agentforge_core.Tool<any, string>;
 
+type StateBuilderFields = Record<string, unknown> & {
+    messages?: BaseMessage[];
+};
+type MessageState = {
+    messages: BaseMessage[];
+};
+type BuiltState<TState extends StateBuilderFields> = TState & Partial<MessageState>;
+interface TestToolCall<TArgs = unknown> {
+    name: string;
+    args: TArgs;
+}
+interface TestToolResult<TResult = string> {
+    name: string;
+    result: TResult;
+}
+interface ReActTestState<TArgs = unknown, TResult = string> {
+    messages: BaseMessage[];
+    thoughts: string[];
+    toolCalls: Array<TestToolCall<TArgs>>;
+    toolResults: Array<TestToolResult<TResult>>;
+    scratchpad: string[];
+    iterations: number;
+    maxIterations: number;
+}
+interface PlanningStep<TStatus extends string = string> {
+    step: string;
+    status: TStatus;
+}
+interface PlanningTestState<TResultMap extends Record<string, unknown> = Record<string, unknown>, TStatus extends string = string> extends StateBuilderFields {
+    messages: BaseMessage[];
+    plan: Array<PlanningStep<TStatus>>;
+    currentStep: number;
+    results: Partial<TResultMap>;
+}
 /**
  * Builder for creating test states
  */
-declare class StateBuilder<T extends Record<string, any> = Record<string, any>> {
+declare class StateBuilder<TState extends StateBuilderFields = StateBuilderFields> {
     private state;
     /**
      * Set a field in the state
      */
-    set<K extends keyof T>(key: K, value: T[K]): this;
+    set<K extends keyof TState>(key: K, value: TState[K]): this;
     /**
      * Set multiple fields in the state
      */
-    setMany(fields: Partial<T>): this;
+    setMany(fields: Partial<TState>): this;
+    private ensureMessages;
     /**
      * Add a message to the messages array
      */
@@ -170,7 +205,7 @@ declare class StateBuilder<T extends Record<string, any> = Record<string, any>>
     /**
      * Add multiple messages
      */
-    addMessages(messages: BaseMessage[]): this;
+    addMessages(messages: ReadonlyArray<BaseMessage>): this;
     /**
      * Add a human message
      */
@@ -186,7 +221,7 @@ declare class StateBuilder<T extends Record<string, any> = Record<string, any>>
     /**
      * Build the state
      */
-    build(): T;
+    build(): BuiltState<TState>;
     /**
      * Reset the builder
      */
@@ -195,48 +230,37 @@ declare class StateBuilder<T extends Record<string, any> = Record<string, any>>
 /**
  * Create a state builder
  */
-declare function createStateBuilder<T extends Record<string, any> = Record<string, any>>(): StateBuilder<T>;
+declare function createStateBuilder<TState extends StateBuilderFields = StateBuilderFields>(): StateBuilder<TState>;
 /**
  * Create a simple conversation state
  */
-declare function createConversationState(messages: string[]): {
-    messages: BaseMessage[];
-};
+declare function createConversationState(messages: ReadonlyArray<string>): MessageState;
 /**
  * Create a ReAct agent state
  */
-declare function createReActState(config?: {
-    messages?: BaseMessage[];
-    thoughts?: string[];
-    toolCalls?: Array<{
-        name: string;
-        args: any;
-    }>;
-    toolResults?: Array<{
-        name: string;
-        result: string;
-    }>;
-    scratchpad?: string[];
-    iterations?: number;
-    maxIterations?: number;
-}): any;
+declare function createReActState<TArgs = unknown, TResult = string>(config?: Partial<ReActTestState<TArgs, TResult>>): ReActTestState<TArgs, TResult>;
 /**
  * Create a planning agent state
  */
-declare function createPlanningState(config?: {
-    messages?: BaseMessage[];
-    plan?: Array<{
-        step: string;
-        status: string;
-    }>;
-    currentStep?: number;
-    results?: Record<string, any>;
-}): any;
+declare function createPlanningState<TResultMap extends Record<string, unknown> = Record<string, unknown>, TStatus extends string = string>(config?: Partial<PlanningTestState<TResultMap, TStatus>>): PlanningTestState<TResultMap, TStatus>;
 
+type ToolCall<TArgs = unknown> = {
+    name: string;
+    args: TArgs;
+};
+type MessageLike<TType extends string = string> = {
+    content: unknown;
+    _getType: () => TType;
+};
+type AssertedMessage<TType extends string = string> = BaseMessage | MessageLike<TType>;
 /**
  * Assert that a value is a message of a specific type
  */
-declare function assertIsMessage(value: any, type?: 'human' | 'ai' | 'system'): asserts value is BaseMessage;
+declare function assertIsMessage(value: unknown): asserts value is AssertedMessage;
+declare function assertIsMessage(value: unknown, type: 'human'): asserts value is AssertedMessage<'human'>;
+declare function assertIsMessage(value: unknown, type: 'ai'): asserts value is AssertedMessage<'ai'>;
+declare function assertIsMessage(value: unknown, type: 'system'): asserts value is AssertedMessage<'system'>;
+declare function assertIsMessage(value: unknown, type: 'tool'): asserts value is AssertedMessage<'tool'>;
 /**
  * Assert that messages array contains a message with specific content
  */
@@ -248,26 +272,24 @@ declare function assertLastMessageContains(messages: BaseMessage[], content: str
 /**
  * Assert that state has required fields
  */
-declare function assertStateHasFields<T extends Record<string, any>>(state: T, fields: (keyof T)[]): void;
+declare function assertStateHasFields<TState extends object>(state: TState, fields: ReadonlyArray<keyof TState & (string | number)>): void;
 /**
  * Assert that a tool was called with specific arguments
  */
-declare function assertToolCalled(toolCalls: Array<{
-    name: string;
-    args: any;
-}>, toolName: string, args?: Record<string, any>): void;
+declare function assertToolCalled(toolCalls: ReadonlyArray<ToolCall>, toolName: string): void;
+declare function assertToolCalled<TArgs extends Record<string, unknown>>(toolCalls: ReadonlyArray<ToolCall<TArgs>>, toolName: string, args: Partial<TArgs>): void;
 /**
  * Assert that execution completed within time limit
  */
-declare function assertCompletesWithin(fn: () => Promise<any>, maxMs: number): Promise<void>;
+declare function assertCompletesWithin(fn: () => Promise<unknown>, maxMs: number): Promise<void>;
 /**
  * Assert that a function throws an error with specific message
  */
-declare function assertThrowsWithMessage(fn: () => Promise<any>, message: string): Promise<void>;
+declare function assertThrowsWithMessage(fn: () => Promise<unknown>, message: string): Promise<void>;
 /**
  * Assert that state matches a snapshot
  */
-declare function assertStateSnapshot(state: any, snapshot: any): void;
+declare function assertStateSnapshot<TState extends object>(state: TState, snapshot: Partial<TState>): void;
 /**
  * Assert that messages have alternating types (human, ai, human, ai, ...)
  */
@@ -275,7 +297,7 @@ declare function assertAlternatingMessages(messages: BaseMessage[]): void;
 /**
  * Assert that an array is not empty
  */
-declare function assertNotEmpty<T>(array: T[]): void;
+declare function assertNotEmpty<T>(array: readonly T[]): void;
 /**
  * Assert that a value is within a range
  */
@@ -287,7 +309,7 @@ declare function assertIterationsWithinLimit(iterations: number, maxIterations:
 /**
  * Assert that a result contains expected keys
  */
-declare function assertHasKeys<T extends Record<string, any>>(obj: T, keys: string[]): void;
+declare function assertHasKeys<TObject extends object>(obj: TObject, keys: ReadonlyArray<keyof TObject & string>): void;
 
 /**
  * Sample conversation: Simple greeting
@@ -705,4 +727,4 @@ declare function createStateDiff(state1: any, state2: any, config?: SnapshotConf
  */
 declare function assertStateChanged(stateBefore: any, stateAfter: any, expectedChanges: string[], config?: SnapshotConfig): void;
 
-export { type AgentTestConfig, type AgentTestResult, AgentTestRunner, type ConversationResult, ConversationSimulator, type ConversationSimulatorConfig, MockLLM, type MockLLMConfig, type MockToolConfig, type SnapshotConfig, StateBuilder, assertAlternatingMessages, assertCompletesWithin, assertHasKeys, assertInRange, assertIsMessage, assertIterationsWithinLimit, assertLastMessageContains, assertMatchesSnapshot, assertMessageContains, assertMessagesMatchSnapshot, assertNotEmpty, assertStateChanged, assertStateHasFields, assertStateSnapshot, assertThrowsWithMessage, assertToolCalled, calculatorTool, compareStates, complexReasoningConversation, createAgentTestRunner, createCalculatorTool, createConversation, createConversationSimulator, createConversationState, createConversationWithSystem, createDelayedTool, createEchoLLM, createEchoTool, createErrorLLM, createErrorTool, createMessageSnapshot, createMockLLM, createMockTool, createPlanningState, createReActState, createSnapshot, createStateBuilder, createStateDiff, databaseQueryTool, errorHandlingConversation, fileReaderTool, getToolByName, getToolsByCategory, longContextConversation, multiTurnConversation, sampleData, sampleTools, searchTool, simpleGreeting, timeTool, toolUsageConversation, weatherTool };
+export { type AgentTestConfig, type AgentTestResult, AgentTestRunner, type AssertedMessage, type ConversationResult, ConversationSimulator, type ConversationSimulatorConfig, MockLLM, type MockLLMConfig, type MockToolConfig, type PlanningStep, type PlanningTestState, type ReActTestState, type SnapshotConfig, StateBuilder, type TestToolCall, type TestToolResult, assertAlternatingMessages, assertCompletesWithin, assertHasKeys, assertInRange, assertIsMessage, assertIterationsWithinLimit, assertLastMessageContains, assertMatchesSnapshot, assertMessageContains, assertMessagesMatchSnapshot, assertNotEmpty, assertStateChanged, assertStateHasFields, assertStateSnapshot, assertThrowsWithMessage, assertToolCalled, calculatorTool, compareStates, complexReasoningConversation, createAgentTestRunner, createCalculatorTool, createConversation, createConversationSimulator, createConversationState, createConversationWithSystem, createDelayedTool, createEchoLLM, createEchoTool, createErrorLLM, createErrorTool, createMessageSnapshot, createMockLLM, createMockTool, createPlanningState, createReActState, createSnapshot, createStateBuilder, createStateDiff, databaseQueryTool, errorHandlingConversation, fileReaderTool, getToolByName, getToolsByCategory, longContextConversation, multiTurnConversation, sampleData, sampleTools, searchTool, simpleGreeting, timeTool, toolUsageConversation, weatherTool };

package/dist/index.d.ts

@@ -150,19 +150,54 @@ declare function createDelayedTool(name?: string, delay?: number): _agentforge_c
  */
 declare function createCalculatorTool(): _agentforge_core.Tool<any, string>;
 
+type StateBuilderFields = Record<string, unknown> & {
+    messages?: BaseMessage[];
+};
+type MessageState = {
+    messages: BaseMessage[];
+};
+type BuiltState<TState extends StateBuilderFields> = TState & Partial<MessageState>;
+interface TestToolCall<TArgs = unknown> {
+    name: string;
+    args: TArgs;
+}
+interface TestToolResult<TResult = string> {
+    name: string;
+    result: TResult;
+}
+interface ReActTestState<TArgs = unknown, TResult = string> {
+    messages: BaseMessage[];
+    thoughts: string[];
+    toolCalls: Array<TestToolCall<TArgs>>;
+    toolResults: Array<TestToolResult<TResult>>;
+    scratchpad: string[];
+    iterations: number;
+    maxIterations: number;
+}
+interface PlanningStep<TStatus extends string = string> {
+    step: string;
+    status: TStatus;
+}
+interface PlanningTestState<TResultMap extends Record<string, unknown> = Record<string, unknown>, TStatus extends string = string> extends StateBuilderFields {
+    messages: BaseMessage[];
+    plan: Array<PlanningStep<TStatus>>;
+    currentStep: number;
+    results: Partial<TResultMap>;
+}
 /**
  * Builder for creating test states
  */
-declare class StateBuilder<T extends Record<string, any> = Record<string, any>> {
+declare class StateBuilder<TState extends StateBuilderFields = StateBuilderFields> {
     private state;
     /**
      * Set a field in the state
      */
-    set<K extends keyof T>(key: K, value: T[K]): this;
+    set<K extends keyof TState>(key: K, value: TState[K]): this;
     /**
      * Set multiple fields in the state
      */
-    setMany(fields: Partial<T>): this;
+    setMany(fields: Partial<TState>): this;
+    private ensureMessages;
     /**
      * Add a message to the messages array
      */
@@ -170,7 +205,7 @@ declare class StateBuilder<T extends Record<string, any> = Record<string, any>>
     /**
      * Add multiple messages
      */
-    addMessages(messages: BaseMessage[]): this;
+    addMessages(messages: ReadonlyArray<BaseMessage>): this;
     /**
      * Add a human message
      */
@@ -186,7 +221,7 @@ declare class StateBuilder<T extends Record<string, any> = Record<string, any>>
     /**
      * Build the state
      */
-    build(): T;
+    build(): BuiltState<TState>;
     /**
      * Reset the builder
      */
@@ -195,48 +230,37 @@ declare class StateBuilder<T extends Record<string, any> = Record<string, any>>
 /**
  * Create a state builder
  */
-declare function createStateBuilder<T extends Record<string, any> = Record<string, any>>(): StateBuilder<T>;
+declare function createStateBuilder<TState extends StateBuilderFields = StateBuilderFields>(): StateBuilder<TState>;
 /**
  * Create a simple conversation state
  */
-declare function createConversationState(messages: string[]): {
-    messages: BaseMessage[];
-};
+declare function createConversationState(messages: ReadonlyArray<string>): MessageState;
 /**
  * Create a ReAct agent state
  */
-declare function createReActState(config?: {
-    messages?: BaseMessage[];
-    thoughts?: string[];
-    toolCalls?: Array<{
-        name: string;
-        args: any;
-    }>;
-    toolResults?: Array<{
-        name: string;
-        result: string;
-    }>;
-    scratchpad?: string[];
-    iterations?: number;
-    maxIterations?: number;
-}): any;
+declare function createReActState<TArgs = unknown, TResult = string>(config?: Partial<ReActTestState<TArgs, TResult>>): ReActTestState<TArgs, TResult>;
 /**
  * Create a planning agent state
  */
-declare function createPlanningState(config?: {
-    messages?: BaseMessage[];
-    plan?: Array<{
-        step: string;
-        status: string;
-    }>;
-    currentStep?: number;
-    results?: Record<string, any>;
-}): any;
+declare function createPlanningState<TResultMap extends Record<string, unknown> = Record<string, unknown>, TStatus extends string = string>(config?: Partial<PlanningTestState<TResultMap, TStatus>>): PlanningTestState<TResultMap, TStatus>;
 
+type ToolCall<TArgs = unknown> = {
+    name: string;
+    args: TArgs;
+};
+type MessageLike<TType extends string = string> = {
+    content: unknown;
+    _getType: () => TType;
+};
+type AssertedMessage<TType extends string = string> = BaseMessage | MessageLike<TType>;
 /**
  * Assert that a value is a message of a specific type
  */
-declare function assertIsMessage(value: any, type?: 'human' | 'ai' | 'system'): asserts value is BaseMessage;
+declare function assertIsMessage(value: unknown): asserts value is AssertedMessage;
+declare function assertIsMessage(value: unknown, type: 'human'): asserts value is AssertedMessage<'human'>;
+declare function assertIsMessage(value: unknown, type: 'ai'): asserts value is AssertedMessage<'ai'>;
+declare function assertIsMessage(value: unknown, type: 'system'): asserts value is AssertedMessage<'system'>;
+declare function assertIsMessage(value: unknown, type: 'tool'): asserts value is AssertedMessage<'tool'>;
 /**
  * Assert that messages array contains a message with specific content
  */
@@ -248,26 +272,24 @@ declare function assertLastMessageContains(messages: BaseMessage[], content: str
 /**
  * Assert that state has required fields
  */
-declare function assertStateHasFields<T extends Record<string, any>>(state: T, fields: (keyof T)[]): void;
+declare function assertStateHasFields<TState extends object>(state: TState, fields: ReadonlyArray<keyof TState & (string | number)>): void;
 /**
  * Assert that a tool was called with specific arguments
  */
-declare function assertToolCalled(toolCalls: Array<{
-    name: string;
-    args: any;
-}>, toolName: string, args?: Record<string, any>): void;
+declare function assertToolCalled(toolCalls: ReadonlyArray<ToolCall>, toolName: string): void;
+declare function assertToolCalled<TArgs extends Record<string, unknown>>(toolCalls: ReadonlyArray<ToolCall<TArgs>>, toolName: string, args: Partial<TArgs>): void;
 /**
  * Assert that execution completed within time limit
  */
-declare function assertCompletesWithin(fn: () => Promise<any>, maxMs: number): Promise<void>;
+declare function assertCompletesWithin(fn: () => Promise<unknown>, maxMs: number): Promise<void>;
 /**
  * Assert that a function throws an error with specific message
  */
-declare function assertThrowsWithMessage(fn: () => Promise<any>, message: string): Promise<void>;
+declare function assertThrowsWithMessage(fn: () => Promise<unknown>, message: string): Promise<void>;
 /**
  * Assert that state matches a snapshot
  */
-declare function assertStateSnapshot(state: any, snapshot: any): void;
+declare function assertStateSnapshot<TState extends object>(state: TState, snapshot: Partial<TState>): void;
 /**
  * Assert that messages have alternating types (human, ai, human, ai, ...)
  */
@@ -275,7 +297,7 @@ declare function assertAlternatingMessages(messages: BaseMessage[]): void;
 /**
  * Assert that an array is not empty
  */
-declare function assertNotEmpty<T>(array: T[]): void;
+declare function assertNotEmpty<T>(array: readonly T[]): void;
 /**
  * Assert that a value is within a range
  */
@@ -287,7 +309,7 @@ declare function assertIterationsWithinLimit(iterations: number, maxIterations:
 /**
  * Assert that a result contains expected keys
  */
-declare function assertHasKeys<T extends Record<string, any>>(obj: T, keys: string[]): void;
+declare function assertHasKeys<TObject extends object>(obj: TObject, keys: ReadonlyArray<keyof TObject & string>): void;
 
 /**
  * Sample conversation: Simple greeting
@@ -705,4 +727,4 @@ declare function createStateDiff(state1: any, state2: any, config?: SnapshotConf
  */
 declare function assertStateChanged(stateBefore: any, stateAfter: any, expectedChanges: string[], config?: SnapshotConfig): void;
 
-export { type AgentTestConfig, type AgentTestResult, AgentTestRunner, type ConversationResult, ConversationSimulator, type ConversationSimulatorConfig, MockLLM, type MockLLMConfig, type MockToolConfig, type SnapshotConfig, StateBuilder, assertAlternatingMessages, assertCompletesWithin, assertHasKeys, assertInRange, assertIsMessage, assertIterationsWithinLimit, assertLastMessageContains, assertMatchesSnapshot, assertMessageContains, assertMessagesMatchSnapshot, assertNotEmpty, assertStateChanged, assertStateHasFields, assertStateSnapshot, assertThrowsWithMessage, assertToolCalled, calculatorTool, compareStates, complexReasoningConversation, createAgentTestRunner, createCalculatorTool, createConversation, createConversationSimulator, createConversationState, createConversationWithSystem, createDelayedTool, createEchoLLM, createEchoTool, createErrorLLM, createErrorTool, createMessageSnapshot, createMockLLM, createMockTool, createPlanningState, createReActState, createSnapshot, createStateBuilder, createStateDiff, databaseQueryTool, errorHandlingConversation, fileReaderTool, getToolByName, getToolsByCategory, longContextConversation, multiTurnConversation, sampleData, sampleTools, searchTool, simpleGreeting, timeTool, toolUsageConversation, weatherTool };
+export { type AgentTestConfig, type AgentTestResult, AgentTestRunner, type AssertedMessage, type ConversationResult, ConversationSimulator, type ConversationSimulatorConfig, MockLLM, type MockLLMConfig, type MockToolConfig, type PlanningStep, type PlanningTestState, type ReActTestState, type SnapshotConfig, StateBuilder, type TestToolCall, type TestToolResult, assertAlternatingMessages, assertCompletesWithin, assertHasKeys, assertInRange, assertIsMessage, assertIterationsWithinLimit, assertLastMessageContains, assertMatchesSnapshot, assertMessageContains, assertMessagesMatchSnapshot, assertNotEmpty, assertStateChanged, assertStateHasFields, assertStateSnapshot, assertThrowsWithMessage, assertToolCalled, calculatorTool, compareStates, complexReasoningConversation, createAgentTestRunner, createCalculatorTool, createConversation, createConversationSimulator, createConversationState, createConversationWithSystem, createDelayedTool, createEchoLLM, createEchoTool, createErrorLLM, createErrorTool, createMessageSnapshot, createMockLLM, createMockTool, createPlanningState, createReActState, createSnapshot, createStateBuilder, createStateDiff, databaseQueryTool, errorHandlingConversation, fileReaderTool, getToolByName, getToolsByCategory, longContextConversation, multiTurnConversation, sampleData, sampleTools, searchTool, simpleGreeting, timeTool, toolUsageConversation, weatherTool };

package/dist/index.js

@@ -1,5 +1,5 @@
 import { BaseChatModel } from '@langchain/core/language_models/chat_models';
-import { HumanMessage, AIMessage, SystemMessage } from '@langchain/core/messages';
+import { HumanMessage, AIMessage, SystemMessage, BaseMessage } from '@langchain/core/messages';
 import { z } from 'zod';
 import { toolBuilder, ToolCategory } from '@agentforge/core';
 import { isatty } from 'tty';
@@ -1632,14 +1632,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;
   }
   /**
@@ -1685,7 +1689,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);
@@ -1697,21 +1701,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 ?? {}
   };
 }
 
@@ -18342,13 +18346,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(HumanMessage);
-  } else if (type3 === "ai") {
-    globalExpect(value).toBeInstanceOf(AIMessage);
+  globalExpect(value).not.toBeNull();
+  globalExpect(typeof value).toBe("object");
+  const samePackageMessage = value instanceof 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) {
@@ -18364,7 +18380,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) {