@copilotkit/react-core 0.37.0 → 0.38.0-beta.0

package/src/lib/copilot-task.ts

@@ -89,11 +89,20 @@
  * ```
  */
 
-import { FunctionCall, Message } from "@copilotkit/shared";
+import {
+  ActionExecutionMessage,
+  CopilotRuntimeClient,
+  Message,
+  Role,
+  TextMessage,
+  convertGqlOutputToMessages,
+  convertMessagesToGqlInput,
+} from "@copilotkit/runtime-client-gql";
 import { FrontendAction } from "../types/frontend-action";
 import { CopilotContextParams } from "../context";
 import { defaultCopilotContextCategories } from "../components";
-import { fetchAndDecodeChatCompletion } from "../utils/fetch-chat-completion";
+import { MessageStatusCode } from "@copilotkit/runtime-client-gql";
+import { actionParametersToJsonSchema } from "@copilotkit/shared";
 
 export interface CopilotTaskConfig {
   /**
@@ -141,11 +150,11 @@ export class CopilotTask<T = any> {
    * @param data The data to use for the task.
    */
   async run(context: CopilotContextParams, data?: T): Promise<void> {
-    const entryPoints = this.includeCopilotActions ? Object.assign({}, context.entryPoints) : {};
+    const actions = this.includeCopilotActions ? Object.assign({}, context.actions) : {};
 
     // merge functions into entry points
     for (const fn of this.actions) {
-      entryPoints[fn.name] = fn;
+      actions[fn.name] = fn;
     }
 
     let contextString = "";
@@ -158,52 +167,46 @@ export class CopilotTask<T = any> {
       contextString += context.getContextString([], defaultCopilotContextCategories);
     }
 
-    const systemMessage: Message = {
-      id: "system",
+    const systemMessage = new TextMessage({
       content: taskSystemMessage(contextString, this.instructions),
-      role: "system",
-    };
+      role: Role.System,
+    });
 
-    const messages = [systemMessage];
+    const messages: Message[] = [systemMessage];
 
-    const response = await fetchAndDecodeChatCompletion({
-      copilotConfig: context.copilotApiConfig,
-      messages: messages,
-      tools: context.getChatCompletionFunctionDescriptions(entryPoints),
+    const runtimeClient = new CopilotRuntimeClient({
+      url: context.copilotApiConfig.chatApiEndpoint,
+      publicApiKey: context.copilotApiConfig.publicApiKey,
       headers: context.copilotApiConfig.headers,
-      body: context.copilotApiConfig.body,
     });
 
-    if (!response.events) {
-      throw new Error("Failed to execute task");
-    }
-
-    const reader = response.events.getReader();
-    let functionCalls: FunctionCall[] = [];
-
-    while (true) {
-      const { done, value } = await reader.read();
-
-      if (done) {
-        break;
-      }
-
-      if (value.type === "function") {
-        functionCalls.push({
-          name: value.name,
-          arguments: JSON.stringify(value.arguments),
-        });
-        break;
-      }
-    }
-
-    if (!functionCalls.length) {
-      throw new Error("No function call occurred");
-    }
+    const response = await runtimeClient
+      .generateCopilotResponse(
+        {
+          frontend: {
+            actions: Object.values(actions).map((action) => ({
+              name: action.name,
+              description: action.description || "",
+              jsonSchema: JSON.stringify(actionParametersToJsonSchema(action.parameters || [])),
+            })),
+          },
+          messages: convertMessagesToGqlInput(messages),
+        },
+        context.copilotApiConfig.properties,
+      )
+      .toPromise();
+
+    const functionCallHandler = context.getFunctionCallHandler(actions);
+    const functionCalls = convertGqlOutputToMessages(
+      response.data?.generateCopilotResponse?.messages || [],
+    ).filter((m): m is ActionExecutionMessage => m instanceof ActionExecutionMessage);
 
-    const functionCallHandler = context.getFunctionCallHandler(entryPoints);
     for (const functionCall of functionCalls) {
-      await functionCallHandler(messages, functionCall);
+      await functionCallHandler({
+        messages,
+        name: functionCall.name,
+        args: functionCall.arguments,
+      });
     }
   }
 }

package/src/utils/extract.ts

@@ -2,13 +2,20 @@ import {
   Action,
   COPILOT_CLOUD_PUBLIC_API_KEY_HEADER,
   MappedParameterTypes,
-  Message,
   Parameter,
+  actionParametersToJsonSchema,
 } from "@copilotkit/shared";
+import {
+  ActionExecutionMessage,
+  Message,
+  Role,
+  TextMessage,
+  convertGqlOutputToMessages,
+} from "@copilotkit/runtime-client-gql";
 import { CopilotContextParams } from "../context";
 import { defaultCopilotContextCategories } from "../components";
-import { fetchAndDecodeChatCompletion } from "./fetch-chat-completion";
-import untruncateJson from "untruncate-json";
+import { CopilotRuntimeClient } from "@copilotkit/runtime-client-gql";
+import { convertMessagesToGqlInput } from "@copilotkit/runtime-client-gql";
 
 interface InitialState<T extends Parameter[] | [] = []> {
   status: "initial";
@@ -75,11 +82,10 @@ export async function extract<const T extends Parameter[]>({
     contextString += context.getContextString([], defaultCopilotContextCategories);
   }
 
-  const systemMessage: Message = {
-    id: "system",
+  const systemMessage: Message = new TextMessage({
     content: makeSystemMessage(contextString, instructions),
-    role: "system",
-  };
+    role: Role.System,
+  });
 
   const headers = {
     ...(context.copilotApiConfig.headers || {}),
@@ -88,23 +94,40 @@ export async function extract<const T extends Parameter[]>({
       : {}),
   };
 
-  const response = await fetchAndDecodeChatCompletion({
-    copilotConfig: context.copilotApiConfig,
-    messages: includeMessages ? [systemMessage, ...messages] : [systemMessage],
-    tools: context.getChatCompletionFunctionDescriptions({ extract: action }),
+  const runtimeClient = new CopilotRuntimeClient({
+    url: context.copilotApiConfig.chatApiEndpoint,
+    publicApiKey: context.copilotApiConfig.publicApiKey,
     headers,
-    body: context.copilotApiConfig.body,
-    toolChoice: { type: "function", function: { name: "extract" } },
-    signal: abortSignal,
   });
 
-  if (!response.events) {
-    throw new Error("extract() failed: Could not fetch chat completion");
-  }
+  const response = CopilotRuntimeClient.asStream(
+    runtimeClient.generateCopilotResponse(
+      {
+        frontend: {
+          actions: [
+            {
+              name: action.name,
+              description: action.description || "",
+              jsonSchema: JSON.stringify(actionParametersToJsonSchema(action.parameters || [])),
+            },
+          ],
+        },
+
+        messages: convertMessagesToGqlInput(
+          includeMessages ? [systemMessage, ...messages] : [systemMessage],
+        ),
+      },
+      context.copilotApiConfig.properties,
+      abortSignal,
+    ),
+  );
+
+  const reader = response.getReader();
 
-  const reader = response.events.getReader();
   let isInitial = true;
 
+  let actionExecutionMessage: ActionExecutionMessage | undefined = undefined;
+
   while (true) {
     const { done, value } = await reader.read();
 
@@ -112,27 +135,32 @@ export async function extract<const T extends Parameter[]>({
       break;
     }
 
-    if (value.type === "partial") {
-      try {
-        let partialArguments = JSON.parse(untruncateJson(value.arguments));
-        stream?.({
-          status: isInitial ? "initial" : "inProgress",
-          args: partialArguments as Partial<MappedParameterTypes<T>>,
-        });
-        isInitial = false;
-      } catch (e) {}
-    }
+    actionExecutionMessage = convertGqlOutputToMessages(
+      value.generateCopilotResponse.messages,
+    ).find((msg) => msg instanceof ActionExecutionMessage) as ActionExecutionMessage | undefined;
 
-    if (value.type === "function") {
-      stream?.({
-        status: "complete",
-        args: value.arguments as MappedParameterTypes<T>,
-      });
-      return value.arguments as MappedParameterTypes<T>;
+    if (!actionExecutionMessage) {
+      continue;
     }
+
+    stream?.({
+      status: isInitial ? "initial" : "inProgress",
+      args: actionExecutionMessage.arguments as Partial<MappedParameterTypes<T>>,
+    });
+
+    isInitial = false;
   }
 
-  throw new Error("extract() failed: No function call occurred");
+  if (!actionExecutionMessage) {
+    throw new Error("extract() failed: No function call occurred");
+  }
+
+  stream?.({
+    status: "complete",
+    args: actionExecutionMessage.arguments as MappedParameterTypes<T>,
+  });
+
+  return actionExecutionMessage.arguments as MappedParameterTypes<T>;
 }
 
 function makeSystemMessage(contextString: string, instructions: string): string {
@@ -147,7 +175,7 @@ The user has provided you with the following context:
 ${contextString}
 \`\`\`
 
-They have also provided you with a function you MUST call to initiate actions on their behalf.
+They have also provided you with a function called extract you MUST call to initiate actions on their behalf.
 
 Please assist them as best you can.
 

package/src/utils/index.ts

@@ -1,8 +1 @@
-export {
-  type FetchChatCompletionParams,
-  fetchChatCompletion,
-  fetchAndDecodeChatCompletion,
-  fetchAndDecodeChatCompletionAsText,
-} from "./fetch-chat-completion";
-
 export { extract } from "./extract";

package/dist/chunk-2EQGN5QK.mjs

@@ -1,137 +0,0 @@
-import {
-  processMessageStream
-} from "./chunk-ZFS5SQUT.mjs";
-import {
-  defaultCopilotContextCategories
-} from "./chunk-LLU5URI5.mjs";
-import {
-  CopilotContext
-} from "./chunk-DY63PD22.mjs";
-import {
-  __async,
-  __spreadValues
-} from "./chunk-SKC7AJIV.mjs";
-
-// src/openai-assistants/hooks/use-copilot-chat-v2.ts
-import { useContext, useMemo, useState } from "react";
-import { parseStreamPart } from "@copilotkit/shared";
-function useCopilotChatV2(options) {
-  const {
-    getContextString,
-    getChatCompletionFunctionDescriptions,
-    getFunctionCallHandler,
-    copilotApiConfig
-  } = useContext(CopilotContext);
-  const [messages, setMessages] = useState([]);
-  const [input, setInput] = useState("");
-  const [threadId, setThreadId] = useState(void 0);
-  const [status, setStatus] = useState("awaiting_message");
-  const [error, setError] = useState(void 0);
-  const systemMessage = useMemo(() => {
-    const systemMessageMaker = options.makeSystemMessage || defaultSystemMessage;
-    const contextString = getContextString([], defaultCopilotContextCategories);
-    return {
-      id: "system",
-      content: systemMessageMaker(contextString),
-      role: "system"
-    };
-  }, [getContextString, options.makeSystemMessage]);
-  const handleInputChange = (e) => {
-    setInput(e.target.value);
-  };
-  const submitMessage = (e) => __async(this, null, function* () {
-    var _a, _b;
-    e.preventDefault();
-    if (input === "") {
-      return;
-    }
-    setStatus("in_progress");
-    setMessages((messages2) => [...messages2, { id: "", role: "user", content: input }]);
-    setInput("");
-    const apiUrl = copilotApiConfig.chatApiEndpointV2;
-    const functions = getChatCompletionFunctionDescriptions();
-    const result = yield fetch(apiUrl, {
-      method: "POST",
-      headers: __spreadValues(__spreadValues({
-        "Content-Type": "application/json"
-      }, copilotApiConfig.headers), options.headers),
-      body: JSON.stringify(__spreadValues(__spreadValues(__spreadValues({
-        // always use user-provided threadId when available:
-        threadId: (_b = (_a = options.threadId) != null ? _a : threadId) != null ? _b : null,
-        message: input
-      }, functions.length > 0 && { functions }), copilotApiConfig.body), options.body))
-    });
-    if (result.body == null) {
-      throw new Error("The response body is empty.");
-    }
-    yield processMessageStream(result.body.getReader(), (message) => {
-      try {
-        const { type, value } = parseStreamPart(message);
-        switch (type) {
-          case "assistant_message": {
-            setMessages((messages2) => [
-              ...messages2,
-              {
-                id: value.id,
-                role: value.role,
-                content: value.content[0].text.value
-              }
-            ]);
-            break;
-          }
-          case "assistant_control_data": {
-            setThreadId(value.threadId);
-            setMessages((messages2) => {
-              const lastMessage = messages2[messages2.length - 1];
-              lastMessage.id = value.messageId;
-              return [...messages2.slice(0, messages2.length - 1), lastMessage];
-            });
-            break;
-          }
-          case "error": {
-            setError(value);
-            break;
-          }
-        }
-      } catch (error2) {
-        setError(error2);
-      }
-    });
-    setStatus("awaiting_message");
-  });
-  return {
-    messages,
-    input,
-    handleInputChange,
-    submitMessage,
-    status,
-    error
-  };
-}
-function defaultSystemMessage(contextString) {
-  return `
-Please act as an efficient, competent, conscientious, and industrious professional assistant.
-
-Help 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.
-Always be polite and respectful, and prefer brevity over verbosity.
-
-The user has provided you with the following context:
-\`\`\`
-${contextString}
-\`\`\`
-
-They have also provided you with functions you can call to initiate actions on their behalf, or functions you can call to receive more information.
-
-Please assist them as best you can.
-
-You 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.
-
-If you would like to call a function, call it without saying anything else.
-`;
-}
-
-export {
-  useCopilotChatV2,
-  defaultSystemMessage
-};
-//# sourceMappingURL=chunk-2EQGN5QK.mjs.map

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

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/openai-assistants/hooks/use-copilot-chat-v2.ts"],"sourcesContent":["import { useContext, useMemo, useState } from \"react\";\nimport { processMessageStream } from \"../utils\";\nimport { Message, parseStreamPart } from \"@copilotkit/shared\";\nimport { CopilotContext } from \"../../context\";\nimport { defaultCopilotContextCategories } from \"../../components\";\n\nexport type AssistantStatus = \"in_progress\" | \"awaiting_message\";\n\nexport interface RequestForwardingOptions {\n  /**\n   * The credentials mode to be used for the fetch request.\n   * Possible values are: 'omit', 'same-origin', 'include'.\n   * Defaults to 'same-origin'.\n   */\n  credentials?: RequestCredentials;\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}\nexport interface UseCopilotChatOptionsV2 extends RequestForwardingOptions {\n  makeSystemMessage?: (contextString: string) => string;\n  threadId?: string | undefined;\n}\n\nexport interface UseCopilotChatV2Result {\n  messages: Message[];\n  input: string;\n  handleInputChange: (e: any) => void;\n  submitMessage: (e: any) => Promise<void>;\n  status: AssistantStatus;\n  error: unknown;\n}\n\nexport function useCopilotChatV2(options: UseCopilotChatOptionsV2): UseCopilotChatV2Result {\n  const {\n    getContextString,\n    getChatCompletionFunctionDescriptions,\n    getFunctionCallHandler,\n    copilotApiConfig,\n  } = useContext(CopilotContext);\n\n  const [messages, setMessages] = useState<Message[]>([]);\n  const [input, setInput] = useState(\"\");\n  const [threadId, setThreadId] = useState<string | undefined>(undefined);\n  const [status, setStatus] = useState<AssistantStatus>(\"awaiting_message\");\n  const [error, setError] = useState<unknown | undefined>(undefined);\n\n  const systemMessage: Message = useMemo(() => {\n    const systemMessageMaker = options.makeSystemMessage || defaultSystemMessage;\n    const contextString = getContextString([], defaultCopilotContextCategories); // TODO: make the context categories configurable\n\n    return {\n      id: \"system\",\n      content: systemMessageMaker(contextString),\n      role: \"system\",\n    };\n  }, [getContextString, options.makeSystemMessage]);\n\n  const handleInputChange = (e: any) => {\n    setInput(e.target.value);\n  };\n\n  const submitMessage = async (e: any) => {\n    e.preventDefault();\n\n    if (input === \"\") {\n      return;\n    }\n\n    setStatus(\"in_progress\");\n\n    setMessages((messages) => [...messages, { id: \"\", role: \"user\", content: input }]);\n\n    setInput(\"\");\n\n    const apiUrl = copilotApiConfig.chatApiEndpointV2;\n\n    const functions = getChatCompletionFunctionDescriptions();\n\n    const result = await fetch(apiUrl, {\n      method: \"POST\",\n      headers: {\n        \"Content-Type\": \"application/json\",\n        ...copilotApiConfig.headers,\n        ...options.headers,\n      },\n      body: JSON.stringify({\n        // always use user-provided threadId when available:\n        threadId: options.threadId ?? threadId ?? null,\n        message: input,\n        ...(functions.length > 0 && { functions: functions }),\n        ...copilotApiConfig.body,\n        ...options.body,\n      }),\n    });\n\n    if (result.body == null) {\n      throw new Error(\"The response body is empty.\");\n    }\n\n    await processMessageStream(result.body.getReader(), (message: string) => {\n      try {\n        const { type, value } = parseStreamPart(message);\n\n        switch (type) {\n          case \"assistant_message\": {\n            // append message:\n            setMessages((messages) => [\n              ...messages,\n              {\n                id: value.id,\n                role: value.role,\n                content: value.content[0].text.value,\n              },\n            ]);\n            break;\n          }\n\n          case \"assistant_control_data\": {\n            setThreadId(value.threadId);\n\n            // set id of last message:\n            setMessages((messages) => {\n              const lastMessage = messages[messages.length - 1];\n              lastMessage.id = value.messageId;\n              return [...messages.slice(0, messages.length - 1), lastMessage];\n            });\n\n            break;\n          }\n\n          case \"error\": {\n            setError(value);\n            break;\n          }\n        }\n      } catch (error) {\n        setError(error);\n      }\n    });\n\n    setStatus(\"awaiting_message\");\n  };\n\n  return {\n    messages,\n    input,\n    handleInputChange,\n    submitMessage,\n    status,\n    error,\n  };\n}\n\nexport function defaultSystemMessage(contextString: string): string {\n  return `\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`;\n}\n"],"mappings":";;;;;;;;;;;;;;;AAAA,SAAS,YAAY,SAAS,gBAAgB;AAE9C,SAAkB,uBAAuB;AA6ClC,SAAS,iBAAiB,SAA0D;AACzF,QAAM;AAAA,IACJ;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,IAAI,WAAW,cAAc;AAE7B,QAAM,CAAC,UAAU,WAAW,IAAI,SAAoB,CAAC,CAAC;AACtD,QAAM,CAAC,OAAO,QAAQ,IAAI,SAAS,EAAE;AACrC,QAAM,CAAC,UAAU,WAAW,IAAI,SAA6B,MAAS;AACtE,QAAM,CAAC,QAAQ,SAAS,IAAI,SAA0B,kBAAkB;AACxE,QAAM,CAAC,OAAO,QAAQ,IAAI,SAA8B,MAAS;AAEjE,QAAM,gBAAyB,QAAQ,MAAM;AAC3C,UAAM,qBAAqB,QAAQ,qBAAqB;AACxD,UAAM,gBAAgB,iBAAiB,CAAC,GAAG,+BAA+B;AAE1E,WAAO;AAAA,MACL,IAAI;AAAA,MACJ,SAAS,mBAAmB,aAAa;AAAA,MACzC,MAAM;AAAA,IACR;AAAA,EACF,GAAG,CAAC,kBAAkB,QAAQ,iBAAiB,CAAC;AAEhD,QAAM,oBAAoB,CAAC,MAAW;AACpC,aAAS,EAAE,OAAO,KAAK;AAAA,EACzB;AAEA,QAAM,gBAAgB,CAAO,MAAW;AA5E1C;AA6EI,MAAE,eAAe;AAEjB,QAAI,UAAU,IAAI;AAChB;AAAA,IACF;AAEA,cAAU,aAAa;AAEvB,gBAAY,CAACA,cAAa,CAAC,GAAGA,WAAU,EAAE,IAAI,IAAI,MAAM,QAAQ,SAAS,MAAM,CAAC,CAAC;AAEjF,aAAS,EAAE;AAEX,UAAM,SAAS,iBAAiB;AAEhC,UAAM,YAAY,sCAAsC;AAExD,UAAM,SAAS,MAAM,MAAM,QAAQ;AAAA,MACjC,QAAQ;AAAA,MACR,SAAS;AAAA,QACP,gBAAgB;AAAA,SACb,iBAAiB,UACjB,QAAQ;AAAA,MAEb,MAAM,KAAK,UAAU;AAAA;AAAA,QAEnB,WAAU,mBAAQ,aAAR,YAAoB,aAApB,YAAgC;AAAA,QAC1C,SAAS;AAAA,SACL,UAAU,SAAS,KAAK,EAAE,UAAqB,IAChD,iBAAiB,OACjB,QAAQ,KACZ;AAAA,IACH,CAAC;AAED,QAAI,OAAO,QAAQ,MAAM;AACvB,YAAM,IAAI,MAAM,6BAA6B;AAAA,IAC/C;AAEA,UAAM,qBAAqB,OAAO,KAAK,UAAU,GAAG,CAAC,YAAoB;AACvE,UAAI;AACF,cAAM,EAAE,MAAM,MAAM,IAAI,gBAAgB,OAAO;AAE/C,gBAAQ,MAAM;AAAA,UACZ,KAAK,qBAAqB;AAExB,wBAAY,CAACA,cAAa;AAAA,cACxB,GAAGA;AAAA,cACH;AAAA,gBACE,IAAI,MAAM;AAAA,gBACV,MAAM,MAAM;AAAA,gBACZ,SAAS,MAAM,QAAQ,CAAC,EAAE,KAAK;AAAA,cACjC;AAAA,YACF,CAAC;AACD;AAAA,UACF;AAAA,UAEA,KAAK,0BAA0B;AAC7B,wBAAY,MAAM,QAAQ;AAG1B,wBAAY,CAACA,cAAa;AACxB,oBAAM,cAAcA,UAASA,UAAS,SAAS,CAAC;AAChD,0BAAY,KAAK,MAAM;AACvB,qBAAO,CAAC,GAAGA,UAAS,MAAM,GAAGA,UAAS,SAAS,CAAC,GAAG,WAAW;AAAA,YAChE,CAAC;AAED;AAAA,UACF;AAAA,UAEA,KAAK,SAAS;AACZ,qBAAS,KAAK;AACd;AAAA,UACF;AAAA,QACF;AAAA,MACF,SAASC,QAAP;AACA,iBAASA,MAAK;AAAA,MAChB;AAAA,IACF,CAAC;AAED,cAAU,kBAAkB;AAAA,EAC9B;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEO,SAAS,qBAAqB,eAA+B;AAClE,SAAO;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQP;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAWF;","names":["messages","error"]}

