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

package/dist/services/router/chat.cjs.js

@@ -11,12 +11,12 @@ var Router__default = /*#__PURE__*/_interopDefaultCompat(Router);
 var z__default = /*#__PURE__*/_interopDefaultCompat(z);
 
 async function createChatRouter(options) {
-  const { chat, httpAuth, userInfo } = options;
+  const { chat, httpAuth, userInfo, conversation } = options;
   const router = Router__default.default();
   const messageSchema = z__default.default.object({
     messages: z__default.default.array(
       z__default.default.object({
-        id: z__default.default.string().uuid().optional().default(uuid.v4),
+        id: z__default.default.uuid().optional().default(uuid.v4),
         role: z__default.default.string(),
         content: z__default.default.string()
       })
@@ -36,13 +36,13 @@ async function createChatRouter(options) {
     validation.validation(messageSchema, "body"),
     async (req, res) => {
       const { messages, conversationId, modelId, stream, tools } = req.body;
-      const userCredentials = await httpAuth.credentials(req);
+      const credentials = await httpAuth.credentials(req);
       const responseMessages = await chat.prompt({
         modelId,
         messages,
         conversationId,
         stream,
-        userCredentials,
+        credentials,
         tools
       });
       res.json({
@@ -57,25 +57,20 @@ async function createChatRouter(options) {
   router.get("/conversations", async (req, res) => {
     const credentials = await httpAuth.credentials(req);
     const { userEntityRef } = await userInfo.getUserInfo(credentials);
-    const conversations = await chat.getConversations({
+    const conversations = await conversation.getConversations({
       userEntityRef
     });
     res.json({ conversations });
   });
-  router.get("/tools", async (req, res) => {
-    const credentials = await httpAuth.credentials(req);
-    const tools = await chat.getAvailableTools({ credentials });
-    res.json({ tools });
-  });
   router.get("/:id", validation.validation(chatSchema, "params"), async (req, res) => {
     const { id } = req.params;
     const credentials = await httpAuth.credentials(req);
     const { userEntityRef } = await userInfo.getUserInfo(credentials);
-    const conversation = await chat.getConversation({
+    const conversationMessages = await conversation.getConversation({
       conversationId: id,
       userEntityRef
     });
-    res.json({ conversation });
+    res.json({ conversation: conversationMessages });
   });
   router.post(
     "/message/:messageId/score",
@@ -94,7 +89,7 @@ async function createChatRouter(options) {
     async (req, res) => {
       const { messageId } = req.params;
       const { score } = req.body;
-      await chat.scoreMessage(messageId, score);
+      await conversation.scoreMessage(messageId, score);
       res.status(204).end();
     }
   );

package/dist/services/router/chat.cjs.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat.cjs.js","sources":["../../../src/services/router/chat.ts"],"sourcesContent":["import express from 'express';\nimport Router from 'express-promise-router';\nimport { ChatService } from '../chat';\nimport z from 'zod';\nimport { validation } from './middleware/validation';\nimport { v4 as uuid } from 'uuid';\nimport {\n  DatabaseService,\n  HttpAuthService,\n  UserInfoService,\n} from '@backstage/backend-plugin-api';\n\nexport type ChatRouterOptions = {\n  chat: ChatService;\n  database: DatabaseService;\n  httpAuth: HttpAuthService;\n  userInfo: UserInfoService;\n};\n\nexport async function createChatRouter(\n  options: ChatRouterOptions,\n): Promise<express.Router> {\n  const { chat, httpAuth, userInfo } = options;\n\n  const router = Router();\n\n  const messageSchema = z.object({\n    messages: z.array(\n      z.object({\n        id: z.string().uuid().optional().default(uuid),\n        role: z.string(),\n        content: z.string(),\n      }),\n    ),\n    modelId: z.string(),\n    conversationId: z.uuid().optional().default(uuid),\n    stream: z.boolean().optional(),\n    tools: z\n      .array(\n        z.object({\n          name: z.string(),\n          provider: z.string(),\n        }),\n      )\n      .optional(),\n  });\n\n  router.post(\n    '/message',\n    validation(messageSchema, 'body'),\n    async (req, res) => {\n      const { messages, conversationId, modelId, stream, tools } = req.body;\n\n      const userCredentials = await httpAuth.credentials(req);\n\n      const responseMessages = await chat.prompt({\n        modelId,\n        messages,\n        conversationId,\n        stream,\n        userCredentials,\n        tools,\n      });\n\n      res.json({\n        messages: responseMessages,\n        conversationId,\n      });\n    },\n  );\n\n  const chatSchema = z.object({\n    id: z.uuid(),\n  });\n\n  router.get('/conversations', async (req, res) => {\n    const credentials = await httpAuth.credentials(req);\n    const { userEntityRef } = await userInfo.getUserInfo(credentials);\n\n    const conversations = await chat.getConversations({\n      userEntityRef,\n    });\n\n    res.json({ conversations });\n  });\n\n  router.get('/tools', async (req, res) => {\n    const credentials = await httpAuth.credentials(req);\n\n    const tools = await chat.getAvailableTools({ credentials });\n\n    res.json({ tools });\n  });\n\n  router.get('/:id', validation(chatSchema, 'params'), async (req, res) => {\n    const { id } = req.params;\n\n    const credentials = await httpAuth.credentials(req);\n    const { userEntityRef } = await userInfo.getUserInfo(credentials);\n\n    const conversation = await chat.getConversation({\n      conversationId: id,\n      userEntityRef,\n    });\n\n    res.json({ conversation });\n  });\n\n  router.post(\n    '/message/:messageId/score',\n    validation(\n      z.object({\n        messageId: z.string().uuid(),\n      }),\n      'params',\n    ),\n    validation(\n      z.object({\n        score: z.number(),\n      }),\n      'body',\n    ),\n    async (req, res) => {\n      const { messageId } = req.params;\n      const { score } = req.body;\n\n      await chat.scoreMessage(messageId, score);\n\n      res.status(204).end();\n    },\n  );\n\n  return router;\n}\n"],"names":["Router","z","uuid","validation"],"mappings":";;;;;;;;;;;;AAmBA,eAAsB,iBACpB,OAAA,EACyB;AACzB,EAAA,MAAM,EAAE,IAAA,EAAM,QAAA,EAAU,QAAA,EAAS,GAAI,OAAA;AAErC,EAAA,MAAM,SAASA,uBAAA,EAAO;AAEtB,EAAA,MAAM,aAAA,GAAgBC,mBAAE,MAAA,CAAO;AAAA,IAC7B,UAAUA,kBAAA,CAAE,KAAA;AAAA,MACVA,mBAAE,MAAA,CAAO;AAAA,QACP,EAAA,EAAIA,mBAAE,MAAA,EAAO,CAAE,MAAK,CAAE,QAAA,EAAS,CAAE,OAAA,CAAQC,OAAI,CAAA;AAAA,QAC7C,IAAA,EAAMD,mBAAE,MAAA,EAAO;AAAA,QACf,OAAA,EAASA,mBAAE,MAAA;AAAO,OACnB;AAAA,KACH;AAAA,IACA,OAAA,EAASA,mBAAE,MAAA,EAAO;AAAA,IAClB,gBAAgBA,kBAAA,CAAE,IAAA,GAAO,QAAA,EAAS,CAAE,QAAQC,OAAI,CAAA;AAAA,IAChD,MAAA,EAAQD,kBAAA,CAAE,OAAA,EAAQ,CAAE,QAAA,EAAS;AAAA,IAC7B,OAAOA,kBAAA,CACJ,KAAA;AAAA,MACCA,mBAAE,MAAA,CAAO;AAAA,QACP,IAAA,EAAMA,mBAAE,MAAA,EAAO;AAAA,QACf,QAAA,EAAUA,mBAAE,MAAA;AAAO,OACpB;AAAA,MAEF,QAAA;AAAS,GACb,CAAA;AAED,EAAA,MAAA,CAAO,IAAA;AAAA,IACL,UAAA;AAAA,IACAE,qBAAA,CAAW,eAAe,MAAM,CAAA;AAAA,IAChC,OAAO,KAAK,GAAA,KAAQ;AAClB,MAAA,MAAM,EAAE,QAAA,EAAU,cAAA,EAAgB,SAAS,MAAA,EAAQ,KAAA,KAAU,GAAA,CAAI,IAAA;AAEjE,MAAA,MAAM,eAAA,GAAkB,MAAM,QAAA,CAAS,WAAA,CAAY,GAAG,CAAA;AAEtD,MAAA,MAAM,gBAAA,GAAmB,MAAM,IAAA,CAAK,MAAA,CAAO;AAAA,QACzC,OAAA;AAAA,QACA,QAAA;AAAA,QACA,cAAA;AAAA,QACA,MAAA;AAAA,QACA,eAAA;AAAA,QACA;AAAA,OACD,CAAA;AAED,MAAA,GAAA,CAAI,IAAA,CAAK;AAAA,QACP,QAAA,EAAU,gBAAA;AAAA,QACV;AAAA,OACD,CAAA;AAAA,IACH;AAAA,GACF;AAEA,EAAA,MAAM,UAAA,GAAaF,mBAAE,MAAA,CAAO;AAAA,IAC1B,EAAA,EAAIA,mBAAE,IAAA;AAAK,GACZ,CAAA;AAED,EAAA,MAAA,CAAO,GAAA,CAAI,gBAAA,EAAkB,OAAO,GAAA,EAAK,GAAA,KAAQ;AAC/C,IAAA,MAAM,WAAA,GAAc,MAAM,QAAA,CAAS,WAAA,CAAY,GAAG,CAAA;AAClD,IAAA,MAAM,EAAE,aAAA,EAAc,GAAI,MAAM,QAAA,CAAS,YAAY,WAAW,CAAA;AAEhE,IAAA,MAAM,aAAA,GAAgB,MAAM,IAAA,CAAK,gBAAA,CAAiB;AAAA,MAChD;AAAA,KACD,CAAA;AAED,IAAA,GAAA,CAAI,IAAA,CAAK,EAAE,aAAA,EAAe,CAAA;AAAA,EAC5B,CAAC,CAAA;AAED,EAAA,MAAA,CAAO,GAAA,CAAI,QAAA,EAAU,OAAO,GAAA,EAAK,GAAA,KAAQ;AACvC,IAAA,MAAM,WAAA,GAAc,MAAM,QAAA,CAAS,WAAA,CAAY,GAAG,CAAA;AAElD,IAAA,MAAM,QAAQ,MAAM,IAAA,CAAK,iBAAA,CAAkB,EAAE,aAAa,CAAA;AAE1D,IAAA,GAAA,CAAI,IAAA,CAAK,EAAE,KAAA,EAAO,CAAA;AAAA,EACpB,CAAC,CAAA;AAED,EAAA,MAAA,CAAO,GAAA,CAAI,QAAQE,qBAAA,CAAW,UAAA,EAAY,QAAQ,CAAA,EAAG,OAAO,KAAK,GAAA,KAAQ;AACvE,IAAA,MAAM,EAAE,EAAA,EAAG,GAAI,GAAA,CAAI,MAAA;AAEnB,IAAA,MAAM,WAAA,GAAc,MAAM,QAAA,CAAS,WAAA,CAAY,GAAG,CAAA;AAClD,IAAA,MAAM,EAAE,aAAA,EAAc,GAAI,MAAM,QAAA,CAAS,YAAY,WAAW,CAAA;AAEhE,IAAA,MAAM,YAAA,GAAe,MAAM,IAAA,CAAK,eAAA,CAAgB;AAAA,MAC9C,cAAA,EAAgB,EAAA;AAAA,MAChB;AAAA,KACD,CAAA;AAED,IAAA,GAAA,CAAI,IAAA,CAAK,EAAE,YAAA,EAAc,CAAA;AAAA,EAC3B,CAAC,CAAA;AAED,EAAA,MAAA,CAAO,IAAA;AAAA,IACL,2BAAA;AAAA,IACAA,qBAAA;AAAA,MACEF,mBAAE,MAAA,CAAO;AAAA,QACP,SAAA,EAAWA,kBAAA,CAAE,MAAA,EAAO,CAAE,IAAA;AAAK,OAC5B,CAAA;AAAA,MACD;AAAA,KACF;AAAA,IACAE,qBAAA;AAAA,MACEF,mBAAE,MAAA,CAAO;AAAA,QACP,KAAA,EAAOA,mBAAE,MAAA;AAAO,OACjB,CAAA;AAAA,MACD;AAAA,KACF;AAAA,IACA,OAAO,KAAK,GAAA,KAAQ;AAClB,MAAA,MAAM,EAAE,SAAA,EAAU,GAAI,GAAA,CAAI,MAAA;AAC1B,MAAA,MAAM,EAAE,KAAA,EAAM,GAAI,GAAA,CAAI,IAAA;AAEtB,MAAA,MAAM,IAAA,CAAK,YAAA,CAAa,SAAA,EAAW,KAAK,CAAA;AAExC,MAAA,GAAA,CAAI,MAAA,CAAO,GAAG,CAAA,CAAE,GAAA,EAAI;AAAA,IACtB;AAAA,GACF;AAEA,EAAA,OAAO,MAAA;AACT;;;;"}
+{"version":3,"file":"chat.cjs.js","sources":["../../../src/services/router/chat.ts"],"sourcesContent":["import express from 'express';\nimport Router from 'express-promise-router';\nimport { ChatService } from '../chat';\nimport z from 'zod';\nimport { validation } from './middleware/validation';\nimport { v4 as uuid } from 'uuid';\nimport {\n  HttpAuthService,\n  UserInfoService,\n} from '@backstage/backend-plugin-api';\nimport { ConversationService } from '../conversation';\n\nexport type ChatRouterOptions = {\n  chat: ChatService;\n  httpAuth: HttpAuthService;\n  userInfo: UserInfoService;\n  conversation: ConversationService;\n};\n\nexport async function createChatRouter(\n  options: ChatRouterOptions,\n): Promise<express.Router> {\n  const { chat, httpAuth, userInfo, conversation } = options;\n\n  const router = Router();\n\n  const messageSchema = z.object({\n    messages: z.array(\n      z.object({\n        id: z.uuid().optional().default(uuid),\n        role: z.string(),\n        content: z.string(),\n      }),\n    ),\n    modelId: z.string(),\n    conversationId: z.uuid().optional().default(uuid),\n    stream: z.boolean().optional(),\n    tools: z\n      .array(\n        z.object({\n          name: z.string(),\n          provider: z.string(),\n        }),\n      )\n      .optional(),\n  });\n\n  router.post(\n    '/message',\n    validation(messageSchema, 'body'),\n    async (req, res) => {\n      const { messages, conversationId, modelId, stream, tools } = req.body;\n\n      const credentials = await httpAuth.credentials(req);\n\n      const responseMessages = await chat.prompt({\n        modelId,\n        messages,\n        conversationId,\n        stream,\n        credentials,\n        tools,\n      });\n\n      res.json({\n        messages: responseMessages,\n        conversationId,\n      });\n    },\n  );\n\n  const chatSchema = z.object({\n    id: z.uuid(),\n  });\n\n  router.get('/conversations', async (req, res) => {\n    const credentials = await httpAuth.credentials(req);\n    const { userEntityRef } = await userInfo.getUserInfo(credentials);\n\n    const conversations = await conversation.getConversations({\n      userEntityRef,\n    });\n\n    res.json({ conversations });\n  });\n\n  router.get('/:id', validation(chatSchema, 'params'), async (req, res) => {\n    const { id } = req.params;\n\n    const credentials = await httpAuth.credentials(req);\n    const { userEntityRef } = await userInfo.getUserInfo(credentials);\n\n    const conversationMessages = await conversation.getConversation({\n      conversationId: id,\n      userEntityRef,\n    });\n\n    res.json({ conversation: conversationMessages });\n  });\n\n  router.post(\n    '/message/:messageId/score',\n    validation(\n      z.object({\n        messageId: z.string().uuid(),\n      }),\n      'params',\n    ),\n    validation(\n      z.object({\n        score: z.number(),\n      }),\n      'body',\n    ),\n    async (req, res) => {\n      const { messageId } = req.params;\n      const { score } = req.body;\n\n      await conversation.scoreMessage(messageId, score);\n\n      res.status(204).end();\n    },\n  );\n\n  return router;\n}\n"],"names":["Router","z","uuid","validation"],"mappings":";;;;;;;;;;;;AAmBA,eAAsB,iBACpB,OAAA,EACyB;AACzB,EAAA,MAAM,EAAE,IAAA,EAAM,QAAA,EAAU,QAAA,EAAU,cAAa,GAAI,OAAA;AAEnD,EAAA,MAAM,SAASA,uBAAA,EAAO;AAEtB,EAAA,MAAM,aAAA,GAAgBC,mBAAE,MAAA,CAAO;AAAA,IAC7B,UAAUA,kBAAA,CAAE,KAAA;AAAA,MACVA,mBAAE,MAAA,CAAO;AAAA,QACP,IAAIA,kBAAA,CAAE,IAAA,GAAO,QAAA,EAAS,CAAE,QAAQC,OAAI,CAAA;AAAA,QACpC,IAAA,EAAMD,mBAAE,MAAA,EAAO;AAAA,QACf,OAAA,EAASA,mBAAE,MAAA;AAAO,OACnB;AAAA,KACH;AAAA,IACA,OAAA,EAASA,mBAAE,MAAA,EAAO;AAAA,IAClB,gBAAgBA,kBAAA,CAAE,IAAA,GAAO,QAAA,EAAS,CAAE,QAAQC,OAAI,CAAA;AAAA,IAChD,MAAA,EAAQD,kBAAA,CAAE,OAAA,EAAQ,CAAE,QAAA,EAAS;AAAA,IAC7B,OAAOA,kBAAA,CACJ,KAAA;AAAA,MACCA,mBAAE,MAAA,CAAO;AAAA,QACP,IAAA,EAAMA,mBAAE,MAAA,EAAO;AAAA,QACf,QAAA,EAAUA,mBAAE,MAAA;AAAO,OACpB;AAAA,MAEF,QAAA;AAAS,GACb,CAAA;AAED,EAAA,MAAA,CAAO,IAAA;AAAA,IACL,UAAA;AAAA,IACAE,qBAAA,CAAW,eAAe,MAAM,CAAA;AAAA,IAChC,OAAO,KAAK,GAAA,KAAQ;AAClB,MAAA,MAAM,EAAE,QAAA,EAAU,cAAA,EAAgB,SAAS,MAAA,EAAQ,KAAA,KAAU,GAAA,CAAI,IAAA;AAEjE,MAAA,MAAM,WAAA,GAAc,MAAM,QAAA,CAAS,WAAA,CAAY,GAAG,CAAA;AAElD,MAAA,MAAM,gBAAA,GAAmB,MAAM,IAAA,CAAK,MAAA,CAAO;AAAA,QACzC,OAAA;AAAA,QACA,QAAA;AAAA,QACA,cAAA;AAAA,QACA,MAAA;AAAA,QACA,WAAA;AAAA,QACA;AAAA,OACD,CAAA;AAED,MAAA,GAAA,CAAI,IAAA,CAAK;AAAA,QACP,QAAA,EAAU,gBAAA;AAAA,QACV;AAAA,OACD,CAAA;AAAA,IACH;AAAA,GACF;AAEA,EAAA,MAAM,UAAA,GAAaF,mBAAE,MAAA,CAAO;AAAA,IAC1B,EAAA,EAAIA,mBAAE,IAAA;AAAK,GACZ,CAAA;AAED,EAAA,MAAA,CAAO,GAAA,CAAI,gBAAA,EAAkB,OAAO,GAAA,EAAK,GAAA,KAAQ;AAC/C,IAAA,MAAM,WAAA,GAAc,MAAM,QAAA,CAAS,WAAA,CAAY,GAAG,CAAA;AAClD,IAAA,MAAM,EAAE,aAAA,EAAc,GAAI,MAAM,QAAA,CAAS,YAAY,WAAW,CAAA;AAEhE,IAAA,MAAM,aAAA,GAAgB,MAAM,YAAA,CAAa,gBAAA,CAAiB;AAAA,MACxD;AAAA,KACD,CAAA;AAED,IAAA,GAAA,CAAI,IAAA,CAAK,EAAE,aAAA,EAAe,CAAA;AAAA,EAC5B,CAAC,CAAA;AAED,EAAA,MAAA,CAAO,GAAA,CAAI,QAAQE,qBAAA,CAAW,UAAA,EAAY,QAAQ,CAAA,EAAG,OAAO,KAAK,GAAA,KAAQ;AACvE,IAAA,MAAM,EAAE,EAAA,EAAG,GAAI,GAAA,CAAI,MAAA;AAEnB,IAAA,MAAM,WAAA,GAAc,MAAM,QAAA,CAAS,WAAA,CAAY,GAAG,CAAA;AAClD,IAAA,MAAM,EAAE,aAAA,EAAc,GAAI,MAAM,QAAA,CAAS,YAAY,WAAW,CAAA;AAEhE,IAAA,MAAM,oBAAA,GAAuB,MAAM,YAAA,CAAa,eAAA,CAAgB;AAAA,MAC9D,cAAA,EAAgB,EAAA;AAAA,MAChB;AAAA,KACD,CAAA;AAED,IAAA,GAAA,CAAI,IAAA,CAAK,EAAE,YAAA,EAAc,oBAAA,EAAsB,CAAA;AAAA,EACjD,CAAC,CAAA;AAED,EAAA,MAAA,CAAO,IAAA;AAAA,IACL,2BAAA;AAAA,IACAA,qBAAA;AAAA,MACEF,mBAAE,MAAA,CAAO;AAAA,QACP,SAAA,EAAWA,kBAAA,CAAE,MAAA,EAAO,CAAE,IAAA;AAAK,OAC5B,CAAA;AAAA,MACD;AAAA,KACF;AAAA,IACAE,qBAAA;AAAA,MACEF,mBAAE,MAAA,CAAO;AAAA,QACP,KAAA,EAAOA,mBAAE,MAAA;AAAO,OACjB,CAAA;AAAA,MACD;AAAA,KACF;AAAA,IACA,OAAO,KAAK,GAAA,KAAQ;AAClB,MAAA,MAAM,EAAE,SAAA,EAAU,GAAI,GAAA,CAAI,MAAA;AAC1B,MAAA,MAAM,EAAE,KAAA,EAAM,GAAI,GAAA,CAAI,IAAA;AAEtB,MAAA,MAAM,YAAA,CAAa,YAAA,CAAa,SAAA,EAAW,KAAK,CAAA;AAEhD,MAAA,GAAA,CAAI,MAAA,CAAO,GAAG,CAAA,CAAE,GAAA,EAAI;AAAA,IACtB;AAAA,GACF;AAEA,EAAA,OAAO,MAAA;AACT;;;;"}

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

@@ -7,6 +7,8 @@ var models = require('./models.cjs.js');
 var rootHttpRouter = require('@backstage/backend-defaults/rootHttpRouter');
 var summary = require('./summary.cjs.js');
 var index = require('./settings/index.cjs.js');
+var agent = require('./agent.cjs.js');
+var tools = require('./tools.cjs.js');
 
 function _interopDefaultCompat (e) { return e && typeof e === 'object' && 'default' in e ? e : { default: e }; }
 
@@ -20,6 +22,8 @@ async function createRouter(options) {
   router.use("/models", await models.createModelRouter(options));
   router.use("/summary", await summary.createSummaryRouter(options));
   router.use("/settings", await index.createSettingsRouter(options));
+  router.use("/agent", await agent.createAgentRouter(options));
+  router.use("/tools", await tools.createToolRouter(options));
   const middleware = rootHttpRouter.MiddlewareFactory.create(options);
   router.use(middleware.error());
   return router;

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

@@ -1 +1 @@
-{"version":3,"file":"index.cjs.js","sources":["../../../src/services/router/index.ts"],"sourcesContent":["import express from 'express';\nimport Router from 'express-promise-router';\nimport { createChatRouter, ChatRouterOptions } from './chat';\nimport { createModelRouter, ModelRouterOptions } from './models';\nimport {\n  LoggerService,\n  RootConfigService,\n} from '@backstage/backend-plugin-api';\nimport { MiddlewareFactory } from '@backstage/backend-defaults/rootHttpRouter';\nimport { SummaryRouterOptions, createSummaryRouter } from './summary';\nimport { createSettingsRouter, SettingsRouterOptions } from './settings';\n\nexport type RouterOptions = ChatRouterOptions &\n  SummaryRouterOptions &\n  ModelRouterOptions &\n  SettingsRouterOptions & {\n    config: RootConfigService;\n    logger: LoggerService;\n  };\n\nexport async function createRouter(\n  options: RouterOptions,\n): Promise<express.Router> {\n  const router = Router();\n  router.use(express.json());\n\n  router.use('/chat', await createChatRouter(options));\n  router.use('/models', await createModelRouter(options));\n  router.use('/summary', await createSummaryRouter(options));\n  router.use('/settings', await createSettingsRouter(options));\n\n  const middleware = MiddlewareFactory.create(options);\n\n  router.use(middleware.error());\n\n  return router;\n}\n"],"names":["Router","express","createChatRouter","createModelRouter","createSummaryRouter","createSettingsRouter","MiddlewareFactory"],"mappings":";;;;;;;;;;;;;;;AAoBA,eAAsB,aACpB,OAAA,EACyB;AACzB,EAAA,MAAM,SAASA,uBAAA,EAAO;AACtB,EAAA,MAAA,CAAO,GAAA,CAAIC,wBAAA,CAAQ,IAAA,EAAM,CAAA;AAEzB,EAAA,MAAA,CAAO,GAAA,CAAI,OAAA,EAAS,MAAMC,qBAAA,CAAiB,OAAO,CAAC,CAAA;AACnD,EAAA,MAAA,CAAO,GAAA,CAAI,SAAA,EAAW,MAAMC,wBAAA,CAAkB,OAAO,CAAC,CAAA;AACtD,EAAA,MAAA,CAAO,GAAA,CAAI,UAAA,EAAY,MAAMC,2BAAA,CAAoB,OAAO,CAAC,CAAA;AACzD,EAAA,MAAA,CAAO,GAAA,CAAI,WAAA,EAAa,MAAMC,0BAAA,CAAqB,OAAO,CAAC,CAAA;AAE3D,EAAA,MAAM,UAAA,GAAaC,gCAAA,CAAkB,MAAA,CAAO,OAAO,CAAA;AAEnD,EAAA,MAAA,CAAO,GAAA,CAAI,UAAA,CAAW,KAAA,EAAO,CAAA;AAE7B,EAAA,OAAO,MAAA;AACT;;;;"}
+{"version":3,"file":"index.cjs.js","sources":["../../../src/services/router/index.ts"],"sourcesContent":["import express from 'express';\nimport Router from 'express-promise-router';\nimport { createChatRouter, ChatRouterOptions } from './chat';\nimport { createModelRouter, ModelRouterOptions } from './models';\nimport {\n  LoggerService,\n  RootConfigService,\n} from '@backstage/backend-plugin-api';\nimport { MiddlewareFactory } from '@backstage/backend-defaults/rootHttpRouter';\nimport { SummaryRouterOptions, createSummaryRouter } from './summary';\nimport { createSettingsRouter, SettingsRouterOptions } from './settings';\nimport { createAgentRouter, AgentRouterOptions } from './agent';\nimport { createToolRouter, ToolRouterOptions } from './tools';\n\nexport type RouterOptions = ChatRouterOptions &\n  SummaryRouterOptions &\n  ModelRouterOptions &\n  SettingsRouterOptions &\n  AgentRouterOptions &\n  ToolRouterOptions & {\n    config: RootConfigService;\n    logger: LoggerService;\n  };\n\nexport async function createRouter(\n  options: RouterOptions,\n): Promise<express.Router> {\n  const router = Router();\n  router.use(express.json());\n\n  router.use('/chat', await createChatRouter(options));\n  router.use('/models', await createModelRouter(options));\n  router.use('/summary', await createSummaryRouter(options));\n  router.use('/settings', await createSettingsRouter(options));\n  router.use('/agent', await createAgentRouter(options));\n  router.use('/tools', await createToolRouter(options));\n\n  const middleware = MiddlewareFactory.create(options);\n\n  router.use(middleware.error());\n\n  return router;\n}\n"],"names":["Router","express","createChatRouter","createModelRouter","createSummaryRouter","createSettingsRouter","createAgentRouter","createToolRouter","MiddlewareFactory"],"mappings":";;;;;;;;;;;;;;;;;AAwBA,eAAsB,aACpB,OAAA,EACyB;AACzB,EAAA,MAAM,SAASA,uBAAA,EAAO;AACtB,EAAA,MAAA,CAAO,GAAA,CAAIC,wBAAA,CAAQ,IAAA,EAAM,CAAA;AAEzB,EAAA,MAAA,CAAO,GAAA,CAAI,OAAA,EAAS,MAAMC,qBAAA,CAAiB,OAAO,CAAC,CAAA;AACnD,EAAA,MAAA,CAAO,GAAA,CAAI,SAAA,EAAW,MAAMC,wBAAA,CAAkB,OAAO,CAAC,CAAA;AACtD,EAAA,MAAA,CAAO,GAAA,CAAI,UAAA,EAAY,MAAMC,2BAAA,CAAoB,OAAO,CAAC,CAAA;AACzD,EAAA,MAAA,CAAO,GAAA,CAAI,WAAA,EAAa,MAAMC,0BAAA,CAAqB,OAAO,CAAC,CAAA;AAC3D,EAAA,MAAA,CAAO,GAAA,CAAI,QAAA,EAAU,MAAMC,uBAAA,CAAkB,OAAO,CAAC,CAAA;AACrD,EAAA,MAAA,CAAO,GAAA,CAAI,QAAA,EAAU,MAAMC,sBAAA,CAAiB,OAAO,CAAC,CAAA;AAEpD,EAAA,MAAM,UAAA,GAAaC,gCAAA,CAAkB,MAAA,CAAO,OAAO,CAAA;AAEnD,EAAA,MAAA,CAAO,GAAA,CAAI,UAAA,CAAW,KAAA,EAAO,CAAA;AAE7B,EAAA,OAAO,MAAA;AACT;;;;"}

package/dist/services/router/models.cjs.js

@@ -7,10 +7,10 @@ function _interopDefaultCompat (e) { return e && typeof e === 'object' && 'defau
 var Router__default = /*#__PURE__*/_interopDefaultCompat(Router);
 
 async function createModelRouter(options) {
-  const { chat } = options;
+  const { model } = options;
   const router = Router__default.default();
   router.get("/", async (_req, res) => {
-    const models = await chat.getAvailableModels();
+    const models = await model.getAvailableModels();
     res.json({ models });
   });
   return router;

package/dist/services/router/models.cjs.js.map

@@ -1 +1 @@
-{"version":3,"file":"models.cjs.js","sources":["../../../src/services/router/models.ts"],"sourcesContent":["import express from 'express';\nimport Router from 'express-promise-router';\nimport { ChatService } from '../chat';\n\nexport type ModelRouterOptions = {\n  chat: ChatService;\n};\n\nexport async function createModelRouter(\n  options: ModelRouterOptions,\n): Promise<express.Router> {\n  const { chat } = options;\n  const router = Router();\n\n  router.get('/', async (_req, res) => {\n    const models = await chat.getAvailableModels();\n    res.json({ models });\n  });\n\n  return router;\n}\n"],"names":["Router"],"mappings":";;;;;;;;AAQA,eAAsB,kBACpB,OAAA,EACyB;AACzB,EAAA,MAAM,EAAE,MAAK,GAAI,OAAA;AACjB,EAAA,MAAM,SAASA,uBAAA,EAAO;AAEtB,EAAA,MAAA,CAAO,GAAA,CAAI,GAAA,EAAK,OAAO,IAAA,EAAM,GAAA,KAAQ;AACnC,IAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,kBAAA,EAAmB;AAC7C,IAAA,GAAA,CAAI,IAAA,CAAK,EAAE,MAAA,EAAQ,CAAA;AAAA,EACrB,CAAC,CAAA;AAED,EAAA,OAAO,MAAA;AACT;;;;"}
+{"version":3,"file":"models.cjs.js","sources":["../../../src/services/router/models.ts"],"sourcesContent":["import express from 'express';\nimport Router from 'express-promise-router';\nimport { ModelService } from '../model';\n\nexport type ModelRouterOptions = {\n  model: ModelService;\n};\n\nexport async function createModelRouter(\n  options: ModelRouterOptions,\n): Promise<express.Router> {\n  const { model } = options;\n  const router = Router();\n\n  router.get('/', async (_req, res) => {\n    const models = await model.getAvailableModels();\n    res.json({ models });\n  });\n\n  return router;\n}\n"],"names":["Router"],"mappings":";;;;;;;;AAQA,eAAsB,kBACpB,OAAA,EACyB;AACzB,EAAA,MAAM,EAAE,OAAM,GAAI,OAAA;AAClB,EAAA,MAAM,SAASA,uBAAA,EAAO;AAEtB,EAAA,MAAA,CAAO,GAAA,CAAI,GAAA,EAAK,OAAO,IAAA,EAAM,GAAA,KAAQ;AACnC,IAAA,MAAM,MAAA,GAAS,MAAM,KAAA,CAAM,kBAAA,EAAmB;AAC9C,IAAA,GAAA,CAAI,IAAA,CAAK,EAAE,MAAA,EAAQ,CAAA;AAAA,EACrB,CAAC,CAAA;AAED,EAAA,OAAO,MAAA;AACT;;;;"}

package/dist/services/router/settings/index.cjs.js

@@ -2,7 +2,6 @@
 
 var express = require('express');
 var Router = require('express-promise-router');
-var rootHttpRouter = require('@backstage/backend-defaults/rootHttpRouter');
 var mcp = require('./mcp.cjs.js');
 var z = require('zod');
 var validation = require('../middleware/validation.cjs.js');
@@ -42,8 +41,6 @@ async function createSettingsRouter(options) {
     await userSettings.setSettingsForType(credentials, type, settings);
     res.status(204).send();
   });
-  const middleware = rootHttpRouter.MiddlewareFactory.create(options);
-  router.use(middleware.error());
   return router;
 }
 

package/dist/services/router/settings/index.cjs.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs.js","sources":["../../../../src/services/router/settings/index.ts"],"sourcesContent":["import express from 'express';\nimport Router from 'express-promise-router';\nimport {\n  LoggerService,\n  RootConfigService,\n} from '@backstage/backend-plugin-api';\nimport { MiddlewareFactory } from '@backstage/backend-defaults/rootHttpRouter';\nimport { createMcpRouter, McpRouterOptions } from './mcp';\nimport { UserSettingsService } from '../../user-settings';\nimport z from 'zod';\nimport { validation } from '../middleware/validation';\n\nexport type SettingsRouterOptions = McpRouterOptions & {\n  config: RootConfigService;\n  logger: LoggerService;\n  userSettings: UserSettingsService;\n};\n\nexport async function createSettingsRouter(\n  options: SettingsRouterOptions,\n): Promise<express.Router> {\n  const { userSettings, httpAuth } = options;\n  const router = Router();\n  router.use(express.json());\n\n  router.use('/mcp', await createMcpRouter(options));\n\n  const settingsSchema = z.object({\n    type: z.string(),\n  });\n\n  router.get('/', validation(settingsSchema, 'query'), async (req, res) => {\n    const credentials = await httpAuth.credentials(req);\n\n    const { type } = req.query;\n\n    const settings = await userSettings.getSettingsForType(\n      credentials,\n      type as string,\n    );\n\n    res.json({\n      settings,\n    });\n  });\n\n  const setSettingsSchema = z.object({\n    type: z.string(),\n    settings: z.any(),\n  });\n\n  router.patch('/', validation(setSettingsSchema, 'body'), async (req, res) => {\n    const credentials = await httpAuth.credentials(req);\n    const { type, settings } = req.body;\n\n    await userSettings.setSettingsForType(credentials, type, settings);\n    res.status(204).send();\n  });\n\n  const middleware = MiddlewareFactory.create(options);\n\n  router.use(middleware.error());\n\n  return router;\n}\n"],"names":["Router","express","createMcpRouter","z","validation","MiddlewareFactory"],"mappings":";;;;;;;;;;;;;;;AAkBA,eAAsB,qBACpB,OAAA,EACyB;AACzB,EAAA,MAAM,EAAE,YAAA,EAAc,QAAA,EAAS,GAAI,OAAA;AACnC,EAAA,MAAM,SAASA,uBAAA,EAAO;AACtB,EAAA,MAAA,CAAO,GAAA,CAAIC,wBAAA,CAAQ,IAAA,EAAM,CAAA;AAEzB,EAAA,MAAA,CAAO,GAAA,CAAI,MAAA,EAAQ,MAAMC,mBAAA,CAAgB,OAAO,CAAC,CAAA;AAEjD,EAAA,MAAM,cAAA,GAAiBC,mBAAE,MAAA,CAAO;AAAA,IAC9B,IAAA,EAAMA,mBAAE,MAAA;AAAO,GAChB,CAAA;AAED,EAAA,MAAA,CAAO,GAAA,CAAI,KAAKC,qBAAA,CAAW,cAAA,EAAgB,OAAO,CAAA,EAAG,OAAO,KAAK,GAAA,KAAQ;AACvE,IAAA,MAAM,WAAA,GAAc,MAAM,QAAA,CAAS,WAAA,CAAY,GAAG,CAAA;AAElD,IAAA,MAAM,EAAE,IAAA,EAAK,GAAI,GAAA,CAAI,KAAA;AAErB,IAAA,MAAM,QAAA,GAAW,MAAM,YAAA,CAAa,kBAAA;AAAA,MAClC,WAAA;AAAA,MACA;AAAA,KACF;AAEA,IAAA,GAAA,CAAI,IAAA,CAAK;AAAA,MACP;AAAA,KACD,CAAA;AAAA,EACH,CAAC,CAAA;AAED,EAAA,MAAM,iBAAA,GAAoBD,mBAAE,MAAA,CAAO;AAAA,IACjC,IAAA,EAAMA,mBAAE,MAAA,EAAO;AAAA,IACf,QAAA,EAAUA,mBAAE,GAAA;AAAI,GACjB,CAAA;AAED,EAAA,MAAA,CAAO,KAAA,CAAM,KAAKC,qBAAA,CAAW,iBAAA,EAAmB,MAAM,CAAA,EAAG,OAAO,KAAK,GAAA,KAAQ;AAC3E,IAAA,MAAM,WAAA,GAAc,MAAM,QAAA,CAAS,WAAA,CAAY,GAAG,CAAA;AAClD,IAAA,MAAM,EAAE,IAAA,EAAM,QAAA,EAAS,GAAI,GAAA,CAAI,IAAA;AAE/B,IAAA,MAAM,YAAA,CAAa,kBAAA,CAAmB,WAAA,EAAa,IAAA,EAAM,QAAQ,CAAA;AACjE,IAAA,GAAA,CAAI,MAAA,CAAO,GAAG,CAAA,CAAE,IAAA,EAAK;AAAA,EACvB,CAAC,CAAA;AAED,EAAA,MAAM,UAAA,GAAaC,gCAAA,CAAkB,MAAA,CAAO,OAAO,CAAA;AAEnD,EAAA,MAAA,CAAO,GAAA,CAAI,UAAA,CAAW,KAAA,EAAO,CAAA;AAE7B,EAAA,OAAO,MAAA;AACT;;;;"}
+{"version":3,"file":"index.cjs.js","sources":["../../../../src/services/router/settings/index.ts"],"sourcesContent":["import express from 'express';\nimport Router from 'express-promise-router';\n\nimport { createMcpRouter, McpRouterOptions } from './mcp';\nimport { UserSettingsService } from '../../user-settings';\nimport z from 'zod';\nimport { validation } from '../middleware/validation';\n\nexport type SettingsRouterOptions = McpRouterOptions & {\n  userSettings: UserSettingsService;\n};\n\nexport async function createSettingsRouter(\n  options: SettingsRouterOptions,\n): Promise<express.Router> {\n  const { userSettings, httpAuth } = options;\n  const router = Router();\n  router.use(express.json());\n\n  router.use('/mcp', await createMcpRouter(options));\n\n  const settingsSchema = z.object({\n    type: z.string(),\n  });\n\n  router.get('/', validation(settingsSchema, 'query'), async (req, res) => {\n    const credentials = await httpAuth.credentials(req);\n\n    const { type } = req.query;\n\n    const settings = await userSettings.getSettingsForType(\n      credentials,\n      type as string,\n    );\n\n    res.json({\n      settings,\n    });\n  });\n\n  const setSettingsSchema = z.object({\n    type: z.string(),\n    settings: z.any(),\n  });\n\n  router.patch('/', validation(setSettingsSchema, 'body'), async (req, res) => {\n    const credentials = await httpAuth.credentials(req);\n    const { type, settings } = req.body;\n\n    await userSettings.setSettingsForType(credentials, type, settings);\n    res.status(204).send();\n  });\n\n  return router;\n}\n"],"names":["Router","express","createMcpRouter","z","validation"],"mappings":";;;;;;;;;;;;;;AAYA,eAAsB,qBACpB,OAAA,EACyB;AACzB,EAAA,MAAM,EAAE,YAAA,EAAc,QAAA,EAAS,GAAI,OAAA;AACnC,EAAA,MAAM,SAASA,uBAAA,EAAO;AACtB,EAAA,MAAA,CAAO,GAAA,CAAIC,wBAAA,CAAQ,IAAA,EAAM,CAAA;AAEzB,EAAA,MAAA,CAAO,GAAA,CAAI,MAAA,EAAQ,MAAMC,mBAAA,CAAgB,OAAO,CAAC,CAAA;AAEjD,EAAA,MAAM,cAAA,GAAiBC,mBAAE,MAAA,CAAO;AAAA,IAC9B,IAAA,EAAMA,mBAAE,MAAA;AAAO,GAChB,CAAA;AAED,EAAA,MAAA,CAAO,GAAA,CAAI,KAAKC,qBAAA,CAAW,cAAA,EAAgB,OAAO,CAAA,EAAG,OAAO,KAAK,GAAA,KAAQ;AACvE,IAAA,MAAM,WAAA,GAAc,MAAM,QAAA,CAAS,WAAA,CAAY,GAAG,CAAA;AAElD,IAAA,MAAM,EAAE,IAAA,EAAK,GAAI,GAAA,CAAI,KAAA;AAErB,IAAA,MAAM,QAAA,GAAW,MAAM,YAAA,CAAa,kBAAA;AAAA,MAClC,WAAA;AAAA,MACA;AAAA,KACF;AAEA,IAAA,GAAA,CAAI,IAAA,CAAK;AAAA,MACP;AAAA,KACD,CAAA;AAAA,EACH,CAAC,CAAA;AAED,EAAA,MAAM,iBAAA,GAAoBD,mBAAE,MAAA,CAAO;AAAA,IACjC,IAAA,EAAMA,mBAAE,MAAA,EAAO;AAAA,IACf,QAAA,EAAUA,mBAAE,GAAA;AAAI,GACjB,CAAA;AAED,EAAA,MAAA,CAAO,KAAA,CAAM,KAAKC,qBAAA,CAAW,iBAAA,EAAmB,MAAM,CAAA,EAAG,OAAO,KAAK,GAAA,KAAQ;AAC3E,IAAA,MAAM,WAAA,GAAc,MAAM,QAAA,CAAS,WAAA,CAAY,GAAG,CAAA;AAClD,IAAA,MAAM,EAAE,IAAA,EAAM,QAAA,EAAS,GAAI,GAAA,CAAI,IAAA;AAE/B,IAAA,MAAM,YAAA,CAAa,kBAAA,CAAmB,WAAA,EAAa,IAAA,EAAM,QAAQ,CAAA;AACjE,IAAA,GAAA,CAAI,MAAA,CAAO,GAAG,CAAA,CAAE,IAAA,EAAK;AAAA,EACvB,CAAC,CAAA;AAED,EAAA,OAAO,MAAA;AACT;;;;"}

package/dist/services/router/summary.cjs.js

@@ -14,16 +14,18 @@ async function createSummaryRouter(options) {
   const router = Router__default.default();
   const contentSchema = z__default.default.object({
     content: z__default.default.string(),
-    length: z__default.default.string().optional()
+    length: z__default.default.string().optional(),
+    prompt: z__default.default.string().optional()
   });
   router.post(
     "/content",
     validation.validation(contentSchema, "body"),
     async (req, res) => {
-      const { content, length } = req.body;
+      const { content, length, prompt } = req.body;
       const summary = await summarizer.summarize({
         content,
-        length
+        length,
+        prompt
       });
       res.json({ summary });
     }

package/dist/services/router/summary.cjs.js.map

@@ -1 +1 @@
-{"version":3,"file":"summary.cjs.js","sources":["../../../src/services/router/summary.ts"],"sourcesContent":["import express from 'express';\nimport Router from 'express-promise-router';\nimport { SummarizerService } from '../summarizer';\nimport { validation } from './middleware/validation';\nimport z from 'zod';\n\nexport type SummaryRouterOptions = {\n  summarizer: SummarizerService;\n};\n\nexport async function createSummaryRouter(\n  options: SummaryRouterOptions,\n): Promise<express.Router> {\n  const { summarizer } = options;\n\n  const router = Router();\n\n  const contentSchema = z.object({\n    content: z.string(),\n    length: z.string().optional(),\n  });\n\n  router.post(\n    '/content',\n    validation(contentSchema, 'body'),\n    async (req, res) => {\n      const { content, length } = req.body;\n\n      const summary = await summarizer.summarize({\n        content,\n        length,\n      });\n      res.json({ summary });\n    },\n  );\n\n  const conversationSchema = z.object({\n    messages: z.array(\n      z.object({\n        role: z.string(),\n        content: z.string(),\n      }),\n    ),\n    length: z.string().optional(),\n  });\n\n  router.post(\n    '/conversation',\n    validation(conversationSchema, 'body'),\n    async (req, res) => {\n      const { messages, length } = req.body;\n      const summary = await summarizer.summarizeConversation({\n        messages,\n        length,\n      });\n      res.json({ summary });\n    },\n  );\n\n  return router;\n}\n"],"names":["Router","z","validation"],"mappings":";;;;;;;;;;;AAUA,eAAsB,oBACpB,OAAA,EACyB;AACzB,EAAA,MAAM,EAAE,YAAW,GAAI,OAAA;AAEvB,EAAA,MAAM,SAASA,uBAAA,EAAO;AAEtB,EAAA,MAAM,aAAA,GAAgBC,mBAAE,MAAA,CAAO;AAAA,IAC7B,OAAA,EAASA,mBAAE,MAAA,EAAO;AAAA,IAClB,MAAA,EAAQA,kBAAA,CAAE,MAAA,EAAO,CAAE,QAAA;AAAS,GAC7B,CAAA;AAED,EAAA,MAAA,CAAO,IAAA;AAAA,IACL,UAAA;AAAA,IACAC,qBAAA,CAAW,eAAe,MAAM,CAAA;AAAA,IAChC,OAAO,KAAK,GAAA,KAAQ;AAClB,MAAA,MAAM,EAAE,OAAA,EAAS,MAAA,EAAO,GAAI,GAAA,CAAI,IAAA;AAEhC,MAAA,MAAM,OAAA,GAAU,MAAM,UAAA,CAAW,SAAA,CAAU;AAAA,QACzC,OAAA;AAAA,QACA;AAAA,OACD,CAAA;AACD,MAAA,GAAA,CAAI,IAAA,CAAK,EAAE,OAAA,EAAS,CAAA;AAAA,IACtB;AAAA,GACF;AAEA,EAAA,MAAM,kBAAA,GAAqBD,mBAAE,MAAA,CAAO;AAAA,IAClC,UAAUA,kBAAA,CAAE,KAAA;AAAA,MACVA,mBAAE,MAAA,CAAO;AAAA,QACP,IAAA,EAAMA,mBAAE,MAAA,EAAO;AAAA,QACf,OAAA,EAASA,mBAAE,MAAA;AAAO,OACnB;AAAA,KACH;AAAA,IACA,MAAA,EAAQA,kBAAA,CAAE,MAAA,EAAO,CAAE,QAAA;AAAS,GAC7B,CAAA;AAED,EAAA,MAAA,CAAO,IAAA;AAAA,IACL,eAAA;AAAA,IACAC,qBAAA,CAAW,oBAAoB,MAAM,CAAA;AAAA,IACrC,OAAO,KAAK,GAAA,KAAQ;AAClB,MAAA,MAAM,EAAE,QAAA,EAAU,MAAA,EAAO,GAAI,GAAA,CAAI,IAAA;AACjC,MAAA,MAAM,OAAA,GAAU,MAAM,UAAA,CAAW,qBAAA,CAAsB;AAAA,QACrD,QAAA;AAAA,QACA;AAAA,OACD,CAAA;AACD,MAAA,GAAA,CAAI,IAAA,CAAK,EAAE,OAAA,EAAS,CAAA;AAAA,IACtB;AAAA,GACF;AAEA,EAAA,OAAO,MAAA;AACT;;;;"}
+{"version":3,"file":"summary.cjs.js","sources":["../../../src/services/router/summary.ts"],"sourcesContent":["import express from 'express';\nimport Router from 'express-promise-router';\nimport { SummarizerService } from '../summarizer';\nimport { validation } from './middleware/validation';\nimport z from 'zod';\n\nexport type SummaryRouterOptions = {\n  summarizer: SummarizerService;\n};\n\nexport async function createSummaryRouter(\n  options: SummaryRouterOptions,\n): Promise<express.Router> {\n  const { summarizer } = options;\n\n  const router = Router();\n\n  const contentSchema = z.object({\n    content: z.string(),\n    length: z.string().optional(),\n    prompt: z.string().optional(),\n  });\n\n  router.post(\n    '/content',\n    validation(contentSchema, 'body'),\n    async (req, res) => {\n      const { content, length, prompt } = req.body;\n\n      const summary = await summarizer.summarize({\n        content,\n        length,\n        prompt,\n      });\n      res.json({ summary });\n    },\n  );\n\n  const conversationSchema = z.object({\n    messages: z.array(\n      z.object({\n        role: z.string(),\n        content: z.string(),\n      }),\n    ),\n    length: z.string().optional(),\n  });\n\n  router.post(\n    '/conversation',\n    validation(conversationSchema, 'body'),\n    async (req, res) => {\n      const { messages, length } = req.body;\n      const summary = await summarizer.summarizeConversation({\n        messages,\n        length,\n      });\n      res.json({ summary });\n    },\n  );\n\n  return router;\n}\n"],"names":["Router","z","validation"],"mappings":";;;;;;;;;;;AAUA,eAAsB,oBACpB,OAAA,EACyB;AACzB,EAAA,MAAM,EAAE,YAAW,GAAI,OAAA;AAEvB,EAAA,MAAM,SAASA,uBAAA,EAAO;AAEtB,EAAA,MAAM,aAAA,GAAgBC,mBAAE,MAAA,CAAO;AAAA,IAC7B,OAAA,EAASA,mBAAE,MAAA,EAAO;AAAA,IAClB,MAAA,EAAQA,kBAAA,CAAE,MAAA,EAAO,CAAE,QAAA,EAAS;AAAA,IAC5B,MAAA,EAAQA,kBAAA,CAAE,MAAA,EAAO,CAAE,QAAA;AAAS,GAC7B,CAAA;AAED,EAAA,MAAA,CAAO,IAAA;AAAA,IACL,UAAA;AAAA,IACAC,qBAAA,CAAW,eAAe,MAAM,CAAA;AAAA,IAChC,OAAO,KAAK,GAAA,KAAQ;AAClB,MAAA,MAAM,EAAE,OAAA,EAAS,MAAA,EAAQ,MAAA,KAAW,GAAA,CAAI,IAAA;AAExC,MAAA,MAAM,OAAA,GAAU,MAAM,UAAA,CAAW,SAAA,CAAU;AAAA,QACzC,OAAA;AAAA,QACA,MAAA;AAAA,QACA;AAAA,OACD,CAAA;AACD,MAAA,GAAA,CAAI,IAAA,CAAK,EAAE,OAAA,EAAS,CAAA;AAAA,IACtB;AAAA,GACF;AAEA,EAAA,MAAM,kBAAA,GAAqBD,mBAAE,MAAA,CAAO;AAAA,IAClC,UAAUA,kBAAA,CAAE,KAAA;AAAA,MACVA,mBAAE,MAAA,CAAO;AAAA,QACP,IAAA,EAAMA,mBAAE,MAAA,EAAO;AAAA,QACf,OAAA,EAASA,mBAAE,MAAA;AAAO,OACnB;AAAA,KACH;AAAA,IACA,MAAA,EAAQA,kBAAA,CAAE,MAAA,EAAO,CAAE,QAAA;AAAS,GAC7B,CAAA;AAED,EAAA,MAAA,CAAO,IAAA;AAAA,IACL,eAAA;AAAA,IACAC,qBAAA,CAAW,oBAAoB,MAAM,CAAA;AAAA,IACrC,OAAO,KAAK,GAAA,KAAQ;AAClB,MAAA,MAAM,EAAE,QAAA,EAAU,MAAA,EAAO,GAAI,GAAA,CAAI,IAAA;AACjC,MAAA,MAAM,OAAA,GAAU,MAAM,UAAA,CAAW,qBAAA,CAAsB;AAAA,QACrD,QAAA;AAAA,QACA;AAAA,OACD,CAAA;AACD,MAAA,GAAA,CAAI,IAAA,CAAK,EAAE,OAAA,EAAS,CAAA;AAAA,IACtB;AAAA,GACF;AAEA,EAAA,OAAO,MAAA;AACT;;;;"}

package/dist/services/router/tools.cjs.js

@@ -0,0 +1,21 @@
+'use strict';
+
+var Router = require('express-promise-router');
+
+function _interopDefaultCompat (e) { return e && typeof e === 'object' && 'default' in e ? e : { default: e }; }
+
+var Router__default = /*#__PURE__*/_interopDefaultCompat(Router);
+
+async function createToolRouter(options) {
+  const { httpAuth, tool } = options;
+  const router = Router__default.default();
+  router.get("/", async (req, res) => {
+    const credentials = await httpAuth.credentials(req);
+    const tools = await tool.getAvailableUserTools({ credentials });
+    res.json({ tools });
+  });
+  return router;
+}
+
+exports.createToolRouter = createToolRouter;
+//# sourceMappingURL=tools.cjs.js.map

package/dist/services/router/tools.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.cjs.js","sources":["../../../src/services/router/tools.ts"],"sourcesContent":["import express from 'express';\nimport Router from 'express-promise-router';\nimport { HttpAuthService } from '@backstage/backend-plugin-api';\nimport { ToolsService } from '../tools';\n\nexport type ToolRouterOptions = {\n  httpAuth: HttpAuthService;\n  tool: ToolsService;\n};\n\nexport async function createToolRouter(\n  options: ToolRouterOptions,\n): Promise<express.Router> {\n  const { httpAuth, tool } = options;\n  const router = Router();\n\n  router.get('/', async (req, res) => {\n    const credentials = await httpAuth.credentials(req);\n\n    const tools = await tool.getAvailableUserTools({ credentials });\n\n    res.json({ tools });\n  });\n\n  return router;\n}\n"],"names":["Router"],"mappings":";;;;;;;;AAUA,eAAsB,iBACpB,OAAA,EACyB;AACzB,EAAA,MAAM,EAAE,QAAA,EAAU,IAAA,EAAK,GAAI,OAAA;AAC3B,EAAA,MAAM,SAASA,uBAAA,EAAO;AAEtB,EAAA,MAAA,CAAO,GAAA,CAAI,GAAA,EAAK,OAAO,GAAA,EAAK,GAAA,KAAQ;AAClC,IAAA,MAAM,WAAA,GAAc,MAAM,QAAA,CAAS,WAAA,CAAY,GAAG,CAAA;AAElD,IAAA,MAAM,QAAQ,MAAM,IAAA,CAAK,qBAAA,CAAsB,EAAE,aAAa,CAAA;AAE9D,IAAA,GAAA,CAAI,IAAA,CAAK,EAAE,KAAA,EAAO,CAAA;AAAA,EACpB,CAAC,CAAA;AAED,EAAA,OAAO,MAAA;AACT;;;;"}

package/dist/services/summarizer.cjs.js

@@ -1,51 +1,56 @@
 'use strict';
 
+var backendPluginApi = require('@backstage/backend-plugin-api');
+var index = require('./agent/index.cjs.js');
 var prompts = require('../constants/prompts.cjs.js');
 var prompts$1 = require('@langchain/core/prompts');
+var uuid = require('uuid');
 
 const createSummarizerService = async ({
   config,
-  models,
-  callback
+  agent,
+  auth
 }) => {
-  const summaryModelId = config.getOptionalString("aiAssistant.conversation.summaryModel") ?? models[0].id;
+  const summaryModelId = config.getOptionalString(
+    "aiAssistant.conversation.summaryModel"
+  );
   const conversationSummaryPrompt = config.getOptionalString("aiAssistant.conversation.summaryPrompt") ?? prompts.DEFAULT_CONVERSATION_SUMMARY_PROMPT;
-  const model = models.find((m) => m.id === summaryModelId);
-  if (!model) {
-    throw new Error(`Summary model with id ${summaryModelId} not found`);
-  }
-  const llm = model.chatModel;
   const summaryPromptTemplate = prompts$1.PromptTemplate.fromTemplate(`
     PURPOSE:
     {summaryPrompt}
 
     Summarize the following content in {length}.
-
-    Content:
-    {content}
   `);
-  const summarize = async (content, summaryPrompt = prompts.DEFAULT_SUMMARY_PROMPT, length = "as few words as possible") => {
+  const summarize = async ({
+    content,
+    length = "as few words as possible",
+    prompt: summaryPrompt = prompts.DEFAULT_SUMMARY_PROMPT
+  }) => {
     const prompt = await summaryPromptTemplate.format({
       summaryPrompt,
       content,
       length
     });
-    const { callbacks } = await callback.getChainCallbacks({
-      conversationId: "summarizer",
-      userId: "system",
-      modelId: summaryModelId
-    });
-    const { metadata } = await callback.getChainMetadata({
-      conversationId: "summarizer",
-      userId: "system",
-      modelId: summaryModelId
-    });
-    const { text } = await llm.invoke(prompt, {
-      callbacks,
-      runName: "summarizer",
-      metadata
+    const credentials = await auth.getOwnServiceCredentials();
+    const messages = await agent.prompt({
+      messages: [{ role: "system", content, score: 0, metadata: {} }],
+      metadata: {
+        conversationId: "summarizer",
+        userId: "system:summarizer",
+        runId: uuid.v4(),
+        runName: "summarizer"
+      },
+      systemPrompt: prompt,
+      modelId: summaryModelId,
+      credentials,
+      tools: []
     });
-    return text.trim();
+    const aiMessages = messages.filter((m) => m.role === "ai");
+    if (aiMessages.length === 0) {
+      throw new Error("Failed to summarize content");
+    }
+    const response = messages.pop();
+    return response.content.trim();
   };
   const summarizeConversation = async ({ messages, length = "as few words as possible" }) => {
     const conversationMessages = messages.filter(
@@ -60,6 +65,20 @@ const createSummarizerService = async ({
   };
   return { summarizeConversation, summarize };
 };
+const summarizerServiceRef = backendPluginApi.createServiceRef({
+  id: "ai-assistant.summarizer-service",
+  defaultFactory: async (service) => backendPluginApi.createServiceFactory({
+    service,
+    deps: {
+      config: backendPluginApi.coreServices.rootConfig,
+      agent: index.agentServiceRef,
+      auth: backendPluginApi.coreServices.auth
+    },
+    factory: async (options) => {
+      return createSummarizerService(options);
+    }
+  })
+});
 
-exports.createSummarizerService = createSummarizerService;
+exports.summarizerServiceRef = summarizerServiceRef;
 //# sourceMappingURL=summarizer.cjs.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"summarizer.cjs.js","sources":["../../src/services/summarizer.ts"],"sourcesContent":["import { RootConfigService } from '@backstage/backend-plugin-api';\nimport { Model } from '@sweetoburrito/backstage-plugin-ai-assistant-node';\nimport {\n  DEFAULT_CONVERSATION_SUMMARY_PROMPT,\n  DEFAULT_SUMMARY_PROMPT,\n} from '../constants/prompts';\nimport { PromptTemplate } from '@langchain/core/prompts';\nimport { Message } from '@sweetoburrito/backstage-plugin-ai-assistant-common';\nimport { CallbackService } from './callbacks';\n\nexport type SummarizerService = {\n  summarizeConversation: (options: {\n    messages: Message[];\n    length?: string;\n  }) => Promise<string>;\n\n  summarize: (options: {\n    content: string;\n    prompt?: string;\n    length?: string;\n  }) => Promise<string>;\n};\n\ntype SummarizerServiceOptions = {\n  config: RootConfigService;\n  models: Model[];\n  callback: CallbackService;\n};\n\nexport const createSummarizerService = async ({\n  config,\n  models,\n  callback,\n}: SummarizerServiceOptions): Promise<SummarizerService> => {\n  const summaryModelId =\n    config.getOptionalString('aiAssistant.conversation.summaryModel') ??\n    models[0].id;\n\n  const conversationSummaryPrompt =\n    config.getOptionalString('aiAssistant.conversation.summaryPrompt') ??\n    DEFAULT_CONVERSATION_SUMMARY_PROMPT;\n\n  const model = models.find(m => m.id === summaryModelId);\n\n  if (!model) {\n    throw new Error(`Summary model with id ${summaryModelId} not found`);\n  }\n\n  const llm = model.chatModel;\n\n  const summaryPromptTemplate = PromptTemplate.fromTemplate(`\n    PURPOSE:\n    {summaryPrompt}\n\n    Summarize the following content in {length}.\n\n    Content:\n    {content}\n  `);\n\n  const summarize: SummarizerService['summarize'] = async (\n    content,\n    summaryPrompt = DEFAULT_SUMMARY_PROMPT,\n    length = 'as few words as possible',\n  ) => {\n    const prompt = await summaryPromptTemplate.format({\n      summaryPrompt,\n      content,\n      length,\n    });\n\n    const { callbacks } = await callback.getChainCallbacks({\n      conversationId: 'summarizer',\n      userId: 'system',\n      modelId: summaryModelId,\n    });\n\n    const { metadata } = await callback.getChainMetadata({\n      conversationId: 'summarizer',\n      userId: 'system',\n      modelId: summaryModelId,\n    });\n\n    const { text } = await llm.invoke(prompt, {\n      callbacks,\n      runName: 'summarizer',\n      metadata,\n    });\n\n    return text.trim();\n  };\n\n  const summarizeConversation: SummarizerService['summarizeConversation'] =\n    async ({ messages, length = 'as few words as possible' }) => {\n      const conversationMessages = messages.filter(\n        msg => msg.role === 'ai' || msg.role === 'human',\n      );\n\n      const conversation = conversationMessages\n        .map(msg => `${msg.role}: ${msg.content}`)\n        .join('\\n');\n\n      return summarize({\n        content: conversation,\n        prompt: conversationSummaryPrompt,\n        length,\n      });\n    };\n\n  return { summarizeConversation, summarize };\n};\n"],"names":["DEFAULT_CONVERSATION_SUMMARY_PROMPT","PromptTemplate","DEFAULT_SUMMARY_PROMPT"],"mappings":";;;;;AA6BO,MAAM,0BAA0B,OAAO;AAAA,EAC5C,MAAA;AAAA,EACA,MAAA;AAAA,EACA;AACF,CAAA,KAA4D;AAC1D,EAAA,MAAM,iBACJ,MAAA,CAAO,iBAAA,CAAkB,uCAAuC,CAAA,IAChE,MAAA,CAAO,CAAC,CAAA,CAAE,EAAA;AAEZ,EAAA,MAAM,yBAAA,GACJ,MAAA,CAAO,iBAAA,CAAkB,wCAAwC,CAAA,IACjEA,2CAAA;AAEF,EAAA,MAAM,QAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,CAAA,KAAK,CAAA,CAAE,OAAO,cAAc,CAAA;AAEtD,EAAA,IAAI,CAAC,KAAA,EAAO;AACV,IAAA,MAAM,IAAI,KAAA,CAAM,CAAA,sBAAA,EAAyB,cAAc,CAAA,UAAA,CAAY,CAAA;AAAA,EACrE;AAEA,EAAA,MAAM,MAAM,KAAA,CAAM,SAAA;AAElB,EAAA,MAAM,qBAAA,GAAwBC,yBAAe,YAAA,CAAa;AAAA;AAAA;;AAAA;;AAAA;AAAA;AAAA,EAAA,CAQzD,CAAA;AAED,EAAA,MAAM,YAA4C,OAChD,OAAA,EACA,aAAA,GAAgBC,8BAAA,EAChB,SAAS,0BAAA,KACN;AACH,IAAA,MAAM,MAAA,GAAS,MAAM,qBAAA,CAAsB,MAAA,CAAO;AAAA,MAChD,aAAA;AAAA,MACA,OAAA;AAAA,MACA;AAAA,KACD,CAAA;AAED,IAAA,MAAM,EAAE,SAAA,EAAU,GAAI,MAAM,SAAS,iBAAA,CAAkB;AAAA,MACrD,cAAA,EAAgB,YAAA;AAAA,MAChB,MAAA,EAAQ,QAAA;AAAA,MACR,OAAA,EAAS;AAAA,KACV,CAAA;AAED,IAAA,MAAM,EAAE,QAAA,EAAS,GAAI,MAAM,SAAS,gBAAA,CAAiB;AAAA,MACnD,cAAA,EAAgB,YAAA;AAAA,MAChB,MAAA,EAAQ,QAAA;AAAA,MACR,OAAA,EAAS;AAAA,KACV,CAAA;AAED,IAAA,MAAM,EAAE,IAAA,EAAK,GAAI,MAAM,GAAA,CAAI,OAAO,MAAA,EAAQ;AAAA,MACxC,SAAA;AAAA,MACA,OAAA,EAAS,YAAA;AAAA,MACT;AAAA,KACD,CAAA;AAED,IAAA,OAAO,KAAK,IAAA,EAAK;AAAA,EACnB,CAAA;AAEA,EAAA,MAAM,wBACJ,OAAO,EAAE,QAAA,EAAU,MAAA,GAAS,4BAA2B,KAAM;AAC3D,IAAA,MAAM,uBAAuB,QAAA,CAAS,MAAA;AAAA,MACpC,CAAA,GAAA,KAAO,GAAA,CAAI,IAAA,KAAS,IAAA,IAAQ,IAAI,IAAA,KAAS;AAAA,KAC3C;AAEA,IAAA,MAAM,YAAA,GAAe,oBAAA,CAClB,GAAA,CAAI,CAAA,GAAA,KAAO,CAAA,EAAG,GAAA,CAAI,IAAI,CAAA,EAAA,EAAK,GAAA,CAAI,OAAO,CAAA,CAAE,CAAA,CACxC,KAAK,IAAI,CAAA;AAEZ,IAAA,OAAO,SAAA,CAAU;AAAA,MACf,OAAA,EAAS,YAAA;AAAA,MACT,MAAA,EAAQ,yBAAA;AAAA,MACR;AAAA,KACD,CAAA;AAAA,EACH,CAAA;AAEF,EAAA,OAAO,EAAE,uBAAuB,SAAA,EAAU;AAC5C;;;;"}
+{"version":3,"file":"summarizer.cjs.js","sources":["../../src/services/summarizer.ts"],"sourcesContent":["import {\n  coreServices,\n  RootConfigService,\n  AuthService,\n} from '@backstage/backend-plugin-api';\nimport { AgentService, agentServiceRef } from './agent';\n\nimport {\n  DEFAULT_CONVERSATION_SUMMARY_PROMPT,\n  DEFAULT_SUMMARY_PROMPT,\n} from '../constants/prompts';\nimport { PromptTemplate } from '@langchain/core/prompts';\nimport { Message } from '@sweetoburrito/backstage-plugin-ai-assistant-common';\nimport {\n  createServiceFactory,\n  createServiceRef,\n  ServiceRef,\n} from '@backstage/backend-plugin-api';\nimport { v4 as uuid } from 'uuid';\n\nexport type SummarizerService = {\n  summarizeConversation: (options: {\n    messages: Message[];\n    length?: string;\n  }) => Promise<string>;\n\n  summarize: (options: {\n    content: string;\n    prompt?: string;\n    length?: string;\n  }) => Promise<string>;\n};\n\nexport type SummarizerServiceOptions = {\n  config: RootConfigService;\n  agent: AgentService;\n  auth: AuthService;\n};\n\nconst createSummarizerService = async ({\n  config,\n  agent,\n  auth,\n}: SummarizerServiceOptions): Promise<SummarizerService> => {\n  const summaryModelId = config.getOptionalString(\n    'aiAssistant.conversation.summaryModel',\n  );\n\n  const conversationSummaryPrompt =\n    config.getOptionalString('aiAssistant.conversation.summaryPrompt') ??\n    DEFAULT_CONVERSATION_SUMMARY_PROMPT;\n\n  const summaryPromptTemplate = PromptTemplate.fromTemplate(`\n    PURPOSE:\n    {summaryPrompt}\n\n    Summarize the following content in {length}.\n  `);\n\n  const summarize: SummarizerService['summarize'] = async ({\n    content,\n    length = 'as few words as possible',\n    prompt: summaryPrompt = DEFAULT_SUMMARY_PROMPT,\n  }) => {\n    const prompt = await summaryPromptTemplate.format({\n      summaryPrompt,\n      content,\n      length,\n    });\n\n    const credentials = await auth.getOwnServiceCredentials();\n\n    const messages = await agent.prompt({\n      messages: [{ role: 'system', content, score: 0, metadata: {} }],\n      metadata: {\n        conversationId: 'summarizer',\n        userId: 'system:summarizer',\n        runId: uuid(),\n        runName: 'summarizer',\n      },\n      systemPrompt: prompt,\n      modelId: summaryModelId,\n      credentials,\n      tools: [],\n    });\n\n    const aiMessages = messages.filter(m => m.role === 'ai');\n\n    if (aiMessages.length === 0) {\n      throw new Error('Failed to summarize content');\n    }\n\n    const response = messages.pop()!;\n\n    return response.content.trim();\n  };\n\n  const summarizeConversation: SummarizerService['summarizeConversation'] =\n    async ({ messages, length = 'as few words as possible' }) => {\n      const conversationMessages = messages.filter(\n        msg => msg.role === 'ai' || msg.role === 'human',\n      );\n\n      const conversation = conversationMessages\n        .map(msg => `${msg.role}: ${msg.content}`)\n        .join('\\n');\n\n      return summarize({\n        content: conversation,\n        prompt: conversationSummaryPrompt,\n        length,\n      });\n    };\n\n  return { summarizeConversation, summarize };\n};\n\nexport const summarizerServiceRef: ServiceRef<\n  SummarizerService,\n  'plugin',\n  'singleton'\n> = createServiceRef<SummarizerService>({\n  id: 'ai-assistant.summarizer-service',\n  defaultFactory: async service =>\n    createServiceFactory({\n      service,\n      deps: {\n        config: coreServices.rootConfig,\n        agent: agentServiceRef,\n        auth: coreServices.auth,\n      },\n      factory: async options => {\n        return createSummarizerService(options);\n      },\n    }),\n});\n"],"names":["DEFAULT_CONVERSATION_SUMMARY_PROMPT","PromptTemplate","DEFAULT_SUMMARY_PROMPT","uuid","createServiceRef","createServiceFactory","coreServices","agentServiceRef"],"mappings":";;;;;;;;AAuCA,MAAM,0BAA0B,OAAO;AAAA,EACrC,MAAA;AAAA,EACA,KAAA;AAAA,EACA;AACF,CAAA,KAA4D;AAC1D,EAAA,MAAM,iBAAiB,MAAA,CAAO,iBAAA;AAAA,IAC5B;AAAA,GACF;AAEA,EAAA,MAAM,yBAAA,GACJ,MAAA,CAAO,iBAAA,CAAkB,wCAAwC,CAAA,IACjEA,2CAAA;AAEF,EAAA,MAAM,qBAAA,GAAwBC,yBAAe,YAAA,CAAa;AAAA;AAAA;;AAAA;AAAA,EAAA,CAKzD,CAAA;AAED,EAAA,MAAM,YAA4C,OAAO;AAAA,IACvD,OAAA;AAAA,IACA,MAAA,GAAS,0BAAA;AAAA,IACT,QAAQ,aAAA,GAAgBC;AAAA,GAC1B,KAAM;AACJ,IAAA,MAAM,MAAA,GAAS,MAAM,qBAAA,CAAsB,MAAA,CAAO;AAAA,MAChD,aAAA;AAAA,MACA,OAAA;AAAA,MACA;AAAA,KACD,CAAA;AAED,IAAA,MAAM,WAAA,GAAc,MAAM,IAAA,CAAK,wBAAA,EAAyB;AAExD,IAAA,MAAM,QAAA,GAAW,MAAM,KAAA,CAAM,MAAA,CAAO;AAAA,MAClC,QAAA,EAAU,CAAC,EAAE,IAAA,EAAM,QAAA,EAAU,OAAA,EAAS,KAAA,EAAO,CAAA,EAAG,QAAA,EAAU,EAAC,EAAG,CAAA;AAAA,MAC9D,QAAA,EAAU;AAAA,QACR,cAAA,EAAgB,YAAA;AAAA,QAChB,MAAA,EAAQ,mBAAA;AAAA,QACR,OAAOC,OAAA,EAAK;AAAA,QACZ,OAAA,EAAS;AAAA,OACX;AAAA,MACA,YAAA,EAAc,MAAA;AAAA,MACd,OAAA,EAAS,cAAA;AAAA,MACT,WAAA;AAAA,MACA,OAAO;AAAC,KACT,CAAA;AAED,IAAA,MAAM,aAAa,QAAA,CAAS,MAAA,CAAO,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,IAAI,CAAA;AAEvD,IAAA,IAAI,UAAA,CAAW,WAAW,CAAA,EAAG;AAC3B,MAAA,MAAM,IAAI,MAAM,6BAA6B,CAAA;AAAA,IAC/C;AAEA,IAAA,MAAM,QAAA,GAAW,SAAS,GAAA,EAAI;AAE9B,IAAA,OAAO,QAAA,CAAS,QAAQ,IAAA,EAAK;AAAA,EAC/B,CAAA;AAEA,EAAA,MAAM,wBACJ,OAAO,EAAE,QAAA,EAAU,MAAA,GAAS,4BAA2B,KAAM;AAC3D,IAAA,MAAM,uBAAuB,QAAA,CAAS,MAAA;AAAA,MACpC,CAAA,GAAA,KAAO,GAAA,CAAI,IAAA,KAAS,IAAA,IAAQ,IAAI,IAAA,KAAS;AAAA,KAC3C;AAEA,IAAA,MAAM,YAAA,GAAe,oBAAA,CAClB,GAAA,CAAI,CAAA,GAAA,KAAO,CAAA,EAAG,GAAA,CAAI,IAAI,CAAA,EAAA,EAAK,GAAA,CAAI,OAAO,CAAA,CAAE,CAAA,CACxC,KAAK,IAAI,CAAA;AAEZ,IAAA,OAAO,SAAA,CAAU;AAAA,MACf,OAAA,EAAS,YAAA;AAAA,MACT,MAAA,EAAQ,yBAAA;AAAA,MACR;AAAA,KACD,CAAA;AAAA,EACH,CAAA;AAEF,EAAA,OAAO,EAAE,uBAAuB,SAAA,EAAU;AAC5C,CAAA;AAEO,MAAM,uBAITC,iCAAA,CAAoC;AAAA,EACtC,EAAA,EAAI,iCAAA;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,MAAMD,6BAAA,CAAa;AAAA,KACrB;AAAA,IACA,OAAA,EAAS,OAAM,OAAA,KAAW;AACxB,MAAA,OAAO,wBAAwB,OAAO,CAAA;AAAA,IACxC;AAAA,GACD;AACL,CAAC;;;;"}

package/dist/services/tools.cjs.js

@@ -0,0 +1,65 @@
+'use strict';
+
+var backendPluginApi = require('@backstage/backend-plugin-api');
+var index = require('./mcp/index.cjs.js');
+var tools = require('@langchain/core/tools');
+
+const createToolsService = async ({
+  mcp,
+  auth
+}) => {
+  const tools$1 = [];
+  const coreTools = [];
+  const registerTools = async (providers) => {
+    tools$1.push(...providers);
+  };
+  const registerCoreTools = async (providers) => {
+    coreTools.push(...providers.map((tool) => ({ ...tool, provider: "core" })));
+  };
+  const getAvailableUserTools = async ({
+    credentials
+  }) => {
+    const mcpTools = await mcp.getTools(credentials);
+    const availableTools = tools$1.concat(mcpTools).concat(coreTools).map((tool) => ({
+      name: tool.name,
+      provider: tool.provider,
+      description: tool.description
+    }));
+    return availableTools;
+  };
+  const getPrincipalTools = async ({
+    credentials,
+    filter = () => true
+  }) => {
+    const isUser = auth.isPrincipal(credentials, "user");
+    if (!isUser) {
+      return tools$1.filter(filter).map((t) => new tools.DynamicStructuredTool(t));
+    }
+    const mcpTools = await mcp.getTools(credentials);
+    const userTools = tools$1.concat(mcpTools);
+    const allTools = userTools.filter(filter).concat(coreTools.filter(filter));
+    return allTools.map((t) => new tools.DynamicStructuredTool(t));
+  };
+  return {
+    registerTools,
+    registerCoreTools,
+    getAvailableUserTools,
+    getPrincipalTools
+  };
+};
+const toolsServiceRef = backendPluginApi.createServiceRef({
+  id: "ai-assistant.tools-service",
+  defaultFactory: async (service) => backendPluginApi.createServiceFactory({
+    service,
+    deps: {
+      mcp: index.mcpServiceRef,
+      auth: backendPluginApi.coreServices.auth
+    },
+    factory: async (options) => {
+      return createToolsService(options);
+    }
+  })
+});
+
+exports.toolsServiceRef = toolsServiceRef;
+//# sourceMappingURL=tools.cjs.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tools.cjs.js","sources":["../../src/services/tools.ts"],"sourcesContent":["import {\n  BackstageCredentials,\n  coreServices,\n  createServiceFactory,\n  createServiceRef,\n  ServiceRef,\n  AuthService,\n} from '@backstage/backend-plugin-api';\nimport {\n  Tool,\n  EnabledTool,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-common';\nimport { McpService, mcpServiceRef } from './mcp';\nimport { DynamicStructuredTool } from '@langchain/core/tools';\n\nexport type ToolsService = {\n  registerTools: (tools: Tool[]) => void;\n  registerCoreTools: (tools: Tool[]) => void;\n  getAvailableUserTools: (options: {\n    credentials: BackstageCredentials;\n  }) => Promise<EnabledTool[]>;\n  getPrincipalTools: (options: {\n    credentials: BackstageCredentials;\n    filter?: (tool: Tool) => boolean;\n  }) => Promise<DynamicStructuredTool[]>;\n};\n\nexport type CreateToolsServiceOptions = {\n  mcp: McpService;\n  auth: AuthService;\n};\n\nconst createToolsService = async ({\n  mcp,\n  auth,\n}: CreateToolsServiceOptions): Promise<ToolsService> => {\n  const tools: Tool[] = [];\n  const coreTools: Tool[] = [];\n\n  const registerTools: ToolsService['registerTools'] = async providers => {\n    tools.push(...providers);\n  };\n\n  const registerCoreTools: ToolsService['registerCoreTools'] =\n    async providers => {\n      coreTools.push(...providers.map(tool => ({ ...tool, provider: 'core' })));\n    };\n\n  const getAvailableUserTools: ToolsService['getAvailableUserTools'] = async ({\n    credentials,\n  }) => {\n    const mcpTools = await mcp.getTools(credentials);\n\n    const availableTools: EnabledTool[] = tools\n      .concat(mcpTools)\n      .concat(coreTools)\n      .map(tool => ({\n        name: tool.name,\n        provider: tool.provider,\n        description: tool.description,\n      }));\n\n    return availableTools;\n  };\n\n  const getPrincipalTools: ToolsService['getPrincipalTools'] = async ({\n    credentials,\n    filter = () => true,\n  }) => {\n    const isUser = auth.isPrincipal(credentials, 'user');\n\n    if (!isUser) {\n      return tools.filter(filter).map(t => new DynamicStructuredTool(t));\n    }\n\n    const mcpTools = await mcp.getTools(credentials);\n\n    const userTools = tools.concat(mcpTools);\n\n    const allTools: Tool[] = userTools\n      .filter(filter)\n      .concat(coreTools.filter(filter));\n    return allTools.map(t => new DynamicStructuredTool(t));\n  };\n\n  return {\n    registerTools,\n    registerCoreTools,\n    getAvailableUserTools,\n    getPrincipalTools,\n  };\n};\n\nexport const toolsServiceRef: ServiceRef<ToolsService, 'plugin', 'singleton'> =\n  createServiceRef<ToolsService>({\n    id: 'ai-assistant.tools-service',\n    defaultFactory: async service =>\n      createServiceFactory({\n        service,\n        deps: {\n          mcp: mcpServiceRef,\n          auth: coreServices.auth,\n        },\n        factory: async options => {\n          return createToolsService(options);\n        },\n      }),\n  });\n"],"names":["tools","DynamicStructuredTool","createServiceRef","createServiceFactory","mcpServiceRef","coreServices"],"mappings":";;;;;;AAgCA,MAAM,qBAAqB,OAAO;AAAA,EAChC,GAAA;AAAA,EACA;AACF,CAAA,KAAwD;AACtD,EAAA,MAAMA,UAAgB,EAAC;AACvB,EAAA,MAAM,YAAoB,EAAC;AAE3B,EAAA,MAAM,aAAA,GAA+C,OAAM,SAAA,KAAa;AACtE,IAAAA,OAAA,CAAM,IAAA,CAAK,GAAG,SAAS,CAAA;AAAA,EACzB,CAAA;AAEA,EAAA,MAAM,iBAAA,GACJ,OAAM,SAAA,KAAa;AACjB,IAAA,SAAA,CAAU,IAAA,CAAK,GAAG,SAAA,CAAU,GAAA,CAAI,CAAA,IAAA,MAAS,EAAE,GAAG,IAAA,EAAM,QAAA,EAAU,MAAA,EAAO,CAAE,CAAC,CAAA;AAAA,EAC1E,CAAA;AAEF,EAAA,MAAM,wBAA+D,OAAO;AAAA,IAC1E;AAAA,GACF,KAAM;AACJ,IAAA,MAAM,QAAA,GAAW,MAAM,GAAA,CAAI,QAAA,CAAS,WAAW,CAAA;AAE/C,IAAA,MAAM,cAAA,GAAgCA,QACnC,MAAA,CAAO,QAAQ,EACf,MAAA,CAAO,SAAS,CAAA,CAChB,GAAA,CAAI,CAAA,IAAA,MAAS;AAAA,MACZ,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,UAAU,IAAA,CAAK,QAAA;AAAA,MACf,aAAa,IAAA,CAAK;AAAA,KACpB,CAAE,CAAA;AAEJ,IAAA,OAAO,cAAA;AAAA,EACT,CAAA;AAEA,EAAA,MAAM,oBAAuD,OAAO;AAAA,IAClE,WAAA;AAAA,IACA,SAAS,MAAM;AAAA,GACjB,KAAM;AACJ,IAAA,MAAM,MAAA,GAAS,IAAA,CAAK,WAAA,CAAY,WAAA,EAAa,MAAM,CAAA;AAEnD,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,OAAOA,OAAA,CAAM,OAAO,MAAM,CAAA,CAAE,IAAI,CAAA,CAAA,KAAK,IAAIC,2BAAA,CAAsB,CAAC,CAAC,CAAA;AAAA,IACnE;AAEA,IAAA,MAAM,QAAA,GAAW,MAAM,GAAA,CAAI,QAAA,CAAS,WAAW,CAAA;AAE/C,IAAA,MAAM,SAAA,GAAYD,OAAA,CAAM,MAAA,CAAO,QAAQ,CAAA;AAEvC,IAAA,MAAM,QAAA,GAAmB,UACtB,MAAA,CAAO,MAAM,EACb,MAAA,CAAO,SAAA,CAAU,MAAA,CAAO,MAAM,CAAC,CAAA;AAClC,IAAA,OAAO,SAAS,GAAA,CAAI,CAAA,CAAA,KAAK,IAAIC,2BAAA,CAAsB,CAAC,CAAC,CAAA;AAAA,EACvD,CAAA;AAEA,EAAA,OAAO;AAAA,IACL,aAAA;AAAA,IACA,iBAAA;AAAA,IACA,qBAAA;AAAA,IACA;AAAA,GACF;AACF,CAAA;AAEO,MAAM,kBACXC,iCAAA,CAA+B;AAAA,EAC7B,EAAA,EAAI,4BAAA;AAAA,EACJ,cAAA,EAAgB,OAAM,OAAA,KACpBC,qCAAA,CAAqB;AAAA,IACnB,OAAA;AAAA,IACA,IAAA,EAAM;AAAA,MACJ,GAAA,EAAKC,mBAAA;AAAA,MACL,MAAMC,6BAAA,CAAa;AAAA,KACrB;AAAA,IACA,OAAA,EAAS,OAAM,OAAA,KAAW;AACxB,MAAA,OAAO,mBAAmB,OAAO,CAAA;AAAA,IACnC;AAAA,GACD;AACL,CAAC;;;;"}

package/dist/services/user-settings.cjs.js

@@ -1,5 +1,6 @@
 'use strict';
 
+var backendPluginApi = require('@backstage/backend-plugin-api');
 var userSettingsStore = require('../database/user-settings-store.cjs.js');
 
 const createUserSettingsService = async ({
@@ -29,6 +30,20 @@ const createUserSettingsService = async ({
     deleteSettingsForType
   };
 };
+const userSettingsServiceRef = backendPluginApi.createServiceRef({
+  id: "ai-assistant.user-settings-service",
+  defaultFactory: async (service) => backendPluginApi.createServiceFactory({
+    service,
+    deps: {
+      userInfo: backendPluginApi.coreServices.userInfo,
+      database: backendPluginApi.coreServices.database
+    },
+    factory: async (options) => {
+      return createUserSettingsService(options);
+    }
+  })
+});
 
 exports.createUserSettingsService = createUserSettingsService;
+exports.userSettingsServiceRef = userSettingsServiceRef;
 //# sourceMappingURL=user-settings.cjs.js.map

package/dist/services/user-settings.cjs.js.map

@@ -1 +1 @@
-{"version":3,"file":"user-settings.cjs.js","sources":["../../src/services/user-settings.ts"],"sourcesContent":["import {\n  BackstageCredentials,\n  DatabaseService,\n  UserInfoService,\n} from '@backstage/backend-plugin-api';\nimport { UserSettingsStore } from '../database/user-settings-store';\n\ntype CreateUserSettingsServiceOptions = {\n  userInfo: UserInfoService;\n  database: DatabaseService;\n};\n\nexport type UserSettingsService = {\n  getSettingsForType: (\n    credentials: BackstageCredentials,\n    type: string,\n  ) => Promise<Record<string, unknown>>;\n  setSettingsForType: (\n    credentials: BackstageCredentials,\n    type: string,\n    settings: Record<string, unknown>,\n  ) => Promise<void>;\n  deleteSettingsForType: (\n    credentials: BackstageCredentials,\n    type: string,\n  ) => Promise<void>;\n};\n\nexport const createUserSettingsService = async ({\n  userInfo,\n  database,\n}: CreateUserSettingsServiceOptions): Promise<UserSettingsService> => {\n  const userSettingsStore = await UserSettingsStore.fromConfig({ database });\n\n  const getSettingsForType: UserSettingsService['getSettingsForType'] = async (\n    credentials,\n    type,\n  ) => {\n    const { userEntityRef } = await userInfo.getUserInfo(credentials);\n    const settings = await userSettingsStore.getUserSettingsByType(\n      userEntityRef,\n      type,\n    );\n\n    return settings ?? {};\n  };\n\n  const setSettingsForType: UserSettingsService['setSettingsForType'] = async (\n    credentials,\n    type,\n    settings,\n  ) => {\n    const { userEntityRef } = await userInfo.getUserInfo(credentials);\n    await userSettingsStore.setUserSettings(userEntityRef, type, settings);\n  };\n\n  const deleteSettingsForType: UserSettingsService['deleteSettingsForType'] =\n    async (credentials, type) => {\n      const { userEntityRef } = await userInfo.getUserInfo(credentials);\n      await userSettingsStore.deleteUserSettings(userEntityRef, type);\n    };\n\n  return {\n    getSettingsForType,\n    setSettingsForType,\n    deleteSettingsForType,\n  };\n};\n"],"names":["userSettingsStore","UserSettingsStore"],"mappings":";;;;AA4BO,MAAM,4BAA4B,OAAO;AAAA,EAC9C,QAAA;AAAA,EACA;AACF,CAAA,KAAsE;AACpE,EAAA,MAAMA,sBAAoB,MAAMC,mCAAA,CAAkB,UAAA,CAAW,EAAE,UAAU,CAAA;AAEzE,EAAA,MAAM,kBAAA,GAAgE,OACpE,WAAA,EACA,IAAA,KACG;AACH,IAAA,MAAM,EAAE,aAAA,EAAc,GAAI,MAAM,QAAA,CAAS,YAAY,WAAW,CAAA;AAChE,IAAA,MAAM,QAAA,GAAW,MAAMD,mBAAA,CAAkB,qBAAA;AAAA,MACvC,aAAA;AAAA,MACA;AAAA,KACF;AAEA,IAAA,OAAO,YAAY,EAAC;AAAA,EACtB,CAAA;AAEA,EAAA,MAAM,kBAAA,GAAgE,OACpE,WAAA,EACA,IAAA,EACA,QAAA,KACG;AACH,IAAA,MAAM,EAAE,aAAA,EAAc,GAAI,MAAM,QAAA,CAAS,YAAY,WAAW,CAAA;AAChE,IAAA,MAAMA,mBAAA,CAAkB,eAAA,CAAgB,aAAA,EAAe,IAAA,EAAM,QAAQ,CAAA;AAAA,EACvE,CAAA;AAEA,EAAA,MAAM,qBAAA,GACJ,OAAO,WAAA,EAAa,IAAA,KAAS;AAC3B,IAAA,MAAM,EAAE,aAAA,EAAc,GAAI,MAAM,QAAA,CAAS,YAAY,WAAW,CAAA;AAChE,IAAA,MAAMA,mBAAA,CAAkB,kBAAA,CAAmB,aAAA,EAAe,IAAI,CAAA;AAAA,EAChE,CAAA;AAEF,EAAA,OAAO;AAAA,IACL,kBAAA;AAAA,IACA,kBAAA;AAAA,IACA;AAAA,GACF;AACF;;;;"}
+{"version":3,"file":"user-settings.cjs.js","sources":["../../src/services/user-settings.ts"],"sourcesContent":["import {\n  createServiceFactory,\n  createServiceRef,\n  coreServices,\n} from '@backstage/backend-plugin-api';\n\nimport type {\n  ServiceRef,\n  BackstageCredentials,\n  DatabaseService,\n  UserInfoService,\n} from '@backstage/backend-plugin-api';\n\nimport { UserSettingsStore } from '../database/user-settings-store';\n\ntype CreateUserSettingsServiceOptions = {\n  userInfo: UserInfoService;\n  database: DatabaseService;\n};\n\nexport type UserSettingsService = {\n  getSettingsForType: (\n    credentials: BackstageCredentials,\n    type: string,\n  ) => Promise<Record<string, unknown>>;\n  setSettingsForType: (\n    credentials: BackstageCredentials,\n    type: string,\n    settings: Record<string, unknown>,\n  ) => Promise<void>;\n  deleteSettingsForType: (\n    credentials: BackstageCredentials,\n    type: string,\n  ) => Promise<void>;\n};\n\nexport const createUserSettingsService = async ({\n  userInfo,\n  database,\n}: CreateUserSettingsServiceOptions): Promise<UserSettingsService> => {\n  const userSettingsStore = await UserSettingsStore.fromConfig({ database });\n\n  const getSettingsForType: UserSettingsService['getSettingsForType'] = async (\n    credentials,\n    type,\n  ) => {\n    const { userEntityRef } = await userInfo.getUserInfo(credentials);\n    const settings = await userSettingsStore.getUserSettingsByType(\n      userEntityRef,\n      type,\n    );\n\n    return settings ?? {};\n  };\n\n  const setSettingsForType: UserSettingsService['setSettingsForType'] = async (\n    credentials,\n    type,\n    settings,\n  ) => {\n    const { userEntityRef } = await userInfo.getUserInfo(credentials);\n    await userSettingsStore.setUserSettings(userEntityRef, type, settings);\n  };\n\n  const deleteSettingsForType: UserSettingsService['deleteSettingsForType'] =\n    async (credentials, type) => {\n      const { userEntityRef } = await userInfo.getUserInfo(credentials);\n      await userSettingsStore.deleteUserSettings(userEntityRef, type);\n    };\n\n  return {\n    getSettingsForType,\n    setSettingsForType,\n    deleteSettingsForType,\n  };\n};\n\nexport const userSettingsServiceRef: ServiceRef<\n  UserSettingsService,\n  'plugin',\n  'singleton'\n> = createServiceRef<UserSettingsService>({\n  id: 'ai-assistant.user-settings-service',\n  defaultFactory: async service =>\n    createServiceFactory({\n      service,\n      deps: {\n        userInfo: coreServices.userInfo,\n        database: coreServices.database,\n      },\n      factory: async options => {\n        return createUserSettingsService(options);\n      },\n    }),\n});\n"],"names":["userSettingsStore","UserSettingsStore","createServiceRef","createServiceFactory","coreServices"],"mappings":";;;;;AAoCO,MAAM,4BAA4B,OAAO;AAAA,EAC9C,QAAA;AAAA,EACA;AACF,CAAA,KAAsE;AACpE,EAAA,MAAMA,sBAAoB,MAAMC,mCAAA,CAAkB,UAAA,CAAW,EAAE,UAAU,CAAA;AAEzE,EAAA,MAAM,kBAAA,GAAgE,OACpE,WAAA,EACA,IAAA,KACG;AACH,IAAA,MAAM,EAAE,aAAA,EAAc,GAAI,MAAM,QAAA,CAAS,YAAY,WAAW,CAAA;AAChE,IAAA,MAAM,QAAA,GAAW,MAAMD,mBAAA,CAAkB,qBAAA;AAAA,MACvC,aAAA;AAAA,MACA;AAAA,KACF;AAEA,IAAA,OAAO,YAAY,EAAC;AAAA,EACtB,CAAA;AAEA,EAAA,MAAM,kBAAA,GAAgE,OACpE,WAAA,EACA,IAAA,EACA,QAAA,KACG;AACH,IAAA,MAAM,EAAE,aAAA,EAAc,GAAI,MAAM,QAAA,CAAS,YAAY,WAAW,CAAA;AAChE,IAAA,MAAMA,mBAAA,CAAkB,eAAA,CAAgB,aAAA,EAAe,IAAA,EAAM,QAAQ,CAAA;AAAA,EACvE,CAAA;AAEA,EAAA,MAAM,qBAAA,GACJ,OAAO,WAAA,EAAa,IAAA,KAAS;AAC3B,IAAA,MAAM,EAAE,aAAA,EAAc,GAAI,MAAM,QAAA,CAAS,YAAY,WAAW,CAAA;AAChE,IAAA,MAAMA,mBAAA,CAAkB,kBAAA,CAAmB,aAAA,EAAe,IAAI,CAAA;AAAA,EAChE,CAAA;AAEF,EAAA,OAAO;AAAA,IACL,kBAAA;AAAA,IACA,kBAAA;AAAA,IACA;AAAA,GACF;AACF;AAEO,MAAM,yBAITE,iCAAA,CAAsC;AAAA,EACxC,EAAA,EAAI,oCAAA;AAAA,EACJ,cAAA,EAAgB,OAAM,OAAA,KACpBC,qCAAA,CAAqB;AAAA,IACnB,OAAA;AAAA,IACA,IAAA,EAAM;AAAA,MACJ,UAAUC,6BAAA,CAAa,QAAA;AAAA,MACvB,UAAUA,6BAAA,CAAa;AAAA,KACzB;AAAA,IACA,OAAA,EAAS,OAAM,OAAA,KAAW;AACxB,MAAA,OAAO,0BAA0B,OAAO,CAAA;AAAA,IAC1C;AAAA,GACD;AACL,CAAC;;;;;"}

package/dist/tools/searchKnowledge.cjs.js

@@ -39,13 +39,8 @@ Do NOT use for general knowledge that doesn't require company-specific informati
             content: "No relevant information found in the knowledge base."
           };
         }
-        const content = results.map((r) => r.content).join("\n---\n");
-        const urls = results.map((r) => r.metadata.url);
         return {
-          content,
-          metadata: {
-            urls
-          }
+          content: JSON.stringify(results)
         };
       }
     }

package/dist/tools/searchKnowledge.cjs.js.map

@@ -1 +1 @@
-{"version":3,"file":"searchKnowledge.cjs.js","sources":["../../src/tools/searchKnowledge.ts"],"sourcesContent":["import {\n  createAssistantTool,\n  VectorStore,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-node';\nimport { Tool } from '@sweetoburrito/backstage-plugin-ai-assistant-common';\nimport z from 'zod';\n\ntype CreateSearchKnowledgeToolOptions = {\n  vectorStore: VectorStore;\n};\n\nexport const createSearchKnowledgeTool = ({\n  vectorStore,\n}: CreateSearchKnowledgeToolOptions): Tool => {\n  const knowledgeTool = createAssistantTool({\n    tool: {\n      name: 'search-knowledge-base',\n      description: `Search the internal knowledge base containing company specific information.\n\nUse this tool when users ask about:\n- General questions about the company or internal information\n\nDo NOT use for general knowledge that doesn't require company-specific information.`,\n      provider: 'core',\n      schema: z.object({\n        query: z.string().describe('The query to search for.'),\n        filter: z\n          .object({\n            source: z.string().optional().describe('Source to filter by.'),\n            id: z.string().optional().describe('ID to filter by.'),\n          })\n          .optional()\n          .describe('Filters to apply to the search.'),\n        amount: z\n          .number()\n          .min(1)\n          .optional()\n          .describe('The number of results to return.'),\n      }),\n      func: async ({ query, filter, amount }) => {\n        const results = await vectorStore.similaritySearch(\n          query,\n          filter,\n          amount,\n        );\n\n        if (results.length === 0) {\n          return {\n            content: 'No relevant information found in the knowledge base.',\n          };\n        }\n\n        const content = results.map(r => r.content).join('\\n---\\n');\n\n        const urls = results.map(r => r.metadata.url);\n\n        return {\n          content,\n          metadata: {\n            urls,\n          },\n        };\n      },\n    },\n  });\n\n  return knowledgeTool;\n};\n"],"names":["createAssistantTool","z"],"mappings":";;;;;;;;;AAWO,MAAM,4BAA4B,CAAC;AAAA,EACxC;AACF,CAAA,KAA8C;AAC5C,EAAA,MAAM,gBAAgBA,kDAAA,CAAoB;AAAA,IACxC,IAAA,EAAM;AAAA,MACJ,IAAA,EAAM,uBAAA;AAAA,MACN,WAAA,EAAa,CAAA;;AAAA;AAAA;;AAAA,mFAAA,CAAA;AAAA,MAMb,QAAA,EAAU,MAAA;AAAA,MACV,MAAA,EAAQC,mBAAE,MAAA,CAAO;AAAA,QACf,KAAA,EAAOA,kBAAA,CAAE,MAAA,EAAO,CAAE,SAAS,0BAA0B,CAAA;AAAA,QACrD,MAAA,EAAQA,mBACL,MAAA,CAAO;AAAA,UACN,QAAQA,kBAAA,CAAE,MAAA,GAAS,QAAA,EAAS,CAAE,SAAS,sBAAsB,CAAA;AAAA,UAC7D,IAAIA,kBAAA,CAAE,MAAA,GAAS,QAAA,EAAS,CAAE,SAAS,kBAAkB;AAAA,SACtD,CAAA,CACA,QAAA,EAAS,CACT,SAAS,iCAAiC,CAAA;AAAA,QAC7C,MAAA,EAAQA,kBAAA,CACL,MAAA,EAAO,CACP,GAAA,CAAI,CAAC,CAAA,CACL,QAAA,EAAS,CACT,QAAA,CAAS,kCAAkC;AAAA,OAC/C,CAAA;AAAA,MACD,MAAM,OAAO,EAAE,KAAA,EAAO,MAAA,EAAQ,QAAO,KAAM;AACzC,QAAA,MAAM,OAAA,GAAU,MAAM,WAAA,CAAY,gBAAA;AAAA,UAChC,KAAA;AAAA,UACA,MAAA;AAAA,UACA;AAAA,SACF;AAEA,QAAA,IAAI,OAAA,CAAQ,WAAW,CAAA,EAAG;AACxB,UAAA,OAAO;AAAA,YACL,OAAA,EAAS;AAAA,WACX;AAAA,QACF;AAEA,QAAA,MAAM,OAAA,GAAU,QAAQ,GAAA,CAAI,CAAA,CAAA,KAAK,EAAE,OAAO,CAAA,CAAE,KAAK,SAAS,CAAA;AAE1D,QAAA,MAAM,OAAO,OAAA,CAAQ,GAAA,CAAI,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,GAAG,CAAA;AAE5C,QAAA,OAAO;AAAA,UACL,OAAA;AAAA,UACA,QAAA,EAAU;AAAA,YACR;AAAA;AACF,SACF;AAAA,MACF;AAAA;AACF,GACD,CAAA;AAED,EAAA,OAAO,aAAA;AACT;;;;"}
+{"version":3,"file":"searchKnowledge.cjs.js","sources":["../../src/tools/searchKnowledge.ts"],"sourcesContent":["import {\n  createAssistantTool,\n  VectorStore,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-node';\nimport { Tool } from '@sweetoburrito/backstage-plugin-ai-assistant-common';\nimport z from 'zod';\n\ntype CreateSearchKnowledgeToolOptions = {\n  vectorStore: VectorStore;\n};\n\nexport const createSearchKnowledgeTool = ({\n  vectorStore,\n}: CreateSearchKnowledgeToolOptions): Tool => {\n  const knowledgeTool = createAssistantTool({\n    tool: {\n      name: 'search-knowledge-base',\n      description: `Search the internal knowledge base containing company specific information.\n\nUse this tool when users ask about:\n- General questions about the company or internal information\n\nDo NOT use for general knowledge that doesn't require company-specific information.`,\n      provider: 'core',\n      schema: z.object({\n        query: z.string().describe('The query to search for.'),\n        filter: z\n          .object({\n            source: z.string().optional().describe('Source to filter by.'),\n            id: z.string().optional().describe('ID to filter by.'),\n          })\n          .optional()\n          .describe('Filters to apply to the search.'),\n        amount: z\n          .number()\n          .min(1)\n          .optional()\n          .describe('The number of results to return.'),\n      }),\n      func: async ({ query, filter, amount }) => {\n        const results = await vectorStore.similaritySearch(\n          query,\n          filter,\n          amount,\n        );\n\n        if (results.length === 0) {\n          return {\n            content: 'No relevant information found in the knowledge base.',\n          };\n        }\n\n        return {\n          content: JSON.stringify(results),\n        };\n      },\n    },\n  });\n\n  return knowledgeTool;\n};\n"],"names":["createAssistantTool","z"],"mappings":";;;;;;;;;AAWO,MAAM,4BAA4B,CAAC;AAAA,EACxC;AACF,CAAA,KAA8C;AAC5C,EAAA,MAAM,gBAAgBA,kDAAA,CAAoB;AAAA,IACxC,IAAA,EAAM;AAAA,MACJ,IAAA,EAAM,uBAAA;AAAA,MACN,WAAA,EAAa,CAAA;;AAAA;AAAA;;AAAA,mFAAA,CAAA;AAAA,MAMb,QAAA,EAAU,MAAA;AAAA,MACV,MAAA,EAAQC,mBAAE,MAAA,CAAO;AAAA,QACf,KAAA,EAAOA,kBAAA,CAAE,MAAA,EAAO,CAAE,SAAS,0BAA0B,CAAA;AAAA,QACrD,MAAA,EAAQA,mBACL,MAAA,CAAO;AAAA,UACN,QAAQA,kBAAA,CAAE,MAAA,GAAS,QAAA,EAAS,CAAE,SAAS,sBAAsB,CAAA;AAAA,UAC7D,IAAIA,kBAAA,CAAE,MAAA,GAAS,QAAA,EAAS,CAAE,SAAS,kBAAkB;AAAA,SACtD,CAAA,CACA,QAAA,EAAS,CACT,SAAS,iCAAiC,CAAA;AAAA,QAC7C,MAAA,EAAQA,kBAAA,CACL,MAAA,EAAO,CACP,GAAA,CAAI,CAAC,CAAA,CACL,QAAA,EAAS,CACT,QAAA,CAAS,kCAAkC;AAAA,OAC/C,CAAA;AAAA,MACD,MAAM,OAAO,EAAE,KAAA,EAAO,MAAA,EAAQ,QAAO,KAAM;AACzC,QAAA,MAAM,OAAA,GAAU,MAAM,WAAA,CAAY,gBAAA;AAAA,UAChC,KAAA;AAAA,UACA,MAAA;AAAA,UACA;AAAA,SACF;AAEA,QAAA,IAAI,OAAA,CAAQ,WAAW,CAAA,EAAG;AACxB,UAAA,OAAO;AAAA,YACL,OAAA,EAAS;AAAA,WACX;AAAA,QACF;AAEA,QAAA,OAAO;AAAA,UACL,OAAA,EAAS,IAAA,CAAK,SAAA,CAAU,OAAO;AAAA,SACjC;AAAA,MACF;AAAA;AACF,GACD,CAAA;AAED,EAAA,OAAO,aAAA;AACT;;;;"}

package/migrations/20260108_data_ingestion_refactor.js

@@ -0,0 +1,27 @@
+const EMBEDDINGS_TABLE = 'embeddings';
+
+/**
+ *
+ * @param {import('knex').knex} knex
+ */
+exports.down = async knex => {
+  await knex.schema.alterTable(EMBEDDINGS_TABLE, table => {
+    table.dropColumn('hash');
+    table.dropColumn('lastUpdated');
+  });
+};
+
+/**
+ *
+ * @param {import('knex').knex} knex
+ */
+exports.up = async knex => {
+  await knex.schema.alterTable(EMBEDDINGS_TABLE, table => {
+    table.string('hash').comment('The content hash of the embedding document');
+
+    table
+      .timestamp('lastUpdated')
+      .defaultTo(knex.fn.now())
+      .comment('Timestamp of the last update to the embedding document');
+  });
+};