@skaldapp/milkdown-copilot 0.0.22 → 1.0.0

package/dist/index.d.ts

@@ -1,3 +1,3 @@
 import type { Ctx } from "@milkdown/kit/ctx";
-export declare const apiKeySlice: import("@milkdown/ctx").SliceType<string, "apiKey">, filenameSlice: import("@milkdown/ctx").SliceType<string, "filename">;
+export declare const apiKeySlice: import("@milkdown/ctx").SliceType<string, "apiKey">, baseURLSlice: import("@milkdown/ctx").SliceType<string, "baseURL">, endpointSlice: import("@milkdown/ctx").SliceType<string, "endpoint">, filenameSlice: import("@milkdown/ctx").SliceType<string, "filename">, modelSlice: import("@milkdown/ctx").SliceType<string, "model">;
 export declare const copilotPlugin: (((ctx: Ctx) => () => undefined) | import("@milkdown/utils").$Prose)[];

package/dist/index.js

@@ -5,26 +5,66 @@ import { DOMParser, DOMSerializer } from "@milkdown/kit/prose/model";
 import { Plugin, PluginKey } from "@milkdown/kit/prose/state";
 import { Decoration, DecorationSet } from "@milkdown/kit/prose/view";
 import { $prose } from "@milkdown/kit/utils";
+import { DEFAULT_COPILOT_MAX_TOKENS as max_tokens, DEFAULT_COPILOT_STOP_SEQUENCE as stop, DEFAULT_COPILOT_STREAM as stream, DEFAULT_COPILOT_TEMPERATURE as temperature, DEFAULT_COPILOT_TOP_P as top_p, } from "@monacopilot/core";
 import { CompletionCopilot } from "monacopilot";
 import { debounce } from "quasar";
-const BR_TAG_REGEX = /^<br \/>|<br \/>$/g, column = NaN, lineNumber = NaN, cursorPosition = { column, lineNumber }, deco = DecorationSet.empty, name = "MilkdownCopilot", key = new PluginKey(name), language = "markdown", message = "", model = "codestral", provider = "mistral", relatedFiles = undefined, second = 1000, technologies = ["vue", "tailwindcss", "comark"];
+const BR_TAG_REGEX = /^<br \/>|<br \/>$/g, column = NaN, lineNumber = NaN, cursorPosition = { column, lineNumber }, deco = DecorationSet.empty, name = "MilkdownCopilot", key = new PluginKey(name), language = "markdown", message = "", method = "POST", relatedFiles = undefined, second = 1000, technologies = ["vue", "tailwindcss", "comark"];
 const init = () => ({ deco, message });
-export const apiKeySlice = createSlice("", "apiKey"), filenameSlice = createSlice("", "filename");
+export const apiKeySlice = createSlice("", "apiKey"), baseURLSlice = createSlice("", "baseURL"), endpointSlice = createSlice("", "endpoint"), filenameSlice = createSlice("", "filename"), modelSlice = createSlice("", "model");
 export const copilotPlugin = [
     (ctx) => {
         ctx.inject(apiKeySlice);
         ctx.inject(filenameSlice);
+        ctx.inject(modelSlice);
+        ctx.inject(baseURLSlice);
+        ctx.inject(endpointSlice);
         return () => undefined;
     },
     $prose((ctx) => {
-        let copilot, localApiKey = "";
+        let copilot, localApiKey = "", localBaseURL = "", localEndpoint = "", localModel = "";
         const getHint = debounce(async (view) => {
-            const apiKey = ctx.get(apiKeySlice), filename = ctx.get(filenameSlice);
-            if (localApiKey !== apiKey) {
+            const apiKey = ctx.get(apiKeySlice), baseURL = ctx.get(baseURLSlice), endpoint = ctx.get(endpointSlice), filename = ctx.get(filenameSlice), model = ctx.get(modelSlice);
+            if (localApiKey !== apiKey ||
+                localBaseURL !== baseURL ||
+                localEndpoint !== endpoint ||
+                localModel !== model) {
                 localApiKey = apiKey;
-                copilot = localApiKey
-                    ? new CompletionCopilot(localApiKey, { model, provider })
-                    : undefined;
+                localBaseURL = baseURL;
+                localEndpoint = endpoint;
+                localModel = model;
+                copilot =
+                    apiKey && baseURL && model
+                        ? new CompletionCopilot(undefined, {
+                            model: async ({ context, fileContent, instruction }) => {
+                                const [prefix = "", suffix = ""] = fileContent.split("<|developer_cursor_is_here|>"), Authorization = `Bearer ${apiKey}`, prompt = `${context}\n${instruction}\n${prefix}`;
+                                let text = null;
+                                try {
+                                    const { choices: [{ message: { content }, },], } = await (await fetch(`${baseURL}/${endpoint || "completions"}`, {
+                                        body: JSON.stringify({
+                                            max_tokens,
+                                            model,
+                                            prompt,
+                                            stop,
+                                            stream,
+                                            ...(endpoint && { suffix }),
+                                            temperature,
+                                            top_p,
+                                        }),
+                                        headers: {
+                                            Authorization,
+                                            "Content-Type": "application/json",
+                                        },
+                                        method,
+                                    })).json();
+                                    text = content;
+                                }
+                                catch (err) {
+                                    console.log(err);
+                                }
+                                return { text };
+                            },
+                        })
+                        : undefined;
             }
             if (copilot && filename) {
                 const { dispatch, state: { schema: { topNodeType }, tr: { doc, selection: { from }, }, }, } = view, { content: after } = doc.slice(from), textAfterCursor = ctx

package/package.json

@@ -1,7 +1,7 @@
 {
   "$schema": "https://www.schemastore.org/package",
   "name": "@skaldapp/milkdown-copilot",
-  "version": "0.0.22",
+  "version": "1.0.0",
   "description": "A Milkdown plugin that provides AI-powered code completions using Mistral AI’s Codestral model with debounced requests and tab-to-accept functionality.",
   "keywords": [
     "milkdown",
@@ -39,12 +39,13 @@
   "dependencies": {
     "@milkdown/ctx": "^7.21.1",
     "@milkdown/kit": "^7.21.1",
+    "@monacopilot/core": "^1.2.12",
     "monacopilot": "^1.2.12",
     "quasar": "^2.19.3"
   },
   "devDependencies": {
-    "@skaldapp/configs": "^1.2.108",
-    "@types/node": "^25.9.0",
+    "@skaldapp/configs": "^1.2.109",
+    "@types/node": "^25.9.1",
     "eslint": "^10.4.0"
   }
 }