@copilotkit/react-core 0.16.0 → 0.17.0-alpha.0

package/dist/openai/chat-completion-stream.d.ts

@@ -0,0 +1,21 @@
+import { ChatCompletionTransportFetchParams } from './chat-completion-transport.js';
+import 'eventemitter3';
+import '../types/base.js';
+import '../context/copilot-context.js';
+import '../types/annotated-function.js';
+import '../types/document-pointer.js';
+import 'react';
+import '../hooks/use-tree.js';
+import 'openai/resources/chat';
+
+interface ChatCompletionStreamConfiguration {
+    url: string;
+    model?: string;
+}
+declare class ChatCompletionStream {
+    private url;
+    constructor(params: ChatCompletionStreamConfiguration);
+    fetch(params: ChatCompletionTransportFetchParams): Promise<ReadableStream<string>>;
+}
+
+export { ChatCompletionStream, ChatCompletionStreamConfiguration };

package/dist/openai/chat-completion-stream.mjs

@@ -0,0 +1,221 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
+var __async = (__this, __arguments, generator) => {
+  return new Promise((resolve, reject) => {
+    var fulfilled = (value) => {
+      try {
+        step(generator.next(value));
+      } catch (e) {
+        reject(e);
+      }
+    };
+    var rejected = (value) => {
+      try {
+        step(generator.throw(value));
+      } catch (e) {
+        reject(e);
+      }
+    };
+    var step = (x) => x.done ? resolve(x.value) : Promise.resolve(x.value).then(fulfilled, rejected);
+    step((generator = generator.apply(__this, __arguments)).next());
+  });
+};
+
+// src/openai/chat-completion-transport.ts
+import EventEmitter from "eventemitter3";
+var DEFAULT_MODEL = "gpt-4-1106-preview";
+var ChatCompletionTransport = class extends EventEmitter {
+  constructor({ url }) {
+    super();
+    this.buffer = new Uint8Array();
+    this.bodyReader = null;
+    this.url = url;
+  }
+  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* ({
+      model,
+      messages,
+      copilotConfig,
+      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(this.url, {
+          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;
+    });
+  }
+};
+
+// 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({
+        url: this.url
+      });
+      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=chat-completion-stream.mjs.map

package/dist/openai/chat-completion-stream.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/openai/chat-completion-transport.ts","../../src/openai/chat-completion-stream.ts"],"sourcesContent":["import EventEmitter from \"eventemitter3\";\nimport { Message, Function } from \"../types\";\nimport { CopilotApiConfig } from \"../context\";\n\nexport interface ChatCompletionTransportConfiguration {\n  url: string;\n}\n\ninterface ChatCompletionTransportEvents {\n  end: void;\n  data: any;\n  error: any;\n}\n\nexport interface ChatCompletionTransportFetchParams {\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  copilotConfig: CopilotApiConfig;\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  private url: string;\n\n  constructor({ url }: ChatCompletionTransportConfiguration) {\n    super();\n    this.url = url;\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    model,\n    messages,\n    copilotConfig,\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(this.url, {\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","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      url: this.url,\n    });\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"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,kBAAkB;AA0BzB,IAAM,gBAAgB;AAEf,IAAM,0BAAN,cAAsC,aAA4C;AAAA,EAKvF,YAAY,EAAE,IAAI,GAAyC;AACzD,UAAM;AALR,SAAQ,SAAS,IAAI,WAAW;AAChC,SAAQ,aAA6D;AAKnE,SAAK,MAAM;AAAA,EACb;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,KAAK,KAAK;AAAA,UACrC,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;;;ACnLO,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;AAAA,QAC5C,KAAK,KAAK;AAAA,MACZ,CAAC;AAED,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;","names":[]}

package/dist/openai/chat-completion-transport.d.ts

