@copilotkit/react-core 1.1.1-feat-runtime-remote-actions.1 → 1.1.1-feat-runtime-remote-actions.3

package/CHANGELOG.md

@@ -1,5 +1,23 @@
 # ui
 
+## 1.1.1-feat-runtime-remote-actions.3
+
+### Patch Changes
+
+- stream intermediate state
+- Updated dependencies
+  - @copilotkit/runtime-client-gql@1.1.1-feat-runtime-remote-actions.3
+  - @copilotkit/shared@1.1.1-feat-runtime-remote-actions.3
+
+## 1.1.1-feat-runtime-remote-actions.2
+
+### Patch Changes
+
+- useAgentState
+- Updated dependencies
+  - @copilotkit/runtime-client-gql@1.1.1-feat-runtime-remote-actions.2
+  - @copilotkit/shared@1.1.1-feat-runtime-remote-actions.2
+
 ## 1.1.1-feat-runtime-remote-actions.1
 
 ### Patch Changes

package/dist/{chunk-E653KLSE.mjs → chunk-2MQX7BJL.mjs}

@@ -1,6 +1,6 @@
 import {
   CopilotContext
-} from "./chunk-ZOXWEKMF.mjs";
+} from "./chunk-DB4VWZ5Q.mjs";
 
 // src/hooks/use-copilot-action.ts
 import { useRef, useContext, useEffect } from "react";
@@ -13,7 +13,7 @@ function useCopilotAction(action, dependencies) {
       actions[idRef.current].handler = action.handler;
       if (typeof action.render === "function") {
         if (chatComponentsCache.current !== null) {
-          chatComponentsCache.current[action.name] = action.render;
+          chatComponentsCache.current.actions[action.name] = action.render;
         }
       }
     }
