@sweetoburrito/backstage-plugin-ai-assistant-node 0.5.1 → 0.6.0

package/dist/extensions.cjs.js

@@ -14,7 +14,11 @@ const modelProviderExtensionPoint = backendPluginApi.createExtensionPoint({
 const toolExtensionPoint = backendPluginApi.createExtensionPoint({
   id: "ai-assistant.tool"
 });
+const callbackProviderExtensionPoint = backendPluginApi.createExtensionPoint({
+  id: "ai-assistant.callback"
+});
 
+exports.callbackProviderExtensionPoint = callbackProviderExtensionPoint;
 exports.dataIngestorExtensionPoint = dataIngestorExtensionPoint;
 exports.embeddingsProviderExtensionPoint = embeddingsProviderExtensionPoint;
 exports.modelProviderExtensionPoint = modelProviderExtensionPoint;

package/dist/extensions.cjs.js.map

@@ -1 +1 @@
-{"version":3,"file":"extensions.cjs.js","sources":["../src/extensions.ts"],"sourcesContent":["import { createExtensionPoint } from '@backstage/backend-plugin-api';\nimport { Ingestor } from './types/ingestor';\nimport { Embeddings } from '@langchain/core/embeddings';\nimport { BaseChatModel } from '@langchain/core/language_models/chat_models';\nimport { Tool } from './types';\nimport { ZodType } from 'zod';\n\nexport type DataIngestorExtensionPoint = {\n  registerIngestor: (ingestor: Ingestor) => void;\n};\n\nexport const dataIngestorExtensionPoint =\n  createExtensionPoint<DataIngestorExtensionPoint>({\n    id: 'ai-assistant.data-ingestor',\n  });\n\nexport type EmbeddingsProvider = {\n  getEmbeddings: () => Promise<Omit<Embeddings, 'caller'>>;\n};\n\nexport type EmbeddingsProviderExtensionPoint = {\n  register: (provider: EmbeddingsProvider) => void;\n};\n\nexport const embeddingsProviderExtensionPoint =\n  createExtensionPoint<EmbeddingsProviderExtensionPoint>({\n    id: 'ai-assistant.embeddings-provider',\n  });\n\nexport type Model = {\n  id: string;\n  chatModel: BaseChatModel;\n};\n\nexport type ModelProviderExtensionPoint = {\n  register: (model: Model) => void;\n};\n\nexport const modelProviderExtensionPoint =\n  createExtensionPoint<ModelProviderExtensionPoint>({\n    id: 'ai-assistant.model-provider',\n  });\n\nexport type ToolExtensionPoint = {\n  register: (tool: Tool<ZodType>) => void;\n};\n\nexport const toolExtensionPoint = createExtensionPoint<ToolExtensionPoint>({\n  id: 'ai-assistant.tool',\n});\n"],"names":["createExtensionPoint"],"mappings":";;;;AAWO,MAAM,6BACXA,qCAAA,CAAiD;AAAA,EAC/C,EAAA,EAAI;AACN,CAAC;AAUI,MAAM,mCACXA,qCAAA,CAAuD;AAAA,EACrD,EAAA,EAAI;AACN,CAAC;AAWI,MAAM,8BACXA,qCAAA,CAAkD;AAAA,EAChD,EAAA,EAAI;AACN,CAAC;AAMI,MAAM,qBAAqBA,qCAAA,CAAyC;AAAA,EACzE,EAAA,EAAI;AACN,CAAC;;;;;;;"}
+{"version":3,"file":"extensions.cjs.js","sources":["../src/extensions.ts"],"sourcesContent":["import { createExtensionPoint } from '@backstage/backend-plugin-api';\nimport { Ingestor } from './types/ingestor';\nimport { Embeddings } from '@langchain/core/embeddings';\nimport { BaseChatModel } from '@langchain/core/language_models/chat_models';\nimport { CallbackProvider, Tool } from './types';\nimport { ZodType } from 'zod';\n\nexport type DataIngestorExtensionPoint = {\n  registerIngestor: (ingestor: Ingestor) => void;\n};\n\nexport const dataIngestorExtensionPoint =\n  createExtensionPoint<DataIngestorExtensionPoint>({\n    id: 'ai-assistant.data-ingestor',\n  });\n\nexport type EmbeddingsProvider = {\n  getEmbeddings: () => Promise<Omit<Embeddings, 'caller'>>;\n};\n\nexport type EmbeddingsProviderExtensionPoint = {\n  register: (provider: EmbeddingsProvider) => void;\n};\n\nexport const embeddingsProviderExtensionPoint =\n  createExtensionPoint<EmbeddingsProviderExtensionPoint>({\n    id: 'ai-assistant.embeddings-provider',\n  });\n\nexport type Model = {\n  id: string;\n  chatModel: BaseChatModel;\n};\n\nexport type ModelProviderExtensionPoint = {\n  register: (model: Model) => void;\n};\n\nexport const modelProviderExtensionPoint =\n  createExtensionPoint<ModelProviderExtensionPoint>({\n    id: 'ai-assistant.model-provider',\n  });\n\nexport type ToolExtensionPoint = {\n  register: (tool: Tool<ZodType>) => void;\n};\n\nexport const toolExtensionPoint = createExtensionPoint<ToolExtensionPoint>({\n  id: 'ai-assistant.tool',\n});\n\nexport type CallbackExtensionPoint = {\n  register: (callback: CallbackProvider) => void;\n};\n\nexport const callbackProviderExtensionPoint =\n  createExtensionPoint<CallbackExtensionPoint>({\n    id: 'ai-assistant.callback',\n  });\n"],"names":["createExtensionPoint"],"mappings":";;;;AAWO,MAAM,6BACXA,qCAAA,CAAiD;AAAA,EAC/C,EAAA,EAAI;AACN,CAAC;AAUI,MAAM,mCACXA,qCAAA,CAAuD;AAAA,EACrD,EAAA,EAAI;AACN,CAAC;AAWI,MAAM,8BACXA,qCAAA,CAAkD;AAAA,EAChD,EAAA,EAAI;AACN,CAAC;AAMI,MAAM,qBAAqBA,qCAAA,CAAyC;AAAA,EACzE,EAAA,EAAI;AACN,CAAC;AAMM,MAAM,iCACXA,qCAAA,CAA6C;AAAA,EAC3C,EAAA,EAAI;AACN,CAAC;;;;;;;;"}

package/dist/index.cjs.js

@@ -3,13 +3,19 @@
 var extensions = require('./extensions.cjs.js');
 var tool = require('./utils/tool.cjs.js');
 var streamToString = require('./utils/stream-to-string.cjs.js');
+var encrypt = require('./utils/encrypt.cjs.js');
+var getUser = require('./utils/get-user.cjs.js');
 
 
 
+exports.callbackProviderExtensionPoint = extensions.callbackProviderExtensionPoint;
 exports.dataIngestorExtensionPoint = extensions.dataIngestorExtensionPoint;
 exports.embeddingsProviderExtensionPoint = extensions.embeddingsProviderExtensionPoint;
 exports.modelProviderExtensionPoint = extensions.modelProviderExtensionPoint;
 exports.toolExtensionPoint = extensions.toolExtensionPoint;
 exports.createAssistantTool = tool.createAssistantTool;
 exports.streamToString = streamToString.streamToString;
+exports.decrypt = encrypt.decrypt;
+exports.encrypt = encrypt.encrypt;
+exports.getUser = getUser.getUser;
 //# sourceMappingURL=index.cjs.js.map

package/dist/index.cjs.js.map

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

package/dist/index.d.ts

@@ -1,8 +1,11 @@
 import * as _backstage_backend_plugin_api from '@backstage/backend-plugin-api';
-import { LoggerService, RootConfigService, SchedulerService } from '@backstage/backend-plugin-api';
+import { LoggerService, RootConfigService, SchedulerService, CacheService, BackstageCredentials } from '@backstage/backend-plugin-api';
 import { Embeddings } from '@langchain/core/embeddings';
 import { ZodType, z } from 'zod';
+import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
+import { JsonValue, Message } from '@sweetoburrito/backstage-plugin-ai-assistant-common';
 import { BaseChatModel } from '@langchain/core/language_models/chat_models';
+import { CatalogService } from '@backstage/plugin-catalog-node';
 
 type EmbeddingsSource = string;
 /**
@@ -68,6 +71,25 @@ type Tool<T extends ZodType<any, any, any> = ZodType<any, any, any>> = {
     func: (params: z.infer<T>) => Promise<string>;
 };
 
+type ChainCallbackOptions = Record<string, JsonValue> & {
+    modelId: string;
+    userId: string;
+    conversationId: string;
+};
+type ChainCallback = (options: ChainCallbackOptions) => BaseCallbackHandler;
+type ScoreCallback = (options: {
+    name: string;
+    message: Message;
+}) => Promise<void>;
+type ChainMetadata = Record<string, JsonValue>;
+type ChainMetadataCallback = (options: ChainCallbackOptions) => Promise<ChainMetadata>;
+type CallbackProvider = {
+    id: string;
+    chainCallback?: ChainCallback;
+    scoreCallback?: ScoreCallback;
+    chainMetadataCallback?: ChainMetadataCallback;
+};
+
 type DataIngestorExtensionPoint = {
     registerIngestor: (ingestor: Ingestor) => void;
 };
@@ -91,6 +113,10 @@ type ToolExtensionPoint = {
     register: (tool: Tool<ZodType>) => void;
 };
 declare const toolExtensionPoint: _backstage_backend_plugin_api.ExtensionPoint<ToolExtensionPoint>;
+type CallbackExtensionPoint = {
+    register: (callback: CallbackProvider) => void;
+};
+declare const callbackProviderExtensionPoint: _backstage_backend_plugin_api.ExtensionPoint<CallbackExtensionPoint>;
 
 type CreateAssistantToolOptions<T extends ZodType<any, any, any>> = {
     tool: Tool<T>;
@@ -99,5 +125,10 @@ declare const createAssistantTool: <T extends ZodType<any, any, any>>(options: C
 
 declare function streamToString(readableStream: NodeJS.ReadableStream | null): Promise<string>;
 
-export { createAssistantTool, dataIngestorExtensionPoint, embeddingsProviderExtensionPoint, modelProviderExtensionPoint, streamToString, toolExtensionPoint };
-export type { AugmentationIndexer, DataIngestionPipeline, DataIngestionPipelineOptions, DataIngestorExtensionPoint, Embedding, EmbeddingDocument, EmbeddingDocumentMetadata, EmbeddingsProvider, EmbeddingsProviderExtensionPoint, EmbeddingsSource, EntityFilterShape, Ingestor, IngestorOptions, Model, ModelProviderExtensionPoint, Tool, ToolExtensionPoint, VectorStore };
+declare const encrypt: (text: string, key: string) => string;
+declare const decrypt: (encryptedText: string, key: string) => string;
+
+declare const getUser: (cache: CacheService, userEntityRef: string, credentials: BackstageCredentials, catalog: CatalogService) => Promise<any>;
+
+export { callbackProviderExtensionPoint, createAssistantTool, dataIngestorExtensionPoint, decrypt, embeddingsProviderExtensionPoint, encrypt, getUser, modelProviderExtensionPoint, streamToString, toolExtensionPoint };
+export type { AugmentationIndexer, CallbackExtensionPoint, CallbackProvider, ChainCallback, ChainCallbackOptions, ChainMetadata, ChainMetadataCallback, DataIngestionPipeline, DataIngestionPipelineOptions, DataIngestorExtensionPoint, Embedding, EmbeddingDocument, EmbeddingDocumentMetadata, EmbeddingsProvider, EmbeddingsProviderExtensionPoint, EmbeddingsSource, EntityFilterShape, Ingestor, IngestorOptions, Model, ModelProviderExtensionPoint, ScoreCallback, Tool, ToolExtensionPoint, VectorStore };

package/dist/utils/encrypt.cjs.js

@@ -0,0 +1,32 @@
+'use strict';
+
+var crypto = require('crypto');
+
+const normalizeKey = (key) => {
+  return crypto.createHash("sha256").update(key).digest();
+};
+const encrypt = (text, key) => {
+  const iv = crypto.randomBytes(16);
+  const normalizedKey = normalizeKey(key);
+  const cipher = crypto.createCipheriv("aes-256-gcm", normalizedKey, iv);
+  let encrypted = cipher.update(text, "utf8", "hex");
+  encrypted += cipher.final("hex");
+  const authTag = cipher.getAuthTag();
+  return `${iv.toString("hex")}:${encrypted}:${authTag.toString("hex")}`;
+};
+const decrypt = (encryptedText, key) => {
+  const parts = encryptedText.split(":");
+  const iv = Buffer.from(parts[0], "hex");
+  const encrypted = parts[1];
+  const authTag = Buffer.from(parts[2], "hex");
+  const normalizedKey = normalizeKey(key);
+  const decipher = crypto.createDecipheriv("aes-256-gcm", normalizedKey, iv);
+  decipher.setAuthTag(authTag);
+  let decrypted = decipher.update(encrypted, "hex", "utf8");
+  decrypted += decipher.final("utf8");
+  return decrypted;
+};
+
+exports.decrypt = decrypt;
+exports.encrypt = encrypt;
+//# sourceMappingURL=encrypt.cjs.js.map

package/dist/utils/encrypt.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"encrypt.cjs.js","sources":["../../src/utils/encrypt.ts"],"sourcesContent":["import {\n  createCipheriv,\n  createDecipheriv,\n  randomBytes,\n  createHash,\n} from 'crypto';\n\nconst normalizeKey = (key: string): Buffer => {\n  // Hash the key to ensure it's always 32 bytes for AES-256\n  return createHash('sha256').update(key).digest();\n};\n\nexport const encrypt = (text: string, key: string): string => {\n  const iv = randomBytes(16);\n  const normalizedKey = normalizeKey(key);\n  const cipher = createCipheriv('aes-256-gcm', normalizedKey, iv);\n  let encrypted = cipher.update(text, 'utf8', 'hex');\n  encrypted += cipher.final('hex');\n  const authTag = cipher.getAuthTag();\n  return `${iv.toString('hex')}:${encrypted}:${authTag.toString('hex')}`;\n};\n\nexport const decrypt = (encryptedText: string, key: string): string => {\n  const parts = encryptedText.split(':');\n  const iv = Buffer.from(parts[0], 'hex');\n  const encrypted = parts[1];\n  const authTag = Buffer.from(parts[2], 'hex');\n  const normalizedKey = normalizeKey(key);\n  const decipher = createDecipheriv('aes-256-gcm', normalizedKey, iv);\n  decipher.setAuthTag(authTag);\n  let decrypted = decipher.update(encrypted, 'hex', 'utf8');\n  decrypted += decipher.final('utf8');\n  return decrypted;\n};\n"],"names":["createHash","randomBytes","createCipheriv","createDecipheriv"],"mappings":";;;;AAOA,MAAM,YAAA,GAAe,CAAC,GAAA,KAAwB;AAE5C,EAAA,OAAOA,kBAAW,QAAQ,CAAA,CAAE,MAAA,CAAO,GAAG,EAAE,MAAA,EAAO;AACjD,CAAA;AAEO,MAAM,OAAA,GAAU,CAAC,IAAA,EAAc,GAAA,KAAwB;AAC5D,EAAA,MAAM,EAAA,GAAKC,mBAAY,EAAE,CAAA;AACzB,EAAA,MAAM,aAAA,GAAgB,aAAa,GAAG,CAAA;AACtC,EAAA,MAAM,MAAA,GAASC,qBAAA,CAAe,aAAA,EAAe,aAAA,EAAe,EAAE,CAAA;AAC9D,EAAA,IAAI,SAAA,GAAY,MAAA,CAAO,MAAA,CAAO,IAAA,EAAM,QAAQ,KAAK,CAAA;AACjD,EAAA,SAAA,IAAa,MAAA,CAAO,MAAM,KAAK,CAAA;AAC/B,EAAA,MAAM,OAAA,GAAU,OAAO,UAAA,EAAW;AAClC,EAAA,OAAO,CAAA,EAAG,EAAA,CAAG,QAAA,CAAS,KAAK,CAAC,CAAA,CAAA,EAAI,SAAS,CAAA,CAAA,EAAI,OAAA,CAAQ,QAAA,CAAS,KAAK,CAAC,CAAA,CAAA;AACtE;AAEO,MAAM,OAAA,GAAU,CAAC,aAAA,EAAuB,GAAA,KAAwB;AACrE,EAAA,MAAM,KAAA,GAAQ,aAAA,CAAc,KAAA,CAAM,GAAG,CAAA;AACrC,EAAA,MAAM,KAAK,MAAA,CAAO,IAAA,CAAK,KAAA,CAAM,CAAC,GAAG,KAAK,CAAA;AACtC,EAAA,MAAM,SAAA,GAAY,MAAM,CAAC,CAAA;AACzB,EAAA,MAAM,UAAU,MAAA,CAAO,IAAA,CAAK,KAAA,CAAM,CAAC,GAAG,KAAK,CAAA;AAC3C,EAAA,MAAM,aAAA,GAAgB,aAAa,GAAG,CAAA;AACtC,EAAA,MAAM,QAAA,GAAWC,uBAAA,CAAiB,aAAA,EAAe,aAAA,EAAe,EAAE,CAAA;AAClE,EAAA,QAAA,CAAS,WAAW,OAAO,CAAA;AAC3B,EAAA,IAAI,SAAA,GAAY,QAAA,CAAS,MAAA,CAAO,SAAA,EAAW,OAAO,MAAM,CAAA;AACxD,EAAA,SAAA,IAAa,QAAA,CAAS,MAAM,MAAM,CAAA;AAClC,EAAA,OAAO,SAAA;AACT;;;;;"}

package/dist/utils/get-user.cjs.js

@@ -0,0 +1,16 @@
+'use strict';
+
+const getUser = async (cache, userEntityRef, credentials, catalog) => {
+  const cached = await cache.get(userEntityRef);
+  if (cached) {
+    return JSON.parse(String(cached));
+  }
+  const user = await catalog.getEntityByRef(userEntityRef, {
+    credentials
+  });
+  await cache.set(userEntityRef, JSON.stringify(user));
+  return user;
+};
+
+exports.getUser = getUser;
+//# sourceMappingURL=get-user.cjs.js.map

package/dist/utils/get-user.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"get-user.cjs.js","sources":["../../src/utils/get-user.ts"],"sourcesContent":["import {\n  BackstageCredentials,\n  CacheService,\n} from '@backstage/backend-plugin-api';\nimport { UserEntity } from '@backstage/catalog-model';\nimport { CatalogService } from '@backstage/plugin-catalog-node';\n\nexport const getUser = async (\n  cache: CacheService,\n  userEntityRef: string,\n  credentials: BackstageCredentials,\n  catalog: CatalogService,\n) => {\n  const cached = await cache.get(userEntityRef);\n\n  if (cached) {\n    return JSON.parse(String(cached));\n  }\n\n  const user = (await catalog.getEntityByRef(userEntityRef, {\n    credentials,\n  })) as UserEntity | undefined;\n  await cache.set(userEntityRef, JSON.stringify(user));\n\n  return user;\n};\n"],"names":[],"mappings":";;AAOO,MAAM,OAAA,GAAU,OACrB,KAAA,EACA,aAAA,EACA,aACA,OAAA,KACG;AACH,EAAA,MAAM,MAAA,GAAS,MAAM,KAAA,CAAM,GAAA,CAAI,aAAa,CAAA;AAE5C,EAAA,IAAI,MAAA,EAAQ;AACV,IAAA,OAAO,IAAA,CAAK,KAAA,CAAM,MAAA,CAAO,MAAM,CAAC,CAAA;AAAA,EAClC;AAEA,EAAA,MAAM,IAAA,GAAQ,MAAM,OAAA,CAAQ,cAAA,CAAe,aAAA,EAAe;AAAA,IACxD;AAAA,GACD,CAAA;AACD,EAAA,MAAM,MAAM,GAAA,CAAI,aAAA,EAAe,IAAA,CAAK,SAAA,CAAU,IAAI,CAAC,CAAA;AAEnD,EAAA,OAAO,IAAA;AACT;;;;"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sweetoburrito/backstage-plugin-ai-assistant-node",
-  "version": "0.5.1",
+  "version": "0.6.0",
   "license": "Apache-2.0",
   "description": "Node.js library for the ai-assistant plugin",
   "main": "dist/index.cjs.js",
@@ -36,7 +36,10 @@
   ],
   "dependencies": {
     "@backstage/backend-plugin-api": "backstage:^",
+    "@backstage/catalog-model": "backstage:^",
+    "@backstage/plugin-catalog-node": "backstage:^",
     "@langchain/core": "^0.3.72",
+    "@sweetoburrito/backstage-plugin-ai-assistant-common": "workspace:^",
     "zod": "^4.1.11"
   },
   "typesVersions": {