@copilotkit/react-core 0.17.0-alpha.3 → 0.17.0-alpha.4

package/dist/chunk-TKXPMMND.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/openai/chat-completion-client.ts"],"names":[],"mappings":";;;;;;;;;AAAA,OAAO,kBAAkB;AAiCzB,IAAM,qBAAqB;AAEpB,IAAM,uBAAN,cAAmC,aAAyC;AAAA,EAMjF,YAAY,QAA2C;AACrD,UAAM;AANR,SAAQ,0BAA0D;AAClE,SAAQ,OAAsC;AAC9C,SAAQ,mBAA2B;AACnC,SAAQ,wBAAgC;AAyExC,SAAQ,SAAS,CAAC,SAA8B;AAE9C,UAAI,KAAK,SAAS,cAAc,CAAC,KAAK,QAAQ,CAAC,EAAE,MAAM,eAAe;AACpE,cAAM,UAAU,KAAK,qBAAqB;AAC1C,YAAI,CAAC,SAAS;AACZ;AAAA,QACF;AAAA,MACF;AAEA,WAAK,OAAO,KAAK,QAAQ,CAAC,EAAE,MAAM,gBAAgB,aAAa;AAE/D,UAAI,KAAK,SAAS,WAAW;AAG3B,YAAI,KAAK,QAAQ,CAAC,EAAE,MAAM,SAAS;AACjC,eAAK,KAAK,WAAW,KAAK,QAAQ,CAAC,EAAE,MAAM,OAAO;AAAA,QACpD;AAEA;AAAA,MACF,WAAW,KAAK,SAAS,YAAY;AAGnC,YAAI,KAAK,QAAQ,CAAC,EAAE,MAAM,cAAe,MAAM;AAC7C,eAAK,mBAAmB,KAAK,QAAQ,CAAC,EAAE,MAAM,cAAe;AAAA,QAC/D;AACA,YAAI,KAAK,QAAQ,CAAC,EAAE,MAAM,cAAe,WAAW;AAClD,eAAK,yBAAyB,KAAK,QAAQ,CAAC,EAAE,MAAM,cAAe;AAAA,QACrE;AACA,aAAK,KAAK,WAAW,KAAK,kBAAkB,KAAK,qBAAqB;AAEtE;AAAA,MACF;AAAA,IACF;AAEA,SAAQ,UAAU,CAAC,UAAe;AAChC,WAAK,KAAK,SAAS,KAAK;AACxB,WAAK,QAAQ;AAAA,IACf;AAEA,SAAQ,QAAQ,MAAM;AACpB,UAAI,KAAK,SAAS,YAAY;AAC5B,cAAM,UAAU,KAAK,qBAAqB;AAC1C,YAAI,CAAC,SAAS;AACZ;AAAA,QACF;AAAA,MACF;AACA,WAAK,KAAK,KAAK;AACf,WAAK,QAAQ;AAAA,IACf;AAAA,EArHA;AAAA,EAEa,MAAM,QAA4C;AAAA;AAC7D,eAAS,mBAAK;AACd,UAAI,OAAO,SAAS,OAAO,SAAS,kBAAkB;AACpD,eAAO,cAAP,OAAO,YAAc,iBAAiB,OAAO,KAAK;AAAA,MACpD,OAAO;AACL,eAAO,cAAP,OAAO,YAAc;AAAA,MACvB;AAEA,aAAO,cAAP,OAAO,YAAc,CAAC;AACtB,aAAO,WAAW,KAAK,YAAY,MAAM;AACzC,aAAO,MAAM,KAAK,UAAU,MAAM;AAAA,IACpC;AAAA;AAAA,EAEQ,YAAY,QAAuD;AACzE,QAAI,YAAY,OAAO;AACvB,UAAM,WAAW,OAAO;AACxB,UAAM,YAAY,OAAO;AACzB,UAAM,qBAAqB,qBAAqB,SAAS;AACzD,QAAI,qBAAqB,WAAW;AAClC,YAAM,IAAI,MAAM,sCAAsC,wBAAwB,WAAW;AAAA,IAC3F;AACA,iBAAa;AAEb,eAAW,WAAW,UAAU;AAC9B,UAAI,QAAQ,SAAS,UAAU;AAC7B,cAAM,YAAY,KAAK,YAAY,OAAO;AAC1C,qBAAa;AAEb,YAAI,YAAY,GAAG;AACjB,gBAAM,IAAI,MAAM,uCAAuC;AAAA,QACzD;AAAA,MACF;AAAA,IACF;AAEA,UAAM,SAAoB,CAAC;AAC3B,QAAI,SAAkB;AAEtB,UAAM,mBAAmB,CAAC,GAAG,QAAQ,EAAE,QAAQ;AAC/C,eAAW,WAAW,kBAAkB;AACtC,UAAI,QAAQ,SAAS,UAAU;AAC7B,eAAO,QAAQ,OAAO;AACtB;AAAA,MACF,WAAW,QAAQ;AACjB;AAAA,MACF;AACA,UAAI,YAAY,KAAK,YAAY,OAAO;AACxC,UAAI,YAAY,WAAW;AACzB,iBAAS;AACT;AAAA,MACF;AACA,aAAO,QAAQ,OAAO;AACtB,mBAAa;AAAA,IACf;AAEA,WAAO;AAAA,EACT;AAAA,EAEc,UAAU,QAA2D;AAAA;AACjF,WAAK,0BAA0B,IAAI,wBAAwB,CAAC,CAAC;AAE7D,WAAK,wBAAwB,GAAG,QAAQ,KAAK,MAAM;AACnD,WAAK,wBAAwB,GAAG,SAAS,KAAK,OAAO;AACrD,WAAK,wBAAwB,GAAG,OAAO,KAAK,KAAK;AAEjD,YAAM,KAAK,wBAAwB,MAAM,MAAM;AAAA,IACjD;AAAA;AAAA,EAoDQ,uBAAgC;AACtC,QAAI,OAAY;AAChB,QAAI;AACF,aAAO,KAAK,MAAM,KAAK,qBAAqB;AAAA,IAC9C,SAAS,OAAP;AACA,WAAK,KAAK,SAAS,KAAK;AACxB,WAAK,QAAQ;AACb,aAAO;AAAA,IACT;AACA,SAAK,KAAK,YAAY;AAAA,MACpB,MAAM,KAAK;AAAA,MACX,WAAW;AAAA,IACb,CAAC;AACD,SAAK,OAAO;AACZ,SAAK,mBAAmB;AACxB,SAAK,wBAAwB;AAC7B,WAAO;AAAA,EACT;AAAA,EAEQ,UAAU;AAChB,QAAI,KAAK,yBAAyB;AAChC,WAAK,wBAAwB,IAAI,QAAQ,KAAK,MAAM;AACpD,WAAK,wBAAwB,IAAI,SAAS,KAAK,OAAO;AACtD,WAAK,wBAAwB,IAAI,OAAO,KAAK,KAAK;AAAA,IACpD;AACA,SAAK,0BAA0B;AAC/B,SAAK,OAAO;AACZ,SAAK,mBAAmB;AACxB,SAAK,wBAAwB;AAAA,EAC/B;AAAA,EAEO,YAAY,SAA0B;AAC3C,QAAI,QAAQ,SAAS;AACnB,aAAO,eAAe,QAAQ,OAAO;AAAA,IACvC,WAAW,QAAQ,eAAe;AAChC,aAAO,eAAe,KAAK,UAAU,QAAQ,aAAa,CAAC;AAAA,IAC7D;AACA,WAAO;AAAA,EACT;AACF;AAEA,IAAM,mBAA8C;AAAA,EAClD,iBAAiB;AAAA,EACjB,qBAAqB;AAAA,EACrB,SAAS;AAAA,EACT,sBAAsB;AAAA,EACtB,aAAa;AAAA,EACb,sBAAsB;AAAA,EACtB,cAAc;AAAA,EACd,kBAAkB;AAAA,EAClB,sBAAsB;AAAA,EACtB,cAAc;AAAA,EACd,kBAAkB;AAAA,EAClB,0BAA0B;AAC5B;AAEA,SAAS,eAAe,MAAsB;AAC5C,SAAO,KAAK,SAAS;AACvB;AAEA,SAAS,qBAAqB,WAA+B;AAC3D,MAAI,UAAU,WAAW,GAAG;AAC1B,WAAO;AAAA,EACT;AACA,QAAM,OAAO,KAAK,UAAU,SAAS;AACrC,SAAO,eAAe,IAAI;AAC5B","sourcesContent":["import EventEmitter from \"eventemitter3\";\nimport { Function, Message, Role } from \"../types\";\nimport {\n  ChatCompletionTransport,\n  ChatCompletionTransportFetchParams,\n} from \"./chat-completion-transport\";\n\ninterface ChatCompletionClientConfiguration {}\n\ninterface ChatCompletionClientEvents {\n  content: string;\n  partial: [string, string];\n  error: any;\n  function: {\n    name: string;\n    arguments: any;\n  };\n  end: void;\n}\n\nexport interface ChatCompletionChunk {\n  choices: {\n    delta: {\n      role: Role;\n      content?: string | null;\n      function_call?: {\n        name?: string;\n        arguments?: string;\n      };\n    };\n  }[];\n}\n\nconst DEFAULT_MAX_TOKENS = 8192;\n\nexport class ChatCompletionClient extends EventEmitter<ChatCompletionClientEvents> {\n  private chatCompletionTransport: ChatCompletionTransport | null = null;\n  private mode: \"function\" | \"message\" | null = null;\n  private functionCallName: string = \"\";\n  private functionCallArguments: string = \"\";\n\n  constructor(params: ChatCompletionClientConfiguration) {\n    super();\n  }\n\n  public async fetch(params: ChatCompletionTransportFetchParams) {\n    params = { ...params };\n    if (params.model && params.model in maxTokensByModel) {\n      params.maxTokens ||= maxTokensByModel[params.model];\n    } else {\n      params.maxTokens ||= DEFAULT_MAX_TOKENS;\n    }\n\n    params.functions ||= [];\n    params.messages = this.buildPrompt(params);\n    return await this.runPrompt(params);\n  }\n\n  private buildPrompt(params: ChatCompletionTransportFetchParams): Message[] {\n    let maxTokens = params.maxTokens!;\n    const messages = params.messages!;\n    const functions = params.functions!;\n    const functionsNumTokens = countFunctionsTokens(functions);\n    if (functionsNumTokens > maxTokens) {\n      throw new Error(`Too many tokens in function calls: ${functionsNumTokens} > ${maxTokens}`);\n    }\n    maxTokens -= functionsNumTokens;\n\n    for (const message of messages) {\n      if (message.role === \"system\") {\n        const numTokens = this.countTokens(message);\n        maxTokens -= numTokens;\n\n        if (maxTokens < 0) {\n          throw new Error(\"Not enough tokens for system message.\");\n        }\n      }\n    }\n\n    const result: Message[] = [];\n    let cutoff: boolean = false;\n\n    const reversedMessages = [...messages].reverse();\n    for (const message of reversedMessages) {\n      if (message.role === \"system\") {\n        result.unshift(message);\n        continue;\n      } else if (cutoff) {\n        continue;\n      }\n      let numTokens = this.countTokens(message);\n      if (maxTokens < numTokens) {\n        cutoff = true;\n        continue;\n      }\n      result.unshift(message);\n      maxTokens -= numTokens;\n    }\n\n    return result;\n  }\n\n  private async runPrompt(params: ChatCompletionTransportFetchParams): Promise<void> {\n    this.chatCompletionTransport = new ChatCompletionTransport({});\n\n    this.chatCompletionTransport.on(\"data\", this.onData);\n    this.chatCompletionTransport.on(\"error\", this.onError);\n    this.chatCompletionTransport.on(\"end\", this.onEnd);\n\n    await this.chatCompletionTransport.fetch(params);\n  }\n\n  private onData = (data: ChatCompletionChunk) => {\n    // In case we are in a function call but the next message is not a function call, flush it.\n    if (this.mode === \"function\" && !data.choices[0].delta.function_call) {\n      const success = this.tryFlushFunctionCall();\n      if (!success) {\n        return;\n      }\n    }\n\n    this.mode = data.choices[0].delta.function_call ? \"function\" : \"message\";\n\n    if (this.mode === \"message\") {\n      // if we get a message, emit the content and return;\n\n      if (data.choices[0].delta.content) {\n        this.emit(\"content\", data.choices[0].delta.content);\n      }\n\n      return;\n    } else if (this.mode === \"function\") {\n      // if we get a function call, we buffer the name and arguments, then emit a partial event.\n\n      if (data.choices[0].delta.function_call!.name) {\n        this.functionCallName = data.choices[0].delta.function_call!.name!;\n      }\n      if (data.choices[0].delta.function_call!.arguments) {\n        this.functionCallArguments += data.choices[0].delta.function_call!.arguments!;\n      }\n      this.emit(\"partial\", this.functionCallName, this.functionCallArguments);\n\n      return;\n    }\n  };\n\n  private onError = (error: any) => {\n    this.emit(\"error\", error);\n    this.cleanup();\n  };\n\n  private onEnd = () => {\n    if (this.mode === \"function\") {\n      const success = this.tryFlushFunctionCall();\n      if (!success) {\n        return;\n      }\n    }\n    this.emit(\"end\");\n    this.cleanup();\n  };\n\n  private tryFlushFunctionCall(): boolean {\n    let args: any = null;\n    try {\n      args = JSON.parse(this.functionCallArguments);\n    } catch (error) {\n      this.emit(\"error\", error);\n      this.cleanup();\n      return false;\n    }\n    this.emit(\"function\", {\n      name: this.functionCallName,\n      arguments: args,\n    });\n    this.mode = null;\n    this.functionCallName = \"\";\n    this.functionCallArguments = \"\";\n    return true;\n  }\n\n  private cleanup() {\n    if (this.chatCompletionTransport) {\n      this.chatCompletionTransport.off(\"data\", this.onData);\n      this.chatCompletionTransport.off(\"error\", this.onError);\n      this.chatCompletionTransport.off(\"end\", this.onEnd);\n    }\n    this.chatCompletionTransport = null;\n    this.mode = null;\n    this.functionCallName = \"\";\n    this.functionCallArguments = \"\";\n  }\n\n  public countTokens(message: Message): number {\n    if (message.content) {\n      return estimateTokens(message.content);\n    } else if (message.function_call) {\n      return estimateTokens(JSON.stringify(message.function_call));\n    }\n    return 0;\n  }\n}\n\nconst maxTokensByModel: { [key: string]: number } = {\n  \"gpt-3.5-turbo\": 4097,\n  \"gpt-3.5-turbo-16k\": 16385,\n  \"gpt-4\": 8192,\n  \"gpt-4-1106-preview\": 8192,\n  \"gpt-4-32k\": 32768,\n  \"gpt-3.5-turbo-0301\": 4097,\n  \"gpt-4-0314\": 8192,\n  \"gpt-4-32k-0314\": 32768,\n  \"gpt-3.5-turbo-0613\": 4097,\n  \"gpt-4-0613\": 8192,\n  \"gpt-4-32k-0613\": 32768,\n  \"gpt-3.5-turbo-16k-0613\": 16385,\n};\n\nfunction estimateTokens(text: string): number {\n  return text.length / 3;\n}\n\nfunction countFunctionsTokens(functions: Function[]): number {\n  if (functions.length === 0) {\n    return 0;\n  }\n  const json = JSON.stringify(functions);\n  return estimateTokens(json);\n}\n"]}

