@copilotkit/react-core 1.57.3 → 1.58.0-canary.thread-id-propagation

package/src/context/threads-context.tsx

@@ -1,69 +0,0 @@
-import React, {
-  createContext,
-  useCallback,
-  useContext,
-  useState,
-  ReactNode,
-  SetStateAction,
-} from "react";
-import { randomUUID } from "@copilotkit/shared";
-
-export interface ThreadsContextValue {
-  threadId: string;
-  setThreadId: (value: SetStateAction<string>) => void;
-  // True when the current threadId was chosen by the caller — either via
-  // the `threadId` prop on <CopilotKit> / <ThreadsProvider>, or via a later
-  // setThreadId() call. False when the provider minted a UUID on first
-  // mount so downstream consumers don't have to treat that placeholder as
-  // a real backend thread.
-  isThreadIdExplicit: boolean;
-}
-
-const ThreadsContext = createContext<ThreadsContextValue | undefined>(
-  undefined,
-);
-
-export interface ThreadsProviderProps {
-  children: ReactNode;
-  threadId?: string;
-}
-
-export function ThreadsProvider({
-  children,
-  threadId: explicitThreadId,
-}: ThreadsProviderProps) {
-  const [internalThreadId, setInternalThreadId] = useState<string>(() =>
-    randomUUID(),
-  );
-  const [internalIsExplicit, setInternalIsExplicit] = useState<boolean>(false);
-
-  const threadId = explicitThreadId ?? internalThreadId;
-  const isThreadIdExplicit = explicitThreadId != null || internalIsExplicit;
-
-  const setThreadId = useCallback((value: SetStateAction<string>) => {
-    setInternalThreadId(value);
-    setInternalIsExplicit(true);
-  }, []);
-
-  return (
-    <ThreadsContext.Provider
-      value={{
-        threadId,
-        setThreadId,
-        isThreadIdExplicit,
-      }}
-    >
-      {children}
-    </ThreadsContext.Provider>
-  );
-}
-
-export function useThreads() {
-  const context = useContext(ThreadsContext);
-  if (!context) {
-    throw new Error("useThreads must be used within ThreadsProvider");
-  }
-  return context;
-}
-
-export { ThreadsContext };

package/src/hooks/__tests__/use-coagent-config.test.ts

