@sweetoburrito/backstage-plugin-ai-assistant-node 0.0.0-snapshot-20251113134620 → 0.0.0-snapshot-20251114125112

package/dist/extensions.cjs.js

@@ -14,11 +14,11 @@ const modelProviderExtensionPoint = backendPluginApi.createExtensionPoint({
 const toolExtensionPoint = backendPluginApi.createExtensionPoint({
   id: "ai-assistant.tool"
 });
-const callbackFactoryExtensionPoint = backendPluginApi.createExtensionPoint({
+const callbackProviderExtensionPoint = backendPluginApi.createExtensionPoint({
   id: "ai-assistant.callback"
 });
 
-exports.callbackFactoryExtensionPoint = callbackFactoryExtensionPoint;
+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 { CallbackFactory, 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: CallbackFactory) => void;\n};\n\nexport const callbackFactoryExtensionPoint =\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,gCACXA,qCAAA,CAA6C;AAAA,EAC3C,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

@@ -4,10 +4,11 @@ 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.callbackFactoryExtensionPoint = extensions.callbackFactoryExtensionPoint;
+exports.callbackProviderExtensionPoint = extensions.callbackProviderExtensionPoint;
 exports.dataIngestorExtensionPoint = extensions.dataIngestorExtensionPoint;
 exports.embeddingsProviderExtensionPoint = extensions.embeddingsProviderExtensionPoint;
 exports.modelProviderExtensionPoint = extensions.modelProviderExtensionPoint;
@@ -16,4 +17,5 @@ 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,10 +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 } from '@sweetoburrito/backstage-plugin-ai-assistant-common';
+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;
 /**
@@ -70,16 +71,24 @@ type Tool<T extends ZodType<any, any, any> = ZodType<any, any, any>> = {
     func: (params: z.infer<T>) => Promise<string>;
 };
 
-type CallbackOptions = Record<string, JsonValue> & {
+type ChainCallbackOptions = Record<string, JsonValue> & {
     modelId: string;
     userId: string;
     conversationId: string;
 };
-type Callback = {
-    metadata: Record<string, JsonValue>;
-    callback: BaseCallbackHandler;
+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 CallbackFactory = (options: CallbackOptions) => Promise<Callback>;
 
 type DataIngestorExtensionPoint = {
     registerIngestor: (ingestor: Ingestor) => void;
@@ -105,9 +114,9 @@ type ToolExtensionPoint = {
 };
 declare const toolExtensionPoint: _backstage_backend_plugin_api.ExtensionPoint<ToolExtensionPoint>;
 type CallbackExtensionPoint = {
-    register: (callback: CallbackFactory) => void;
+    register: (callback: CallbackProvider) => void;
 };
-declare const callbackFactoryExtensionPoint: _backstage_backend_plugin_api.ExtensionPoint<CallbackExtensionPoint>;
+declare const callbackProviderExtensionPoint: _backstage_backend_plugin_api.ExtensionPoint<CallbackExtensionPoint>;
 
 type CreateAssistantToolOptions<T extends ZodType<any, any, any>> = {
     tool: Tool<T>;
@@ -119,5 +128,7 @@ declare function streamToString(readableStream: NodeJS.ReadableStream | null): P
 declare const encrypt: (text: string, key: string) => string;
 declare const decrypt: (encryptedText: string, key: string) => string;
 
-export { callbackFactoryExtensionPoint, createAssistantTool, dataIngestorExtensionPoint, decrypt, embeddingsProviderExtensionPoint, encrypt, modelProviderExtensionPoint, streamToString, toolExtensionPoint };
-export type { AugmentationIndexer, Callback, CallbackExtensionPoint, CallbackFactory, CallbackOptions, DataIngestionPipeline, DataIngestionPipelineOptions, DataIngestorExtensionPoint, Embedding, EmbeddingDocument, EmbeddingDocumentMetadata, EmbeddingsProvider, EmbeddingsProviderExtensionPoint, EmbeddingsSource, EntityFilterShape, Ingestor, IngestorOptions, Model, ModelProviderExtensionPoint, Tool, ToolExtensionPoint, VectorStore };
+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/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.0.0-snapshot-20251113134620",
+  "version": "0.0.0-snapshot-20251114125112",
   "license": "Apache-2.0",
   "description": "Node.js library for the ai-assistant plugin",
   "main": "dist/index.cjs.js",
@@ -36,6 +36,8 @@
   ],
   "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"