package/dist/chunk-VK25FIML.mjs

@@ -0,0 +1,144 @@
+import { __async, __spreadValues } from './chunk-MRXNTQOX.mjs';
+import EventEmitter from 'eventemitter3';
+
+var DEFAULT_MODEL = "gpt-4-1106-preview";
+var ChatCompletionTransport = class extends EventEmitter {
+  constructor(params) {
+    super();
+    this.buffer = new Uint8Array();
+    this.bodyReader = null;
+  }
+  cleanup() {
+    return __async(this, null, function* () {
+      if (this.bodyReader) {
+        try {
+          yield this.bodyReader.cancel();
+        } catch (error) {
+          console.warn("Failed to cancel body reader:", error);
+        }
+      }
+      this.bodyReader = null;
+      this.buffer = new Uint8Array();
+    });
+  }
+  fetch(_0) {
+    return __async(this, arguments, function* ({
+      copilotConfig,
+      model,
+      messages,
+      functions,
+      temperature,
+      headers,
+      body,
+      signal
+    }) {
+      yield this.cleanup();
+      temperature || (temperature = 0.5);
+      functions || (functions = []);
+      model || (model = DEFAULT_MODEL);
+      const cleanedMessages = messages.map((message) => {
+        const { content, role, name, function_call } = message;
+        return { content, role, name, function_call };
+      });
+      try {
+        const response = yield fetch(copilotConfig.chatApiEndpoint, {
+          method: "POST",
+          headers: __spreadValues(__spreadValues({
+            "Content-Type": "application/json"
+          }, copilotConfig.headers), headers ? __spreadValues({}, headers) : {}),
+          body: JSON.stringify(__spreadValues(__spreadValues(__spreadValues(__spreadValues(__spreadValues({
+            model,
+            messages: cleanedMessages,
+            stream: true
+          }, functions.length ? { functions } : {}), temperature ? { temperature } : {}), functions.length != 0 ? { function_call: "auto" } : {}), copilotConfig.body), body ? __spreadValues({}, body) : {})),
+          signal
+        });
+        if (!response.ok) {
+          try {
+            const errorText = yield response.text();
+            yield this.cleanup();
+            const msg = `Error ${response.status}: ${errorText}`;
+            this.emit("error", new Error(msg));
+          } catch (_error) {
+            yield this.cleanup();
+            const msg = `Error ${response.status}: ${response.statusText}`;
+            this.emit("error", new Error(msg));
+          }
+          return;
+        }
+        if (response.body == null) {
+          yield this.cleanup();
+          const msg = "Response body is null";
+          this.emit("error", new Error(msg));
+          return;
+        }
+        this.bodyReader = response.body.getReader();
+        yield this.streamBody();
+      } catch (error) {
+        yield this.cleanup();
+        this.emit("error", error);
+        return;
+      }
+    });
+  }
+  streamBody() {
+    return __async(this, null, function* () {
+      while (true) {
+        try {
+          const { done, value } = yield this.bodyReader.read();
+          if (done) {
+            yield this.cleanup();
+            this.emit("end");
+            return;
+          }
+          const shouldContinue = yield this.processData(value);
+          if (!shouldContinue) {
+            return;
+          }
+        } catch (error) {
+          yield this.cleanup();
+          this.emit("error", error);
+          return;
+        }
+      }
+    });
+  }
+  processData(data) {
+    return __async(this, null, function* () {
+      const newBuffer = new Uint8Array(this.buffer.length + data.length);
+      newBuffer.set(this.buffer);
+      newBuffer.set(data, this.buffer.length);
+      this.buffer = newBuffer;
+      const dataString = new TextDecoder("utf-8").decode(this.buffer);
+      let lines = dataString.split("\n").filter((line) => line.trim() !== "");
+      if (!dataString.endsWith("\n")) {
+        const lastLine = lines.pop() || "";
+        const remainingBytes = new TextEncoder().encode(lastLine);
+        this.buffer = new Uint8Array(remainingBytes);
+      } else {
+        this.buffer = new Uint8Array();
+      }
+      for (const line of lines) {
+        const cleanedLine = line.replace(/^data: /, "");
+        if (cleanedLine === "[DONE]") {
+          yield this.cleanup();
+          this.emit("end");
+          return false;
+        }
+        let json;
+        try {
+          json = JSON.parse(cleanedLine);
+        } catch (error) {
+          console.error("Failed to parse JSON:", error);
+          continue;
+        }
+        this.emit("data", json);
+      }
+      return true;
+    });
+  }
+};
+
+export { ChatCompletionTransport };
+//# sourceMappingURL=out.js.map
+//# sourceMappingURL=chunk-VK25FIML.mjs.map

