@sweetoburrito/backstage-plugin-ai-assistant-backend-module-model-provider-google-vertex-ai 0.0.0-snapshot-20251029150521

package/README.md

@@ -0,0 +1,54 @@
+# @sweetoburrito/backstage-plugin-ai-assistant-backend-module-model-provider-google-vertex-ai
+
+The model-provider-google-vertex-ai backend module for the Backstage AI Assistant plugin.
+
+This module lets the AI Assistant backend call Google-hosted (GCP) Vertex AI models through a configuration-driven provider so the rest of the plugin remains model-agnostic.
+
+## Features
+
+- Connects Backstage AI Assistant to GCP-hosted Vertex AI LLM models (Vertex AI Studio).
+- Configuration via Backstage `app-config.yaml` and environment variables.
+
+## When to use
+
+Use this module when you want the AI Assistant backend to use models hosted in GCP Vertex AI (for example: Gemini-family models
+deployed in Google Vertex AI Studio) in the Backstage AI assistant.
+
+## Configuration
+
+Add the provider configuration in your `app-config.local`.
+
+```yaml
+aiAssistant:
+  models:
+    googleVertexAi:
+      apiKey: ${VERTEX-AI-API-KEY}
+      models:
+        - 'gemini-2.5-flash-lite'
+```
+
+## Install
+
+Install the plugin into your backstage backend with the following command
+
+```sh
+yarn workspace backend add @sweetoburrito/backstage-plugin-ai-assistant-backend-module-model-provider-google-vertex-ai
+```
+
+Add it to your backend
+
+```diff
+// packages/backend/src/index.ts
+
+backend.add(import('@backstage/plugin-events-backend'));
+backend.add(import('@backstage/plugin-signals-backend'));
+
+backend.add(import('@sweetoburrito/backstage-plugin-ai-assistant-backend'));
+
+++backend.add(
+++  import(
+++    '@sweetoburrito/backstage-plugin-ai-assistant-backend-module-model-provider-google-vertex-ai'
+++  ),
+++);
+
+```

package/config.d.ts