@@ -0,0 +1,40 @@
+import EventEmitter from 'eventemitter3';
+import { Message, Function } from '../types/base.js';
+import { CopilotApiConfig } from '../context/copilot-context.js';
+import '../types/annotated-function.js';
+import '../types/document-pointer.js';
+import 'react';
+import '../hooks/use-tree.js';
+import 'openai/resources/chat';
+
+interface ChatCompletionTransportConfiguration {
+    url: string;
+}
+interface ChatCompletionTransportEvents {
+    end: void;
+    data: any;
+    error: any;
+}
+interface ChatCompletionTransportFetchParams {
+    model?: string;
+    messages: Message[];
+    functions?: Function[];
+    temperature?: number;
+    maxTokens?: number;
+    headers?: Record<string, string> | Headers;
+    body?: object;
+    copilotConfig: CopilotApiConfig;
+    signal?: AbortSignal;
+}
+declare class ChatCompletionTransport extends EventEmitter<ChatCompletionTransportEvents> {
+    private buffer;
+    private bodyReader;
+    private url;
+    constructor({ url }: ChatCompletionTransportConfiguration);
+    private cleanup;
+    fetch({ model, messages, copilotConfig, functions, temperature, headers, body, signal, }: ChatCompletionTransportFetchParams): Promise<void>;
+    private streamBody;
+    private processData;
+}
+
+export { ChatCompletionTransport, ChatCompletionTransportConfiguration, ChatCompletionTransportFetchParams };

package/dist/openai/chat-completion-transport.mjs

@@ -0,0 +1,181 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
+var __async = (__this, __arguments, generator) => {
+  return new Promise((resolve, reject) => {
+    var fulfilled = (value) => {
+      try {
+        step(generator.next(value));
+      } catch (e) {
+        reject(e);
+      }
+    };
+    var rejected = (value) => {
+      try {
+        step(generator.throw(value));
+      } catch (e) {
+        reject(e);
+      }
+    };
+    var step = (x) => x.done ? resolve(x.value) : Promise.resolve(x.value).then(fulfilled, rejected);
+    step((generator = generator.apply(__this, __arguments)).next());
+  });
+};
+
+// src/openai/chat-completion-transport.ts
+import EventEmitter from "eventemitter3";
+var DEFAULT_MODEL = "gpt-4-1106-preview";
+var ChatCompletionTransport = class extends EventEmitter {
+  constructor({ url }) {
+    super();
+    this.buffer = new Uint8Array();
+    this.bodyReader = null;
+    this.url = url;
+  }
+  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* ({
+      model,
+      messages,
+      copilotConfig,
+      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(this.url, {
+          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=chat-completion-transport.mjs.map

package/dist/openai/chat-completion-transport.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/openai/chat-completion-transport.ts"],"sourcesContent":["import EventEmitter from \"eventemitter3\";\nimport { Message, Function } from \"../types\";\nimport { CopilotApiConfig } from \"../context\";\n\nexport interface ChatCompletionTransportConfiguration {\n  url: string;\n}\n\ninterface ChatCompletionTransportEvents {\n  end: void;\n  data: any;\n  error: any;\n}\n\nexport interface ChatCompletionTransportFetchParams {\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  copilotConfig: CopilotApiConfig;\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  private url: string;\n\n  constructor({ url }: ChatCompletionTransportConfiguration) {\n    super();\n    this.url = url;\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    model,\n    messages,\n    copilotConfig,\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(this.url, {\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"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,kBAAkB;AA0BzB,IAAM,gBAAgB;AAEf,IAAM,0BAAN,cAAsC,aAA4C;AAAA,EAKvF,YAAY,EAAE,IAAI,GAAyC;AACzD,UAAM;AALR,SAAQ,SAAS,IAAI,WAAW;AAChC,SAAQ,aAA6D;AAKnE,SAAK,MAAM;AAAA,EACb;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,KAAK,KAAK;AAAA,UACrC,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;","names":[]}

package/dist/openai/index.d.ts

@@ -0,0 +1,10 @@
+export { ChatCompletionStream, ChatCompletionStreamConfiguration } from './chat-completion-stream.js';
+export { ChatCompletionTransportFetchParams } from './chat-completion-transport.js';
+import 'eventemitter3';
+import '../types/base.js';
+import '../context/copilot-context.js';
+import '../types/annotated-function.js';
+import '../types/document-pointer.js';
+import 'react';
+import '../hooks/use-tree.js';
+import 'openai/resources/chat';