package/dist/chunk-VK25FIML.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/openai/chat-completion-transport.ts"],"names":[],"mappings":";;;;;;AAAA,OAAO,kBAAkB;AAwBzB,IAAM,gBAAgB;AAEf,IAAM,0BAAN,cAAsC,aAA4C;AAAA,EAIvF,YAAY,QAA8C;AACxD,UAAM;AAJR,SAAQ,SAAS,IAAI,WAAW;AAChC,SAAQ,aAA6D;AAAA,EAIrE;AAAA,EAEc,UAAU;AAAA;AACtB,UAAI,KAAK,YAAY;AACnB,YAAI;AACF,gBAAM,KAAK,WAAW,OAAO;AAAA,QAC/B,SAAS,OAAP;AACA,kBAAQ,KAAK,iCAAiC,KAAK;AAAA,QACrD;AAAA,MACF;AACA,WAAK,aAAa;AAClB,WAAK,SAAS,IAAI,WAAW;AAAA,IAC/B;AAAA;AAAA,EAEa,MAAM,IASmC;AAAA,+CATnC;AAAA,MACjB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,GAAsD;AACpD,YAAM,KAAK,QAAQ;AAEnB,oCAAgB;AAChB,gCAAc,CAAC;AACf,wBAAU;AAGV,YAAM,kBAAkB,SAAS,IAAI,CAAC,YAAY;AAChD,cAAM,EAAE,SAAS,MAAM,MAAM,cAAc,IAAI;AAC/C,eAAO,EAAE,SAAS,MAAM,MAAM,cAAc;AAAA,MAC9C,CAAC;AAED,UAAI;AACF,cAAM,WAAW,MAAM,MAAM,cAAc,iBAAiB;AAAA,UAC1D,QAAQ;AAAA,UACR,SAAS;AAAA,YACP,gBAAgB;AAAA,aACb,cAAc,UACb,UAAU,mBAAK,WAAY,CAAC;AAAA,UAElC,MAAM,KAAK,UAAU;AAAA,YACnB;AAAA,YACA,UAAU;AAAA,YACV,QAAQ;AAAA,aACJ,UAAU,SAAS,EAAE,UAAU,IAAI,CAAC,IACpC,cAAc,EAAE,YAAY,IAAI,CAAC,IACjC,UAAU,UAAU,IAAI,EAAE,eAAe,OAAO,IAAI,CAAC,IACtD,cAAc,OACb,OAAO,mBAAK,QAAS,CAAC,EAC3B;AAAA,UACD;AAAA,QACF,CAAC;AAED,YAAI,CAAC,SAAS,IAAI;AAChB,cAAI;AACF,kBAAM,YAAY,MAAM,SAAS,KAAK;AACtC,kBAAM,KAAK,QAAQ;AACnB,kBAAM,MAAM,SAAS,SAAS,WAAW;AACzC,iBAAK,KAAK,SAAS,IAAI,MAAM,GAAG,CAAC;AAAA,UACnC,SAAS,QAAP;AACA,kBAAM,KAAK,QAAQ;AACnB,kBAAM,MAAM,SAAS,SAAS,WAAW,SAAS;AAClD,iBAAK,KAAK,SAAS,IAAI,MAAM,GAAG,CAAC;AAAA,UACnC;AACA;AAAA,QACF;AAEA,YAAI,SAAS,QAAQ,MAAM;AACzB,gBAAM,KAAK,QAAQ;AACnB,gBAAM,MAAM;AACZ,eAAK,KAAK,SAAS,IAAI,MAAM,GAAG,CAAC;AACjC;AAAA,QACF;AAEA,aAAK,aAAa,SAAS,KAAK,UAAU;AAE1C,cAAM,KAAK,WAAW;AAAA,MACxB,SAAS,OAAP;AACA,cAAM,KAAK,QAAQ;AACnB,aAAK,KAAK,SAAS,KAAK;AACxB;AAAA,MACF;AAAA,IACF;AAAA;AAAA,EAEc,aAAa;AAAA;AACzB,aAAO,MAAM;AACX,YAAI;AACF,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,KAAK,WAAY,KAAK;AAEpD,cAAI,MAAM;AACR,kBAAM,KAAK,QAAQ;AACnB,iBAAK,KAAK,KAAK;AACf;AAAA,UACF;AAEA,gBAAM,iBAAiB,MAAM,KAAK,YAAY,KAAK;AAEnD,cAAI,CAAC,gBAAgB;AACnB;AAAA,UACF;AAAA,QACF,SAAS,OAAP;AACA,gBAAM,KAAK,QAAQ;AACnB,eAAK,KAAK,SAAS,KAAK;AACxB;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA;AAAA,EAEc,YAAY,MAAoC;AAAA;AAE5D,YAAM,YAAY,IAAI,WAAW,KAAK,OAAO,SAAS,KAAK,MAAM;AACjE,gBAAU,IAAI,KAAK,MAAM;AACzB,gBAAU,IAAI,MAAM,KAAK,OAAO,MAAM;AACtC,WAAK,SAAS;AAEd,YAAM,aAAa,IAAI,YAAY,OAAO,EAAE,OAAO,KAAK,MAAM;AAE9D,UAAI,QAAQ,WAAW,MAAM,IAAI,EAAE,OAAO,CAAC,SAAS,KAAK,KAAK,MAAM,EAAE;AAGtE,UAAI,CAAC,WAAW,SAAS,IAAI,GAAG;AAC9B,cAAM,WAAW,MAAM,IAAI,KAAK;AAChC,cAAM,iBAAiB,IAAI,YAAY,EAAE,OAAO,QAAQ;AACxD,aAAK,SAAS,IAAI,WAAW,cAAc;AAAA,MAC7C,OAAO;AACL,aAAK,SAAS,IAAI,WAAW;AAAA,MAC/B;AAEA,iBAAW,QAAQ,OAAO;AACxB,cAAM,cAAc,KAAK,QAAQ,WAAW,EAAE;AAE9C,YAAI,gBAAgB,UAAU;AAC5B,gBAAM,KAAK,QAAQ;AACnB,eAAK,KAAK,KAAK;AACf,iBAAO;AAAA,QACT;AAEA,YAAI;AACJ,YAAI;AACF,iBAAO,KAAK,MAAM,WAAW;AAAA,QAC/B,SAAS,OAAP;AACA,kBAAQ,MAAM,yBAAyB,KAAK;AAC5C;AAAA,QACF;AAEA,aAAK,KAAK,QAAQ,IAAI;AAAA,MACxB;AACA,aAAO;AAAA,IACT;AAAA;AACF","sourcesContent":["import EventEmitter from \"eventemitter3\";\nimport { Message, Function } from \"../types\";\nimport { CopilotApiConfig } from \"../context\";\n\nexport interface ChatCompletionTransportConfiguration {}\n\ninterface ChatCompletionTransportEvents {\n  end: void;\n  data: any;\n  error: any;\n}\n\nexport interface ChatCompletionTransportFetchParams {\n  copilotConfig: CopilotApiConfig;\n  model?: string;\n  messages: Message[];\n  functions?: Function[];\n  temperature?: number;\n  maxTokens?: number;\n  headers?: Record<string, string> | Headers;\n  body?: object;\n  signal?: AbortSignal;\n}\n\nconst DEFAULT_MODEL = \"gpt-4-1106-preview\";\n\nexport class ChatCompletionTransport extends EventEmitter<ChatCompletionTransportEvents> {\n  private buffer = new Uint8Array();\n  private bodyReader: ReadableStreamDefaultReader<Uint8Array> | null = null;\n\n  constructor(params: ChatCompletionTransportConfiguration) {\n    super();\n  }\n\n  private async cleanup() {\n    if (this.bodyReader) {\n      try {\n        await this.bodyReader.cancel();\n      } catch (error) {\n        console.warn(\"Failed to cancel body reader:\", error);\n      }\n    }\n    this.bodyReader = null;\n    this.buffer = new Uint8Array();\n  }\n\n  public async fetch({\n    copilotConfig,\n    model,\n    messages,\n    functions,\n    temperature,\n    headers,\n    body,\n    signal,\n  }: ChatCompletionTransportFetchParams): Promise<void> {\n    await this.cleanup();\n\n    temperature ||= 0.5;\n    functions ||= [];\n    model ||= DEFAULT_MODEL;\n\n    // clean up any extra properties from messages\n    const cleanedMessages = messages.map((message) => {\n      const { content, role, name, function_call } = message;\n      return { content, role, name, function_call };\n    });\n\n    try {\n      const response = await fetch(copilotConfig.chatApiEndpoint, {\n        method: \"POST\",\n        headers: {\n          \"Content-Type\": \"application/json\",\n          ...copilotConfig.headers,\n          ...(headers ? { ...headers } : {}),\n        },\n        body: JSON.stringify({\n          model,\n          messages: cleanedMessages,\n          stream: true,\n          ...(functions.length ? { functions } : {}),\n          ...(temperature ? { temperature } : {}),\n          ...(functions.length != 0 ? { function_call: \"auto\" } : {}),\n          ...copilotConfig.body,\n          ...(body ? { ...body } : {}),\n        }),\n        signal,\n      });\n\n      if (!response.ok) {\n        try {\n          const errorText = await response.text();\n          await this.cleanup();\n          const msg = `Error ${response.status}: ${errorText}`;\n          this.emit(\"error\", new Error(msg));\n        } catch (_error) {\n          await this.cleanup();\n          const msg = `Error ${response.status}: ${response.statusText}`;\n          this.emit(\"error\", new Error(msg));\n        }\n        return;\n      }\n\n      if (response.body == null) {\n        await this.cleanup();\n        const msg = \"Response body is null\";\n        this.emit(\"error\", new Error(msg));\n        return;\n      }\n\n      this.bodyReader = response.body.getReader();\n\n      await this.streamBody();\n    } catch (error) {\n      await this.cleanup();\n      this.emit(\"error\", error);\n      return;\n    }\n  }\n\n  private async streamBody() {\n    while (true) {\n      try {\n        const { done, value } = await this.bodyReader!.read();\n\n        if (done) {\n          await this.cleanup();\n          this.emit(\"end\");\n          return;\n        }\n\n        const shouldContinue = await this.processData(value);\n\n        if (!shouldContinue) {\n          return;\n        }\n      } catch (error) {\n        await this.cleanup();\n        this.emit(\"error\", error);\n        return;\n      }\n    }\n  }\n\n  private async processData(data: Uint8Array): Promise<boolean> {\n    // Append new data to the temp buffer\n    const newBuffer = new Uint8Array(this.buffer.length + data.length);\n    newBuffer.set(this.buffer);\n    newBuffer.set(data, this.buffer.length);\n    this.buffer = newBuffer;\n\n    const dataString = new TextDecoder(\"utf-8\").decode(this.buffer);\n\n    let lines = dataString.split(\"\\n\").filter((line) => line.trim() !== \"\");\n\n    // If the last line isn't complete, keep it in the buffer for next time\n    if (!dataString.endsWith(\"\\n\")) {\n      const lastLine = lines.pop() || \"\";\n      const remainingBytes = new TextEncoder().encode(lastLine);\n      this.buffer = new Uint8Array(remainingBytes);\n    } else {\n      this.buffer = new Uint8Array();\n    }\n\n    for (const line of lines) {\n      const cleanedLine = line.replace(/^data: /, \"\");\n\n      if (cleanedLine === \"[DONE]\") {\n        await this.cleanup();\n        this.emit(\"end\");\n        return false;\n      }\n\n      let json;\n      try {\n        json = JSON.parse(cleanedLine);\n      } catch (error) {\n        console.error(\"Failed to parse JSON:\", error);\n        continue;\n      }\n\n      this.emit(\"data\", json);\n    }\n    return true;\n  }\n}\n"]}

