@copilotkit/react-textarea 1.1.3-feat-runtime-remote-actions.2 → 1.2.1

package/dist/hooks/base-copilot-textarea-implementation/use-populate-copilot-textarea-ref.mjs

@@ -1,9 +1,144 @@
-import {
-  usePopulateCopilotTextareaRef
-} from "../../chunk-QJNFNPWH.mjs";
-import "../../chunk-5UNJXFUO.mjs";
-import "../../chunk-ECR45NSD.mjs";
-import "../../chunk-MRXNTQOX.mjs";
+// src/hooks/base-copilot-textarea-implementation/use-populate-copilot-textarea-ref.ts
+import React from "react";
+import { ReactEditor } from "slate-react";
+
+// src/lib/get-text-around-cursor.ts
+import { Editor, Path, Range, Text, Element, Point } from "slate";
+function getFullEditorTextWithNewlines(editor) {
+  const fullDocumentRange = {
+    anchor: Editor.start(editor, []),
+    focus: Editor.end(editor, [])
+  };
+  return extractTextWithNewlines(editor, fullDocumentRange);
+}
+function extractTextWithNewlines(editor, range) {
+  const voids = false;
+  const [start, end] = Range.edges(range);
+  let text = "";
+  let lastBlock = null;
+  for (const [node, path] of Editor.nodes(editor, {
+    at: range,
+    match: Text.isText,
+    voids
+  })) {
+    let t = node.text;
+    const [block] = Editor.above(editor, {
+      at: path,
+      match: (n) => Element.isElement(n) && n.type === "paragraph"
+    }) || [null];
+    if (lastBlock !== block && block) {
+      if (lastBlock) {
+        text += "\n";
+      }
+      lastBlock = block;
+    }
+    if (Path.equals(path, end.path)) {
+      t = t.slice(0, end.offset);
+    }
+    if (Path.equals(path, start.path)) {
+      t = t.slice(start.offset);
+    }
+    text += t;
+  }
+  return text;
+}
+
+// src/lib/slatejs-edits/replace-text.ts
+import { Editor as Editor2, Transforms } from "slate";
+function replaceEditorText(editor, newText) {
+  Transforms.delete(editor, {
+    at: {
+      anchor: Editor2.start(editor, []),
+      focus: Editor2.end(editor, [])
+    }
+  });
+  if (newText && newText !== "") {
+    Transforms.insertNodes(
+      editor,
+      [
+        {
+          type: "paragraph",
+          children: [{ text: newText }]
+        }
+      ],
+      {
+        at: [0]
+      }
+    );
+  }
+}
+
+// src/hooks/base-copilot-textarea-implementation/use-populate-copilot-textarea-ref.ts
+function usePopulateCopilotTextareaRef(editor, ref) {
+  React.useImperativeHandle(ref, () => {
+    class Combined {
+      constructor(customMethods2, editorHtmlElement2) {
+        this.customMethods = customMethods2;
+        this.editorHtmlElement = editorHtmlElement2;
+      }
+      get(target, propKey) {
+        if (this.isKeyOfCustomMethods(propKey)) {
+          const value = this.customMethods[propKey];
+          if (typeof value === "function") {
+            return value.bind(this.customMethods);
+          }
+          return value;
+        } else if (this.isKeyOfHTMLElement(propKey)) {
+          const value = this.editorHtmlElement[propKey];
+          if (typeof value === "function") {
+            return value.bind(this.editorHtmlElement);
+          }
+          return value;
+        }
+      }
+      set(target, propKey, value) {
+        if (this.isKeyOfCustomMethods(propKey)) {
+          this.customMethods[propKey] = value;
+        } else if (this.isKeyOfHTMLElement(propKey)) {
+          this.editorHtmlElement[propKey] = value;
+        } else {
+          target[propKey] = value;
+        }
+        return true;
+      }
+      isKeyOfCustomMethods(key) {
+        return key in this.customMethods;
+      }
+      isKeyOfHTMLElement(key) {
+        return key in this.editorHtmlElement;
+      }
+    }
+    const handler = {
+      get(target, propKey) {
+        return target.get(target, propKey);
+      },
+      set(target, propKey, value) {
+        return target.set(target, propKey, value);
+      }
+    };
+    class CustomMethods {
+      constructor(editor2) {
+        this.editor = editor2;
+      }
+      focus() {
+        ReactEditor.focus(this.editor);
+      }
+      blur() {
+        ReactEditor.blur(this.editor);
+      }
+      get value() {
+        return getFullEditorTextWithNewlines(this.editor);
+      }
+      set value(value) {
+        replaceEditorText(this.editor, value);
+      }
+    }
+    const editorHtmlElement = ReactEditor.toDOMNode(editor, editor);
+    const customMethods = new CustomMethods(editor);
+    const combined = new Combined(customMethods, editorHtmlElement);
+    return new Proxy(combined, handler);
+  }, [editor]);
+}
 export {
   usePopulateCopilotTextareaRef
 };

