@copilotkit/react-core 1.5.0-tyler-reset-chat.0 → 1.5.0

package/src/components/toast/toast-provider.tsx

@@ -1,7 +1,7 @@
-import { useCopilotContext } from "../../context";
 import { GraphQLError } from "@copilotkit/runtime-client-gql";
 import React, { createContext, useContext, useState, useCallback } from "react";
-import { ExclamationMarkIcon } from "./exclamation-mark-icon";
+import { ErrorToast } from "../error-boundary/error-utils";
+import { PartialBy } from "@copilotkit/shared";
 
 interface Toast {
   id: string;
@@ -12,7 +12,7 @@ interface Toast {
 
 interface ToastContextValue {
   toasts: Toast[];
-  addToast: (toast: Omit<Toast, "id">) => void;
+  addToast: (toast: PartialBy<Toast, "id">) => void;
   addGraphQLErrorsToast: (errors: GraphQLError[]) => void;
   removeToast: (id: string) => void;
   enabled: boolean;
@@ -36,10 +36,13 @@ export function ToastProvider({
   children: React.ReactNode;
 }) {
   const [toasts, setToasts] = useState<Toast[]>([]);
-  const addToast = useCallback((toast: Omit<Toast, "id">) => {
-    const id = Math.random().toString(36).substring(2, 9);
+  const addToast = useCallback((toast: PartialBy<Toast, "id">) => {
+    const id = toast.id ?? Math.random().toString(36).substring(2, 9);
 
-    setToasts((currentToasts) => [...currentToasts, { ...toast, id }]);
+    setToasts((currentToasts) => {
+      if (currentToasts.find((toast) => toast.id === id)) return currentToasts;
+      return [...currentToasts, { ...toast, id }];
+    });
 
     if (toast.duration) {
       setTimeout(() => {
@@ -54,51 +57,9 @@ export function ToastProvider({
     //   return;
     // }
 
-    const errorsToRender = errors.map((error, idx) => {
-      const message = error.message;
-      const code = error.extensions?.code as string;
-
-      return (
-        <div
-          key={idx}
-          style={{
-            marginTop: idx === 0 ? 0 : 10,
-            marginBottom: 14,
-          }}
-        >
-          <ExclamationMarkIcon style={{ marginBottom: 4 }} />
-
-          {code && (
-            <div
-              style={{
-                fontWeight: "600",
-                marginBottom: 4,
-              }}
-            >
-              Copilot Cloud Error:{" "}
-              <span style={{ fontFamily: "monospace", fontWeight: "normal" }}>{code}</span>
-            </div>
-          )}
-          <div>{message}</div>
-        </div>
-      );
-    });
-
     addToast({
       type: "error",
-      message: (
-        <div
-          style={{
-            fontSize: "13px",
-            maxWidth: "600px",
-          }}
-        >
-          {errorsToRender}
-          <div style={{ fontSize: "11px", opacity: 0.75 }}>
-            NOTE: This is a Copilot Cloud error, and it only displays during local development.
-          </div>
-        </div>
-      ),
+      message: <ErrorToast errors={errors} />,
     });
   }, []);
 

package/src/context/copilot-context.tsx

@@ -1,5 +1,9 @@
 import { CopilotCloudConfig, FunctionCallHandler } from "@copilotkit/shared";
-import { ActionRenderProps, FrontendAction } from "../types/frontend-action";
+import {
+  ActionRenderProps,
+  CatchAllActionRenderProps,
+  FrontendAction,
+} from "../types/frontend-action";
 import React from "react";
 import { TreeNodeId } from "../hooks/use-tree";
 import { DocumentPointer } from "../types";
@@ -68,7 +72,9 @@ export interface CopilotApiConfig {
   credentials?: RequestCredentials;
 }
 
-export type InChatRenderFunction = (props: ActionRenderProps<any>) => string | JSX.Element;
+export type InChatRenderFunction = (
+  props: ActionRenderProps<any> | CatchAllActionRenderProps<any>,
+) => string | JSX.Element;
 export type CoagentInChatRenderFunction = (
   props: CoAgentStateRenderProps<any>,
 ) => string | JSX.Element | undefined | null;
@@ -132,6 +138,13 @@ export interface CopilotContextParams {
   // agents
   coagentStates: Record<string, CoagentState>;
   setCoagentStates: React.Dispatch<React.SetStateAction<Record<string, CoagentState>>>;
+  coagentStatesRef: React.RefObject<Record<string, CoagentState>>;
+  setCoagentStatesWithRef: (
+    value:
+      | Record<string, CoagentState>
+      | ((prev: Record<string, CoagentState>) => Record<string, CoagentState>),
+  ) => void;
+
   agentSession: AgentSession | null;
   setAgentSession: React.Dispatch<React.SetStateAction<AgentSession | null>>;
 
@@ -198,6 +211,8 @@ const emptyCopilotContext: CopilotContextParams = {
   showDevConsole: "auto",
   coagentStates: {},
   setCoagentStates: () => {},
+  coagentStatesRef: { current: {} },
+  setCoagentStatesWithRef: () => {},
 
   agentSession: null,
   setAgentSession: () => {},