package/dist/chunk-VNRDQJXW.mjs

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

package/dist/chunk-VNRDQJXW.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"names":[],"mappings":""}

package/dist/chunk-VUY2K2DI.mjs

@@ -0,0 +1,135 @@
+import { __spreadProps, __spreadValues } from './chunk-MRXNTQOX.mjs';
+import { nanoid } from 'nanoid';
+import { useReducer, useCallback } from 'react';
+
+var removeNode = (nodes, id) => {
+  return nodes.reduce((result, node) => {
+    if (node.id !== id) {
+      const newNode = __spreadProps(__spreadValues({}, node), { children: removeNode(node.children, id) });
+      result.push(newNode);
+    }
+    return result;
+  }, []);
+};
+var addNode = (nodes, newNode, parentId) => {
+  if (!parentId) {
+    return [...nodes, newNode];
+  }
+  return nodes.map((node) => {
+    if (node.id === parentId) {
+      return __spreadProps(__spreadValues({}, node), { children: [...node.children, newNode] });
+    } else if (node.children.length) {
+      return __spreadProps(__spreadValues({}, node), { children: addNode(node.children, newNode, parentId) });
+    }
+    return node;
+  });
+};
+var treeIndentationRepresentation = (index, indentLevel) => {
+  if (indentLevel === 0) {
+    return (index + 1).toString();
+  } else if (indentLevel === 1) {
+    return String.fromCharCode(65 + index);
+  } else if (indentLevel === 2) {
+    return String.fromCharCode(97 + index);
+  } else {
+    return "-";
+  }
+};
+var printNode = (node, prefix = "", indentLevel = 0) => {
+  const indent = " ".repeat(3).repeat(indentLevel);
+  const prefixPlusIndentLength = prefix.length + indent.length;
+  const subsequentLinesPrefix = " ".repeat(prefixPlusIndentLength);
+  const valueLines = node.value.split("\n");
+  const outputFirstLine = `${indent}${prefix}${valueLines[0]}`;
+  const outputSubsequentLines = valueLines.slice(1).map((line) => `${subsequentLinesPrefix}${line}`).join("\n");
+  let output = `${outputFirstLine}
+`;
+  if (outputSubsequentLines) {
+    output += `${outputSubsequentLines}
+`;
+  }
+  const childPrePrefix = " ".repeat(prefix.length);
+  node.children.forEach(
+    (child, index) => output += printNode(
+      child,
+      `${childPrePrefix}${treeIndentationRepresentation(index, indentLevel + 1)}. `,
+      indentLevel + 1
+    )
+  );
+  return output;
+};
+function treeReducer(state, action) {
+  switch (action.type) {
+    case "ADD_NODE": {
+      const { value, parentId, id: newNodeId } = action;
+      const newNode = {
+        id: newNodeId,
+        value,
+        children: [],
+        categories: new Set(action.categories)
+      };
+      try {
+        return addNode(state, newNode, parentId);
+      } catch (error) {
+        console.error(`Error while adding node with id ${newNodeId}: ${error}`);
+        return state;
+      }
+    }
+    case "REMOVE_NODE":
+      return removeNode(state, action.id);
+    default:
+      return state;
+  }
+}
+var useTree = () => {
+  const [tree, dispatch] = useReducer(treeReducer, []);
+  const addElement = useCallback(
+    (value, categories, parentId) => {
+      const newNodeId = nanoid();
+      dispatch({
+        type: "ADD_NODE",
+        value,
+        parentId,
+        id: newNodeId,
+        categories
+      });
+      return newNodeId;
+    },
+    []
+  );
+  const removeElement = useCallback((id) => {
+    dispatch({ type: "REMOVE_NODE", id });
+  }, []);
+  const printTree = useCallback(
+    (categories) => {
+      const categoriesSet = new Set(categories);
+      let output = "";
+      tree.forEach((node, index) => {
+        if (!setsHaveIntersection(categoriesSet, node.categories)) {
+          return;
+        }
+        if (index !== 0) {
+          output += "\n";
+        }
+        output += printNode(node, `${treeIndentationRepresentation(index, 0)}. `);
+      });
+      return output;
+    },
+    [tree]
+  );
+  return { tree, addElement, printTree, removeElement };
+};
+var use_tree_default = useTree;
+function setsHaveIntersection(setA, setB) {
+  const [smallerSet, largerSet] = setA.size <= setB.size ? [setA, setB] : [setB, setA];
+  for (let item of smallerSet) {
+    if (largerSet.has(item)) {
+      return true;
+    }
+  }
+  return false;
+}
+
+export { use_tree_default };
+//# sourceMappingURL=out.js.map
+//# sourceMappingURL=chunk-VUY2K2DI.mjs.map

