@theia/ai-llamafile 1.46.0-next.241

package/README.md

@@ -0,0 +1,57 @@
+# AI Llamafile Integration
+
+The AI Llamafile package provides an integration that allows users to manage and interact with Llamafile language models within Theia IDE.
+
+## Features
+
+- Start and stop Llamafile language servers.
+
+## Commands
+
+### Start Llamafile
+
+- **Command ID:** `llamafile.start`
+- **Label:** `Start Llamafile`
+- **Functionality:** Allows you to start a Llamafile language server by selecting from a list of configured Llamafiles.
+
+### Stop Llamafile
+
+- **Command ID:** `llamafile.stop`
+- **Label:** `Stop Llamafile`
+- **Functionality:** Allows you to stop a running Llamafile language server by selecting from a list of currently running Llamafiles.
+
+## Usage
+
+1. **Starting a Llamafile Language Server:**
+
+    - Use the command palette to invoke `Start Llamafile`.
+    - A quick pick menu will appear with a list of configured Llamafiles.
+    - Select a Llamafile to start its language server.
+
+2. **Stopping a Llamafile Language Server:**
+    - Use the command palette to invoke `Stop Llamafile`.
+    - A quick pick menu will display a list of currently running Llamafiles.
+    - Select a Llamafile to stop its language server.
+
+## Dependencies
+
+This extension depends on the `@theia/ai-core` package for AI-related services and functionalities.
+
+## Configuration
+
+Make sure to configure your Llamafiles properly within the preference settings.
+This setting is an array of objects, where each object defines a llamafile with a user-friendly name, the file uri, and the port to start the server on.
+
+Example Configuration:
+
+```json
+{
+    "ai-features.llamafile.llamafiles": [
+        {
+            "name": "MyLlamaFile",
+            "uri": "file:///path/to/my.llamafile",
+            "port": 30000
+        }
+    ]
+}
+```

package/lib/browser/llamafile-command-contribution.d.ts

@@ -0,0 +1,21 @@
+import { AICommandHandlerFactory } from '@theia/ai-core/lib/browser/ai-command-handler-factory';
+import { CommandContribution, CommandRegistry, MessageService } from '@theia/core';
+import { PreferenceService, QuickInputService } from '@theia/core/lib/browser';
+import { LlamafileManager } from '../common/llamafile-manager';
+export declare const StartLlamafileCommand: {
+    id: string;
+    label: string;
+};
+export declare const StopLlamafileCommand: {
+    id: string;
+    label: string;
+};
+export declare class LlamafileCommandContribution implements CommandContribution {
+    protected readonly quickInputService: QuickInputService;
+    protected readonly commandHandlerFactory: AICommandHandlerFactory;
+    protected preferenceService: PreferenceService;
+    protected messageService: MessageService;
+    protected llamafileManager: LlamafileManager;
+    registerCommands(commandRegistry: CommandRegistry): void;
+}
+//# sourceMappingURL=llamafile-command-contribution.d.ts.map

package/lib/browser/llamafile-command-contribution.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"llamafile-command-contribution.d.ts","sourceRoot":"","sources":["../../src/browser/llamafile-command-contribution.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,uBAAuB,EAAE,MAAM,uDAAuD,CAAC;AAChG,OAAO,EAAE,mBAAmB,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AACnF,OAAO,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAE/E,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAI/D,eAAO,MAAM,qBAAqB;;;CAGjC,CAAC;AACF,eAAO,MAAM,oBAAoB;;;CAGhC,CAAC;AAEF,qBACa,4BAA6B,YAAW,mBAAmB;IAGpE,SAAS,CAAC,QAAQ,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;IAGxD,SAAS,CAAC,QAAQ,CAAC,qBAAqB,EAAE,uBAAuB,CAAC;IAGlE,SAAS,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;IAG/C,SAAS,CAAC,cAAc,EAAE,cAAc,CAAC;IAGzC,SAAS,CAAC,gBAAgB,EAAE,gBAAgB,CAAC;IAE7C,gBAAgB,CAAC,eAAe,EAAE,eAAe,GAAG,IAAI;CA0C3D"}

package/lib/browser/llamafile-command-contribution.js

