@sweetoburrito/backstage-plugin-ai-assistant-backend 0.13.0 → 0.14.0

package/dist/plugin.cjs.js

@@ -1,26 +1,28 @@
 'use strict';
 
 var backendPluginApi = require('@backstage/backend-plugin-api');
-var index$1 = require('./services/router/index.cjs.js');
+var index$2 = require('./services/router/index.cjs.js');
 var backstagePluginAiAssistantNode = require('@sweetoburrito/backstage-plugin-ai-assistant-node');
 var ingestor = require('./services/ingestor.cjs.js');
-var chat = require('./services/chat.cjs.js');
 var migrations = require('./database/migrations.cjs.js');
 var pgVectorStore = require('./database/pg-vector-store.cjs.js');
-var pluginSignalsNode = require('@backstage/plugin-signals-node');
 var searchKnowledge = require('./tools/searchKnowledge.cjs.js');
-var pluginCatalogNode = require('@backstage/plugin-catalog-node');
-var index = require('./services/mcp/index.cjs.js');
+var index$1 = require('./services/mcp/index.cjs.js');
+var chat = require('./services/chat.cjs.js');
 var callbacks = require('./services/callbacks.cjs.js');
 var summarizer = require('./services/summarizer.cjs.js');
 var userSettings = require('./services/user-settings.cjs.js');