@@ -21,7 +21,7 @@ function useCopilotAction(action, dependencies) {
   useEffect(() => {
     setAction(idRef.current, action);
     if (chatComponentsCache.current !== null && action.render !== void 0) {
-      chatComponentsCache.current[action.name] = action.render;
+      chatComponentsCache.current.actions[action.name] = action.render;
     }
     return () => {
       removeAction(idRef.current);
@@ -44,4 +44,4 @@ function useCopilotAction(action, dependencies) {
 export {
   useCopilotAction
 };
-//# sourceMappingURL=chunk-E653KLSE.mjs.map
+//# sourceMappingURL=chunk-2MQX7BJL.mjs.map

package/dist/chunk-2MQX7BJL.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/hooks/use-copilot-action.ts"],"sourcesContent":["import { useRef, useContext, useEffect } from \"react\";\nimport { FrontendAction } from \"../types/frontend-action\";\nimport { CopilotContext } from \"../context/copilot-context\";\nimport { Parameter, randomId } from \"@copilotkit/shared\";\n\n// We implement useCopilotAction dependency handling so that\n// the developer has the option to not provide any dependencies.\n// In this case, we assume they want to update the handler on each rerender.\n// To avoid getting stuck in an infinite loop, we update the handler directly,\n// skipping React state updates.\n// This is ok in this case, because the handler is not part of any UI that\n// needs to be updated.\n// useCallback, useMemo or other memoization techniques are not suitable here,\n// because they will cause a infinite rerender loop.\nexport function useCopilotAction<const T extends Parameter[] | [] = []>(\n  action: FrontendAction<T>,\n  dependencies?: any[],\n): void {\n  const { setAction, removeAction, actions, chatComponentsCache } = useContext(CopilotContext);\n  const idRef = useRef<string>(randomId());\n\n  // If the developer doesn't provide dependencies, we assume they want to\n  // update handler and render function when the action object changes.\n  // This ensures that any captured variables in the handler are up to date.\n  if (dependencies === undefined) {\n    if (actions[idRef.current]) {\n      actions[idRef.current].handler = action.handler as any;\n      if (typeof action.render === \"function\") {\n        if (chatComponentsCache.current !== null) {\n          chatComponentsCache.current.actions[action.name] = action.render;\n        }\n      }\n    }\n  }\n\n  useEffect(() => {\n    setAction(idRef.current, action as any);\n    if (chatComponentsCache.current !== null && action.render !== undefined) {\n      chatComponentsCache.current.actions[action.name] = action.render;\n    }\n    return () => {\n      // NOTE: For now, we don't remove the chatComponentsCache entry when the action is removed.\n      // This is because we currently don't have access to the messages array in CopilotContext.\n      removeAction(idRef.current);\n    };\n  }, [\n    setAction,\n    removeAction,\n    action.description,\n    action.name,\n    // This should be faster than deep equality checking\n    // In addition, all major JS engines guarantee the order of object keys\n    JSON.stringify(action.parameters),\n    // include render only if it's a string\n    typeof action.render === \"string\" ? action.render : undefined,\n    // dependencies set by the developer\n    ...(dependencies || []),\n  ]);\n}\n\n// Usage Example:\n// useCopilotAction({\n//   name: \"myAction\",\n//   parameters: [\n//     { name: \"arg1\", type: \"string\", enum: [\"option1\", \"option2\", \"option3\"], required: false },\n//     { name: \"arg2\", type: \"number\" },\n//     {\n//       name: \"arg3\",\n//       type: \"object\",\n//       attributes: [\n//         { name: \"nestedArg1\", type: \"boolean\" },\n//         { name: \"xyz\", required: false },\n//       ],\n//     },\n//     { name: \"arg4\", type: \"number[]\" },\n//   ],\n//   handler: ({ arg1, arg2, arg3, arg4 }) => {\n//     const x = arg3.nestedArg1;\n//     const z = arg3.xyz;\n//     console.log(arg1, arg2, arg3);\n//   },\n// });\n\n// useCopilotAction({\n//   name: \"myAction\",\n//   handler: () => {\n//     console.log(\"No parameters provided.\");\n//   },\n// });\n"],"mappings":";;;;;AAAA,SAAS,QAAQ,YAAY,iBAAiB;AAG9C,SAAoB,gBAAgB;AAW7B,SAAS,iBACd,QACA,cACM;AACN,QAAM,EAAE,WAAW,cAAc,SAAS,oBAAoB,IAAI,WAAW,cAAc;AAC3F,QAAM,QAAQ,OAAe,SAAS,CAAC;AAKvC,MAAI,iBAAiB,QAAW;AAC9B,QAAI,QAAQ,MAAM,OAAO,GAAG;AAC1B,cAAQ,MAAM,OAAO,EAAE,UAAU,OAAO;AACxC,UAAI,OAAO,OAAO,WAAW,YAAY;AACvC,YAAI,oBAAoB,YAAY,MAAM;AACxC,8BAAoB,QAAQ,QAAQ,OAAO,IAAI,IAAI,OAAO;AAAA,QAC5D;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,YAAU,MAAM;AACd,cAAU,MAAM,SAAS,MAAa;AACtC,QAAI,oBAAoB,YAAY,QAAQ,OAAO,WAAW,QAAW;AACvE,0BAAoB,QAAQ,QAAQ,OAAO,IAAI,IAAI,OAAO;AAAA,IAC5D;AACA,WAAO,MAAM;AAGX,mBAAa,MAAM,OAAO;AAAA,IAC5B;AAAA,EACF,GAAG;AAAA,IACD;AAAA,IACA;AAAA,IACA,OAAO;AAAA,IACP,OAAO;AAAA;AAAA;AAAA,IAGP,KAAK,UAAU,OAAO,UAAU;AAAA;AAAA,IAEhC,OAAO,OAAO,WAAW,WAAW,OAAO,SAAS;AAAA;AAAA,IAEpD,GAAI,gBAAgB,CAAC;AAAA,EACvB,CAAC;AACH;","names":[]}

package/dist/{chunk-FNUGRFGB.mjs → chunk-33ERMMNX.mjs}

@@ -1,6 +1,6 @@
 import {
   CopilotContext
-} from "./chunk-ZOXWEKMF.mjs";
+} from "./chunk-DB4VWZ5Q.mjs";
 
 // src/hooks/use-copilot-readable.ts
 import { useContext, useEffect, useRef } from "react";
@@ -25,4 +25,4 @@ function useCopilotReadable({ description, value, parentId, categories, convert
 export {
   useCopilotReadable
 };
-//# sourceMappingURL=chunk-FNUGRFGB.mjs.map
+//# sourceMappingURL=chunk-33ERMMNX.mjs.map

package/dist/{chunk-YSQQRY2B.mjs → chunk-3K7JAFGM.mjs}

@@ -1,6 +1,6 @@
 import {
   CopilotContext
-} from "./chunk-ZOXWEKMF.mjs";
+} from "./chunk-DB4VWZ5Q.mjs";
 
 // src/hooks/use-make-copilot-document-readable.ts
 import { useContext, useEffect, useRef } from "react";
@@ -20,4 +20,4 @@ function useMakeCopilotDocumentReadable(document, categories, dependencies = [])
 export {
   useMakeCopilotDocumentReadable
 };
-//# sourceMappingURL=chunk-YSQQRY2B.mjs.map
+//# sourceMappingURL=chunk-3K7JAFGM.mjs.map

package/dist/{chunk-AEMRMXIF.mjs → chunk-52M7642J.mjs}

@@ -34,7 +34,7 @@ function useChat(options) {
     isLoading,
     actions,
     onFunctionCall,
-    setAgentState
+    setAgentStates
   } = options;
   const abortControllerRef = useRef();
   const threadIdRef = useRef(null);
@@ -130,14 +130,20 @@ function useChat(options) {
             newMessages.push(message);
             if (message instanceof AgentStateMessage) {
               if (message.running) {
-                setAgentState(message);
+                setAgentStates((prevAgentStates) => __spreadProps(__spreadValues({}, prevAgentStates), {
+                  [message.agentName]: message
+                }));
                 agentSessionRef.current = {
                   threadId: message.threadId,
                   agentName: message.agentName,
                   nodeName: message.nodeName
                 };
               } else {
-                setAgentState(null);
+                setAgentStates((prevAgentStates) => {
+                  const newAgentStates = __spreadValues({}, prevAgentStates);
+                  delete newAgentStates[message.agentName];
+                  return newAgentStates;
+                });
                 agentSessionRef.current = null;
               }
             }
@@ -164,7 +170,22 @@ function useChat(options) {
           }
         }
         if (newMessages.length > 0) {
-          setMessages([...previousMessages, ...newMessages]);
+          const filteredMessages = [...previousMessages, ...newMessages].reduce(
+            (acc, message) => {
+              if (message instanceof AgentStateMessage && // Check if the current message is an AgentStateMessage
+              acc.length > 0 && // Ensure there is at least one message in the accumulator
+              acc[acc.length - 1] instanceof AgentStateMessage && // Check if the last message in the accumulator is also an AgentStateMessage
+              acc[acc.length - 1].agentName === message.agentName && // Check if the agentName is the same
+              acc[acc.length - 1].nodeName === message.nodeName) {
+                acc[acc.length - 1] = message;
+              } else {
+                acc.push(message);
+              }
+              return acc;
+            },
+            []
+          );
+          setMessages(filteredMessages);
         }
       }
       if (
@@ -218,4 +239,4 @@ function useChat(options) {
 export {
   useChat
 };
-//# sourceMappingURL=chunk-AEMRMXIF.mjs.map
+//# sourceMappingURL=chunk-52M7642J.mjs.map

package/dist/chunk-52M7642J.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/hooks/use-chat.ts"],"sourcesContent":["import { useRef } from \"react\";\nimport {\n  FunctionCallHandler,\n  COPILOT_CLOUD_PUBLIC_API_KEY_HEADER,\n  Action,\n  actionParametersToJsonSchema,\n} from \"@copilotkit/shared\";\nimport {\n  Message,\n  TextMessage,\n  ActionExecutionMessage,\n  ResultMessage,\n  CopilotRuntimeClient,\n  convertMessagesToGqlInput,\n  convertGqlOutputToMessages,\n  MessageStatusCode,\n  MessageRole,\n  Role,\n  CopilotRequestType,\n  AgentStateMessage,\n} from \"@copilotkit/runtime-client-gql\";\n\nimport { CopilotApiConfig } from \"../context\";\n\nexport type UseChatOptions = {\n  /**\n   * System messages of the chat. Defaults to an empty array.\n   */\n  initialMessages?: Message[];\n  /**\n   * Callback function to be called when a function call is received.\n   * If the function returns a `ChatRequest` object, the request will be sent\n   * automatically to the API and will be used to update the chat.\n   */\n  onFunctionCall?: FunctionCallHandler;\n  /**\n   * Function definitions to be sent to the API.\n   */\n  actions: Action[];\n\n  /**\n   * The CopilotKit API configuration.\n   */\n  copilotConfig: CopilotApiConfig;\n\n  /**\n   * The current list of messages in the chat.\n   */\n  messages: Message[];\n  /**\n   * The setState-powered method to update the chat messages.\n   */\n  setMessages: React.Dispatch<React.SetStateAction<Message[]>>;\n\n  /**\n   * A callback to get the latest system message.\n   */\n  makeSystemMessageCallback: () => TextMessage;\n\n  /**\n   * Whether the API request is in progress\n   */\n  isLoading: boolean;\n\n  /**\n   * setState-powered method to update the isChatLoading value\n   */\n  setIsLoading: React.Dispatch<React.SetStateAction<boolean>>;\n\n  /**\n   * setState-powered method to update the agent states\n   */\n  setAgentStates: React.Dispatch<React.SetStateAction<Record<string, AgentStateMessage | null>>>;\n};\n\nexport type UseChatHelpers = {\n  /**\n   * Append a user message to the chat list. This triggers the API call to fetch\n   * the assistant's response.\n   * @param message The message to append\n   */\n  append: (message: Message) => Promise<void>;\n  /**\n   * Reload the last AI chat response for the given chat history. If the last\n   * message isn't from the assistant, it will request the API to generate a\n   * new response.\n   */\n  reload: () => Promise<void>;\n  /**\n   * Abort the current request immediately, keep the generated tokens if any.\n   */\n  stop: () => void;\n};\n\ninterface AgentSession {\n  threadId: string;\n  agentName: string;\n  nodeName: string;\n}\n\nexport function useChat(options: UseChatOptions): UseChatHelpers {\n  const {\n    messages,\n    setMessages,\n    makeSystemMessageCallback,\n    copilotConfig,\n    setIsLoading,\n    initialMessages,\n    isLoading,\n    actions,\n    onFunctionCall,\n    setAgentStates,\n  } = options;\n\n  const abortControllerRef = useRef<AbortController>();\n  const threadIdRef = useRef<string | null>(null);\n  const runIdRef = useRef<string | null>(null);\n  const agentSessionRef = useRef<AgentSession | null>(null);\n\n  const publicApiKey = copilotConfig.publicApiKey;\n\n  const headers = {\n    ...(copilotConfig.headers || {}),\n    ...(publicApiKey ? { [COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {}),\n  };\n\n  const runtimeClient = new CopilotRuntimeClient({\n    url: copilotConfig.chatApiEndpoint,\n    publicApiKey: copilotConfig.publicApiKey,\n    headers,\n    credentials: copilotConfig.credentials,\n  });\n\n  const runChatCompletion = async (previousMessages: Message[]): Promise<Message[]> => {\n    setIsLoading(true);\n\n    // this message is just a placeholder. It will disappear once the first real message\n    // is received\n    let newMessages: Message[] = [\n      new TextMessage({\n        content: \"\",\n        role: Role.Assistant,\n      }),\n    ];\n    const abortController = new AbortController();\n    abortControllerRef.current = abortController;\n\n    setMessages([...previousMessages, ...newMessages]);\n\n    const systemMessage = makeSystemMessageCallback();\n\n    const messagesWithContext = [systemMessage, ...(initialMessages || []), ...previousMessages];\n\n    const stream = CopilotRuntimeClient.asStream(\n      runtimeClient.generateCopilotResponse({\n        data: {\n          frontend: {\n            actions: actions.map((action) => ({\n              name: action.name,\n              description: action.description || \"\",\n              jsonSchema: JSON.stringify(actionParametersToJsonSchema(action.parameters || [])),\n            })),\n          },\n          threadId: threadIdRef.current,\n          runId: runIdRef.current,\n          messages: convertMessagesToGqlInput(messagesWithContext),\n          ...(copilotConfig.cloud\n            ? {\n                cloud: {\n                  ...(copilotConfig.cloud.guardrails?.input?.restrictToTopic?.enabled\n                    ? {\n                        guardrails: {\n                          inputValidationRules: {\n                            allowList:\n                              copilotConfig.cloud.guardrails.input.restrictToTopic.validTopics,\n                            denyList:\n                              copilotConfig.cloud.guardrails.input.restrictToTopic.invalidTopics,\n                          },\n                        },\n                      }\n                    : {}),\n                },\n              }\n            : {}),\n          metadata: {\n            requestType: CopilotRequestType.Chat,\n          },\n          ...(agentSessionRef.current\n            ? {\n                agentSession: {\n                  threadId: agentSessionRef.current.threadId,\n                  agentName: agentSessionRef.current.agentName,\n                  nodeName: agentSessionRef.current.nodeName,\n                },\n              }\n            : {}),\n        },\n        properties: copilotConfig.properties,\n        signal: abortControllerRef.current?.signal,\n      }),\n    );\n\n    const guardrailsEnabled =\n      copilotConfig.cloud?.guardrails?.input?.restrictToTopic.enabled || false;\n\n    const reader = stream.getReader();\n\n    let results: { [id: string]: string } = {};\n\n    try {\n      while (true) {\n        const { done, value } = await reader.read();\n\n        if (done) {\n          break;\n        }\n\n        if (!value?.generateCopilotResponse) {\n          continue;\n        }\n\n        threadIdRef.current = value.generateCopilotResponse.threadId || null;\n        runIdRef.current = value.generateCopilotResponse.runId || null;\n\n        const messages = convertGqlOutputToMessages(value.generateCopilotResponse.messages);\n\n        if (messages.length === 0) {\n          continue;\n        }\n\n        newMessages = [];\n\n        // request failed, display error message\n        if (\n          value.generateCopilotResponse.status?.__typename === \"FailedResponseStatus\" &&\n          value.generateCopilotResponse.status.reason === \"GUARDRAILS_VALIDATION_FAILED\"\n        ) {\n          newMessages = [\n            new TextMessage({\n              role: MessageRole.Assistant,\n              content: value.generateCopilotResponse.status.details?.guardrailsReason || \"\",\n            }),\n          ];\n        }\n\n        // add messages to the chat\n        else {\n          for (const message of messages) {\n            newMessages.push(message);\n\n            if (message instanceof AgentStateMessage) {\n              if (message.running) {\n                setAgentStates((prevAgentStates) => ({\n                  ...prevAgentStates,\n                  [message.agentName]: message,\n                }));\n                agentSessionRef.current = {\n                  threadId: message.threadId,\n                  agentName: message.agentName,\n                  nodeName: message.nodeName,\n                };\n              } else {\n                setAgentStates((prevAgentStates) => {\n                  const newAgentStates = { ...prevAgentStates };\n                  delete newAgentStates[message.agentName];\n                  return newAgentStates;\n                });\n                agentSessionRef.current = null;\n              }\n            }\n\n            // execute regular action executions\n            if (\n              message instanceof ActionExecutionMessage &&\n              message.status.code !== MessageStatusCode.Pending &&\n              message.scope === \"client\" &&\n              onFunctionCall\n            ) {\n              if (!(message.id in results)) {\n                // Do not execute a function call if guardrails are enabled but the status is not known\n                if (guardrailsEnabled && value.generateCopilotResponse.status === undefined) {\n                  break;\n                }\n                // execute action\n                const result = await onFunctionCall({\n                  messages: previousMessages,\n                  name: message.name,\n                  args: message.arguments,\n                });\n                results[message.id] = result;\n              }\n\n              // add the result message\n              newMessages.push(\n                new ResultMessage({\n                  result: ResultMessage.encodeResult(results[message.id]),\n                  actionExecutionId: message.id,\n                  actionName: message.name,\n                }),\n              );\n            }\n          }\n        }\n\n        if (newMessages.length > 0) {\n          // Construct filteredMessages inline to remove adjacent AgentStateMessage instances\n          // with the same agentName, keeping only the last one.\n          const filteredMessages = [...previousMessages, ...newMessages].reduce(\n            (acc: Message[], message: Message) => {\n              if (\n                message instanceof AgentStateMessage && // Check if the current message is an AgentStateMessage\n                acc.length > 0 && // Ensure there is at least one message in the accumulator\n                acc[acc.length - 1] instanceof AgentStateMessage && // Check if the last message in the accumulator is also an AgentStateMessage\n                (acc[acc.length - 1] as AgentStateMessage).agentName === message.agentName && // Check if the agentName is the same\n                (acc[acc.length - 1] as AgentStateMessage).nodeName === message.nodeName // Check if the nodeName is the same\n              ) {\n                // If the conditions are met, replace the last message in the accumulator with the current message\n                acc[acc.length - 1] = message;\n              } else {\n                // Otherwise, add the current message to the accumulator\n                acc.push(message);\n              }\n              return acc; // Return the accumulator for the next iteration\n            },\n            [],\n          );\n\n          // Update the state with the filtered messages\n          setMessages(filteredMessages);\n        }\n      }\n\n      if (\n        // if we have client side results\n        Object.values(results).length ||\n        // or the last message we received is a result\n        (newMessages.length && newMessages[newMessages.length - 1] instanceof ResultMessage)\n      ) {\n        // run the completion again and return the result\n\n        // wait for next tick to make sure all the react state updates\n        // - tried using react-dom's flushSync, but it did not work\n        await new Promise((resolve) => setTimeout(resolve, 10));\n\n        return await runChatCompletion([...previousMessages, ...newMessages]);\n      } else {\n        return newMessages.slice();\n      }\n    } finally {\n      setIsLoading(false);\n    }\n  };\n\n  const runChatCompletionAndHandleFunctionCall = async (messages: Message[]): Promise<void> => {\n    await runChatCompletion(messages);\n  };\n\n  const append = async (message: Message): Promise<void> => {\n    if (isLoading) {\n      return;\n    }\n\n    const newMessages = [...messages, message];\n    setMessages(newMessages);\n    return runChatCompletionAndHandleFunctionCall(newMessages);\n  };\n\n  const reload = async (): Promise<void> => {\n    if (isLoading || messages.length === 0) {\n      return;\n    }\n    let newMessages = [...messages];\n    const lastMessage = messages[messages.length - 1];\n\n    if (lastMessage instanceof TextMessage && lastMessage.role === \"assistant\") {\n      newMessages = newMessages.slice(0, -1);\n    }\n\n    setMessages(newMessages);\n\n    return runChatCompletionAndHandleFunctionCall(newMessages);\n  };\n\n  const stop = (): void => {\n    abortControllerRef.current?.abort();\n  };\n\n  return {\n    append,\n    reload,\n    stop,\n  };\n}\n"],"mappings":";;;;;;;AAAA,SAAS,cAAc;AACvB;AAAA,EAEE;AAAA,EAEA;AAAA,OACK;AACP;AAAA,EAEE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAgFA,SAAS,QAAQ,SAAyC;AAC/D,QAAM;AAAA,IACJ;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,IAAI;AAEJ,QAAM,qBAAqB,OAAwB;AACnD,QAAM,cAAc,OAAsB,IAAI;AAC9C,QAAM,WAAW,OAAsB,IAAI;AAC3C,QAAM,kBAAkB,OAA4B,IAAI;AAExD,QAAM,eAAe,cAAc;AAEnC,QAAM,UAAU,kCACV,cAAc,WAAW,CAAC,IAC1B,eAAe,EAAE,CAAC,mCAAmC,GAAG,aAAa,IAAI,CAAC;AAGhF,QAAM,gBAAgB,IAAI,qBAAqB;AAAA,IAC7C,KAAK,cAAc;AAAA,IACnB,cAAc,cAAc;AAAA,IAC5B;AAAA,IACA,aAAa,cAAc;AAAA,EAC7B,CAAC;AAED,QAAM,oBAAoB,CAAO,qBAAoD;AArIvF;AAsII,iBAAa,IAAI;AAIjB,QAAI,cAAyB;AAAA,MAC3B,IAAI,YAAY;AAAA,QACd,SAAS;AAAA,QACT,MAAM,KAAK;AAAA,MACb,CAAC;AAAA,IACH;AACA,UAAM,kBAAkB,IAAI,gBAAgB;AAC5C,uBAAmB,UAAU;AAE7B,gBAAY,CAAC,GAAG,kBAAkB,GAAG,WAAW,CAAC;AAEjD,UAAM,gBAAgB,0BAA0B;AAEhD,UAAM,sBAAsB,CAAC,eAAe,GAAI,mBAAmB,CAAC,GAAI,GAAG,gBAAgB;AAE3F,UAAM,SAAS,qBAAqB;AAAA,MAClC,cAAc,wBAAwB;AAAA,QACpC,MAAM;AAAA,UACJ,UAAU;AAAA,YACR,SAAS,QAAQ,IAAI,CAAC,YAAY;AAAA,cAChC,MAAM,OAAO;AAAA,cACb,aAAa,OAAO,eAAe;AAAA,cACnC,YAAY,KAAK,UAAU,6BAA6B,OAAO,cAAc,CAAC,CAAC,CAAC;AAAA,YAClF,EAAE;AAAA,UACJ;AAAA,UACA,UAAU,YAAY;AAAA,UACtB,OAAO,SAAS;AAAA,UAChB,UAAU,0BAA0B,mBAAmB;AAAA,WACnD,cAAc,QACd;AAAA,UACE,OAAO,qBACD,+BAAc,MAAM,eAApB,mBAAgC,UAAhC,mBAAuC,oBAAvC,mBAAwD,WACxD;AAAA,YACE,YAAY;AAAA,cACV,sBAAsB;AAAA,gBACpB,WACE,cAAc,MAAM,WAAW,MAAM,gBAAgB;AAAA,gBACvD,UACE,cAAc,MAAM,WAAW,MAAM,gBAAgB;AAAA,cACzD;AAAA,YACF;AAAA,UACF,IACA,CAAC;AAAA,QAET,IACA,CAAC,IA5BD;AAAA,UA6BJ,UAAU;AAAA,YACR,aAAa,mBAAmB;AAAA,UAClC;AAAA,YACI,gBAAgB,UAChB;AAAA,UACE,cAAc;AAAA,YACZ,UAAU,gBAAgB,QAAQ;AAAA,YAClC,WAAW,gBAAgB,QAAQ;AAAA,YACnC,UAAU,gBAAgB,QAAQ;AAAA,UACpC;AAAA,QACF,IACA,CAAC;AAAA,QAEP,YAAY,cAAc;AAAA,QAC1B,SAAQ,wBAAmB,YAAnB,mBAA4B;AAAA,MACtC,CAAC;AAAA,IACH;AAEA,UAAM,sBACJ,+BAAc,UAAd,mBAAqB,eAArB,mBAAiC,UAAjC,mBAAwC,gBAAgB,YAAW;AAErE,UAAM,SAAS,OAAO,UAAU;AAEhC,QAAI,UAAoC,CAAC;AAEzC,QAAI;AACF,aAAO,MAAM;AACX,cAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAE1C,YAAI,MAAM;AACR;AAAA,QACF;AAEA,YAAI,EAAC,+BAAO,0BAAyB;AACnC;AAAA,QACF;AAEA,oBAAY,UAAU,MAAM,wBAAwB,YAAY;AAChE,iBAAS,UAAU,MAAM,wBAAwB,SAAS;AAE1D,cAAMA,YAAW,2BAA2B,MAAM,wBAAwB,QAAQ;AAElF,YAAIA,UAAS,WAAW,GAAG;AACzB;AAAA,QACF;AAEA,sBAAc,CAAC;AAGf,cACE,WAAM,wBAAwB,WAA9B,mBAAsC,gBAAe,0BACrD,MAAM,wBAAwB,OAAO,WAAW,gCAChD;AACA,wBAAc;AAAA,YACZ,IAAI,YAAY;AAAA,cACd,MAAM,YAAY;AAAA,cAClB,WAAS,WAAM,wBAAwB,OAAO,YAArC,mBAA8C,qBAAoB;AAAA,YAC7E,CAAC;AAAA,UACH;AAAA,QACF,OAGK;AACH,qBAAW,WAAWA,WAAU;AAC9B,wBAAY,KAAK,OAAO;AAExB,gBAAI,mBAAmB,mBAAmB;AACxC,kBAAI,QAAQ,SAAS;AACnB,+BAAe,CAAC,oBAAqB,iCAChC,kBADgC;AAAA,kBAEnC,CAAC,QAAQ,SAAS,GAAG;AAAA,gBACvB,EAAE;AACF,gCAAgB,UAAU;AAAA,kBACxB,UAAU,QAAQ;AAAA,kBAClB,WAAW,QAAQ;AAAA,kBACnB,UAAU,QAAQ;AAAA,gBACpB;AAAA,cACF,OAAO;AACL,+BAAe,CAAC,oBAAoB;AAClC,wBAAM,iBAAiB,mBAAK;AAC5B,yBAAO,eAAe,QAAQ,SAAS;AACvC,yBAAO;AAAA,gBACT,CAAC;AACD,gCAAgB,UAAU;AAAA,cAC5B;AAAA,YACF;AAGA,gBACE,mBAAmB,0BACnB,QAAQ,OAAO,SAAS,kBAAkB,WAC1C,QAAQ,UAAU,YAClB,gBACA;AACA,kBAAI,EAAE,QAAQ,MAAM,UAAU;AAE5B,oBAAI,qBAAqB,MAAM,wBAAwB,WAAW,QAAW;AAC3E;AAAA,gBACF;AAEA,sBAAM,SAAS,MAAM,eAAe;AAAA,kBAClC,UAAU;AAAA,kBACV,MAAM,QAAQ;AAAA,kBACd,MAAM,QAAQ;AAAA,gBAChB,CAAC;AACD,wBAAQ,QAAQ,EAAE,IAAI;AAAA,cACxB;AAGA,0BAAY;AAAA,gBACV,IAAI,cAAc;AAAA,kBAChB,QAAQ,cAAc,aAAa,QAAQ,QAAQ,EAAE,CAAC;AAAA,kBACtD,mBAAmB,QAAQ;AAAA,kBAC3B,YAAY,QAAQ;AAAA,gBACtB,CAAC;AAAA,cACH;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAEA,YAAI,YAAY,SAAS,GAAG;AAG1B,gBAAM,mBAAmB,CAAC,GAAG,kBAAkB,GAAG,WAAW,EAAE;AAAA,YAC7D,CAAC,KAAgB,YAAqB;AACpC,kBACE,mBAAmB;AAAA,cACnB,IAAI,SAAS;AAAA,cACb,IAAI,IAAI,SAAS,CAAC,aAAa;AAAA,cAC9B,IAAI,IAAI,SAAS,CAAC,EAAwB,cAAc,QAAQ;AAAA,cAChE,IAAI,IAAI,SAAS,CAAC,EAAwB,aAAa,QAAQ,UAChE;AAEA,oBAAI,IAAI,SAAS,CAAC,IAAI;AAAA,cACxB,OAAO;AAEL,oBAAI,KAAK,OAAO;AAAA,cAClB;AACA,qBAAO;AAAA,YACT;AAAA,YACA,CAAC;AAAA,UACH;AAGA,sBAAY,gBAAgB;AAAA,QAC9B;AAAA,MACF;AAEA;AAAA;AAAA,QAEE,OAAO,OAAO,OAAO,EAAE;AAAA,QAEtB,YAAY,UAAU,YAAY,YAAY,SAAS,CAAC,aAAa;AAAA,QACtE;AAKA,cAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,EAAE,CAAC;AAEtD,eAAO,MAAM,kBAAkB,CAAC,GAAG,kBAAkB,GAAG,WAAW,CAAC;AAAA,MACtE,OAAO;AACL,eAAO,YAAY,MAAM;AAAA,MAC3B;AAAA,IACF,UAAE;AACA,mBAAa,KAAK;AAAA,IACpB;AAAA,EACF;AAEA,QAAM,yCAAyC,CAAOA,cAAuC;AAC3F,UAAM,kBAAkBA,SAAQ;AAAA,EAClC;AAEA,QAAM,SAAS,CAAO,YAAoC;AACxD,QAAI,WAAW;AACb;AAAA,IACF;AAEA,UAAM,cAAc,CAAC,GAAG,UAAU,OAAO;AACzC,gBAAY,WAAW;AACvB,WAAO,uCAAuC,WAAW;AAAA,EAC3D;AAEA,QAAM,SAAS,MAA2B;AACxC,QAAI,aAAa,SAAS,WAAW,GAAG;AACtC;AAAA,IACF;AACA,QAAI,cAAc,CAAC,GAAG,QAAQ;AAC9B,UAAM,cAAc,SAAS,SAAS,SAAS,CAAC;AAEhD,QAAI,uBAAuB,eAAe,YAAY,SAAS,aAAa;AAC1E,oBAAc,YAAY,MAAM,GAAG,EAAE;AAAA,IACvC;AAEA,gBAAY,WAAW;AAEvB,WAAO,uCAAuC,WAAW;AAAA,EAC3D;AAEA,QAAM,OAAO,MAAY;AA/X3B;AAgYI,6BAAmB,YAAnB,mBAA4B;AAAA,EAC9B;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;","names":["messages"]}

package/dist/chunk-6YOQY4WD.mjs

@@ -0,0 +1,20 @@
+import {
+  useCopilotContext
+} from "./chunk-DB4VWZ5Q.mjs";
+
+// src/hooks/use-copilot-chat-ui.ts
+import { useEffect } from "react";
+function useCopilotChatUI(chatUI, dependencies) {
+  const { chatUI: chatUIFromContext, setChatUI } = useCopilotContext();
+  useEffect(() => {
+    setChatUI([...chatUIFromContext, chatUI]);
+    return () => {
+      setChatUI(chatUIFromContext.filter((ui) => ui !== chatUI));
+    };
+  }, [chatUI.agentName, chatUI.nodeName, setChatUI]);
+}
+
+export {
+  useCopilotChatUI
+};
+//# sourceMappingURL=chunk-6YOQY4WD.mjs.map

package/dist/chunk-6YOQY4WD.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/hooks/use-copilot-chat-ui.ts"],"sourcesContent":["import { randomId } from \"@copilotkit/shared\";\nimport { useCopilotContext } from \"../context\";\nimport { useEffect, useRef } from \"react\";\n\nexport interface UseCopilotChatUIRenderProps<S = any> {\n  state: S;\n  nodeName: string;\n  agentName: string;\n}\n\nexport interface CopilotChatUI<S = any> {\n  agentName: string;\n  nodeName?: string;\n  render: string | ((props: UseCopilotChatUIRenderProps<S>) => string | React.ReactElement);\n}\n\nexport function useCopilotChatUI<S = any>(chatUI: CopilotChatUI<S>, dependencies?: any[]) {\n  const { chatUI: chatUIFromContext, setChatUI } = useCopilotContext();\n\n  useEffect(() => {\n    setChatUI([...chatUIFromContext, chatUI]);\n    return () => {\n      setChatUI(chatUIFromContext.filter((ui) => ui !== chatUI));\n    };\n  }, [chatUI.agentName, chatUI.nodeName, setChatUI]);\n}\n"],"mappings":";;;;;AAEA,SAAS,iBAAyB;AAc3B,SAAS,iBAA0B,QAA0B,cAAsB;AACxF,QAAM,EAAE,QAAQ,mBAAmB,UAAU,IAAI,kBAAkB;AAEnE,YAAU,MAAM;AACd,cAAU,CAAC,GAAG,mBAAmB,MAAM,CAAC;AACxC,WAAO,MAAM;AACX,gBAAU,kBAAkB,OAAO,CAAC,OAAO,OAAO,MAAM,CAAC;AAAA,IAC3D;AAAA,EACF,GAAG,CAAC,OAAO,WAAW,OAAO,UAAU,SAAS,CAAC;AACnD;","names":[]}