package/dist/chunk-4OIVQMEM.mjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/utils/extract.ts"],"sourcesContent":["import {\n  Action,\n  COPILOT_CLOUD_PUBLIC_API_KEY_HEADER,\n  MappedParameterTypes,\n  Message,\n  Parameter,\n} from \"@copilotkit/shared\";\nimport { CopilotContextParams } from \"../context\";\nimport { defaultCopilotContextCategories } from \"../components\";\nimport { fetchAndDecodeChatCompletion } from \"./fetch-chat-completion\";\nimport untruncateJson from \"untruncate-json\";\n\ninterface InitialState<T extends Parameter[] | [] = []> {\n  status: \"initial\";\n  args: Partial<MappedParameterTypes<T>>;\n}\n\ninterface InProgressState<T extends Parameter[] | [] = []> {\n  status: \"inProgress\";\n  args: Partial<MappedParameterTypes<T>>;\n}\n\ninterface CompleteState<T extends Parameter[] | [] = []> {\n  status: \"complete\";\n  args: MappedParameterTypes<T>;\n}\n\ntype StreamHandlerArgs<T extends Parameter[] | [] = []> =\n  | InitialState<T>\n  | InProgressState<T>\n  | CompleteState<T>;\n\ninterface ExtractOptions<T extends Parameter[]> {\n  context: CopilotContextParams;\n  instructions: string;\n  parameters: T;\n  include?: IncludeOptions;\n  data?: any;\n  abortSignal?: AbortSignal;\n  stream?: (args: StreamHandlerArgs<T>) => void;\n}\n\ninterface IncludeOptions {\n  readable?: boolean;\n  messages?: boolean;\n}\n\nexport async function extract<const T extends Parameter[]>({\n  context,\n  instructions,\n  parameters,\n  include,\n  data,\n  abortSignal,\n  stream,\n}: ExtractOptions<T>): Promise<MappedParameterTypes<T>> {\n  const { messages } = context;\n\n  const action: Action<any> = {\n    name: \"extract\",\n    parameters,\n    handler: (args: any) => {},\n  };\n\n  const includeReadable = include?.readable ?? false;\n  const includeMessages = include?.messages ?? false;\n\n  let contextString = \"\";\n\n  if (data) {\n    contextString = (typeof data === \"string\" ? data : JSON.stringify(data)) + \"\\n\\n\";\n  }\n\n  if (includeReadable) {\n    contextString += context.getContextString([], defaultCopilotContextCategories);\n  }\n\n  const systemMessage: Message = {\n    id: \"system\",\n    content: makeSystemMessage(contextString, instructions),\n    role: \"system\",\n  };\n\n  const headers = {\n    ...(context.copilotApiConfig.headers || {}),\n    ...(context.copilotApiConfig.publicApiKey\n      ? { [COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: context.copilotApiConfig.publicApiKey }\n      : {}),\n  };\n\n  const response = await fetchAndDecodeChatCompletion({\n    copilotConfig: context.copilotApiConfig,\n    messages: includeMessages ? [systemMessage, ...messages] : [systemMessage],\n    tools: context.getChatCompletionFunctionDescriptions({ extract: action }),\n    headers,\n    body: context.copilotApiConfig.body,\n    toolChoice: { type: \"function\", function: { name: \"extract\" } },\n    signal: abortSignal,\n  });\n\n  if (!response.events) {\n    throw new Error(\"extract() failed: Could not fetch chat completion\");\n  }\n\n  const reader = response.events.getReader();\n  let isInitial = true;\n\n  while (true) {\n    const { done, value } = await reader.read();\n\n    if (done) {\n      break;\n    }\n\n    if (value.type === \"partial\") {\n      try {\n        let partialArguments = JSON.parse(untruncateJson(value.arguments));\n        stream?.({\n          status: isInitial ? \"initial\" : \"inProgress\",\n          args: partialArguments as Partial<MappedParameterTypes<T>>,\n        });\n        isInitial = false;\n      } catch (e) {}\n    }\n\n    if (value.type === \"function\") {\n      stream?.({\n        status: \"complete\",\n        args: value.arguments as MappedParameterTypes<T>,\n      });\n      return value.arguments as MappedParameterTypes<T>;\n    }\n  }\n\n  throw new Error(\"extract() failed: No function call occurred\");\n}\n\nfunction makeSystemMessage(contextString: string, instructions: string): string {\n  return `\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 a function you MUST call to initiate actions on their behalf.\n\nPlease assist them as best you can.\n\nThis is not a conversation, so please do not ask questions. Just call the function without saying anything else.\n\nThe user has given you the following task to complete:\n\n\\`\\`\\`\n${instructions}\n\\`\\`\\`\n\nAny additional messages provided are for providing context only and should not be used to ask questions or engage in conversation.\n`;\n}\n"],"mappings":";;;;;;;;;;;;AAAA;AAAA,EAEE;AAAA,OAIK;AAIP,OAAO,oBAAoB;AAqC3B,SAAsB,QAAqC,IAQH;AAAA,6CARG;AAAA,IACzD;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAAwD;AAvDxD;AAwDE,UAAM,EAAE,SAAS,IAAI;AAErB,UAAM,SAAsB;AAAA,MAC1B,MAAM;AAAA,MACN;AAAA,MACA,SAAS,CAAC,SAAc;AAAA,MAAC;AAAA,IAC3B;AAEA,UAAM,mBAAkB,wCAAS,aAAT,YAAqB;AAC7C,UAAM,mBAAkB,wCAAS,aAAT,YAAqB;AAE7C,QAAI,gBAAgB;AAEpB,QAAI,MAAM;AACR,uBAAiB,OAAO,SAAS,WAAW,OAAO,KAAK,UAAU,IAAI,KAAK;AAAA,IAC7E;AAEA,QAAI,iBAAiB;AACnB,uBAAiB,QAAQ,iBAAiB,CAAC,GAAG,+BAA+B;AAAA,IAC/E;AAEA,UAAM,gBAAyB;AAAA,MAC7B,IAAI;AAAA,MACJ,SAAS,kBAAkB,eAAe,YAAY;AAAA,MACtD,MAAM;AAAA,IACR;AAEA,UAAM,UAAU,kCACV,QAAQ,iBAAiB,WAAW,CAAC,IACrC,QAAQ,iBAAiB,eACzB,EAAE,CAAC,mCAAmC,GAAG,QAAQ,iBAAiB,aAAa,IAC/E,CAAC;AAGP,UAAM,WAAW,MAAM,6BAA6B;AAAA,MAClD,eAAe,QAAQ;AAAA,MACvB,UAAU,kBAAkB,CAAC,eAAe,GAAG,QAAQ,IAAI,CAAC,aAAa;AAAA,MACzE,OAAO,QAAQ,sCAAsC,EAAE,SAAS,OAAO,CAAC;AAAA,MACxE;AAAA,MACA,MAAM,QAAQ,iBAAiB;AAAA,MAC/B,YAAY,EAAE,MAAM,YAAY,UAAU,EAAE,MAAM,UAAU,EAAE;AAAA,MAC9D,QAAQ;AAAA,IACV,CAAC;AAED,QAAI,CAAC,SAAS,QAAQ;AACpB,YAAM,IAAI,MAAM,mDAAmD;AAAA,IACrE;AAEA,UAAM,SAAS,SAAS,OAAO,UAAU;AACzC,QAAI,YAAY;AAEhB,WAAO,MAAM;AACX,YAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAE1C,UAAI,MAAM;AACR;AAAA,MACF;AAEA,UAAI,MAAM,SAAS,WAAW;AAC5B,YAAI;AACF,cAAI,mBAAmB,KAAK,MAAM,eAAe,MAAM,SAAS,CAAC;AACjE,2CAAS;AAAA,YACP,QAAQ,YAAY,YAAY;AAAA,YAChC,MAAM;AAAA,UACR;AACA,sBAAY;AAAA,QACd,SAAS,GAAP;AAAA,QAAW;AAAA,MACf;AAEA,UAAI,MAAM,SAAS,YAAY;AAC7B,yCAAS;AAAA,UACP,QAAQ;AAAA,UACR,MAAM,MAAM;AAAA,QACd;AACA,eAAO,MAAM;AAAA,MACf;AAAA,IACF;AAEA,UAAM,IAAI,MAAM,6CAA6C;AAAA,EAC/D;AAAA;AAEA,SAAS,kBAAkB,eAAuB,cAA8B;AAC9E,SAAO;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQP;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYA;AAAA;AAAA;AAAA;AAAA;AAKF;","names":[]}

