@sweetoburrito/backstage-plugin-ai-assistant-backend-module-embeddings-provider-ollama 0.2.1

package/README.md

@@ -0,0 +1,5 @@
+# @sweetoburrito/backstage-plugin-ai-assistant-backend-module-embeddings-provider-ollama
+
+The embeddings-provider-ollama backend module for the ai-assistant plugin.
+
+_This plugin was created through the Backstage CLI_

package/config.d.ts

@@ -0,0 +1,14 @@
+export interface Config {
+  aiAssistant: {
+    embeddings: {
+      ollama: {
+        baseUrl: string;
+        model: string;
+        /**
+         * @visibility secret
+         */
+        apiKey: string;
+      };
+    };
+  };
+}

package/dist/index.cjs.js

@@ -0,0 +1,10 @@
+'use strict';
+
+Object.defineProperty(exports, '__esModule', { value: true });
+
+var module$1 = require('./module.cjs.js');
+
+
+
+exports.default = module$1.aiAssistantModuleEmbeddingsProviderOllama;
+//# sourceMappingURL=index.cjs.js.map

package/dist/index.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.cjs.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;;;;;;;"}

package/dist/index.d.ts

@@ -0,0 +1,5 @@
+import * as _backstage_backend_plugin_api from '@backstage/backend-plugin-api';
+
+declare const aiAssistantModuleEmbeddingsProviderOllama: _backstage_backend_plugin_api.BackendFeature;
+
+export { aiAssistantModuleEmbeddingsProviderOllama as default };

package/dist/module.cjs.js

@@ -0,0 +1,39 @@
+'use strict';
+
+var backendPluginApi = require('@backstage/backend-plugin-api');
+var backstagePluginAiAssistantNode = require('@sweetoburrito/backstage-plugin-ai-assistant-node');
+var ollama = require('@langchain/ollama');
+
+const aiAssistantModuleEmbeddingsProviderOllama = backendPluginApi.createBackendModule({
+  pluginId: "ai-assistant",
+  moduleId: "embeddings-provider-ollama",
+  register(reg) {
+    reg.registerInit({
+      deps: {
+        logger: backendPluginApi.coreServices.logger,
+        config: backendPluginApi.coreServices.rootConfig,
+        embeddingsProvider: backstagePluginAiAssistantNode.embeddingsProviderExtensionPoint
+      },
+      async init({ embeddingsProvider, config }) {
+        const model = config.getString("aiAssistant.embeddings.ollama.model");
+        const apiKey = config.getString("aiAssistant.embeddings.ollama.apiKey");
+        const baseUrl = config.getString(
+          "aiAssistant.embeddings.ollama.baseUrl"
+        );
+        const embeddings = new ollama.OllamaEmbeddings({
+          baseUrl,
+          model,
+          headers: {
+            Authorization: `Bearer ${apiKey}`
+          }
+        });
+        return embeddingsProvider.register({
+          getEmbeddings: async () => embeddings
+        });
+      }
+    });
+  }
+});
+
+exports.aiAssistantModuleEmbeddingsProviderOllama = aiAssistantModuleEmbeddingsProviderOllama;
+//# sourceMappingURL=module.cjs.js.map