package/dist/{chunk-4U2G3YLF.mjs → chunk-73KBO5JG.mjs}

@@ -1,6 +1,6 @@
 import {
   defaultCopilotContextCategories
-} from "./chunk-AWRZLWWD.mjs";
+} from "./chunk-IF6P5ZXR.mjs";
 import {
   __async
 } from "./chunk-SKC7AJIV.mjs";
@@ -115,4 +115,4 @@ ${instructions}
 export {
   CopilotTask
 };
-//# sourceMappingURL=chunk-4U2G3YLF.mjs.map
+//# sourceMappingURL=chunk-73KBO5JG.mjs.map

package/dist/chunk-AIWDXM7L.mjs

@@ -0,0 +1 @@
+//# sourceMappingURL=chunk-AIWDXM7L.mjs.map

package/dist/{chunk-ZOXWEKMF.mjs → chunk-DB4VWZ5Q.mjs}

@@ -10,7 +10,7 @@ var emptyCopilotContext = {
   },
   removeAction: () => {
   },
-  chatComponentsCache: { current: {} },
+  chatComponentsCache: { current: { actions: {}, chatUI: {} } },
   getContextString: (documents, categories) => returnAndThrowInDebug(""),
   addContext: () => "",
   removeContext: () => {
@@ -44,8 +44,11 @@ var emptyCopilotContext = {
   removeChatSuggestionConfiguration: () => {
   },
   showDevConsole: "auto",
-  agentState: null,
-  setAgentState: () => {
+  agentStates: {},
+  setAgentStates: () => {
+  },
+  chatUI: [],
+  setChatUI: () => {
   }
 };
 var CopilotContext = React.createContext(emptyCopilotContext);
@@ -61,4 +64,4 @@ export {
   CopilotContext,
   useCopilotContext
 };
-//# sourceMappingURL=chunk-ZOXWEKMF.mjs.map
+//# sourceMappingURL=chunk-DB4VWZ5Q.mjs.map

package/dist/chunk-DB4VWZ5Q.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/context/copilot-context.tsx"],"sourcesContent":["import { CopilotCloudConfig, FunctionCallHandler } from \"@copilotkit/shared\";\nimport { AgentStateMessage, Message } from \"@copilotkit/runtime-client-gql\";\nimport { ActionRenderProps, FrontendAction } from \"../types/frontend-action\";\nimport React from \"react\";\nimport { TreeNodeId } from \"../hooks/use-tree\";\nimport { DocumentPointer } from \"../types\";\nimport { CopilotChatSuggestionConfiguration } from \"../types/chat-suggestion-configuration\";\nimport { CopilotChatUI } from \"../hooks/use-copilot-chat-ui\";\n\n/**\n * Interface for the configuration of the Copilot API.\n */\nexport interface CopilotApiConfig {\n  /**\n   * The public API key for Copilot Cloud.\n   */\n  publicApiKey?: string;\n\n  /**\n   * The configuration for Copilot Cloud.\n   */\n  cloud?: CopilotCloudConfig;\n\n  /**\n   * The endpoint for the chat API.\n   */\n  chatApiEndpoint: string;\n\n  /**\n   * The endpoint for the Copilot transcribe audio service.\n   */\n  transcribeAudioUrl?: string;\n\n  /**\n   * The endpoint for the Copilot text to speech service.\n   */\n  textToSpeechUrl?: string;\n\n  /**\n   * additional headers to be sent with the request\n   * @default {}\n   * @example\n   * ```\n   * {\n   *   'Authorization': 'Bearer your_token_here'\n   * }\n   * ```\n   */\n  headers: Record<string, string>;\n\n  /**\n   * Custom properties to be sent with the request\n   * @default {}\n   * @example\n   * ```\n   * {\n   *   'user_id': 'user_id'\n   * }\n   * ```\n   */\n  properties?: Record<string, any>;\n\n  /**\n   * Indicates whether the user agent should send or receive cookies from the other domain\n   * in the case of cross-origin requests.\n   */\n  credentials?: RequestCredentials;\n}\n\nexport type InChatRenderFunction = (props: ActionRenderProps<any>) => string | JSX.Element;\n\nexport interface ChatComponentsCache {\n  actions: Record<string, InChatRenderFunction | string>;\n  chatUI: Record<string, Function | string>;\n}\n\nexport interface CopilotContextParams {\n  // function-calling\n  actions: Record<string, FrontendAction<any>>;\n  setAction: (id: string, action: FrontendAction<any>) => void;\n  removeAction: (id: string) => void;\n  chatComponentsCache: React.RefObject<ChatComponentsCache>;\n\n  getFunctionCallHandler: (\n    customEntryPoints?: Record<string, FrontendAction<any>>,\n  ) => FunctionCallHandler;\n\n  // text context\n  addContext: (context: string, parentId?: string, categories?: string[]) => TreeNodeId;\n  removeContext: (id: TreeNodeId) => void;\n  getContextString: (documents: DocumentPointer[], categories: string[]) => string;\n\n  // document context\n  addDocumentContext: (documentPointer: DocumentPointer, categories?: string[]) => TreeNodeId;\n  removeDocumentContext: (documentId: string) => void;\n  getDocumentsContext: (categories: string[]) => DocumentPointer[];\n\n  // chat\n  messages: Message[];\n  setMessages: React.Dispatch<React.SetStateAction<Message[]>>;\n\n  isLoading: boolean;\n  setIsLoading: React.Dispatch<React.SetStateAction<boolean>>;\n\n  chatSuggestionConfiguration: { [key: string]: CopilotChatSuggestionConfiguration };\n  addChatSuggestionConfiguration: (\n    id: string,\n    suggestion: CopilotChatSuggestionConfiguration,\n  ) => void;\n  removeChatSuggestionConfiguration: (id: string) => void;\n\n  chatInstructions: string;\n  setChatInstructions: React.Dispatch<React.SetStateAction<string>>;\n\n  // api endpoints\n  copilotApiConfig: CopilotApiConfig;\n\n  showDevConsole: boolean | \"auto\";\n\n  // agent states\n  agentStates: Record<string, AgentStateMessage | null>;\n  setAgentStates: React.Dispatch<React.SetStateAction<Record<string, AgentStateMessage | null>>>;\n\n  // chat ui\n  chatUI: CopilotChatUI[];\n  setChatUI: React.Dispatch<React.SetStateAction<CopilotChatUI[]>>;\n}\n\nconst emptyCopilotContext: CopilotContextParams = {\n  actions: {},\n  setAction: () => {},\n  removeAction: () => {},\n\n  chatComponentsCache: { current: { actions: {}, chatUI: {} } },\n  getContextString: (documents: DocumentPointer[], categories: string[]) =>\n    returnAndThrowInDebug(\"\"),\n  addContext: () => \"\",\n  removeContext: () => {},\n\n  getFunctionCallHandler: () => returnAndThrowInDebug(async () => {}),\n\n  messages: [],\n  setMessages: () => returnAndThrowInDebug([]),\n\n  isLoading: false,\n  setIsLoading: () => returnAndThrowInDebug(false),\n\n  chatInstructions: \"\",\n  setChatInstructions: () => returnAndThrowInDebug(\"\"),\n\n  getDocumentsContext: (categories: string[]) => returnAndThrowInDebug([]),\n  addDocumentContext: () => returnAndThrowInDebug(\"\"),\n  removeDocumentContext: () => {},\n\n  copilotApiConfig: new (class implements CopilotApiConfig {\n    get chatApiEndpoint(): string {\n      throw new Error(\"Remember to wrap your app in a `<CopilotKit> {...} </CopilotKit>` !!!\");\n    }\n\n    get headers(): Record<string, string> {\n      return {};\n    }\n    get body(): Record<string, any> {\n      return {};\n    }\n  })(),\n\n  chatSuggestionConfiguration: {},\n  addChatSuggestionConfiguration: () => {},\n  removeChatSuggestionConfiguration: () => {},\n  showDevConsole: \"auto\",\n  agentStates: {},\n  setAgentStates: () => {},\n  chatUI: [],\n  setChatUI: () => {},\n};\n\nexport const CopilotContext = React.createContext<CopilotContextParams>(emptyCopilotContext);\n\nexport function useCopilotContext(): CopilotContextParams {\n  return React.useContext(CopilotContext);\n}\n\nfunction returnAndThrowInDebug<T>(value: T): T {\n  throw new Error(\"Remember to wrap your app in a `<CopilotKit> {...} </CopilotKit>` !!!\");\n  return value;\n}\n"],"mappings":";;;;;AAGA,OAAO,WAAW;AA6HlB,IAAM,sBAA4C;AAAA,EAChD,SAAS,CAAC;AAAA,EACV,WAAW,MAAM;AAAA,EAAC;AAAA,EAClB,cAAc,MAAM;AAAA,EAAC;AAAA,EAErB,qBAAqB,EAAE,SAAS,EAAE,SAAS,CAAC,GAAG,QAAQ,CAAC,EAAE,EAAE;AAAA,EAC5D,kBAAkB,CAAC,WAA8B,eAC/C,sBAAsB,EAAE;AAAA,EAC1B,YAAY,MAAM;AAAA,EAClB,eAAe,MAAM;AAAA,EAAC;AAAA,EAEtB,wBAAwB,MAAM,sBAAsB,MAAY;AAAA,EAAC,EAAC;AAAA,EAElE,UAAU,CAAC;AAAA,EACX,aAAa,MAAM,sBAAsB,CAAC,CAAC;AAAA,EAE3C,WAAW;AAAA,EACX,cAAc,MAAM,sBAAsB,KAAK;AAAA,EAE/C,kBAAkB;AAAA,EAClB,qBAAqB,MAAM,sBAAsB,EAAE;AAAA,EAEnD,qBAAqB,CAAC,eAAyB,sBAAsB,CAAC,CAAC;AAAA,EACvE,oBAAoB,MAAM,sBAAsB,EAAE;AAAA,EAClD,uBAAuB,MAAM;AAAA,EAAC;AAAA,EAE9B,kBAAkB,IAAK,MAAkC;AAAA,IACvD,IAAI,kBAA0B;AAC5B,YAAM,IAAI,MAAM,uEAAuE;AAAA,IACzF;AAAA,IAEA,IAAI,UAAkC;AACpC,aAAO,CAAC;AAAA,IACV;AAAA,IACA,IAAI,OAA4B;AAC9B,aAAO,CAAC;AAAA,IACV;AAAA,EACF,EAAG;AAAA,EAEH,6BAA6B,CAAC;AAAA,EAC9B,gCAAgC,MAAM;AAAA,EAAC;AAAA,EACvC,mCAAmC,MAAM;AAAA,EAAC;AAAA,EAC1C,gBAAgB;AAAA,EAChB,aAAa,CAAC;AAAA,EACd,gBAAgB,MAAM;AAAA,EAAC;AAAA,EACvB,QAAQ,CAAC;AAAA,EACT,WAAW,MAAM;AAAA,EAAC;AACpB;AAEO,IAAM,iBAAiB,MAAM,cAAoC,mBAAmB;AAEpF,SAAS,oBAA0C;AACxD,SAAO,MAAM,WAAW,cAAc;AACxC;AAEA,SAAS,sBAAyB,OAAa;AAC7C,QAAM,IAAI,MAAM,uEAAuE;AACvF,SAAO;AACT;","names":[]}

