@sweetoburrito/backstage-plugin-ai-assistant-backend 0.8.0 → 0.10.0

package/config.d.ts

@@ -1,5 +1,6 @@
 import { HumanDuration } from '@backstage/types';
 import { SchedulerServiceTaskScheduleDefinitionConfig } from '@backstage/backend-plugin-api';
+import { McpServerConfig } from '@sweetoburrito/backstage-plugin-ai-assistant-common';
 
 export interface Config {
   aiAssistant: {
@@ -28,5 +29,12 @@ export interface Config {
     ingestion?: {
       schedule?: SchedulerServiceTaskScheduleDefinitionConfig;
     };
+    mcp: {
+      /**
+       * @visibility secret
+       */
+      encryptionKey: string;
+      servers?: Array<McpServerConfig>;
+    };
   };
 }

package/dist/constants/prompts.cjs.js

@@ -1,21 +1,30 @@
 'use strict';
 
-const DEFAULT_SUMMARY_PROMPT = "Summarize this conversation in a concise manner. The summary should capture the main points. Return the summary only, without any additional text.";
-const DEFAULT_SYSTEM_PROMPT = `
+const DEFAULT_CONVERSATION_SUMMARY_PROMPT = "Summarize this conversation in a concise manner. The summary should capture the main points. Return the summary only, without any additional text. Do not include any introductions or other part of the conversation that doesn't contribute to the summary or form part of the overall conversation as part of the summary.";
+const DEFAULT_SUMMARY_PROMPT = "Summarize the following content in a concise manner. The summary should capture the main points. Return the summary only, without any additional text. Do not include any introductions or other part of the content that doesn't contribute to the summary.";
+const DEFAULT_IDENTITY_PROMPT = `
 You are a helpful assistant that answers questions based on provided context from various documents. The context may come from sources such as internal wikis, code repositories, technical documentation, or other structured or unstructured data.
-
-Rules:
+`;
+const DEFAULT_FORMATTING_PROMPT = `
+CRITICAL FORMATTING RULES - MUST ALWAYS FOLLOW:
+1. **ALWAYS use proper markdown formatting in ALL responses**
+2. **NEVER output plain URLs** - ALWAYS convert them to clickable markdown links using [description](url) syntax
+3. **For images, ALWAYS use markdown image syntax**: ![alt text](image-url)
+4. **For all URLs, ALWAYS format as**: [descriptive text](url) - never just paste the raw URL
+5. Use headings (##, ###), bullet points, numbered lists, and **bold**/*italic* text appropriately
+6. Format code with backticks: \`inline code\` or \`\`\`language for code blocks
+7. Structure responses clearly with proper spacing and organization
+`;
+const DEFAULT_SYSTEM_PROMPT = `
+Content Rules:
 1. Always base your answers on the provided context. Do not make up information.
 2. When relevant, cite or reference the source information provided in the context.
-3. Format answers clearly and concisely. Use bullet points for lists when appropriate.
-4. Maintain a professional, friendly, and helpful tone.
-5. Return only the relevant information without any filler or unnecessary details.
-6. If you don't know the answer, admit it and suggest ways to find the information.
-7. Always return a well-structured response using markdown.
-8. **Actively use available tools** to enhance your responses:
-9. Adapt your approach based on the specific tools and capabilities available in the current session
-10. When you have a link to an image, include it in your response using markdown format ![description](image_url).
-11. When you have a link to a document, include it in your response using markdown format [description](document_url).
+3. Maintain a professional, friendly, and helpful tone.
+4. Return only the relevant information without any filler or unnecessary details.
+5. If you don't know the answer, admit it and suggest ways to find the information.
+6. **Actively use available tools** to enhance your responses
+7. Adapt your approach based on the specific tools and capabilities available in the current session
+8. When you do not have the information needed to answer, use the tools provided to gather more context before responding.
 `;
 const DEFAULT_TOOL_GUIDELINE = `
 TOOL USAGE GUIDELINES:
@@ -27,6 +36,9 @@ TOOL USAGE GUIDELINES:
 - If a tool fails, try an alternative approach before using another tool
 `;
 
+exports.DEFAULT_CONVERSATION_SUMMARY_PROMPT = DEFAULT_CONVERSATION_SUMMARY_PROMPT;
+exports.DEFAULT_FORMATTING_PROMPT = DEFAULT_FORMATTING_PROMPT;
+exports.DEFAULT_IDENTITY_PROMPT = DEFAULT_IDENTITY_PROMPT;
 exports.DEFAULT_SUMMARY_PROMPT = DEFAULT_SUMMARY_PROMPT;
 exports.DEFAULT_SYSTEM_PROMPT = DEFAULT_SYSTEM_PROMPT;
 exports.DEFAULT_TOOL_GUIDELINE = DEFAULT_TOOL_GUIDELINE;

package/dist/constants/prompts.cjs.js.map

@@ -1 +1 @@
-{"version":3,"file":"prompts.cjs.js","sources":["../../src/constants/prompts.ts"],"sourcesContent":["export const DEFAULT_SUMMARY_PROMPT =\n  'Summarize this conversation in a concise manner. The summary should capture the main points. Return the summary only, without any additional text.';\n\nexport const DEFAULT_SYSTEM_PROMPT = `\nYou are a helpful assistant that answers questions based on provided context from various documents. The context may come from sources such as internal wikis, code repositories, technical documentation, or other structured or unstructured data.\n\nRules:\n1. Always base your answers on the provided context. Do not make up information.\n2. When relevant, cite or reference the source information provided in the context.\n3. Format answers clearly and concisely. Use bullet points for lists when appropriate.\n4. Maintain a professional, friendly, and helpful tone.\n5. Return only the relevant information without any filler or unnecessary details.\n6. If you don't know the answer, admit it and suggest ways to find the information.\n7. Always return a well-structured response using markdown.\n8. **Actively use available tools** to enhance your responses:\n9. Adapt your approach based on the specific tools and capabilities available in the current session\n10. When you have a link to an image, include it in your response using markdown format ![description](image_url).\n11. When you have a link to a document, include it in your response using markdown format [description](document_url).\n`;\n\nexport const DEFAULT_TOOL_GUIDELINE = `\nTOOL USAGE GUIDELINES:\n- Only use tools when explicitly needed to answer the user's question\n- Read tool descriptions carefully before using them\n- If you can answer without tools, do so\n- IMPORTANT: When using tools, always explain why you're using each tool\n- Use tools in logical sequence, not randomly\n- If a tool fails, try an alternative approach before using another tool\n`;\n"],"names":[],"mappings":";;AAAO,MAAM,sBAAA,GACX;AAEK,MAAM,qBAAA,GAAwB;AAAA;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAiB9B,MAAM,sBAAA,GAAyB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;;;;"}
+{"version":3,"file":"prompts.cjs.js","sources":["../../src/constants/prompts.ts"],"sourcesContent":["export const DEFAULT_CONVERSATION_SUMMARY_PROMPT =\n  \"Summarize this conversation in a concise manner. The summary should capture the main points. Return the summary only, without any additional text. Do not include any introductions or other part of the conversation that doesn't contribute to the summary or form part of the overall conversation as part of the summary.\";\n\nexport const DEFAULT_SUMMARY_PROMPT =\n  \"Summarize the following content in a concise manner. The summary should capture the main points. Return the summary only, without any additional text. Do not include any introductions or other part of the content that doesn't contribute to the summary.\";\n\nexport const DEFAULT_IDENTITY_PROMPT = `\nYou are a helpful assistant that answers questions based on provided context from various documents. The context may come from sources such as internal wikis, code repositories, technical documentation, or other structured or unstructured data.\n`;\n\nexport const DEFAULT_FORMATTING_PROMPT = `\nCRITICAL FORMATTING RULES - MUST ALWAYS FOLLOW:\n1. **ALWAYS use proper markdown formatting in ALL responses**\n2. **NEVER output plain URLs** - ALWAYS convert them to clickable markdown links using [description](url) syntax\n3. **For images, ALWAYS use markdown image syntax**: ![alt text](image-url)\n4. **For all URLs, ALWAYS format as**: [descriptive text](url) - never just paste the raw URL\n5. Use headings (##, ###), bullet points, numbered lists, and **bold**/*italic* text appropriately\n6. Format code with backticks: \\`inline code\\` or \\`\\`\\`language for code blocks\n7. Structure responses clearly with proper spacing and organization\n`;\n\nexport const DEFAULT_SYSTEM_PROMPT = `\nContent Rules:\n1. Always base your answers on the provided context. Do not make up information.\n2. When relevant, cite or reference the source information provided in the context.\n3. Maintain a professional, friendly, and helpful tone.\n4. Return only the relevant information without any filler or unnecessary details.\n5. If you don't know the answer, admit it and suggest ways to find the information.\n6. **Actively use available tools** to enhance your responses\n7. Adapt your approach based on the specific tools and capabilities available in the current session\n8. When you do not have the information needed to answer, use the tools provided to gather more context before responding.\n`;\n\nexport const DEFAULT_TOOL_GUIDELINE = `\nTOOL USAGE GUIDELINES:\n- Only use tools when explicitly needed to answer the user's question\n- Read tool descriptions carefully before using them\n- If you can answer without tools, do so\n- IMPORTANT: When using tools, always explain why you're using each tool\n- Use tools in logical sequence, not randomly\n- If a tool fails, try an alternative approach before using another tool\n`;\n"],"names":[],"mappings":";;AAAO,MAAM,mCAAA,GACX;AAEK,MAAM,sBAAA,GACX;AAEK,MAAM,uBAAA,GAA0B;AAAA;AAAA;AAIhC,MAAM,yBAAA,GAA4B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAWlC,MAAM,qBAAA,GAAwB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAY9B,MAAM,sBAAA,GAAyB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;;;;;;;"}

package/dist/database/chat-store.cjs.js

@@ -33,7 +33,9 @@ class ChatStore {
       role: row.role,
       content: row.content,
       id: row.id,
-      metadata: row.metadata
+      metadata: row.metadata,
+      score: row.score,
+      traceId: row.trace_id
     }));
     return chatMessages;
   }
