@copilotkitnext/agent 1.51.4-next.7 → 1.51.4

package/src/__tests__/property-overrides.test.ts

@@ -46,7 +46,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["model"],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",
@@ -71,7 +73,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["model"],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const customModel = {
         modelId: "custom-model",
@@ -100,7 +104,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["model"],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",
@@ -127,7 +133,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["toolChoice"],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",
@@ -152,7 +160,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["toolChoice"],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",
@@ -177,7 +187,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["toolChoice"],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",
@@ -202,7 +214,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["toolChoice"],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",
@@ -211,13 +225,18 @@ describe("Property Overrides - Edge Cases", () => {
         tools: [],
         context: [],
         state: {},
-        forwardedProps: { toolChoice: { type: "tool", toolName: "specificTool" } },
+        forwardedProps: {
+          toolChoice: { type: "tool", toolName: "specificTool" },
+        },
       };
 
       await collectEvents(agent["run"](input));
 
       const callArgs = vi.mocked(streamText).mock.calls[0][0];
-      expect(callArgs.toolChoice).toEqual({ type: "tool", toolName: "specificTool" });
+      expect(callArgs.toolChoice).toEqual({
+        type: "tool",
+        toolName: "specificTool",
+      });
     });
 
     it("should ignore invalid toolChoice values", async () => {
@@ -227,7 +246,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["toolChoice"],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",
@@ -254,7 +275,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["stopSequences"],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",
@@ -279,7 +302,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["stopSequences"],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",
@@ -304,7 +329,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["stopSequences"],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",
@@ -347,7 +374,9 @@ describe("Property Overrides - Edge Cases", () => {
         ],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",
@@ -388,7 +417,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["temperature"],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",
@@ -417,7 +448,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["temperature"], // Only temperature is overridable
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",
@@ -448,7 +481,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["temperature"],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",
@@ -473,7 +508,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["temperature"],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",
@@ -498,7 +535,9 @@ describe("Property Overrides - Edge Cases", () => {
         overridableProperties: ["temperature"],
       });
 
-      vi.mocked(streamText).mockReturnValue(mockStreamTextResponse([finish()]) as any);
+      vi.mocked(streamText).mockReturnValue(
+        mockStreamTextResponse([finish()]) as any,
+      );
 
       const input: RunAgentInput = {
         threadId: "thread1",

package/src/__tests__/state-tools.test.ts

@@ -2,7 +2,14 @@ import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
 import { BasicAgent } from "../index";
 import { EventType, type RunAgentInput } from "@ag-ui/client";
 import { streamText } from "ai";
-import { mockStreamTextResponse, toolCallStreamingStart, toolCall, toolResult, finish, collectEvents } from "./test-helpers";
+import {
+  mockStreamTextResponse,
+  toolCallStreamingStart,
+  toolCall,
+  toolResult,
+  finish,
+  collectEvents,
+} from "./test-helpers";
 
 // Mock the ai module
 vi.mock("ai", () => ({
@@ -43,7 +50,10 @@ describe("State Update Tools", () => {
         mockStreamTextResponse([
           toolCallStreamingStart("call1", "AGUISendStateSnapshot"),
           toolCall("call1", "AGUISendStateSnapshot"),
-          toolResult("call1", "AGUISendStateSnapshot", { success: true, snapshot: newState }),
+          toolResult("call1", "AGUISendStateSnapshot", {
+            success: true,
+            snapshot: newState,
+          }),
           finish(),
         ]) as any,
       );
@@ -60,7 +70,9 @@ describe("State Update Tools", () => {
       const events = await collectEvents(agent["run"](input));
 
       // Find STATE_SNAPSHOT event
-      const snapshotEvent = events.find((e: any) => e.type === EventType.STATE_SNAPSHOT);
+      const snapshotEvent = events.find(
+        (e: any) => e.type === EventType.STATE_SNAPSHOT,
+      );
       expect(snapshotEvent).toBeDefined();
       expect(snapshotEvent).toMatchObject({
         type: EventType.STATE_SNAPSHOT,
@@ -77,7 +89,10 @@ describe("State Update Tools", () => {
         mockStreamTextResponse([
           toolCallStreamingStart("call1", "AGUISendStateSnapshot"),
           toolCall("call1", "AGUISendStateSnapshot"),
-          toolResult("call1", "AGUISendStateSnapshot", { success: true, snapshot: { value: 1 } }),
+          toolResult("call1", "AGUISendStateSnapshot", {
+            success: true,
+            snapshot: { value: 1 },
+          }),
           finish(),
         ]) as any,
       );
@@ -94,8 +109,12 @@ describe("State Update Tools", () => {
       const events = await collectEvents(agent["run"](input));
 
       // Should have both STATE_SNAPSHOT and TOOL_CALL_RESULT
-      const snapshotEvent = events.find((e: any) => e.type === EventType.STATE_SNAPSHOT);
-      const toolResultEvent = events.find((e: any) => e.type === EventType.TOOL_CALL_RESULT);
+      const snapshotEvent = events.find(
+        (e: any) => e.type === EventType.STATE_SNAPSHOT,
+      );
+      const toolResultEvent = events.find(
+        (e: any) => e.type === EventType.TOOL_CALL_RESULT,
+      );
 
       expect(snapshotEvent).toBeDefined();
       expect(toolResultEvent).toBeDefined();
@@ -138,7 +157,9 @@ describe("State Update Tools", () => {
       const events = await collectEvents(agent["run"](input));
 
       // Find STATE_DELTA event
-      const deltaEvent = events.find((e: any) => e.type === EventType.STATE_DELTA);
+      const deltaEvent = events.find(
+        (e: any) => e.type === EventType.STATE_DELTA,
+      );
       expect(deltaEvent).toBeDefined();
       expect(deltaEvent).toMatchObject({
         type: EventType.STATE_DELTA,
@@ -173,7 +194,9 @@ describe("State Update Tools", () => {
 
       const events = await collectEvents(agent["run"](input));
 
-      const deltaEvent = events.find((e: any) => e.type === EventType.STATE_DELTA);
+      const deltaEvent = events.find(
+        (e: any) => e.type === EventType.STATE_DELTA,
+      );
       expect(deltaEvent?.delta).toEqual(delta);
     });
 
@@ -204,7 +227,9 @@ describe("State Update Tools", () => {
 
       const events = await collectEvents(agent["run"](input));
 
-      const deltaEvent = events.find((e: any) => e.type === EventType.STATE_DELTA);
+      const deltaEvent = events.find(
+        (e: any) => e.type === EventType.STATE_DELTA,
+      );
       expect(deltaEvent?.delta).toEqual(delta);
     });
 
@@ -235,7 +260,9 @@ describe("State Update Tools", () => {
 
       const events = await collectEvents(agent["run"](input));
 
-      const deltaEvent = events.find((e: any) => e.type === EventType.STATE_DELTA);
+      const deltaEvent = events.find(
+        (e: any) => e.type === EventType.STATE_DELTA,
+      );
       expect(deltaEvent?.delta).toEqual(delta);
     });
 
@@ -270,7 +297,9 @@ describe("State Update Tools", () => {
 
       const events = await collectEvents(agent["run"](input));
 
-      const deltaEvent = events.find((e: any) => e.type === EventType.STATE_DELTA);
+      const deltaEvent = events.find(
+        (e: any) => e.type === EventType.STATE_DELTA,
+      );
       expect(deltaEvent?.delta).toEqual(delta);
     });
 
@@ -302,8 +331,12 @@ describe("State Update Tools", () => {
       const events = await collectEvents(agent["run"](input));
 
       // Should have both STATE_DELTA and TOOL_CALL_RESULT
-      const deltaEvent = events.find((e: any) => e.type === EventType.STATE_DELTA);
-      const toolResultEvent = events.find((e: any) => e.type === EventType.TOOL_CALL_RESULT);
+      const deltaEvent = events.find(
+        (e: any) => e.type === EventType.STATE_DELTA,
+      );
+      const toolResultEvent = events.find(
+        (e: any) => e.type === EventType.TOOL_CALL_RESULT,
+      );
 
       expect(deltaEvent).toBeDefined();
       expect(toolResultEvent).toBeDefined();
@@ -324,10 +357,16 @@ describe("State Update Tools", () => {
         mockStreamTextResponse([
           toolCallStreamingStart("call1", "AGUISendStateSnapshot"),
           toolCall("call1", "AGUISendStateSnapshot"),
-          toolResult("call1", "AGUISendStateSnapshot", { success: true, snapshot: { value: 1 } }),
+          toolResult("call1", "AGUISendStateSnapshot", {
+            success: true,
+            snapshot: { value: 1 },
+          }),
           toolCallStreamingStart("call2", "AGUISendStateDelta"),
           toolCall("call2", "AGUISendStateDelta"),
-          toolResult("call2", "AGUISendStateDelta", { success: true, delta: [{ op: "replace", path: "/value", value: 2 }] }),
+          toolResult("call2", "AGUISendStateDelta", {
+            success: true,
+            delta: [{ op: "replace", path: "/value", value: 2 }],
+          }),
           finish(),
         ]) as any,
       );
@@ -343,8 +382,12 @@ describe("State Update Tools", () => {
 
       const events = await collectEvents(agent["run"](input));
 
-      const snapshotEvents = events.filter((e: any) => e.type === EventType.STATE_SNAPSHOT);
-      const deltaEvents = events.filter((e: any) => e.type === EventType.STATE_DELTA);
+      const snapshotEvents = events.filter(
+        (e: any) => e.type === EventType.STATE_SNAPSHOT,
+      );
+      const deltaEvents = events.filter(
+        (e: any) => e.type === EventType.STATE_DELTA,
+      );
 
       expect(snapshotEvents).toHaveLength(1);
       expect(deltaEvents).toHaveLength(1);
@@ -382,7 +425,9 @@ describe("State Update Tools", () => {
       const events = await collectEvents(agent["run"](input));
 
       const stateEvents = events.filter(
-        (e: any) => e.type === EventType.STATE_SNAPSHOT || e.type === EventType.STATE_DELTA,
+        (e: any) =>
+          e.type === EventType.STATE_SNAPSHOT ||
+          e.type === EventType.STATE_DELTA,
       );
 
       expect(stateEvents).toHaveLength(0);

package/src/__tests__/test-helpers.ts

@@ -46,7 +46,10 @@ export function textDelta(text: string): MockStreamEvent {
 /**
  * Helper to create a tool call streaming start event
  */
-export function toolCallStreamingStart(toolCallId: string, toolName: string): MockStreamEvent {
+export function toolCallStreamingStart(
+  toolCallId: string,
+  toolName: string,
+): MockStreamEvent {
   return {
     type: "tool-input-start",
     id: toolCallId,
@@ -57,7 +60,10 @@ export function toolCallStreamingStart(toolCallId: string, toolName: string): Mo
 /**
  * Helper to create a tool call delta event
  */
-export function toolCallDelta(toolCallId: string, argsTextDelta: string): MockStreamEvent {
+export function toolCallDelta(
+  toolCallId: string,
+  argsTextDelta: string,
+): MockStreamEvent {
   return {
     type: "tool-input-delta",
     id: toolCallId,
@@ -68,7 +74,11 @@ export function toolCallDelta(toolCallId: string, argsTextDelta: string): MockSt
 /**
  * Helper to create a tool call event
  */
-export function toolCall(toolCallId: string, toolName: string, input: unknown = {}): MockStreamEvent {
+export function toolCall(
+  toolCallId: string,
+  toolName: string,
+  input: unknown = {},
+): MockStreamEvent {
   return {
     type: "tool-call",
     toolCallId,
@@ -80,7 +90,11 @@ export function toolCall(toolCallId: string, toolName: string, input: unknown =
 /**
  * Helper to create a tool result event
  */
-export function toolResult(toolCallId: string, toolName: string, output: any): MockStreamEvent {
+export function toolResult(
+  toolCallId: string,
+  toolName: string,
+  output: any,
+): MockStreamEvent {
   return {
     type: "tool-result",
     toolCallId,
@@ -112,7 +126,9 @@ export function error(errorMessage: string): MockStreamEvent {
 /**
  * Collects all events from an Observable into an array
  */
-export async function collectEvents<T>(observable: { subscribe: (observer: any) => any }): Promise<T[]> {
+export async function collectEvents<T>(observable: {
+  subscribe: (observer: any) => any;
+}): Promise<T[]> {
   return new Promise((resolve, reject) => {
     const events: T[] = [];
     const subscription = observable.subscribe({

package/src/__tests__/utils.test.ts

@@ -155,7 +155,7 @@ describe("convertMessagesToVercelAISDKMessages", () => {
             type: "function",
             function: {
               name: "getTool",
-              arguments: '{}',
+              arguments: "{}",
             },
           },
         ],
@@ -331,7 +331,9 @@ describe("convertToolsToVercelAITools", () => {
       },
     ];
 
-    expect(() => convertToolsToVercelAITools(tools)).toThrow("Invalid JSON schema");
+    expect(() => convertToolsToVercelAITools(tools)).toThrow(
+      "Invalid JSON schema",
+    );
   });
 
   it("should handle multiple tools", () => {