package/dist/hooks/base-copilot-textarea-implementation/use-populate-copilot-textarea-ref.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}
+{"version":3,"sources":["../../../src/hooks/base-copilot-textarea-implementation/use-populate-copilot-textarea-ref.ts","../../../src/lib/get-text-around-cursor.ts","../../../src/lib/slatejs-edits/replace-text.ts"],"sourcesContent":["import React from \"react\";\nimport { Editor } from \"slate\";\nimport { ReactEditor } from \"slate-react\";\nimport { getFullEditorTextWithNewlines } from \"../../lib/get-text-around-cursor\";\nimport { replaceEditorText } from \"../../lib/slatejs-edits/replace-text\";\nimport { HTMLCopilotTextAreaElement } from \"../../types\";\nimport { CustomEditor } from \"../../types/base/custom-editor\";\n\nexport function usePopulateCopilotTextareaRef(\n  editor: Editor,\n  ref: React.Ref<HTMLCopilotTextAreaElement>,\n) {\n  React.useImperativeHandle(ref, () => {\n    class Combined {\n      constructor(\n        private customMethods: CustomMethods,\n        private editorHtmlElement: HTMLElement,\n      ) {}\n\n      [key: string]: any;\n\n      get(target: any, propKey: string): any {\n        if (this.isKeyOfCustomMethods(propKey)) {\n          const value = this.customMethods[propKey];\n          if (typeof value === \"function\") {\n            return value.bind(this.customMethods);\n          }\n          return value;\n        } else if (this.isKeyOfHTMLElement(propKey)) {\n          const value = this.editorHtmlElement[propKey];\n          if (typeof value === \"function\") {\n            return value.bind(this.editorHtmlElement);\n          }\n          return value;\n        }\n      }\n\n      set(target: any, propKey: string, value: any): boolean {\n        if (this.isKeyOfCustomMethods(propKey)) {\n          (this.customMethods as any)[propKey] = value;\n        } else if (this.isKeyOfHTMLElement(propKey)) {\n          (this.editorHtmlElement as any)[propKey] = value;\n        } else {\n          // Default behavior (optional)\n          target[propKey] = value;\n        }\n        return true;\n      }\n\n      private isKeyOfCustomMethods(key: string): key is keyof CustomMethods {\n        return key in this.customMethods;\n      }\n\n      private isKeyOfHTMLElement(key: string): key is keyof HTMLElement {\n        return key in this.editorHtmlElement;\n      }\n    }\n\n    const handler = {\n      get(target: any, propKey: keyof CustomMethods | keyof HTMLElement) {\n        return target.get(target, propKey);\n      },\n      set(target: any, propKey: keyof CustomMethods | keyof HTMLElement, value: any) {\n        return target.set(target, propKey, value);\n      },\n    };\n\n    class CustomMethods {\n      constructor(private editor: CustomEditor) {}\n\n      focus() {\n        ReactEditor.focus(this.editor);\n      }\n\n      blur() {\n        ReactEditor.blur(this.editor);\n      }\n\n      get value() {\n        return getFullEditorTextWithNewlines(this.editor);\n      }\n      set value(value: string) {\n        replaceEditorText(this.editor, value);\n      }\n    }\n\n    const editorHtmlElement = ReactEditor.toDOMNode(editor, editor);\n    const customMethods = new CustomMethods(editor);\n\n    const combined = new Combined(customMethods, editorHtmlElement);\n    return new Proxy(combined, handler);\n  }, [editor]);\n}\n","import { Editor, Node, Path, Range, Text, Element, BasePoint, BaseRange, Point } from \"slate\";\nimport { EditorAutocompleteState } from \"../types/base/editor-autocomplete-state\";\n\nexport interface EditorTextState {\n  selection: BaseRange;\n\n  textBeforeCursor: string;\n  selectedText: string;\n  textAfterCursor: string;\n}\n\nexport function getTextAroundCollapsedCursor(editor: Editor): EditorAutocompleteState | null {\n  const { selection } = editor;\n  if (!selection || !Range.isCollapsed(selection)) {\n    return null;\n  }\n\n  const cursorPoint = selection.anchor;\n\n  // Create two ranges: one before the anchor and one after\n  const beforeRange: Range = {\n    anchor: Editor.start(editor, []),\n    focus: cursorPoint,\n  };\n  const afterRange: Range = {\n    anchor: cursorPoint,\n    focus: Editor.end(editor, []),\n  };\n\n  // Extract text for these ranges\n  const before = extractTextWithNewlines(editor, beforeRange);\n  const after = extractTextWithNewlines(editor, afterRange);\n\n  return {\n    cursorPoint: cursorPoint,\n    textBeforeCursor: before,\n    textAfterCursor: after,\n  };\n}\n\nexport function getTextAroundSelection(editor: Editor): EditorTextState | null {\n  const { selection } = editor;\n  if (!selection) {\n    return null;\n  }\n\n  const wellOrderedSelection = wellOrderedRange(selection);\n\n  // Create two ranges: one before the anchor and one after\n  const beforeRange: Range = {\n    anchor: Editor.start(editor, []),\n    focus: wellOrderedSelection.anchor,\n  };\n  const afterRange: Range = {\n    anchor: wellOrderedSelection.focus,\n    focus: Editor.end(editor, []),\n  };\n\n  // Extract text for these ranges\n  const before = extractTextWithNewlines(editor, beforeRange);\n  const after = extractTextWithNewlines(editor, afterRange);\n  const selectedText = extractTextWithNewlines(editor, wellOrderedSelection);\n\n  return {\n    selection: wellOrderedSelection,\n    textBeforeCursor: before,\n    selectedText,\n    textAfterCursor: after,\n  };\n}\n\nexport function getFullEditorTextWithNewlines(editor: Editor): string {\n  const fullDocumentRange: Range = {\n    anchor: Editor.start(editor, []),\n    focus: Editor.end(editor, []),\n  };\n  return extractTextWithNewlines(editor, fullDocumentRange);\n}\n\n// Helper function to extract text with newlines\nexport function extractTextWithNewlines(editor: Editor, range: Range): string {\n  const voids = false;\n  const [start, end] = Range.edges(range);\n  let text = \"\";\n  let lastBlock: Node | null = null;\n\n  for (const [node, path] of Editor.nodes(editor, {\n    at: range,\n    match: Text.isText,\n    voids,\n  })) {\n    let t = node.text;\n\n    // Determine the parent block of the current text node\n    const [block] = Editor.above(editor, {\n      at: path,\n      match: (n) => Element.isElement(n) && n.type === \"paragraph\",\n    }) || [null];\n\n    // If we encounter a new block, prepend a newline\n    if (lastBlock !== block && block) {\n      // check that lastBlock is not null to avoid adding a newline at the beginning\n      if (lastBlock) {\n        text += \"\\n\";\n      }\n      lastBlock = block;\n    }\n\n    if (Path.equals(path, end.path)) {\n      t = t.slice(0, end.offset);\n    }\n\n    if (Path.equals(path, start.path)) {\n      t = t.slice(start.offset);\n    }\n\n    text += t;\n  }\n\n  return text;\n}\n\nfunction wellOrderedRange(range: BaseRange): BaseRange {\n  const { anchor, focus } = range;\n  // if anchor is before focus, return range as is\n  if (Point.isBefore(anchor, focus)) {\n    return range;\n  }\n\n  // if focus is before anchor, return range with anchor and focus swapped\n  return {\n    anchor: focus,\n    focus: anchor,\n  };\n}\n","import { Editor, Transforms } from \"slate\";\n\nexport function replaceEditorText(editor: Editor, newText: string) {\n  // clear all previous text\n  Transforms.delete(editor, {\n    at: {\n      anchor: Editor.start(editor, []),\n      focus: Editor.end(editor, []),\n    },\n  });\n\n  // insert new text\n  if (newText && newText !== \"\") {\n    // don't insert empty text - results in strange visual behavior\n    Transforms.insertNodes(\n      editor,\n      [\n        {\n          type: \"paragraph\",\n          children: [{ text: newText }],\n        },\n      ],\n      {\n        at: [0],\n      },\n    );\n  }\n}\n"],"mappings":";AAAA,OAAO,WAAW;AAElB,SAAS,mBAAmB;;;ACF5B,SAAS,QAAc,MAAM,OAAO,MAAM,SAA+B,aAAa;AAuE/E,SAAS,8BAA8B,QAAwB;AACpE,QAAM,oBAA2B;AAAA,IAC/B,QAAQ,OAAO,MAAM,QAAQ,CAAC,CAAC;AAAA,IAC/B,OAAO,OAAO,IAAI,QAAQ,CAAC,CAAC;AAAA,EAC9B;AACA,SAAO,wBAAwB,QAAQ,iBAAiB;AAC1D;AAGO,SAAS,wBAAwB,QAAgB,OAAsB;AAC5E,QAAM,QAAQ;AACd,QAAM,CAAC,OAAO,GAAG,IAAI,MAAM,MAAM,KAAK;AACtC,MAAI,OAAO;AACX,MAAI,YAAyB;AAE7B,aAAW,CAAC,MAAM,IAAI,KAAK,OAAO,MAAM,QAAQ;AAAA,IAC9C,IAAI;AAAA,IACJ,OAAO,KAAK;AAAA,IACZ;AAAA,EACF,CAAC,GAAG;AACF,QAAI,IAAI,KAAK;AAGb,UAAM,CAAC,KAAK,IAAI,OAAO,MAAM,QAAQ;AAAA,MACnC,IAAI;AAAA,MACJ,OAAO,CAAC,MAAM,QAAQ,UAAU,CAAC,KAAK,EAAE,SAAS;AAAA,IACnD,CAAC,KAAK,CAAC,IAAI;AAGX,QAAI,cAAc,SAAS,OAAO;AAEhC,UAAI,WAAW;AACb,gBAAQ;AAAA,MACV;AACA,kBAAY;AAAA,IACd;AAEA,QAAI,KAAK,OAAO,MAAM,IAAI,IAAI,GAAG;AAC/B,UAAI,EAAE,MAAM,GAAG,IAAI,MAAM;AAAA,IAC3B;AAEA,QAAI,KAAK,OAAO,MAAM,MAAM,IAAI,GAAG;AACjC,UAAI,EAAE,MAAM,MAAM,MAAM;AAAA,IAC1B;AAEA,YAAQ;AAAA,EACV;AAEA,SAAO;AACT;;;ACxHA,SAAS,UAAAA,SAAQ,kBAAkB;AAE5B,SAAS,kBAAkB,QAAgB,SAAiB;AAEjE,aAAW,OAAO,QAAQ;AAAA,IACxB,IAAI;AAAA,MACF,QAAQA,QAAO,MAAM,QAAQ,CAAC,CAAC;AAAA,MAC/B,OAAOA,QAAO,IAAI,QAAQ,CAAC,CAAC;AAAA,IAC9B;AAAA,EACF,CAAC;AAGD,MAAI,WAAW,YAAY,IAAI;AAE7B,eAAW;AAAA,MACT;AAAA,MACA;AAAA,QACE;AAAA,UACE,MAAM;AAAA,UACN,UAAU,CAAC,EAAE,MAAM,QAAQ,CAAC;AAAA,QAC9B;AAAA,MACF;AAAA,MACA;AAAA,QACE,IAAI,CAAC,CAAC;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACF;;;AFnBO,SAAS,8BACd,QACA,KACA;AACA,QAAM,oBAAoB,KAAK,MAAM;AAAA,IACnC,MAAM,SAAS;AAAA,MACb,YACUC,gBACAC,oBACR;AAFQ,6BAAAD;AACA,iCAAAC;AAAA,MACP;AAAA,MAIH,IAAI,QAAa,SAAsB;AACrC,YAAI,KAAK,qBAAqB,OAAO,GAAG;AACtC,gBAAM,QAAQ,KAAK,cAAc,OAAO;AACxC,cAAI,OAAO,UAAU,YAAY;AAC/B,mBAAO,MAAM,KAAK,KAAK,aAAa;AAAA,UACtC;AACA,iBAAO;AAAA,QACT,WAAW,KAAK,mBAAmB,OAAO,GAAG;AAC3C,gBAAM,QAAQ,KAAK,kBAAkB,OAAO;AAC5C,cAAI,OAAO,UAAU,YAAY;AAC/B,mBAAO,MAAM,KAAK,KAAK,iBAAiB;AAAA,UAC1C;AACA,iBAAO;AAAA,QACT;AAAA,MACF;AAAA,MAEA,IAAI,QAAa,SAAiB,OAAqB;AACrD,YAAI,KAAK,qBAAqB,OAAO,GAAG;AACtC,UAAC,KAAK,cAAsB,OAAO,IAAI;AAAA,QACzC,WAAW,KAAK,mBAAmB,OAAO,GAAG;AAC3C,UAAC,KAAK,kBAA0B,OAAO,IAAI;AAAA,QAC7C,OAAO;AAEL,iBAAO,OAAO,IAAI;AAAA,QACpB;AACA,eAAO;AAAA,MACT;AAAA,MAEQ,qBAAqB,KAAyC;AACpE,eAAO,OAAO,KAAK;AAAA,MACrB;AAAA,MAEQ,mBAAmB,KAAuC;AAChE,eAAO,OAAO,KAAK;AAAA,MACrB;AAAA,IACF;AAEA,UAAM,UAAU;AAAA,MACd,IAAI,QAAa,SAAkD;AACjE,eAAO,OAAO,IAAI,QAAQ,OAAO;AAAA,MACnC;AAAA,MACA,IAAI,QAAa,SAAkD,OAAY;AAC7E,eAAO,OAAO,IAAI,QAAQ,SAAS,KAAK;AAAA,MAC1C;AAAA,IACF;AAAA,IAEA,MAAM,cAAc;AAAA,MAClB,YAAoBC,SAAsB;AAAtB,sBAAAA;AAAA,MAAuB;AAAA,MAE3C,QAAQ;AACN,oBAAY,MAAM,KAAK,MAAM;AAAA,MAC/B;AAAA,MAEA,OAAO;AACL,oBAAY,KAAK,KAAK,MAAM;AAAA,MAC9B;AAAA,MAEA,IAAI,QAAQ;AACV,eAAO,8BAA8B,KAAK,MAAM;AAAA,MAClD;AAAA,MACA,IAAI,MAAM,OAAe;AACvB,0BAAkB,KAAK,QAAQ,KAAK;AAAA,MACtC;AAAA,IACF;AAEA,UAAM,oBAAoB,YAAY,UAAU,QAAQ,MAAM;AAC9D,UAAM,gBAAgB,IAAI,cAAc,MAAM;AAE9C,UAAM,WAAW,IAAI,SAAS,eAAe,iBAAiB;AAC9D,WAAO,IAAI,MAAM,UAAU,OAAO;AAAA,EACpC,GAAG,CAAC,MAAM,CAAC;AACb;","names":["Editor","customMethods","editorHtmlElement","editor"]}

package/dist/hooks/index.d.mts

@@ -0,0 +1,2 @@
+
+export {  }

package/dist/hooks/index.d.ts

@@ -1,2 +1,2 @@
 
-export { }
+export {  }

package/dist/hooks/index.mjs

@@ -1,2 +1 @@
-import "../chunk-H4VKQGVU.mjs";
 //# sourceMappingURL=index.mjs.map

package/dist/hooks/make-autosuggestions-function/use-make-standard-autosuggestions-function.d.mts

@@ -0,0 +1,22 @@
+import { AutosuggestionsBareFunction } from '../../types/base/autosuggestions-bare-function.mjs';
+import { SuggestionsApiConfig } from '../../types/autosuggestions-config/suggestions-api-config.mjs';
+import '@copilotkit/react-core';
+import '@copilotkit/runtime-client-gql';
+import '../../types/autosuggestions-config/subtypes/make-system-prompt.mjs';
+
+/**
+ * Returns a memoized function that sends a request to the specified API endpoint to get an autosuggestion for the user's input.
+ * The function takes in the text before and after the cursor, and an abort signal.
+ * It sends a POST request to the API endpoint with the messages array containing the system message, few shot messages, and user messages.
+ * The function returns the suggestion from the API response.
+ *
+ * @param textareaPurpose - The purpose of the textarea. This is included in the system message.
+ * @param apiEndpoint - The API endpoint to send the autosuggestion request to.
+ * @param makeSystemMessage - A function that takes in a context string and returns a system message to include in the autosuggestion request.
+ * @param fewShotMessages - An array of few shot messages to include in the autosuggestion request.
+ * @param contextCategories - The categories of context strings we want to include. By default, we include the (default) "global" context category.
+ * @returns A memoized function that sends a request to the specified API endpoint to get an autosuggestion for the user's input.
+ */
+declare function useMakeStandardAutosuggestionFunction(textareaPurpose: string, contextCategories: string[], apiConfig: SuggestionsApiConfig): AutosuggestionsBareFunction;
+
+export { useMakeStandardAutosuggestionFunction };