@@ -0,0 +1,104 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LlamafileCommandContribution = exports.StopLlamafileCommand = exports.StartLlamafileCommand = void 0;
+const tslib_1 = require("tslib");
+// *****************************************************************************
+// Copyright (C) 2024 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+const ai_command_handler_factory_1 = require("@theia/ai-core/lib/browser/ai-command-handler-factory");
+const core_1 = require("@theia/core");
+const browser_1 = require("@theia/core/lib/browser");
+const inversify_1 = require("@theia/core/shared/inversify");
+const llamafile_manager_1 = require("../common/llamafile-manager");
+const llamafile_preferences_1 = require("./llamafile-preferences");
+exports.StartLlamafileCommand = {
+    id: 'llamafile.start',
+    label: 'Start Llamafile',
+};
+exports.StopLlamafileCommand = {
+    id: 'llamafile.stop',
+    label: 'Stop Llamafile',
+};
+let LlamafileCommandContribution = class LlamafileCommandContribution {
+    registerCommands(commandRegistry) {
+        commandRegistry.registerCommand(exports.StartLlamafileCommand, this.commandHandlerFactory({
+            execute: async () => {
+                try {
+                    const llamaFiles = this.preferenceService.get(llamafile_preferences_1.PREFERENCE_LLAMAFILE);
+                    if (llamaFiles === undefined || llamaFiles.length === 0) {
+                        this.messageService.error('No Llamafiles configured.');
+                        return;
+                    }
+                    const options = llamaFiles.map(llamaFile => ({ label: llamaFile.name }));
+                    const result = await this.quickInputService.showQuickPick(options);
+                    if (result === undefined) {
+                        return;
+                    }
+                    this.llamafileManager.startServer(result.label);
+                }
+                catch (error) {
+                    console.error('Something went wrong during the llamafile start.', error);
+                    this.messageService.error(`Something went wrong during the llamafile start: ${error.message}.\nFor more information, see the console.`);
+                }
+            }
+        }));
+        commandRegistry.registerCommand(exports.StopLlamafileCommand, this.commandHandlerFactory({
+            execute: async () => {
+                try {
+                    const llamaFiles = await this.llamafileManager.getStartedLlamafiles();
+                    if (llamaFiles === undefined || llamaFiles.length === 0) {
+                        this.messageService.error('No Llamafiles running.');
+                        return;
+                    }
+                    const options = llamaFiles.map(llamaFile => ({ label: llamaFile }));
+                    const result = await this.quickInputService.showQuickPick(options);
+                    if (result === undefined) {
+                        return;
+                    }
+                    this.llamafileManager.stopServer(result.label);
+                }
+                catch (error) {
+                    console.error('Something went wrong during the llamafile stop.', error);
+                    this.messageService.error(`Something went wrong during the llamafile stop: ${error.message}.\nFor more information, see the console.`);
+                }
+            }
+        }));
+    }
+};
+exports.LlamafileCommandContribution = LlamafileCommandContribution;
+tslib_1.__decorate([
+    (0, inversify_1.inject)(browser_1.QuickInputService),
+    tslib_1.__metadata("design:type", Object)
+], LlamafileCommandContribution.prototype, "quickInputService", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.inject)(ai_command_handler_factory_1.AICommandHandlerFactory),
+    tslib_1.__metadata("design:type", Function)
+], LlamafileCommandContribution.prototype, "commandHandlerFactory", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.inject)(browser_1.PreferenceService),
+    tslib_1.__metadata("design:type", Object)
+], LlamafileCommandContribution.prototype, "preferenceService", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.inject)(core_1.MessageService),
+    tslib_1.__metadata("design:type", core_1.MessageService)
+], LlamafileCommandContribution.prototype, "messageService", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.inject)(llamafile_manager_1.LlamafileManager),
+    tslib_1.__metadata("design:type", Object)
+], LlamafileCommandContribution.prototype, "llamafileManager", void 0);
+exports.LlamafileCommandContribution = LlamafileCommandContribution = tslib_1.__decorate([
+    (0, inversify_1.injectable)()
+], LlamafileCommandContribution);
+//# sourceMappingURL=llamafile-command-contribution.js.map

package/lib/browser/llamafile-command-contribution.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llamafile-command-contribution.js","sourceRoot":"","sources":["../../src/browser/llamafile-command-contribution.ts"],"names":[],"mappings":";;;;AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;AAChF,sGAAgG;AAChG,sCAAmF;AACnF,qDAA+E;AAC/E,4DAAkE;AAClE,mEAA+D;AAC/D,mEAA+D;AAGlD,QAAA,qBAAqB,GAAG;IACjC,EAAE,EAAE,iBAAiB;IACrB,KAAK,EAAE,iBAAiB;CAC3B,CAAC;AACW,QAAA,oBAAoB,GAAG;IAChC,EAAE,EAAE,gBAAgB;IACpB,KAAK,EAAE,gBAAgB;CAC1B,CAAC;AAGK,IAAM,4BAA4B,GAAlC,MAAM,4BAA4B;IAiBrC,gBAAgB,CAAC,eAAgC;QAC7C,eAAe,CAAC,eAAe,CAAC,6BAAqB,EAAE,IAAI,CAAC,qBAAqB,CAAC;YAC9E,OAAO,EAAE,KAAK,IAAI,EAAE;gBAChB,IAAI,CAAC;oBACD,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAmB,4CAAoB,CAAC,CAAC;oBACtF,IAAI,UAAU,KAAK,SAAS,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;wBACtD,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;wBACvD,OAAO;oBACX,CAAC;oBACD,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;oBACzE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;oBACnE,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;wBACvB,OAAO;oBACX,CAAC;oBACD,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACpD,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACb,OAAO,CAAC,KAAK,CAAC,kDAAkD,EAAE,KAAK,CAAC,CAAC;oBACzE,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,oDAAoD,KAAK,CAAC,OAAO,2CAA2C,CAAC,CAAC;gBAC5I,CAAC;YACL,CAAC;SACJ,CAAC,CAAC,CAAC;QACJ,eAAe,CAAC,eAAe,CAAC,4BAAoB,EAAE,IAAI,CAAC,qBAAqB,CAAC;YAC7E,OAAO,EAAE,KAAK,IAAI,EAAE;gBAChB,IAAI,CAAC;oBACD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,oBAAoB,EAAE,CAAC;oBACtE,IAAI,UAAU,KAAK,SAAS,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;wBACtD,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;wBACpD,OAAO;oBACX,CAAC;oBACD,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;oBACpE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;oBACnE,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;wBACvB,OAAO;oBACX,CAAC;oBACD,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACnD,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACb,OAAO,CAAC,KAAK,CAAC,iDAAiD,EAAE,KAAK,CAAC,CAAC;oBACxE,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,mDAAmD,KAAK,CAAC,OAAO,2CAA2C,CAAC,CAAC;gBAC3I,CAAC;YACL,CAAC;SACJ,CAAC,CAAC,CAAC;IACR,CAAC;CACJ,CAAA;AA3DY,oEAA4B;AAGlB;IADlB,IAAA,kBAAM,EAAC,2BAAiB,CAAC;;uEAC8B;AAGrC;IADlB,IAAA,kBAAM,EAAC,oDAAuB,CAAC;;2EACkC;AAGxD;IADT,IAAA,kBAAM,EAAC,2BAAiB,CAAC;;uEACqB;AAGrC;IADT,IAAA,kBAAM,EAAC,qBAAc,CAAC;sCACG,qBAAc;oEAAC;AAG/B;IADT,IAAA,kBAAM,EAAC,oCAAgB,CAAC;;sEACoB;uCAfpC,4BAA4B;IADxC,IAAA,sBAAU,GAAE;GACA,4BAA4B,CA2DxC"}