package/dist/chunk-7GFKOIO7.mjs

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

package/dist/chunk-B2H3NC4E.mjs

@@ -1,204 +0,0 @@
-import {
-  fetchAndDecodeChatCompletion
-} from "./chunk-MJKBCG4U.mjs";
-import {
-  __async,
-  __spreadProps,
-  __spreadValues
-} from "./chunk-SKC7AJIV.mjs";
-
-// src/hooks/use-chat.ts
-import { useRef } from "react";
-import {
-  encodeResult,
-  COPILOT_CLOUD_PUBLIC_API_KEY_HEADER
-} from "@copilotkit/shared";
-import { nanoid } from "nanoid";
-import untruncateJson from "untruncate-json";
-function useChat(options) {
-  const { messages, setMessages, makeSystemMessageCallback } = options;
-  const abortControllerRef = useRef();
-  const threadIdRef = useRef(null);
-  const runIdRef = useRef(null);
-  const publicApiKey = options.copilotConfig.publicApiKey;
-  const headers = __spreadValues(__spreadValues({}, options.headers || {}), publicApiKey ? { [COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {});
-  const runChatCompletion = (messages2) => __async(this, null, function* () {
-    options.setIsLoading(true);
-    const newMessages = [
-      {
-        id: nanoid(),
-        createdAt: /* @__PURE__ */ new Date(),
-        content: "",
-        role: "assistant"
-      }
-    ];
-    const abortController = new AbortController();
-    abortControllerRef.current = abortController;
-    setMessages([...messages2, ...newMessages]);
-    const copilotConfigBody = options.copilotConfig.body || {};
-    if (threadIdRef.current) {
-      copilotConfigBody.threadId = threadIdRef.current;
-    }
-    if (runIdRef.current) {
-      copilotConfigBody.runId = runIdRef.current;
-    }
-    const systemMessage = makeSystemMessageCallback();
-    const messagesWithContext = [systemMessage, ...options.initialMessages || [], ...messages2];
-    const response = yield fetchAndDecodeChatCompletion({
-      copilotConfig: __spreadProps(__spreadValues({}, options.copilotConfig), { body: copilotConfigBody }),
-      messages: messagesWithContext,
-      tools: options.tools,
-      headers,
-      signal: abortController.signal
-    });
-    if (response.headers.get("threadid")) {
-      threadIdRef.current = response.headers.get("threadid");
-    }
-    if (response.headers.get("runid")) {
-      runIdRef.current = response.headers.get("runid");
-    }
-    if (!response.events) {
-      setMessages([
-        ...messages2,
-        {
-          id: nanoid(),
-          createdAt: /* @__PURE__ */ new Date(),
-          content: response.statusText,
-          role: "assistant"
-        }
-      ]);
-      options.setIsLoading(false);
-      throw new Error("Failed to fetch chat completion");
-    }
-    const reader = response.events.getReader();
-    let feedback = false;
-    try {
-      while (true) {
-        const { done, value } = yield reader.read();
-        if (done) {
-          break;
-        }
-        let currentMessage = Object.assign({}, newMessages[newMessages.length - 1]);
-        if (value.type === "content") {
-          if (currentMessage.function_call || currentMessage.role === "function") {
-            currentMessage = {
-              id: nanoid(),
-              createdAt: /* @__PURE__ */ new Date(),
-              content: "",
-              role: "assistant"
-            };
-            newMessages.push(currentMessage);
-          }
-          currentMessage.content += value.content;
-          newMessages[newMessages.length - 1] = currentMessage;
-          setMessages([...messages2, ...newMessages]);
-        } else if (value.type === "result") {
-          currentMessage = {
-            id: nanoid(),
-            role: "function",
-            content: value.content,
-            name: value.name
-          };
-          newMessages.push(currentMessage);
-          setMessages([...messages2, ...newMessages]);
-          feedback = true;
-        } else if (value.type === "function" || value.type === "partial") {
-          if (currentMessage.content != "" || currentMessage.function_call || currentMessage.role == "function") {
-            currentMessage = {
-              id: nanoid(),
-              createdAt: /* @__PURE__ */ new Date(),
-              content: "",
-              role: "assistant"
-            };
-            newMessages.push(currentMessage);
-          }
-          if (value.type === "function") {
-            currentMessage.function_call = {
-              name: value.name,
-              arguments: JSON.stringify(value.arguments),
-              scope: value.scope
-            };
-          } else if (value.type === "partial") {
-            let partialArguments = {};
-            try {
-              partialArguments = JSON.parse(untruncateJson(value.arguments));
-            } catch (e) {
-            }
-            currentMessage.partialFunctionCall = {
-              name: value.name,
-              arguments: partialArguments
-            };
-          }
-          newMessages[newMessages.length - 1] = currentMessage;
-          setMessages([...messages2, ...newMessages]);
-          if (value.type === "function") {
-            try {
-              if (options.onFunctionCall && value.scope === "client") {
-                const result = yield options.onFunctionCall(
-                  messages2,
-                  currentMessage.function_call
-                );
-                currentMessage = {
-                  id: nanoid(),
-                  role: "function",
-                  content: encodeResult(result),
-                  name: currentMessage.function_call.name
-                };
-                newMessages.push(currentMessage);
-                setMessages([...messages2, ...newMessages]);
-                feedback = true;
-              }
-            } catch (error) {
-              console.error("Failed to execute function call", error);
-            }
-          }
-        }
-      }
-      if (feedback) {
-        yield new Promise((resolve) => setTimeout(resolve, 10));
-        return yield runChatCompletion([...messages2, ...newMessages]);
-      } else {
-        return newMessages.slice();
-      }
-    } finally {
-      options.setIsLoading(false);
-    }
-  });
-  const runChatCompletionAndHandleFunctionCall = (messages2) => __async(this, null, function* () {
-    yield runChatCompletion(messages2);
-  });
-  const append = (message) => __async(this, null, function* () {
-    if (options.isLoading) {
-      return;
-    }
-    const newMessages = [...messages, message];
-    setMessages(newMessages);
-    return runChatCompletionAndHandleFunctionCall(newMessages);
-  });
-  const reload = () => __async(this, null, function* () {
-    if (options.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 {
-    append,
-    reload,
-    stop
-  };
-}
-
-export {
-  useChat
-};
-//# sourceMappingURL=chunk-B2H3NC4E.mjs.map