package/dist/chunk-VUY2K2DI.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/hooks/use-tree.ts"],"names":[],"mappings":";;;;;;AAAA,SAAS,cAAc;AACvB,SAAS,aAAa,kBAAkB;AAkCxC,IAAM,aAAa,CAAC,OAAa,OAAyB;AACxD,SAAO,MAAM,OAAO,CAAC,QAAc,SAAS;AAC1C,QAAI,KAAK,OAAO,IAAI;AAClB,YAAM,UAAU,iCAAK,OAAL,EAAW,UAAU,WAAW,KAAK,UAAU,EAAE,EAAE;AACnE,aAAO,KAAK,OAAO;AAAA,IACrB;AACA,WAAO;AAAA,EACT,GAAG,CAAC,CAAC;AACP;AAEA,IAAM,UAAU,CAAC,OAAa,SAAmB,aAAgC;AAC/E,MAAI,CAAC,UAAU;AACb,WAAO,CAAC,GAAG,OAAO,OAAO;AAAA,EAC3B;AACA,SAAO,MAAM,IAAI,CAAC,SAAS;AACzB,QAAI,KAAK,OAAO,UAAU;AACxB,aAAO,iCAAK,OAAL,EAAW,UAAU,CAAC,GAAG,KAAK,UAAU,OAAO,EAAE;AAAA,IAC1D,WAAW,KAAK,SAAS,QAAQ;AAC/B,aAAO,iCAAK,OAAL,EAAW,UAAU,QAAQ,KAAK,UAAU,SAAS,QAAQ,EAAE;AAAA,IACxE;AACA,WAAO;AAAA,EACT,CAAC;AACH;AAEA,IAAM,gCAAgC,CAAC,OAAe,gBAAgC;AACpF,MAAI,gBAAgB,GAAG;AACrB,YAAQ,QAAQ,GAAG,SAAS;AAAA,EAC9B,WAAW,gBAAgB,GAAG;AAC5B,WAAO,OAAO,aAAa,KAAK,KAAK;AAAA,EACvC,WAAW,gBAAgB,GAAG;AAC5B,WAAO,OAAO,aAAa,KAAK,KAAK;AAAA,EACvC,OAAO;AACL,WAAO;AAAA,EACT;AACF;AAEA,IAAM,YAAY,CAAC,MAAgB,SAAS,IAAI,cAAc,MAAc;AAC1E,QAAM,SAAS,IAAI,OAAO,CAAC,EAAE,OAAO,WAAW;AAE/C,QAAM,yBAAyB,OAAO,SAAS,OAAO;AACtD,QAAM,wBAAwB,IAAI,OAAO,sBAAsB;AAE/D,QAAM,aAAa,KAAK,MAAM,MAAM,IAAI;AAExC,QAAM,kBAAkB,GAAG,SAAS,SAAS,WAAW,CAAC;AACzD,QAAM,wBAAwB,WAC3B,MAAM,CAAC,EACP,IAAI,CAAC,SAAS,GAAG,wBAAwB,MAAM,EAC/C,KAAK,IAAI;AAEZ,MAAI,SAAS,GAAG;AAAA;AAChB,MAAI,uBAAuB;AACzB,cAAU,GAAG;AAAA;AAAA,EACf;AAEA,QAAM,iBAAiB,IAAI,OAAO,OAAO,MAAM;AAE/C,OAAK,SAAS;AAAA,IACZ,CAAC,OAAO,UACL,UAAU;AAAA,MACT;AAAA,MACA,GAAG,iBAAiB,8BAA8B,OAAO,cAAc,CAAC;AAAA,MACxE,cAAc;AAAA,IAChB;AAAA,EACJ;AACA,SAAO;AACT;AAcA,SAAS,YAAY,OAAa,QAAsB;AACtD,UAAQ,OAAO,MAAM;AAAA,IACnB,KAAK,YAAY;AACf,YAAM,EAAE,OAAO,UAAU,IAAI,UAAU,IAAI;AAC3C,YAAM,UAAoB;AAAA,QACxB,IAAI;AAAA,QACJ;AAAA,QACA,UAAU,CAAC;AAAA,QACX,YAAY,IAAI,IAAI,OAAO,UAAU;AAAA,MACvC;AAEA,UAAI;AACF,eAAO,QAAQ,OAAO,SAAS,QAAQ;AAAA,MACzC,SAAS,OAAP;AACA,gBAAQ,MAAM,mCAAmC,cAAc,OAAO;AACtE,eAAO;AAAA,MACT;AAAA,IACF;AAAA,IACA,KAAK;AACH,aAAO,WAAW,OAAO,OAAO,EAAE;AAAA,IACpC;AACE,aAAO;AAAA,EACX;AACF;AAGA,IAAM,UAAU,MAAqB;AACnC,QAAM,CAAC,MAAM,QAAQ,IAAI,WAAW,aAAa,CAAC,CAAC;AAEnD,QAAM,aAAa;AAAA,IACjB,CAAC,OAAe,YAAsB,aAAkC;AACtE,YAAM,YAAY,OAAO;AACzB,eAAS;AAAA,QACP,MAAM;AAAA,QACN;AAAA,QACA;AAAA,QACA,IAAI;AAAA,QACJ;AAAA,MACF,CAAC;AACD,aAAO;AAAA,IACT;AAAA,IACA,CAAC;AAAA,EACH;AAEA,QAAM,gBAAgB,YAAY,CAAC,OAAyB;AAC1D,aAAS,EAAE,MAAM,eAAe,GAAG,CAAC;AAAA,EACtC,GAAG,CAAC,CAAC;AAEL,QAAM,YAAY;AAAA,IAChB,CAAC,eAAiC;AAChC,YAAM,gBAAgB,IAAI,IAAI,UAAU;AAExC,UAAI,SAAS;AACb,WAAK,QAAQ,CAAC,MAAM,UAAU;AAE5B,YAAI,CAAC,qBAAqB,eAAe,KAAK,UAAU,GAAG;AACzD;AAAA,QACF;AAGA,YAAI,UAAU,GAAG;AACf,oBAAU;AAAA,QACZ;AAEA,kBAAU,UAAU,MAAM,GAAG,8BAA8B,OAAO,CAAC,KAAK;AAAA,MAC1E,CAAC;AACD,aAAO;AAAA,IACT;AAAA,IACA,CAAC,IAAI;AAAA,EACP;AAEA,SAAO,EAAE,MAAM,YAAY,WAAW,cAAc;AACtD;AAEA,IAAO,mBAAQ;AAEf,SAAS,qBAAwB,MAAc,MAAuB;AACpE,QAAM,CAAC,YAAY,SAAS,IAAI,KAAK,QAAQ,KAAK,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,IAAI;AAEnF,WAAS,QAAQ,YAAY;AAC3B,QAAI,UAAU,IAAI,IAAI,GAAG;AACvB,aAAO;AAAA,IACT;AAAA,EACF;AAEA,SAAO;AACT","sourcesContent":["import { nanoid } from \"nanoid\";\nimport { useCallback, useReducer } from \"react\";\n\nexport type TreeNodeId = string;\n\nexport interface TreeNode {\n  id: TreeNodeId;\n  value: string;\n  children: TreeNode[];\n  parentId?: TreeNodeId;\n  categories: Set<string>;\n}\n\nexport type Tree = TreeNode[];\n\nexport interface UseTreeReturn {\n  tree: Tree;\n  addElement: (value: string, categories: string[], parentId?: TreeNodeId) => TreeNodeId;\n  printTree: (categories: string[]) => string;\n  removeElement: (id: TreeNodeId) => void;\n}\n\nconst findNode = (nodes: Tree, id: TreeNodeId): TreeNode | undefined => {\n  for (const node of nodes) {\n    if (node.id === id) {\n      return node;\n    }\n    const result = findNode(node.children, id);\n    if (result) {\n      return result;\n    }\n  }\n  return undefined;\n};\n\nconst removeNode = (nodes: Tree, id: TreeNodeId): Tree => {\n  return nodes.reduce((result: Tree, node) => {\n    if (node.id !== id) {\n      const newNode = { ...node, children: removeNode(node.children, id) };\n      result.push(newNode);\n    }\n    return result;\n  }, []);\n};\n\nconst addNode = (nodes: Tree, newNode: TreeNode, parentId?: TreeNodeId): Tree => {\n  if (!parentId) {\n    return [...nodes, newNode];\n  }\n  return nodes.map((node) => {\n    if (node.id === parentId) {\n      return { ...node, children: [...node.children, newNode] };\n    } else if (node.children.length) {\n      return { ...node, children: addNode(node.children, newNode, parentId) };\n    }\n    return node;\n  });\n};\n\nconst treeIndentationRepresentation = (index: number, indentLevel: number): string => {\n  if (indentLevel === 0) {\n    return (index + 1).toString();\n  } else if (indentLevel === 1) {\n    return String.fromCharCode(65 + index); // 65 is the ASCII value for 'A'\n  } else if (indentLevel === 2) {\n    return String.fromCharCode(97 + index); // 97 is the ASCII value for 'a'\n  } else {\n    return \"-\";\n  }\n};\n\nconst printNode = (node: TreeNode, prefix = \"\", indentLevel = 0): string => {\n  const indent = \" \".repeat(3).repeat(indentLevel);\n\n  const prefixPlusIndentLength = prefix.length + indent.length;\n  const subsequentLinesPrefix = \" \".repeat(prefixPlusIndentLength);\n\n  const valueLines = node.value.split(\"\\n\");\n\n  const outputFirstLine = `${indent}${prefix}${valueLines[0]}`;\n  const outputSubsequentLines = valueLines\n    .slice(1)\n    .map((line) => `${subsequentLinesPrefix}${line}`)\n    .join(\"\\n\");\n\n  let output = `${outputFirstLine}\\n`;\n  if (outputSubsequentLines) {\n    output += `${outputSubsequentLines}\\n`;\n  }\n\n  const childPrePrefix = \" \".repeat(prefix.length);\n\n  node.children.forEach(\n    (child, index) =>\n      (output += printNode(\n        child,\n        `${childPrePrefix}${treeIndentationRepresentation(index, indentLevel + 1)}. `,\n        indentLevel + 1,\n      )),\n  );\n  return output;\n};\n\n// Action types\ntype Action =\n  | {\n      type: \"ADD_NODE\";\n      value: string;\n      parentId?: string;\n      id: string;\n      categories: string[];\n    }\n  | { type: \"REMOVE_NODE\"; id: string };\n\n// Reducer function\nfunction treeReducer(state: Tree, action: Action): Tree {\n  switch (action.type) {\n    case \"ADD_NODE\": {\n      const { value, parentId, id: newNodeId } = action;\n      const newNode: TreeNode = {\n        id: newNodeId,\n        value,\n        children: [],\n        categories: new Set(action.categories),\n      };\n\n      try {\n        return addNode(state, newNode, parentId);\n      } catch (error) {\n        console.error(`Error while adding node with id ${newNodeId}: ${error}`);\n        return state;\n      }\n    }\n    case \"REMOVE_NODE\":\n      return removeNode(state, action.id);\n    default:\n      return state;\n  }\n}\n\n// useTree hook\nconst useTree = (): UseTreeReturn => {\n  const [tree, dispatch] = useReducer(treeReducer, []);\n\n  const addElement = useCallback(\n    (value: string, categories: string[], parentId?: string): TreeNodeId => {\n      const newNodeId = nanoid(); // Generate new ID outside of dispatch\n      dispatch({\n        type: \"ADD_NODE\",\n        value,\n        parentId,\n        id: newNodeId,\n        categories: categories,\n      });\n      return newNodeId; // Return the new ID\n    },\n    [],\n  );\n\n  const removeElement = useCallback((id: TreeNodeId): void => {\n    dispatch({ type: \"REMOVE_NODE\", id });\n  }, []);\n\n  const printTree = useCallback(\n    (categories: string[]): string => {\n      const categoriesSet = new Set(categories);\n\n      let output = \"\";\n      tree.forEach((node, index) => {\n        // if the node does not have any of the desired categories, continue to the next node\n        if (!setsHaveIntersection(categoriesSet, node.categories)) {\n          return;\n        }\n\n        // add a new line before each node except the first one\n        if (index !== 0) {\n          output += \"\\n\";\n        }\n\n        output += printNode(node, `${treeIndentationRepresentation(index, 0)}. `);\n      });\n      return output;\n    },\n    [tree],\n  );\n\n  return { tree, addElement, printTree, removeElement };\n};\n\nexport default useTree;\n\nfunction setsHaveIntersection<T>(setA: Set<T>, setB: Set<T>): boolean {\n  const [smallerSet, largerSet] = setA.size <= setB.size ? [setA, setB] : [setB, setA];\n\n  for (let item of smallerSet) {\n    if (largerSet.has(item)) {\n      return true;\n    }\n  }\n\n  return false;\n}\n"]}

