@copilotkit/react-core 0.18.0-alpha.1 → 0.18.0-alpha.3

package/src/hooks/use-chat.ts

@@ -1,7 +1,7 @@
 import { useRef, useState } from "react";
-import { Message, Function, FunctionCallHandler } from "../types";
+import { Message, Function, FunctionCallHandler, FunctionCall } from "@copilotkit/shared";
 import { nanoid } from "nanoid";
-import { ChatCompletionClient } from "../openai/chat-completion-client";
+import { fetchAndDecodeChatCompletion } from "../utils/fetch-chat-completion";
 import { CopilotApiConfig } from "../context";
 
 export type UseChatOptions = {
@@ -87,84 +87,66 @@ export function useChat(options: UseChatOptionsWithCopilotConfig): UseChatHelper
   const abortControllerRef = useRef<AbortController>();
 
   const runChatCompletion = async (messages: Message[]): Promise<Message> => {
-    return new Promise<Message>((resolve, reject) => {
-      setIsLoading(true);
-
-      const abortController = new AbortController();
-      abortControllerRef.current = abortController;
-
-      const assistantMessage: Message = {
-        id: nanoid(),
-        createdAt: new Date(),
-        content: "",
-        role: "assistant",
-      };
-
-      // Assistant messages are always copied when using setState
-      setMessages([...messages, { ...assistantMessage }]);
-
-      const messagesWithContext = [...(options.initialMessages || []), ...messages];
-
-      const client = new ChatCompletionClient({});
-
-      const cleanup = () => {
-        client.off("content");
-        client.off("end");
-        client.off("error");
-        client.off("function");
-
-        abortControllerRef.current = undefined;
-      };
-
-      abortController.signal.addEventListener("abort", () => {
-        setIsLoading(false);
-        cleanup();
-        reject(new DOMException("Aborted", "AbortError"));
-      });
-
-      client.on("content", (content) => {
-        assistantMessage.content += content;
-        setMessages([...messages, { ...assistantMessage }]);
-      });
+    setIsLoading(true);
+    const assistantMessage: Message = {
+      id: nanoid(),
+      createdAt: new Date(),
+      content: "",
+      role: "assistant",
+    };
+    const abortController = new AbortController();
+    abortControllerRef.current = abortController;
+
+    setMessages([...messages, { ...assistantMessage }]);
+
+    const messagesWithContext = [...(options.initialMessages || []), ...messages];
+    const response = await fetchAndDecodeChatCompletion({
+      copilotConfig: options.copilotConfig,
+      messages: messagesWithContext,
+      functions: options.functions,
+      headers: options.headers,
+      signal: abortController.signal,
+    });
 
-      client.on("end", () => {
-        setIsLoading(false);
-        cleanup();
-        resolve({ ...assistantMessage });
-      });
+    if (!response.events) {
+      throw new Error("Failed to fetch chat completion");
+    }
 
-      client.on("error", (error) => {
-        setIsLoading(false);
-        cleanup();
-        reject(error);
-      });
-
-      client.on("function", async (functionCall) => {
-        assistantMessage.function_call = {
-          name: functionCall.name,
-          arguments: JSON.stringify(functionCall.arguments),
-        };
-        setMessages([...messages, { ...assistantMessage }]);
-        // quit early if we get a function call
+    const reader = response.events.getReader();
+
+    while (true) {
+      try {
+        const { done, value } = await reader.read();
+
+        if (done) {
+          setIsLoading(false);
+          return { ...assistantMessage };
+        }
+
+        if (value.type === "content") {
+          assistantMessage.content += value.content;
+          setMessages([...messages, { ...assistantMessage }]);
+        } else if (value.type === "function") {
+          assistantMessage.function_call = {
+            name: value.name,
+            arguments: JSON.stringify(value.arguments),
+          };
+          setMessages([...messages, { ...assistantMessage }]);
+          // quit early if we get a function call
+          setIsLoading(false);
+          return { ...assistantMessage };
+        }
+      } catch (error) {
         setIsLoading(false);
-        cleanup();
-        resolve({ ...assistantMessage });
-      });
-
-      client.fetch({
-        copilotConfig: options.copilotConfig,
-        messages: messagesWithContext,
-        functions: options.functions,
-        headers: options.headers,
-        signal: abortController.signal,
-      });
-    });
+        throw error;
+      }
+    }
   };
 
   const runChatCompletionAndHandleFunctionCall = async (messages: Message[]): Promise<void> => {
     const message = await runChatCompletion(messages);
     if (message.function_call && options.onFunctionCall) {
-      await options.onFunctionCall(messages, message.function_call);
+      await options.onFunctionCall(messages, message.function_call as FunctionCall);
     }
   };
 

package/src/hooks/use-copilot-chat.ts

@@ -1,9 +1,9 @@
 import { useMemo, useContext } from "react";
-import { CopilotContext, CopilotContextParams } from "../context/copilot-context";
-import { Message, SystemMessageFunction } from "../types";
+import { CopilotContext } from "../context/copilot-context";
+import { Message, Function } from "@copilotkit/shared";
+import { SystemMessageFunction } from "../types";
 import { UseChatOptions, useChat } from "./use-chat";
 import { defaultCopilotContextCategories } from "../components";
-import { ChatCompletionCreateParams } from "openai/resources/chat";
 
 export interface UseCopilotChatOptions extends UseChatOptions {
   makeSystemMessage?: SystemMessageFunction;
@@ -43,7 +43,7 @@ export function useCopilotChat({
     };
   }, [getContextString, makeSystemMessage]);
 
-  const functionDescriptions: ChatCompletionCreateParams.Function[] = useMemo(() => {
+  const functionDescriptions: Function[] = useMemo(() => {
     return getChatCompletionFunctionDescriptions();
   }, [getChatCompletionFunctionDescriptions]);
 

package/src/hooks/use-make-copilot-actionable.ts

@@ -2,7 +2,7 @@
 
 import { useRef, useContext, useEffect, useMemo } from "react";
 import { CopilotContext } from "../context/copilot-context";
-import { AnnotatedFunction } from "../types/annotated-function";
+import { AnnotatedFunction } from "@copilotkit/shared";
 import { nanoid } from "nanoid";
 
 export function useMakeCopilotActionable<ActionInput extends any[]>(

package/src/index.tsx

@@ -3,4 +3,9 @@ export * from "./context";
 export * from "./hooks";
 export * from "./types";
 export * from "./openai-assistants";
-export * from "./openai";
+export {
+  type FetchChatCompletionParams,
+  fetchChatCompletion,
+  fetchAndDecodeChatCompletion,
+  fetchAndDecodeChatCompletionAsText,
+} from "./utils/fetch-chat-completion";

package/src/types/index.ts

@@ -1,5 +1,2 @@
-export type { AnnotatedFunctionArgument } from "./annotated-function";
-export type { AnnotatedFunction } from "./annotated-function";
 export type { DocumentPointer } from "./document-pointer";
-export type { Role, Message, Function, FunctionCall, FunctionCallHandler } from "./base";
 export type { SystemMessageFunction } from "./system-message";

package/src/utils/fetch-chat-completion.ts

@@ -0,0 +1,95 @@
+import {
+  Message,
+  Function,
+  ChatCompletionEvent,
+  decodeChatCompletion,
+  parseChatCompletion,
+  decodeChatCompletionAsText,
+} from "@copilotkit/shared";
+import { CopilotApiConfig } from "../context";
+
+export interface FetchChatCompletionParams {
+  copilotConfig: CopilotApiConfig;
+  model?: string;
+  messages: Message[];
+  functions?: Function[];
+  temperature?: number;
+  maxTokens?: number;
+  headers?: Record<string, string> | Headers;
+  body?: object;
+  signal?: AbortSignal;
+}
+
+export async function fetchChatCompletion({
+  copilotConfig,
+  model,
+  messages,
+  functions,
+  temperature,
+  headers,
+  body,
+  signal,
+}: FetchChatCompletionParams): Promise<Response> {
+  temperature ||= 0.5;
+  functions ||= [];
+
+  // clean up any extra properties from messages
+  const cleanedMessages = messages.map((message) => {
+    const { content, role, name, function_call } = message;
+    return { content, role, name, function_call };
+  });
+
+  const response = await fetch(copilotConfig.chatApiEndpoint, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      ...copilotConfig.headers,
+      ...(headers ? { ...headers } : {}),
+    },
+    body: JSON.stringify({
+      model,
+      messages: cleanedMessages,
+      stream: true,
+      ...(functions.length ? { functions } : {}),
+      ...(temperature ? { temperature } : {}),
+      ...(functions.length != 0 ? { function_call: "auto" } : {}),
+      ...copilotConfig.body,
+      ...(body ? { ...body } : {}),
+    }),
+    signal,
+  });
+
+  return response;
+}
+
+export interface DecodedChatCompletionResponse extends Response {
+  events: ReadableStream<ChatCompletionEvent> | null;
+}
+
+export async function fetchAndDecodeChatCompletion(
+  params: FetchChatCompletionParams,
+): Promise<DecodedChatCompletionResponse> {
+  const response = await fetchChatCompletion(params);
+  if (!response.ok || !response.body) {
+    return { ...response, events: null };
+  }
+  const events = await decodeChatCompletion(parseChatCompletion(response.body));
+  return { ...response, events };
+}
+
+export interface DecodedChatCompletionResponseAsText extends Response {
+  events: ReadableStream<string> | null;
+}
+
+export async function fetchAndDecodeChatCompletionAsText(
+  params: FetchChatCompletionParams,
+): Promise<DecodedChatCompletionResponseAsText> {
+  const response = await fetchChatCompletion(params);
+  if (!response.ok || !response.body) {
+    return { ...response, events: null };
+  }
+  const events = await decodeChatCompletionAsText(
+    decodeChatCompletion(parseChatCompletion(response.body)),
+  );
+  return { ...response, events };
+}

package/tsup.config.ts

@@ -2,10 +2,9 @@ import { defineConfig, Options } from "tsup";
 
 export default defineConfig((options: Options) => ({
   entry: ["src/**/*.{ts,tsx}"],
-  format: ["esm"],
+  format: ["esm", "cjs"],
   dts: true,
   minify: false,
-  clean: true,
   external: ["react"],
   sourcemap: true,
   exclude: [

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

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/context/copilot-context.tsx"],"names":[],"mappings":";;;;;AAGA,OAAO,WAAW;AAmElB,IAAM,sBAA4C;AAAA,EAChD,aAAa,CAAC;AAAA,EACd,eAAe,MAAM;AAAA,EAAC;AAAA,EACtB,kBAAkB,MAAM;AAAA,EAAC;AAAA,EACzB,uCAAuC,MAAM,sBAAsB,CAAC,CAAC;AAAA,EACrE,wBAAwB,MAAM,sBAAsB,MAAY;AAAA,EAAC,EAAC;AAAA,EAElE,kBAAkB,CAAC,WAA8B,eAC/C,sBAAsB,EAAE;AAAA,EAC1B,YAAY,MAAM;AAAA,EAClB,eAAe,MAAM;AAAA,EAAC;AAAA,EAEtB,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;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAAA,IACA,IAAI,oBAA4B;AAC9B,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAAA,IACA,IAAI,UAAkC;AACpC,aAAO,CAAC;AAAA,IACV;AAAA,IACA,IAAI,OAA4B;AAC9B,aAAO,CAAC;AAAA,IACV;AAAA,EACF,EAAG;AACL;AAEO,IAAM,iBAAiB,MAAM,cAAoC,mBAAmB;AAE3F,SAAS,sBAAyB,OAAa;AAC7C,QAAM,IAAI;AAAA,IACR;AAAA,EACF;AACA,SAAO;AACT","sourcesContent":["\"use client\";\n\nimport { FunctionCallHandler } from \"../types\";\nimport React from \"react\";\nimport { TreeNodeId } from \"../hooks/use-tree\";\nimport { AnnotatedFunction } from \"../types/annotated-function\";\nimport { ChatCompletionCreateParams } from \"openai/resources/chat\";\nimport { DocumentPointer } from \"../types\";\n\n/**\n * Interface for the configuration of the Copilot API.\n */\nexport interface CopilotApiConfig {\n  /**\n   * The endpoint for the chat API.\n   */\n  chatApiEndpoint: string;\n\n  /**\n   * The endpoint for the chat API v2.\n   */\n  chatApiEndpointV2: 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   * Additional body params to be sent with the request\n   * @default {}\n   * @example\n   * ```\n   * {\n   *   'message': 'Hello, world!'\n   * }\n   * ```\n   */\n  body: Record<string, any>;\n}\n\nexport interface CopilotContextParams {\n  // function-calling\n  entryPoints: Record<string, AnnotatedFunction<any[]>>;\n  setEntryPoint: (id: string, entryPoint: AnnotatedFunction<any[]>) => void;\n  removeEntryPoint: (id: string) => void;\n  getChatCompletionFunctionDescriptions: () => ChatCompletionCreateParams.Function[];\n  getFunctionCallHandler: () => 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  // api endpoints\n  copilotApiConfig: CopilotApiConfig;\n}\n\nconst emptyCopilotContext: CopilotContextParams = {\n  entryPoints: {},\n  setEntryPoint: () => {},\n  removeEntryPoint: () => {},\n  getChatCompletionFunctionDescriptions: () => returnAndThrowInDebug([]),\n  getFunctionCallHandler: () => returnAndThrowInDebug(async () => {}),\n\n  getContextString: (documents: DocumentPointer[], categories: string[]) =>\n    returnAndThrowInDebug(\"\"),\n  addContext: () => \"\",\n  removeContext: () => {},\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(\n        \"Remember to wrap your app in a `<CopilotProvider> {...} </CopilotProvider>` !!!\",\n      );\n    }\n    get chatApiEndpointV2(): string {\n      throw new Error(\n        \"Remember to wrap your app in a `<CopilotProvider> {...} </CopilotProvider>` !!!\",\n      );\n    }\n    get headers(): Record<string, string> {\n      return {};\n    }\n    get body(): Record<string, any> {\n      return {};\n    }\n  })(),\n};\n\nexport const CopilotContext = React.createContext<CopilotContextParams>(emptyCopilotContext);\n\nfunction returnAndThrowInDebug<T>(value: T): T {\n  throw new Error(\n    \"Remember to wrap your app in a `<CopilotProvider> {...} </CopilotProvider>` !!!\",\n  );\n  return value;\n}\n"]}

package/dist/chunk-6SOQYBNX.mjs

@@ -1,44 +0,0 @@
-import { ChatCompletionTransport } from './chunk-VK25FIML.mjs';
-import { __async, __spreadValues } from './chunk-MRXNTQOX.mjs';
-
-// src/openai/chat-completion-stream.ts
-var ChatCompletionStream = class {
-  constructor(params) {
-    this.url = params.url;
-  }
-  fetch(params) {
-    return __async(this, null, function* () {
-      params = __spreadValues({}, params);
-      params.functions = void 0;
-      const transport = new ChatCompletionTransport({});
-      const cleanup = () => {
-        transport.off("data");
-        transport.off("end");
-        transport.off("error");
-      };
-      const stream = new ReadableStream({
-        start: (controller) => {
-          transport.on("data", (data) => {
-            if (data.choices[0].delta.content) {
-              controller.enqueue(data.choices[0].delta.content);
-            }
-          });
-          transport.on("error", (error) => {
-            controller.error(error);
-            cleanup();
-          });
-          transport.on("end", () => {
-            controller.close();
-            cleanup();
-          });
-        }
-      });
-      transport.fetch(params);
-      return stream;
-    });
-  }
-};
-
-export { ChatCompletionStream };
-//# sourceMappingURL=out.js.map
-//# sourceMappingURL=chunk-6SOQYBNX.mjs.map

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

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/openai/chat-completion-stream.ts"],"names":[],"mappings":";;;;;;;;;AAUO,IAAM,uBAAN,MAA2B;AAAA,EAGhC,YAAY,QAA2C;AACrD,SAAK,MAAM,OAAO;AAAA,EACpB;AAAA,EAEa,MAAM,QAA6E;AAAA;AAC9F,eAAS,mBAAK;AACd,aAAO,YAAY;AAEnB,YAAM,YAAY,IAAI,wBAAwB,CAAC,CAAC;AAEhD,YAAM,UAAU,MAAM;AACpB,kBAAU,IAAI,MAAM;AACpB,kBAAU,IAAI,KAAK;AACnB,kBAAU,IAAI,OAAO;AAAA,MACvB;AAEA,YAAM,SAAS,IAAI,eAAuB;AAAA,QACxC,OAAO,CAAC,eAAe;AACrB,oBAAU,GAAG,QAAQ,CAAC,SAAS;AAC7B,gBAAI,KAAK,QAAQ,CAAC,EAAE,MAAM,SAAS;AACjC,yBAAW,QAAQ,KAAK,QAAQ,CAAC,EAAE,MAAM,OAAO;AAAA,YAClD;AAAA,UACF,CAAC;AAED,oBAAU,GAAG,SAAS,CAAC,UAAU;AAC/B,uBAAW,MAAM,KAAK;AACtB,oBAAQ;AAAA,UACV,CAAC;AAED,oBAAU,GAAG,OAAO,MAAM;AACxB,uBAAW,MAAM;AACjB,oBAAQ;AAAA,UACV,CAAC;AAAA,QACH;AAAA,MACF,CAAC;AAED,gBAAU,MAAM,MAAM;AAEtB,aAAO;AAAA,IACT;AAAA;AACF","sourcesContent":["import {\n  ChatCompletionTransport,\n  ChatCompletionTransportFetchParams,\n} from \"./chat-completion-transport\";\n\nexport interface ChatCompletionStreamConfiguration {\n  url: string;\n  model?: string;\n}\n\nexport class ChatCompletionStream {\n  private url: string;\n\n  constructor(params: ChatCompletionStreamConfiguration) {\n    this.url = params.url;\n  }\n\n  public async fetch(params: ChatCompletionTransportFetchParams): Promise<ReadableStream<string>> {\n    params = { ...params };\n    params.functions = undefined;\n\n    const transport = new ChatCompletionTransport({});\n\n    const cleanup = () => {\n      transport.off(\"data\");\n      transport.off(\"end\");\n      transport.off(\"error\");\n    };\n\n    const stream = new ReadableStream<string>({\n      start: (controller) => {\n        transport.on(\"data\", (data) => {\n          if (data.choices[0].delta.content) {\n            controller.enqueue(data.choices[0].delta.content);\n          }\n        });\n\n        transport.on(\"error\", (error) => {\n          controller.error(error);\n          cleanup();\n        });\n\n        transport.on(\"end\", () => {\n          controller.close();\n          cleanup();\n        });\n      },\n    });\n\n    transport.fetch(params);\n\n    return stream;\n  }\n}\n"]}

package/dist/chunk-D4NQCVHE.mjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/components/copilot-provider/copilot-provider.tsx"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;AAEA,SAAS,aAAa,gBAAgB;AA0JlC;AA3GG,SAAS,gBAAgB,IAA2D;AAA3D,eAAE,WAjDlC,IAiDgC,IAAe,kBAAf,IAAe,CAAb;AAIhC,QAAM,CAAC,aAAa,cAAc,IAAI,SAAmD,CAAC,CAAC;AAE3F,QAAM,EAAE,YAAY,eAAe,UAAU,IAAI,iBAAQ;AAEzD,QAAM;AAAA,IACJ,YAAY;AAAA,IACZ,eAAe;AAAA,IACf,aAAa;AAAA,EACf,IAAI,gCAAsC;AAE1C,QAAM,gBAAgB,YAAY,CAAC,IAAY,eAAyC;AACtF,mBAAe,CAAC,eAAe;AAC7B,aAAO,iCACF,aADE;AAAA,QAEL,CAAC,EAAE,GAAG;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH,GAAG,CAAC,CAAC;AAEL,QAAM,mBAAmB,YAAY,CAAC,OAAe;AACnD,mBAAe,CAAC,eAAe;AAC7B,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,wCAAwC,YAAY,MAAM;AAC9D,WAAO,qCAAqC,OAAO,OAAO,WAAW,CAAC;AAAA,EACxE,GAAG,CAAC,WAAW,CAAC;AAEhB,QAAM,yBAAyB,YAAY,MAAM;AAC/C,WAAO,iCAAiC,OAAO,OAAO,WAAW,CAAC;AAAA,EACpE,GAAG,CAAC,WAAW,CAAC;AAEhB,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;AAGA,MAAI;AACJ,MAAI,qBAAqB,OAAO;AAC9B,uBAAmB,IAAI;AAAA,MACrB,MAAM;AAAA,MACN,MAAM,qBAAqB,GAAG,MAAM;AAAA,MACpC,CAAC;AAAA,MACD,CAAC;AAAA,IACH;AAAA,EACF,OAAO;AACL,uBAAmB,MAAM;AAAA,EAC3B;AAEA,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,MACF;AAAA,MAEC;AAAA;AAAA,EACH;AAEJ;AAEO,IAAM,kCAAkC,CAAC,QAAQ;AAExD,SAAS,iCACP,aACqB;AACrB,SAAO,CAAO,cAAc,iBAAiB;AAC3C,QAAI,4BAAsE,CAAC;AAC3E,aAAS,cAAc,aAAa;AAClC,gCAA0B,WAAW,IAAI,IAAI;AAAA,IAC/C;AAEA,UAAM,qBAAqB,0BAA0B,aAAa,QAAQ,EAAE;AAC5E,QAAI,oBAAoB;AACtB,UAAI,wBAA+C,CAAC;AACpD,UAAI,aAAa,WAAW;AAC1B,gCAAwB,KAAK,MAAM,aAAa,SAAS;AAAA,MAC3D;AAEA,YAAM,uBAA8B,CAAC;AACrC,eAAS,OAAO,mBAAmB,qBAAqB;AACtD,6BAAqB;AAAA,UACnB,sBAAsB,IAAI,IAA0C;AAAA,QACtE;AAAA,MACF;AAEA,YAAM,mBAAmB,eAAe,GAAG,oBAAoB;AAAA,IAmBjE;AAAA,EACF;AACF;AAEA,SAAS,qCACP,aACuC;AACvC,SAAO,YAAY,IAAI,yCAAyC;AAClE;AAEA,SAAS,0CACP,mBACqC;AAErC,MAAI,aAAqC,CAAC;AAC1C,WAAS,OAAO,kBAAkB,qBAAqB;AAErD,QAA2C,UAArC,QAAM,SA9OhB,IA8O+C,IAAlB,0BAAkB,IAAlB,CAAnB,QAAM;AACZ,eAAW,IAAI,IAAI,IAAI;AAAA,EACzB;AAEA,MAAI,yBAAmC,CAAC;AACxC,WAAS,OAAO,kBAAkB,qBAAqB;AACrD,QAAI,IAAI,UAAU;AAChB,6BAAuB,KAAK,IAAI,IAAI;AAAA,IACtC;AAAA,EACF;AAGA,MAAI,yBAA8D;AAAA,IAChE,MAAM,kBAAkB;AAAA,IACxB,aAAa,kBAAkB;AAAA,IAC/B,YAAY;AAAA,MACV,MAAM;AAAA,MACN,YAAY;AAAA,MACZ,UAAU;AAAA,IACZ;AAAA,EACF;AAEA,SAAO;AACT","sourcesContent":["\"use client\";\n\nimport { useCallback, useState } from \"react\";\nimport { CopilotContext, CopilotApiConfig } from \"../../context/copilot-context\";\nimport useTree from \"../../hooks/use-tree\";\nimport { AnnotatedFunction } from \"../../types/annotated-function\";\nimport { ChatCompletionCreateParams } from \"openai/resources/chat\";\nimport { DocumentPointer, FunctionCallHandler } from \"../../types\";\nimport useFlatCategoryStore from \"../../hooks/use-flat-category-store\";\nimport { StandardCopilotApiConfig } from \"./standard-copilot-api-config\";\nimport { CopilotProviderProps } from \"./copilot-provider-props\";\n\n/**\n * The CopilotProvider component.\n * This component provides the Copilot context to its children.\n * It can be configured either with a chat API endpoint or a CopilotApiConfig.\n *\n * NOTE: The backend can use OpenAI, or you can bring your own LLM.\n * For examples of the backend api implementation, see `examples/next-openai` usage (under `src/api/copilotkit`),\n * or read the documentation at https://docs.copilotkit.ai\n * In particular, Getting-Started > Quickstart-Backend: https://docs.copilotkit.ai/getting-started/quickstart-backend\n *\n * Example usage:\n * ```\n * <CopilotProvider chatApiEndpoint=\"https://your.copilotkit.api\">\n *    <App />\n * </CopilotProvider>\n * ```\n *\n * or\n *\n * ```\n * const copilotApiConfig = new StandardCopilotApiConfig(\n *  \"https://your.copilotkit.api/v1\",\n *  \"https://your.copilotkit.api/v2\",\n *  {},\n *  {}\n *  );\n *\n * // ...\n *\n * <CopilotProvider chatApiConfig={copilotApiConfig}>\n *    <App />\n * </CopilotProvider>\n * ```\n *\n * @param props - The props for the component.\n * @returns The CopilotProvider component.\n */\nexport function CopilotProvider({ children, ...props }: CopilotProviderProps): JSX.Element {\n  // Compute all the functions and properties that we need to pass\n  // to the CopilotContext.\n\n  const [entryPoints, setEntryPoints] = useState<Record<string, AnnotatedFunction<any[]>>>({});\n\n  const { addElement, removeElement, printTree } = useTree();\n\n  const {\n    addElement: addDocument,\n    removeElement: removeDocument,\n    allElements: allDocuments,\n  } = useFlatCategoryStore<DocumentPointer>();\n\n  const setEntryPoint = useCallback((id: string, entryPoint: AnnotatedFunction<any[]>) => {\n    setEntryPoints((prevPoints) => {\n      return {\n        ...prevPoints,\n        [id]: entryPoint,\n      };\n    });\n  }, []);\n\n  const removeEntryPoint = useCallback((id: string) => {\n    setEntryPoints((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 getChatCompletionFunctionDescriptions = useCallback(() => {\n    return entryPointsToChatCompletionFunctions(Object.values(entryPoints));\n  }, [entryPoints]);\n\n  const getFunctionCallHandler = useCallback(() => {\n    return entryPointsToFunctionCallHandler(Object.values(entryPoints));\n  }, [entryPoints]);\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  // get the appropriate CopilotApiConfig from the props\n  let copilotApiConfig: CopilotApiConfig;\n  if (\"chatApiEndpoint\" in props) {\n    copilotApiConfig = new StandardCopilotApiConfig(\n      props.chatApiEndpoint,\n      props.chatApiEndpointV2 || `${props.chatApiEndpoint}/v2`,\n      {},\n      {},\n    );\n  } else {\n    copilotApiConfig = props.chatApiConfig;\n  }\n\n  return (\n    <CopilotContext.Provider\n      value={{\n        entryPoints,\n        getChatCompletionFunctionDescriptions,\n        getFunctionCallHandler,\n        setEntryPoint,\n        removeEntryPoint,\n        getContextString,\n        addContext,\n        removeContext,\n        getDocumentsContext,\n        addDocumentContext,\n        removeDocumentContext,\n        copilotApiConfig: copilotApiConfig,\n      }}\n    >\n      {children}\n    </CopilotContext.Provider>\n  );\n}\n\nexport const defaultCopilotContextCategories = [\"global\"];\n\nfunction entryPointsToFunctionCallHandler(\n  entryPoints: AnnotatedFunction<any[]>[],\n): FunctionCallHandler {\n  return async (chatMessages, functionCall) => {\n    let entrypointsByFunctionName: Record<string, AnnotatedFunction<any[]>> = {};\n    for (let entryPoint of entryPoints) {\n      entrypointsByFunctionName[entryPoint.name] = entryPoint;\n    }\n\n    const entryPointFunction = entrypointsByFunctionName[functionCall.name || \"\"];\n    if (entryPointFunction) {\n      let functionCallArguments: Record<string, any>[] = [];\n      if (functionCall.arguments) {\n        functionCallArguments = JSON.parse(functionCall.arguments);\n      }\n\n      const paramsInCorrectOrder: any[] = [];\n      for (let arg of entryPointFunction.argumentAnnotations) {\n        paramsInCorrectOrder.push(\n          functionCallArguments[arg.name as keyof typeof functionCallArguments],\n        );\n      }\n\n      await entryPointFunction.implementation(...paramsInCorrectOrder);\n\n      // commented out becasue for now we don't want to return anything\n      // const result = await entryPointFunction.implementation(\n      //   ...parsedFunctionCallArguments\n      // );\n      // const functionResponse: ChatRequest = {\n      //   messages: [\n      //     ...chatMessages,\n      //     {\n      //       id: nanoid(),\n      //       name: functionCall.name,\n      //       role: 'function' as const,\n      //       content: JSON.stringify(result),\n      //     },\n      //   ],\n      // };\n\n      // return functionResponse;\n    }\n  };\n}\n\nfunction entryPointsToChatCompletionFunctions(\n  entryPoints: AnnotatedFunction<any[]>[],\n): ChatCompletionCreateParams.Function[] {\n  return entryPoints.map(annotatedFunctionToChatCompletionFunction);\n}\n\nfunction annotatedFunctionToChatCompletionFunction(\n  annotatedFunction: AnnotatedFunction<any[]>,\n): ChatCompletionCreateParams.Function {\n  // Create the parameters object based on the argumentAnnotations\n  let parameters: { [key: string]: any } = {};\n  for (let arg of annotatedFunction.argumentAnnotations) {\n    // isolate the args we should forward inline\n    let { name, required, ...forwardedArgs } = arg;\n    parameters[arg.name] = forwardedArgs;\n  }\n\n  let requiredParameterNames: string[] = [];\n  for (let arg of annotatedFunction.argumentAnnotations) {\n    if (arg.required) {\n      requiredParameterNames.push(arg.name);\n    }\n  }\n\n  // Create the ChatCompletionFunctions object\n  let chatCompletionFunction: ChatCompletionCreateParams.Function = {\n    name: annotatedFunction.name,\n    description: annotatedFunction.description,\n    parameters: {\n      type: \"object\",\n      properties: parameters,\n      required: requiredParameterNames,\n    },\n  };\n\n  return chatCompletionFunction;\n}\n"]}

package/dist/chunk-FP4EXCGS.mjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/hooks/use-make-copilot-actionable.ts"],"names":[],"mappings":";;;;;AAEA,SAAS,QAAQ,YAAY,WAAW,eAAe;AAGvD,SAAS,cAAc;AAEhB,SAAS,yBACd,mBACA,cACA;AACA,QAAM,QAAQ,OAAO,OAAO,CAAC;AAC7B,QAAM,EAAE,eAAe,iBAAiB,IAAI,WAAW,cAAc;AAErE,QAAM,4BAA4D;AAAA,IAChE,OAAO;AAAA,MACL,MAAM,kBAAkB;AAAA,MACxB,aAAa,kBAAkB;AAAA,MAC/B,qBAAqB,kBAAkB;AAAA,MACvC,gBAAgB,kBAAkB;AAAA,IACpC;AAAA,IACA;AAAA,EACF;AAEA,YAAU,MAAM;AACd,kBAAc,MAAM,SAAS,yBAAqD;AAElF,WAAO,MAAM;AACX,uBAAiB,MAAM,OAAO;AAAA,IAChC;AAAA,EACF,GAAG,CAAC,2BAA2B,eAAe,gBAAgB,CAAC;AACjE","sourcesContent":["\"use client\";\n\nimport { useRef, useContext, useEffect, useMemo } from \"react\";\nimport { CopilotContext } from \"../context/copilot-context\";\nimport { AnnotatedFunction } from \"../types/annotated-function\";\nimport { nanoid } from \"nanoid\";\n\nexport function useMakeCopilotActionable<ActionInput extends any[]>(\n  annotatedFunction: AnnotatedFunction<ActionInput>,\n  dependencies: any[],\n) {\n  const idRef = useRef(nanoid()); // generate a unique id\n  const { setEntryPoint, removeEntryPoint } = useContext(CopilotContext);\n\n  const memoizedAnnotatedFunction: AnnotatedFunction<ActionInput> = useMemo(\n    () => ({\n      name: annotatedFunction.name,\n      description: annotatedFunction.description,\n      argumentAnnotations: annotatedFunction.argumentAnnotations,\n      implementation: annotatedFunction.implementation,\n    }),\n    dependencies,\n  );\n\n  useEffect(() => {\n    setEntryPoint(idRef.current, memoizedAnnotatedFunction as AnnotatedFunction<any[]>);\n\n    return () => {\n      removeEntryPoint(idRef.current);\n    };\n  }, [memoizedAnnotatedFunction, setEntryPoint, removeEntryPoint]);\n}\n"]}

