@sweetoburrito/backstage-plugin-ai-assistant-backend 0.9.0 → 0.11.0

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

@@ -1 +1 @@
-{"version":3,"file":"searchKnowledge.cjs.js","sources":["../../../src/services/tools/searchKnowledge.ts"],"sourcesContent":["import {\n  createAssistantTool,\n  Tool,\n  VectorStore,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-node';\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      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        if (results.length === 0) {\n          return 'No relevant information found.';\n        }\n        return results.map(r => r.content).join('\\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,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;AACA,QAAA,IAAI,OAAA,CAAQ,WAAW,CAAA,EAAG;AACxB,UAAA,OAAO,gCAAA;AAAA,QACT;AACA,QAAA,OAAO,QAAQ,GAAA,CAAI,CAAA,CAAA,KAAK,EAAE,OAAO,CAAA,CAAE,KAAK,SAAS,CAAA;AAAA,MACnD;AAAA;AACF,GACD,CAAA;AAED,EAAA,OAAO,aAAA;AACT;;;;"}
+{"version":3,"file":"searchKnowledge.cjs.js","sources":["../../../src/services/tools/searchKnowledge.ts"],"sourcesContent":["import {\n  createAssistantTool,\n  Tool,\n  VectorStore,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-node';\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      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        console.log(results);\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,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,OAAA,CAAQ,IAAI,OAAO,CAAA;AAEnB,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;;;;"}

package/migrations/20251029_mcp.js

@@ -0,0 +1,42 @@
+const mcpConfigTable = 'user_mcp_config';
+
+/**
+ *
+ * @param {import('knex').knex} knex
+ */
+
+exports.down = async knex => {
+  const hasMcpConfigTable = await knex.schema.hasTable(mcpConfigTable);
+
+  if (hasMcpConfigTable) {
+    await knex.schema.dropTable(mcpConfigTable);
+  }
+};
+
+/**
+ *
+ * @param {import('knex').knex} knex
+ */
+
+exports.up = async knex => {
+  await knex.schema.createTable(mcpConfigTable, table => {
+    table
+      .uuid('id')
+      .primary()
+      .notNullable()
+      .defaultTo(knex.raw('gen_random_uuid()'));
+    table
+      .text('userRef')
+      .notNullable()
+      .comment('Reference to the user who sent the message');
+    table
+      .text('name')
+      .notNullable()
+      .comment('Name of the MCP server configuration');
+    table
+      .text('encryptedOptions')
+      .notNullable()
+      .comment('Encrypted MCP server configuration');
+    table.timestamps(true, true);
+  });
+};

package/migrations/20251029_score.js

@@ -0,0 +1,52 @@
+/**
+ *
+ * @param {import('knex').knex} knex
+ */
+
+exports.down = async knex => {
+  const hasScoreColumn = await knex.schema.hasColumn('message', 'score');
+  const hasTraceIdColumn = await knex.schema.hasColumn('message', 'trace_id');
+
+  if (hasScoreColumn || hasTraceIdColumn) {
+    await knex.schema.alterTable('message', table => {
+      if (hasScoreColumn) {
+        table.dropColumn('score');
+      }
+      if (hasTraceIdColumn) {
+        table.dropColumn('trace_id');
+      }
+    });
+  }
+};
+
+/**
+ *
+ * @param {import('knex').knex} knex
+ */
+
+exports.up = async knex => {
+  const hasScoreColumn = await knex.schema.hasColumn('message', 'score');
+  const hasTraceIdColumn = await knex.schema.hasColumn('message', 'trace_id');
+
+  if (!hasScoreColumn || !hasTraceIdColumn) {
+    await knex.schema.alterTable('message', table => {
+      if (!hasScoreColumn) {
+        table
+          .float('score')
+          .nullable()
+          .defaultTo(0)
+          .comment(
+            'User feedback score for the message: 0 = no feedback, 1 = helpful, -1 = not helpful',
+          );
+      }
+      if (!hasTraceIdColumn) {
+        table
+          .string('trace_id')
+          .nullable()
+          .comment(
+            'Langfuse trace ID for the message, used for scoring and observability',
+          );
+      }
+    });
+  }
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sweetoburrito/backstage-plugin-ai-assistant-backend",
-  "version": "0.9.0",
+  "version": "0.11.0",
   "license": "Apache-2.0",
   "main": "dist/index.cjs.js",
   "types": "dist/index.d.ts",
@@ -41,9 +41,10 @@
     "@backstage/plugin-signals-node": "backstage:^",
     "@langchain/core": "^0.3.72",
     "@langchain/langgraph": "^0.4.9",
+    "@langchain/mcp-adapters": "^1.0.0",
     "@langchain/textsplitters": "^0.1.0",
-    "@sweetoburrito/backstage-plugin-ai-assistant-common": "^0.5.0",
-    "@sweetoburrito/backstage-plugin-ai-assistant-node": "^0.5.1",
+    "@sweetoburrito/backstage-plugin-ai-assistant-common": "^0.6.0",
+    "@sweetoburrito/backstage-plugin-ai-assistant-node": "^0.7.0",
     "express": "^4.17.1",
     "express-promise-router": "^4.1.0",
     "knex": "^3.1.0",
@@ -59,12 +60,14 @@
     "@backstage/plugin-events-backend": "backstage:^",
     "@backstage/plugin-signals-backend": "backstage:^",
     "@backstage/types": "backstage:^",
+    "@sweetoburrito/backstage-plugin-ai-assistant-backend-module-callback-provider-langfuse": "workspace:^",
     "@sweetoburrito/backstage-plugin-ai-assistant-backend-module-embeddings-provider-azure-open-ai": "workspace:^",
     "@sweetoburrito/backstage-plugin-ai-assistant-backend-module-embeddings-provider-ollama": "workspace:^",
     "@sweetoburrito/backstage-plugin-ai-assistant-backend-module-ingestor-azure-devops": "workspace:^",
     "@sweetoburrito/backstage-plugin-ai-assistant-backend-module-ingestor-catalog": "workspace:^",
     "@sweetoburrito/backstage-plugin-ai-assistant-backend-module-ingestor-github": "workspace:^",
     "@sweetoburrito/backstage-plugin-ai-assistant-backend-module-model-provider-azure-ai": "workspace:^",
+    "@sweetoburrito/backstage-plugin-ai-assistant-backend-module-model-provider-google-vertex-ai": "workspace:^",
     "@sweetoburrito/backstage-plugin-ai-assistant-backend-module-model-provider-ollama": "workspace:^",
     "@sweetoburrito/backstage-plugin-ai-assistant-backend-module-tool-provider-backstage": "workspace:^",
     "@types/express": "^4.0.0",