package/dist/chunk-WMXA2TIX.mjs

@@ -0,0 +1,102 @@
+import { ChatCompletionClient } from './chunk-TKXPMMND.mjs';
+import { __async, __spreadValues } from './chunk-MRXNTQOX.mjs';
+import { useState } from 'react';
+import { nanoid } from 'nanoid';
+
+function useChat(options) {
+  const [messages, setMessages] = useState([]);
+  const [input, setInput] = useState("");
+  const [isLoading, setIsLoading] = useState(false);
+  const runChatCompletion = (messages2) => __async(this, null, function* () {
+    return new Promise((resolve, reject) => {
+      setIsLoading(true);
+      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");
+      };
+      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
+      });
+    });
+  });
+  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 = () => {
+    throw new Error("Not implemented");
+  };
+  return {
+    messages,
+    append,
+    reload,
+    stop,
+    isLoading,
+    input,
+    setInput
+  };
+}
+
+export { useChat };
+//# sourceMappingURL=out.js.map
+//# sourceMappingURL=chunk-WMXA2TIX.mjs.map

package/dist/chunk-WMXA2TIX.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/hooks/use-chat.ts"],"names":["messages"],"mappings":";;;;;;;;;AAAA,SAAS,gBAAgB;AAEzB,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;AAEhD,QAAM,oBAAoB,CAAOA,cAA0C;AACzE,WAAO,IAAI,QAAiB,CAAC,SAAS,WAAW;AAC/C,mBAAa,IAAI;AAEjB,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;AAAA,MACvB;AAEA,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,MACnB,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;AACvB,UAAM,IAAI,MAAM,iBAAiB;AAAA,EACnC;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF","sourcesContent":["import { 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\n  const runChatCompletion = async (messages: Message[]): Promise<Message> => {\n    return new Promise<Message>((resolve, reject) => {\n      setIsLoading(true);\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\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      });\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    throw new Error(\"Not implemented\");\n  };\n\n  return {\n    messages,\n    append,\n    reload,\n    stop,\n    isLoading,\n    input,\n    setInput,\n  };\n}\n"]}

package/dist/chunk-YULKJPY3.mjs

@@ -0,0 +1,70 @@
+import { nanoid } from 'nanoid';
+import { useReducer, useCallback } from 'react';
+
+// src/hooks/use-flat-category-store.ts
+var useFlatCategoryStore = () => {
+  const [elements, dispatch] = useReducer(flatCategoryStoreReducer, /* @__PURE__ */ new Map());
+  const addElement = useCallback((value, categories) => {
+    const newId = nanoid();
+    dispatch({
+      type: "ADD_ELEMENT",
+      value,
+      id: newId,
+      categories
+    });
+    return newId;
+  }, []);
+  const removeElement = useCallback((id) => {
+    dispatch({ type: "REMOVE_ELEMENT", id });
+  }, []);
+  const allElements = useCallback(
+    (categories) => {
+      const categoriesSet = new Set(categories);
+      const result = [];
+      elements.forEach((element) => {
+        if (setsHaveIntersection(categoriesSet, element.categories)) {
+          result.push(element.value);
+        }
+      });
+      return result;
+    },
+    [elements]
+  );
+  return { addElement, removeElement, allElements };
+};
+var use_flat_category_store_default = useFlatCategoryStore;
+function flatCategoryStoreReducer(state, action) {
+  switch (action.type) {
+    case "ADD_ELEMENT": {
+      const { value, id, categories } = action;
+      const newElement = {
+        id,
+        value,
+        categories: new Set(categories)
+      };
+      const newState = new Map(state);
+      newState.set(id, newElement);
+      return newState;
+    }
+    case "REMOVE_ELEMENT": {
+      const newState = new Map(state);
+      newState.delete(action.id);
+      return newState;
+    }
+    default:
+      return state;
+  }
+}
+function setsHaveIntersection(setA, setB) {
+  const [smallerSet, largerSet] = setA.size <= setB.size ? [setA, setB] : [setB, setA];
+  for (let item of smallerSet) {
+    if (largerSet.has(item)) {
+      return true;
+    }
+  }
+  return false;
+}
+
+export { use_flat_category_store_default };
+//# sourceMappingURL=out.js.map
+//# sourceMappingURL=chunk-YULKJPY3.mjs.map

package/dist/chunk-YULKJPY3.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/hooks/use-flat-category-store.ts"],"names":[],"mappings":";AAAA,SAAS,cAAc;AACvB,SAAS,aAAa,kBAAkB;AAgBxC,IAAM,uBAAuB,MAAwC;AACnE,QAAM,CAAC,UAAU,QAAQ,IAAI,WAE3B,0BAA0B,oBAAI,IAAsD,CAAC;AAEvF,QAAM,aAAa,YAAY,CAAC,OAAU,eAA8C;AACtF,UAAM,QAAQ,OAAO;AACrB,aAAS;AAAA,MACP,MAAM;AAAA,MACN;AAAA,MACA,IAAI;AAAA,MACJ;AAAA,IACF,CAAC;AACD,WAAO;AAAA,EACT,GAAG,CAAC,CAAC;AAEL,QAAM,gBAAgB,YAAY,CAAC,OAAkC;AACnE,aAAS,EAAE,MAAM,kBAAkB,GAAG,CAAC;AAAA,EACzC,GAAG,CAAC,CAAC;AAEL,QAAM,cAAc;AAAA,IAClB,CAAC,eAA8B;AAC7B,YAAM,gBAAgB,IAAI,IAAI,UAAU;AACxC,YAAM,SAAc,CAAC;AACrB,eAAS,QAAQ,CAAC,YAAY;AAC5B,YAAI,qBAAqB,eAAe,QAAQ,UAAU,GAAG;AAC3D,iBAAO,KAAK,QAAQ,KAAK;AAAA,QAC3B;AAAA,MACF,CAAC;AACD,aAAO;AAAA,IACT;AAAA,IACA,CAAC,QAAQ;AAAA,EACX;AAEA,SAAO,EAAE,YAAY,eAAe,YAAY;AAClD;AAEA,IAAO,kCAAQ;AAaf,SAAS,yBACP,OACA,QACuD;AACvD,UAAQ,OAAO,MAAM;AAAA,IACnB,KAAK,eAAe;AAClB,YAAM,EAAE,OAAO,IAAI,WAAW,IAAI;AAClC,YAAM,aAA0C;AAAA,QAC9C;AAAA,QACA;AAAA,QACA,YAAY,IAAI,IAAI,UAAU;AAAA,MAChC;AACA,YAAM,WAAW,IAAI,IAAI,KAAK;AAC9B,eAAS,IAAI,IAAI,UAAU;AAC3B,aAAO;AAAA,IACT;AAAA,IACA,KAAK,kBAAkB;AACrB,YAAM,WAAW,IAAI,IAAI,KAAK;AAC9B,eAAS,OAAO,OAAO,EAAE;AACzB,aAAO;AAAA,IACT;AAAA,IACA;AACE,aAAO;AAAA,EACX;AACF;AAEA,SAAS,qBAAwB,MAAc,MAAuB;AACpE,QAAM,CAAC,YAAY,SAAS,IAAI,KAAK,QAAQ,KAAK,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,IAAI;AAEnF,WAAS,QAAQ,YAAY;AAC3B,QAAI,UAAU,IAAI,IAAI,GAAG;AACvB,aAAO;AAAA,IACT;AAAA,EACF;AAEA,SAAO;AACT","sourcesContent":["import { nanoid } from \"nanoid\";\nimport { useCallback, useReducer } from \"react\";\n\nexport type FlatCategoryStoreId = string;\n\nexport interface UseFlatCategoryStoreReturn<T> {\n  addElement: (value: T, categories: string[]) => FlatCategoryStoreId;\n  removeElement: (id: FlatCategoryStoreId) => void;\n  allElements: (categories: string[]) => T[];\n}\n\ninterface FlatCategoryStoreElement<T> {\n  id: FlatCategoryStoreId;\n  value: T;\n  categories: Set<string>;\n}\n\nconst useFlatCategoryStore = <T>(): UseFlatCategoryStoreReturn<T> => {\n  const [elements, dispatch] = useReducer<\n    React.Reducer<Map<FlatCategoryStoreId, FlatCategoryStoreElement<T>>, Action<T>>\n  >(flatCategoryStoreReducer, new Map<FlatCategoryStoreId, FlatCategoryStoreElement<T>>());\n\n  const addElement = useCallback((value: T, categories: string[]): FlatCategoryStoreId => {\n    const newId = nanoid();\n    dispatch({\n      type: \"ADD_ELEMENT\",\n      value,\n      id: newId,\n      categories,\n    });\n    return newId;\n  }, []);\n\n  const removeElement = useCallback((id: FlatCategoryStoreId): void => {\n    dispatch({ type: \"REMOVE_ELEMENT\", id });\n  }, []);\n\n  const allElements = useCallback(\n    (categories: string[]): T[] => {\n      const categoriesSet = new Set(categories);\n      const result: T[] = [];\n      elements.forEach((element) => {\n        if (setsHaveIntersection(categoriesSet, element.categories)) {\n          result.push(element.value);\n        }\n      });\n      return result;\n    },\n    [elements],\n  );\n\n  return { addElement, removeElement, allElements };\n};\n\nexport default useFlatCategoryStore;\n\n// Action types\ntype Action<T> =\n  | {\n      type: \"ADD_ELEMENT\";\n      value: T;\n      id: FlatCategoryStoreId;\n      categories: string[];\n    }\n  | { type: \"REMOVE_ELEMENT\"; id: FlatCategoryStoreId };\n\n// Reducer\nfunction flatCategoryStoreReducer<T>(\n  state: Map<FlatCategoryStoreId, FlatCategoryStoreElement<T>>,\n  action: Action<T>,\n): Map<FlatCategoryStoreId, FlatCategoryStoreElement<T>> {\n  switch (action.type) {\n    case \"ADD_ELEMENT\": {\n      const { value, id, categories } = action;\n      const newElement: FlatCategoryStoreElement<T> = {\n        id,\n        value,\n        categories: new Set(categories),\n      };\n      const newState = new Map(state);\n      newState.set(id, newElement);\n      return newState;\n    }\n    case \"REMOVE_ELEMENT\": {\n      const newState = new Map(state);\n      newState.delete(action.id);\n      return newState;\n    }\n    default:\n      return state;\n  }\n}\n\nfunction setsHaveIntersection<T>(setA: Set<T>, setB: Set<T>): boolean {\n  const [smallerSet, largerSet] = setA.size <= setB.size ? [setA, setB] : [setB, setA];\n\n  for (let item of smallerSet) {\n    if (largerSet.has(item)) {\n      return true;\n    }\n  }\n\n  return false;\n}\n"]}

package/dist/components/copilot-provider/copilot-provider-props.mjs

@@ -1,2 +1,3 @@
-"use client";
+
+//# sourceMappingURL=out.js.map
 //# sourceMappingURL=copilot-provider-props.mjs.map

package/dist/components/copilot-provider/copilot-provider-props.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}
+{"version":3,"sources":[],"names":[],"mappings":""}