package/dist/chunk-FY7XNRJP.mjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/hooks/use-copilot-chat.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,SAAS,SAAS,kBAAkB;AAsB7B,SAAS,eAAe,IAIiB;AAJjB,eAC7B;AAAA;AAAA,IACA;AAAA,EAxBF,IAsB+B,IAG1B,oBAH0B,IAG1B;AAAA,IAFH;AAAA,IACA;AAAA;AAGA,QAAM;AAAA,IACJ;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,IAAI,WAAW,cAAc;AAE7B,QAAM,gBAAyB,QAAQ,MAAM;AAC3C,UAAM,qBAAqB,qBAAqB;AAChD,UAAM,gBAAgB,iBAAiB,CAAC,GAAG,+BAA+B;AAE1E,WAAO;AAAA,MACL,IAAI;AAAA,MACJ,SAAS,mBAAmB,eAAe,sBAAsB;AAAA,MACjE,MAAM;AAAA,IACR;AAAA,EACF,GAAG,CAAC,kBAAkB,iBAAiB,CAAC;AAExC,QAAM,uBAA8D,QAAQ,MAAM;AAChF,WAAO,sCAAsC;AAAA,EAC/C,GAAG,CAAC,qCAAqC,CAAC;AAE1C,QAAM,EAAE,UAAU,QAAQ,QAAQ,MAAM,WAAW,OAAO,SAAS,IAAI,QAAQ,iCAC1E,UAD0E;AAAA,IAE7E,eAAe;AAAA,IACf,IAAI,QAAQ;AAAA,IACZ,iBAAiB,CAAC,aAAa,EAAE,OAAO,QAAQ,mBAAmB,CAAC,CAAC;AAAA,IACrE,WAAW;AAAA,IACX,gBAAgB,uBAAuB;AAAA,IACvC,SAAS,mBAAK,QAAQ;AAAA,IACtB,MAAM,mBACD,QAAQ;AAAA,EAEf,EAAC;AAED,QAAM,kBAAkB,SAAS;AAAA,IAC/B,CAAC,YAAY,QAAQ,SAAS,UAAU,QAAQ,SAAS;AAAA,EAC3D;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;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","sourcesContent":["import { useMemo, useContext } from \"react\";\nimport { CopilotContext, CopilotContextParams } from \"../context/copilot-context\";\nimport { Message, SystemMessageFunction } from \"../types\";\nimport { UseChatOptions, useChat } from \"./use-chat\";\nimport { defaultCopilotContextCategories } from \"../components\";\nimport { ChatCompletionCreateParams } from \"openai/resources/chat\";\n\nexport interface UseCopilotChatOptions extends UseChatOptions {\n  makeSystemMessage?: SystemMessageFunction;\n  additionalInstructions?: string;\n}\n\nexport interface UseCopilotChatReturn {\n  visibleMessages: Message[];\n  append: (message: Message) => Promise<void>;\n  reload: () => Promise<void>;\n  stop: () => void;\n  isLoading: boolean;\n  input: string;\n  setInput: React.Dispatch<React.SetStateAction<string>>;\n}\n\nexport function useCopilotChat({\n  makeSystemMessage,\n  additionalInstructions,\n  ...options\n}: UseCopilotChatOptions): UseCopilotChatReturn {\n  const {\n    getContextString,\n    getChatCompletionFunctionDescriptions,\n    getFunctionCallHandler,\n    copilotApiConfig,\n  } = useContext(CopilotContext);\n\n  const systemMessage: Message = useMemo(() => {\n    const systemMessageMaker = makeSystemMessage || defaultSystemMessage;\n    const contextString = getContextString([], defaultCopilotContextCategories); // TODO: make the context categories configurable\n\n    return {\n      id: \"system\",\n      content: systemMessageMaker(contextString, additionalInstructions),\n      role: \"system\",\n    };\n  }, [getContextString, makeSystemMessage]);\n\n  const functionDescriptions: ChatCompletionCreateParams.Function[] = useMemo(() => {\n    return getChatCompletionFunctionDescriptions();\n  }, [getChatCompletionFunctionDescriptions]);\n\n  const { messages, append, reload, stop, isLoading, input, setInput } = useChat({\n    ...options,\n    copilotConfig: copilotApiConfig,\n    id: options.id,\n    initialMessages: [systemMessage].concat(options.initialMessages || []),\n    functions: functionDescriptions,\n    onFunctionCall: getFunctionCallHandler(),\n    headers: { ...options.headers },\n    body: {\n      ...options.body,\n    },\n  });\n\n  const visibleMessages = messages.filter(\n    (message) => message.role === \"user\" || message.role === \"assistant\",\n  );\n\n  return {\n    visibleMessages,\n    append,\n    reload,\n    stop,\n    isLoading,\n    input,\n    setInput,\n  };\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"]}