package/lib/browser/llamafile-frontend-application-contribution.d.ts

@@ -0,0 +1,18 @@
+import { FrontendApplicationContribution, PreferenceService } from '@theia/core/lib/browser';
+import { LlamafileManager, LlamafileModelDescription } from '../common/llamafile-manager';
+import { RequestSetting } from '@theia/ai-core/lib/browser/ai-core-preferences';
+export declare class LlamafileFrontendApplicationContribution implements FrontendApplicationContribution {
+    protected preferenceService: PreferenceService;
+    protected llamafileManager: LlamafileManager;
+    private _knownLlamaFiles;
+    onStart(): void;
+    protected getLLamaFileModelDescriptions(llamafiles: LlamafileEntry[]): LlamafileModelDescription[];
+    protected handleLlamaFilePreferenceChange(newModels: LlamafileEntry[]): void;
+    protected handleRequestSettingsChange(newSettings: RequestSetting[]): void;
+}
+export interface LlamafileEntry {
+    name: string;
+    uri: string;
+    port: number;
+}
+//# sourceMappingURL=llamafile-frontend-application-contribution.d.ts.map

package/lib/browser/llamafile-frontend-application-contribution.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"llamafile-frontend-application-contribution.d.ts","sourceRoot":"","sources":["../../src/browser/llamafile-frontend-application-contribution.ts"],"names":[],"mappings":"AAgBA,OAAO,EAAE,+BAA+B,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAE7F,OAAO,EAAE,gBAAgB,EAAE,yBAAyB,EAAE,MAAM,6BAA6B,CAAC;AAE1F,OAAO,EAAoC,cAAc,EAAE,MAAM,gDAAgD,CAAC;AAGlH,qBACa,wCAAyC,YAAW,+BAA+B;IAG5F,SAAS,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;IAG/C,SAAS,CAAC,gBAAgB,EAAE,gBAAgB,CAAC;IAE7C,OAAO,CAAC,gBAAgB,CAA0C;IAElE,OAAO,IAAI,IAAI;IAqBf,SAAS,CAAC,6BAA6B,CAAC,UAAU,EAAE,cAAc,EAAE,GAAG,yBAAyB,EAAE;IAoBlG,SAAS,CAAC,+BAA+B,CAAC,SAAS,EAAE,cAAc,EAAE,GAAG,IAAI;IAgB5E,SAAS,CAAC,2BAA2B,CAAC,WAAW,EAAE,cAAc,EAAE,GAAG,IAAI;CAO7E;AAED,MAAM,WAAW,cAAc;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;CAChB"}

package/lib/browser/llamafile-frontend-application-contribution.js