package/dist/{chunk-PSBBWX7X.mjs → chunk-H4XENJME.mjs}

@@ -1,6 +1,6 @@
 import {
   defaultCopilotContextCategories
-} from "./chunk-AWRZLWWD.mjs";
+} from "./chunk-IF6P5ZXR.mjs";
 import {
   __async,
   __spreadValues
@@ -150,4 +150,4 @@ Any additional messages provided are for providing context only and should not b
 export {
   extract
 };
-//# sourceMappingURL=chunk-PSBBWX7X.mjs.map
+//# sourceMappingURL=chunk-H4XENJME.mjs.map

package/dist/{chunk-AWRZLWWD.mjs → chunk-IF6P5ZXR.mjs}

@@ -6,7 +6,7 @@ import {
 } from "./chunk-6U3UH3KO.mjs";
 import {
   CopilotContext
-} from "./chunk-ZOXWEKMF.mjs";
+} from "./chunk-DB4VWZ5Q.mjs";
 import {
   __async,
   __objRest,
@@ -32,7 +32,7 @@ function CopilotKit(_a) {
   }
   const chatApiEndpoint = props.runtimeUrl || COPILOT_CLOUD_CHAT_URL;
   const [actions, setActions] = useState({});
-  const chatComponentsCache = useRef({});
+  const chatComponentsCache = useRef({ actions: {}, chatUI: {} });
   const { addElement, removeElement, printTree } = use_tree_default();
   const [messages, setMessages] = useState([]);
   const [isLoading, setIsLoading] = useState(false);
@@ -146,7 +146,8 @@ ${nonDocumentStrings}`;
       return rest;
     });
   };
-  const [agentState, setAgentState] = useState(null);
+  const [agentStates, setAgentStates] = useState({});
+  const [chatUI, setChatUI] = useState([]);
   return /* @__PURE__ */ jsx(
     CopilotContext.Provider,
     {
@@ -173,8 +174,10 @@ ${nonDocumentStrings}`;
         chatInstructions,
         setChatInstructions,
         showDevConsole: props.showDevConsole === void 0 ? "auto" : props.showDevConsole,
-        agentState,
-        setAgentState
+        agentStates,
+        setAgentStates,
+        chatUI,
+        setChatUI
       },
       children
     }