+var model = require('./services/model.cjs.js');
+var tools = require('./services/tools.cjs.js');
+var conversation = require('./services/conversation.cjs.js');
+var index = require('./services/agent/index.cjs.js');
 
 const aiAssistantPlugin = backendPluginApi.createBackendPlugin({
   pluginId: "ai-assistant",
   register(env) {
     const ingestors = [];
     const models = [];
-    const tools = [];
+    const tools$1 = [];
     const callbacks$1 = [];
     let embeddingsProvider;
     env.registerExtensionPoint(backstagePluginAiAssistantNode.dataIngestorExtensionPoint, {
@@ -50,11 +52,11 @@ const aiAssistantPlugin = backendPluginApi.createBackendPlugin({
     });
     env.registerExtensionPoint(backstagePluginAiAssistantNode.toolExtensionPoint, {
       register: (tool) => {
-        const existingTool = tools.find((t) => t.name === tool.name);
+        const existingTool = tools$1.find((t) => t.name === tool.name);
         if (existingTool) {
           throw new Error(`Tool with name ${tool.name} is already registered.`);
         }
-        tools.push(tool);
+        tools$1.push(tool);
       }
     });
     env.registerExtensionPoint(backstagePluginAiAssistantNode.callbackProviderExtensionPoint, {
@@ -71,13 +73,18 @@ const aiAssistantPlugin = backendPluginApi.createBackendPlugin({
         scheduler: backendPluginApi.coreServices.scheduler,
         httpAuth: backendPluginApi.coreServices.httpAuth,
         userInfo: backendPluginApi.coreServices.userInfo,
-        signals: pluginSignalsNode.signalsServiceRef,
-        catalog: pluginCatalogNode.catalogServiceRef,
-        cache: backendPluginApi.coreServices.cache,
-        auth: backendPluginApi.coreServices.auth
+        callback: callbacks.callbackServiceRef,
+        summarizer: summarizer.summarizerServiceRef,
+        model: model.modelServiceRef,
+        mcp: index$1.mcpServiceRef,
+        tool: tools.toolsServiceRef,
+        chat: chat.chatServiceRef,
+        userSettings: userSettings.userSettingsServiceRef,
+        conversation: conversation.conversationServiceRef,
+        agent: index.agentServiceRef
       },
       async init(options) {
-        const { httpRouter, database, config } = options;
+        const { httpRouter, database, callback, model, tool } = options;
         const client = await database.getClient();
         await migrations.applyDatabaseMigrations(client);
         const vectorStore = await pgVectorStore.PgVectorStore.fromConfig(options);
@@ -90,35 +97,12 @@ const aiAssistantPlugin = backendPluginApi.createBackendPlugin({
           vectorStore,
           ingestors
         });
-        const mcp = await index.createMcpService(options);
         const searchKnowledgeTool = searchKnowledge.createSearchKnowledgeTool({ vectorStore });
-        tools.push(searchKnowledgeTool);
-        const callback = await callbacks.createCallbackService({
-          callbacks: callbacks$1
-        });
-        const summarizer$1 = await summarizer.createSummarizerService({
-          config,
-          models,
-          callback
-        });
-        const chat$1 = await chat.createChatService({
-          ...options,
-          models,
-          tools,
-          mcp,
-          callback,
-          summarizer: summarizer$1
-        });
-        const userSettings$1 = await userSettings.createUserSettingsService(options);
-        httpRouter.use(
-          await index$1.createRouter({
-            ...options,
-            chat: chat$1,
-            mcp,
-            summarizer: summarizer$1,
-            userSettings: userSettings$1
-          })
-        );
+        tool.registerTools(tools$1);
+        tool.registerCoreTools([searchKnowledgeTool]);
+        callback.registerCallbacks(callbacks$1);
+        model.registerModels(models);
+        httpRouter.use(await index$2.createRouter(options));
         dataIngestionPipeline.start();
       }
     });

package/dist/plugin.cjs.js.map

@@ -1 +1 @@
-{"version":3,"file":"plugin.cjs.js","sources":["../src/plugin.ts"],"sourcesContent":["import {\n  coreServices,\n  createBackendPlugin,\n} from '@backstage/backend-plugin-api';\nimport { createRouter } from './services/router';\nimport {\n  dataIngestorExtensionPoint,\n  EmbeddingsProvider,\n  embeddingsProviderExtensionPoint,\n  Ingestor,\n  Model,\n  modelProviderExtensionPoint,\n  toolExtensionPoint,\n  callbackProviderExtensionPoint,\n  CallbackProvider,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-node';\nimport { Tool } from '@sweetoburrito/backstage-plugin-ai-assistant-common';\nimport { createDataIngestionPipeline } from './services/ingestor';\nimport { createChatService } from './services/chat';\nimport { applyDatabaseMigrations } from './database/migrations';\nimport { PgVectorStore } from './database';\nimport { signalsServiceRef } from '@backstage/plugin-signals-node';\nimport { createSearchKnowledgeTool } from './tools/searchKnowledge';\nimport { catalogServiceRef } from '@backstage/plugin-catalog-node';\nimport { createMcpService } from './services/mcp';\nimport { createCallbackService } from './services/callbacks';\nimport { createSummarizerService } from './services/summarizer';\nimport { createUserSettingsService } from './services/user-settings';\n/**\n * aiAssistantPlugin backend plugin\n *\n * @public\n */\n\nexport const aiAssistantPlugin = createBackendPlugin({\n  pluginId: 'ai-assistant',\n  register(env) {\n    const ingestors: Ingestor[] = [];\n    const models: Model[] = [];\n    const tools: Tool[] = [];\n    const callbacks: CallbackProvider[] = [];\n\n    let embeddingsProvider: EmbeddingsProvider;\n\n    env.registerExtensionPoint(dataIngestorExtensionPoint, {\n      registerIngestor: ingestor => {\n        const existingIngestor = ingestors.find(i => i.id === ingestor.id);\n        if (existingIngestor) {\n          throw new Error(\n            `Ingestor with id ${ingestor.id} is already registered.`,\n          );\n        }\n        ingestors.push(ingestor);\n      },\n    });\n\n    env.registerExtensionPoint(embeddingsProviderExtensionPoint, {\n      register: provider => {\n        embeddingsProvider = provider;\n      },\n    });\n\n    env.registerExtensionPoint(modelProviderExtensionPoint, {\n      register: model => {\n        const existingModel = models.find(m => m.id === model.id);\n        if (existingModel) {\n          throw new Error(`Model with id ${model.id} is already registered.`);\n        }\n        models.push(model);\n      },\n    });\n\n    env.registerExtensionPoint(toolExtensionPoint, {\n      register: tool => {\n        const existingTool = tools.find(t => t.name === tool.name);\n        if (existingTool) {\n          throw new Error(`Tool with name ${tool.name} is already registered.`);\n        }\n        tools.push(tool);\n      },\n    });\n\n    env.registerExtensionPoint(callbackProviderExtensionPoint, {\n      register: callbackProvider => {\n        callbacks.push(callbackProvider);\n      },\n    });\n\n    env.registerInit({\n      deps: {\n        httpRouter: coreServices.httpRouter,\n        database: coreServices.database,\n        logger: coreServices.logger,\n        config: coreServices.rootConfig,\n        scheduler: coreServices.scheduler,\n        httpAuth: coreServices.httpAuth,\n        userInfo: coreServices.userInfo,\n        signals: signalsServiceRef,\n        catalog: catalogServiceRef,\n        cache: coreServices.cache,\n        auth: coreServices.auth,\n      },\n\n      async init(options) {\n        const { httpRouter, database, config } = options;\n\n        const client = await database.getClient();\n\n        await applyDatabaseMigrations(client);\n\n        const vectorStore = await PgVectorStore.fromConfig(options);\n\n        if (!embeddingsProvider) {\n          throw new Error('No Embeddings Provider was registered.');\n        }\n\n        vectorStore.connectEmbeddings(await embeddingsProvider.getEmbeddings());\n\n        const dataIngestionPipeline = createDataIngestionPipeline({\n          ...options,\n          vectorStore,\n          ingestors,\n        });\n\n        const mcp = await createMcpService(options);\n\n        const searchKnowledgeTool = createSearchKnowledgeTool({ vectorStore });\n        tools.push(searchKnowledgeTool);\n\n        const callback = await createCallbackService({\n          callbacks,\n        });\n\n        const summarizer = await createSummarizerService({\n          config,\n          models,\n          callback,\n        });\n\n        const chat = await createChatService({\n          ...options,\n          models,\n          tools,\n          mcp,\n          callback,\n          summarizer,\n        });\n\n        const userSettings = await createUserSettingsService(options);\n\n        httpRouter.use(\n          await createRouter({\n            ...options,\n            chat,\n            mcp,\n            summarizer,\n            userSettings,\n          }),\n        );\n        dataIngestionPipeline.start();\n      },\n    });\n  },\n});\n"],"names":["createBackendPlugin","callbacks","dataIngestorExtensionPoint","embeddingsProviderExtensionPoint","modelProviderExtensionPoint","toolExtensionPoint","callbackProviderExtensionPoint","coreServices","signalsServiceRef","catalogServiceRef","applyDatabaseMigrations","PgVectorStore","createDataIngestionPipeline","createMcpService","createSearchKnowledgeTool","createCallbackService","summarizer","createSummarizerService","chat","createChatService","userSettings","createUserSettingsService","createRouter"],"mappings":";;;;;;;;;;;;;;;;;AAkCO,MAAM,oBAAoBA,oCAAA,CAAoB;AAAA,EACnD,QAAA,EAAU,cAAA;AAAA,EACV,SAAS,GAAA,EAAK;AACZ,IAAA,MAAM,YAAwB,EAAC;AAC/B,IAAA,MAAM,SAAkB,EAAC;AACzB,IAAA,MAAM,QAAgB,EAAC;AACvB,IAAA,MAAMC,cAAgC,EAAC;AAEvC,IAAA,IAAI,kBAAA;AAEJ,IAAA,GAAA,CAAI,uBAAuBC,yDAAA,EAA4B;AAAA,MACrD,kBAAkB,CAAA,QAAA,KAAY;AAC5B,QAAA,MAAM,mBAAmB,SAAA,CAAU,IAAA,CAAK,OAAK,CAAA,CAAE,EAAA,KAAO,SAAS,EAAE,CAAA;AACjE,QAAA,IAAI,gBAAA,EAAkB;AACpB,UAAA,MAAM,IAAI,KAAA;AAAA,YACR,CAAA,iBAAA,EAAoB,SAAS,EAAE,CAAA,uBAAA;AAAA,WACjC;AAAA,QACF;AACA,QAAA,SAAA,CAAU,KAAK,QAAQ,CAAA;AAAA,MACzB;AAAA,KACD,CAAA;AAED,IAAA,GAAA,CAAI,uBAAuBC,+DAAA,EAAkC;AAAA,MAC3D,UAAU,CAAA,QAAA,KAAY;AACpB,QAAA,kBAAA,GAAqB,QAAA;AAAA,MACvB;AAAA,KACD,CAAA;AAED,IAAA,GAAA,CAAI,uBAAuBC,0DAAA,EAA6B;AAAA,MACtD,UAAU,CAAA,KAAA,KAAS;AACjB,QAAA,MAAM,gBAAgB,MAAA,CAAO,IAAA,CAAK,OAAK,CAAA,CAAE,EAAA,KAAO,MAAM,EAAE,CAAA;AACxD,QAAA,IAAI,aAAA,EAAe;AACjB,UAAA,MAAM,IAAI,KAAA,CAAM,CAAA,cAAA,EAAiB,KAAA,CAAM,EAAE,CAAA,uBAAA,CAAyB,CAAA;AAAA,QACpE;AACA,QAAA,MAAA,CAAO,KAAK,KAAK,CAAA;AAAA,MACnB;AAAA,KACD,CAAA;AAED,IAAA,GAAA,CAAI,uBAAuBC,iDAAA,EAAoB;AAAA,MAC7C,UAAU,CAAA,IAAA,KAAQ;AAChB,QAAA,MAAM,eAAe,KAAA,CAAM,IAAA,CAAK,OAAK,CAAA,CAAE,IAAA,KAAS,KAAK,IAAI,CAAA;AACzD,QAAA,IAAI,YAAA,EAAc;AAChB,UAAA,MAAM,IAAI,KAAA,CAAM,CAAA,eAAA,EAAkB,IAAA,CAAK,IAAI,CAAA,uBAAA,CAAyB,CAAA;AAAA,QACtE;AACA,QAAA,KAAA,CAAM,KAAK,IAAI,CAAA;AAAA,MACjB;AAAA,KACD,CAAA;AAED,IAAA,GAAA,CAAI,uBAAuBC,6DAAA,EAAgC;AAAA,MACzD,UAAU,CAAA,gBAAA,KAAoB;AAC5B,QAAAL,WAAA,CAAU,KAAK,gBAAgB,CAAA;AAAA,MACjC;AAAA,KACD,CAAA;AAED,IAAA,GAAA,CAAI,YAAA,CAAa;AAAA,MACf,IAAA,EAAM;AAAA,QACJ,YAAYM,6BAAA,CAAa,UAAA;AAAA,QACzB,UAAUA,6BAAA,CAAa,QAAA;AAAA,QACvB,QAAQA,6BAAA,CAAa,MAAA;AAAA,QACrB,QAAQA,6BAAA,CAAa,UAAA;AAAA,QACrB,WAAWA,6BAAA,CAAa,SAAA;AAAA,QACxB,UAAUA,6BAAA,CAAa,QAAA;AAAA,QACvB,UAAUA,6BAAA,CAAa,QAAA;AAAA,QACvB,OAAA,EAASC,mCAAA;AAAA,QACT,OAAA,EAASC,mCAAA;AAAA,QACT,OAAOF,6BAAA,CAAa,KAAA;AAAA,QACpB,MAAMA,6BAAA,CAAa;AAAA,OACrB;AAAA,MAEA,MAAM,KAAK,OAAA,EAAS;AAClB,QAAA,MAAM,EAAE,UAAA,EAAY,QAAA,EAAU,MAAA,EAAO,GAAI,OAAA;AAEzC,QAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,SAAA,EAAU;AAExC,QAAA,MAAMG,mCAAwB,MAAM,CAAA;AAEpC,QAAA,MAAM,WAAA,GAAc,MAAMC,2BAAA,CAAc,UAAA,CAAW,OAAO,CAAA;AAE1D,QAAA,IAAI,CAAC,kBAAA,EAAoB;AACvB,UAAA,MAAM,IAAI,MAAM,wCAAwC,CAAA;AAAA,QAC1D;AAEA,QAAA,WAAA,CAAY,iBAAA,CAAkB,MAAM,kBAAA,CAAmB,aAAA,EAAe,CAAA;AAEtE,QAAA,MAAM,wBAAwBC,oCAAA,CAA4B;AAAA,UACxD,GAAG,OAAA;AAAA,UACH,WAAA;AAAA,UACA;AAAA,SACD,CAAA;AAED,QAAA,MAAM,GAAA,GAAM,MAAMC,sBAAA,CAAiB,OAAO,CAAA;AAE1C,QAAA,MAAM,mBAAA,GAAsBC,yCAAA,CAA0B,EAAE,WAAA,EAAa,CAAA;AACrE,QAAA,KAAA,CAAM,KAAK,mBAAmB,CAAA;AAE9B,QAAA,MAAM,QAAA,GAAW,MAAMC,+BAAA,CAAsB;AAAA,qBAC3Cd;AAAA,SACD,CAAA;AAED,QAAA,MAAMe,YAAA,GAAa,MAAMC,kCAAA,CAAwB;AAAA,UAC/C,MAAA;AAAA,UACA,MAAA;AAAA,UACA;AAAA,SACD,CAAA;AAED,QAAA,MAAMC,MAAA,GAAO,MAAMC,sBAAA,CAAkB;AAAA,UACnC,GAAG,OAAA;AAAA,UACH,MAAA;AAAA,UACA,KAAA;AAAA,UACA,GAAA;AAAA,UACA,QAAA;AAAA,sBACAH;AAAA,SACD,CAAA;AAED,QAAA,MAAMI,cAAA,GAAe,MAAMC,sCAAA,CAA0B,OAAO,CAAA;AAE5D,QAAA,UAAA,CAAW,GAAA;AAAA,UACT,MAAMC,oBAAA,CAAa;AAAA,YACjB,GAAG,OAAA;AAAA,kBACHJ,MAAA;AAAA,YACA,GAAA;AAAA,wBACAF,YAAA;AAAA,0BACAI;AAAA,WACD;AAAA,SACH;AACA,QAAA,qBAAA,CAAsB,KAAA,EAAM;AAAA,MAC9B;AAAA,KACD,CAAA;AAAA,EACH;AACF,CAAC;;;;"}
+{"version":3,"file":"plugin.cjs.js","sources":["../src/plugin.ts"],"sourcesContent":["import {\n  coreServices,\n  createBackendPlugin,\n} from '@backstage/backend-plugin-api';\nimport { createRouter } from './services/router';\nimport {\n  dataIngestorExtensionPoint,\n  EmbeddingsProvider,\n  embeddingsProviderExtensionPoint,\n  Ingestor,\n  Model,\n  modelProviderExtensionPoint,\n  toolExtensionPoint,\n  callbackProviderExtensionPoint,\n  CallbackProvider,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-node';\nimport { Tool } from '@sweetoburrito/backstage-plugin-ai-assistant-common';\nimport { createDataIngestionPipeline } from './services/ingestor';\nimport { applyDatabaseMigrations } from './database/migrations';\nimport { PgVectorStore } from './database';\nimport { createSearchKnowledgeTool } from './tools/searchKnowledge';\nimport {\n  callbackServiceRef,\n  summarizerServiceRef,\n  modelServiceRef,\n  mcpServiceRef,\n  toolsServiceRef,\n  chatServiceRef,\n  userSettingsServiceRef,\n  conversationServiceRef,\n  agentServiceRef,\n} from './services';\n\n/**\n * aiAssistantPlugin backend plugin\n *\n * @public\n */\n\nexport const aiAssistantPlugin = createBackendPlugin({\n  pluginId: 'ai-assistant',\n  register(env) {\n    const ingestors: Ingestor[] = [];\n    const models: Model[] = [];\n    const tools: Tool[] = [];\n    const callbacks: CallbackProvider[] = [];\n\n    let embeddingsProvider: EmbeddingsProvider;\n\n    env.registerExtensionPoint(dataIngestorExtensionPoint, {\n      registerIngestor: ingestor => {\n        const existingIngestor = ingestors.find(i => i.id === ingestor.id);\n        if (existingIngestor) {\n          throw new Error(\n            `Ingestor with id ${ingestor.id} is already registered.`,\n          );\n        }\n        ingestors.push(ingestor);\n      },\n    });\n\n    env.registerExtensionPoint(embeddingsProviderExtensionPoint, {\n      register: provider => {\n        embeddingsProvider = provider;\n      },\n    });\n\n    env.registerExtensionPoint(modelProviderExtensionPoint, {\n      register: model => {\n        const existingModel = models.find(m => m.id === model.id);\n        if (existingModel) {\n          throw new Error(`Model with id ${model.id} is already registered.`);\n        }\n        models.push(model);\n      },\n    });\n\n    env.registerExtensionPoint(toolExtensionPoint, {\n      register: tool => {\n        const existingTool = tools.find(t => t.name === tool.name);\n        if (existingTool) {\n          throw new Error(`Tool with name ${tool.name} is already registered.`);\n        }\n        tools.push(tool);\n      },\n    });\n\n    env.registerExtensionPoint(callbackProviderExtensionPoint, {\n      register: callbackProvider => {\n        callbacks.push(callbackProvider);\n      },\n    });\n\n    env.registerInit({\n      deps: {\n        httpRouter: coreServices.httpRouter,\n        database: coreServices.database,\n        logger: coreServices.logger,\n        config: coreServices.rootConfig,\n        scheduler: coreServices.scheduler,\n        httpAuth: coreServices.httpAuth,\n        userInfo: coreServices.userInfo,\n        callback: callbackServiceRef,\n        summarizer: summarizerServiceRef,\n        model: modelServiceRef,\n        mcp: mcpServiceRef,\n        tool: toolsServiceRef,\n        chat: chatServiceRef,\n        userSettings: userSettingsServiceRef,\n        conversation: conversationServiceRef,\n        agent: agentServiceRef,\n      },\n\n      async init(options) {\n        const { httpRouter, database, callback, model, tool } = options;\n\n        const client = await database.getClient();\n\n        await applyDatabaseMigrations(client);\n\n        const vectorStore = await PgVectorStore.fromConfig(options);\n\n        if (!embeddingsProvider) {\n          throw new Error('No Embeddings Provider was registered.');\n        }\n\n        vectorStore.connectEmbeddings(await embeddingsProvider.getEmbeddings());\n\n        const dataIngestionPipeline = createDataIngestionPipeline({\n          ...options,\n          vectorStore,\n          ingestors,\n        });\n\n        const searchKnowledgeTool = createSearchKnowledgeTool({ vectorStore });\n\n        tool.registerTools(tools);\n        tool.registerCoreTools([searchKnowledgeTool]);\n        callback.registerCallbacks(callbacks);\n        model.registerModels(models);\n\n        httpRouter.use(await createRouter(options));\n        dataIngestionPipeline.start();\n      },\n    });\n  },\n});\n"],"names":["createBackendPlugin","tools","callbacks","dataIngestorExtensionPoint","embeddingsProviderExtensionPoint","modelProviderExtensionPoint","toolExtensionPoint","callbackProviderExtensionPoint","coreServices","callbackServiceRef","summarizerServiceRef","modelServiceRef","mcpServiceRef","toolsServiceRef","chatServiceRef","userSettingsServiceRef","conversationServiceRef","agentServiceRef","applyDatabaseMigrations","PgVectorStore","createDataIngestionPipeline","createSearchKnowledgeTool","createRouter"],"mappings":";;;;;;;;;;;;;;;;;;;AAuCO,MAAM,oBAAoBA,oCAAA,CAAoB;AAAA,EACnD,QAAA,EAAU,cAAA;AAAA,EACV,SAAS,GAAA,EAAK;AACZ,IAAA,MAAM,YAAwB,EAAC;AAC/B,IAAA,MAAM,SAAkB,EAAC;AACzB,IAAA,MAAMC,UAAgB,EAAC;AACvB,IAAA,MAAMC,cAAgC,EAAC;AAEvC,IAAA,IAAI,kBAAA;AAEJ,IAAA,GAAA,CAAI,uBAAuBC,yDAAA,EAA4B;AAAA,MACrD,kBAAkB,CAAA,QAAA,KAAY;AAC5B,QAAA,MAAM,mBAAmB,SAAA,CAAU,IAAA,CAAK,OAAK,CAAA,CAAE,EAAA,KAAO,SAAS,EAAE,CAAA;AACjE,QAAA,IAAI,gBAAA,EAAkB;AACpB,UAAA,MAAM,IAAI,KAAA;AAAA,YACR,CAAA,iBAAA,EAAoB,SAAS,EAAE,CAAA,uBAAA;AAAA,WACjC;AAAA,QACF;AACA,QAAA,SAAA,CAAU,KAAK,QAAQ,CAAA;AAAA,MACzB;AAAA,KACD,CAAA;AAED,IAAA,GAAA,CAAI,uBAAuBC,+DAAA,EAAkC;AAAA,MAC3D,UAAU,CAAA,QAAA,KAAY;AACpB,QAAA,kBAAA,GAAqB,QAAA;AAAA,MACvB;AAAA,KACD,CAAA;AAED,IAAA,GAAA,CAAI,uBAAuBC,0DAAA,EAA6B;AAAA,MACtD,UAAU,CAAA,KAAA,KAAS;AACjB,QAAA,MAAM,gBAAgB,MAAA,CAAO,IAAA,CAAK,OAAK,CAAA,CAAE,EAAA,KAAO,MAAM,EAAE,CAAA;AACxD,QAAA,IAAI,aAAA,EAAe;AACjB,UAAA,MAAM,IAAI,KAAA,CAAM,CAAA,cAAA,EAAiB,KAAA,CAAM,EAAE,CAAA,uBAAA,CAAyB,CAAA;AAAA,QACpE;AACA,QAAA,MAAA,CAAO,KAAK,KAAK,CAAA;AAAA,MACnB;AAAA,KACD,CAAA;AAED,IAAA,GAAA,CAAI,uBAAuBC,iDAAA,EAAoB;AAAA,MAC7C,UAAU,CAAA,IAAA,KAAQ;AAChB,QAAA,MAAM,eAAeL,OAAA,CAAM,IAAA,CAAK,OAAK,CAAA,CAAE,IAAA,KAAS,KAAK,IAAI,CAAA;AACzD,QAAA,IAAI,YAAA,EAAc;AAChB,UAAA,MAAM,IAAI,KAAA,CAAM,CAAA,eAAA,EAAkB,IAAA,CAAK,IAAI,CAAA,uBAAA,CAAyB,CAAA;AAAA,QACtE;AACA,QAAAA,OAAA,CAAM,KAAK,IAAI,CAAA;AAAA,MACjB;AAAA,KACD,CAAA;AAED,IAAA,GAAA,CAAI,uBAAuBM,6DAAA,EAAgC;AAAA,MACzD,UAAU,CAAA,gBAAA,KAAoB;AAC5B,QAAAL,WAAA,CAAU,KAAK,gBAAgB,CAAA;AAAA,MACjC;AAAA,KACD,CAAA;AAED,IAAA,GAAA,CAAI,YAAA,CAAa;AAAA,MACf,IAAA,EAAM;AAAA,QACJ,YAAYM,6BAAA,CAAa,UAAA;AAAA,QACzB,UAAUA,6BAAA,CAAa,QAAA;AAAA,QACvB,QAAQA,6BAAA,CAAa,MAAA;AAAA,QACrB,QAAQA,6BAAA,CAAa,UAAA;AAAA,QACrB,WAAWA,6BAAA,CAAa,SAAA;AAAA,QACxB,UAAUA,6BAAA,CAAa,QAAA;AAAA,QACvB,UAAUA,6BAAA,CAAa,QAAA;AAAA,QACvB,QAAA,EAAUC,4BAAA;AAAA,QACV,UAAA,EAAYC,+BAAA;AAAA,QACZ,KAAA,EAAOC,qBAAA;AAAA,QACP,GAAA,EAAKC,qBAAA;AAAA,QACL,IAAA,EAAMC,qBAAA;AAAA,QACN,IAAA,EAAMC,mBAAA;AAAA,QACN,YAAA,EAAcC,mCAAA;AAAA,QACd,YAAA,EAAcC,mCAAA;AAAA,QACd,KAAA,EAAOC;AAAA,OACT;AAAA,MAEA,MAAM,KAAK,OAAA,EAAS;AAClB,QAAA,MAAM,EAAE,UAAA,EAAY,QAAA,EAAU,QAAA,EAAU,KAAA,EAAO,MAAK,GAAI,OAAA;AAExD,QAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,SAAA,EAAU;AAExC,QAAA,MAAMC,mCAAwB,MAAM,CAAA;AAEpC,QAAA,MAAM,WAAA,GAAc,MAAMC,2BAAA,CAAc,UAAA,CAAW,OAAO,CAAA;AAE1D,QAAA,IAAI,CAAC,kBAAA,EAAoB;AACvB,UAAA,MAAM,IAAI,MAAM,wCAAwC,CAAA;AAAA,QAC1D;AAEA,QAAA,WAAA,CAAY,iBAAA,CAAkB,MAAM,kBAAA,CAAmB,aAAA,EAAe,CAAA;AAEtE,QAAA,MAAM,wBAAwBC,oCAAA,CAA4B;AAAA,UACxD,GAAG,OAAA;AAAA,UACH,WAAA;AAAA,UACA;AAAA,SACD,CAAA;AAED,QAAA,MAAM,mBAAA,GAAsBC,yCAAA,CAA0B,EAAE,WAAA,EAAa,CAAA;AAErE,QAAA,IAAA,CAAK,cAAcpB,OAAK,CAAA;AACxB,QAAA,IAAA,CAAK,iBAAA,CAAkB,CAAC,mBAAmB,CAAC,CAAA;AAC5C,QAAA,QAAA,CAAS,kBAAkBC,WAAS,CAAA;AACpC,QAAA,KAAA,CAAM,eAAe,MAAM,CAAA;AAE3B,QAAA,UAAA,CAAW,GAAA,CAAI,MAAMoB,oBAAA,CAAa,OAAO,CAAC,CAAA;AAC1C,QAAA,qBAAA,CAAsB,KAAA,EAAM;AAAA,MAC9B;AAAA,KACD,CAAA;AAAA,EACH;AACF,CAAC;;;;"}

package/dist/services/agent/helpers/message-parser.cjs.js

@@ -0,0 +1,31 @@
+'use strict';
+
+var uuid = require('uuid');
+
+const parseLangchainMessage = (message, traceId) => {
+  const id = uuid.validate(message.id) ? message.id : uuid.v4();
+  const role = message.getType();
+  const content = typeof message.content === "string" ? message.content : JSON.stringify(message.content);
+  const metadata = {};
+  if (role === "ai") {
+    const aiMessage = message;
+    metadata.toolCalls = aiMessage.tool_calls || [];
+    metadata.finishReason = aiMessage.response_metadata.finish_reason || void 0;
+    metadata.modelName = aiMessage.response_metadata.model_name || void 0;
+  }
+  if (role === "tool") {
+    const toolMessage = message;
+    metadata.name = toolMessage.name || "";
+  }
+  return {
+    id,
+    role,
+    content,
+    metadata,
+    score: 0,
+    traceId
+  };
+};
+
+exports.parseLangchainMessage = parseLangchainMessage;
+//# sourceMappingURL=message-parser.cjs.js.map

package/dist/services/agent/helpers/message-parser.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"message-parser.cjs.js","sources":["../../../../src/services/agent/helpers/message-parser.ts"],"sourcesContent":["import { AIMessage, ToolMessage, BaseMessage } from '@langchain/core/messages';\n\nimport {\n  JsonObject,\n  Message,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-common';\nimport { v4 as uuid, validate } from 'uuid';\n\nexport const parseLangchainMessage = (\n  message: BaseMessage,\n  traceId: string,\n): Message => {\n  const id = validate(message.id) ? message.id : uuid();\n  const role = message.getType();\n  const content =\n    typeof message.content === 'string'\n      ? message.content\n      : JSON.stringify(message.content);\n\n  const metadata: JsonObject = {};\n\n  if (role === 'ai') {\n    const aiMessage = message as AIMessage;\n\n    metadata.toolCalls = aiMessage.tool_calls || [];\n    metadata.finishReason =\n      aiMessage.response_metadata.finish_reason || undefined;\n    metadata.modelName = aiMessage.response_metadata.model_name || undefined;\n  }\n\n  if (role === 'tool') {\n    const toolMessage = message as ToolMessage;\n    metadata.name = toolMessage.name || '';\n  }\n\n  return {\n    id,\n    role,\n    content,\n    metadata,\n    score: 0,\n    traceId,\n  };\n};\n"],"names":["validate","uuid"],"mappings":";;;;AAQO,MAAM,qBAAA,GAAwB,CACnC,OAAA,EACA,OAAA,KACY;AACZ,EAAA,MAAM,KAAKA,aAAA,CAAS,OAAA,CAAQ,EAAE,CAAA,GAAI,OAAA,CAAQ,KAAKC,OAAA,EAAK;AACpD,EAAA,MAAM,IAAA,GAAO,QAAQ,OAAA,EAAQ;AAC7B,EAAA,MAAM,OAAA,GACJ,OAAO,OAAA,CAAQ,OAAA,KAAY,QAAA,GACvB,QAAQ,OAAA,GACR,IAAA,CAAK,SAAA,CAAU,OAAA,CAAQ,OAAO,CAAA;AAEpC,EAAA,MAAM,WAAuB,EAAC;AAE9B,EAAA,IAAI,SAAS,IAAA,EAAM;AACjB,IAAA,MAAM,SAAA,GAAY,OAAA;AAElB,IAAA,QAAA,CAAS,SAAA,GAAY,SAAA,CAAU,UAAA,IAAc,EAAC;AAC9C,IAAA,QAAA,CAAS,YAAA,GACP,SAAA,CAAU,iBAAA,CAAkB,aAAA,IAAiB,MAAA;AAC/C,IAAA,QAAA,CAAS,SAAA,GAAY,SAAA,CAAU,iBAAA,CAAkB,UAAA,IAAc,MAAA;AAAA,EACjE;AAEA,EAAA,IAAI,SAAS,MAAA,EAAQ;AACnB,IAAA,MAAM,WAAA,GAAc,OAAA;AACpB,IAAA,QAAA,CAAS,IAAA,GAAO,YAAY,IAAA,IAAQ,EAAA;AAAA,EACtC;AAEA,EAAA,OAAO;AAAA,IACL,EAAA;AAAA,IACA,IAAA;AAAA,IACA,OAAA;AAAA,IACA,QAAA;AAAA,IACA,KAAA,EAAO,CAAA;AAAA,IACP;AAAA,GACF;AACF;;;;"}

package/dist/services/agent/helpers/tool-filter.cjs.js

@@ -0,0 +1,13 @@
+'use strict';
+
+const toolFilter = (t, enabledTools) => {
+  if (enabledTools === void 0) return true;
+  if (enabledTools.length === 0) return false;
+  const enabled = enabledTools.find(
+    (enabledTool) => enabledTool.name === t.name && enabledTool.provider === t.provider
+  );
+  return !!enabled;
+};
+
+exports.toolFilter = toolFilter;
+//# sourceMappingURL=tool-filter.cjs.js.map

package/dist/services/agent/helpers/tool-filter.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-filter.cjs.js","sources":["../../../../src/services/agent/helpers/tool-filter.ts"],"sourcesContent":["import {\n  EnabledTool,\n  Tool,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-common';\n\nexport const toolFilter = (\n  t: Tool,\n  enabledTools: EnabledTool[] | undefined,\n): boolean => {\n  if (enabledTools === undefined) return true;\n\n  // If empty array, no tools should be enabled\n  if (enabledTools.length === 0) return false;\n  // Otherwise, only allow tools that are in the enabled list\n  const enabled = enabledTools.find(\n    enabledTool =>\n      enabledTool.name === t.name && enabledTool.provider === t.provider,\n  );\n  return !!enabled;\n};\n"],"names":[],"mappings":";;AAKO,MAAM,UAAA,GAAa,CACxB,CAAA,EACA,YAAA,KACY;AACZ,EAAA,IAAI,YAAA,KAAiB,QAAW,OAAO,IAAA;AAGvC,EAAA,IAAI,YAAA,CAAa,MAAA,KAAW,CAAA,EAAG,OAAO,KAAA;AAEtC,EAAA,MAAM,UAAU,YAAA,CAAa,IAAA;AAAA,IAC3B,iBACE,WAAA,CAAY,IAAA,KAAS,EAAE,IAAA,IAAQ,WAAA,CAAY,aAAa,CAAA,CAAE;AAAA,GAC9D;AACA,EAAA,OAAO,CAAC,CAAC,OAAA;AACX;;;;"}

package/dist/services/agent/index.cjs.js

@@ -0,0 +1,140 @@
+'use strict';
+
+var backendPluginApi = require('@backstage/backend-plugin-api');
+var prebuilt = require('@langchain/langgraph/prebuilt');
+var toolFilter = require('./helpers/tool-filter.cjs.js');
+var prompts = require('../../constants/prompts.cjs.js');
+var prompts$1 = require('@langchain/core/prompts');
+var callbacks = require('../callbacks.cjs.js');
+var model = require('../model.cjs.js');
+var tools = require('../tools.cjs.js');
+var messageParser = require('./helpers/message-parser.cjs.js');
+
+const createAgentService = ({
+  model,
+  tool,
+  config,
+  callback
+}) => {
+  const identityPrompt = config.getOptionalString("aiAssistant.prompt.identity") || prompts.DEFAULT_IDENTITY_PROMPT;
+  const formattingPrompt = config.getOptionalString("aiAssistant.prompt.formatting") || prompts.DEFAULT_FORMATTING_PROMPT;
+  const contentPrompt = config.getOptionalString("aiAssistant.prompt.content") || prompts.DEFAULT_SYSTEM_PROMPT;
+  const defaultBasePrompt = `${identityPrompt}
+
+${formattingPrompt}
+
+${contentPrompt}`;
+  const toolGuideline = config.getOptionalString("aiAssistant.prompt.toolGuideline") || prompts.DEFAULT_TOOL_GUIDELINE;
+  const systemPromptTemplate = prompts$1.SystemMessagePromptTemplate.fromTemplate(`
+    PURPOSE:
+    {systemPrompt}
+
+    TOOL USAGE GUIDELINES:
+    {toolGuideline}
+
+    Available tools:
+    {toolList}
+
+    Context:
+    {context}`);
+  const createAgent = async (options) => {
+    const {
+      modelId,
+      credentials,
+      tools: enabledTools,
+      systemPrompt = defaultBasePrompt,
+      context = "none",
+      metadata: { conversationId, userId, runId, runName }
+    } = options;
+    const { chatModel: llm, id: resolvedModelId } = model.getModel(
+      modelId ?? "default"
+    );
+    const tools = await tool.getPrincipalTools({
+      credentials,
+      filter: (t) => {
+        return toolFilter.toolFilter(t, enabledTools);
+      }
+    });
+    const toolList = tools.map((t) => `- ${t.name}: ${t.description}`).join("\n");
+    const agentPrompt = await systemPromptTemplate.formatMessages({
+      toolGuideline,
+      toolList,
+      context,
+      systemPrompt
+    });
+    const agent = prebuilt.createReactAgent({
+      llm,
+      tools,
+      prompt: agentPrompt[0].text
+    });
+    const { callbacks } = await callback.getChainCallbacks({
+      userId,
+      conversationId,
+      modelId: resolvedModelId
+    });
+    const { metadata } = await callback.getChainMetadata({
+      userId,
+      conversationId,
+      modelId: resolvedModelId
+    });
+    agent.config = {
+      ...agent.config,
+      callbacks,
+      metadata,
+      runId,
+      runName
+    };
+    return agent;
+  };
+  const stream = async (options) => {
+    const { messages, onStreamChunk } = options;
+    const agent = await createAgent(options);
+    const promptStream = await agent.stream(
+      {
+        messages
+      },
+      {
+        streamMode: ["values"]
+      }
+    );
+    for await (const [, chunk] of promptStream) {
+      const { messages: promptMessages } = chunk;
+      const parsedMessages = promptMessages.map(
+        (m) => messageParser.parseLangchainMessage(m, options.metadata.runId)
+      );
+      onStreamChunk(parsedMessages);
+    }
+  };
+  const prompt = async (options) => {
+    const {
+      messages,
+      metadata: { runId }
+    } = options;
+    const agent = await createAgent(options);
+    const result = await agent.invoke({
+      messages
+    });
+    return result.messages.map(
+      (m) => messageParser.parseLangchainMessage(m, runId)
+    );
+  };
+  return { prompt, stream };
+};
+const agentServiceRef = backendPluginApi.createServiceRef({
+  id: "ai-assistant.conversation-service",
+  defaultFactory: async (service) => backendPluginApi.createServiceFactory({
+    service,
+    deps: {
+      config: backendPluginApi.coreServices.rootConfig,
+      model: model.modelServiceRef,
+      tool: tools.toolsServiceRef,
+      callback: callbacks.callbackServiceRef
+    },
+    factory: async (options) => {
+      return createAgentService(options);
+    }
+  })
+});
+
+exports.agentServiceRef = agentServiceRef;
+//# sourceMappingURL=index.cjs.js.map

package/dist/services/agent/index.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.cjs.js","sources":["../../../src/services/agent/index.ts"],"sourcesContent":["import {\n  BackstageCredentials,\n  coreServices,\n  createServiceFactory,\n  createServiceRef,\n  RootConfigService,\n  ServiceRef,\n} from '@backstage/backend-plugin-api';\nimport { createReactAgent } from '@langchain/langgraph/prebuilt';\nimport {\n  EnabledTool,\n  Message,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-common';\nimport { toolFilter } from './helpers/tool-filter';\nimport {\n  DEFAULT_FORMATTING_PROMPT,\n  DEFAULT_IDENTITY_PROMPT,\n  DEFAULT_SYSTEM_PROMPT,\n  DEFAULT_TOOL_GUIDELINE,\n} from '../../constants/prompts';\nimport { SystemMessagePromptTemplate } from '@langchain/core/prompts';\n\nimport { CallbackService, callbackServiceRef } from '../callbacks';\nimport { modelServiceRef, ModelService } from '../model';\nimport { toolsServiceRef, ToolsService } from '../tools';\n\nimport { BaseMessage } from '@langchain/core/messages';\nimport { parseLangchainMessage } from './helpers/message-parser';\n\ntype PromptOptions = {\n  credentials: BackstageCredentials;\n  metadata: {\n    conversationId: string;\n    userId: string;\n    runName: string;\n    runId: string;\n  };\n  messages: Message[];\n  modelId?: string;\n  tools?: EnabledTool[];\n  systemPrompt?: string;\n  context?: string;\n};\n\ntype StreamOptions = PromptOptions & {\n  onStreamChunk: (messages: Message[]) => void;\n};\n\nexport type AgentService = {\n  prompt: (options: PromptOptions) => Promise<Message[]>;\n  stream: (options: StreamOptions) => Promise<void>;\n};\n\ntype AgentServiceOptions = {\n  model: ModelService;\n  config: RootConfigService;\n  tool: ToolsService;\n  callback: CallbackService;\n};\n\nconst createAgentService = ({\n  model,\n  tool,\n  config,\n  callback,\n}: AgentServiceOptions): AgentService => {\n  const identityPrompt =\n    config.getOptionalString('aiAssistant.prompt.identity') ||\n    DEFAULT_IDENTITY_PROMPT;\n\n  const formattingPrompt =\n    config.getOptionalString('aiAssistant.prompt.formatting') ||\n    DEFAULT_FORMATTING_PROMPT;\n\n  const contentPrompt =\n    config.getOptionalString('aiAssistant.prompt.content') ||\n    DEFAULT_SYSTEM_PROMPT;\n\n  const defaultBasePrompt = `${identityPrompt}\\n\\n${formattingPrompt}\\n\\n${contentPrompt}`;\n\n  const toolGuideline =\n    config.getOptionalString('aiAssistant.prompt.toolGuideline') ||\n    DEFAULT_TOOL_GUIDELINE;\n\n  const systemPromptTemplate = SystemMessagePromptTemplate.fromTemplate(`\n    PURPOSE:\n    {systemPrompt}\n\n    TOOL USAGE GUIDELINES:\n    {toolGuideline}\n\n    Available tools:\n    {toolList}\n\n    Context:\n    {context}`);\n\n  const createAgent = async (\n    options: PromptOptions,\n  ): Promise<ReturnType<typeof createReactAgent>> => {\n    const {\n      modelId,\n      credentials,\n      tools: enabledTools,\n      systemPrompt = defaultBasePrompt,\n      context = 'none',\n      metadata: { conversationId, userId, runId, runName },\n    } = options;\n\n    const { chatModel: llm, id: resolvedModelId } = model.getModel(\n      modelId ?? 'default',\n    );\n\n    const tools = await tool.getPrincipalTools({\n      credentials,\n      filter: t => {\n        return toolFilter(t, enabledTools);\n      },\n    });\n\n    const toolList = tools.map(t => `- ${t.name}: ${t.description}`).join('\\n');\n\n    const agentPrompt = await systemPromptTemplate.formatMessages({\n      toolGuideline,\n      toolList,\n      context,\n      systemPrompt,\n    });\n\n    const agent = createReactAgent({\n      llm,\n      tools,\n      prompt: agentPrompt[0].text,\n    });\n\n    const { callbacks } = await callback.getChainCallbacks({\n      userId,\n      conversationId,\n      modelId: resolvedModelId,\n    });\n\n    const { metadata } = await callback.getChainMetadata({\n      userId,\n      conversationId,\n      modelId: resolvedModelId,\n    });\n\n    agent.config = {\n      ...agent.config,\n      callbacks,\n      metadata,\n      runId,\n      runName,\n    };\n\n    return agent;\n  };\n\n  const stream: AgentService['stream'] = async options => {\n    const { messages, onStreamChunk } = options;\n\n    const agent = await createAgent(options);\n\n    const promptStream = await agent.stream(\n      {\n        messages,\n      },\n      {\n        streamMode: ['values'],\n      },\n    );\n\n    for await (const [, chunk] of promptStream) {\n      const { messages: promptMessages } = chunk;\n\n      const parsedMessages: Message[] = (promptMessages as BaseMessage[]).map(\n        m => parseLangchainMessage(m, options.metadata.runId),\n      );\n\n      onStreamChunk(parsedMessages);\n    }\n  };\n\n  const prompt: AgentService['prompt'] = async options => {\n    const {\n      messages,\n      metadata: { runId },\n    } = options;\n    const agent = await createAgent(options);\n\n    const result = await agent.invoke({\n      messages,\n    });\n\n    return (result.messages as BaseMessage[]).map(m =>\n      parseLangchainMessage(m, runId),\n    );\n  };\n\n  return { prompt, stream };\n};\n\nexport const agentServiceRef: ServiceRef<AgentService, 'plugin', 'singleton'> =\n  createServiceRef<AgentService>({\n    id: 'ai-assistant.conversation-service',\n    defaultFactory: async service =>\n      createServiceFactory({\n        service,\n        deps: {\n          config: coreServices.rootConfig,\n          model: modelServiceRef,\n          tool: toolsServiceRef,\n          callback: callbackServiceRef,\n        },\n        factory: async options => {\n          return createAgentService(options);\n        },\n      }),\n  });\n"],"names":["DEFAULT_IDENTITY_PROMPT","DEFAULT_FORMATTING_PROMPT","DEFAULT_SYSTEM_PROMPT","DEFAULT_TOOL_GUIDELINE","SystemMessagePromptTemplate","toolFilter","createReactAgent","parseLangchainMessage","createServiceRef","createServiceFactory","coreServices","modelServiceRef","toolsServiceRef","callbackServiceRef"],"mappings":";;;;;;;;;;;;AA4DA,MAAM,qBAAqB,CAAC;AAAA,EAC1B,KAAA;AAAA,EACA,IAAA;AAAA,EACA,MAAA;AAAA,EACA;AACF,CAAA,KAAyC;AACvC,EAAA,MAAM,cAAA,GACJ,MAAA,CAAO,iBAAA,CAAkB,6BAA6B,CAAA,IACtDA,+BAAA;AAEF,EAAA,MAAM,gBAAA,GACJ,MAAA,CAAO,iBAAA,CAAkB,+BAA+B,CAAA,IACxDC,iCAAA;AAEF,EAAA,MAAM,aAAA,GACJ,MAAA,CAAO,iBAAA,CAAkB,4BAA4B,CAAA,IACrDC,6BAAA;AAEF,EAAA,MAAM,iBAAA,GAAoB,GAAG,cAAc;;AAAA,EAAO,gBAAgB;;AAAA,EAAO,aAAa,CAAA,CAAA;AAEtF,EAAA,MAAM,aAAA,GACJ,MAAA,CAAO,iBAAA,CAAkB,kCAAkC,CAAA,IAC3DC,8BAAA;AAEF,EAAA,MAAM,oBAAA,GAAuBC,sCAA4B,YAAA,CAAa;AAAA;AAAA;;AAAA;AAAA;;AAAA;AAAA;;AAAA;AAAA,aAAA,CAW1D,CAAA;AAEZ,EAAA,MAAM,WAAA,GAAc,OAClB,OAAA,KACiD;AACjD,IAAA,MAAM;AAAA,MACJ,OAAA;AAAA,MACA,WAAA;AAAA,MACA,KAAA,EAAO,YAAA;AAAA,MACP,YAAA,GAAe,iBAAA;AAAA,MACf,OAAA,GAAU,MAAA;AAAA,MACV,QAAA,EAAU,EAAE,cAAA,EAAgB,MAAA,EAAQ,OAAO,OAAA;AAAQ,KACrD,GAAI,OAAA;AAEJ,IAAA,MAAM,EAAE,SAAA,EAAW,GAAA,EAAK,EAAA,EAAI,eAAA,KAAoB,KAAA,CAAM,QAAA;AAAA,MACpD,OAAA,IAAW;AAAA,KACb;AAEA,IAAA,MAAM,KAAA,GAAQ,MAAM,IAAA,CAAK,iBAAA,CAAkB;AAAA,MACzC,WAAA;AAAA,MACA,QAAQ,CAAA,CAAA,KAAK;AACX,QAAA,OAAOC,qBAAA,CAAW,GAAG,YAAY,CAAA;AAAA,MACnC;AAAA,KACD,CAAA;AAED,IAAA,MAAM,QAAA,GAAW,KAAA,CAAM,GAAA,CAAI,CAAA,CAAA,KAAK,CAAA,EAAA,EAAK,CAAA,CAAE,IAAI,CAAA,EAAA,EAAK,CAAA,CAAE,WAAW,CAAA,CAAE,CAAA,CAAE,KAAK,IAAI,CAAA;AAE1E,IAAA,MAAM,WAAA,GAAc,MAAM,oBAAA,CAAqB,cAAA,CAAe;AAAA,MAC5D,aAAA;AAAA,MACA,QAAA;AAAA,MACA,OAAA;AAAA,MACA;AAAA,KACD,CAAA;AAED,IAAA,MAAM,QAAQC,yBAAA,CAAiB;AAAA,MAC7B,GAAA;AAAA,MACA,KAAA;AAAA,MACA,MAAA,EAAQ,WAAA,CAAY,CAAC,CAAA,CAAE;AAAA,KACxB,CAAA;AAED,IAAA,MAAM,EAAE,SAAA,EAAU,GAAI,MAAM,SAAS,iBAAA,CAAkB;AAAA,MACrD,MAAA;AAAA,MACA,cAAA;AAAA,MACA,OAAA,EAAS;AAAA,KACV,CAAA;AAED,IAAA,MAAM,EAAE,QAAA,EAAS,GAAI,MAAM,SAAS,gBAAA,CAAiB;AAAA,MACnD,MAAA;AAAA,MACA,cAAA;AAAA,MACA,OAAA,EAAS;AAAA,KACV,CAAA;AAED,IAAA,KAAA,CAAM,MAAA,GAAS;AAAA,MACb,GAAG,KAAA,CAAM,MAAA;AAAA,MACT,SAAA;AAAA,MACA,QAAA;AAAA,MACA,KAAA;AAAA,MACA;AAAA,KACF;AAEA,IAAA,OAAO,KAAA;AAAA,EACT,CAAA;AAEA,EAAA,MAAM,MAAA,GAAiC,OAAM,OAAA,KAAW;AACtD,IAAA,MAAM,EAAE,QAAA,EAAU,aAAA,EAAc,GAAI,OAAA;AAEpC,IAAA,MAAM,KAAA,GAAQ,MAAM,WAAA,CAAY,OAAO,CAAA;AAEvC,IAAA,MAAM,YAAA,GAAe,MAAM,KAAA,CAAM,MAAA;AAAA,MAC/B;AAAA,QACE;AAAA,OACF;AAAA,MACA;AAAA,QACE,UAAA,EAAY,CAAC,QAAQ;AAAA;AACvB,KACF;AAEA,IAAA,WAAA,MAAiB,GAAG,KAAK,CAAA,IAAK,YAAA,EAAc;AAC1C,MAAA,MAAM,EAAE,QAAA,EAAU,cAAA,EAAe,GAAI,KAAA;AAErC,MAAA,MAAM,iBAA6B,cAAA,CAAiC,GAAA;AAAA,QAClE,CAAA,CAAA,KAAKC,mCAAA,CAAsB,CAAA,EAAG,OAAA,CAAQ,SAAS,KAAK;AAAA,OACtD;AAEA,MAAA,aAAA,CAAc,cAAc,CAAA;AAAA,IAC9B;AAAA,EACF,CAAA;AAEA,EAAA,MAAM,MAAA,GAAiC,OAAM,OAAA,KAAW;AACtD,IAAA,MAAM;AAAA,MACJ,QAAA;AAAA,MACA,QAAA,EAAU,EAAE,KAAA;AAAM,KACpB,GAAI,OAAA;AACJ,IAAA,MAAM,KAAA,GAAQ,MAAM,WAAA,CAAY,OAAO,CAAA;AAEvC,IAAA,MAAM,MAAA,GAAS,MAAM,KAAA,CAAM,MAAA,CAAO;AAAA,MAChC;AAAA,KACD,CAAA;AAED,IAAA,OAAQ,OAAO,QAAA,CAA2B,GAAA;AAAA,MAAI,CAAA,CAAA,KAC5CA,mCAAA,CAAsB,CAAA,EAAG,KAAK;AAAA,KAChC;AAAA,EACF,CAAA;AAEA,EAAA,OAAO,EAAE,QAAQ,MAAA,EAAO;AAC1B,CAAA;AAEO,MAAM,kBACXC,iCAAA,CAA+B;AAAA,EAC7B,EAAA,EAAI,mCAAA;AAAA,EACJ,cAAA,EAAgB,OAAM,OAAA,KACpBC,qCAAA,CAAqB;AAAA,IACnB,OAAA;AAAA,IACA,IAAA,EAAM;AAAA,MACJ,QAAQC,6BAAA,CAAa,UAAA;AAAA,MACrB,KAAA,EAAOC,qBAAA;AAAA,MACP,IAAA,EAAMC,qBAAA;AAAA,MACN,QAAA,EAAUC;AAAA,KACZ;AAAA,IACA,OAAA,EAAS,OAAM,OAAA,KAAW;AACxB,MAAA,OAAO,mBAAmB,OAAO,CAAA;AAAA,IACnC;AAAA,GACD;AACL,CAAC;;;;"}

package/dist/services/callbacks.cjs.js

@@ -1,8 +1,9 @@
 'use strict';
 
-const createCallbackService = async ({
-  callbacks
-}) => {
+var backendPluginApi = require('@backstage/backend-plugin-api');
+
+const createCallbackService = async ({}) => {
+  const callbacks = [];
   const getChainCallbacks = async (options) => {
     const callbackHandlers = [];
     for (const { chainCallback } of callbacks) {
@@ -27,10 +28,7 @@ const createCallbackService = async ({
     }
     return { metadata };
   };
-  const handleScoreCallbacks = async ({
-    name,
-    message
-  }) => {
+  const handleScoreCallbacks = async ({ name, message }) => {
     callbacks.forEach(async ({ scoreCallback }) => {
       if (!scoreCallback) {
         return;
@@ -38,12 +36,28 @@ const createCallbackService = async ({
       scoreCallback({ name, message });
     });
   };
+  const registerCallbacks = (newCallbacks) => {
+    newCallbacks.forEach((callback) => {
+      callbacks.push(callback);
+    });
+  };
   return {
     getChainCallbacks,
     getChainMetadata,
-    handleScoreCallbacks
+    handleScoreCallbacks,
+    registerCallbacks
   };
 };
+const callbackServiceRef = backendPluginApi.createServiceRef({
+  id: "ai-assistant.callback-service",
+  defaultFactory: async (service) => backendPluginApi.createServiceFactory({
+    service,
+    deps: {},
+    factory: async (options) => {
+      return createCallbackService(options);
+    }
+  })
+});
 
-exports.createCallbackService = createCallbackService;
+exports.callbackServiceRef = callbackServiceRef;
 //# sourceMappingURL=callbacks.cjs.js.map

package/dist/services/callbacks.cjs.js.map

@@ -1 +1 @@
-{"version":3,"file":"callbacks.cjs.js","sources":["../../src/services/callbacks.ts"],"sourcesContent":["import {\n  CallbackProvider,\n  ChainMetadata,\n  ChainCallbackOptions,\n  ChainCallback,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-node';\n\nexport type CallbackService = {\n  getChainCallbacks: (options: ChainCallbackOptions) => Promise<{\n    callbacks: ReturnType<ChainCallback>[];\n  }>;\n\n  getChainMetadata: (\n    options: ChainCallbackOptions,\n  ) => Promise<{ metadata: ChainMetadata }>;\n\n  handleScoreCallbacks: NonNullable<CallbackProvider['scoreCallback']>;\n};\n\nexport type CreateCallbackServiceOptions = {\n  callbacks: CallbackProvider[];\n};\n\nexport const createCallbackService = async ({\n  callbacks,\n}: CreateCallbackServiceOptions): Promise<CallbackService> => {\n  const getChainCallbacks: CallbackService['getChainCallbacks'] =\n    async options => {\n      const callbackHandlers: ReturnType<ChainCallback>[] = [];\n\n      for (const { chainCallback } of callbacks) {\n        if (!chainCallback) {\n          continue;\n        }\n\n        const callbackHandler = await chainCallback(options);\n\n        callbackHandlers.push(callbackHandler);\n      }\n\n      return {\n        callbacks: callbackHandlers,\n      };\n    };\n\n  const getChainMetadata: CallbackService['getChainMetadata'] =\n    async options => {\n      const metadata: ChainMetadata = {};\n\n      for (const { chainMetadataCallback } of callbacks) {\n        if (!chainMetadataCallback) {\n          continue;\n        }\n\n        const callbackData = await chainMetadataCallback(options);\n\n        Object.assign(metadata, callbackData.metadata);\n      }\n\n      return { metadata };\n    };\n\n  const handleScoreCallbacks: CallbackService['handleScoreCallbacks'] = async ({\n    name,\n    message,\n  }) => {\n    callbacks.forEach(async ({ scoreCallback }) => {\n      if (!scoreCallback) {\n        return;\n      }\n\n      scoreCallback({ name, message });\n    });\n  };\n\n  return {\n    getChainCallbacks,\n    getChainMetadata,\n    handleScoreCallbacks,\n  };\n};\n"],"names":[],"mappings":";;AAuBO,MAAM,wBAAwB,OAAO;AAAA,EAC1C;AACF,CAAA,KAA8D;AAC5D,EAAA,MAAM,iBAAA,GACJ,OAAM,OAAA,KAAW;AACf,IAAA,MAAM,mBAAgD,EAAC;AAEvD,IAAA,KAAA,MAAW,EAAE,aAAA,EAAc,IAAK,SAAA,EAAW;AACzC,MAAA,IAAI,CAAC,aAAA,EAAe;AAClB,QAAA;AAAA,MACF;AAEA,MAAA,MAAM,eAAA,GAAkB,MAAM,aAAA,CAAc,OAAO,CAAA;AAEnD,MAAA,gBAAA,CAAiB,KAAK,eAAe,CAAA;AAAA,IACvC;AAEA,IAAA,OAAO;AAAA,MACL,SAAA,EAAW;AAAA,KACb;AAAA,EACF,CAAA;AAEF,EAAA,MAAM,gBAAA,GACJ,OAAM,OAAA,KAAW;AACf,IAAA,MAAM,WAA0B,EAAC;AAEjC,IAAA,KAAA,MAAW,EAAE,qBAAA,EAAsB,IAAK,SAAA,EAAW;AACjD,MAAA,IAAI,CAAC,qBAAA,EAAuB;AAC1B,QAAA;AAAA,MACF;AAEA,MAAA,MAAM,YAAA,GAAe,MAAM,qBAAA,CAAsB,OAAO,CAAA;AAExD,MAAA,MAAA,CAAO,MAAA,CAAO,QAAA,EAAU,YAAA,CAAa,QAAQ,CAAA;AAAA,IAC/C;AAEA,IAAA,OAAO,EAAE,QAAA,EAAS;AAAA,EACpB,CAAA;AAEF,EAAA,MAAM,uBAAgE,OAAO;AAAA,IAC3E,IAAA;AAAA,IACA;AAAA,GACF,KAAM;AACJ,IAAA,SAAA,CAAU,OAAA,CAAQ,OAAO,EAAE,aAAA,EAAc,KAAM;AAC7C,MAAA,IAAI,CAAC,aAAA,EAAe;AAClB,QAAA;AAAA,MACF;AAEA,MAAA,aAAA,CAAc,EAAE,IAAA,EAAM,OAAA,EAAS,CAAA;AAAA,IACjC,CAAC,CAAA;AAAA,EACH,CAAA;AAEA,EAAA,OAAO;AAAA,IACL,iBAAA;AAAA,IACA,gBAAA;AAAA,IACA;AAAA,GACF;AACF;;;;"}
+{"version":3,"file":"callbacks.cjs.js","sources":["../../src/services/callbacks.ts"],"sourcesContent":["import {\n  CallbackProvider,\n  ChainMetadata,\n  ChainCallbackOptions,\n  ChainCallback,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-node';\n\nimport {\n  createServiceFactory,\n  createServiceRef,\n  ServiceRef,\n} from '@backstage/backend-plugin-api';\n\nexport type CallbackService = {\n  getChainCallbacks: (options: ChainCallbackOptions) => Promise<{\n    callbacks: ReturnType<ChainCallback>[];\n  }>;\n\n  getChainMetadata: (\n    options: ChainCallbackOptions,\n  ) => Promise<{ metadata: ChainMetadata }>;\n\n  handleScoreCallbacks: NonNullable<CallbackProvider['scoreCallback']>;\n\n  registerCallbacks: (callbacks: CallbackProvider[]) => void;\n};\n\nexport type CreateCallbackServiceOptions = {};\n\nconst createCallbackService =\n  async ({}: CreateCallbackServiceOptions): Promise<CallbackService> => {\n    const callbacks: CallbackProvider[] = [];\n\n    const getChainCallbacks: CallbackService['getChainCallbacks'] =\n      async options => {\n        const callbackHandlers: ReturnType<ChainCallback>[] = [];\n\n        for (const { chainCallback } of callbacks) {\n          if (!chainCallback) {\n            continue;\n          }\n\n          const callbackHandler = await chainCallback(options);\n\n          callbackHandlers.push(callbackHandler);\n        }\n\n        return {\n          callbacks: callbackHandlers,\n        };\n      };\n\n    const getChainMetadata: CallbackService['getChainMetadata'] =\n      async options => {\n        const metadata: ChainMetadata = {};\n\n        for (const { chainMetadataCallback } of callbacks) {\n          if (!chainMetadataCallback) {\n            continue;\n          }\n\n          const callbackData = await chainMetadataCallback(options);\n\n          Object.assign(metadata, callbackData.metadata);\n        }\n\n        return { metadata };\n      };\n\n    const handleScoreCallbacks: CallbackService['handleScoreCallbacks'] =\n      async ({ name, message }) => {\n        callbacks.forEach(async ({ scoreCallback }) => {\n          if (!scoreCallback) {\n            return;\n          }\n\n          scoreCallback({ name, message });\n        });\n      };\n\n    const registerCallbacks: CallbackService['registerCallbacks'] =\n      newCallbacks => {\n        newCallbacks.forEach(callback => {\n          callbacks.push(callback);\n        });\n      };\n\n    return {\n      getChainCallbacks,\n      getChainMetadata,\n      handleScoreCallbacks,\n      registerCallbacks,\n    };\n  };\n\nexport const callbackServiceRef: ServiceRef<\n  CallbackService,\n  'plugin',\n  'singleton'\n> = createServiceRef<CallbackService>({\n  id: 'ai-assistant.callback-service',\n  defaultFactory: async service =>\n    createServiceFactory({\n      service,\n      deps: {},\n      factory: async options => {\n        return createCallbackService(options);\n      },\n    }),\n});\n"],"names":["createServiceRef","createServiceFactory"],"mappings":";;;;AA6BA,MAAM,qBAAA,GACJ,OAAO,EAAC,KAA8D;AACpE,EAAA,MAAM,YAAgC,EAAC;AAEvC,EAAA,MAAM,iBAAA,GACJ,OAAM,OAAA,KAAW;AACf,IAAA,MAAM,mBAAgD,EAAC;AAEvD,IAAA,KAAA,MAAW,EAAE,aAAA,EAAc,IAAK,SAAA,EAAW;AACzC,MAAA,IAAI,CAAC,aAAA,EAAe;AAClB,QAAA;AAAA,MACF;AAEA,MAAA,MAAM,eAAA,GAAkB,MAAM,aAAA,CAAc,OAAO,CAAA;AAEnD,MAAA,gBAAA,CAAiB,KAAK,eAAe,CAAA;AAAA,IACvC;AAEA,IAAA,OAAO;AAAA,MACL,SAAA,EAAW;AAAA,KACb;AAAA,EACF,CAAA;AAEF,EAAA,MAAM,gBAAA,GACJ,OAAM,OAAA,KAAW;AACf,IAAA,MAAM,WAA0B,EAAC;AAEjC,IAAA,KAAA,MAAW,EAAE,qBAAA,EAAsB,IAAK,SAAA,EAAW;AACjD,MAAA,IAAI,CAAC,qBAAA,EAAuB;AAC1B,QAAA;AAAA,MACF;AAEA,MAAA,MAAM,YAAA,GAAe,MAAM,qBAAA,CAAsB,OAAO,CAAA;AAExD,MAAA,MAAA,CAAO,MAAA,CAAO,QAAA,EAAU,YAAA,CAAa,QAAQ,CAAA;AAAA,IAC/C;AAEA,IAAA,OAAO,EAAE,QAAA,EAAS;AAAA,EACpB,CAAA;AAEF,EAAA,MAAM,oBAAA,GACJ,OAAO,EAAE,IAAA,EAAM,SAAQ,KAAM;AAC3B,IAAA,SAAA,CAAU,OAAA,CAAQ,OAAO,EAAE,aAAA,EAAc,KAAM;AAC7C,MAAA,IAAI,CAAC,aAAA,EAAe;AAClB,QAAA;AAAA,MACF;AAEA,MAAA,aAAA,CAAc,EAAE,IAAA,EAAM,OAAA,EAAS,CAAA;AAAA,IACjC,CAAC,CAAA;AAAA,EACH,CAAA;AAEF,EAAA,MAAM,oBACJ,CAAA,YAAA,KAAgB;AACd,IAAA,YAAA,CAAa,QAAQ,CAAA,QAAA,KAAY;AAC/B,MAAA,SAAA,CAAU,KAAK,QAAQ,CAAA;AAAA,IACzB,CAAC,CAAA;AAAA,EACH,CAAA;AAEF,EAAA,OAAO;AAAA,IACL,iBAAA;AAAA,IACA,gBAAA;AAAA,IACA,oBAAA;AAAA,IACA;AAAA,GACF;AACF,CAAA;AAEK,MAAM,qBAITA,iCAAA,CAAkC;AAAA,EACpC,EAAA,EAAI,+BAAA;AAAA,EACJ,cAAA,EAAgB,OAAM,OAAA,KACpBC,qCAAA,CAAqB;AAAA,IACnB,OAAA;AAAA,IACA,MAAM,EAAC;AAAA,IACP,OAAA,EAAS,OAAM,OAAA,KAAW;AACxB,MAAA,OAAO,sBAAsB,OAAO,CAAA;AAAA,IACtC;AAAA,GACD;AACL,CAAC;;;;"}