@@ -0,0 +1,113 @@
+"use strict";
+// *****************************************************************************
+// Copyright (C) 2024 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LlamafileFrontendApplicationContribution = void 0;
+const tslib_1 = require("tslib");
+const browser_1 = require("@theia/core/lib/browser");
+const inversify_1 = require("@theia/core/shared/inversify");
+const llamafile_manager_1 = require("../common/llamafile-manager");
+const llamafile_preferences_1 = require("./llamafile-preferences");
+const ai_core_preferences_1 = require("@theia/ai-core/lib/browser/ai-core-preferences");
+const LLAMAFILE_PROVIDER_ID = 'llamafile';
+let LlamafileFrontendApplicationContribution = class LlamafileFrontendApplicationContribution {
+    constructor() {
+        this._knownLlamaFiles = new Map();
+    }
+    onStart() {
+        this.preferenceService.ready.then(() => {
+            const llamafiles = this.preferenceService.get(llamafile_preferences_1.PREFERENCE_LLAMAFILE, []);
+            const validLlamafiles = llamafiles.filter(LlamafileEntry.is);
+            const LlamafileModelDescriptions = this.getLLamaFileModelDescriptions(validLlamafiles);
+            this.llamafileManager.addLanguageModels(LlamafileModelDescriptions);
+            validLlamafiles.forEach(model => this._knownLlamaFiles.set(model.name, model));
+            this.preferenceService.onPreferenceChanged(event => {
+                if (event.preferenceName === llamafile_preferences_1.PREFERENCE_LLAMAFILE) {
+                    const newModels = event.newValue.filter((llamafileEntry) => LlamafileEntry.is(llamafileEntry));
+                    this.handleLlamaFilePreferenceChange(newModels);
+                }
+                else if (event.preferenceName === ai_core_preferences_1.PREFERENCE_NAME_REQUEST_SETTINGS) {
+                    this.handleRequestSettingsChange(event.newValue);
+                }
+            });
+        });
+    }
+    getLLamaFileModelDescriptions(llamafiles) {
+        const requestSettings = this.preferenceService.get(ai_core_preferences_1.PREFERENCE_NAME_REQUEST_SETTINGS, []);
+        return llamafiles.map(llamafile => {
+            var _a;
+            const matchingSettings = requestSettings.filter(setting => (!setting.providerId || setting.providerId === LLAMAFILE_PROVIDER_ID) &&
+                setting.modelId === llamafile.name);
+            if (matchingSettings.length > 1) {
+                console.warn(`Multiple entries found for model "${llamafile.name}". Using the first match.`);
+            }
+            return {
+                name: llamafile.name,
+                uri: llamafile.uri,
+                port: llamafile.port,
+                defaultRequestSettings: (_a = matchingSettings[0]) === null || _a === void 0 ? void 0 : _a.requestSettings
+            };
+        });
+    }
+    handleLlamaFilePreferenceChange(newModels) {
+        const llamafilesToAdd = newModels.filter(llamafile => !this._knownLlamaFiles.has(llamafile.name) ||
+            !LlamafileEntry.equals(this._knownLlamaFiles.get(llamafile.name), llamafile));
+        const llamafileIdsToRemove = [...this._knownLlamaFiles.values()].filter(llamafile => !newModels.find(newModel => LlamafileEntry.equals(newModel, llamafile)))
+            .map(llamafile => llamafile.name);
+        this.llamafileManager.removeLanguageModels(llamafileIdsToRemove);
+        llamafileIdsToRemove.forEach(id => this._knownLlamaFiles.delete(id));
+        this.llamafileManager.addLanguageModels(this.getLLamaFileModelDescriptions(llamafilesToAdd));
+        llamafilesToAdd.forEach(model => this._knownLlamaFiles.set(model.name, model));
+    }
+    handleRequestSettingsChange(newSettings) {
+        const llamafiles = Array.from(this._knownLlamaFiles.values());
+        const llamafileModelDescriptions = this.getLLamaFileModelDescriptions(llamafiles);
+        llamafileModelDescriptions.forEach(llamafileModelDescription => {
+            this.llamafileManager.updateRequestSettings(llamafileModelDescription.name, llamafileModelDescription.defaultRequestSettings);
+        });
+    }
+};
+exports.LlamafileFrontendApplicationContribution = LlamafileFrontendApplicationContribution;
+tslib_1.__decorate([
+    (0, inversify_1.inject)(browser_1.PreferenceService),
+    tslib_1.__metadata("design:type", Object)
+], LlamafileFrontendApplicationContribution.prototype, "preferenceService", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.inject)(llamafile_manager_1.LlamafileManager),
+    tslib_1.__metadata("design:type", Object)
+], LlamafileFrontendApplicationContribution.prototype, "llamafileManager", void 0);
+exports.LlamafileFrontendApplicationContribution = LlamafileFrontendApplicationContribution = tslib_1.__decorate([
+    (0, inversify_1.injectable)()
+], LlamafileFrontendApplicationContribution);
+var LlamafileEntry;
+(function (LlamafileEntry) {
+    function equals(a, b) {
+        return (a.name === b.name &&
+            a.uri === b.uri &&
+            a.port === b.port);
+    }
+    LlamafileEntry.equals = equals;
+    function is(entry) {
+        return (typeof entry === 'object' &&
+            // eslint-disable-next-line no-null/no-null
+            entry !== null &&
+            'name' in entry && typeof entry.name === 'string' &&
+            'uri' in entry && typeof entry.uri === 'string' &&
+            'port' in entry && typeof entry.port === 'number');
+    }
+    LlamafileEntry.is = is;
+})(LlamafileEntry || (LlamafileEntry = {}));
+//# sourceMappingURL=llamafile-frontend-application-contribution.js.map