@@ -1,352 +0,0 @@
-import { vi } from "vitest";
-import { renderHook, waitFor } from "@testing-library/react";
-import { useCoAgent } from "../use-coagent";
-import type { AgentSubscriber } from "@ag-ui/client";
-
-// Mock functions for v2
-const mockSetState = vi.fn();
-const mockRunAgent = vi.fn();
-const mockAbortRun = vi.fn();
-const mockSubscribe = vi.fn();
-const mockSetProperties = vi.fn();
-
-// Store the last subscriber for triggering events
-let lastSubscriber: AgentSubscriber | null = null;
-
-const mockAgent = {
-  agentId: "test-agent",
-  state: { count: 0 },
-  isRunning: false,
-  threadId: "thread-123",
-  setState: mockSetState,
-  runAgent: mockRunAgent,
-  abortRun: mockAbortRun,
-  subscribe: mockSubscribe.mockImplementation((subscriber: AgentSubscriber) => {
-    lastSubscriber = subscriber;
-    return {
-      unsubscribe: vi.fn(),
-    };
-  }),
-};
-
-vi.mock("../../v2", () => ({
-  useAgent: vi.fn(() => ({ agent: mockAgent })),
-  useCopilotKit: vi.fn(() => ({
-    copilotkit: {
-      setProperties: mockSetProperties,
-    },
-  })),
-}));
-
-// Mock other dependencies
-const mockAppendMessage = vi.fn();
-const mockRunChatCompletion = vi.fn();
-
-vi.mock("../use-copilot-chat_internal", () => ({
-  useCopilotChat: () => ({
-    appendMessage: mockAppendMessage,
-    runChatCompletion: mockRunChatCompletion,
-  }),
-}));
-
-vi.mock("../use-copilot-runtime-client", () => ({
-  useCopilotRuntimeClient: () => ({
-    loadAgentState: vi.fn().mockResolvedValue({
-      data: { loadAgentState: { state: "{}", threadExists: false } },
-      error: null,
-    }),
-  }),
-}));
-
-vi.mock("../../context", () => ({
-  useCopilotContext: () => ({
-    availableAgents: [],
-    coagentStates: {},
-    coagentStatesRef: { current: {} },
-    threadId: "test-thread",
-    copilotApiConfig: {
-      headers: {},
-      chatApiEndpoint: "test-endpoint",
-      publicApiKey: "test-key",
-    },
-    showDevConsole: false,
-  }),
-  useCopilotMessagesContext: () => ({
-    messages: [],
-  }),
-}));
-
-vi.mock("../../components/toast/toast-provider", () => ({
-  useToast: () => ({
-    setBannerError: vi.fn(),
-  }),
-}));
-
-vi.mock("../../components/error-boundary/error-utils", () => ({
-  useAsyncCallback: (fn: any) => fn,
-}));
-
-vi.mock("../../components/copilot-provider/copilot-messages", () => ({
-  useMessagesTap: () => ({
-    getMessagesFromTap: vi.fn(() => []),
-    updateTapMessages: vi.fn(),
-  }),
-}));
-
-describe("useCoAgent config synchronization", () => {
-  beforeEach(() => {
-    vi.clearAllMocks();
-    lastSubscriber = null;
-  });
-
-  it("should call setProperties when config changes", async () => {
-    const { rerender } = renderHook(
-      ({ config }) =>
-        useCoAgent({
-          name: "test-agent",
-          initialState: { count: 0 },
-          config,
-        }),
-      {
-        initialProps: { config: { configurable: { model: "gpt-4" } } },
-      },
-    );
-
-    // Clear the initial calls
-    mockSetProperties.mockClear();
-
-    // Change config
-    rerender({ config: { configurable: { model: "gpt-4o" } } });
-
-    // Wait for effect to complete and verify setProperties was called
-    await waitFor(() => {
-      expect(mockSetProperties).toHaveBeenCalledWith({
-        configurable: { model: "gpt-4o" },
-      });
-    });
-  });
-
-  it("should not call setProperties when config is unchanged", () => {
-    const config = { configurable: { model: "gpt-4" } };
-
-    const { rerender } = renderHook(
-      ({ config }) =>
-        useCoAgent({
-          name: "test-agent",
-          initialState: { count: 0 },
-          config,
-        }),
-      {
-        initialProps: { config },
-      },
-    );
-
-    // Clear the initial calls
-    mockSetProperties.mockClear();
-
-    // Re-render with same config reference
-    rerender({ config });
-
-    // Should not have called setProperties
-    expect(mockSetProperties).not.toHaveBeenCalled();
-  });
-
-  it("should handle backward compatibility with configurable prop", async () => {
-    const { rerender } = renderHook(
-      ({ configurable }) =>
-        useCoAgent({
-          name: "test-agent",
-          initialState: { count: 0 },
-          configurable,
-        }),
-      {
-        initialProps: { configurable: { model: "gpt-4" } },
-      },
-    );
-
-    // Clear the initial calls
-    mockSetProperties.mockClear();
-
-    // Change configurable prop
-    rerender({ configurable: { model: "gpt-4o" } });
-
-    // Wait for effect to complete and verify setProperties was called
-    await waitFor(() => {
-      expect(mockSetProperties).toHaveBeenCalledWith({
-        configurable: { model: "gpt-4o" },
-      });
-    });
-  });
-
-  it("should not call setProperties when both config and configurable are undefined", () => {
-    renderHook(() =>
-      useCoAgent({
-        name: "test-agent",
-        initialState: { count: 0 },
-      }),
-    );
-
-    // Should not have called setProperties
-    expect(mockSetProperties).not.toHaveBeenCalled();
-  });
-
-  it("should handle deeply nested config changes", async () => {
-    const { rerender } = renderHook(
-      ({ config }) =>
-        useCoAgent({
-          name: "test-agent",
-          initialState: { count: 0 },
-          config,
-        }),
-      {
-        initialProps: {
-          config: {
-            configurable: {
-              model: "gpt-4",
-              settings: { temperature: 0.5 },
-            },
-          },
-        },
-      },
-    );
-
-    // Clear the initial calls
-    mockSetProperties.mockClear();
-
-    // Change nested config
-    rerender({
-      config: {
-        configurable: {
-          model: "gpt-4",
-          settings: { temperature: 0.7 }, // Only nested property changed
-        },
-      },
-    });
-
-    // Wait for effect to complete and verify setProperties was called
-    await waitFor(() => {
-      expect(mockSetProperties).toHaveBeenCalledWith({
-        configurable: {
-          model: "gpt-4",
-          settings: { temperature: 0.7 },
-        },
-      });
-    });
-  });
-
-  describe("State Management", () => {
-    // Helper to create mock subscriber params
-    const createMockParams = (stateOverride: any = {}) => ({
-      messages: [],
-      state: stateOverride,
-      agent: mockAgent as any,
-      input: {} as any,
-    });
-
-    it("should initialize agent state via onRunInitialized event", () => {
-      // Set agent to have NO state (empty object)
-      mockAgent.state = {} as any;
-
-      renderHook(() =>
-        useCoAgent({
-          name: "test-agent",
-          initialState: { count: 42 },
-        }),
-      );
-
-      // Verify subscription was created
-      expect(mockSubscribe).toHaveBeenCalled();
-      expect(lastSubscriber).toBeTruthy();
-
-      // Clear any initial state calls
-      mockSetState.mockClear();
-
-      // Trigger onRunInitialized with no run state
-      lastSubscriber?.onRunInitialized?.(createMockParams({}));
-
-      // Should set state to initialState
-      expect(mockSetState).toHaveBeenCalledWith({ count: 42 });
-
-      // Reset agent state
-      mockAgent.state = { count: 0 };
-    });
-
-    it("should preserve existing agent state on onRunInitialized", () => {
-      // Set agent to have existing state
-      mockAgent.state = { count: 100 };
-
-      renderHook(() =>
-        useCoAgent({
-          name: "test-agent",
-          initialState: { count: 42 },
-        }),
-      );
-
-      // Clear any initial state calls
-      mockSetState.mockClear();
-
-      // Trigger onRunInitialized with no new state
-      lastSubscriber?.onRunInitialized?.(createMockParams({}));
-
-      // Should NOT override existing state
-      expect(mockSetState).not.toHaveBeenCalled();
-
-      // Reset agent state
-      mockAgent.state = { count: 0 };
-    });
-
-    it("should prioritize run state over initialState", () => {
-      renderHook(() =>
-        useCoAgent({
-          name: "test-agent",
-          initialState: { count: 42 },
-        }),
-      );
-
-      // Clear any initial state calls
-      mockSetState.mockClear();
-
-      // Trigger onRunInitialized with state from the run
-      lastSubscriber?.onRunInitialized?.(createMockParams({ count: 999 }));
-
-      // Should use run state, not initialState
-      expect(mockSetState).toHaveBeenCalledWith({ count: 999 });
-    });
-
-    it("should handle setState with object updates", () => {
-      const { result } = renderHook(() =>
-        useCoAgent({
-          name: "test-agent",
-          initialState: { count: 0 },
-        }),
-      );
-
-      // Update state with object
-      result.current.setState({ count: 5 });
-
-      // Should merge with existing state
-      expect(mockSetState).toHaveBeenCalledWith({ count: 5 });
-    });
-
-    it("should handle setState with function updaters", () => {
-      // Set current agent state
-      mockAgent.state = { count: 10 };
-
-      const { result } = renderHook(() =>
-        useCoAgent({
-          name: "test-agent",
-          initialState: { count: 0 },
-        }),
-      );
-
-      // Update state with function
-      result.current.setState((prev) => ({ count: (prev?.count || 0) + 5 }));
-
-      // Should call function with current state and set result
-      expect(mockSetState).toHaveBeenCalledWith({ count: 15 });
-
-      // Reset agent state
-      mockAgent.state = { count: 0 };
-    });
-  });
-});

