@copilotkit/vue 2.0.0-rc.6 → 2.0.0-rc.8

package/src/v2/providers/CopilotKitProvider.vue

@@ -2,11 +2,13 @@
 import {
   GENERATE_SANDBOXED_UI_DESCRIPTION,
   randomUUID,
+  ɵanySchema,
 } from "@copilotkit/shared";
 import {
   computed,
   h,
   onMounted,
+  onUnmounted,
   provide,
   ref,
   shallowRef,
@@ -14,7 +16,7 @@ import {
   watch,
 } from "vue";
 import type { Component, FunctionalComponent } from "vue";
-import { z } from "zod";
+import type { z } from "zod";
 import type { AbstractAgent } from "@ag-ui/client";
 import { ToolCallStatus } from "@copilotkit/core";
 import type {
@@ -23,7 +25,7 @@ import type {
   CopilotKitHeadersSource,
   FrontendTool,
 } from "@copilotkit/core";
-import { ɵwithHeaderDefaults } from "@copilotkit/core";
+import { ɵresolveLicenseState, ɵwithHeaderDefaults } from "@copilotkit/core";
 import { schemaToJsonSchema, shouldEnableInspector } from "@copilotkit/shared";
 import type {
   RuntimeEntitlementResponse,
@@ -54,11 +56,7 @@ import {
 } from "./keys";
 import type { VueInspectorOpenRequest } from "./keys";
 import { MARKDOWN_RENDERER_KEY } from "./markdown-renderer";
-import {
-  LicenseContextKey,
-  createLicenseContextValue,
-} from "./license-context";
-import type { LicenseContextValue } from "./license-context";
+import { LicenseContextKey } from "./license-context";
 import CopilotKitInspector from "../components/CopilotKitInspector.vue";
 import LicenseWarningBanner from "../components/LicenseWarningBanner.vue";
 import type {
@@ -365,7 +363,7 @@ const processedHumanInTheLoop = computed(() => {
 
       renderToolCalls.push({
         name: tool.name,
-        args: tool.parameters ?? z.any(),
+        args: tool.parameters ?? ɵanySchema,
         render,
         ...(tool.agentId && { agentId: tool.agentId }),
       } as VueToolCallRenderer<unknown>);
@@ -392,7 +390,8 @@ const allRenderToolCalls = computed(() => {
   ];
   for (const tool of [...props.frontendTools, ...builtInFrontendTools.value]) {
     if (tool.render) {
-      const args = tool.parameters ?? (tool.name === "*" ? z.any() : undefined);
+      const args =
+        tool.parameters ?? (tool.name === "*" ? ɵanySchema : undefined);
       if (args) {
         combined.push({
           name: tool.name,
@@ -469,7 +468,7 @@ const builtInActivityRenderers = computed<
   const renderers: VueActivityMessageRenderer<unknown>[] = [
     {
       activityType: MCPAppsActivityType,
-      content: MCPAppsActivityContentSchema as unknown as z.ZodSchema<unknown>,
+      content: MCPAppsActivityContentSchema,
       render: (props.mcpApps?.allowSameOriginFallback
         ? SameOriginMCPAppsActivityRenderer
         : MCPAppsActivityRenderer) as unknown as VueActivityMessageRenderer<unknown>["render"],
@@ -479,7 +478,7 @@ const builtInActivityRenderers = computed<
   if (openGenerativeUIActive.value) {
     renderers.push({
       activityType: OpenGenerativeUIActivityType,
-      content: OpenGenerativeUIContentSchema as unknown as z.ZodSchema<unknown>,
+      content: OpenGenerativeUIContentSchema,
       render:
         OpenGenerativeUIActivityRenderer as unknown as VueActivityMessageRenderer<unknown>["render"],
     });
@@ -894,73 +893,26 @@ provide(CopilotKitKey, {
 provide(SandboxFunctionsKey, sandboxFunctions);
 provide(MARKDOWN_RENDERER_KEY, props.markdownRenderer);
 
-// License context — driven by structured and legacy Runtime authority.
-const retryableRuntimeEntitlementFailure = computed(
-  () =>
-    runtimeEntitlements.value?.status !== "ready" &&
-    runtimeEntitlements.value?.error.retryable === true,
-);
-const hasNonReadyRuntimeEntitlement = computed(
-  () =>
-    runtimeEntitlements.value !== undefined &&
-    runtimeEntitlements.value.status !== "ready",
-);
-const hasLegacyRuntimeEntitlementFallback = computed(
-  () =>
-    runtimeLicenseStatus.value === "valid" ||
-    runtimeLicenseStatus.value === "expiring",
-);
-const runtimeEntitlementRetryInProgress = computed(
-  () =>
-    retryableRuntimeEntitlementFailure.value &&
-    runtimeEntitlementRetryPending.value &&
-    !hasLegacyRuntimeEntitlementFallback.value,
-);
-// Only a terminal failure denies features. A retryable failure (a timeout,
-// a network error, a 5xx) says nothing about what the project may use.
-const terminalRuntimeEntitlementFailure = computed(
-  () =>
-    hasNonReadyRuntimeEntitlement.value &&
-    !retryableRuntimeEntitlementFailure.value &&
-    !hasLegacyRuntimeEntitlementFallback.value,
-);
-const licenseContextValue = computed<LicenseContextValue>(() => {
-  const runtimeLicenseContext = createLicenseContextValue(
-    runtimeEntitlementRetryInProgress.value
-      ? undefined
-      : runtimeLicenseStatus.value,
-    runtimeEntitlements.value,
-  );
-  if (!terminalRuntimeEntitlementFailure.value) {
-    return runtimeLicenseContext;
-  }
-
-  return {
-    ...runtimeLicenseContext,
-    checkFeature: () => false,
-    getLimit: () => null,
-  };
-});
+// License context and banner, from the runtime license status and the
+// structured entitlement response.
+const licenseState = computed(() =>
+  ɵresolveLicenseState({
+    status: runtimeLicenseStatus.value,
+    entitlements: runtimeEntitlements.value,
+    retryPending: runtimeEntitlementRetryPending.value,
+    publicKey: resolvedPublicKey.value,
+  }),
+);
+const licenseContextValue = computed(() => licenseState.value.context);
 provide(LicenseContextKey, licenseContextValue);
+const licenseBannerType = computed(() => licenseState.value.bannerType);
 
-const runtimeLicenseWarningStatus = computed(() =>
-  runtimeEntitlementRetryInProgress.value
-    ? undefined
-    : (licenseContextValue.value.status ?? undefined),
-);
-const showNoLicenseBanner = computed(
-  () =>
-    runtimeLicenseWarningStatus.value === "none" && !resolvedPublicKey.value,
-);
-const showExpiredBanner = computed(
-  () => runtimeLicenseWarningStatus.value === "expired",
-);
-const showInvalidBanner = computed(
-  () => runtimeLicenseWarningStatus.value === "invalid",
-);
-const showExpiringBanner = computed(
-  () => runtimeLicenseWarningStatus.value === "expiring",
-);
+// The provider created this core, so it stops the core's runs, timers and
+// subscriptions. `onUnmounted` runs after the children unmount, so their
+// cleanup still reaches a live core.
+onUnmounted(() => {
+  copilotkit.value.dispose();
+});
 </script>
 
 <template>
@@ -971,8 +923,5 @@ const showExpiringBanner = computed(
     :open-request="inspectorOpenRequest"
   />
   <!-- License warnings — driven by server-reported status -->
-  <LicenseWarningBanner v-if="showNoLicenseBanner" type="no_license" />
-  <LicenseWarningBanner v-if="showExpiredBanner" type="expired" />
-  <LicenseWarningBanner v-if="showInvalidBanner" type="invalid" />
-  <LicenseWarningBanner v-if="showExpiringBanner" type="expiring" />
+  <LicenseWarningBanner v-if="licenseBannerType" :type="licenseBannerType" />
 </template>

package/src/v2/providers/__tests__/CopilotKitProvider.headersAtSendTime.test.ts

@@ -7,12 +7,12 @@ import { mount } from "@vue/test-utils";
 import { defineComponent, h, nextTick } from "vue";
 import { afterEach, describe, expect, it, vi } from "vitest";
 import { HttpAgent } from "@ag-ui/client";
+import { ɵtranscribeAudio } from "@copilotkit/core";
 import type { CopilotKitHeadersSource } from "@copilotkit/core";
 import CopilotKitProvider from "../CopilotKitProvider.vue";
 import { useCopilotKit } from "../useCopilotKit";
 import { useAgent } from "../../hooks/use-agent";
 import { useThreads } from "../../hooks/use-threads";
-import { transcribeAudio } from "../../lib/transcription-client";
 import type { CopilotKitCoreVue } from "../../lib/vue-core";
 
 type Call = { url: string; auth: string | null; publicApiKey: string | null };
@@ -418,14 +418,14 @@ describe("CopilotKitProvider — headers builder evaluated at send time (#1937)"
     // Force an actual resolution while signed in, so the header source's
     // last snapshot is non-empty — otherwise a stale-merge bug has nothing
     // stale to leak, and the assertion below would pass vacuously.
-    await transcribeAudio(getCore(), new Blob(["a"], { type: "audio/webm" }));
+    await ɵtranscribeAudio(getCore(), new Blob(["a"], { type: "audio/webm" }));
     const firstCall = calls.find((c) => c.url.endsWith("/transcribe"));
     expect(firstCall?.auth).toBe("Bearer x");
 
     calls.length = 0;
     signedIn = false;
 
-    await transcribeAudio(getCore(), new Blob(["a"], { type: "audio/webm" }));
+    await ɵtranscribeAudio(getCore(), new Blob(["a"], { type: "audio/webm" }));
     const secondCall = calls.find((c) => c.url.endsWith("/transcribe"));
     expect(secondCall).toBeDefined();
     expect(secondCall!.auth).toBeNull();

package/src/v2/providers/__tests__/CopilotKitProvider.learning.test.ts

@@ -478,7 +478,9 @@ describe("CopilotKitProvider learning prop", () => {
         const app = renderApp(true);
 
         await app.unmount();
-        expect(stop).toHaveBeenCalledTimes(1);
+        // The learning watcher stops capture, then disposing the core stops
+        // it again. Stopping twice is safe; capture must end either way.
+        expect(stop).toHaveBeenCalled();
         expect(History.prototype.pushState).toBe(nativePushState);
       });
 

package/src/v2/providers/__tests__/CopilotKitProvider.test.ts

@@ -810,6 +810,23 @@ describe("CopilotKitProvider", () => {
     });
   });
 
+  describe("Unmount", () => {
+    it("disposes the core it created, so subscribers stop hearing from it", () => {
+      const { getCore, wrapper } = mountWithProvider(() => h("div"));
+      const core = getCore();
+      const onPropertiesChanged = vi.fn();
+      core.subscribe({ onPropertiesChanged });
+
+      core.setProperties({ before: "unmount" });
+      expect(onPropertiesChanged).toHaveBeenCalledTimes(1);
+
+      wrapper.unmount();
+      core.setProperties({ after: "unmount" });
+
+      expect(onPropertiesChanged).toHaveBeenCalledTimes(1);
+    });
+  });
+
   describe("inspector visibility", () => {
     async function settleInspectorLoad(): Promise<void> {
       await vi.dynamicImportSettled();
@@ -886,39 +903,6 @@ describe("CopilotKitProvider", () => {
       view.unmount();
     });
 
-    it("does not let legacy showDevConsole disable the development Inspector", async () => {
-      const view = render(CopilotKitProvider, {
-        props: {
-          runtimeUrl: "/api/copilotkit",
-          showDevConsole: false,
-        },
-        slots: { default: "child" },
-      });
-
-      await waitFor(() => {
-        expect(
-          view.container.querySelector("cpk-web-inspector"),
-        ).not.toBeNull();
-      });
-      view.unmount();
-    });
-
-    it("does not let legacy showDevConsole enable the production Inspector", async () => {
-      vi.stubEnv("NODE_ENV", "production");
-      const view = render(CopilotKitProvider, {
-        props: {
-          runtimeUrl: "/api/copilotkit",
-          showDevConsole: "auto",
-        },
-        slots: { default: "child" },
-      });
-
-      await settleInspectorLoad();
-      expect(view.container.querySelector("cpk-web-inspector")).toBeNull();
-      expect(defineWebInspector).not.toHaveBeenCalled();
-      view.unmount();
-    });
-
     it("does not render or load the inspector during SSR", async () => {
       vi.stubGlobal("window", undefined);
       const html = await renderToString(

package/src/v2/providers/types.ts

@@ -1,53 +1,7 @@
-export const CopilotChatDefaultLabels = {
-  chatInputPlaceholder: "Type a message...",
-  chatInputToolbarStartTranscribeButtonLabel: "Transcribe",
-  chatInputToolbarCancelTranscribeButtonLabel: "Cancel",
-  chatInputToolbarFinishTranscribeButtonLabel: "Finish",
-  chatInputToolbarAddButtonLabel: "Add photos or files",
-  chatInputToolbarToolsButtonLabel: "Tools",
-  chatInputToolbarSendButtonLabel: "Send message",
-  chatInputToolbarStopButtonLabel: "Stop",
-  assistantMessageToolbarCopyCodeLabel: "Copy",
-  assistantMessageToolbarCopyCodeCopiedLabel: "Copied",
-  assistantMessageToolbarCopyMessageLabel: "Copy",
-  assistantMessageToolbarInspectorLabel: "View in Inspector",
-  assistantMessageToolbarInspectorDescription:
-    "Open this message in the Inspector",
-  assistantMessageToolbarInspectorLocalOnlyLabel: "Local only",
-  assistantMessageToolbarInspectorLocalOnlyDescription:
-    "Only visible on localhost or loopback hosts in development. Never shown in production.",
-  assistantMessageToolbarInspectorTitle: "CopilotKit Inspector",
-  assistantMessageToolbarInspectorHideLabel: "Hide this icon",
-  assistantMessageToolbarInspectorHideDescription: "Until you reload this page",
-  assistantMessageToolbarThumbsUpLabel: "Good response",
-  assistantMessageToolbarThumbsDownLabel: "Bad response",
-  assistantMessageFeedbackCommentPlaceholder: "Add a comment (optional)",
-  assistantMessageFeedbackCommentSubmitLabel: "Send comment",
-  assistantMessageToolbarReadAloudLabel: "Read aloud",
-  assistantMessageToolbarRegenerateLabel: "Regenerate",
-  userMessageToolbarCopyMessageLabel: "Copy",
-  userMessageToolbarEditMessageLabel: "Edit",
-  userMessageToolbarPreviousBranchLabel: "Previous branch",
-  userMessageToolbarNextBranchLabel: "Next branch",
-  chatDisclaimerText:
-    "AI can make mistakes. Please verify important information.",
-  chatToggleOpenLabel: "Open chat",
-  chatToggleCloseLabel: "Close chat",
-  chatScrollToBottomLabel: "Scroll to bottom",
-  modalHeaderTitle: "CopilotKit Chat",
-  welcomeMessageText: "How can I help you today?",
-  reasoningMessageThinkingLabel: "Thinking…",
-  chatInputSlashMenuLabel: "Slash commands",
-  chatInputSlashMenuEmptyText: "No commands found",
-  attachmentRemoveButtonLabel: "Remove attachment",
-  attachmentPlayVideoButtonLabel: "Play video",
-  attachmentPreviewCloseButtonLabel: "Close preview",
-  modalHeaderCloseButtonLabel: "Close",
-  modalHeaderThreadsButtonLabel: "Open threads",
-  sidebarLabel: "Copilot chat sidebar",
-} as const;
+import type { CopilotChatLabels } from "@copilotkit/core";
 
-export type CopilotChatLabels = typeof CopilotChatDefaultLabels;
+export { CopilotChatDefaultLabels } from "@copilotkit/core";
+export type { CopilotChatLabels } from "@copilotkit/core";
 
 export interface CopilotChatConfigurationValue {
   labels: CopilotChatLabels;

package/src/v2/types/__tests__/defineToolCallRenderer.test.ts

@@ -2,11 +2,30 @@ import { describe, it, expect } from "vitest";
 import { defineComponent, h } from "vue";
 import { z } from "zod";
 import { ToolCallStatus } from "@copilotkit/core";
+import type { VNode } from "vue";
 import { defineToolCallRenderer } from "../defineToolCallRenderer";
-import type { VueToolCallRenderer } from "../vue-tool-call-renderer";
+import type {
+  VueToolCallRenderer,
+  VueToolCallRendererRenderProps,
+} from "../vue-tool-call-renderer";
+
+/**
+ * Calls a renderer built from a render function. `render` is typed as a
+ * function or a component, and TypeScript cannot tell which one this test
+ * passed, so the narrowing cast lives here once.
+ */
+function callRender<T>(
+  renderer: VueToolCallRenderer<T>,
+  props: VueToolCallRendererRenderProps<T>,
+) {
+  const render = renderer.render as (
+    props: VueToolCallRendererRenderProps<T>,
+  ) => VNode;
+  return render(props);
+}
 
 describe("defineToolCallRenderer", () => {
-  it("defaults wildcard args to z.any and preserves runtime tool name", () => {
+  it("defaults wildcard args to a schema that accepts any value and preserves runtime tool name", () => {
     const wildcardRenderer = defineToolCallRenderer({
       name: "*",
       render: ({ name, args, status }) =>
@@ -18,19 +37,11 @@ describe("defineToolCallRenderer", () => {
     });
 
     expect(wildcardRenderer.name).toBe("*");
-    expect(wildcardRenderer.args.safeParse({ anything: true }).success).toBe(
-      true,
-    );
-
-    const rendered = (
-      wildcardRenderer.render as (props: {
-        name: string;
-        toolCallId: string;
-        args: Record<string, unknown>;
-        status: ToolCallStatus;
-        result: string | undefined;
-      }) => ReturnType<typeof h>
-    )({
+    expect(
+      wildcardRenderer.args["~standard"].validate({ anything: true }),
+    ).toEqual({ value: { anything: true } });
+
+    const rendered = callRender(wildcardRenderer, {
       name: "customTool",
       toolCallId: "tc-wildcard-1",
       args: { x: 1 },
@@ -53,15 +64,7 @@ describe("defineToolCallRenderer", () => {
         ),
     });
 
-    const rendered = (
-      wildcardRenderer.render as (props: {
-        name: string;
-        toolCallId: string;
-        args: Record<string, unknown>;
-        status: ToolCallStatus;
-        result: string | undefined;
-      }) => ReturnType<typeof h>
-    )({
+    const rendered = callRender(wildcardRenderer, {
       name: "customTool",
       toolCallId: "tc-123",
       args: {},
@@ -136,15 +139,7 @@ describe("defineToolCallRenderer", () => {
       },
     });
 
-    const rendered = (
-      typedRenderer.render as (props: {
-        name: string;
-        toolCallId: string;
-        args: { location: string; units?: "celsius" | "fahrenheit" };
-        status: ToolCallStatus;
-        result: string | undefined;
-      }) => ReturnType<typeof h>
-    )({
+    const rendered = callRender(typedRenderer, {
       name: "get_weather",
       toolCallId: "tc-weather-1",
       args: { location: "Paris", units: "celsius" },

package/src/v2/types/defineToolCallRenderer.ts

@@ -1,4 +1,4 @@
-import { z } from "zod";
+import { ɵanySchema } from "@copilotkit/shared";
 import type { StandardSchemaV1, InferSchemaOutput } from "@copilotkit/shared";
 import type { VueToolCallRenderer } from "./vue-tool-call-renderer";
 import type { VueToolCallRendererRenderFn } from "./vue-tool-call-renderer";
@@ -25,11 +25,12 @@ export function defineToolCallRenderer<S extends StandardSchemaV1>(def: {
   agentId?: string;
   group?: boolean;
 }): VueToolCallRenderer<unknown> {
-  const argsSchema = def.name === "*" && !def.args ? z.any() : def.args;
+  // Only a "*" renderer may leave out `args`; it accepts any arguments.
+  const args = def.args ?? ɵanySchema;
   return {
     name: def.name,
-    args: argsSchema as StandardSchemaV1<any, unknown>,
-    render: def.render as VueToolCallRenderer<unknown>["render"],
+    args,
+    render: def.render,
     ...(def.agentId ? { agentId: def.agentId } : {}),
     ...(def.group ? { group: true } : {}),
   };

package/src/v2/types/human-in-the-loop.ts

@@ -1,37 +1,13 @@
-import type { FrontendTool, ToolCallStatus } from "@copilotkit/core";
+import type { FrontendTool, HumanInTheLoopRenderProps } from "@copilotkit/core";
 import type { Component, VNodeChild } from "vue";
 
-export type VueHumanInTheLoopRenderProps<T> =
-  | {
-      name: string;
-      description: string;
-      toolCallId: string;
-      agentId?: string;
-      args: Partial<T>;
-      status: ToolCallStatus.InProgress;
-      result: undefined;
-      respond: undefined;
-    }
-  | {
-      name: string;
-      description: string;
-      toolCallId: string;
-      agentId?: string;
-      args: T;
-      status: ToolCallStatus.Executing;
-      result: undefined;
-      respond: (result: unknown) => Promise<void>;
-    }
-  | {
-      name: string;
-      description: string;
-      toolCallId: string;
-      agentId?: string;
-      args: T;
-      status: ToolCallStatus.Complete;
-      result: string;
-      respond: undefined;
-    };
+/**
+ * Props a Vue human-in-the-loop renderer receives. Same as core's
+ * `HumanInTheLoopRenderProps`.
+ *
+ * @typeParam T - The shape of the tool arguments.
+ */
+export type VueHumanInTheLoopRenderProps<T> = HumanInTheLoopRenderProps<T>;
 
 export type VueHumanInTheLoopRenderFn<T> =
   | ((props: VueHumanInTheLoopRenderProps<T>) => VNodeChild)

package/src/v2/types/interrupt.ts

@@ -1,31 +1,7 @@
-import type { Interrupt, RunAgentResult } from "@ag-ui/client";
-
-export interface InterruptEvent<TValue = unknown> {
-  name: string;
-  value: TValue;
-}
-
-export type InterruptResolveFn = (
-  payload?: unknown,
-  interruptId?: string,
-) => Promise<RunAgentResult | void>;
-export type InterruptCancelFn = (
-  interruptId?: string,
-) => Promise<RunAgentResult | void>;
-
-export interface InterruptHandlerProps<TValue = unknown> {
-  event: InterruptEvent<TValue>;
-  interrupt: Interrupt | null;
-  interrupts: Interrupt[];
-  resolve: InterruptResolveFn;
-  cancel: InterruptCancelFn;
-}
-
-export interface InterruptRenderProps<TValue = unknown, TResult = unknown> {
-  event: InterruptEvent<TValue>;
-  interrupt: Interrupt | null;
-  interrupts: Interrupt[];
-  result: TResult;
-  resolve: InterruptResolveFn;
-  cancel: InterruptCancelFn;
-}
+export type {
+  InterruptEvent,
+  InterruptResolveFn,
+  InterruptCancelFn,
+  InterruptHandlerProps,
+  InterruptRenderProps,
+} from "@copilotkit/core";

package/src/v2/types/vue-activity-message-renderer.ts

@@ -1,22 +1,31 @@
-import type { ActivityMessage } from "@ag-ui/core";
-import type { AbstractAgent } from "@ag-ui/client";
-import type { z } from "zod";
+import type {
+  ActivityMessageRenderProps,
+  ActivityMessageRendererDef,
+} from "@copilotkit/core";
 import type { Component, VNodeChild } from "vue";
 
-export interface VueActivityMessageRendererProps<TActivityContent> {
-  activityType: string;
-  content: TActivityContent;
-  message: ActivityMessage;
-  agent: AbstractAgent | undefined;
-}
+/**
+ * Props a Vue activity message renderer receives. Same as core's
+ * `ActivityMessageRenderProps`.
+ *
+ * @typeParam TActivityContent - The shape of the activity content.
+ */
+export type VueActivityMessageRendererProps<TActivityContent> =
+  ActivityMessageRenderProps<TActivityContent>;
 
 export type VueActivityMessageRendererRenderFn<TActivityContent> =
   | ((props: VueActivityMessageRendererProps<TActivityContent>) => VNodeChild)
   | Component<VueActivityMessageRendererProps<TActivityContent>>;
 
-export interface VueActivityMessageRenderer<TActivityContent> {
-  activityType: string;
-  agentId?: string;
-  content: z.ZodSchema<TActivityContent>;
-  render: VueActivityMessageRendererRenderFn<TActivityContent>;
-}
+/**
+ * A Vue activity message renderer registration. `content` takes any
+ * Standard Schema (zod, valibot, arktype, and so on). The chat validates the
+ * activity content with it and passes the result to `render`.
+ *
+ * @typeParam TActivityContent - The shape of the activity content.
+ */
+export type VueActivityMessageRenderer<TActivityContent> =
+  ActivityMessageRendererDef<
+    TActivityContent,
+    VueActivityMessageRendererRenderFn<TActivityContent>
+  >;

package/src/v2/types/vue-custom-message-renderer.ts

@@ -1,24 +1,20 @@
-import type { Message } from "@ag-ui/core";
+import type {
+  CustomMessageRendererDef,
+  CustomMessageRendererPosition,
+  CustomMessageRenderProps,
+} from "@copilotkit/core";
 import type { Component, VNodeChild } from "vue";
 
-export type VueCustomMessageRendererPosition = "before" | "after";
+/** Where a custom message renderer draws. Same as core's type. */
+export type VueCustomMessageRendererPosition = CustomMessageRendererPosition;
 
-export interface VueCustomMessageRendererProps {
-  message: Message;
-  position: VueCustomMessageRendererPosition;
-  runId: string;
-  messageIndex: number;
-  messageIndexInRun: number;
-  numberOfMessagesInRun: number;
-  agentId: string;
-  stateSnapshot: unknown;
-}
+/** Props a Vue custom message renderer receives. Same as core's type. */
+export type VueCustomMessageRendererProps = CustomMessageRenderProps;
 
 export type VueCustomMessageRendererRenderFn =
   | ((props: VueCustomMessageRendererProps) => VNodeChild)
   | Component<VueCustomMessageRendererProps>;
 
-export interface VueCustomMessageRenderer {
-  agentId?: string;
-  render: VueCustomMessageRendererRenderFn;
-}
+/** A Vue custom message renderer registration. */
+export type VueCustomMessageRenderer =
+  CustomMessageRendererDef<VueCustomMessageRendererRenderFn>;