package/lib/browser/llamafile-frontend-application-contribution.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llamafile-frontend-application-contribution.js","sourceRoot":"","sources":["../../src/browser/llamafile-frontend-application-contribution.ts"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;;;AAEhF,qDAA6F;AAC7F,4DAAkE;AAClE,mEAA0F;AAC1F,mEAA+D;AAC/D,wFAAkH;AAElH,MAAM,qBAAqB,GAAG,WAAW,CAAC;AAEnC,IAAM,wCAAwC,GAA9C,MAAM,wCAAwC;IAA9C;QAQK,qBAAgB,GAAgC,IAAI,GAAG,EAAE,CAAC;IAkEtE,CAAC;IAhEG,OAAO;QACH,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE;YACnC,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAmB,4CAAoB,EAAE,EAAE,CAAC,CAAC;YAC1F,MAAM,eAAe,GAAG,UAAU,CAAC,MAAM,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;YAE7D,MAAM,0BAA0B,GAAG,IAAI,CAAC,6BAA6B,CAAC,eAAe,CAAC,CAAC;YAEvF,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,0BAA0B,CAAC,CAAC;YACpE,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;YAE/E,IAAI,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,KAAK,CAAC,EAAE;gBAC/C,IAAI,KAAK,CAAC,cAAc,KAAK,4CAAoB,EAAE,CAAC;oBAChD,MAAM,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,cAAuB,EAAE,EAAE,CAAC,cAAc,CAAC,EAAE,CAAC,cAAc,CAAC,CAAqB,CAAC;oBAC5H,IAAI,CAAC,+BAA+B,CAAC,SAAS,CAAC,CAAC;gBACpD,CAAC;qBAAM,IAAI,KAAK,CAAC,cAAc,KAAK,sDAAgC,EAAE,CAAC;oBACnE,IAAI,CAAC,2BAA2B,CAAC,KAAK,CAAC,QAA4B,CAAC,CAAC;gBACzE,CAAC;YACL,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;IACP,CAAC;IAES,6BAA6B,CAAC,UAA4B;QAChE,MAAM,eAAe,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAmB,sDAAgC,EAAE,EAAE,CAAC,CAAC;QAC3G,OAAO,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE;;YAC9B,MAAM,gBAAgB,GAAG,eAAe,CAAC,MAAM,CAC3C,OAAO,CAAC,EAAE,CACN,CAAC,CAAC,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,KAAK,qBAAqB,CAAC;gBACrE,OAAO,CAAC,OAAO,KAAK,SAAS,CAAC,IAAI,CACzC,CAAC;YACF,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9B,OAAO,CAAC,IAAI,CAAC,qCAAqC,SAAS,CAAC,IAAI,2BAA2B,CAAC,CAAC;YACjG,CAAC;YACD,OAAO;gBACH,IAAI,EAAE,SAAS,CAAC,IAAI;gBACpB,GAAG,EAAE,SAAS,CAAC,GAAG;gBAClB,IAAI,EAAE,SAAS,CAAC,IAAI;gBACpB,sBAAsB,EAAE,MAAA,gBAAgB,CAAC,CAAC,CAAC,0CAAE,eAAe;aAC/D,CAAC;QACN,CAAC,CAAC,CAAC;IACP,CAAC;IAES,+BAA+B,CAAC,SAA2B;QACjE,MAAM,eAAe,GAAG,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CACjD,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC;YAC1C,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAE,EAAE,SAAS,CAAC,CAAC,CAAC;QAEnF,MAAM,oBAAoB,GAAG,CAAC,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAChF,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,cAAc,CAAC,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC;aACvE,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAEtC,IAAI,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,oBAAoB,CAAC,CAAC;QACjE,oBAAoB,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAErE,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,IAAI,CAAC,6BAA6B,CAAC,eAAe,CAAC,CAAC,CAAC;QAC7F,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;IACnF,CAAC;IAES,2BAA2B,CAAC,WAA6B;QAC/D,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9D,MAAM,0BAA0B,GAAG,IAAI,CAAC,6BAA6B,CAAC,UAAU,CAAC,CAAC;QAClF,0BAA0B,CAAC,OAAO,CAAC,yBAAyB,CAAC,EAAE;YAC3D,IAAI,CAAC,gBAAgB,CAAC,qBAAqB,CAAC,yBAAyB,CAAC,IAAI,EAAE,yBAAyB,CAAC,sBAAsB,CAAC,CAAC;QAClI,CAAC,CAAC,CAAC;IACP,CAAC;CACJ,CAAA;AA1EY,4FAAwC;AAGvC;IADT,IAAA,kBAAM,EAAC,2BAAiB,CAAC;;mFACqB;AAGrC;IADT,IAAA,kBAAM,EAAC,oCAAgB,CAAC;;kFACoB;mDANpC,wCAAwC;IADpD,IAAA,sBAAU,GAAE;GACA,wCAAwC,CA0EpD;AAQD,IAAU,cAAc,CAmBvB;AAnBD,WAAU,cAAc;IACpB,SAAgB,MAAM,CAAC,CAAiB,EAAE,CAAiB;QACvD,OAAO,CACH,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI;YACjB,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,GAAG;YACf,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,CACpB,CAAC;IACN,CAAC;IANe,qBAAM,SAMrB,CAAA;IAED,SAAgB,EAAE,CAAC,KAAc;QAC7B,OAAO,CACH,OAAO,KAAK,KAAK,QAAQ;YACzB,2CAA2C;YAC3C,KAAK,KAAK,IAAI;YACd,MAAM,IAAI,KAAK,IAAI,OAAQ,KAAwB,CAAC,IAAI,KAAK,QAAQ;YACrE,KAAK,IAAI,KAAK,IAAI,OAAQ,KAAwB,CAAC,GAAG,KAAK,QAAQ;YACnE,MAAM,IAAI,KAAK,IAAI,OAAQ,KAAwB,CAAC,IAAI,KAAK,QAAQ,CACxE,CAAC;IACN,CAAC;IATe,iBAAE,KASjB,CAAA;AACL,CAAC,EAnBS,cAAc,KAAd,cAAc,QAmBvB"}

package/lib/browser/llamafile-frontend-module.d.ts

@@ -0,0 +1,4 @@
+import { ContainerModule } from '@theia/core/shared/inversify';
+declare const _default: ContainerModule;
+export default _default;
+//# sourceMappingURL=llamafile-frontend-module.d.ts.map