package/dist/chunk-LMZN4NSM.mjs

@@ -1,113 +0,0 @@
-import { ChatCompletionClient } from './chunk-TKXPMMND.mjs';
-import { __async, __spreadValues } from './chunk-MRXNTQOX.mjs';
-import { useState, useRef } from 'react';
-import { nanoid } from 'nanoid';
-
-function useChat(options) {
-  const [messages, setMessages] = useState([]);
-  const [input, setInput] = useState("");
-  const [isLoading, setIsLoading] = useState(false);
-  const abortControllerRef = useRef();
-  const runChatCompletion = (messages2) => __async(this, null, function* () {
-    return new Promise((resolve, reject) => {
-      setIsLoading(true);
-      const abortController = new AbortController();
-      abortControllerRef.current = abortController;
-      const assistantMessage = {
-        id: nanoid(),
-        createdAt: /* @__PURE__ */ new Date(),
-        content: "",
-        role: "assistant"
-      };
-      setMessages([...messages2, __spreadValues({}, assistantMessage)]);
-      const messagesWithContext = [...options.initialMessages || [], ...messages2];
-      const client = new ChatCompletionClient({});
-      const cleanup = () => {
-        client.off("content");
-        client.off("end");
-        client.off("error");
-        client.off("function");
-        abortControllerRef.current = void 0;
-      };
-      abortController.signal.addEventListener("abort", () => {
-        setIsLoading(false);
-        cleanup();
-        reject(new DOMException("Aborted", "AbortError"));
-      });
-      client.on("content", (content) => {
-        assistantMessage.content += content;
-        setMessages([...messages2, __spreadValues({}, assistantMessage)]);
-      });
-      client.on("end", () => {
-        setIsLoading(false);
-        cleanup();
-        resolve(__spreadValues({}, assistantMessage));
-      });
-      client.on("error", (error) => {
-        setIsLoading(false);
-        cleanup();
-        reject(error);
-      });
-      client.on("function", (functionCall) => __async(this, null, function* () {
-        assistantMessage.function_call = {
-          name: functionCall.name,
-          arguments: JSON.stringify(functionCall.arguments)
-        };
-        setMessages([...messages2, __spreadValues({}, assistantMessage)]);
-        setIsLoading(false);
-        cleanup();
-        resolve(__spreadValues({}, assistantMessage));
-      }));
-      client.fetch({
-        copilotConfig: options.copilotConfig,
-        messages: messagesWithContext,
-        functions: options.functions,
-        headers: options.headers,
-        signal: abortController.signal
-      });
-    });
-  });
-  const runChatCompletionAndHandleFunctionCall = (messages2) => __async(this, null, function* () {
-    const message = yield runChatCompletion(messages2);
-    if (message.function_call && options.onFunctionCall) {
-      yield options.onFunctionCall(messages2, message.function_call);
-    }
-  });
-  const append = (message) => __async(this, null, function* () {
-    if (isLoading) {
-      return;
-    }
-    const newMessages = [...messages, message];
-    setMessages(newMessages);
-    return runChatCompletionAndHandleFunctionCall(newMessages);
-  });
-  const reload = () => __async(this, null, function* () {
-    if (isLoading || messages.length === 0) {
-      return;
-    }
-    let newMessages = [...messages];
-    const lastMessage = messages[messages.length - 1];
-    if (lastMessage.role === "assistant") {
-      newMessages = newMessages.slice(0, -1);
-    }
-    setMessages(newMessages);
-    return runChatCompletionAndHandleFunctionCall(newMessages);
-  });
-  const stop = () => {
-    var _a;
-    (_a = abortControllerRef.current) == null ? void 0 : _a.abort();
-  };
-  return {
-    messages,
-    append,
-    reload,
-    stop,
-    isLoading,
-    input,
-    setInput
-  };
-}
-
-export { useChat };
-//# sourceMappingURL=out.js.map
-//# sourceMappingURL=chunk-LMZN4NSM.mjs.map