package/dist/module.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"module.cjs.js","sources":["../src/module.ts"],"sourcesContent":["import {\n  coreServices,\n  createBackendModule,\n} from '@backstage/backend-plugin-api';\nimport { embeddingsProviderExtensionPoint } from '@sweetoburrito/backstage-plugin-ai-assistant-node';\nimport { OllamaEmbeddings } from '@langchain/ollama';\n\nexport const aiAssistantModuleEmbeddingsProviderOllama = createBackendModule({\n  pluginId: 'ai-assistant',\n  moduleId: 'embeddings-provider-ollama',\n  register(reg) {\n    reg.registerInit({\n      deps: {\n        logger: coreServices.logger,\n        config: coreServices.rootConfig,\n        embeddingsProvider: embeddingsProviderExtensionPoint,\n      },\n      async init({ embeddingsProvider, config }) {\n        const model = config.getString('aiAssistant.embeddings.ollama.model');\n        const apiKey = config.getString('aiAssistant.embeddings.ollama.apiKey');\n        const baseUrl = config.getString(\n          'aiAssistant.embeddings.ollama.baseUrl',\n        );\n\n        const embeddings = new OllamaEmbeddings({\n          baseUrl,\n          model,\n          headers: {\n            Authorization: `Bearer ${apiKey}`,\n          },\n        });\n\n        return embeddingsProvider.register({\n          getEmbeddings: async () => embeddings,\n        });\n      },\n    });\n  },\n});\n"],"names":["createBackendModule","coreServices","embeddingsProviderExtensionPoint","OllamaEmbeddings"],"mappings":";;;;;;AAOO,MAAM,4CAA4CA,oCAAA,CAAoB;AAAA,EAC3E,QAAA,EAAU,cAAA;AAAA,EACV,QAAA,EAAU,4BAAA;AAAA,EACV,SAAS,GAAA,EAAK;AACZ,IAAA,GAAA,CAAI,YAAA,CAAa;AAAA,MACf,IAAA,EAAM;AAAA,QACJ,QAAQC,6BAAA,CAAa,MAAA;AAAA,QACrB,QAAQA,6BAAA,CAAa,UAAA;AAAA,QACrB,kBAAA,EAAoBC;AAAA,OACtB;AAAA,MACA,MAAM,IAAA,CAAK,EAAE,kBAAA,EAAoB,QAAO,EAAG;AACzC,QAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,SAAA,CAAU,qCAAqC,CAAA;AACpE,QAAA,MAAM,MAAA,GAAS,MAAA,CAAO,SAAA,CAAU,sCAAsC,CAAA;AACtE,QAAA,MAAM,UAAU,MAAA,CAAO,SAAA;AAAA,UACrB;AAAA,SACF;AAEA,QAAA,MAAM,UAAA,GAAa,IAAIC,uBAAA,CAAiB;AAAA,UACtC,OAAA;AAAA,UACA,KAAA;AAAA,UACA,OAAA,EAAS;AAAA,YACP,aAAA,EAAe,UAAU,MAAM,CAAA;AAAA;AACjC,SACD,CAAA;AAED,QAAA,OAAO,mBAAmB,QAAA,CAAS;AAAA,UACjC,eAAe,YAAY;AAAA,SAC5B,CAAA;AAAA,MACH;AAAA,KACD,CAAA;AAAA,EACH;AACF,CAAC;;;;"}

package/package.json

@@ -0,0 +1,51 @@
+{
+  "name": "@sweetoburrito/backstage-plugin-ai-assistant-backend-module-embeddings-provider-ollama",
+  "version": "0.2.1",
+  "license": "Apache-2.0",
+  "description": "The embeddings-provider-ollama backend module for the ai-assistant plugin.",
+  "main": "dist/index.cjs.js",
+  "types": "dist/index.d.ts",
+  "publishConfig": {
+    "access": "public",
+    "main": "dist/index.cjs.js",
+    "types": "dist/index.d.ts"
+  },
+  "backstage": {
+    "role": "backend-plugin-module",
+    "pluginId": "ai-assistant",
+    "pluginPackage": "@sweetoburrito/backstage-plugin-ai-assistant-backend",
+    "features": {
+      ".": "@backstage/BackendFeature"
+    }
+  },
+  "scripts": {
+    "start": "backstage-cli package start",
+    "build": "backstage-cli package build",
+    "lint": "backstage-cli package lint",
+    "test": "backstage-cli package test",
+    "clean": "backstage-cli package clean",
+    "prepack": "backstage-cli package prepack",
+    "postpack": "backstage-cli package postpack"
+  },
+  "dependencies": {
+    "@backstage/backend-plugin-api": "^1.4.1",
+    "@langchain/ollama": "^0.2.3",
+    "@sweetoburrito/backstage-plugin-ai-assistant-node": "workspace:^"
+  },
+  "devDependencies": {
+    "@backstage/backend-test-utils": "^1.7.0",
+    "@backstage/cli": "^0.33.1"
+  },
+  "configSchema": "config.d.ts",
+  "files": [
+    "dist",
+    "config.d.ts"
+  ],
+  "typesVersions": {
+    "*": {
+      "package.json": [
+        "package.json"
+      ]
+    }
+  }
+}