package/lib/browser/llamafile-frontend-module.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"llamafile-frontend-module.d.ts","sourceRoot":"","sources":["../../src/browser/llamafile-frontend-module.ts"],"names":[],"mappings":"AAiBA,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;;AAO/D,wBAoBG"}

package/lib/browser/llamafile-frontend-module.js

@@ -0,0 +1,46 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+// *****************************************************************************
+// Copyright (C) 2024 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+const core_1 = require("@theia/core");
+const browser_1 = require("@theia/core/lib/browser");
+const inversify_1 = require("@theia/core/shared/inversify");
+const output_channel_1 = require("@theia/output/lib/browser/output-channel");
+const llamafile_manager_1 = require("../common/llamafile-manager");
+const llamafile_command_contribution_1 = require("./llamafile-command-contribution");
+const llamafile_frontend_application_contribution_1 = require("./llamafile-frontend-application-contribution");
+const llamafile_preferences_1 = require("./llamafile-preferences");
+exports.default = new inversify_1.ContainerModule(bind => {
+    bind(browser_1.FrontendApplicationContribution).to(llamafile_frontend_application_contribution_1.LlamafileFrontendApplicationContribution).inSingletonScope();
+    bind(core_1.CommandContribution).to(llamafile_command_contribution_1.LlamafileCommandContribution).inSingletonScope();
+    bind(llamafile_manager_1.LlamafileManager).toDynamicValue(ctx => {
+        const connection = ctx.container.get(browser_1.RemoteConnectionProvider);
+        const outputChannelManager = ctx.container.get(output_channel_1.OutputChannelManager);
+        const client = {
+            error: (llamafileName, message) => {
+                const channel = outputChannelManager.getChannel(`${llamafileName}-llamafile`);
+                channel.appendLine(message, output_channel_1.OutputChannelSeverity.Error);
+            },
+            log: (llamafileName, message) => {
+                const channel = outputChannelManager.getChannel(`${llamafileName}-llamafile`);
+                channel.appendLine(message, output_channel_1.OutputChannelSeverity.Info);
+            }
+        };
+        return connection.createProxy(llamafile_manager_1.LlamafileManagerPath, client);
+    }).inSingletonScope();
+    (0, llamafile_preferences_1.bindAILlamafilePreferences)(bind);
+});
+//# sourceMappingURL=llamafile-frontend-module.js.map

package/lib/browser/llamafile-frontend-module.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llamafile-frontend-module.js","sourceRoot":"","sources":["../../src/browser/llamafile-frontend-module.ts"],"names":[],"mappings":";;AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;AAChF,sCAAkD;AAClD,qDAA+H;AAC/H,4DAA+D;AAC/D,6EAAuG;AACvG,mEAAmH;AACnH,qFAAgF;AAChF,+GAAyG;AACzG,mEAAqE;AAErE,kBAAe,IAAI,2BAAe,CAAC,IAAI,CAAC,EAAE;IACtC,IAAI,CAAC,yCAA+B,CAAC,CAAC,EAAE,CAAC,sFAAwC,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACtG,IAAI,CAAC,0BAAmB,CAAC,CAAC,EAAE,CAAC,6DAA4B,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC9E,IAAI,CAAC,oCAAgB,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,EAAE;QACxC,MAAM,UAAU,GAAG,GAAG,CAAC,SAAS,CAAC,GAAG,CAA4B,kCAAwB,CAAC,CAAC;QAC1F,MAAM,oBAAoB,GAAG,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,qCAAoB,CAAC,CAAC;QACrE,MAAM,MAAM,GAAiC;YACzC,KAAK,EAAE,CAAC,aAAa,EAAE,OAAO,EAAE,EAAE;gBAC9B,MAAM,OAAO,GAAG,oBAAoB,CAAC,UAAU,CAAC,GAAG,aAAa,YAAY,CAAC,CAAC;gBAC9E,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,sCAAqB,CAAC,KAAK,CAAC,CAAC;YAC7D,CAAC;YACD,GAAG,EAAE,CAAC,aAAa,EAAE,OAAO,EAAE,EAAE;gBAC5B,MAAM,OAAO,GAAG,oBAAoB,CAAC,UAAU,CAAC,GAAG,aAAa,YAAY,CAAC,CAAC;gBAC9E,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,sCAAqB,CAAC,IAAI,CAAC,CAAC;YAC5D,CAAC;SACJ,CAAC;QACF,OAAO,UAAU,CAAC,WAAW,CAAmB,wCAAoB,EAAE,MAAM,CAAC,CAAC;IAClF,CAAC,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAEtB,IAAA,kDAA0B,EAAC,IAAI,CAAC,CAAC;AACrC,CAAC,CAAC,CAAC"}

package/lib/browser/llamafile-preferences.d.ts

@@ -0,0 +1,7 @@
+import { PreferenceSchema } from '@theia/core/lib/browser';
+import { interfaces } from '@theia/core/shared/inversify';
+export declare const AI_LLAMAFILE_PREFERENCES_TITLE = "\u2728 AI LlamaFile";
+export declare const PREFERENCE_LLAMAFILE = "ai-features.llamafile.llamafiles";
+export declare const aiLlamafilePreferencesSchema: PreferenceSchema;
+export declare function bindAILlamafilePreferences(bind: interfaces.Bind): void;
+//# sourceMappingURL=llamafile-preferences.d.ts.map