package/dist/chunk-LMZN4NSM.mjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/hooks/use-chat.ts"],"names":["messages"],"mappings":";;;;;;;;;AAAA,SAAS,QAAQ,gBAAgB;AAEjC,SAAS,cAAc;AAgFhB,SAAS,QAAQ,SAA0D;AAChF,QAAM,CAAC,UAAU,WAAW,IAAI,SAAoB,CAAC,CAAC;AACtD,QAAM,CAAC,OAAO,QAAQ,IAAI,SAAS,EAAE;AACrC,QAAM,CAAC,WAAW,YAAY,IAAI,SAAS,KAAK;AAChD,QAAM,qBAAqB,OAAwB;AAEnD,QAAM,oBAAoB,CAAOA,cAA0C;AACzE,WAAO,IAAI,QAAiB,CAAC,SAAS,WAAW;AAC/C,mBAAa,IAAI;AAEjB,YAAM,kBAAkB,IAAI,gBAAgB;AAC5C,yBAAmB,UAAU;AAE7B,YAAM,mBAA4B;AAAA,QAChC,IAAI,OAAO;AAAA,QACX,WAAW,oBAAI,KAAK;AAAA,QACpB,SAAS;AAAA,QACT,MAAM;AAAA,MACR;AAGA,kBAAY,CAAC,GAAGA,WAAU,mBAAK,iBAAkB,CAAC;AAElD,YAAM,sBAAsB,CAAC,GAAI,QAAQ,mBAAmB,CAAC,GAAI,GAAGA,SAAQ;AAE5E,YAAM,SAAS,IAAI,qBAAqB,CAAC,CAAC;AAE1C,YAAM,UAAU,MAAM;AACpB,eAAO,IAAI,SAAS;AACpB,eAAO,IAAI,KAAK;AAChB,eAAO,IAAI,OAAO;AAClB,eAAO,IAAI,UAAU;AAErB,2BAAmB,UAAU;AAAA,MAC/B;AAEA,sBAAgB,OAAO,iBAAiB,SAAS,MAAM;AACrD,qBAAa,KAAK;AAClB,gBAAQ;AACR,eAAO,IAAI,aAAa,WAAW,YAAY,CAAC;AAAA,MAClD,CAAC;AAED,aAAO,GAAG,WAAW,CAAC,YAAY;AAChC,yBAAiB,WAAW;AAC5B,oBAAY,CAAC,GAAGA,WAAU,mBAAK,iBAAkB,CAAC;AAAA,MACpD,CAAC;AAED,aAAO,GAAG,OAAO,MAAM;AACrB,qBAAa,KAAK;AAClB,gBAAQ;AACR,gBAAQ,mBAAK,iBAAkB;AAAA,MACjC,CAAC;AAED,aAAO,GAAG,SAAS,CAAC,UAAU;AAC5B,qBAAa,KAAK;AAClB,gBAAQ;AACR,eAAO,KAAK;AAAA,MACd,CAAC;AAED,aAAO,GAAG,YAAY,CAAO,iBAAiB;AAC5C,yBAAiB,gBAAgB;AAAA,UAC/B,MAAM,aAAa;AAAA,UACnB,WAAW,KAAK,UAAU,aAAa,SAAS;AAAA,QAClD;AACA,oBAAY,CAAC,GAAGA,WAAU,mBAAK,iBAAkB,CAAC;AAElD,qBAAa,KAAK;AAClB,gBAAQ;AACR,gBAAQ,mBAAK,iBAAkB;AAAA,MACjC,EAAC;AAED,aAAO,MAAM;AAAA,QACX,eAAe,QAAQ;AAAA,QACvB,UAAU;AAAA,QACV,WAAW,QAAQ;AAAA,QACnB,SAAS,QAAQ;AAAA,QACjB,QAAQ,gBAAgB;AAAA,MAC1B,CAAC;AAAA,IACH,CAAC;AAAA,EACH;AAEA,QAAM,yCAAyC,CAAOA,cAAuC;AAC3F,UAAM,UAAU,MAAM,kBAAkBA,SAAQ;AAChD,QAAI,QAAQ,iBAAiB,QAAQ,gBAAgB;AACnD,YAAM,QAAQ,eAAeA,WAAU,QAAQ,aAAa;AAAA,IAC9D;AAAA,EACF;AAEA,QAAM,SAAS,CAAO,YAAoC;AACxD,QAAI,WAAW;AACb;AAAA,IACF;AACA,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,YAAY,SAAS,aAAa;AACpC,oBAAc,YAAY,MAAM,GAAG,EAAE;AAAA,IACvC;AACA,gBAAY,WAAW;AAEvB,WAAO,uCAAuC,WAAW;AAAA,EAC3D;AAEA,QAAM,OAAO,MAAY;AAlM3B;AAmMI,6BAAmB,YAAnB,mBAA4B;AAAA,EAC9B;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF","sourcesContent":["import { useRef, useState } from \"react\";\nimport { Message, Function, FunctionCallHandler } from \"../types\";\nimport { nanoid } from \"nanoid\";\nimport { ChatCompletionClient } from \"../openai/chat-completion-client\";\nimport { CopilotApiConfig } from \"../context\";\n\nexport type UseChatOptions = {\n  /**\n   * The API endpoint that accepts a `{ messages: Message[] }` object and returns\n   * a stream of tokens of the AI chat response. Defaults to `/api/chat`.\n   */\n  api?: string;\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   * 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   * HTTP headers to be sent with the API request.\n   */\n  headers?: Record<string, string> | Headers;\n  /**\n   * Extra body object to be sent with the API request.\n   * @example\n   * Send a `sessionId` to the API along with the messages.\n   * ```js\n   * useChat({\n   *   body: {\n   *     sessionId: '123',\n   *   }\n   * })\n   * ```\n   */\n  body?: object;\n  /**\n   * Function definitions to be sent to the API.\n   */\n  functions?: Function[];\n};\n\nexport type UseChatHelpers = {\n  /** Current messages in the chat */\n  messages: Message[];\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  /** The current value of the input */\n  input: string;\n  /** setState-powered method to update the input value */\n  setInput: React.Dispatch<React.SetStateAction<string>>;\n  /** Whether the API request is in progress */\n  isLoading: boolean;\n};\n\nexport type UseChatOptionsWithCopilotConfig = UseChatOptions & {\n  copilotConfig: CopilotApiConfig;\n};\n\nexport function useChat(options: UseChatOptionsWithCopilotConfig): UseChatHelpers {\n  const [messages, setMessages] = useState<Message[]>([]);\n  const [input, setInput] = useState(\"\");\n  const [isLoading, setIsLoading] = useState(false);\n  const abortControllerRef = useRef<AbortController>();\n\n  const runChatCompletion = async (messages: Message[]): Promise<Message> => {\n    return new Promise<Message>((resolve, reject) => {\n      setIsLoading(true);\n\n      const abortController = new AbortController();\n      abortControllerRef.current = abortController;\n\n      const assistantMessage: Message = {\n        id: nanoid(),\n        createdAt: new Date(),\n        content: \"\",\n        role: \"assistant\",\n      };\n\n      // Assistant messages are always copied when using setState\n      setMessages([...messages, { ...assistantMessage }]);\n\n      const messagesWithContext = [...(options.initialMessages || []), ...messages];\n\n      const client = new ChatCompletionClient({});\n\n      const cleanup = () => {\n        client.off(\"content\");\n        client.off(\"end\");\n        client.off(\"error\");\n        client.off(\"function\");\n\n        abortControllerRef.current = undefined;\n      };\n\n      abortController.signal.addEventListener(\"abort\", () => {\n        setIsLoading(false);\n        cleanup();\n        reject(new DOMException(\"Aborted\", \"AbortError\"));\n      });\n\n      client.on(\"content\", (content) => {\n        assistantMessage.content += content;\n        setMessages([...messages, { ...assistantMessage }]);\n      });\n\n      client.on(\"end\", () => {\n        setIsLoading(false);\n        cleanup();\n        resolve({ ...assistantMessage });\n      });\n\n      client.on(\"error\", (error) => {\n        setIsLoading(false);\n        cleanup();\n        reject(error);\n      });\n\n      client.on(\"function\", async (functionCall) => {\n        assistantMessage.function_call = {\n          name: functionCall.name,\n          arguments: JSON.stringify(functionCall.arguments),\n        };\n        setMessages([...messages, { ...assistantMessage }]);\n        // quit early if we get a function call\n        setIsLoading(false);\n        cleanup();\n        resolve({ ...assistantMessage });\n      });\n\n      client.fetch({\n        copilotConfig: options.copilotConfig,\n        messages: messagesWithContext,\n        functions: options.functions,\n        headers: options.headers,\n        signal: abortController.signal,\n      });\n    });\n  };\n\n  const runChatCompletionAndHandleFunctionCall = async (messages: Message[]): Promise<void> => {\n    const message = await runChatCompletion(messages);\n    if (message.function_call && options.onFunctionCall) {\n      await options.onFunctionCall(messages, message.function_call);\n    }\n  };\n\n  const append = async (message: Message): Promise<void> => {\n    if (isLoading) {\n      return;\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.role === \"assistant\") {\n      newMessages = newMessages.slice(0, -1);\n    }\n    setMessages(newMessages);\n\n    return runChatCompletionAndHandleFunctionCall(newMessages);\n  };\n\n  const stop = (): void => {\n    abortControllerRef.current?.abort();\n  };\n\n  return {\n    messages,\n    append,\n    reload,\n    stop,\n    isLoading,\n    input,\n    setInput,\n  };\n}\n"]}

package/dist/chunk-PUFR5OAI.mjs

@@ -1,3 +0,0 @@
-
-//# sourceMappingURL=out.js.map
-//# sourceMappingURL=chunk-PUFR5OAI.mjs.map