@@ -210,4 +213,4 @@ export {
   CopilotKit,
   defaultCopilotContextCategories
 };
-//# sourceMappingURL=chunk-AWRZLWWD.mjs.map
+//# sourceMappingURL=chunk-IF6P5ZXR.mjs.map

package/dist/chunk-IF6P5ZXR.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/components/copilot-provider/copilotkit.tsx"],"sourcesContent":["/**\n * This component will typically wrap your entire application (or a sub-tree of your application where you want to have a copilot). It provides the copilot context to all other components and hooks.\n *\n * ## Example\n *\n * <Tabs items={[\"Copilot Cloud (Recommended)\", \"Self-hosted Runtime\"]}>\n *   <Tabs.Tab>\n *\n *     You can get your Copilot Cloud API key for free by <LinkToCopilotCloud>signing up here</LinkToCopilotCloud>.\n *\n *     ```tsx\n *     import { CopilotKit } from \"@copilotkit/react-core\";\n *\n *     <CopilotKit publicApiKey=\"<your-public-api-key>\">\n *       // ... your app ...\n *     </CopilotKit>\n *     ```\n *   </Tabs.Tab>\n *\n *   <Tabs.Tab>\n *     You can find more information about self-hosting CopilotKit [here](/concepts/copilot-runtime).\n *\n *     ```tsx\n *     import { CopilotKit } from \"@copilotkit/react-core\";\n *\n *     <CopilotKit runtimeUrl=\"<your-runtime-url>\">\n *       // ... your app ...\n *     </CopilotKit>\n *     ```\n *   </Tabs.Tab>\n * </Tabs>\n */\nimport { useCallback, useRef, useState } from \"react\";\nimport {\n  CopilotContext,\n  CopilotApiConfig,\n  InChatRenderFunction,\n  ChatComponentsCache,\n} from \"../../context/copilot-context\";\nimport useTree from \"../../hooks/use-tree\";\nimport { CopilotChatSuggestionConfiguration, DocumentPointer } from \"../../types\";\nimport { flushSync } from \"react-dom\";\nimport {\n  COPILOT_CLOUD_CHAT_URL,\n  CopilotCloudConfig,\n  FunctionCallHandler,\n} from \"@copilotkit/shared\";\nimport { AgentStateMessage, Message } from \"@copilotkit/runtime-client-gql\";\n\nimport { FrontendAction } from \"../../types/frontend-action\";\nimport useFlatCategoryStore from \"../../hooks/use-flat-category-store\";\nimport { CopilotKitProps } from \"./copilotkit-props\";\nimport { CopilotChatUI } from \"../../hooks/use-copilot-chat-ui\";\n\nexport function CopilotKit({ children, ...props }: CopilotKitProps) {\n  // Compute all the functions and properties that we need to pass\n  // to the CopilotContext.\n\n  if (!props.runtimeUrl && !props.publicApiKey) {\n    throw new Error(\n      \"Please provide either a runtimeUrl or a publicApiKey to the CopilotKit component.\",\n    );\n  }\n\n  const chatApiEndpoint = props.runtimeUrl || COPILOT_CLOUD_CHAT_URL;\n\n  const [actions, setActions] = useState<Record<string, FrontendAction<any>>>({});\n  const chatComponentsCache = useRef<ChatComponentsCache>({ actions: {}, chatUI: {} });\n  const { addElement, removeElement, printTree } = useTree();\n  const [messages, setMessages] = useState<Message[]>([]);\n  const [isLoading, setIsLoading] = useState(false);\n  const [chatInstructions, setChatInstructions] = useState(\"\");\n\n  const {\n    addElement: addDocument,\n    removeElement: removeDocument,\n    allElements: allDocuments,\n  } = useFlatCategoryStore<DocumentPointer>();\n\n  const setAction = useCallback((id: string, action: FrontendAction<any>) => {\n    setActions((prevPoints) => {\n      return {\n        ...prevPoints,\n        [id]: action,\n      };\n    });\n  }, []);\n\n  const removeAction = useCallback((id: string) => {\n    setActions((prevPoints) => {\n      const newPoints = { ...prevPoints };\n      delete newPoints[id];\n      return newPoints;\n    });\n  }, []);\n\n  const getContextString = useCallback(\n    (documents: DocumentPointer[], categories: string[]) => {\n      const documentsString = documents\n        .map((document) => {\n          return `${document.name} (${document.sourceApplication}):\\n${document.getContents()}`;\n        })\n        .join(\"\\n\\n\");\n\n      const nonDocumentStrings = printTree(categories);\n\n      return `${documentsString}\\n\\n${nonDocumentStrings}`;\n    },\n    [printTree],\n  );\n\n  const addContext = useCallback(\n    (\n      context: string,\n      parentId?: string,\n      categories: string[] = defaultCopilotContextCategories,\n    ) => {\n      return addElement(context, categories, parentId);\n    },\n    [addElement],\n  );\n\n  const removeContext = useCallback(\n    (id: string) => {\n      removeElement(id);\n    },\n    [removeElement],\n  );\n\n  const getFunctionCallHandler = useCallback(\n    (customEntryPoints?: Record<string, FrontendAction<any>>) => {\n      return entryPointsToFunctionCallHandler(Object.values(customEntryPoints || actions));\n    },\n    [actions],\n  );\n\n  const getDocumentsContext = useCallback(\n    (categories: string[]) => {\n      return allDocuments(categories);\n    },\n    [allDocuments],\n  );\n\n  const addDocumentContext = useCallback(\n    (documentPointer: DocumentPointer, categories: string[] = defaultCopilotContextCategories) => {\n      return addDocument(documentPointer, categories);\n    },\n    [addDocument],\n  );\n\n  const removeDocumentContext = useCallback(\n    (documentId: string) => {\n      removeDocument(documentId);\n    },\n    [removeDocument],\n  );\n\n  if (!props.publicApiKey) {\n    if (props.cloudRestrictToTopic) {\n      throw new Error(\n        \"To use the cloudRestrictToTopic feature, please sign up at https://copilotkit.ai and provide a publicApiKey.\",\n      );\n    }\n  }\n\n  let cloud: CopilotCloudConfig | undefined = undefined;\n  if (props.publicApiKey) {\n    cloud = {\n      guardrails: {\n        input: {\n          restrictToTopic: {\n            enabled: props.cloudRestrictToTopic ? true : false,\n            validTopics: props.cloudRestrictToTopic?.validTopics || [],\n            invalidTopics: props.cloudRestrictToTopic?.invalidTopics || [],\n          },\n        },\n      },\n    };\n  }\n\n  // get the appropriate CopilotApiConfig from the props\n  const copilotApiConfig: CopilotApiConfig = {\n    publicApiKey: props.publicApiKey,\n    ...(cloud ? { cloud } : {}),\n    chatApiEndpoint: chatApiEndpoint,\n    headers: props.headers || {},\n    properties: props.properties || {},\n    transcribeAudioUrl: props.transcribeAudioUrl,\n    textToSpeechUrl: props.textToSpeechUrl,\n    credentials: props.credentials,\n  };\n\n  const [chatSuggestionConfiguration, setChatSuggestionConfiguration] = useState<{\n    [key: string]: CopilotChatSuggestionConfiguration;\n  }>({});\n\n  const addChatSuggestionConfiguration = (\n    id: string,\n    suggestion: CopilotChatSuggestionConfiguration,\n  ) => {\n    setChatSuggestionConfiguration((prev) => ({ ...prev, [id]: suggestion }));\n  };\n\n  const removeChatSuggestionConfiguration = (id: string) => {\n    setChatSuggestionConfiguration((prev) => {\n      const { [id]: _, ...rest } = prev;\n      return rest;\n    });\n  };\n\n  const [agentStates, setAgentStates] = useState<Record<string, AgentStateMessage | null>>({});\n\n  const [chatUI, setChatUI] = useState<CopilotChatUI[]>([]);\n\n  return (\n    <CopilotContext.Provider\n      value={{\n        actions,\n        chatComponentsCache,\n        getFunctionCallHandler,\n        setAction,\n        removeAction,\n        getContextString,\n        addContext,\n        removeContext,\n        getDocumentsContext,\n        addDocumentContext,\n        removeDocumentContext,\n        copilotApiConfig: copilotApiConfig,\n        messages,\n        setMessages,\n        isLoading,\n        setIsLoading,\n        chatSuggestionConfiguration,\n        addChatSuggestionConfiguration,\n        removeChatSuggestionConfiguration,\n        chatInstructions,\n        setChatInstructions,\n        showDevConsole: props.showDevConsole === undefined ? \"auto\" : props.showDevConsole,\n        agentStates,\n        setAgentStates,\n        chatUI,\n        setChatUI,\n      }}\n    >\n      {children}\n    </CopilotContext.Provider>\n  );\n}\n\nexport const defaultCopilotContextCategories = [\"global\"];\n\nfunction entryPointsToFunctionCallHandler(actions: FrontendAction<any>[]): FunctionCallHandler {\n  return async ({ messages, name, args }) => {\n    let actionsByFunctionName: Record<string, FrontendAction<any>> = {};\n    for (let action of actions) {\n      actionsByFunctionName[action.name] = action;\n    }\n\n    const action = actionsByFunctionName[name];\n    let result: any = undefined;\n    if (action) {\n      await new Promise<void>((resolve, reject) => {\n        flushSync(async () => {\n          try {\n            result = await action.handler(args);\n            resolve();\n          } catch (error) {\n            reject(error);\n          }\n        });\n      });\n      await new Promise((resolve) => setTimeout(resolve, 20));\n    }\n    return result;\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAgCA,SAAS,aAAa,QAAQ,gBAAgB;AAS9C,SAAS,iBAAiB;AAC1B;AAAA,EACE;AAAA,OAGK;AAyKH;AAjKG,SAAS,WAAW,IAAyC;AAAzC,eAAE,WAtD7B,IAsD2B,IAAe,kBAAf,IAAe,CAAb;AAtD7B,MAAAA,KAAAC;AA0DE,MAAI,CAAC,MAAM,cAAc,CAAC,MAAM,cAAc;AAC5C,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAEA,QAAM,kBAAkB,MAAM,cAAc;AAE5C,QAAM,CAAC,SAAS,UAAU,IAAI,SAA8C,CAAC,CAAC;AAC9E,QAAM,sBAAsB,OAA4B,EAAE,SAAS,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC;AACnF,QAAM,EAAE,YAAY,eAAe,UAAU,IAAI,iBAAQ;AACzD,QAAM,CAAC,UAAU,WAAW,IAAI,SAAoB,CAAC,CAAC;AACtD,QAAM,CAAC,WAAW,YAAY,IAAI,SAAS,KAAK;AAChD,QAAM,CAAC,kBAAkB,mBAAmB,IAAI,SAAS,EAAE;AAE3D,QAAM;AAAA,IACJ,YAAY;AAAA,IACZ,eAAe;AAAA,IACf,aAAa;AAAA,EACf,IAAI,gCAAsC;AAE1C,QAAM,YAAY,YAAY,CAAC,IAAY,WAAgC;AACzE,eAAW,CAAC,eAAe;AACzB,aAAO,iCACF,aADE;AAAA,QAEL,CAAC,EAAE,GAAG;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH,GAAG,CAAC,CAAC;AAEL,QAAM,eAAe,YAAY,CAAC,OAAe;AAC/C,eAAW,CAAC,eAAe;AACzB,YAAM,YAAY,mBAAK;AACvB,aAAO,UAAU,EAAE;AACnB,aAAO;AAAA,IACT,CAAC;AAAA,EACH,GAAG,CAAC,CAAC;AAEL,QAAM,mBAAmB;AAAA,IACvB,CAAC,WAA8B,eAAyB;AACtD,YAAM,kBAAkB,UACrB,IAAI,CAAC,aAAa;AACjB,eAAO,GAAG,SAAS,SAAS,SAAS;AAAA,EAAwB,SAAS,YAAY;AAAA,MACpF,CAAC,EACA,KAAK,MAAM;AAEd,YAAM,qBAAqB,UAAU,UAAU;AAE/C,aAAO,GAAG;AAAA;AAAA,EAAsB;AAAA,IAClC;AAAA,IACA,CAAC,SAAS;AAAA,EACZ;AAEA,QAAM,aAAa;AAAA,IACjB,CACE,SACA,UACA,aAAuB,oCACpB;AACH,aAAO,WAAW,SAAS,YAAY,QAAQ;AAAA,IACjD;AAAA,IACA,CAAC,UAAU;AAAA,EACb;AAEA,QAAM,gBAAgB;AAAA,IACpB,CAAC,OAAe;AACd,oBAAc,EAAE;AAAA,IAClB;AAAA,IACA,CAAC,aAAa;AAAA,EAChB;AAEA,QAAM,yBAAyB;AAAA,IAC7B,CAAC,sBAA4D;AAC3D,aAAO,iCAAiC,OAAO,OAAO,qBAAqB,OAAO,CAAC;AAAA,IACrF;AAAA,IACA,CAAC,OAAO;AAAA,EACV;AAEA,QAAM,sBAAsB;AAAA,IAC1B,CAAC,eAAyB;AACxB,aAAO,aAAa,UAAU;AAAA,IAChC;AAAA,IACA,CAAC,YAAY;AAAA,EACf;AAEA,QAAM,qBAAqB;AAAA,IACzB,CAAC,iBAAkC,aAAuB,oCAAoC;AAC5F,aAAO,YAAY,iBAAiB,UAAU;AAAA,IAChD;AAAA,IACA,CAAC,WAAW;AAAA,EACd;AAEA,QAAM,wBAAwB;AAAA,IAC5B,CAAC,eAAuB;AACtB,qBAAe,UAAU;AAAA,IAC3B;AAAA,IACA,CAAC,cAAc;AAAA,EACjB;AAEA,MAAI,CAAC,MAAM,cAAc;AACvB,QAAI,MAAM,sBAAsB;AAC9B,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,MAAI,QAAwC;AAC5C,MAAI,MAAM,cAAc;AACtB,YAAQ;AAAA,MACN,YAAY;AAAA,QACV,OAAO;AAAA,UACL,iBAAiB;AAAA,YACf,SAAS,MAAM,uBAAuB,OAAO;AAAA,YAC7C,eAAaD,MAAA,MAAM,yBAAN,gBAAAA,IAA4B,gBAAe,CAAC;AAAA,YACzD,iBAAeC,MAAA,MAAM,yBAAN,gBAAAA,IAA4B,kBAAiB,CAAC;AAAA,UAC/D;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAGA,QAAM,mBAAqC;AAAA,IACzC,cAAc,MAAM;AAAA,KAChB,QAAQ,EAAE,MAAM,IAAI,CAAC,IAFgB;AAAA,IAGzC;AAAA,IACA,SAAS,MAAM,WAAW,CAAC;AAAA,IAC3B,YAAY,MAAM,cAAc,CAAC;AAAA,IACjC,oBAAoB,MAAM;AAAA,IAC1B,iBAAiB,MAAM;AAAA,IACvB,aAAa,MAAM;AAAA,EACrB;AAEA,QAAM,CAAC,6BAA6B,8BAA8B,IAAI,SAEnE,CAAC,CAAC;AAEL,QAAM,iCAAiC,CACrC,IACA,eACG;AACH,mCAA+B,CAAC,SAAU,iCAAK,OAAL,EAAW,CAAC,EAAE,GAAG,WAAW,EAAE;AAAA,EAC1E;AAEA,QAAM,oCAAoC,CAAC,OAAe;AACxD,mCAA+B,CAAC,SAAS;AACvC,YAA6BD,MAAA,MAApB,EA7Mf,CA6Me,KAAK,EA7MpB,IA6MmCA,KAAT,iBAASA,KAAT,CAAX;AACT,aAAO;AAAA,IACT,CAAC;AAAA,EACH;AAEA,QAAM,CAAC,aAAa,cAAc,IAAI,SAAmD,CAAC,CAAC;AAE3F,QAAM,CAAC,QAAQ,SAAS,IAAI,SAA0B,CAAC,CAAC;AAExD,SACE;AAAA,IAAC,eAAe;AAAA,IAAf;AAAA,MACC,OAAO;AAAA,QACL;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA,gBAAgB,MAAM,mBAAmB,SAAY,SAAS,MAAM;AAAA,QACpE;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF;AAAA,MAEC;AAAA;AAAA,EACH;AAEJ;AAEO,IAAM,kCAAkC,CAAC,QAAQ;AAExD,SAAS,iCAAiC,SAAqD;AAC7F,SAAO,CAAO,OAA6B,eAA7B,KAA6B,WAA7B,EAAE,UAAU,MAAM,KAAK,GAAM;AACzC,QAAI,wBAA6D,CAAC;AAClE,aAASE,WAAU,SAAS;AAC1B,4BAAsBA,QAAO,IAAI,IAAIA;AAAA,IACvC;AAEA,UAAM,SAAS,sBAAsB,IAAI;AACzC,QAAI,SAAc;AAClB,QAAI,QAAQ;AACV,YAAM,IAAI,QAAc,CAAC,SAAS,WAAW;AAC3C,kBAAU,MAAY;AACpB,cAAI;AACF,qBAAS,MAAM,OAAO,QAAQ,IAAI;AAClC,oBAAQ;AAAA,UACV,SAAS,OAAP;AACA,mBAAO,KAAK;AAAA,UACd;AAAA,QACF,EAAC;AAAA,MACH,CAAC;AACD,YAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,EAAE,CAAC;AAAA,IACxD;AACA,WAAO;AAAA,EACT;AACF;","names":["_a","_b","action"]}