package/dist/hooks/make-autosuggestions-function/use-make-standard-autosuggestions-function.js

@@ -121,7 +121,7 @@ function useMakeStandardAutosuggestionFunction(textareaPurpose, contextCategorie
               actions: [],
               url: window.location.href
             },
-            messages: (0, import_runtime_client_gql.convertMessagesToGqlInput)((0, import_runtime_client_gql.filterAgentStateMessages)(messages)),
+            messages: (0, import_runtime_client_gql.convertMessagesToGqlInput)(messages),
             metadata: {
               requestType: import_runtime_client_gql.CopilotRequestType.TextareaCompletion
             },
@@ -142,6 +142,7 @@ function useMakeStandardAutosuggestionFunction(textareaPurpose, contextCategorie
           }
           if (message instanceof import_runtime_client_gql.TextMessage) {
             result += message.content;
+            console.log(message.content);
           }
         }
         return result;

package/dist/hooks/make-autosuggestions-function/use-make-standard-autosuggestions-function.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/hooks/make-autosuggestions-function/use-make-standard-autosuggestions-function.tsx","../../../src/lib/retry.tsx"],"sourcesContent":["import { COPILOT_CLOUD_PUBLIC_API_KEY_HEADER } from \"@copilotkit/shared\";\nimport { useCopilotContext } from \"@copilotkit/react-core\";\nimport { useCallback } from \"react\";\nimport { AutosuggestionsBareFunction } from \"../../types\";\nimport { retry } from \"../../lib/retry\";\nimport { InsertionEditorState } from \"../../types/base/autosuggestions-bare-function\";\nimport { SuggestionsApiConfig } from \"../../types/autosuggestions-config/suggestions-api-config\";\nimport {\n  CopilotRuntimeClient,\n  Message,\n  Role,\n  TextMessage,\n  convertGqlOutputToMessages,\n  convertMessagesToGqlInput,\n  filterAgentStateMessages,\n  CopilotRequestType,\n} from \"@copilotkit/runtime-client-gql\";\n\n/**\n * Returns a memoized function that sends a request to the specified API endpoint to get an autosuggestion for the user's input.\n * The function takes in the text before and after the cursor, and an abort signal.\n * It sends a POST request to the API endpoint with the messages array containing the system message, few shot messages, and user messages.\n * The function returns the suggestion from the API response.\n *\n * @param textareaPurpose - The purpose of the textarea. This is included in the system message.\n * @param apiEndpoint - The API endpoint to send the autosuggestion request to.\n * @param makeSystemMessage - A function that takes in a context string and returns a system message to include in the autosuggestion request.\n * @param fewShotMessages - An array of few shot messages to include in the autosuggestion request.\n * @param contextCategories - The categories of context strings we want to include. By default, we include the (default) \"global\" context category.\n * @returns A memoized function that sends a request to the specified API endpoint to get an autosuggestion for the user's input.\n */\nexport function useMakeStandardAutosuggestionFunction(\n  textareaPurpose: string,\n  contextCategories: string[],\n  apiConfig: SuggestionsApiConfig,\n): AutosuggestionsBareFunction {\n  const { getContextString, copilotApiConfig } = useCopilotContext();\n  const { chatApiEndpoint: url, publicApiKey, credentials, properties } = copilotApiConfig;\n  const headers = {\n    ...copilotApiConfig.headers,\n    ...(publicApiKey ? { [COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {}),\n  };\n  const { maxTokens, stop } = apiConfig;\n\n  return useCallback(\n    async (editorState: InsertionEditorState, abortSignal: AbortSignal) => {\n      const res = await retry(async () => {\n        const messages: Message[] = [\n          new TextMessage({\n            role: Role.System,\n            content: apiConfig.makeSystemPrompt(\n              textareaPurpose,\n              getContextString([], contextCategories),\n            ),\n          }),\n          ...apiConfig.fewShotMessages,\n          new TextMessage({\n            role: Role.User,\n            content: editorState.textAfterCursor,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<TextAfterCursor>${editorState.textAfterCursor}</TextAfterCursor>`,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<TextBeforeCursor>${editorState.textBeforeCursor}</TextBeforeCursor>`,\n          }),\n        ];\n\n        const runtimeClient = new CopilotRuntimeClient({\n          url,\n          publicApiKey,\n          headers,\n          credentials,\n        });\n\n        const response = await runtimeClient\n          .generateCopilotResponse({\n            data: {\n              frontend: {\n                actions: [],\n                url: window.location.href,\n              },\n              messages: convertMessagesToGqlInput(filterAgentStateMessages(messages)),\n              metadata: {\n                requestType: CopilotRequestType.TextareaCompletion,\n              },\n              forwardedParameters: {\n                maxTokens,\n                stop,\n              },\n            },\n            properties,\n            signal: abortSignal,\n          })\n          .toPromise();\n\n        let result = \"\";\n        for (const message of convertGqlOutputToMessages(\n          response.data?.generateCopilotResponse?.messages ?? [],\n        )) {\n          if (abortSignal.aborted) {\n            break;\n          }\n          if (message instanceof TextMessage) {\n            result += message.content;\n          }\n        }\n\n        return result;\n      });\n\n      return res;\n    },\n    [apiConfig, getContextString, contextCategories, textareaPurpose],\n  );\n}\n","export function retry<T>(\n  fn: () => Promise<T>,\n  retriesLeft: number = 2,\n  interval: number = 200,\n  backoff: number = 1.5,\n): Promise<T> {\n  return new Promise((resolve, reject) => {\n    fn()\n      .then(resolve)\n      .catch((error) => {\n        if (retriesLeft === 1) {\n          reject(error);\n          return;\n        }\n\n        setTimeout(() => {\n          retry(fn, retriesLeft - 1, interval * backoff, backoff)\n            .then(resolve)\n            .catch(reject);\n        }, interval);\n      });\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,oBAAoD;AACpD,wBAAkC;AAClC,mBAA4B;;;ACFrB,SAAS,MACd,IACA,cAAsB,GACtB,WAAmB,KACnB,UAAkB,KACN;AACZ,SAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,OAAG,EACA,KAAK,OAAO,EACZ,MAAM,CAAC,UAAU;AAChB,UAAI,gBAAgB,GAAG;AACrB,eAAO,KAAK;AACZ;AAAA,MACF;AAEA,iBAAW,MAAM;AACf,cAAM,IAAI,cAAc,GAAG,WAAW,SAAS,OAAO,EACnD,KAAK,OAAO,EACZ,MAAM,MAAM;AAAA,MACjB,GAAG,QAAQ;AAAA,IACb,CAAC;AAAA,EACL,CAAC;AACH;;;ADfA,gCASO;AAeA,SAAS,sCACd,iBACA,mBACA,WAC6B;AAC7B,QAAM,EAAE,kBAAkB,iBAAiB,QAAI,qCAAkB;AACjE,QAAM,EAAE,iBAAiB,KAAK,cAAc,aAAa,WAAW,IAAI;AACxE,QAAM,UAAU,kCACX,iBAAiB,UAChB,eAAe,EAAE,CAAC,iDAAmC,GAAG,aAAa,IAAI,CAAC;AAEhF,QAAM,EAAE,WAAW,KAAK,IAAI;AAE5B,aAAO;AAAA,IACL,CAAO,aAAmC,gBAA6B;AACrE,YAAM,MAAM,MAAM,MAAM,MAAY;AA9C1C;AA+CQ,cAAM,WAAsB;AAAA,UAC1B,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,UAAU;AAAA,cACjB;AAAA,cACA,iBAAiB,CAAC,GAAG,iBAAiB;AAAA,YACxC;AAAA,UACF,CAAC;AAAA,UACD,GAAG,UAAU;AAAA,UACb,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,YAAY;AAAA,UACvB,CAAC;AAAA,UACD,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,oBAAoB,YAAY;AAAA,UAC3C,CAAC;AAAA,UACD,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,qBAAqB,YAAY;AAAA,UAC5C,CAAC;AAAA,QACH;AAEA,cAAM,gBAAgB,IAAI,+CAAqB;AAAA,UAC7C;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,QACF,CAAC;AAED,cAAM,WAAW,MAAM,cACpB,wBAAwB;AAAA,UACvB,MAAM;AAAA,YACJ,UAAU;AAAA,cACR,SAAS,CAAC;AAAA,cACV,KAAK,OAAO,SAAS;AAAA,YACvB;AAAA,YACA,cAAU,yDAA0B,oDAAyB,QAAQ,CAAC;AAAA,YACtE,UAAU;AAAA,cACR,aAAa,6CAAmB;AAAA,YAClC;AAAA,YACA,qBAAqB;AAAA,cACnB;AAAA,cACA;AAAA,YACF;AAAA,UACF;AAAA,UACA;AAAA,UACA,QAAQ;AAAA,QACV,CAAC,EACA,UAAU;AAEb,YAAI,SAAS;AACb,mBAAW,eAAW;AAAA,WACpB,0BAAS,SAAT,mBAAe,4BAAf,mBAAwC,aAAxC,YAAoD,CAAC;AAAA,QACvD,GAAG;AACD,cAAI,YAAY,SAAS;AACvB;AAAA,UACF;AACA,cAAI,mBAAmB,uCAAa;AAClC,sBAAU,QAAQ;AAAA,UACpB;AAAA,QACF;AAEA,eAAO;AAAA,MACT,EAAC;AAED,aAAO;AAAA,IACT;AAAA,IACA,CAAC,WAAW,kBAAkB,mBAAmB,eAAe;AAAA,EAClE;AACF;","names":[]}
+{"version":3,"sources":["../../../src/hooks/make-autosuggestions-function/use-make-standard-autosuggestions-function.tsx","../../../src/lib/retry.tsx"],"sourcesContent":["import { COPILOT_CLOUD_PUBLIC_API_KEY_HEADER } from \"@copilotkit/shared\";\nimport { useCopilotContext } from \"@copilotkit/react-core\";\nimport { useCallback } from \"react\";\nimport { AutosuggestionsBareFunction } from \"../../types\";\nimport { retry } from \"../../lib/retry\";\nimport { InsertionEditorState } from \"../../types/base/autosuggestions-bare-function\";\nimport { SuggestionsApiConfig } from \"../../types/autosuggestions-config/suggestions-api-config\";\nimport {\n  CopilotRuntimeClient,\n  Message,\n  Role,\n  TextMessage,\n  convertGqlOutputToMessages,\n  convertMessagesToGqlInput,\n  CopilotRequestType,\n} from \"@copilotkit/runtime-client-gql\";\n\n/**\n * Returns a memoized function that sends a request to the specified API endpoint to get an autosuggestion for the user's input.\n * The function takes in the text before and after the cursor, and an abort signal.\n * It sends a POST request to the API endpoint with the messages array containing the system message, few shot messages, and user messages.\n * The function returns the suggestion from the API response.\n *\n * @param textareaPurpose - The purpose of the textarea. This is included in the system message.\n * @param apiEndpoint - The API endpoint to send the autosuggestion request to.\n * @param makeSystemMessage - A function that takes in a context string and returns a system message to include in the autosuggestion request.\n * @param fewShotMessages - An array of few shot messages to include in the autosuggestion request.\n * @param contextCategories - The categories of context strings we want to include. By default, we include the (default) \"global\" context category.\n * @returns A memoized function that sends a request to the specified API endpoint to get an autosuggestion for the user's input.\n */\nexport function useMakeStandardAutosuggestionFunction(\n  textareaPurpose: string,\n  contextCategories: string[],\n  apiConfig: SuggestionsApiConfig,\n): AutosuggestionsBareFunction {\n  const { getContextString, copilotApiConfig } = useCopilotContext();\n  const { chatApiEndpoint: url, publicApiKey, credentials, properties } = copilotApiConfig;\n  const headers = {\n    ...copilotApiConfig.headers,\n    ...(publicApiKey ? { [COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {}),\n  };\n  const { maxTokens, stop } = apiConfig;\n\n  return useCallback(\n    async (editorState: InsertionEditorState, abortSignal: AbortSignal) => {\n      const res = await retry(async () => {\n        const messages: Message[] = [\n          new TextMessage({\n            role: Role.System,\n            content: apiConfig.makeSystemPrompt(\n              textareaPurpose,\n              getContextString([], contextCategories),\n            ),\n          }),\n          ...apiConfig.fewShotMessages,\n          new TextMessage({\n            role: Role.User,\n            content: editorState.textAfterCursor,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<TextAfterCursor>${editorState.textAfterCursor}</TextAfterCursor>`,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<TextBeforeCursor>${editorState.textBeforeCursor}</TextBeforeCursor>`,\n          }),\n        ];\n\n        const runtimeClient = new CopilotRuntimeClient({\n          url,\n          publicApiKey,\n          headers,\n          credentials,\n        });\n\n        const response = await runtimeClient\n          .generateCopilotResponse({\n            data: {\n              frontend: {\n                actions: [],\n                url: window.location.href,\n              },\n              messages: convertMessagesToGqlInput(messages),\n              metadata: {\n                requestType: CopilotRequestType.TextareaCompletion,\n              },\n              forwardedParameters: {\n                maxTokens,\n                stop,\n              },\n            },\n            properties,\n            signal: abortSignal,\n          })\n          .toPromise();\n\n        let result = \"\";\n        for (const message of convertGqlOutputToMessages(\n          response.data?.generateCopilotResponse?.messages ?? [],\n        )) {\n          if (abortSignal.aborted) {\n            break;\n          }\n          if (message instanceof TextMessage) {\n            result += message.content;\n            console.log(message.content);\n          }\n        }\n\n        return result;\n      });\n\n      return res;\n    },\n    [apiConfig, getContextString, contextCategories, textareaPurpose],\n  );\n}\n","export function retry<T>(\n  fn: () => Promise<T>,\n  retriesLeft: number = 2,\n  interval: number = 200,\n  backoff: number = 1.5,\n): Promise<T> {\n  return new Promise((resolve, reject) => {\n    fn()\n      .then(resolve)\n      .catch((error) => {\n        if (retriesLeft === 1) {\n          reject(error);\n          return;\n        }\n\n        setTimeout(() => {\n          retry(fn, retriesLeft - 1, interval * backoff, backoff)\n            .then(resolve)\n            .catch(reject);\n        }, interval);\n      });\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,oBAAoD;AACpD,wBAAkC;AAClC,mBAA4B;;;ACFrB,SAAS,MACd,IACA,cAAsB,GACtB,WAAmB,KACnB,UAAkB,KACN;AACZ,SAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,OAAG,EACA,KAAK,OAAO,EACZ,MAAM,CAAC,UAAU;AAChB,UAAI,gBAAgB,GAAG;AACrB,eAAO,KAAK;AACZ;AAAA,MACF;AAEA,iBAAW,MAAM;AACf,cAAM,IAAI,cAAc,GAAG,WAAW,SAAS,OAAO,EACnD,KAAK,OAAO,EACZ,MAAM,MAAM;AAAA,MACjB,GAAG,QAAQ;AAAA,IACb,CAAC;AAAA,EACL,CAAC;AACH;;;ADfA,gCAQO;AAeA,SAAS,sCACd,iBACA,mBACA,WAC6B;AAC7B,QAAM,EAAE,kBAAkB,iBAAiB,QAAI,qCAAkB;AACjE,QAAM,EAAE,iBAAiB,KAAK,cAAc,aAAa,WAAW,IAAI;AACxE,QAAM,UAAU,kCACX,iBAAiB,UAChB,eAAe,EAAE,CAAC,iDAAmC,GAAG,aAAa,IAAI,CAAC;AAEhF,QAAM,EAAE,WAAW,KAAK,IAAI;AAE5B,aAAO;AAAA,IACL,CAAO,aAAmC,gBAA6B;AACrE,YAAM,MAAM,MAAM,MAAM,MAAY;AA7C1C;AA8CQ,cAAM,WAAsB;AAAA,UAC1B,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,UAAU;AAAA,cACjB;AAAA,cACA,iBAAiB,CAAC,GAAG,iBAAiB;AAAA,YACxC;AAAA,UACF,CAAC;AAAA,UACD,GAAG,UAAU;AAAA,UACb,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,YAAY;AAAA,UACvB,CAAC;AAAA,UACD,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,oBAAoB,YAAY,eAAe;AAAA,UAC1D,CAAC;AAAA,UACD,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,qBAAqB,YAAY,gBAAgB;AAAA,UAC5D,CAAC;AAAA,QACH;AAEA,cAAM,gBAAgB,IAAI,+CAAqB;AAAA,UAC7C;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,QACF,CAAC;AAED,cAAM,WAAW,MAAM,cACpB,wBAAwB;AAAA,UACvB,MAAM;AAAA,YACJ,UAAU;AAAA,cACR,SAAS,CAAC;AAAA,cACV,KAAK,OAAO,SAAS;AAAA,YACvB;AAAA,YACA,cAAU,qDAA0B,QAAQ;AAAA,YAC5C,UAAU;AAAA,cACR,aAAa,6CAAmB;AAAA,YAClC;AAAA,YACA,qBAAqB;AAAA,cACnB;AAAA,cACA;AAAA,YACF;AAAA,UACF;AAAA,UACA;AAAA,UACA,QAAQ;AAAA,QACV,CAAC,EACA,UAAU;AAEb,YAAI,SAAS;AACb,mBAAW,eAAW;AAAA,WACpB,0BAAS,SAAT,mBAAe,4BAAf,mBAAwC,aAAxC,YAAoD,CAAC;AAAA,QACvD,GAAG;AACD,cAAI,YAAY,SAAS;AACvB;AAAA,UACF;AACA,cAAI,mBAAmB,uCAAa;AAClC,sBAAU,QAAQ;AAClB,oBAAQ,IAAI,QAAQ,OAAO;AAAA,UAC7B;AAAA,QACF;AAEA,eAAO;AAAA,MACT,EAAC;AAED,aAAO;AAAA,IACT;AAAA,IACA,CAAC,WAAW,kBAAkB,mBAAmB,eAAe;AAAA,EAClE;AACF;","names":[]}

package/dist/hooks/make-autosuggestions-function/use-make-standard-autosuggestions-function.mjs

@@ -1,8 +1,143 @@
+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/hooks/make-autosuggestions-function/use-make-standard-autosuggestions-function.tsx
+import { COPILOT_CLOUD_PUBLIC_API_KEY_HEADER } from "@copilotkit/shared";
+import { useCopilotContext } from "@copilotkit/react-core";
+import { useCallback } from "react";
+
+// src/lib/retry.tsx
+function retry(fn, retriesLeft = 2, interval = 200, backoff = 1.5) {
+  return new Promise((resolve, reject) => {
+    fn().then(resolve).catch((error) => {
+      if (retriesLeft === 1) {
+        reject(error);
+        return;
+      }
+      setTimeout(() => {
+        retry(fn, retriesLeft - 1, interval * backoff, backoff).then(resolve).catch(reject);
+      }, interval);
+    });
+  });
+}
+
+// src/hooks/make-autosuggestions-function/use-make-standard-autosuggestions-function.tsx
 import {
-  useMakeStandardAutosuggestionFunction
-} from "../../chunk-D2SUDM2O.mjs";
-import "../../chunk-KNQIEOFP.mjs";
-import "../../chunk-MRXNTQOX.mjs";
+  CopilotRuntimeClient,
+  Role,
+  TextMessage,
+  convertGqlOutputToMessages,
+  convertMessagesToGqlInput,
+  CopilotRequestType
+} from "@copilotkit/runtime-client-gql";
+function useMakeStandardAutosuggestionFunction(textareaPurpose, contextCategories, apiConfig) {
+  const { getContextString, copilotApiConfig } = useCopilotContext();
+  const { chatApiEndpoint: url, publicApiKey, credentials, properties } = copilotApiConfig;
+  const headers = __spreadValues(__spreadValues({}, copilotApiConfig.headers), publicApiKey ? { [COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {});
+  const { maxTokens, stop } = apiConfig;
+  return useCallback(
+    (editorState, abortSignal) => __async(this, null, function* () {
+      const res = yield retry(() => __async(this, null, function* () {
+        var _a, _b, _c;
+        const messages = [
+          new TextMessage({
+            role: Role.System,
+            content: apiConfig.makeSystemPrompt(
+              textareaPurpose,
+              getContextString([], contextCategories)
+            )
+          }),
+          ...apiConfig.fewShotMessages,
+          new TextMessage({
+            role: Role.User,
+            content: editorState.textAfterCursor
+          }),
+          new TextMessage({
+            role: Role.User,
+            content: `<TextAfterCursor>${editorState.textAfterCursor}</TextAfterCursor>`
+          }),
+          new TextMessage({
+            role: Role.User,
+            content: `<TextBeforeCursor>${editorState.textBeforeCursor}</TextBeforeCursor>`
+          })
+        ];
+        const runtimeClient = new CopilotRuntimeClient({
+          url,
+          publicApiKey,
+          headers,
+          credentials
+        });
+        const response = yield runtimeClient.generateCopilotResponse({
+          data: {
+            frontend: {
+              actions: [],
+              url: window.location.href
+            },
+            messages: convertMessagesToGqlInput(messages),
+            metadata: {
+              requestType: CopilotRequestType.TextareaCompletion
+            },
+            forwardedParameters: {
+              maxTokens,
+              stop
+            }
+          },
+          properties,
+          signal: abortSignal
+        }).toPromise();
+        let result = "";
+        for (const message of convertGqlOutputToMessages(
+          (_c = (_b = (_a = response.data) == null ? void 0 : _a.generateCopilotResponse) == null ? void 0 : _b.messages) != null ? _c : []
+        )) {
+          if (abortSignal.aborted) {
+            break;
+          }
+          if (message instanceof TextMessage) {
+            result += message.content;
+            console.log(message.content);
+          }
+        }
+        return result;
+      }));
+      return res;
+    }),
+    [apiConfig, getContextString, contextCategories, textareaPurpose]
+  );
+}
 export {
   useMakeStandardAutosuggestionFunction
 };

package/dist/hooks/make-autosuggestions-function/use-make-standard-autosuggestions-function.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}
+{"version":3,"sources":["../../../src/hooks/make-autosuggestions-function/use-make-standard-autosuggestions-function.tsx","../../../src/lib/retry.tsx"],"sourcesContent":["import { COPILOT_CLOUD_PUBLIC_API_KEY_HEADER } from \"@copilotkit/shared\";\nimport { useCopilotContext } from \"@copilotkit/react-core\";\nimport { useCallback } from \"react\";\nimport { AutosuggestionsBareFunction } from \"../../types\";\nimport { retry } from \"../../lib/retry\";\nimport { InsertionEditorState } from \"../../types/base/autosuggestions-bare-function\";\nimport { SuggestionsApiConfig } from \"../../types/autosuggestions-config/suggestions-api-config\";\nimport {\n  CopilotRuntimeClient,\n  Message,\n  Role,\n  TextMessage,\n  convertGqlOutputToMessages,\n  convertMessagesToGqlInput,\n  CopilotRequestType,\n} from \"@copilotkit/runtime-client-gql\";\n\n/**\n * Returns a memoized function that sends a request to the specified API endpoint to get an autosuggestion for the user's input.\n * The function takes in the text before and after the cursor, and an abort signal.\n * It sends a POST request to the API endpoint with the messages array containing the system message, few shot messages, and user messages.\n * The function returns the suggestion from the API response.\n *\n * @param textareaPurpose - The purpose of the textarea. This is included in the system message.\n * @param apiEndpoint - The API endpoint to send the autosuggestion request to.\n * @param makeSystemMessage - A function that takes in a context string and returns a system message to include in the autosuggestion request.\n * @param fewShotMessages - An array of few shot messages to include in the autosuggestion request.\n * @param contextCategories - The categories of context strings we want to include. By default, we include the (default) \"global\" context category.\n * @returns A memoized function that sends a request to the specified API endpoint to get an autosuggestion for the user's input.\n */\nexport function useMakeStandardAutosuggestionFunction(\n  textareaPurpose: string,\n  contextCategories: string[],\n  apiConfig: SuggestionsApiConfig,\n): AutosuggestionsBareFunction {\n  const { getContextString, copilotApiConfig } = useCopilotContext();\n  const { chatApiEndpoint: url, publicApiKey, credentials, properties } = copilotApiConfig;\n  const headers = {\n    ...copilotApiConfig.headers,\n    ...(publicApiKey ? { [COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {}),\n  };\n  const { maxTokens, stop } = apiConfig;\n\n  return useCallback(\n    async (editorState: InsertionEditorState, abortSignal: AbortSignal) => {\n      const res = await retry(async () => {\n        const messages: Message[] = [\n          new TextMessage({\n            role: Role.System,\n            content: apiConfig.makeSystemPrompt(\n              textareaPurpose,\n              getContextString([], contextCategories),\n            ),\n          }),\n          ...apiConfig.fewShotMessages,\n          new TextMessage({\n            role: Role.User,\n            content: editorState.textAfterCursor,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<TextAfterCursor>${editorState.textAfterCursor}</TextAfterCursor>`,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<TextBeforeCursor>${editorState.textBeforeCursor}</TextBeforeCursor>`,\n          }),\n        ];\n\n        const runtimeClient = new CopilotRuntimeClient({\n          url,\n          publicApiKey,\n          headers,\n          credentials,\n        });\n\n        const response = await runtimeClient\n          .generateCopilotResponse({\n            data: {\n              frontend: {\n                actions: [],\n                url: window.location.href,\n              },\n              messages: convertMessagesToGqlInput(messages),\n              metadata: {\n                requestType: CopilotRequestType.TextareaCompletion,\n              },\n              forwardedParameters: {\n                maxTokens,\n                stop,\n              },\n            },\n            properties,\n            signal: abortSignal,\n          })\n          .toPromise();\n\n        let result = \"\";\n        for (const message of convertGqlOutputToMessages(\n          response.data?.generateCopilotResponse?.messages ?? [],\n        )) {\n          if (abortSignal.aborted) {\n            break;\n          }\n          if (message instanceof TextMessage) {\n            result += message.content;\n            console.log(message.content);\n          }\n        }\n\n        return result;\n      });\n\n      return res;\n    },\n    [apiConfig, getContextString, contextCategories, textareaPurpose],\n  );\n}\n","export function retry<T>(\n  fn: () => Promise<T>,\n  retriesLeft: number = 2,\n  interval: number = 200,\n  backoff: number = 1.5,\n): Promise<T> {\n  return new Promise((resolve, reject) => {\n    fn()\n      .then(resolve)\n      .catch((error) => {\n        if (retriesLeft === 1) {\n          reject(error);\n          return;\n        }\n\n        setTimeout(() => {\n          retry(fn, retriesLeft - 1, interval * backoff, backoff)\n            .then(resolve)\n            .catch(reject);\n        }, interval);\n      });\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,SAAS,2CAA2C;AACpD,SAAS,yBAAyB;AAClC,SAAS,mBAAmB;;;ACFrB,SAAS,MACd,IACA,cAAsB,GACtB,WAAmB,KACnB,UAAkB,KACN;AACZ,SAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,OAAG,EACA,KAAK,OAAO,EACZ,MAAM,CAAC,UAAU;AAChB,UAAI,gBAAgB,GAAG;AACrB,eAAO,KAAK;AACZ;AAAA,MACF;AAEA,iBAAW,MAAM;AACf,cAAM,IAAI,cAAc,GAAG,WAAW,SAAS,OAAO,EACnD,KAAK,OAAO,EACZ,MAAM,MAAM;AAAA,MACjB,GAAG,QAAQ;AAAA,IACb,CAAC;AAAA,EACL,CAAC;AACH;;;ADfA;AAAA,EACE;AAAA,EAEA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAeA,SAAS,sCACd,iBACA,mBACA,WAC6B;AAC7B,QAAM,EAAE,kBAAkB,iBAAiB,IAAI,kBAAkB;AACjE,QAAM,EAAE,iBAAiB,KAAK,cAAc,aAAa,WAAW,IAAI;AACxE,QAAM,UAAU,kCACX,iBAAiB,UAChB,eAAe,EAAE,CAAC,mCAAmC,GAAG,aAAa,IAAI,CAAC;AAEhF,QAAM,EAAE,WAAW,KAAK,IAAI;AAE5B,SAAO;AAAA,IACL,CAAO,aAAmC,gBAA6B;AACrE,YAAM,MAAM,MAAM,MAAM,MAAY;AA7C1C;AA8CQ,cAAM,WAAsB;AAAA,UAC1B,IAAI,YAAY;AAAA,YACd,MAAM,KAAK;AAAA,YACX,SAAS,UAAU;AAAA,cACjB;AAAA,cACA,iBAAiB,CAAC,GAAG,iBAAiB;AAAA,YACxC;AAAA,UACF,CAAC;AAAA,UACD,GAAG,UAAU;AAAA,UACb,IAAI,YAAY;AAAA,YACd,MAAM,KAAK;AAAA,YACX,SAAS,YAAY;AAAA,UACvB,CAAC;AAAA,UACD,IAAI,YAAY;AAAA,YACd,MAAM,KAAK;AAAA,YACX,SAAS,oBAAoB,YAAY,eAAe;AAAA,UAC1D,CAAC;AAAA,UACD,IAAI,YAAY;AAAA,YACd,MAAM,KAAK;AAAA,YACX,SAAS,qBAAqB,YAAY,gBAAgB;AAAA,UAC5D,CAAC;AAAA,QACH;AAEA,cAAM,gBAAgB,IAAI,qBAAqB;AAAA,UAC7C;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,QACF,CAAC;AAED,cAAM,WAAW,MAAM,cACpB,wBAAwB;AAAA,UACvB,MAAM;AAAA,YACJ,UAAU;AAAA,cACR,SAAS,CAAC;AAAA,cACV,KAAK,OAAO,SAAS;AAAA,YACvB;AAAA,YACA,UAAU,0BAA0B,QAAQ;AAAA,YAC5C,UAAU;AAAA,cACR,aAAa,mBAAmB;AAAA,YAClC;AAAA,YACA,qBAAqB;AAAA,cACnB;AAAA,cACA;AAAA,YACF;AAAA,UACF;AAAA,UACA;AAAA,UACA,QAAQ;AAAA,QACV,CAAC,EACA,UAAU;AAEb,YAAI,SAAS;AACb,mBAAW,WAAW;AAAA,WACpB,0BAAS,SAAT,mBAAe,4BAAf,mBAAwC,aAAxC,YAAoD,CAAC;AAAA,QACvD,GAAG;AACD,cAAI,YAAY,SAAS;AACvB;AAAA,UACF;AACA,cAAI,mBAAmB,aAAa;AAClC,sBAAU,QAAQ;AAClB,oBAAQ,IAAI,QAAQ,OAAO;AAAA,UAC7B;AAAA,QACF;AAEA,eAAO;AAAA,MACT,EAAC;AAED,aAAO;AAAA,IACT;AAAA,IACA,CAAC,WAAW,kBAAkB,mBAAmB,eAAe;AAAA,EAClE;AACF;","names":[]}

package/dist/hooks/make-autosuggestions-function/use-make-standard-insertion-function.d.mts

@@ -0,0 +1,23 @@
+import { Generator_InsertionOrEditingSuggestion } from '../../types/base/autosuggestions-bare-function.mjs';
+import { InsertionsApiConfig } from '../../types/autosuggestions-config/insertions-api-config.mjs';
+import { EditingApiConfig } from '../../types/autosuggestions-config/editing-api-config.mjs';
+import '@copilotkit/react-core';
+import '@copilotkit/runtime-client-gql';
+import '../../types/autosuggestions-config/subtypes/make-system-prompt.mjs';
+
+/**
+ * Returns a memoized function that sends a request to the specified API endpoint to get an autosuggestion for the user's input.
+ * The function takes in the text before and after the cursor, and an abort signal.
+ * It sends a POST request to the API endpoint with the messages array containing the system message, few shot messages, and user messages.
+ * The function returns the suggestion from the API response.
+ *
+ * @param textareaPurpose - The purpose of the textarea. This is included in the system message.
+ * @param apiEndpoint - The API endpoint to send the autosuggestion request to.
+ * @param makeSystemMessage - A function that takes in a context string and returns a system message to include in the autosuggestion request.
+ * @param fewShotMessages - An array of few shot messages to include in the autosuggestion request.
+ * @param contextCategories - The categories of context strings we want to include. By default, we include the (default) "global" context category.
+ * @returns A memoized function that sends a request to the specified API endpoint to get an autosuggestion for the user's input.
+ */
+declare function useMakeStandardInsertionOrEditingFunction(textareaPurpose: string, contextCategories: string[], insertionApiConfig: InsertionsApiConfig, editingApiConfig: EditingApiConfig): Generator_InsertionOrEditingSuggestion;
+
+export { useMakeStandardInsertionOrEditingFunction };

package/dist/hooks/make-autosuggestions-function/use-make-standard-insertion-function.js

@@ -151,7 +151,7 @@ function useMakeStandardInsertionOrEditingFunction(textareaPurpose, contextCateg
                 actions: [],
                 url: window.location.href
               },
-              messages: (0, import_runtime_client_gql.convertMessagesToGqlInput)((0, import_runtime_client_gql.filterAgentStateMessages)(messages)),
+              messages: (0, import_runtime_client_gql.convertMessagesToGqlInput)(messages),
               metadata: {
                 requestType: import_runtime_client_gql.CopilotRequestType.TextareaCompletion
               }
@@ -207,7 +207,7 @@ function useMakeStandardInsertionOrEditingFunction(textareaPurpose, contextCateg
                 actions: [],
                 url: window.location.href
               },
-              messages: (0, import_runtime_client_gql.convertMessagesToGqlInput)((0, import_runtime_client_gql.filterAgentStateMessages)(messages)),
+              messages: (0, import_runtime_client_gql.convertMessagesToGqlInput)(messages),
               metadata: {
                 requestType: import_runtime_client_gql.CopilotRequestType.TextareaCompletion
               }

package/dist/hooks/make-autosuggestions-function/use-make-standard-insertion-function.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/hooks/make-autosuggestions-function/use-make-standard-insertion-function.tsx","../../../src/lib/retry.tsx"],"sourcesContent":["import { COPILOT_CLOUD_PUBLIC_API_KEY_HEADER } from \"@copilotkit/shared\";\nimport { useCopilotContext } from \"@copilotkit/react-core\";\nimport { useCallback } from \"react\";\nimport {\n  CopilotRuntimeClient,\n  Message,\n  Role,\n  TextMessage,\n  convertGqlOutputToMessages,\n  convertMessagesToGqlInput,\n  filterAgentStateMessages,\n  CopilotRequestType,\n} from \"@copilotkit/runtime-client-gql\";\nimport { retry } from \"../../lib/retry\";\nimport {\n  EditingEditorState,\n  Generator_InsertionOrEditingSuggestion,\n} from \"../../types/base/autosuggestions-bare-function\";\nimport { InsertionsApiConfig } from \"../../types/autosuggestions-config/insertions-api-config\";\nimport { EditingApiConfig } from \"../../types/autosuggestions-config/editing-api-config\";\nimport { DocumentPointer } from \"@copilotkit/react-core\";\n\n/**\n * Returns a memoized function that sends a request to the specified API endpoint to get an autosuggestion for the user's input.\n * The function takes in the text before and after the cursor, and an abort signal.\n * It sends a POST request to the API endpoint with the messages array containing the system message, few shot messages, and user messages.\n * The function returns the suggestion from the API response.\n *\n * @param textareaPurpose - The purpose of the textarea. This is included in the system message.\n * @param apiEndpoint - The API endpoint to send the autosuggestion request to.\n * @param makeSystemMessage - A function that takes in a context string and returns a system message to include in the autosuggestion request.\n * @param fewShotMessages - An array of few shot messages to include in the autosuggestion request.\n * @param contextCategories - The categories of context strings we want to include. By default, we include the (default) \"global\" context category.\n * @returns A memoized function that sends a request to the specified API endpoint to get an autosuggestion for the user's input.\n */\nexport function useMakeStandardInsertionOrEditingFunction(\n  textareaPurpose: string,\n  contextCategories: string[],\n  insertionApiConfig: InsertionsApiConfig,\n  editingApiConfig: EditingApiConfig,\n): Generator_InsertionOrEditingSuggestion {\n  const { getContextString, copilotApiConfig } = useCopilotContext();\n  const headers = {\n    ...(copilotApiConfig.publicApiKey\n      ? { [COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: copilotApiConfig.publicApiKey }\n      : {}),\n  };\n\n  const runtimeClient = new CopilotRuntimeClient({\n    url: copilotApiConfig.chatApiEndpoint,\n    publicApiKey: copilotApiConfig.publicApiKey,\n    headers,\n    credentials: copilotApiConfig.credentials,\n  });\n\n  async function runtimeClientResponseToStringStream(\n    responsePromise: ReturnType<typeof runtimeClient.generateCopilotResponse>,\n  ) {\n    const messagesStream = await CopilotRuntimeClient.asStream(responsePromise);\n\n    return new ReadableStream({\n      async start(controller) {\n        const reader = messagesStream.getReader();\n        let sentContent = \"\";\n\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) {\n            break;\n          }\n\n          const messages = convertGqlOutputToMessages(value.generateCopilotResponse.messages);\n\n          let newContent = \"\";\n\n          for (const message of messages) {\n            if (message instanceof TextMessage) {\n              newContent += message.content;\n            }\n          }\n          if (newContent) {\n            const contentToSend = newContent.slice(sentContent.length);\n            controller.enqueue(contentToSend);\n            sentContent += contentToSend;\n          }\n        }\n        controller.close();\n      },\n    });\n  }\n\n  const insertionFunction = useCallback(\n    async (\n      editorState: EditingEditorState,\n      insertionPrompt: string,\n      documents: DocumentPointer[],\n      abortSignal: AbortSignal,\n    ) => {\n      const res = await retry(async () => {\n        const messages: Message[] = [\n          new TextMessage({\n            role: Role.System,\n            content: insertionApiConfig.makeSystemPrompt(\n              textareaPurpose,\n              getContextString(documents, contextCategories),\n            ),\n          }),\n          ...insertionApiConfig.fewShotMessages,\n          new TextMessage({\n            role: Role.User,\n            content: `<TextAfterCursor>${editorState.textAfterCursor}</TextAfterCursor>`,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<TextBeforeCursor>${editorState.textBeforeCursor}</TextBeforeCursor>`,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<InsertionPrompt>${insertionPrompt}</InsertionPrompt>`,\n          }),\n        ];\n\n        return runtimeClientResponseToStringStream(\n          runtimeClient.generateCopilotResponse({\n            data: {\n              frontend: {\n                actions: [],\n                url: window.location.href,\n              },\n              messages: convertMessagesToGqlInput(filterAgentStateMessages(messages)),\n              metadata: {\n                requestType: CopilotRequestType.TextareaCompletion,\n              },\n            },\n            properties: copilotApiConfig.properties,\n            signal: abortSignal,\n          }),\n        );\n      });\n\n      return res;\n    },\n    [insertionApiConfig, getContextString, contextCategories, textareaPurpose],\n  );\n\n  const editingFunction = useCallback(\n    async (\n      editorState: EditingEditorState,\n      editingPrompt: string,\n      documents: DocumentPointer[],\n      abortSignal: AbortSignal,\n    ) => {\n      const res = await retry(async () => {\n        const messages: Message[] = [\n          new TextMessage({\n            role: Role.System,\n            content: editingApiConfig.makeSystemPrompt(\n              textareaPurpose,\n              getContextString(documents, contextCategories),\n            ),\n          }),\n          ...editingApiConfig.fewShotMessages,\n          new TextMessage({\n            role: Role.User,\n            content: `<TextBeforeCursor>${editorState.textBeforeCursor}</TextBeforeCursor>`,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<TextToEdit>${editorState.selectedText}</TextToEdit>`,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<TextAfterCursor>${editorState.textAfterCursor}</TextAfterCursor>`,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<EditingPrompt>${editingPrompt}</EditingPrompt>`,\n          }),\n        ];\n\n        const runtimeClient = new CopilotRuntimeClient({\n          url: copilotApiConfig.chatApiEndpoint,\n          publicApiKey: copilotApiConfig.publicApiKey,\n          headers,\n          credentials: copilotApiConfig.credentials,\n        });\n\n        return runtimeClientResponseToStringStream(\n          runtimeClient.generateCopilotResponse({\n            data: {\n              frontend: {\n                actions: [],\n                url: window.location.href,\n              },\n              messages: convertMessagesToGqlInput(filterAgentStateMessages(messages)),\n              metadata: {\n                requestType: CopilotRequestType.TextareaCompletion,\n              },\n            },\n            properties: copilotApiConfig.properties,\n            signal: abortSignal,\n          }),\n        );\n      });\n\n      return res;\n    },\n    [editingApiConfig, getContextString, contextCategories, textareaPurpose],\n  );\n\n  const insertionOrEditingFunction = useCallback(\n    async (\n      editorState: EditingEditorState,\n      insertionPrompt: string,\n      documents: DocumentPointer[],\n      abortSignal: AbortSignal,\n    ) => {\n      if (editorState.selectedText === \"\") {\n        return await insertionFunction(editorState, insertionPrompt, documents, abortSignal);\n      } else {\n        return await editingFunction(editorState, insertionPrompt, documents, abortSignal);\n      }\n    },\n    [insertionFunction, editingFunction],\n  );\n\n  return insertionOrEditingFunction;\n}\n","export function retry<T>(\n  fn: () => Promise<T>,\n  retriesLeft: number = 2,\n  interval: number = 200,\n  backoff: number = 1.5,\n): Promise<T> {\n  return new Promise((resolve, reject) => {\n    fn()\n      .then(resolve)\n      .catch((error) => {\n        if (retriesLeft === 1) {\n          reject(error);\n          return;\n        }\n\n        setTimeout(() => {\n          retry(fn, retriesLeft - 1, interval * backoff, backoff)\n            .then(resolve)\n            .catch(reject);\n        }, interval);\n      });\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,oBAAoD;AACpD,wBAAkC;AAClC,mBAA4B;AAC5B,gCASO;;;ACZA,SAAS,MACd,IACA,cAAsB,GACtB,WAAmB,KACnB,UAAkB,KACN;AACZ,SAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,OAAG,EACA,KAAK,OAAO,EACZ,MAAM,CAAC,UAAU;AAChB,UAAI,gBAAgB,GAAG;AACrB,eAAO,KAAK;AACZ;AAAA,MACF;AAEA,iBAAW,MAAM;AACf,cAAM,IAAI,cAAc,GAAG,WAAW,SAAS,OAAO,EACnD,KAAK,OAAO,EACZ,MAAM,MAAM;AAAA,MACjB,GAAG,QAAQ;AAAA,IACb,CAAC;AAAA,EACL,CAAC;AACH;;;ADaO,SAAS,0CACd,iBACA,mBACA,oBACA,kBACwC;AACxC,QAAM,EAAE,kBAAkB,iBAAiB,QAAI,qCAAkB;AACjE,QAAM,UAAU,mBACV,iBAAiB,eACjB,EAAE,CAAC,iDAAmC,GAAG,iBAAiB,aAAa,IACvE,CAAC;AAGP,QAAM,gBAAgB,IAAI,+CAAqB;AAAA,IAC7C,KAAK,iBAAiB;AAAA,IACtB,cAAc,iBAAiB;AAAA,IAC/B;AAAA,IACA,aAAa,iBAAiB;AAAA,EAChC,CAAC;AAED,WAAe,oCACb,iBACA;AAAA;AACA,YAAM,iBAAiB,MAAM,+CAAqB,SAAS,eAAe;AAE1E,aAAO,IAAI,eAAe;AAAA,QAClB,MAAM,YAAY;AAAA;AACtB,kBAAM,SAAS,eAAe,UAAU;AACxC,gBAAI,cAAc;AAElB,mBAAO,MAAM;AACX,oBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,kBAAI,MAAM;AACR;AAAA,cACF;AAEA,oBAAM,eAAW,sDAA2B,MAAM,wBAAwB,QAAQ;AAElF,kBAAI,aAAa;AAEjB,yBAAW,WAAW,UAAU;AAC9B,oBAAI,mBAAmB,uCAAa;AAClC,gCAAc,QAAQ;AAAA,gBACxB;AAAA,cACF;AACA,kBAAI,YAAY;AACd,sBAAM,gBAAgB,WAAW,MAAM,YAAY,MAAM;AACzD,2BAAW,QAAQ,aAAa;AAChC,+BAAe;AAAA,cACjB;AAAA,YACF;AACA,uBAAW,MAAM;AAAA,UACnB;AAAA;AAAA,MACF,CAAC;AAAA,IACH;AAAA;AAEA,QAAM,wBAAoB;AAAA,IACxB,CACE,aACA,iBACA,WACA,gBACG;AACH,YAAM,MAAM,MAAM,MAAM,MAAY;AAClC,cAAM,WAAsB;AAAA,UAC1B,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,mBAAmB;AAAA,cAC1B;AAAA,cACA,iBAAiB,WAAW,iBAAiB;AAAA,YAC/C;AAAA,UACF,CAAC;AAAA,UACD,GAAG,mBAAmB;AAAA,UACtB,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,oBAAoB,YAAY;AAAA,UAC3C,CAAC;AAAA,UACD,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,qBAAqB,YAAY;AAAA,UAC5C,CAAC;AAAA,UACD,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,oBAAoB;AAAA,UAC/B,CAAC;AAAA,QACH;AAEA,eAAO;AAAA,UACL,cAAc,wBAAwB;AAAA,YACpC,MAAM;AAAA,cACJ,UAAU;AAAA,gBACR,SAAS,CAAC;AAAA,gBACV,KAAK,OAAO,SAAS;AAAA,cACvB;AAAA,cACA,cAAU,yDAA0B,oDAAyB,QAAQ,CAAC;AAAA,cACtE,UAAU;AAAA,gBACR,aAAa,6CAAmB;AAAA,cAClC;AAAA,YACF;AAAA,YACA,YAAY,iBAAiB;AAAA,YAC7B,QAAQ;AAAA,UACV,CAAC;AAAA,QACH;AAAA,MACF,EAAC;AAED,aAAO;AAAA,IACT;AAAA,IACA,CAAC,oBAAoB,kBAAkB,mBAAmB,eAAe;AAAA,EAC3E;AAEA,QAAM,sBAAkB;AAAA,IACtB,CACE,aACA,eACA,WACA,gBACG;AACH,YAAM,MAAM,MAAM,MAAM,MAAY;AAClC,cAAM,WAAsB;AAAA,UAC1B,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,iBAAiB;AAAA,cACxB;AAAA,cACA,iBAAiB,WAAW,iBAAiB;AAAA,YAC/C;AAAA,UACF,CAAC;AAAA,UACD,GAAG,iBAAiB;AAAA,UACpB,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,qBAAqB,YAAY;AAAA,UAC5C,CAAC;AAAA,UACD,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,eAAe,YAAY;AAAA,UACtC,CAAC;AAAA,UACD,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,oBAAoB,YAAY;AAAA,UAC3C,CAAC;AAAA,UACD,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,kBAAkB;AAAA,UAC7B,CAAC;AAAA,QACH;AAEA,cAAMA,iBAAgB,IAAI,+CAAqB;AAAA,UAC7C,KAAK,iBAAiB;AAAA,UACtB,cAAc,iBAAiB;AAAA,UAC/B;AAAA,UACA,aAAa,iBAAiB;AAAA,QAChC,CAAC;AAED,eAAO;AAAA,UACLA,eAAc,wBAAwB;AAAA,YACpC,MAAM;AAAA,cACJ,UAAU;AAAA,gBACR,SAAS,CAAC;AAAA,gBACV,KAAK,OAAO,SAAS;AAAA,cACvB;AAAA,cACA,cAAU,yDAA0B,oDAAyB,QAAQ,CAAC;AAAA,cACtE,UAAU;AAAA,gBACR,aAAa,6CAAmB;AAAA,cAClC;AAAA,YACF;AAAA,YACA,YAAY,iBAAiB;AAAA,YAC7B,QAAQ;AAAA,UACV,CAAC;AAAA,QACH;AAAA,MACF,EAAC;AAED,aAAO;AAAA,IACT;AAAA,IACA,CAAC,kBAAkB,kBAAkB,mBAAmB,eAAe;AAAA,EACzE;AAEA,QAAM,iCAA6B;AAAA,IACjC,CACE,aACA,iBACA,WACA,gBACG;AACH,UAAI,YAAY,iBAAiB,IAAI;AACnC,eAAO,MAAM,kBAAkB,aAAa,iBAAiB,WAAW,WAAW;AAAA,MACrF,OAAO;AACL,eAAO,MAAM,gBAAgB,aAAa,iBAAiB,WAAW,WAAW;AAAA,MACnF;AAAA,IACF;AAAA,IACA,CAAC,mBAAmB,eAAe;AAAA,EACrC;AAEA,SAAO;AACT;","names":["runtimeClient"]}
+{"version":3,"sources":["../../../src/hooks/make-autosuggestions-function/use-make-standard-insertion-function.tsx","../../../src/lib/retry.tsx"],"sourcesContent":["import { COPILOT_CLOUD_PUBLIC_API_KEY_HEADER } from \"@copilotkit/shared\";\nimport { useCopilotContext } from \"@copilotkit/react-core\";\nimport { useCallback } from \"react\";\nimport {\n  CopilotRuntimeClient,\n  Message,\n  Role,\n  TextMessage,\n  convertGqlOutputToMessages,\n  convertMessagesToGqlInput,\n  CopilotRequestType,\n} from \"@copilotkit/runtime-client-gql\";\nimport { retry } from \"../../lib/retry\";\nimport {\n  EditingEditorState,\n  Generator_InsertionOrEditingSuggestion,\n} from \"../../types/base/autosuggestions-bare-function\";\nimport { InsertionsApiConfig } from \"../../types/autosuggestions-config/insertions-api-config\";\nimport { EditingApiConfig } from \"../../types/autosuggestions-config/editing-api-config\";\nimport { DocumentPointer } from \"@copilotkit/react-core\";\n\n/**\n * Returns a memoized function that sends a request to the specified API endpoint to get an autosuggestion for the user's input.\n * The function takes in the text before and after the cursor, and an abort signal.\n * It sends a POST request to the API endpoint with the messages array containing the system message, few shot messages, and user messages.\n * The function returns the suggestion from the API response.\n *\n * @param textareaPurpose - The purpose of the textarea. This is included in the system message.\n * @param apiEndpoint - The API endpoint to send the autosuggestion request to.\n * @param makeSystemMessage - A function that takes in a context string and returns a system message to include in the autosuggestion request.\n * @param fewShotMessages - An array of few shot messages to include in the autosuggestion request.\n * @param contextCategories - The categories of context strings we want to include. By default, we include the (default) \"global\" context category.\n * @returns A memoized function that sends a request to the specified API endpoint to get an autosuggestion for the user's input.\n */\nexport function useMakeStandardInsertionOrEditingFunction(\n  textareaPurpose: string,\n  contextCategories: string[],\n  insertionApiConfig: InsertionsApiConfig,\n  editingApiConfig: EditingApiConfig,\n): Generator_InsertionOrEditingSuggestion {\n  const { getContextString, copilotApiConfig } = useCopilotContext();\n  const headers = {\n    ...(copilotApiConfig.publicApiKey\n      ? { [COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: copilotApiConfig.publicApiKey }\n      : {}),\n  };\n\n  const runtimeClient = new CopilotRuntimeClient({\n    url: copilotApiConfig.chatApiEndpoint,\n    publicApiKey: copilotApiConfig.publicApiKey,\n    headers,\n    credentials: copilotApiConfig.credentials,\n  });\n\n  async function runtimeClientResponseToStringStream(\n    responsePromise: ReturnType<typeof runtimeClient.generateCopilotResponse>,\n  ) {\n    const messagesStream = await CopilotRuntimeClient.asStream(responsePromise);\n\n    return new ReadableStream({\n      async start(controller) {\n        const reader = messagesStream.getReader();\n        let sentContent = \"\";\n\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) {\n            break;\n          }\n\n          const messages = convertGqlOutputToMessages(value.generateCopilotResponse.messages);\n\n          let newContent = \"\";\n\n          for (const message of messages) {\n            if (message instanceof TextMessage) {\n              newContent += message.content;\n            }\n          }\n          if (newContent) {\n            const contentToSend = newContent.slice(sentContent.length);\n            controller.enqueue(contentToSend);\n            sentContent += contentToSend;\n          }\n        }\n        controller.close();\n      },\n    });\n  }\n\n  const insertionFunction = useCallback(\n    async (\n      editorState: EditingEditorState,\n      insertionPrompt: string,\n      documents: DocumentPointer[],\n      abortSignal: AbortSignal,\n    ) => {\n      const res = await retry(async () => {\n        const messages: Message[] = [\n          new TextMessage({\n            role: Role.System,\n            content: insertionApiConfig.makeSystemPrompt(\n              textareaPurpose,\n              getContextString(documents, contextCategories),\n            ),\n          }),\n          ...insertionApiConfig.fewShotMessages,\n          new TextMessage({\n            role: Role.User,\n            content: `<TextAfterCursor>${editorState.textAfterCursor}</TextAfterCursor>`,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<TextBeforeCursor>${editorState.textBeforeCursor}</TextBeforeCursor>`,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<InsertionPrompt>${insertionPrompt}</InsertionPrompt>`,\n          }),\n        ];\n\n        return runtimeClientResponseToStringStream(\n          runtimeClient.generateCopilotResponse({\n            data: {\n              frontend: {\n                actions: [],\n                url: window.location.href,\n              },\n              messages: convertMessagesToGqlInput(messages),\n              metadata: {\n                requestType: CopilotRequestType.TextareaCompletion,\n              },\n            },\n            properties: copilotApiConfig.properties,\n            signal: abortSignal,\n          }),\n        );\n      });\n\n      return res;\n    },\n    [insertionApiConfig, getContextString, contextCategories, textareaPurpose],\n  );\n\n  const editingFunction = useCallback(\n    async (\n      editorState: EditingEditorState,\n      editingPrompt: string,\n      documents: DocumentPointer[],\n      abortSignal: AbortSignal,\n    ) => {\n      const res = await retry(async () => {\n        const messages: Message[] = [\n          new TextMessage({\n            role: Role.System,\n            content: editingApiConfig.makeSystemPrompt(\n              textareaPurpose,\n              getContextString(documents, contextCategories),\n            ),\n          }),\n          ...editingApiConfig.fewShotMessages,\n          new TextMessage({\n            role: Role.User,\n            content: `<TextBeforeCursor>${editorState.textBeforeCursor}</TextBeforeCursor>`,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<TextToEdit>${editorState.selectedText}</TextToEdit>`,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<TextAfterCursor>${editorState.textAfterCursor}</TextAfterCursor>`,\n          }),\n          new TextMessage({\n            role: Role.User,\n            content: `<EditingPrompt>${editingPrompt}</EditingPrompt>`,\n          }),\n        ];\n\n        const runtimeClient = new CopilotRuntimeClient({\n          url: copilotApiConfig.chatApiEndpoint,\n          publicApiKey: copilotApiConfig.publicApiKey,\n          headers,\n          credentials: copilotApiConfig.credentials,\n        });\n\n        return runtimeClientResponseToStringStream(\n          runtimeClient.generateCopilotResponse({\n            data: {\n              frontend: {\n                actions: [],\n                url: window.location.href,\n              },\n              messages: convertMessagesToGqlInput(messages),\n              metadata: {\n                requestType: CopilotRequestType.TextareaCompletion,\n              },\n            },\n            properties: copilotApiConfig.properties,\n            signal: abortSignal,\n          }),\n        );\n      });\n\n      return res;\n    },\n    [editingApiConfig, getContextString, contextCategories, textareaPurpose],\n  );\n\n  const insertionOrEditingFunction = useCallback(\n    async (\n      editorState: EditingEditorState,\n      insertionPrompt: string,\n      documents: DocumentPointer[],\n      abortSignal: AbortSignal,\n    ) => {\n      if (editorState.selectedText === \"\") {\n        return await insertionFunction(editorState, insertionPrompt, documents, abortSignal);\n      } else {\n        return await editingFunction(editorState, insertionPrompt, documents, abortSignal);\n      }\n    },\n    [insertionFunction, editingFunction],\n  );\n\n  return insertionOrEditingFunction;\n}\n","export function retry<T>(\n  fn: () => Promise<T>,\n  retriesLeft: number = 2,\n  interval: number = 200,\n  backoff: number = 1.5,\n): Promise<T> {\n  return new Promise((resolve, reject) => {\n    fn()\n      .then(resolve)\n      .catch((error) => {\n        if (retriesLeft === 1) {\n          reject(error);\n          return;\n        }\n\n        setTimeout(() => {\n          retry(fn, retriesLeft - 1, interval * backoff, backoff)\n            .then(resolve)\n            .catch(reject);\n        }, interval);\n      });\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,oBAAoD;AACpD,wBAAkC;AAClC,mBAA4B;AAC5B,gCAQO;;;ACXA,SAAS,MACd,IACA,cAAsB,GACtB,WAAmB,KACnB,UAAkB,KACN;AACZ,SAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,OAAG,EACA,KAAK,OAAO,EACZ,MAAM,CAAC,UAAU;AAChB,UAAI,gBAAgB,GAAG;AACrB,eAAO,KAAK;AACZ;AAAA,MACF;AAEA,iBAAW,MAAM;AACf,cAAM,IAAI,cAAc,GAAG,WAAW,SAAS,OAAO,EACnD,KAAK,OAAO,EACZ,MAAM,MAAM;AAAA,MACjB,GAAG,QAAQ;AAAA,IACb,CAAC;AAAA,EACL,CAAC;AACH;;;ADYO,SAAS,0CACd,iBACA,mBACA,oBACA,kBACwC;AACxC,QAAM,EAAE,kBAAkB,iBAAiB,QAAI,qCAAkB;AACjE,QAAM,UAAU,mBACV,iBAAiB,eACjB,EAAE,CAAC,iDAAmC,GAAG,iBAAiB,aAAa,IACvE,CAAC;AAGP,QAAM,gBAAgB,IAAI,+CAAqB;AAAA,IAC7C,KAAK,iBAAiB;AAAA,IACtB,cAAc,iBAAiB;AAAA,IAC/B;AAAA,IACA,aAAa,iBAAiB;AAAA,EAChC,CAAC;AAED,WAAe,oCACb,iBACA;AAAA;AACA,YAAM,iBAAiB,MAAM,+CAAqB,SAAS,eAAe;AAE1E,aAAO,IAAI,eAAe;AAAA,QAClB,MAAM,YAAY;AAAA;AACtB,kBAAM,SAAS,eAAe,UAAU;AACxC,gBAAI,cAAc;AAElB,mBAAO,MAAM;AACX,oBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,kBAAI,MAAM;AACR;AAAA,cACF;AAEA,oBAAM,eAAW,sDAA2B,MAAM,wBAAwB,QAAQ;AAElF,kBAAI,aAAa;AAEjB,yBAAW,WAAW,UAAU;AAC9B,oBAAI,mBAAmB,uCAAa;AAClC,gCAAc,QAAQ;AAAA,gBACxB;AAAA,cACF;AACA,kBAAI,YAAY;AACd,sBAAM,gBAAgB,WAAW,MAAM,YAAY,MAAM;AACzD,2BAAW,QAAQ,aAAa;AAChC,+BAAe;AAAA,cACjB;AAAA,YACF;AACA,uBAAW,MAAM;AAAA,UACnB;AAAA;AAAA,MACF,CAAC;AAAA,IACH;AAAA;AAEA,QAAM,wBAAoB;AAAA,IACxB,CACE,aACA,iBACA,WACA,gBACG;AACH,YAAM,MAAM,MAAM,MAAM,MAAY;AAClC,cAAM,WAAsB;AAAA,UAC1B,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,mBAAmB;AAAA,cAC1B;AAAA,cACA,iBAAiB,WAAW,iBAAiB;AAAA,YAC/C;AAAA,UACF,CAAC;AAAA,UACD,GAAG,mBAAmB;AAAA,UACtB,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,oBAAoB,YAAY,eAAe;AAAA,UAC1D,CAAC;AAAA,UACD,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,qBAAqB,YAAY,gBAAgB;AAAA,UAC5D,CAAC;AAAA,UACD,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,oBAAoB,eAAe;AAAA,UAC9C,CAAC;AAAA,QACH;AAEA,eAAO;AAAA,UACL,cAAc,wBAAwB;AAAA,YACpC,MAAM;AAAA,cACJ,UAAU;AAAA,gBACR,SAAS,CAAC;AAAA,gBACV,KAAK,OAAO,SAAS;AAAA,cACvB;AAAA,cACA,cAAU,qDAA0B,QAAQ;AAAA,cAC5C,UAAU;AAAA,gBACR,aAAa,6CAAmB;AAAA,cAClC;AAAA,YACF;AAAA,YACA,YAAY,iBAAiB;AAAA,YAC7B,QAAQ;AAAA,UACV,CAAC;AAAA,QACH;AAAA,MACF,EAAC;AAED,aAAO;AAAA,IACT;AAAA,IACA,CAAC,oBAAoB,kBAAkB,mBAAmB,eAAe;AAAA,EAC3E;AAEA,QAAM,sBAAkB;AAAA,IACtB,CACE,aACA,eACA,WACA,gBACG;AACH,YAAM,MAAM,MAAM,MAAM,MAAY;AAClC,cAAM,WAAsB;AAAA,UAC1B,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,iBAAiB;AAAA,cACxB;AAAA,cACA,iBAAiB,WAAW,iBAAiB;AAAA,YAC/C;AAAA,UACF,CAAC;AAAA,UACD,GAAG,iBAAiB;AAAA,UACpB,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,qBAAqB,YAAY,gBAAgB;AAAA,UAC5D,CAAC;AAAA,UACD,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,eAAe,YAAY,YAAY;AAAA,UAClD,CAAC;AAAA,UACD,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,oBAAoB,YAAY,eAAe;AAAA,UAC1D,CAAC;AAAA,UACD,IAAI,sCAAY;AAAA,YACd,MAAM,+BAAK;AAAA,YACX,SAAS,kBAAkB,aAAa;AAAA,UAC1C,CAAC;AAAA,QACH;AAEA,cAAMA,iBAAgB,IAAI,+CAAqB;AAAA,UAC7C,KAAK,iBAAiB;AAAA,UACtB,cAAc,iBAAiB;AAAA,UAC/B;AAAA,UACA,aAAa,iBAAiB;AAAA,QAChC,CAAC;AAED,eAAO;AAAA,UACLA,eAAc,wBAAwB;AAAA,YACpC,MAAM;AAAA,cACJ,UAAU;AAAA,gBACR,SAAS,CAAC;AAAA,gBACV,KAAK,OAAO,SAAS;AAAA,cACvB;AAAA,cACA,cAAU,qDAA0B,QAAQ;AAAA,cAC5C,UAAU;AAAA,gBACR,aAAa,6CAAmB;AAAA,cAClC;AAAA,YACF;AAAA,YACA,YAAY,iBAAiB;AAAA,YAC7B,QAAQ;AAAA,UACV,CAAC;AAAA,QACH;AAAA,MACF,EAAC;AAED,aAAO;AAAA,IACT;AAAA,IACA,CAAC,kBAAkB,kBAAkB,mBAAmB,eAAe;AAAA,EACzE;AAEA,QAAM,iCAA6B;AAAA,IACjC,CACE,aACA,iBACA,WACA,gBACG;AACH,UAAI,YAAY,iBAAiB,IAAI;AACnC,eAAO,MAAM,kBAAkB,aAAa,iBAAiB,WAAW,WAAW;AAAA,MACrF,OAAO;AACL,eAAO,MAAM,gBAAgB,aAAa,iBAAiB,WAAW,WAAW;AAAA,MACnF;AAAA,IACF;AAAA,IACA,CAAC,mBAAmB,eAAe;AAAA,EACrC;AAEA,SAAO;AACT;","names":["runtimeClient"]}