package/lib/browser/llamafile-preferences.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"llamafile-preferences.d.ts","sourceRoot":"","sources":["../../src/browser/llamafile-preferences.ts"],"names":[],"mappings":"AAgBA,OAAO,EAA0B,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AACnF,OAAO,EAAE,UAAU,EAAE,MAAM,8BAA8B,CAAC;AAE1D,eAAO,MAAM,8BAA8B,wBAAmB,CAAC;AAC/D,eAAO,MAAM,oBAAoB,qCAAqC,CAAC;AAEvE,eAAO,MAAM,4BAA4B,EAAE,gBAmC1C,CAAC;AAEF,wBAAgB,0BAA0B,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,GAAG,IAAI,CAEtE"}

package/lib/browser/llamafile-preferences.js

@@ -0,0 +1,62 @@
+"use strict";
+// *****************************************************************************
+// Copyright (C) 2024 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.bindAILlamafilePreferences = exports.aiLlamafilePreferencesSchema = exports.PREFERENCE_LLAMAFILE = exports.AI_LLAMAFILE_PREFERENCES_TITLE = void 0;
+const browser_1 = require("@theia/core/lib/browser");
+exports.AI_LLAMAFILE_PREFERENCES_TITLE = '✨ AI LlamaFile';
+exports.PREFERENCE_LLAMAFILE = 'ai-features.llamafile.llamafiles';
+exports.aiLlamafilePreferencesSchema = {
+    type: 'object',
+    properties: {
+        [exports.PREFERENCE_LLAMAFILE]: {
+            title: exports.AI_LLAMAFILE_PREFERENCES_TITLE,
+            markdownDescription: 'This setting allows you to configure and manage LlamaFile models in Theia IDE.\
+            \n\
+            Each entry requires a user-friendly `name`, the file `uri` pointing to your LlamaFile, and the `port` on which it will run.\
+            \n\
+            To start a LlamaFile, use the "Start LlamaFile" command, which enables you to select the desired model.\
+            \n\
+            If you edit an entry (e.g., change the port), any running instance will stop, and you will need to manually start it again.\
+            \n\
+            [Learn more about configuring and managing LlamaFiles in the Theia IDE documentation](https://theia-ide.org/docs/user_ai/#llamafile-models).',
+            type: 'array',
+            default: [],
+            items: {
+                type: 'object',
+                properties: {
+                    name: {
+                        type: 'string',
+                        description: 'The model name to use for this Llamafile.'
+                    },
+                    uri: {
+                        type: 'string',
+                        description: 'The file uri to the Llamafile.'
+                    },
+                    port: {
+                        type: 'number',
+                        description: 'The port to use to start the server.'
+                    }
+                }
+            }
+        }
+    }
+};
+function bindAILlamafilePreferences(bind) {
+    bind(browser_1.PreferenceContribution).toConstantValue({ schema: exports.aiLlamafilePreferencesSchema });
+}
+exports.bindAILlamafilePreferences = bindAILlamafilePreferences;
+//# sourceMappingURL=llamafile-preferences.js.map

package/lib/browser/llamafile-preferences.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llamafile-preferences.js","sourceRoot":"","sources":["../../src/browser/llamafile-preferences.ts"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;;AAEhF,qDAAmF;AAGtE,QAAA,8BAA8B,GAAG,gBAAgB,CAAC;AAClD,QAAA,oBAAoB,GAAG,kCAAkC,CAAC;AAE1D,QAAA,4BAA4B,GAAqB;IAC1D,IAAI,EAAE,QAAQ;IACd,UAAU,EAAE;QACR,CAAC,4BAAoB,CAAC,EAAE;YACpB,KAAK,EAAE,sCAA8B;YACrC,mBAAmB,EAAE;;;;;;;;yJAQwH;YAC7I,IAAI,EAAE,OAAO;YACb,OAAO,EAAE,EAAE;YACX,KAAK,EAAE;gBACH,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACR,IAAI,EAAE;wBACF,IAAI,EAAE,QAAQ;wBACd,WAAW,EAAE,2CAA2C;qBAC3D;oBACD,GAAG,EAAE;wBACD,IAAI,EAAE,QAAQ;wBACd,WAAW,EAAE,gCAAgC;qBAChD;oBACD,IAAI,EAAE;wBACF,IAAI,EAAE,QAAQ;wBACd,WAAW,EAAE,sCAAsC;qBACtD;iBACJ;aACJ;SACJ;KACJ;CACJ,CAAC;AAEF,SAAgB,0BAA0B,CAAC,IAAqB;IAC5D,IAAI,CAAC,gCAAsB,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,oCAA4B,EAAE,CAAC,CAAC;AAC3F,CAAC;AAFD,gEAEC"}

package/lib/common/llamafile-language-model.d.ts