package/dist/{chunk-6X3OMCLF.mjs → chunk-RFFNNNVK.mjs}

@@ -1,12 +1,12 @@
 import {
   defaultCopilotContextCategories
-} from "./chunk-AWRZLWWD.mjs";
+} from "./chunk-IF6P5ZXR.mjs";
 import {
   useChat
-} from "./chunk-AEMRMXIF.mjs";
+} from "./chunk-52M7642J.mjs";
 import {
   CopilotContext
-} from "./chunk-ZOXWEKMF.mjs";
+} from "./chunk-DB4VWZ5Q.mjs";
 import {
   __objRest,
   __spreadProps,
@@ -32,7 +32,7 @@ function useCopilotChat(_a = {}) {
     setIsLoading,
     chatInstructions,
     actions,
-    setAgentState
+    setAgentStates
   } = useContext(CopilotContext);
   const latestGetContextString = useUpdatedRef(getContextString);
   const deleteMessage = useCallback(
@@ -59,7 +59,7 @@ function useCopilotChat(_a = {}) {
     makeSystemMessageCallback,
     isLoading,
     setIsLoading,
-    setAgentState
+    setAgentStates
   }));
   return {
     visibleMessages: messages,
@@ -106,4 +106,4 @@ export {
   useCopilotChat,
   defaultSystemMessage
 };
-//# sourceMappingURL=chunk-6X3OMCLF.mjs.map
+//# sourceMappingURL=chunk-RFFNNNVK.mjs.map

package/dist/chunk-RFFNNNVK.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/hooks/use-copilot-chat.ts"],"sourcesContent":["/**\n * `useCopilotChat` is a React hook that lets you directly interact with the\n * Copilot instance. Use to implement a fully custom UI (headless UI) or to\n * programmatically interact with the Copilot instance managed by the default\n * UI.\n *\n * ## Usage\n *\n * ### Simple Usage\n *\n * ```tsx\n * import { useCopilotChat } from \"@copilotkit/react-core\";\n * import { Role, TextMessage } from \"@copilotkit/runtime-client-gql\";\n *\n * export function YourComponent() {\n *   const { appendMessage } = useCopilotChat();\n *\n *   appendMessage(\n *     new TextMessage({\n *       content: \"Hello World\",\n *       role: Role.User,\n *     }),\n *   );\n * }\n * ```\n *\n * `useCopilotChat` returns an object with the following properties:\n *\n * ```tsx\n * const {\n *   visibleMessages, // An array of messages that are currently visible in the chat.\n *   appendMessage, // A function to append a message to the chat.\n *   setMessages, // A function to set the messages in the chat.\n *   deleteMessage, // A function to delete a message from the chat.\n *   reloadMessages, // A function to reload the messages from the API.\n *   stopGeneration, // A function to stop the generation of the next message.\n *   isLoading, // A boolean indicating if the chat is loading.\n * } = useCopilotChat();\n * ```\n */\nimport { useContext, useRef, useEffect, useCallback } from \"react\";\nimport { CopilotContext } from \"../context/copilot-context\";\nimport { Message, Role, TextMessage } from \"@copilotkit/runtime-client-gql\";\nimport { SystemMessageFunction } from \"../types\";\nimport { useChat } from \"./use-chat\";\nimport { defaultCopilotContextCategories } from \"../components\";\nimport { MessageStatusCode } from \"@copilotkit/runtime-client-gql\";\n\nexport interface UseCopilotChatOptions {\n  /**\n   * A unique identifier for the chat. If not provided, a random one will be\n   * generated. When provided, the `useChat` hook with the same `id` will\n   * have shared states across components.\n   */\n  id?: string;\n\n  /**\n   * HTTP headers to be sent with the API request.\n   */\n  headers?: Record<string, string> | Headers;\n  /**\n   * System messages of the chat. Defaults to an empty array.\n   */\n  initialMessages?: Message[];\n\n  /**\n   * A function to generate the system message. Defaults to `defaultSystemMessage`.\n   */\n  makeSystemMessage?: SystemMessageFunction;\n}\n\nexport interface UseCopilotChatReturn {\n  visibleMessages: Message[];\n  appendMessage: (message: Message) => Promise<void>;\n  setMessages: (messages: Message[]) => void;\n  deleteMessage: (messageId: string) => void;\n  reloadMessages: () => Promise<void>;\n  stopGeneration: () => void;\n  isLoading: boolean;\n}\n\nexport function useCopilotChat({\n  makeSystemMessage,\n  ...options\n}: UseCopilotChatOptions = {}): UseCopilotChatReturn {\n  const {\n    getContextString,\n    getFunctionCallHandler,\n    copilotApiConfig,\n    messages,\n    setMessages,\n    isLoading,\n    setIsLoading,\n    chatInstructions,\n    actions,\n    setAgentStates,\n  } = useContext(CopilotContext);\n\n  // We need to ensure that makeSystemMessageCallback always uses the latest\n  // useCopilotReadable data.\n  const latestGetContextString = useUpdatedRef(getContextString);\n  const deleteMessage = useCallback(\n    (messageId: string) => {\n      setMessages((prev) => prev.filter((message) => message.id !== messageId));\n    },\n    [setMessages],\n  );\n\n  const makeSystemMessageCallback = useCallback(() => {\n    const systemMessageMaker = makeSystemMessage || defaultSystemMessage;\n    // this always gets the latest context string\n    const contextString = latestGetContextString.current([], defaultCopilotContextCategories); // TODO: make the context categories configurable\n\n    return new TextMessage({\n      content: systemMessageMaker(contextString, chatInstructions),\n      role: Role.System,\n    });\n  }, [getContextString, makeSystemMessage, chatInstructions]);\n\n  const { append, reload, stop } = useChat({\n    ...options,\n    actions: Object.values(actions),\n    copilotConfig: copilotApiConfig,\n    initialMessages: options.initialMessages || [],\n    onFunctionCall: getFunctionCallHandler(),\n    messages,\n    setMessages,\n    makeSystemMessageCallback,\n    isLoading,\n    setIsLoading,\n    setAgentStates,\n  });\n\n  return {\n    visibleMessages: messages,\n    appendMessage: append,\n    setMessages,\n    reloadMessages: reload,\n    stopGeneration: stop,\n    deleteMessage,\n    isLoading,\n  };\n}\n\n// store `value` in a ref and update\n// it whenever it changes.\nfunction useUpdatedRef<T>(value: T) {\n  const ref = useRef(value);\n\n  useEffect(() => {\n    ref.current = value;\n  }, [value]);\n\n  return ref;\n}\n\nexport function defaultSystemMessage(\n  contextString: string,\n  additionalInstructions?: string,\n): string {\n  return (\n    `\nPlease act as an efficient, competent, conscientious, and industrious professional assistant.\n\nHelp the user achieve their goals, and you do so in a way that is as efficient as possible, without unnecessary fluff, but also without sacrificing professionalism.\nAlways be polite and respectful, and prefer brevity over verbosity.\n\nThe user has provided you with the following context:\n\\`\\`\\`\n${contextString}\n\\`\\`\\`\n\nThey have also provided you with functions you can call to initiate actions on their behalf, or functions you can call to receive more information.\n\nPlease assist them as best you can.\n\nYou can ask them for clarifying questions if needed, but don't be annoying about it. If you can reasonably 'fill in the blanks' yourself, do so.\n\nIf you would like to call a function, call it without saying anything else.\n` + (additionalInstructions ? `\\n\\n${additionalInstructions}` : \"\")\n  );\n}\n"],"mappings":";;;;;;;;;;;;;;;;AAwCA,SAAS,YAAY,QAAQ,WAAW,mBAAmB;AAE3D,SAAkB,MAAM,mBAAmB;AAuCpC,SAAS,eAAe,KAGJ,CAAC,GAAyB;AAHtB,eAC7B;AAAA;AAAA,EAlFF,IAiF+B,IAE1B,oBAF0B,IAE1B;AAAA,IADH;AAAA;AAGA,QAAM;AAAA,IACJ;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,IAAI,WAAW,cAAc;AAI7B,QAAM,yBAAyB,cAAc,gBAAgB;AAC7D,QAAM,gBAAgB;AAAA,IACpB,CAAC,cAAsB;AACrB,kBAAY,CAAC,SAAS,KAAK,OAAO,CAAC,YAAY,QAAQ,OAAO,SAAS,CAAC;AAAA,IAC1E;AAAA,IACA,CAAC,WAAW;AAAA,EACd;AAEA,QAAM,4BAA4B,YAAY,MAAM;AAClD,UAAM,qBAAqB,qBAAqB;AAEhD,UAAM,gBAAgB,uBAAuB,QAAQ,CAAC,GAAG,+BAA+B;AAExF,WAAO,IAAI,YAAY;AAAA,MACrB,SAAS,mBAAmB,eAAe,gBAAgB;AAAA,MAC3D,MAAM,KAAK;AAAA,IACb,CAAC;AAAA,EACH,GAAG,CAAC,kBAAkB,mBAAmB,gBAAgB,CAAC;AAE1D,QAAM,EAAE,QAAQ,QAAQ,KAAK,IAAI,QAAQ,iCACpC,UADoC;AAAA,IAEvC,SAAS,OAAO,OAAO,OAAO;AAAA,IAC9B,eAAe;AAAA,IACf,iBAAiB,QAAQ,mBAAmB,CAAC;AAAA,IAC7C,gBAAgB,uBAAuB;AAAA,IACvC;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,EAAC;AAED,SAAO;AAAA,IACL,iBAAiB;AAAA,IACjB,eAAe;AAAA,IACf;AAAA,IACA,gBAAgB;AAAA,IAChB,gBAAgB;AAAA,IAChB;AAAA,IACA;AAAA,EACF;AACF;AAIA,SAAS,cAAiB,OAAU;AAClC,QAAM,MAAM,OAAO,KAAK;AAExB,YAAU,MAAM;AACd,QAAI,UAAU;AAAA,EAChB,GAAG,CAAC,KAAK,CAAC;AAEV,SAAO;AACT;AAEO,SAAS,qBACd,eACA,wBACQ;AACR,SACE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAUG,yBAAyB;AAAA;AAAA,EAAO,2BAA2B;AAEhE;","names":[]}