@@ -0,0 +1,13 @@
+export interface Config {
+  aiAssistant: {
+    models: {
+      googleVertexAi: {
+        /**
+         * @visibility secret
+         */
+        apiKey: string;
+        models: 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.aiAssistantModuleModelProviderGoogleVertexAi;
+//# 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 aiAssistantModuleModelProviderGoogleVertexAi: _backstage_backend_plugin_api.BackendFeature;
+
+export { aiAssistantModuleModelProviderGoogleVertexAi as default };

package/dist/module.cjs.js

@@ -0,0 +1,40 @@
+'use strict';
+
+var backendPluginApi = require('@backstage/backend-plugin-api');
+var backstagePluginAiAssistantNode = require('@sweetoburrito/backstage-plugin-ai-assistant-node');
+var googleVertexai = require('@langchain/google-vertexai');
+
+const aiAssistantModuleModelProviderGoogleVertexAi = backendPluginApi.createBackendModule(
+  {
+    pluginId: "ai-assistant",
+    moduleId: "model-provider-google-vertex-ai",
+    register(reg) {
+      reg.registerInit({
+        deps: {
+          config: backendPluginApi.coreServices.rootConfig,
+          modelProvider: backstagePluginAiAssistantNode.modelProviderExtensionPoint
+        },
+        async init({ config, modelProvider }) {
+          const vertexAiConfig = config.getConfig(
+            "aiAssistant.models.googleVertexAi"
+          );
+          const apiKey = vertexAiConfig.getString("apiKey");
+          const modelIds = vertexAiConfig.getStringArray("models");
+          const models = modelIds.map((modelId) => {
+            return {
+              id: modelId,
+              chatModel: new googleVertexai.ChatVertexAI({
+                model: modelId,
+                apiKey
+              })
+            };
+          });
+          models.forEach((model) => modelProvider.register(model));
+        }
+      });
+    }
+  }
+);
+
+exports.aiAssistantModuleModelProviderGoogleVertexAi = aiAssistantModuleModelProviderGoogleVertexAi;
+//# 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 {\n  Model,\n  modelProviderExtensionPoint,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-node';\nimport { ChatVertexAI } from '@langchain/google-vertexai';\n\nexport const aiAssistantModuleModelProviderGoogleVertexAi = createBackendModule(\n  {\n    pluginId: 'ai-assistant',\n    moduleId: 'model-provider-google-vertex-ai',\n    register(reg) {\n      reg.registerInit({\n        deps: {\n          config: coreServices.rootConfig,\n          modelProvider: modelProviderExtensionPoint,\n        },\n        async init({ config, modelProvider }) {\n          const vertexAiConfig = config.getConfig(\n            'aiAssistant.models.googleVertexAi',\n          );\n\n          const apiKey = vertexAiConfig.getString('apiKey');\n          const modelIds = vertexAiConfig.getStringArray('models');\n\n          const models: Model[] = modelIds.map(modelId => {\n            return {\n              id: modelId,\n              chatModel: new ChatVertexAI({\n                model: modelId,\n                apiKey: apiKey,\n              }),\n            };\n          });\n\n          models.forEach(model => modelProvider.register(model));\n        },\n      });\n    },\n  },\n);\n"],"names":["createBackendModule","coreServices","modelProviderExtensionPoint","ChatVertexAI"],"mappings":";;;;;;AAUO,MAAM,4CAAA,GAA+CA,oCAAA;AAAA,EAC1D;AAAA,IACE,QAAA,EAAU,cAAA;AAAA,IACV,QAAA,EAAU,iCAAA;AAAA,IACV,SAAS,GAAA,EAAK;AACZ,MAAA,GAAA,CAAI,YAAA,CAAa;AAAA,QACf,IAAA,EAAM;AAAA,UACJ,QAAQC,6BAAA,CAAa,UAAA;AAAA,UACrB,aAAA,EAAeC;AAAA,SACjB;AAAA,QACA,MAAM,IAAA,CAAK,EAAE,MAAA,EAAQ,eAAc,EAAG;AACpC,UAAA,MAAM,iBAAiB,MAAA,CAAO,SAAA;AAAA,YAC5B;AAAA,WACF;AAEA,UAAA,MAAM,MAAA,GAAS,cAAA,CAAe,SAAA,CAAU,QAAQ,CAAA;AAChD,UAAA,MAAM,QAAA,GAAW,cAAA,CAAe,cAAA,CAAe,QAAQ,CAAA;AAEvD,UAAA,MAAM,MAAA,GAAkB,QAAA,CAAS,GAAA,CAAI,CAAA,OAAA,KAAW;AAC9C,YAAA,OAAO;AAAA,cACL,EAAA,EAAI,OAAA;AAAA,cACJ,SAAA,EAAW,IAAIC,2BAAA,CAAa;AAAA,gBAC1B,KAAA,EAAO,OAAA;AAAA,gBACP;AAAA,eACD;AAAA,aACH;AAAA,UACF,CAAC,CAAA;AAED,UAAA,MAAA,CAAO,OAAA,CAAQ,CAAA,KAAA,KAAS,aAAA,CAAc,QAAA,CAAS,KAAK,CAAC,CAAA;AAAA,QACvD;AAAA,OACD,CAAA;AAAA,IACH;AAAA;AAEJ;;;;"}

package/package.json

@@ -0,0 +1,51 @@
+{
+  "name": "@sweetoburrito/backstage-plugin-ai-assistant-backend-module-model-provider-google-vertex-ai",
+  "version": "0.0.0-snapshot-20251029150521",
+  "license": "Apache-2.0",
+  "description": "The model-provider-google-vertex-ai 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.3",
+    "@langchain/google-vertexai": "^0.2.18",
+    "@sweetoburrito/backstage-plugin-ai-assistant-node": "0.0.0-snapshot-20251029150521"
+  },
+  "devDependencies": {
+    "@backstage/backend-test-utils": "^1.9.0",
+    "@backstage/cli": "^0.34.3"
+  },
+  "configSchema": "config.d.ts",
+  "files": [
+    "dist",
+    "config.d.ts"
+  ],
+  "typesVersions": {
+    "*": {
+      "package.json": [
+        "package.json"
+      ]
+    }
+  }
+}