@@ -0,0 +1,25 @@
+import { LanguageModel, LanguageModelRequest, LanguageModelResponse } from '@theia/ai-core';
+import { CancellationToken } from '@theia/core';
+export declare class LlamafileLanguageModel implements LanguageModel {
+    readonly name: string;
+    readonly uri: string;
+    readonly port: number;
+    defaultRequestSettings?: {
+        [key: string]: unknown;
+    } | undefined;
+    readonly providerId = "llamafile";
+    readonly vendor: string;
+    /**
+     * @param name the unique name for this language model. It will be used to identify the model in the UI.
+     * @param uri the URI pointing to the Llamafile model location.
+     * @param port the port on which the Llamafile model server operates.
+     * @param defaultRequestSettings optional default settings for requests made using this model.
+     */
+    constructor(name: string, uri: string, port: number, defaultRequestSettings?: {
+        [key: string]: unknown;
+    } | undefined);
+    get id(): string;
+    protected getSettings(request: LanguageModelRequest): Record<string, unknown>;
+    request(request: LanguageModelRequest, cancellationToken?: CancellationToken): Promise<LanguageModelResponse>;
+}
+//# sourceMappingURL=llamafile-language-model.d.ts.map

package/lib/common/llamafile-language-model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"llamafile-language-model.d.ts","sourceRoot":"","sources":["../../src/common/llamafile-language-model.ts"],"names":[],"mappings":"AAgBA,OAAO,EAAE,aAAa,EAAE,oBAAoB,EAAE,qBAAqB,EAAmC,MAAM,gBAAgB,CAAC;AAC7H,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAChD,qBAAa,sBAAuB,YAAW,aAAa;aAYpC,IAAI,EAAE,MAAM;aACZ,GAAG,EAAE,MAAM;aACX,IAAI,EAAE,MAAM;IACrB,sBAAsB,CAAC;;;IAblC,QAAQ,CAAC,UAAU,eAAe;IAClC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAa;IAEpC;;;;;OAKG;gBAEiB,IAAI,EAAE,MAAM,EACZ,GAAG,EAAE,MAAM,EACX,IAAI,EAAE,MAAM,EACrB,sBAAsB,CAAC;;iBAA4B;IAG9D,IAAI,EAAE,IAAI,MAAM,CAEf;IACD,SAAS,CAAC,WAAW,CAAC,OAAO,EAAE,oBAAoB,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAavE,OAAO,CAAC,OAAO,EAAE,oBAAoB,EAAE,iBAAiB,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,qBAAqB,CAAC;CAyEtH"}

package/lib/common/llamafile-language-model.js

@@ -0,0 +1,122 @@
+"use strict";
+// *****************************************************************************
+// Copyright (C) 2024 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LlamafileLanguageModel = void 0;
+class LlamafileLanguageModel {
+    /**
+     * @param name the unique name for this language model. It will be used to identify the model in the UI.
+     * @param uri the URI pointing to the Llamafile model location.
+     * @param port the port on which the Llamafile model server operates.
+     * @param defaultRequestSettings optional default settings for requests made using this model.
+     */
+    constructor(name, uri, port, defaultRequestSettings) {
+        this.name = name;
+        this.uri = uri;
+        this.port = port;
+        this.defaultRequestSettings = defaultRequestSettings;
+        this.providerId = 'llamafile';
+        this.vendor = 'Mozilla';
+    }
+    get id() {
+        return this.name;
+    }
+    getSettings(request) {
+        const settings = request.settings ? request.settings : this.defaultRequestSettings;
+        if (!settings) {
+            return {
+                n_predict: 200,
+                stream: true,
+                stop: ['</s>', 'Llama:', 'User:', '<|eot_id|>'],
+                cache_prompt: true,
+            };
+        }
+        return settings;
+    }
+    async request(request, cancellationToken) {
+        const settings = this.getSettings(request);
+        try {
+            let prompt = request.messages.map(message => {
+                switch (message.actor) {
+                    case 'user':
+                        return `User: ${message.query}`;
+                    case 'ai':
+                        return `Llama: ${message.query}`;
+                    case 'system':
+                        return `${message.query.replace(/\n\n/g, '\n')}`;
+                }
+            }).join('\n');
+            prompt += '\nLlama:';
+            const response = await fetch(`http://localhost:${this.port}/completion`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+                body: JSON.stringify({
+                    prompt: prompt,
+                    ...settings
+                }),
+            });
+            if (!response.ok) {
+                throw new Error(`HTTP error! status: ${response.status}`);
+            }
+            if (!response.body) {
+                throw new Error('Response body is undefined');
+            }
+            const reader = response.body.getReader();
+            const decoder = new TextDecoder();
+            return {
+                stream: {
+                    [Symbol.asyncIterator]() {
+                        return {
+                            async next() {
+                                if (cancellationToken === null || cancellationToken === void 0 ? void 0 : cancellationToken.isCancellationRequested) {
+                                    reader.cancel();
+                                    return { value: undefined, done: true };
+                                }
+                                const { value, done } = await reader.read();
+                                if (done) {
+                                    return { value: undefined, done: true };
+                                }
+                                const read = decoder.decode(value, { stream: true });
+                                const chunk = read.split('\n').filter(l => l.length !== 0).reduce((acc, line) => {
+                                    try {
+                                        const parsed = JSON.parse(line.substring(6));
+                                        acc += parsed.content;
+                                        return acc;
+                                    }
+                                    catch (error) {
+                                        console.error('Error parsing JSON:', error);
+                                        return acc;
+                                    }
+                                }, '');
+                                return { value: { content: chunk }, done: false };
+                            }
+                        };
+                    }
+                }
+            };
+        }
+        catch (error) {
+            console.error('Error:', error);
+            return {
+                text: `Error: ${error}`
+            };
+        }
+    }
+}
+exports.LlamafileLanguageModel = LlamafileLanguageModel;
+//# sourceMappingURL=llamafile-language-model.js.map