package/dist/chunk-TX4ZVFGR.mjs

@@ -0,0 +1,43 @@
+import {
+  useCopilotChat
+} from "./chunk-RFFNNNVK.mjs";
+import {
+  useCopilotContext
+} from "./chunk-DB4VWZ5Q.mjs";
+import {
+  __spreadProps,
+  __spreadValues
+} from "./chunk-SKC7AJIV.mjs";
+
+// src/hooks/use-agent-state.ts
+function useAgentState(agentName) {
+  const { agentStates, setAgentStates } = useCopilotContext();
+  const { appendMessage } = useCopilotChat();
+  const setSpecificAgentState = (newState) => {
+    setAgentStates((prevAgentStates) => {
+      const currentState2 = prevAgentStates[agentName];
+      if (!currentState2) {
+        throw new Error(`Agent state ${agentName} not found`);
+      }
+      const updatedState = typeof newState === "function" ? newState(currentState2) : newState;
+      const agentStateMessage = __spreadProps(__spreadValues({}, currentState2), { state: updatedState });
+      appendMessage(agentStateMessage);
+      return __spreadProps(__spreadValues({}, prevAgentStates), {
+        [agentName]: agentStateMessage
+      });
+    });
+  };
+  const currentState = agentStates[agentName];
+  return [
+    currentState == null ? void 0 : currentState.state,
+    setSpecificAgentState,
+    currentState ? { nodeName: currentState.nodeName } : void 0
+  ];
+}
+var use_agent_state_default = useAgentState;
+
+export {
+  useAgentState,
+  use_agent_state_default
+};
+//# sourceMappingURL=chunk-TX4ZVFGR.mjs.map