@@ -44,6 +46,7 @@ class ChatStore {
       role: msg.role,
       content: msg.content,
       metadata: msg.metadata,
+      trace_id: msg.traceId,
       userRef,
       created_at: this.client.fn.now()
     }));
@@ -53,7 +56,9 @@ class ChatStore {
     await this.messageTable().where({ id: message.id }).update({
       role: message.role,
       content: message.content,
-      metadata: message.metadata
+      metadata: message.metadata,
+      score: message.score,
+      trace_id: message.traceId
     });
   }
   async getConversation(conversationId, userRef) {
@@ -90,6 +95,21 @@ class ChatStore {
     }));
     return conversations;
   }
+  async getMessageById(messageId) {
+    const row = await this.messageTable().where({ id: messageId }).first();
+    if (!row) {
+      return null;
+    }
+    const message = {
+      id: row.id,
+      role: row.role,
+      content: row.content,
+      metadata: row.metadata,
+      score: row.score,
+      traceId: row.trace_id
+    };
+    return message;
+  }
 }
 
 exports.ChatStore = ChatStore;

package/dist/database/chat-store.cjs.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat-store.cjs.js","sources":["../../src/database/chat-store.ts"],"sourcesContent":["import { DatabaseService } from '@backstage/backend-plugin-api';\nimport {\n  Message,\n  Conversation,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-common';\n\nimport { Knex } from 'knex';\n\nconst MESSAGE_TABLE_NAME = 'message';\nconst CONVERSATION_TABLE_NAME = 'conversation';\n\nexport type ChatStoreOptions = {\n  database: DatabaseService;\n};\n\nexport class ChatStore {\n  /**\n   * Creates an instance of ChatStore.\n   * @param client - The Knex client to interact with the PostgreSQL database.\n   */\n  constructor(private readonly client: Knex) {}\n\n  static async fromConfig({ database }: ChatStoreOptions) {\n    const client = await database.getClient();\n    return new ChatStore(client);\n  }\n\n  messageTable() {\n    return this.client(MESSAGE_TABLE_NAME);\n  }\n\n  conversationTable() {\n    return this.client(CONVERSATION_TABLE_NAME);\n  }\n\n  async getChatMessages(\n    conversationId: string,\n    userRef: string,\n    limit?: number,\n    excludeRoles?: Message['role'][],\n  ): Promise<Required<Message>[]> {\n    let query = this.messageTable()\n      .where({ conversation_id: conversationId, userRef })\n      .select('*')\n      .orderBy('created_at', 'asc');\n\n    if (typeof limit === 'number') {\n      query = query.limit(limit).orderBy('created_at', 'desc');\n    }\n\n    if (excludeRoles && excludeRoles.length > 0) {\n      query = query.whereNotIn('role', excludeRoles);\n    }\n\n    const rows = await query;\n\n    const chatMessages: Required<Message>[] = rows.map(row => ({\n      role: row.role,\n      content: row.content,\n      id: row.id,\n      metadata: row.metadata,\n    }));\n\n    return chatMessages;\n  }\n\n  async addChatMessage(\n    messages: Message[],\n    userRef: string,\n    conversationId: string,\n  ): Promise<void> {\n    const rows = messages.map(msg => ({\n      id: msg.id,\n      conversation_id: conversationId,\n      role: msg.role,\n      content: msg.content,\n      metadata: msg.metadata,\n      userRef,\n      created_at: this.client.fn.now(),\n    }));\n\n    await this.messageTable().insert(rows);\n  }\n\n  async updateMessage(message: Required<Message>) {\n    await this.messageTable().where({ id: message.id }).update({\n      role: message.role,\n      content: message.content,\n      metadata: message.metadata,\n    });\n  }\n\n  async getConversation(\n    conversationId: string,\n    userRef: string,\n  ): Promise<Conversation> {\n    const row = await this.conversationTable()\n      .where({ id: conversationId, userRef })\n      .first();\n\n    if (!row) {\n      throw new Error('Conversation not found');\n    }\n\n    const conversation: Conversation = {\n      id: row.id,\n      title: row.title,\n      userRef: row.userRef,\n    };\n\n    return conversation;\n  }\n\n  async createConversation(conversation: Conversation) {\n    await this.conversationTable().insert({\n      id: conversation.id,\n      title: conversation.title,\n      userRef: conversation.userRef,\n    });\n  }\n\n  async updateConversation(conversation: Conversation) {\n    await this.conversationTable().where({ id: conversation.id }).update({\n      title: conversation.title,\n      userRef: conversation.userRef,\n    });\n  }\n\n  async getConversations(userRef: string): Promise<Conversation[]> {\n    const rows = await this.conversationTable()\n      .where({ userRef })\n      .select('*')\n      .orderBy('created_at', 'desc');\n\n    const conversations: Conversation[] = rows.map(row => ({\n      id: row.id,\n      title: row.title,\n      userRef: row.userRef,\n    }));\n\n    return conversations;\n  }\n}\n"],"names":[],"mappings":";;AAQA,MAAM,kBAAA,GAAqB,SAAA;AAC3B,MAAM,uBAAA,GAA0B,cAAA;AAMzB,MAAM,SAAA,CAAU;AAAA;AAAA;AAAA;AAAA;AAAA,EAKrB,YAA6B,MAAA,EAAc;AAAd,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA;AAAA,EAAe;AAAA,EAE5C,aAAa,UAAA,CAAW,EAAE,QAAA,EAAS,EAAqB;AACtD,IAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,SAAA,EAAU;AACxC,IAAA,OAAO,IAAI,UAAU,MAAM,CAAA;AAAA,EAC7B;AAAA,EAEA,YAAA,GAAe;AACb,IAAA,OAAO,IAAA,CAAK,OAAO,kBAAkB,CAAA;AAAA,EACvC;AAAA,EAEA,iBAAA,GAAoB;AAClB,IAAA,OAAO,IAAA,CAAK,OAAO,uBAAuB,CAAA;AAAA,EAC5C;AAAA,EAEA,MAAM,eAAA,CACJ,cAAA,EACA,OAAA,EACA,OACA,YAAA,EAC8B;AAC9B,IAAA,IAAI,QAAQ,IAAA,CAAK,YAAA,EAAa,CAC3B,KAAA,CAAM,EAAE,eAAA,EAAiB,cAAA,EAAgB,OAAA,EAAS,EAClD,MAAA,CAAO,GAAG,CAAA,CACV,OAAA,CAAQ,cAAc,KAAK,CAAA;AAE9B,IAAA,IAAI,OAAO,UAAU,QAAA,EAAU;AAC7B,MAAA,KAAA,GAAQ,MAAM,KAAA,CAAM,KAAK,CAAA,CAAE,OAAA,CAAQ,cAAc,MAAM,CAAA;AAAA,IACzD;AAEA,IAAA,IAAI,YAAA,IAAgB,YAAA,CAAa,MAAA,GAAS,CAAA,EAAG;AAC3C,MAAA,KAAA,GAAQ,KAAA,CAAM,UAAA,CAAW,MAAA,EAAQ,YAAY,CAAA;AAAA,IAC/C;AAEA,IAAA,MAAM,OAAO,MAAM,KAAA;AAEnB,IAAA,MAAM,YAAA,GAAoC,IAAA,CAAK,GAAA,CAAI,CAAA,GAAA,MAAQ;AAAA,MACzD,MAAM,GAAA,CAAI,IAAA;AAAA,MACV,SAAS,GAAA,CAAI,OAAA;AAAA,MACb,IAAI,GAAA,CAAI,EAAA;AAAA,MACR,UAAU,GAAA,CAAI;AAAA,KAChB,CAAE,CAAA;AAEF,IAAA,OAAO,YAAA;AAAA,EACT;AAAA,EAEA,MAAM,cAAA,CACJ,QAAA,EACA,OAAA,EACA,cAAA,EACe;AACf,IAAA,MAAM,IAAA,GAAO,QAAA,CAAS,GAAA,CAAI,CAAA,GAAA,MAAQ;AAAA,MAChC,IAAI,GAAA,CAAI,EAAA;AAAA,MACR,eAAA,EAAiB,cAAA;AAAA,MACjB,MAAM,GAAA,CAAI,IAAA;AAAA,MACV,SAAS,GAAA,CAAI,OAAA;AAAA,MACb,UAAU,GAAA,CAAI,QAAA;AAAA,MACd,OAAA;AAAA,MACA,UAAA,EAAY,IAAA,CAAK,MAAA,CAAO,EAAA,CAAG,GAAA;AAAI,KACjC,CAAE,CAAA;AAEF,IAAA,MAAM,IAAA,CAAK,YAAA,EAAa,CAAE,MAAA,CAAO,IAAI,CAAA;AAAA,EACvC;AAAA,EAEA,MAAM,cAAc,OAAA,EAA4B;AAC9C,IAAA,MAAM,IAAA,CAAK,YAAA,EAAa,CAAE,KAAA,CAAM,EAAE,IAAI,OAAA,CAAQ,EAAA,EAAI,CAAA,CAAE,MAAA,CAAO;AAAA,MACzD,MAAM,OAAA,CAAQ,IAAA;AAAA,MACd,SAAS,OAAA,CAAQ,OAAA;AAAA,MACjB,UAAU,OAAA,CAAQ;AAAA,KACnB,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,eAAA,CACJ,cAAA,EACA,OAAA,EACuB;AACvB,IAAA,MAAM,GAAA,GAAM,MAAM,IAAA,CAAK,iBAAA,EAAkB,CACtC,KAAA,CAAM,EAAE,EAAA,EAAI,cAAA,EAAgB,OAAA,EAAS,CAAA,CACrC,KAAA,EAAM;AAET,IAAA,IAAI,CAAC,GAAA,EAAK;AACR,MAAA,MAAM,IAAI,MAAM,wBAAwB,CAAA;AAAA,IAC1C;AAEA,IAAA,MAAM,YAAA,GAA6B;AAAA,MACjC,IAAI,GAAA,CAAI,EAAA;AAAA,MACR,OAAO,GAAA,CAAI,KAAA;AAAA,MACX,SAAS,GAAA,CAAI;AAAA,KACf;AAEA,IAAA,OAAO,YAAA;AAAA,EACT;AAAA,EAEA,MAAM,mBAAmB,YAAA,EAA4B;AACnD,IAAA,MAAM,IAAA,CAAK,iBAAA,EAAkB,CAAE,MAAA,CAAO;AAAA,MACpC,IAAI,YAAA,CAAa,EAAA;AAAA,MACjB,OAAO,YAAA,CAAa,KAAA;AAAA,MACpB,SAAS,YAAA,CAAa;AAAA,KACvB,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,mBAAmB,YAAA,EAA4B;AACnD,IAAA,MAAM,IAAA,CAAK,iBAAA,EAAkB,CAAE,KAAA,CAAM,EAAE,IAAI,YAAA,CAAa,EAAA,EAAI,CAAA,CAAE,MAAA,CAAO;AAAA,MACnE,OAAO,YAAA,CAAa,KAAA;AAAA,MACpB,SAAS,YAAA,CAAa;AAAA,KACvB,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,iBAAiB,OAAA,EAA0C;AAC/D,IAAA,MAAM,IAAA,GAAO,MAAM,IAAA,CAAK,iBAAA,GACrB,KAAA,CAAM,EAAE,OAAA,EAAS,EACjB,MAAA,CAAO,GAAG,CAAA,CACV,OAAA,CAAQ,cAAc,MAAM,CAAA;AAE/B,IAAA,MAAM,aAAA,GAAgC,IAAA,CAAK,GAAA,CAAI,CAAA,GAAA,MAAQ;AAAA,MACrD,IAAI,GAAA,CAAI,EAAA;AAAA,MACR,OAAO,GAAA,CAAI,KAAA;AAAA,MACX,SAAS,GAAA,CAAI;AAAA,KACf,CAAE,CAAA;AAEF,IAAA,OAAO,aAAA;AAAA,EACT;AACF;;;;"}
+{"version":3,"file":"chat-store.cjs.js","sources":["../../src/database/chat-store.ts"],"sourcesContent":["import { DatabaseService } from '@backstage/backend-plugin-api';\nimport {\n  Message,\n  Conversation,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-common';\n\nimport { Knex } from 'knex';\n\nconst MESSAGE_TABLE_NAME = 'message';\nconst CONVERSATION_TABLE_NAME = 'conversation';\n\nexport type ChatStoreOptions = {\n  database: DatabaseService;\n};\n\nexport class ChatStore {\n  /**\n   * Creates an instance of ChatStore.\n   * @param client - The Knex client to interact with the PostgreSQL database.\n   */\n  constructor(private readonly client: Knex) {}\n\n  static async fromConfig({ database }: ChatStoreOptions) {\n    const client = await database.getClient();\n    return new ChatStore(client);\n  }\n\n  messageTable() {\n    return this.client(MESSAGE_TABLE_NAME);\n  }\n\n  conversationTable() {\n    return this.client(CONVERSATION_TABLE_NAME);\n  }\n\n  async getChatMessages(\n    conversationId: string,\n    userRef: string,\n    limit?: number,\n    excludeRoles?: Message['role'][],\n  ): Promise<Required<Message>[]> {\n    let query = this.messageTable()\n      .where({ conversation_id: conversationId, userRef })\n      .select('*')\n      .orderBy('created_at', 'asc');\n\n    if (typeof limit === 'number') {\n      query = query.limit(limit).orderBy('created_at', 'desc');\n    }\n\n    if (excludeRoles && excludeRoles.length > 0) {\n      query = query.whereNotIn('role', excludeRoles);\n    }\n\n    const rows = await query;\n\n    const chatMessages: Required<Message>[] = rows.map(row => ({\n      role: row.role,\n      content: row.content,\n      id: row.id,\n      metadata: row.metadata,\n      score: row.score,\n      traceId: row.trace_id,\n    }));\n\n    return chatMessages;\n  }\n\n  async addChatMessage(\n    messages: Message[],\n    userRef: string,\n    conversationId: string,\n  ): Promise<void> {\n    const rows = messages.map(msg => ({\n      id: msg.id,\n      conversation_id: conversationId,\n      role: msg.role,\n      content: msg.content,\n      metadata: msg.metadata,\n      trace_id: msg.traceId,\n      userRef,\n      created_at: this.client.fn.now(),\n    }));\n\n    await this.messageTable().insert(rows);\n  }\n\n  async updateMessage(message: Required<Message>) {\n    await this.messageTable().where({ id: message.id }).update({\n      role: message.role,\n      content: message.content,\n      metadata: message.metadata,\n      score: message.score,\n      trace_id: message.traceId,\n    });\n  }\n\n  async getConversation(\n    conversationId: string,\n    userRef: string,\n  ): Promise<Conversation> {\n    const row = await this.conversationTable()\n      .where({ id: conversationId, userRef })\n      .first();\n\n    if (!row) {\n      throw new Error('Conversation not found');\n    }\n\n    const conversation: Conversation = {\n      id: row.id,\n      title: row.title,\n      userRef: row.userRef,\n    };\n\n    return conversation;\n  }\n\n  async createConversation(conversation: Conversation) {\n    await this.conversationTable().insert({\n      id: conversation.id,\n      title: conversation.title,\n      userRef: conversation.userRef,\n    });\n  }\n\n  async updateConversation(conversation: Conversation) {\n    await this.conversationTable().where({ id: conversation.id }).update({\n      title: conversation.title,\n      userRef: conversation.userRef,\n    });\n  }\n\n  async getConversations(userRef: string): Promise<Conversation[]> {\n    const rows = await this.conversationTable()\n      .where({ userRef })\n      .select('*')\n      .orderBy('created_at', 'desc');\n\n    const conversations: Conversation[] = rows.map(row => ({\n      id: row.id,\n      title: row.title,\n      userRef: row.userRef,\n    }));\n\n    return conversations;\n  }\n\n  async getMessageById(messageId: string): Promise<Required<Message> | null> {\n    const row = await this.messageTable().where({ id: messageId }).first();\n\n    if (!row) {\n      return null;\n    }\n\n    const message: Required<Message> = {\n      id: row.id,\n      role: row.role,\n      content: row.content,\n      metadata: row.metadata,\n      score: row.score,\n      traceId: row.trace_id,\n    };\n\n    return message;\n  }\n}\n"],"names":[],"mappings":";;AAQA,MAAM,kBAAA,GAAqB,SAAA;AAC3B,MAAM,uBAAA,GAA0B,cAAA;AAMzB,MAAM,SAAA,CAAU;AAAA;AAAA;AAAA;AAAA;AAAA,EAKrB,YAA6B,MAAA,EAAc;AAAd,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA;AAAA,EAAe;AAAA,EAE5C,aAAa,UAAA,CAAW,EAAE,QAAA,EAAS,EAAqB;AACtD,IAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,SAAA,EAAU;AACxC,IAAA,OAAO,IAAI,UAAU,MAAM,CAAA;AAAA,EAC7B;AAAA,EAEA,YAAA,GAAe;AACb,IAAA,OAAO,IAAA,CAAK,OAAO,kBAAkB,CAAA;AAAA,EACvC;AAAA,EAEA,iBAAA,GAAoB;AAClB,IAAA,OAAO,IAAA,CAAK,OAAO,uBAAuB,CAAA;AAAA,EAC5C;AAAA,EAEA,MAAM,eAAA,CACJ,cAAA,EACA,OAAA,EACA,OACA,YAAA,EAC8B;AAC9B,IAAA,IAAI,QAAQ,IAAA,CAAK,YAAA,EAAa,CAC3B,KAAA,CAAM,EAAE,eAAA,EAAiB,cAAA,EAAgB,OAAA,EAAS,EAClD,MAAA,CAAO,GAAG,CAAA,CACV,OAAA,CAAQ,cAAc,KAAK,CAAA;AAE9B,IAAA,IAAI,OAAO,UAAU,QAAA,EAAU;AAC7B,MAAA,KAAA,GAAQ,MAAM,KAAA,CAAM,KAAK,CAAA,CAAE,OAAA,CAAQ,cAAc,MAAM,CAAA;AAAA,IACzD;AAEA,IAAA,IAAI,YAAA,IAAgB,YAAA,CAAa,MAAA,GAAS,CAAA,EAAG;AAC3C,MAAA,KAAA,GAAQ,KAAA,CAAM,UAAA,CAAW,MAAA,EAAQ,YAAY,CAAA;AAAA,IAC/C;AAEA,IAAA,MAAM,OAAO,MAAM,KAAA;AAEnB,IAAA,MAAM,YAAA,GAAoC,IAAA,CAAK,GAAA,CAAI,CAAA,GAAA,MAAQ;AAAA,MACzD,MAAM,GAAA,CAAI,IAAA;AAAA,MACV,SAAS,GAAA,CAAI,OAAA;AAAA,MACb,IAAI,GAAA,CAAI,EAAA;AAAA,MACR,UAAU,GAAA,CAAI,QAAA;AAAA,MACd,OAAO,GAAA,CAAI,KAAA;AAAA,MACX,SAAS,GAAA,CAAI;AAAA,KACf,CAAE,CAAA;AAEF,IAAA,OAAO,YAAA;AAAA,EACT;AAAA,EAEA,MAAM,cAAA,CACJ,QAAA,EACA,OAAA,EACA,cAAA,EACe;AACf,IAAA,MAAM,IAAA,GAAO,QAAA,CAAS,GAAA,CAAI,CAAA,GAAA,MAAQ;AAAA,MAChC,IAAI,GAAA,CAAI,EAAA;AAAA,MACR,eAAA,EAAiB,cAAA;AAAA,MACjB,MAAM,GAAA,CAAI,IAAA;AAAA,MACV,SAAS,GAAA,CAAI,OAAA;AAAA,MACb,UAAU,GAAA,CAAI,QAAA;AAAA,MACd,UAAU,GAAA,CAAI,OAAA;AAAA,MACd,OAAA;AAAA,MACA,UAAA,EAAY,IAAA,CAAK,MAAA,CAAO,EAAA,CAAG,GAAA;AAAI,KACjC,CAAE,CAAA;AAEF,IAAA,MAAM,IAAA,CAAK,YAAA,EAAa,CAAE,MAAA,CAAO,IAAI,CAAA;AAAA,EACvC;AAAA,EAEA,MAAM,cAAc,OAAA,EAA4B;AAC9C,IAAA,MAAM,IAAA,CAAK,YAAA,EAAa,CAAE,KAAA,CAAM,EAAE,IAAI,OAAA,CAAQ,EAAA,EAAI,CAAA,CAAE,MAAA,CAAO;AAAA,MACzD,MAAM,OAAA,CAAQ,IAAA;AAAA,MACd,SAAS,OAAA,CAAQ,OAAA;AAAA,MACjB,UAAU,OAAA,CAAQ,QAAA;AAAA,MAClB,OAAO,OAAA,CAAQ,KAAA;AAAA,MACf,UAAU,OAAA,CAAQ;AAAA,KACnB,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,eAAA,CACJ,cAAA,EACA,OAAA,EACuB;AACvB,IAAA,MAAM,GAAA,GAAM,MAAM,IAAA,CAAK,iBAAA,EAAkB,CACtC,KAAA,CAAM,EAAE,EAAA,EAAI,cAAA,EAAgB,OAAA,EAAS,CAAA,CACrC,KAAA,EAAM;AAET,IAAA,IAAI,CAAC,GAAA,EAAK;AACR,MAAA,MAAM,IAAI,MAAM,wBAAwB,CAAA;AAAA,IAC1C;AAEA,IAAA,MAAM,YAAA,GAA6B;AAAA,MACjC,IAAI,GAAA,CAAI,EAAA;AAAA,MACR,OAAO,GAAA,CAAI,KAAA;AAAA,MACX,SAAS,GAAA,CAAI;AAAA,KACf;AAEA,IAAA,OAAO,YAAA;AAAA,EACT;AAAA,EAEA,MAAM,mBAAmB,YAAA,EAA4B;AACnD,IAAA,MAAM,IAAA,CAAK,iBAAA,EAAkB,CAAE,MAAA,CAAO;AAAA,MACpC,IAAI,YAAA,CAAa,EAAA;AAAA,MACjB,OAAO,YAAA,CAAa,KAAA;AAAA,MACpB,SAAS,YAAA,CAAa;AAAA,KACvB,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,mBAAmB,YAAA,EAA4B;AACnD,IAAA,MAAM,IAAA,CAAK,iBAAA,EAAkB,CAAE,KAAA,CAAM,EAAE,IAAI,YAAA,CAAa,EAAA,EAAI,CAAA,CAAE,MAAA,CAAO;AAAA,MACnE,OAAO,YAAA,CAAa,KAAA;AAAA,MACpB,SAAS,YAAA,CAAa;AAAA,KACvB,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,iBAAiB,OAAA,EAA0C;AAC/D,IAAA,MAAM,IAAA,GAAO,MAAM,IAAA,CAAK,iBAAA,GACrB,KAAA,CAAM,EAAE,OAAA,EAAS,EACjB,MAAA,CAAO,GAAG,CAAA,CACV,OAAA,CAAQ,cAAc,MAAM,CAAA;AAE/B,IAAA,MAAM,aAAA,GAAgC,IAAA,CAAK,GAAA,CAAI,CAAA,GAAA,MAAQ;AAAA,MACrD,IAAI,GAAA,CAAI,EAAA;AAAA,MACR,OAAO,GAAA,CAAI,KAAA;AAAA,MACX,SAAS,GAAA,CAAI;AAAA,KACf,CAAE,CAAA;AAEF,IAAA,OAAO,aAAA;AAAA,EACT;AAAA,EAEA,MAAM,eAAe,SAAA,EAAsD;AACzE,IAAA,MAAM,GAAA,GAAM,MAAM,IAAA,CAAK,YAAA,EAAa,CAAE,KAAA,CAAM,EAAE,EAAA,EAAI,SAAA,EAAW,CAAA,CAAE,KAAA,EAAM;AAErE,IAAA,IAAI,CAAC,GAAA,EAAK;AACR,MAAA,OAAO,IAAA;AAAA,IACT;AAEA,IAAA,MAAM,OAAA,GAA6B;AAAA,MACjC,IAAI,GAAA,CAAI,EAAA;AAAA,MACR,MAAM,GAAA,CAAI,IAAA;AAAA,MACV,SAAS,GAAA,CAAI,OAAA;AAAA,MACb,UAAU,GAAA,CAAI,QAAA;AAAA,MACd,OAAO,GAAA,CAAI,KAAA;AAAA,MACX,SAAS,GAAA,CAAI;AAAA,KACf;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AACF;;;;"}

package/dist/database/mcp-store.cjs.js

@@ -0,0 +1,46 @@
+'use strict';
+
+const MCP_TABLE_NAME = "user_mcp_config";
+class McpStore {
+  /**
+   * Creates an instance of ChatStore.
+   * @param client - The Knex client to interact with the PostgreSQL database.
+   */
+  constructor(client) {
+    this.client = client;
+  }
+  static async fromConfig({ database }) {
+    const client = await database.getClient();
+    return new McpStore(client);
+  }
+  mcpTable() {
+    return this.client(MCP_TABLE_NAME);
+  }
+  async getUserUserMcpConfigNames(userRef) {
+    const rows = await this.mcpTable().where({ userRef }).select("name");
+    return rows.map((row) => row.name);
+  }
+  async getUserMcpConfigs(userRef) {
+    const rows = await this.mcpTable().where({ userRef }).select();
+    return rows.map((row) => ({
+      name: row.name,
+      encryptedOptions: row.encryptedOptions
+    }));
+  }
+  async createUserMcpConfig(userRef, name, encryptedOptions) {
+    await this.mcpTable().insert({
+      userRef,
+      name,
+      encryptedOptions
+    });
+  }
+  async updateUserMcpConfig(userRef, name, encryptedOptions) {
+    await this.mcpTable().where({ userRef, name }).update({ encryptedOptions });
+  }
+  async deleteUserMcpConfig(userRef, name) {
+    await this.mcpTable().where({ userRef, name }).delete();
+  }
+}
+
+exports.McpStore = McpStore;
+//# sourceMappingURL=mcp-store.cjs.js.map

package/dist/database/mcp-store.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-store.cjs.js","sources":["../../src/database/mcp-store.ts"],"sourcesContent":["import { DatabaseService } from '@backstage/backend-plugin-api';\n\nimport { Knex } from 'knex';\n\nconst MCP_TABLE_NAME = 'user_mcp_config';\n\nexport type McpStoreOptions = {\n  database: DatabaseService;\n};\n\nexport class McpStore {\n  /**\n   * Creates an instance of ChatStore.\n   * @param client - The Knex client to interact with the PostgreSQL database.\n   */\n  constructor(private readonly client: Knex) {}\n\n  static async fromConfig({ database }: McpStoreOptions) {\n    const client = await database.getClient();\n    return new McpStore(client);\n  }\n\n  mcpTable() {\n    return this.client(MCP_TABLE_NAME);\n  }\n\n  async getUserUserMcpConfigNames(userRef: string): Promise<string[]> {\n    const rows = await this.mcpTable().where({ userRef }).select('name');\n    return rows.map(row => row.name);\n  }\n\n  async getUserMcpConfigs(\n    userRef: string,\n  ): Promise<Array<{ name: string; encryptedOptions: string }>> {\n    const rows = await this.mcpTable().where({ userRef }).select();\n    return rows.map(row => ({\n      name: row.name,\n      encryptedOptions: row.encryptedOptions,\n    }));\n  }\n\n  async createUserMcpConfig(\n    userRef: string,\n    name: string,\n    encryptedOptions: string,\n  ): Promise<void> {\n    await this.mcpTable().insert({\n      userRef,\n      name,\n      encryptedOptions,\n    });\n  }\n\n  async updateUserMcpConfig(\n    userRef: string,\n    name: string,\n    encryptedOptions: string,\n  ): Promise<void> {\n    await this.mcpTable().where({ userRef, name }).update({ encryptedOptions });\n  }\n\n  async deleteUserMcpConfig(userRef: string, name: string): Promise<void> {\n    await this.mcpTable().where({ userRef, name }).delete();\n  }\n}\n"],"names":[],"mappings":";;AAIA,MAAM,cAAA,GAAiB,iBAAA;AAMhB,MAAM,QAAA,CAAS;AAAA;AAAA;AAAA;AAAA;AAAA,EAKpB,YAA6B,MAAA,EAAc;AAAd,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA;AAAA,EAAe;AAAA,EAE5C,aAAa,UAAA,CAAW,EAAE,QAAA,EAAS,EAAoB;AACrD,IAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,SAAA,EAAU;AACxC,IAAA,OAAO,IAAI,SAAS,MAAM,CAAA;AAAA,EAC5B;AAAA,EAEA,QAAA,GAAW;AACT,IAAA,OAAO,IAAA,CAAK,OAAO,cAAc,CAAA;AAAA,EACnC;AAAA,EAEA,MAAM,0BAA0B,OAAA,EAAoC;AAClE,IAAA,MAAM,IAAA,GAAO,MAAM,IAAA,CAAK,QAAA,EAAS,CAAE,KAAA,CAAM,EAAE,OAAA,EAAS,CAAA,CAAE,MAAA,CAAO,MAAM,CAAA;AACnE,IAAA,OAAO,IAAA,CAAK,GAAA,CAAI,CAAA,GAAA,KAAO,GAAA,CAAI,IAAI,CAAA;AAAA,EACjC;AAAA,EAEA,MAAM,kBACJ,OAAA,EAC4D;AAC5D,IAAA,MAAM,IAAA,GAAO,MAAM,IAAA,CAAK,QAAA,EAAS,CAAE,MAAM,EAAE,OAAA,EAAS,CAAA,CAAE,MAAA,EAAO;AAC7D,IAAA,OAAO,IAAA,CAAK,IAAI,CAAA,GAAA,MAAQ;AAAA,MACtB,MAAM,GAAA,CAAI,IAAA;AAAA,MACV,kBAAkB,GAAA,CAAI;AAAA,KACxB,CAAE,CAAA;AAAA,EACJ;AAAA,EAEA,MAAM,mBAAA,CACJ,OAAA,EACA,IAAA,EACA,gBAAA,EACe;AACf,IAAA,MAAM,IAAA,CAAK,QAAA,EAAS,CAAE,MAAA,CAAO;AAAA,MAC3B,OAAA;AAAA,MACA,IAAA;AAAA,MACA;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,mBAAA,CACJ,OAAA,EACA,IAAA,EACA,gBAAA,EACe;AACf,IAAA,MAAM,IAAA,CAAK,QAAA,EAAS,CAAE,KAAA,CAAM,EAAE,OAAA,EAAS,IAAA,EAAM,CAAA,CAAE,MAAA,CAAO,EAAE,gBAAA,EAAkB,CAAA;AAAA,EAC5E;AAAA,EAEA,MAAM,mBAAA,CAAoB,OAAA,EAAiB,IAAA,EAA6B;AACtE,IAAA,MAAM,IAAA,CAAK,UAAS,CAAE,KAAA,CAAM,EAAE,OAAA,EAAS,IAAA,EAAM,CAAA,CAAE,MAAA,EAAO;AAAA,EACxD;AACF;;;;"}

package/dist/plugin.cjs.js

@@ -10,6 +10,9 @@ var pgVectorStore = require('./database/pg-vector-store.cjs.js');
 var pluginSignalsNode = require('@backstage/plugin-signals-node');
 var searchKnowledge = require('./services/tools/searchKnowledge.cjs.js');
 var pluginCatalogNode = require('@backstage/plugin-catalog-node');
+var mcp = require('./services/mcp.cjs.js');
+var callbacks = require('./services/callbacks.cjs.js');
+var summarizer = require('./services/summarizer.cjs.js');
 
 const aiAssistantPlugin = backendPluginApi.createBackendPlugin({
   pluginId: "ai-assistant",
@@ -17,6 +20,7 @@ const aiAssistantPlugin = backendPluginApi.createBackendPlugin({
     const ingestors = [];
     const models = [];
     const tools = [];
+    const callbacks$1 = [];
     let embeddingsProvider;
     env.registerExtensionPoint(backstagePluginAiAssistantNode.dataIngestorExtensionPoint, {
       registerIngestor: (ingestor) => {
@@ -52,6 +56,11 @@ const aiAssistantPlugin = backendPluginApi.createBackendPlugin({
         tools.push(tool);
       }
     });
+    env.registerExtensionPoint(backstagePluginAiAssistantNode.callbackProviderExtensionPoint, {
+      register: (callbackProvider) => {
+        callbacks$1.push(callbackProvider);
+      }
+    });
     env.registerInit({
       deps: {
         httpRouter: backendPluginApi.coreServices.httpRouter,
@@ -67,7 +76,7 @@ const aiAssistantPlugin = backendPluginApi.createBackendPlugin({
         auth: backendPluginApi.coreServices.auth
       },
       async init(options) {
-        const { httpRouter, database } = options;
+        const { httpRouter, database, config } = options;
         const client = await database.getClient();
         await migrations.applyDatabaseMigrations(client);
         const vectorStore = await pgVectorStore.PgVectorStore.fromConfig(options);
@@ -80,14 +89,28 @@ const aiAssistantPlugin = backendPluginApi.createBackendPlugin({
           vectorStore,
           ingestors
         });
+        const mcp$1 = await mcp.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
+          tools,
+          mcp: mcp$1,
+          callback,
+          summarizer: summarizer$1
         });
-        httpRouter.use(await index.createRouter({ ...options, chat: chat$1 }));
+        httpRouter.use(
+          await index.createRouter({ ...options, chat: chat$1, mcp: mcp$1, summarizer: summarizer$1 })
+        );
         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  Tool,\n  toolExtensionPoint,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-node';\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 './services/tools/searchKnowledge';\nimport { catalogServiceRef } from '@backstage/plugin-catalog-node';\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\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.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 } = options;\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        tools.push(searchKnowledgeTool);\n\n        const chat = await createChatService({\n          ...options,\n          models,\n          tools,\n        });\n\n        httpRouter.use(await createRouter({ ...options, chat }));\n        dataIngestionPipeline.start();\n      },\n    });\n  },\n});\n"],"names":["createBackendPlugin","dataIngestorExtensionPoint","embeddingsProviderExtensionPoint","modelProviderExtensionPoint","toolExtensionPoint","coreServices","signalsServiceRef","catalogServiceRef","applyDatabaseMigrations","PgVectorStore","createDataIngestionPipeline","createSearchKnowledgeTool","chat","createChatService","createRouter"],"mappings":";;;;;;;;;;;;;AA6BO,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;AAEvB,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,YAAA,CAAa;AAAA,MACf,IAAA,EAAM;AAAA,QACJ,YAAYC,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,EAAS,GAAI,OAAA;AACjC,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,mBAAA,GAAsBC,yCAAA,CAA0B,EAAE,WAAA,EAAa,CAAA;AACrE,QAAA,KAAA,CAAM,KAAK,mBAAmB,CAAA;AAE9B,QAAA,MAAMC,MAAA,GAAO,MAAMC,sBAAA,CAAkB;AAAA,UACnC,GAAG,OAAA;AAAA,UACH,MAAA;AAAA,UACA;AAAA,SACD,CAAA;AAED,QAAA,UAAA,CAAW,GAAA,CAAI,MAAMC,kBAAA,CAAa,EAAE,GAAG,OAAA,QAASF,MAAA,EAAM,CAAC,CAAA;AACvD,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  Tool,\n  toolExtensionPoint,\n  callbackProviderExtensionPoint,\n  CallbackProvider,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-node';\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 './services/tools/searchKnowledge';\nimport { catalogServiceRef } from '@backstage/plugin-catalog-node';\nimport { createMcpService } from './services/mcp';\nimport { createCallbackService } from './services/callbacks';\nimport { createSummarizerService } from './services/summarizer';\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        httpRouter.use(\n          await createRouter({ ...options, chat, mcp, summarizer }),\n        );\n        dataIngestionPipeline.start();\n      },\n    });\n  },\n});\n"],"names":["createBackendPlugin","callbacks","dataIngestorExtensionPoint","embeddingsProviderExtensionPoint","modelProviderExtensionPoint","toolExtensionPoint","callbackProviderExtensionPoint","coreServices","signalsServiceRef","catalogServiceRef","applyDatabaseMigrations","PgVectorStore","createDataIngestionPipeline","mcp","createMcpService","createSearchKnowledgeTool","createCallbackService","summarizer","createSummarizerService","chat","createChatService","createRouter"],"mappings":";;;;;;;;;;;;;;;;AAiCO,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,MAAMC,KAAA,GAAM,MAAMC,oBAAA,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,qBAC3Cf;AAAA,SACD,CAAA;AAED,QAAA,MAAMgB,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,eACAP,KAAA;AAAA,UACA,QAAA;AAAA,sBACAI;AAAA,SACD,CAAA;AAED,QAAA,UAAA,CAAW,GAAA;AAAA,UACT,MAAMI,mBAAa,EAAE,GAAG,eAASF,MAAA,OAAMN,KAAA,cAAKI,cAAY;AAAA,SAC1D;AACA,QAAA,qBAAA,CAAsB,KAAA,EAAM;AAAA,MAC9B;AAAA,KACD,CAAA;AAAA,EACH;AACF,CAAC;;;;"}

package/dist/services/callbacks.cjs.js

@@ -0,0 +1,49 @@
+'use strict';
+
+const createCallbackService = async ({
+  callbacks
+}) => {
+  const getChainCallbacks = async (options) => {
+    const callbackHandlers = [];
+    for (const { chainCallback } of callbacks) {
+      if (!chainCallback) {
+        continue;
+      }
+      const callbackHandler = await chainCallback(options);
+      callbackHandlers.push(callbackHandler);
+    }
+    return {
+      callbacks: callbackHandlers
+    };
+  };
+  const getChainMetadata = async (options) => {
+    const metadata = {};
+    for (const { chainMetadataCallback } of callbacks) {
+      if (!chainMetadataCallback) {
+        continue;
+      }
+      const callbackData = await chainMetadataCallback(options);
+      Object.assign(metadata, callbackData.metadata);
+    }
+    return { metadata };
+  };
+  const handleScoreCallbacks = async ({
+    name,
+    message
+  }) => {
+    callbacks.forEach(async ({ scoreCallback }) => {
+      if (!scoreCallback) {
+        return;
+      }
+      scoreCallback({ name, message });
+    });
+  };
+  return {
+    getChainCallbacks,
+    getChainMetadata,
+    handleScoreCallbacks
+  };
+};
+
+exports.createCallbackService = createCallbackService;
+//# sourceMappingURL=callbacks.cjs.js.map

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

@@ -0,0 +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;;;;"}

package/dist/services/chat.cjs.js

@@ -2,10 +2,10 @@
 
 var chatStore = require('../database/chat-store.cjs.js');
 var prompts = require('../constants/prompts.cjs.js');
+var backstagePluginAiAssistantNode = require('@sweetoburrito/backstage-plugin-ai-assistant-node');
 var tools = require('@langchain/core/tools');
 var prebuilt = require('@langchain/langgraph/prebuilt');
 var prompts$1 = require('@langchain/core/prompts');
-var summarizer = require('./summarizer.cjs.js');
 var uuid = require('uuid');
 
 const createChatService = async ({
@@ -17,14 +17,24 @@ const createChatService = async ({
   config,
   catalog,
   cache,
-  auth
+  auth,
+  mcp,
+  userInfo,
+  callback,
+  summarizer
 }) => {
   logger.info(`Available models: ${models.map((m) => m.id).join(", ")}`);
-  const system = config.getOptionalString("aiAssistant.prompt.system") || prompts.DEFAULT_SYSTEM_PROMPT;
+  logger.info(`Available tools: ${tools$1.map((t) => t.name).join(", ")}`);
+  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 combinedBasePrompt = `${identityPrompt}
+
+${formattingPrompt}
+
+${contentPrompt}`;
   const toolGuideline = config.getOptionalString("aiAssistant.prompt.toolGuideline") || prompts.DEFAULT_TOOL_GUIDELINE;
   const chatStore$1 = await chatStore.ChatStore.fromConfig({ database });
-  const summarizer$1 = await summarizer.createSummarizerService({ config, models });
-  const agentTools = tools$1.map((tool) => new tools.DynamicStructuredTool(tool));
   const systemPromptTemplate = prompts$1.SystemMessagePromptTemplate.fromTemplate(`
     PURPOSE:
     {basePrompt}
@@ -73,7 +83,10 @@ const createChatService = async ({
       chatStore$1.addChatMessage(messages, userRef, conversationId);
       return;
     }
-    const summary = await summarizer$1.summarize(recentMessages, "25 characters");
+    const summary = await summarizer.summarizeConversation({
+      messages: recentMessages,
+      length: "25 characters"
+    });
     conversation.title = summary;
     chatStore$1.updateConversation(conversation);
     chatStore$1.addChatMessage(messages, userRef, conversationId);
@@ -91,12 +104,13 @@ const createChatService = async ({
     messages,
     modelId,
     stream = true,
-    userEntityRef
+    userCredentials
   }) => {
     const model = models.find((m) => m.id === modelId)?.chatModel;
     if (!model) {
       throw new Error(`Model with id ${modelId} not found`);
     }
+    const { userEntityRef } = await userInfo.getUserInfo(userCredentials);
     const streamFn = async () => {
       const recentConversationMessages = await chatStore$1.getChatMessages(
         conversationId,
@@ -105,15 +119,18 @@ const createChatService = async ({
         ["tool"]
       );
       const credentials = await auth.getOwnServiceCredentials();
-      const user = await getUser(cache, userEntityRef, credentials, catalog);
+      const user = await backstagePluginAiAssistantNode.getUser(cache, userEntityRef, credentials, catalog);
+      const mcpTools = await mcp.getTools(userCredentials);
+      const agentTools = tools$1.map((tool) => new tools.DynamicStructuredTool(tool)).concat(mcpTools.map((tool) => new tools.DynamicStructuredTool(tool)));
+      const messagesWithoutSystem = messages.filter((m) => m.role !== "system");
       addMessages(
-        messages,
+        messagesWithoutSystem,
         userEntityRef,
         conversationId,
         recentConversationMessages
       );
       const systemPrompt = await systemPromptTemplate.formatMessages({
-        basePrompt: system,
+        basePrompt: combinedBasePrompt,
         toolGuideline,
         toolList: agentTools.map((tool) => `- ${tool.name}: ${tool.description}`).join("\n"),
         context: `none`,
@@ -124,22 +141,45 @@ const createChatService = async ({
         tools: agentTools,
         prompt: systemPrompt[0].text
       });
+      const { callbacks } = await callback.getChainCallbacks({
+        conversationId,
+        userId: userEntityRef,
+        modelId
+      });
+      const { metadata: promptMetadata } = await callback.getChainMetadata({
+        conversationId,
+        userId: userEntityRef,
+        modelId
+      });
+      const traceId = uuid.v4();
       const promptStream = await agent.stream(
         {
           messages: [...recentConversationMessages, ...messages]
         },
-        { streamMode: ["values"] }
+        {
+          streamMode: ["values"],
+          runName: "ai-assistant-chat",
+          runId: traceId,
+          metadata: promptMetadata,
+          callbacks
+        }
       );
       const responseMessages = [];
       for await (const [, chunk] of promptStream) {
         const { messages: promptMessages } = chunk;
-        const newMessages = promptMessages.filter((m) => responseMessages.findIndex((rm) => rm.id === m.id) === -1).filter(
+        const newMessages = promptMessages.filter(
+          (m) => responseMessages.findIndex(
+            (rm) => m.id === rm.metadata.langGraphId
+          ) === -1
+        ).filter(
           (m) => recentConversationMessages.findIndex((rm) => rm.id === m.id) === -1
         ).filter((m) => m.getType() !== "human").map((m) => {
-          const id = m.id ?? "";
+          const id = uuid.v4();
           const role = m.getType();
           const content = typeof m.content === "string" ? m.content : JSON.stringify(m.content);
-          const metadata = {};
+          const metadata = {
+            langGraphId: m.id ?? ""
+          };
           if (role === "ai") {
             const aiMessage = m;
             metadata.toolCalls = aiMessage.tool_calls || [];
@@ -154,14 +194,18 @@ const createChatService = async ({
             id,
             role,
             content,
-            metadata
+            metadata,
+            score: 0,
+            traceId
           };
         });
         for await (const m of newMessages) {
-          const parts = m.content.split(" ");
+          const words = m.content.split(" ");
+          const chunkSize = 5;
           let messageBuilder = "";
-          for await (const part of parts) {
-            messageBuilder = messageBuilder.concat(part).concat(" ");
+          for (let i = 0; i < words.length; i += chunkSize) {
+            const wordChunk = words.slice(i, i + chunkSize).join(" ");
+            messageBuilder = messageBuilder.concat(wordChunk).concat(" ");
             m.content = messageBuilder;
             await new Promise((resolve) => setTimeout(resolve, 50));
             signals.publish({
@@ -176,9 +220,6 @@ const createChatService = async ({
         }
         responseMessages.push(...newMessages);
       }
-      responseMessages.forEach((m) => {
-        m.id = uuid.v4();
-      });
       addMessages(responseMessages, userEntityRef, conversationId, [
         ...recentConversationMessages,
         ...messages
@@ -205,25 +246,30 @@ const createChatService = async ({
     const conversations = await chatStore$1.getConversations(userEntityRef);
     return conversations;
   };
+  const scoreMessage = async (messageId, score) => {
+    const message = await chatStore$1.getMessageById(messageId);
+    if (!message) {
+      throw new Error(`Message with id ${messageId} not found`);
+    }
+    const updatedMessage = {
+      ...message,
+      score
+    };
+    chatStore$1.updateMessage(updatedMessage);
+    callback.handleScoreCallbacks({
+      name: "helpfulness",
+      message: updatedMessage
+    });
+  };
   return {
     prompt,
     getAvailableModels,
     getConversation,
     getConversations,
-    addMessages
+    addMessages,
+    scoreMessage
   };
 };
-async function getUser(cache, userEntityRef, credentials, catalog) {
-  const cached = await cache.get(userEntityRef);
-  if (cached) {
-    return JSON.parse(String(cached));
-  }
-  const user = await catalog.getEntityByRef(userEntityRef, {
-    credentials
-  });
-  await cache.set(userEntityRef, JSON.stringify(user));
-  return user;
-}
 
 exports.createChatService = createChatService;
 //# sourceMappingURL=chat.cjs.js.map