package/src/hooks/__tests__/use-coagent-state-render-bridge.helpers.test.ts

@@ -1,107 +0,0 @@
-import {
-  ClaimAction,
-  resolveClaim,
-  selectSnapshot,
-  type ClaimsByMessageId,
-  type SnapshotCaches,
-  type StateRenderContext,
-} from "../use-coagent-state-render-bridge.helpers";
-
-describe("useCoAgentStateRenderBridge helpers", () => {
-  describe("resolveClaim", () => {
-    it("updates runId for an existing claim", () => {
-      const claims: ClaimsByMessageId = {
-        msg1: { stateRenderId: "render-a", runId: "pending" },
-      };
-      const context: StateRenderContext = {
-        agentId: "agent-1",
-        stateRenderId: "render-a",
-        messageId: "msg1",
-        runId: "run-123",
-      };
-
-      const result = resolveClaim({
-        claims,
-        context,
-        stateSnapshot: { step: 1 },
-      });
-      expect(result.action).toBe(ClaimAction.Existing);
-      expect(result.canRender).toBe(true);
-      expect(result.updateRunId).toBe("run-123");
-    });
-
-    it("overrides when a newer message index arrives", () => {
-      const claims: ClaimsByMessageId = {
-        msg1: {
-          stateRenderId: "render-a",
-          runId: "run-1",
-          messageIndex: 0,
-          stateSnapshot: { step: 2 },
-        },
-      };
-      const context: StateRenderContext = {
-        agentId: "agent-1",
-        stateRenderId: "render-a",
-        messageId: "msg2",
-        runId: "run-1",
-        messageIndex: 1,
-      };
-
-      const result = resolveClaim({
-        claims,
-        context,
-        stateSnapshot: { step: 2 },
-      });
-      expect(result.action).toBe(ClaimAction.Override);
-      expect(result.canRender).toBe(true);
-      expect(result.nextClaim).toMatchObject({
-        stateRenderId: "render-a",
-        runId: "run-1",
-        messageIndex: 1,
-      });
-      expect(result.lockOthers).toBe(true);
-    });
-  });
-
-  describe("selectSnapshot", () => {
-    const caches: SnapshotCaches = {
-      byStateRenderAndRun: {},
-      byMessageId: {},
-    };
-
-    it("prefers the explicit state snapshot prop", () => {
-      const result = selectSnapshot({
-        messageId: "msg1",
-        stateRenderId: "render-a",
-        effectiveRunId: "run-1",
-        stateSnapshotProp: '{"phase":"planning"}',
-        agentState: { phase: "ignored" },
-        agentMessages: [{ id: "assistant-1", role: "assistant" }],
-        caches,
-      });
-
-      expect(result.snapshot).toEqual({ phase: "planning" });
-      expect(result.hasSnapshotKeys).toBe(true);
-    });
-
-    it("falls back to cached snapshots for non-latest assistant messages", () => {
-      const result = selectSnapshot({
-        messageId: "msg-old",
-        stateRenderId: "render-a",
-        effectiveRunId: "run-1",
-        agentState: { phase: "live" },
-        agentMessages: [{ id: "assistant-latest", role: "assistant" }],
-        caches: {
-          byStateRenderAndRun: {
-            "render-a::run-1": { phase: "cached" },
-          },
-          byMessageId: {},
-        },
-      });
-
-      expect(result.snapshot).toBeUndefined();
-      expect(result.cachedSnapshot).toEqual({ phase: "cached" });
-      expect(result.snapshotForClaim).toEqual({ phase: "cached" });
-    });
-  });
-});