package/dist/chunk-TX4ZVFGR.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/hooks/use-agent-state.ts"],"sourcesContent":["import { useCopilotContext } from \"../context\";\nimport { useCopilotChat } from \"./use-copilot-chat\";\n\nexport type AgentStateInfo = {\n  nodeName: string;\n};\n\ntype SetAgentStateAction<T = any> = (newState: T | ((prevState: T | undefined) => T)) => void;\n\nexport type UseAgentStateReturnType<T = any> = [\n  T | undefined,\n  SetAgentStateAction<T>,\n  AgentStateInfo | undefined,\n];\n\nexport function useAgentState<T = any>(agentName: string): UseAgentStateReturnType<T> {\n  const { agentStates, setAgentStates } = useCopilotContext();\n\n  const { appendMessage } = useCopilotChat();\n\n  const setSpecificAgentState: SetAgentStateAction<T> = (newState) => {\n    setAgentStates((prevAgentStates) => {\n      const currentState = prevAgentStates[agentName];\n      if (!currentState) {\n        throw new Error(`Agent state ${agentName} not found`);\n      }\n\n      const updatedState =\n        typeof newState === \"function\" ? (newState as Function)(currentState) : newState;\n\n      const agentStateMessage = { ...currentState, state: updatedState };\n\n      appendMessage(agentStateMessage);\n\n      return {\n        ...prevAgentStates,\n        [agentName]: agentStateMessage,\n      };\n    });\n  };\n\n  const currentState = agentStates[agentName];\n\n  return [\n    currentState?.state as T,\n    setSpecificAgentState,\n    currentState ? { nodeName: currentState.nodeName } : undefined,\n  ];\n}\n\nexport default useAgentState;\n"],"mappings":";;;;;;;;;;;;AAeO,SAAS,cAAuB,WAA+C;AACpF,QAAM,EAAE,aAAa,eAAe,IAAI,kBAAkB;AAE1D,QAAM,EAAE,cAAc,IAAI,eAAe;AAEzC,QAAM,wBAAgD,CAAC,aAAa;AAClE,mBAAe,CAAC,oBAAoB;AAClC,YAAMA,gBAAe,gBAAgB,SAAS;AAC9C,UAAI,CAACA,eAAc;AACjB,cAAM,IAAI,MAAM,eAAe,qBAAqB;AAAA,MACtD;AAEA,YAAM,eACJ,OAAO,aAAa,aAAc,SAAsBA,aAAY,IAAI;AAE1E,YAAM,oBAAoB,iCAAKA,gBAAL,EAAmB,OAAO,aAAa;AAEjE,oBAAc,iBAAiB;AAE/B,aAAO,iCACF,kBADE;AAAA,QAEL,CAAC,SAAS,GAAG;AAAA,MACf;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,eAAe,YAAY,SAAS;AAE1C,SAAO;AAAA,IACL,6CAAc;AAAA,IACd;AAAA,IACA,eAAe,EAAE,UAAU,aAAa,SAAS,IAAI;AAAA,EACvD;AACF;AAEA,IAAO,0BAAQ;","names":["currentState"]}

package/dist/components/copilot-provider/copilotkit.js

@@ -94,7 +94,7 @@ var emptyCopilotContext = {
   },
   removeAction: () => {
   },
-  chatComponentsCache: { current: {} },
+  chatComponentsCache: { current: { actions: {}, chatUI: {} } },
   getContextString: (documents, categories) => returnAndThrowInDebug(""),
   addContext: () => "",
   removeContext: () => {
@@ -128,8 +128,11 @@ var emptyCopilotContext = {
   removeChatSuggestionConfiguration: () => {
   },
   showDevConsole: "auto",
-  agentState: null,
-  setAgentState: () => {
+  agentStates: {},
+  setAgentStates: () => {
+  },
+  chatUI: [],
+  setChatUI: () => {
   }
 };
 var CopilotContext = import_react.default.createContext(emptyCopilotContext);
@@ -351,7 +354,7 @@ function CopilotKit(_a) {
   }
   const chatApiEndpoint = props.runtimeUrl || import_shared3.COPILOT_CLOUD_CHAT_URL;
   const [actions, setActions] = (0, import_react4.useState)({});
-  const chatComponentsCache = (0, import_react4.useRef)({});
+  const chatComponentsCache = (0, import_react4.useRef)({ actions: {}, chatUI: {} });
   const { addElement, removeElement, printTree } = use_tree_default();
   const [messages, setMessages] = (0, import_react4.useState)([]);
   const [isLoading, setIsLoading] = (0, import_react4.useState)(false);
@@ -465,7 +468,8 @@ ${nonDocumentStrings}`;
       return rest;
     });
   };
-  const [agentState, setAgentState] = (0, import_react4.useState)(null);
+  const [agentStates, setAgentStates] = (0, import_react4.useState)({});
+  const [chatUI, setChatUI] = (0, import_react4.useState)([]);
   return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
     CopilotContext.Provider,
     {
@@ -492,8 +496,10 @@ ${nonDocumentStrings}`;
         chatInstructions,
         setChatInstructions,
         showDevConsole: props.showDevConsole === void 0 ? "auto" : props.showDevConsole,
-        agentState,
-        setAgentState
+        agentStates,
+        setAgentStates,
+        chatUI,
+        setChatUI
       },
       children
     }