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

package/README.md

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

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.aiAssistantModuleModelProviderOllama;
+//# 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 aiAssistantModuleModelProviderOllama: _backstage_backend_plugin_api.BackendFeature;
+
+export { aiAssistantModuleModelProviderOllama as default };

package/dist/module.cjs.js

@@ -0,0 +1,40 @@
+'use strict';
+
+var backendPluginApi = require('@backstage/backend-plugin-api');
+var ollama = require('@langchain/ollama');
+var backstagePluginAiAssistantNode = require('@sweetoburrito/backstage-plugin-ai-assistant-node');
+
+const aiAssistantModuleModelProviderOllama = backendPluginApi.createBackendModule({
+  pluginId: "ai-assistant",
+  moduleId: "model-provider-ollama",
+  register(reg) {
+    reg.registerInit({
+      deps: {
+        config: backendPluginApi.coreServices.rootConfig,
+        modelProvider: backstagePluginAiAssistantNode.modelProviderExtensionPoint
+      },
+      async init({ config, modelProvider }) {
+        const ollamaConfig = config.getConfig("aiAssistant.models.ollama");
+        const baseUrl = ollamaConfig.getString("baseUrl");
+        const apiKey = ollamaConfig.getString("apiKey");
+        const modelIds = ollamaConfig.getStringArray("models");
+        const models = modelIds.map((modelId) => {
+          return {
+            id: modelId,
+            chatModel: new ollama.ChatOllama({
+              baseUrl,
+              model: modelId,
+              headers: {
+                Authorization: `Bearer ${apiKey}`
+              }
+            })
+          };
+        });
+        models.forEach((model) => modelProvider.register(model));
+      }
+    });
+  }
+});
+
+exports.aiAssistantModuleModelProviderOllama = aiAssistantModuleModelProviderOllama;
+//# 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 { ChatOllama } from '@langchain/ollama';\nimport {\n  Model,\n  modelProviderExtensionPoint,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-node';\n\nexport const aiAssistantModuleModelProviderOllama = createBackendModule({\n  pluginId: 'ai-assistant',\n  moduleId: 'model-provider-ollama',\n  register(reg) {\n    reg.registerInit({\n      deps: {\n        config: coreServices.rootConfig,\n        modelProvider: modelProviderExtensionPoint,\n      },\n      async init({ config, modelProvider }) {\n        const ollamaConfig = config.getConfig('aiAssistant.models.ollama');\n\n        const baseUrl = ollamaConfig.getString('baseUrl');\n        const apiKey = ollamaConfig.getString('apiKey');\n        const modelIds = ollamaConfig.getStringArray('models');\n\n        const models: Model[] = modelIds.map(modelId => {\n          return {\n            id: modelId,\n            chatModel: new ChatOllama({\n              baseUrl,\n              model: modelId,\n              headers: {\n                Authorization: `Bearer ${apiKey}`,\n              },\n            }),\n          };\n        });\n\n        models.forEach(model => modelProvider.register(model));\n      },\n    });\n  },\n});\n"],"names":["createBackendModule","coreServices","modelProviderExtensionPoint","ChatOllama"],"mappings":";;;;;;AAUO,MAAM,uCAAuCA,oCAAA,CAAoB;AAAA,EACtE,QAAA,EAAU,cAAA;AAAA,EACV,QAAA,EAAU,uBAAA;AAAA,EACV,SAAS,GAAA,EAAK;AACZ,IAAA,GAAA,CAAI,YAAA,CAAa;AAAA,MACf,IAAA,EAAM;AAAA,QACJ,QAAQC,6BAAA,CAAa,UAAA;AAAA,QACrB,aAAA,EAAeC;AAAA,OACjB;AAAA,MACA,MAAM,IAAA,CAAK,EAAE,MAAA,EAAQ,eAAc,EAAG;AACpC,QAAA,MAAM,YAAA,GAAe,MAAA,CAAO,SAAA,CAAU,2BAA2B,CAAA;AAEjE,QAAA,MAAM,OAAA,GAAU,YAAA,CAAa,SAAA,CAAU,SAAS,CAAA;AAChD,QAAA,MAAM,MAAA,GAAS,YAAA,CAAa,SAAA,CAAU,QAAQ,CAAA;AAC9C,QAAA,MAAM,QAAA,GAAW,YAAA,CAAa,cAAA,CAAe,QAAQ,CAAA;AAErD,QAAA,MAAM,MAAA,GAAkB,QAAA,CAAS,GAAA,CAAI,CAAA,OAAA,KAAW;AAC9C,UAAA,OAAO;AAAA,YACL,EAAA,EAAI,OAAA;AAAA,YACJ,SAAA,EAAW,IAAIC,iBAAA,CAAW;AAAA,cACxB,OAAA;AAAA,cACA,KAAA,EAAO,OAAA;AAAA,cACP,OAAA,EAAS;AAAA,gBACP,aAAA,EAAe,UAAU,MAAM,CAAA;AAAA;AACjC,aACD;AAAA,WACH;AAAA,QACF,CAAC,CAAA;AAED,QAAA,MAAA,CAAO,OAAA,CAAQ,CAAA,KAAA,KAAS,aAAA,CAAc,QAAA,CAAS,KAAK,CAAC,CAAA;AAAA,MACvD;AAAA,KACD,CAAA;AAAA,EACH;AACF,CAAC;;;;"}

package/package.json

@@ -0,0 +1,49 @@
+{
+  "name": "@sweetoburrito/backstage-plugin-ai-assistant-backend-module-model-provider-ollama",
+  "version": "0.2.1",
+  "license": "Apache-2.0",
+  "description": "The model-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"
+  },
+  "files": [
+    "dist"
+  ],
+  "typesVersions": {
+    "*": {
+      "package.json": [
+        "package.json"
+      ]
+    }
+  }
+}