@sweetoburrito/backstage-plugin-ai-assistant-backend 0.0.0-snapshot-20251029080430

package/dist/services/chat.cjs.js

@@ -0,0 +1,258 @@
+'use strict';
+
+var chatStore = require('../database/chat-store.cjs.js');
+var prompts = require('../constants/prompts.cjs.js');
+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 langchain = require('@langfuse/langchain');
+var uuid = require('uuid');
+
+const createChatService = async ({
+  models,
+  tools: tools$1,
+  logger,
+  database,
+  signals,
+  config,
+  catalog,
+  cache,
+  auth,
+  langfuseEnabled
+}) => {
+  logger.info(`Available models: ${models.map((m) => m.id).join(", ")}`);
+  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,
+    langfuseEnabled
+  });
+  const agentTools = tools$1.map((tool) => new tools.DynamicStructuredTool(tool));
+  const systemPromptTemplate = prompts$1.SystemMessagePromptTemplate.fromTemplate(`
+    PURPOSE:
+    {basePrompt}
+
+    TOOL USAGE GUIDELINES:
+    {toolGuideline}
+
+    Available tools:
+    {toolList}
+
+    Calling User:
+    {user}
+
+    Context:
+    {context}`);
+  const addMessages = async (messages, userRef, conversationId, recentConversationMessages) => {
+    const recentMessages = recentConversationMessages || await chatStore$1.getChatMessages(conversationId, userRef, 5, ["tool"]);
+    const conversationSize = (recentMessages?.length ?? 0) + messages.length;
+    if (recentMessages.length === 0) {
+      const conversation2 = {
+        id: conversationId,
+        title: "New Conversation",
+        userRef
+      };
+      chatStore$1.createConversation(conversation2);
+      chatStore$1.addChatMessage(messages, userRef, conversationId);
+      signals.publish({
+        channel: `ai-assistant.chat.conversation-details-update`,
+        message: { conversation: conversation2 },
+        recipients: {
+          type: "user",
+          entityRef: userRef
+        }
+      });
+      return;
+    }
+    if (conversationSize < 5) {
+      chatStore$1.addChatMessage(messages, userRef, conversationId);
+      return;
+    }
+    const conversation = await chatStore$1.getConversation(
+      conversationId,
+      userRef
+    );
+    if (conversation.title !== "New Conversation") {
+      chatStore$1.addChatMessage(messages, userRef, conversationId);
+      return;
+    }
+    const summary = await summarizer$1.summarize(recentMessages, "25 characters");
+    conversation.title = summary;
+    chatStore$1.updateConversation(conversation);
+    chatStore$1.addChatMessage(messages, userRef, conversationId);
+    signals.publish({
+      channel: `ai-assistant.chat.conversation-details-update`,
+      message: { conversation },
+      recipients: {
+        type: "user",
+        entityRef: userRef
+      }
+    });
+  };
+  const prompt = async ({
+    conversationId,
+    messages,
+    modelId,
+    stream = true,
+    userEntityRef
+  }) => {
+    const model = models.find((m) => m.id === modelId)?.chatModel;
+    if (!model) {
+      throw new Error(`Model with id ${modelId} not found`);
+    }
+    const streamFn = async () => {
+      const recentConversationMessages = await chatStore$1.getChatMessages(
+        conversationId,
+        userEntityRef,
+        10,
+        ["tool"]
+      );
+      const credentials = await auth.getOwnServiceCredentials();
+      const user = await getUser(cache, userEntityRef, credentials, catalog);
+      addMessages(
+        messages,
+        userEntityRef,
+        conversationId,
+        recentConversationMessages
+      );
+      const systemPrompt = await systemPromptTemplate.formatMessages({
+        basePrompt: combinedBasePrompt,
+        toolGuideline,
+        toolList: agentTools.map((tool) => `- ${tool.name}: ${tool.description}`).join("\n"),
+        context: `none`,
+        user
+      });
+      const agent = prebuilt.createReactAgent({
+        llm: model,
+        tools: agentTools,
+        prompt: systemPrompt[0].text
+      });
+      const langfuseHandler = langfuseEnabled ? new langchain.CallbackHandler({
+        sessionId: conversationId,
+        userId: userEntityRef,
+        tags: ["backstage-ai-assistant", "chat"]
+      }) : void 0;
+      const promptStream = await agent.stream(
+        {
+          messages: [...recentConversationMessages, ...messages]
+        },
+        {
+          streamMode: ["values"],
+          runName: "ai-assistant-chat",
+          metadata: {
+            langfuseUserId: userEntityRef,
+            langfuseSessionId: conversationId,
+            langfuseTags: ["ai-assistant", "chat", modelId]
+          },
+          callbacks: langfuseHandler ? [langfuseHandler] : []
+        }
+      );
+      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(
+          (m) => recentConversationMessages.findIndex((rm) => rm.id === m.id) === -1
+        ).filter((m) => m.getType() !== "human").map((m) => {
+          const id = m.id ?? "";
+          const role = m.getType();
+          const content = typeof m.content === "string" ? m.content : JSON.stringify(m.content);
+          const metadata = {};
+          if (role === "ai") {
+            const aiMessage = m;
+            metadata.toolCalls = aiMessage.tool_calls || [];
+            metadata.finishReason = aiMessage.response_metadata.finish_reason || void 0;
+            metadata.modelName = aiMessage.response_metadata.model_name || void 0;
+          }
+          if (role === "tool") {
+            const toolMessage = m;
+            metadata.name = toolMessage.name || "";
+          }
+          return {
+            id,
+            role,
+            content,
+            metadata
+          };
+        });
+        for await (const m of newMessages) {
+          const words = m.content.split(" ");
+          const chunkSize = 5;
+          let messageBuilder = "";
+          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({
+              channel: `ai-assistant.chat.conversation-stream:${conversationId}`,
+              message: { messages: [m] },
+              recipients: {
+                type: "user",
+                entityRef: userEntityRef
+              }
+            });
+          }
+        }
+        responseMessages.push(...newMessages);
+      }
+      addMessages(
+        responseMessages.map((m) => ({ ...m, id: uuid.v4() })),
+        userEntityRef,
+        conversationId,
+        [...recentConversationMessages, ...messages]
+      );
+      return responseMessages;
+    };
+    const result = streamFn();
+    return stream ? [] : result;
+  };
+  const getAvailableModels = async () => {
+    return models.map((x) => x.id);
+  };
+  const getConversation = async (options) => {
+    const { conversationId, userEntityRef } = options;
+    const conversation = await chatStore$1.getChatMessages(
+      conversationId,
+      userEntityRef
+    );
+    return conversation;
+  };
+  const getConversations = async ({
+    userEntityRef
+  }) => {
+    const conversations = await chatStore$1.getConversations(userEntityRef);
+    return conversations;
+  };
+  return {
+    prompt,
+    getAvailableModels,
+    getConversation,
+    getConversations,
+    addMessages
+  };
+};
+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

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

@@ -0,0 +1 @@
+{"version":3,"file":"chat.cjs.js","sources":["../../src/services/chat.ts"],"sourcesContent":["import { Model } from '@sweetoburrito/backstage-plugin-ai-assistant-node';\nimport { CatalogService } from '@backstage/plugin-catalog-node';\nimport { UserEntity } from '@backstage/catalog-model';\nimport {\n  LoggerService,\n  RootConfigService,\n  DatabaseService,\n  AuthService,\n} from '@backstage/backend-plugin-api';\nimport { ChatStore } from '../database/chat-store';\nimport {\n  Conversation,\n  Message,\n  JsonObject,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-common';\nimport { SignalsService } from '@backstage/plugin-signals-node';\nimport {\n  DEFAULT_FORMATTING_PROMPT,\n  DEFAULT_IDENTITY_PROMPT,\n  DEFAULT_SYSTEM_PROMPT,\n  DEFAULT_TOOL_GUIDELINE,\n} from '../constants/prompts';\nimport { Tool } from '@sweetoburrito/backstage-plugin-ai-assistant-node';\nimport { DynamicStructuredTool } from '@langchain/core/tools';\nimport { createReactAgent } from '@langchain/langgraph/prebuilt';\nimport { SystemMessagePromptTemplate } from '@langchain/core/prompts';\nimport { createSummarizerService } from './summarizer';\nimport { CallbackHandler } from '@langfuse/langchain';\nimport { v4 as uuid } from 'uuid';\nimport type {\n  BackstageCredentials,\n  CacheService,\n} from '@backstage/backend-plugin-api';\nimport { AIMessage, ToolMessage } from '@langchain/core/messages';\n\nexport type ChatServiceOptions = {\n  models: Model[];\n  tools: Tool[];\n  logger: LoggerService;\n  config: RootConfigService;\n  database: DatabaseService;\n  signals: SignalsService;\n  catalog: CatalogService;\n  cache: CacheService;\n  auth: AuthService;\n  langfuseEnabled: boolean;\n};\n\ntype PromptOptions = {\n  modelId: string;\n  messages: Message[];\n  conversationId: string;\n  stream?: boolean;\n  userEntityRef: string;\n};\n\ntype GetConversationOptions = {\n  conversationId: string;\n  userEntityRef: string;\n};\n\ntype GetConversationsOptions = {\n  userEntityRef: string;\n};\n\nexport type ChatService = {\n  prompt: (options: PromptOptions) => Promise<Required<Message>[]>;\n  getAvailableModels: () => Promise<string[]>;\n  getConversation: (\n    options: GetConversationOptions,\n  ) => Promise<Required<Message>[]>;\n  getConversations: (\n    options: GetConversationsOptions,\n  ) => Promise<Conversation[]>;\n  addMessages: (\n    messages: Message[],\n    userRef: string,\n    conversationId: string,\n    recentConversationMessages?: Message[],\n  ) => Promise<void>;\n};\n\nexport const createChatService = async ({\n  models,\n  tools,\n  logger,\n  database,\n  signals,\n  config,\n  catalog,\n  cache,\n  auth,\n  langfuseEnabled,\n}: ChatServiceOptions): Promise<ChatService> => {\n  logger.info(`Available models: ${models.map(m => m.id).join(', ')}`);\n  logger.info(`Available tools: ${tools.map(t => t.name).join(', ')}`);\n\n  const identityPrompt =\n    config.getOptionalString('aiAssistant.prompt.identity') ||\n    DEFAULT_IDENTITY_PROMPT;\n\n  const formattingPrompt =\n    config.getOptionalString('aiAssistant.prompt.formatting') ||\n    DEFAULT_FORMATTING_PROMPT;\n\n  const contentPrompt =\n    config.getOptionalString('aiAssistant.prompt.content') ||\n    DEFAULT_SYSTEM_PROMPT;\n\n  const combinedBasePrompt = `${identityPrompt}\\n\\n${formattingPrompt}\\n\\n${contentPrompt}`;\n\n  const toolGuideline =\n    config.getOptionalString('aiAssistant.prompt.toolGuideline') ||\n    DEFAULT_TOOL_GUIDELINE;\n\n  const chatStore = await ChatStore.fromConfig({ database });\n  const summarizer = await createSummarizerService({\n    config,\n    models,\n    langfuseEnabled,\n  });\n\n  const agentTools = tools.map(tool => new DynamicStructuredTool(tool));\n\n  const systemPromptTemplate = SystemMessagePromptTemplate.fromTemplate(`\n    PURPOSE:\n    {basePrompt}\n\n    TOOL USAGE GUIDELINES:\n    {toolGuideline}\n\n    Available tools:\n    {toolList}\n\n    Calling User:\n    {user}\n\n    Context:\n    {context}`);\n\n  const addMessages: ChatService['addMessages'] = async (\n    messages,\n    userRef,\n    conversationId,\n    recentConversationMessages,\n  ) => {\n    // If we have recentConversationMessages, use them; otherwise, fetch the last 5 messages\n    const recentMessages =\n      recentConversationMessages ||\n      (await chatStore.getChatMessages(conversationId, userRef, 5, ['tool']));\n\n    const conversationSize = (recentMessages?.length ?? 0) + messages.length;\n\n    if (recentMessages.length === 0) {\n      const conversation: Conversation = {\n        id: conversationId,\n        title: 'New Conversation',\n        userRef,\n      };\n      chatStore.createConversation(conversation);\n      chatStore.addChatMessage(messages, userRef, conversationId);\n\n      signals.publish({\n        channel: `ai-assistant.chat.conversation-details-update`,\n        message: { conversation },\n        recipients: {\n          type: 'user',\n          entityRef: userRef,\n        },\n      });\n      return;\n    }\n\n    if (conversationSize < 5) {\n      chatStore.addChatMessage(messages, userRef, conversationId);\n      return;\n    }\n\n    const conversation = await chatStore.getConversation(\n      conversationId,\n      userRef,\n    );\n\n    if (conversation.title !== 'New Conversation') {\n      chatStore.addChatMessage(messages, userRef, conversationId);\n      return;\n    }\n\n    const summary = await summarizer.summarize(recentMessages, '25 characters');\n\n    conversation.title = summary;\n\n    chatStore.updateConversation(conversation);\n    chatStore.addChatMessage(messages, userRef, conversationId);\n\n    signals.publish({\n      channel: `ai-assistant.chat.conversation-details-update`,\n      message: { conversation },\n      recipients: {\n        type: 'user',\n        entityRef: userRef,\n      },\n    });\n  };\n\n  const prompt: ChatService['prompt'] = async ({\n    conversationId,\n    messages,\n    modelId,\n    stream = true,\n    userEntityRef,\n  }: PromptOptions) => {\n    const model = models.find(m => m.id === modelId)?.chatModel;\n\n    if (!model) {\n      throw new Error(`Model with id ${modelId} not found`);\n    }\n\n    const streamFn = async () => {\n      const recentConversationMessages = await chatStore.getChatMessages(\n        conversationId,\n        userEntityRef,\n        10,\n        ['tool'],\n      );\n\n      const credentials = await auth.getOwnServiceCredentials();\n      const user = await getUser(cache, userEntityRef, credentials, catalog);\n\n      addMessages(\n        messages,\n        userEntityRef,\n        conversationId,\n        recentConversationMessages,\n      );\n\n      const systemPrompt = await systemPromptTemplate.formatMessages({\n        basePrompt: combinedBasePrompt,\n        toolGuideline,\n        toolList: agentTools\n          .map(tool => `- ${tool.name}: ${tool.description}`)\n          .join('\\n'),\n        context: `none`,\n        user,\n      });\n\n      const agent = createReactAgent({\n        llm: model,\n        tools: agentTools,\n        prompt: systemPrompt[0].text,\n      });\n\n      // Initialize Langfuse CallbackHandler for tracing if credentials are available\n      const langfuseHandler = langfuseEnabled\n        ? new CallbackHandler({\n            sessionId: conversationId,\n            userId: userEntityRef,\n            tags: ['backstage-ai-assistant', 'chat'],\n          })\n        : undefined;\n\n      const promptStream = await agent.stream(\n        {\n          messages: [...recentConversationMessages, ...messages],\n        },\n        {\n          streamMode: ['values'],\n          runName: 'ai-assistant-chat',\n          metadata: {\n            langfuseUserId: userEntityRef,\n            langfuseSessionId: conversationId,\n            langfuseTags: ['ai-assistant', 'chat', modelId],\n          },\n          callbacks: langfuseHandler ? [langfuseHandler] : [],\n        },\n      );\n\n      const responseMessages: Required<Message>[] = [];\n\n      for await (const [, chunk] of promptStream) {\n        const { messages: promptMessages } = chunk;\n\n        const newMessages: Required<Message>[] = promptMessages\n          .filter(m => responseMessages.findIndex(rm => rm.id === m.id) === -1)\n          .filter(\n            m =>\n              recentConversationMessages.findIndex(rm => rm.id === m.id) === -1,\n          )\n          .filter(m => m.getType() !== 'human')\n          .map(m => {\n            const id = m.id ?? '';\n            const role = m.getType();\n            const content =\n              typeof m.content === 'string'\n                ? m.content\n                : JSON.stringify(m.content);\n\n            const metadata: JsonObject = {};\n\n            if (role === 'ai') {\n              const aiMessage = m as AIMessage;\n              metadata.toolCalls = aiMessage.tool_calls || [];\n              metadata.finishReason =\n                aiMessage.response_metadata.finish_reason || undefined;\n              metadata.modelName =\n                aiMessage.response_metadata.model_name || undefined;\n            }\n\n            if (role === 'tool') {\n              const toolMessage = m as ToolMessage;\n              metadata.name = toolMessage.name || '';\n            }\n\n            return {\n              id,\n              role,\n              content,\n              metadata,\n            };\n          });\n\n        // Simulate streaming until langchain messages error is better understood\n        for await (const m of newMessages) {\n          const words = m.content.split(' ');\n          const chunkSize = 5; // Send 5 words at a time\n          let messageBuilder = '';\n\n          for (let i = 0; i < words.length; i += chunkSize) {\n            const wordChunk = words.slice(i, i + chunkSize).join(' ');\n            messageBuilder = messageBuilder.concat(wordChunk).concat(' ');\n            m.content = messageBuilder;\n\n            await new Promise(resolve => setTimeout(resolve, 50));\n\n            signals.publish({\n              channel: `ai-assistant.chat.conversation-stream:${conversationId}`,\n              message: { messages: [m] },\n              recipients: {\n                type: 'user',\n                entityRef: userEntityRef,\n              },\n            });\n          }\n        }\n\n        responseMessages.push(...newMessages);\n      }\n\n      addMessages(\n        responseMessages.map(m => ({ ...m, id: uuid() })),\n        userEntityRef,\n        conversationId,\n        [...recentConversationMessages, ...messages],\n      );\n\n      return responseMessages;\n    };\n\n    const result = streamFn();\n\n    return stream ? [] : result;\n  };\n\n  const getAvailableModels: ChatService['getAvailableModels'] = async () => {\n    return models.map(x => x.id);\n  };\n\n  const getConversation: ChatService['getConversation'] = async (\n    options: GetConversationOptions,\n  ) => {\n    const { conversationId, userEntityRef } = options;\n\n    const conversation = await chatStore.getChatMessages(\n      conversationId,\n      userEntityRef,\n    );\n\n    return conversation;\n  };\n\n  const getConversations: ChatService['getConversations'] = async ({\n    userEntityRef,\n  }: GetConversationsOptions) => {\n    const conversations = await chatStore.getConversations(userEntityRef);\n\n    return conversations;\n  };\n  return {\n    prompt,\n    getAvailableModels,\n    getConversation,\n    getConversations,\n    addMessages,\n  };\n};\n\nasync function getUser(\n  cache: CacheService,\n  userEntityRef: string,\n  credentials: BackstageCredentials,\n  catalog: CatalogService,\n) {\n  const cached = await cache.get(userEntityRef);\n\n  if (cached) {\n    return JSON.parse(String(cached));\n  }\n\n  const user = (await catalog.getEntityByRef(userEntityRef, {\n    credentials,\n  })) as UserEntity | undefined;\n  await cache.set(userEntityRef, JSON.stringify(user));\n\n  return user;\n}\n"],"names":["tools","DEFAULT_IDENTITY_PROMPT","DEFAULT_FORMATTING_PROMPT","DEFAULT_SYSTEM_PROMPT","DEFAULT_TOOL_GUIDELINE","chatStore","ChatStore","summarizer","createSummarizerService","DynamicStructuredTool","SystemMessagePromptTemplate","conversation","createReactAgent","CallbackHandler","uuid"],"mappings":";;;;;;;;;;;AAkFO,MAAM,oBAAoB,OAAO;AAAA,EACtC,MAAA;AAAA,SACAA,OAAA;AAAA,EACA,MAAA;AAAA,EACA,QAAA;AAAA,EACA,OAAA;AAAA,EACA,MAAA;AAAA,EACA,OAAA;AAAA,EACA,KAAA;AAAA,EACA,IAAA;AAAA,EACA;AACF,CAAA,KAAgD;AAC9C,EAAA,MAAA,CAAO,IAAA,CAAK,CAAA,kBAAA,EAAqB,MAAA,CAAO,GAAA,CAAI,CAAA,CAAA,KAAK,CAAA,CAAE,EAAE,CAAA,CAAE,IAAA,CAAK,IAAI,CAAC,CAAA,CAAE,CAAA;AACnE,EAAA,MAAA,CAAO,IAAA,CAAK,CAAA,iBAAA,EAAoBA,OAAA,CAAM,GAAA,CAAI,CAAA,CAAA,KAAK,CAAA,CAAE,IAAI,CAAA,CAAE,IAAA,CAAK,IAAI,CAAC,CAAA,CAAE,CAAA;AAEnE,EAAA,MAAM,cAAA,GACJ,MAAA,CAAO,iBAAA,CAAkB,6BAA6B,CAAA,IACtDC,+BAAA;AAEF,EAAA,MAAM,gBAAA,GACJ,MAAA,CAAO,iBAAA,CAAkB,+BAA+B,CAAA,IACxDC,iCAAA;AAEF,EAAA,MAAM,aAAA,GACJ,MAAA,CAAO,iBAAA,CAAkB,4BAA4B,CAAA,IACrDC,6BAAA;AAEF,EAAA,MAAM,kBAAA,GAAqB,GAAG,cAAc;;AAAA,EAAO,gBAAgB;;AAAA,EAAO,aAAa,CAAA,CAAA;AAEvF,EAAA,MAAM,aAAA,GACJ,MAAA,CAAO,iBAAA,CAAkB,kCAAkC,CAAA,IAC3DC,8BAAA;AAEF,EAAA,MAAMC,cAAY,MAAMC,mBAAA,CAAU,UAAA,CAAW,EAAE,UAAU,CAAA;AACzD,EAAA,MAAMC,YAAA,GAAa,MAAMC,kCAAA,CAAwB;AAAA,IAC/C,MAAA;AAAA,IACA,MAAA;AAAA,IACA;AAAA,GACD,CAAA;AAED,EAAA,MAAM,aAAaR,OAAA,CAAM,GAAA,CAAI,UAAQ,IAAIS,2BAAA,CAAsB,IAAI,CAAC,CAAA;AAEpE,EAAA,MAAM,oBAAA,GAAuBC,sCAA4B,YAAA,CAAa;AAAA;AAAA;;AAAA;AAAA;;AAAA;AAAA;;AAAA;AAAA;;AAAA;AAAA,aAAA,CAc1D,CAAA;AAEZ,EAAA,MAAM,WAAA,GAA0C,OAC9C,QAAA,EACA,OAAA,EACA,gBACA,0BAAA,KACG;AAEH,IAAA,MAAM,cAAA,GACJ,0BAAA,IACC,MAAML,WAAA,CAAU,eAAA,CAAgB,gBAAgB,OAAA,EAAS,CAAA,EAAG,CAAC,MAAM,CAAC,CAAA;AAEvE,IAAA,MAAM,gBAAA,GAAA,CAAoB,cAAA,EAAgB,MAAA,IAAU,CAAA,IAAK,QAAA,CAAS,MAAA;AAElE,IAAA,IAAI,cAAA,CAAe,WAAW,CAAA,EAAG;AAC/B,MAAA,MAAMM,aAAAA,GAA6B;AAAA,QACjC,EAAA,EAAI,cAAA;AAAA,QACJ,KAAA,EAAO,kBAAA;AAAA,QACP;AAAA,OACF;AACA,MAAAN,WAAA,CAAU,mBAAmBM,aAAY,CAAA;AACzC,MAAAN,WAAA,CAAU,cAAA,CAAe,QAAA,EAAU,OAAA,EAAS,cAAc,CAAA;AAE1D,MAAA,OAAA,CAAQ,OAAA,CAAQ;AAAA,QACd,OAAA,EAAS,CAAA,6CAAA,CAAA;AAAA,QACT,OAAA,EAAS,EAAE,YAAA,EAAAM,aAAAA,EAAa;AAAA,QACxB,UAAA,EAAY;AAAA,UACV,IAAA,EAAM,MAAA;AAAA,UACN,SAAA,EAAW;AAAA;AACb,OACD,CAAA;AACD,MAAA;AAAA,IACF;AAEA,IAAA,IAAI,mBAAmB,CAAA,EAAG;AACxB,MAAAN,WAAA,CAAU,cAAA,CAAe,QAAA,EAAU,OAAA,EAAS,cAAc,CAAA;AAC1D,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,YAAA,GAAe,MAAMA,WAAA,CAAU,eAAA;AAAA,MACnC,cAAA;AAAA,MACA;AAAA,KACF;AAEA,IAAA,IAAI,YAAA,CAAa,UAAU,kBAAA,EAAoB;AAC7C,MAAAA,WAAA,CAAU,cAAA,CAAe,QAAA,EAAU,OAAA,EAAS,cAAc,CAAA;AAC1D,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,MAAME,YAAA,CAAW,SAAA,CAAU,gBAAgB,eAAe,CAAA;AAE1E,IAAA,YAAA,CAAa,KAAA,GAAQ,OAAA;AAErB,IAAAF,WAAA,CAAU,mBAAmB,YAAY,CAAA;AACzC,IAAAA,WAAA,CAAU,cAAA,CAAe,QAAA,EAAU,OAAA,EAAS,cAAc,CAAA;AAE1D,IAAA,OAAA,CAAQ,OAAA,CAAQ;AAAA,MACd,OAAA,EAAS,CAAA,6CAAA,CAAA;AAAA,MACT,OAAA,EAAS,EAAE,YAAA,EAAa;AAAA,MACxB,UAAA,EAAY;AAAA,QACV,IAAA,EAAM,MAAA;AAAA,QACN,SAAA,EAAW;AAAA;AACb,KACD,CAAA;AAAA,EACH,CAAA;AAEA,EAAA,MAAM,SAAgC,OAAO;AAAA,IAC3C,cAAA;AAAA,IACA,QAAA;AAAA,IACA,OAAA;AAAA,IACA,MAAA,GAAS,IAAA;AAAA,IACT;AAAA,GACF,KAAqB;AACnB,IAAA,MAAM,QAAQ,MAAA,CAAO,IAAA,CAAK,OAAK,CAAA,CAAE,EAAA,KAAO,OAAO,CAAA,EAAG,SAAA;AAElD,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,cAAA,EAAiB,OAAO,CAAA,UAAA,CAAY,CAAA;AAAA,IACtD;AAEA,IAAA,MAAM,WAAW,YAAY;AAC3B,MAAA,MAAM,0BAAA,GAA6B,MAAMA,WAAA,CAAU,eAAA;AAAA,QACjD,cAAA;AAAA,QACA,aAAA;AAAA,QACA,EAAA;AAAA,QACA,CAAC,MAAM;AAAA,OACT;AAEA,MAAA,MAAM,WAAA,GAAc,MAAM,IAAA,CAAK,wBAAA,EAAyB;AACxD,MAAA,MAAM,OAAO,MAAM,OAAA,CAAQ,KAAA,EAAO,aAAA,EAAe,aAAa,OAAO,CAAA;AAErE,MAAA,WAAA;AAAA,QACE,QAAA;AAAA,QACA,aAAA;AAAA,QACA,cAAA;AAAA,QACA;AAAA,OACF;AAEA,MAAA,MAAM,YAAA,GAAe,MAAM,oBAAA,CAAqB,cAAA,CAAe;AAAA,QAC7D,UAAA,EAAY,kBAAA;AAAA,QACZ,aAAA;AAAA,QACA,QAAA,EAAU,UAAA,CACP,GAAA,CAAI,CAAA,IAAA,KAAQ,CAAA,EAAA,EAAK,IAAA,CAAK,IAAI,CAAA,EAAA,EAAK,IAAA,CAAK,WAAW,CAAA,CAAE,CAAA,CACjD,KAAK,IAAI,CAAA;AAAA,QACZ,OAAA,EAAS,CAAA,IAAA,CAAA;AAAA,QACT;AAAA,OACD,CAAA;AAED,MAAA,MAAM,QAAQO,yBAAA,CAAiB;AAAA,QAC7B,GAAA,EAAK,KAAA;AAAA,QACL,KAAA,EAAO,UAAA;AAAA,QACP,MAAA,EAAQ,YAAA,CAAa,CAAC,CAAA,CAAE;AAAA,OACzB,CAAA;AAGD,MAAA,MAAM,eAAA,GAAkB,eAAA,GACpB,IAAIC,yBAAA,CAAgB;AAAA,QAClB,SAAA,EAAW,cAAA;AAAA,QACX,MAAA,EAAQ,aAAA;AAAA,QACR,IAAA,EAAM,CAAC,wBAAA,EAA0B,MAAM;AAAA,OACxC,CAAA,GACD,MAAA;AAEJ,MAAA,MAAM,YAAA,GAAe,MAAM,KAAA,CAAM,MAAA;AAAA,QAC/B;AAAA,UACE,QAAA,EAAU,CAAC,GAAG,0BAAA,EAA4B,GAAG,QAAQ;AAAA,SACvD;AAAA,QACA;AAAA,UACE,UAAA,EAAY,CAAC,QAAQ,CAAA;AAAA,UACrB,OAAA,EAAS,mBAAA;AAAA,UACT,QAAA,EAAU;AAAA,YACR,cAAA,EAAgB,aAAA;AAAA,YAChB,iBAAA,EAAmB,cAAA;AAAA,YACnB,YAAA,EAAc,CAAC,cAAA,EAAgB,MAAA,EAAQ,OAAO;AAAA,WAChD;AAAA,UACA,SAAA,EAAW,eAAA,GAAkB,CAAC,eAAe,IAAI;AAAC;AACpD,OACF;AAEA,MAAA,MAAM,mBAAwC,EAAC;AAE/C,MAAA,WAAA,MAAiB,GAAG,KAAK,CAAA,IAAK,YAAA,EAAc;AAC1C,QAAA,MAAM,EAAE,QAAA,EAAU,cAAA,EAAe,GAAI,KAAA;AAErC,QAAA,MAAM,WAAA,GAAmC,cAAA,CACtC,MAAA,CAAO,CAAA,CAAA,KAAK,gBAAA,CAAiB,SAAA,CAAU,CAAA,EAAA,KAAM,EAAA,CAAG,EAAA,KAAO,CAAA,CAAE,EAAE,CAAA,KAAM,EAAE,CAAA,CACnE,MAAA;AAAA,UACC,CAAA,CAAA,KACE,2BAA2B,SAAA,CAAU,CAAA,EAAA,KAAM,GAAG,EAAA,KAAO,CAAA,CAAE,EAAE,CAAA,KAAM;AAAA,SACnE,CACC,OAAO,CAAA,CAAA,KAAK,CAAA,CAAE,SAAQ,KAAM,OAAO,CAAA,CACnC,GAAA,CAAI,CAAA,CAAA,KAAK;AACR,UAAA,MAAM,EAAA,GAAK,EAAE,EAAA,IAAM,EAAA;AACnB,UAAA,MAAM,IAAA,GAAO,EAAE,OAAA,EAAQ;AACvB,UAAA,MAAM,OAAA,GACJ,OAAO,CAAA,CAAE,OAAA,KAAY,QAAA,GACjB,EAAE,OAAA,GACF,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,OAAO,CAAA;AAE9B,UAAA,MAAM,WAAuB,EAAC;AAE9B,UAAA,IAAI,SAAS,IAAA,EAAM;AACjB,YAAA,MAAM,SAAA,GAAY,CAAA;AAClB,YAAA,QAAA,CAAS,SAAA,GAAY,SAAA,CAAU,UAAA,IAAc,EAAC;AAC9C,YAAA,QAAA,CAAS,YAAA,GACP,SAAA,CAAU,iBAAA,CAAkB,aAAA,IAAiB,MAAA;AAC/C,YAAA,QAAA,CAAS,SAAA,GACP,SAAA,CAAU,iBAAA,CAAkB,UAAA,IAAc,MAAA;AAAA,UAC9C;AAEA,UAAA,IAAI,SAAS,MAAA,EAAQ;AACnB,YAAA,MAAM,WAAA,GAAc,CAAA;AACpB,YAAA,QAAA,CAAS,IAAA,GAAO,YAAY,IAAA,IAAQ,EAAA;AAAA,UACtC;AAEA,UAAA,OAAO;AAAA,YACL,EAAA;AAAA,YACA,IAAA;AAAA,YACA,OAAA;AAAA,YACA;AAAA,WACF;AAAA,QACF,CAAC,CAAA;AAGH,QAAA,WAAA,MAAiB,KAAK,WAAA,EAAa;AACjC,UAAA,MAAM,KAAA,GAAQ,CAAA,CAAE,OAAA,CAAQ,KAAA,CAAM,GAAG,CAAA;AACjC,UAAA,MAAM,SAAA,GAAY,CAAA;AAClB,UAAA,IAAI,cAAA,GAAiB,EAAA;AAErB,UAAA,KAAA,IAAS,IAAI,CAAA,EAAG,CAAA,GAAI,KAAA,CAAM,MAAA,EAAQ,KAAK,SAAA,EAAW;AAChD,YAAA,MAAM,SAAA,GAAY,MAAM,KAAA,CAAM,CAAA,EAAG,IAAI,SAAS,CAAA,CAAE,KAAK,GAAG,CAAA;AACxD,YAAA,cAAA,GAAiB,cAAA,CAAe,MAAA,CAAO,SAAS,CAAA,CAAE,OAAO,GAAG,CAAA;AAC5D,YAAA,CAAA,CAAE,OAAA,GAAU,cAAA;AAEZ,YAAA,MAAM,IAAI,OAAA,CAAQ,CAAA,OAAA,KAAW,UAAA,CAAW,OAAA,EAAS,EAAE,CAAC,CAAA;AAEpD,YAAA,OAAA,CAAQ,OAAA,CAAQ;AAAA,cACd,OAAA,EAAS,yCAAyC,cAAc,CAAA,CAAA;AAAA,cAChE,OAAA,EAAS,EAAE,QAAA,EAAU,CAAC,CAAC,CAAA,EAAE;AAAA,cACzB,UAAA,EAAY;AAAA,gBACV,IAAA,EAAM,MAAA;AAAA,gBACN,SAAA,EAAW;AAAA;AACb,aACD,CAAA;AAAA,UACH;AAAA,QACF;AAEA,QAAA,gBAAA,CAAiB,IAAA,CAAK,GAAG,WAAW,CAAA;AAAA,MACtC;AAEA,MAAA,WAAA;AAAA,QACE,gBAAA,CAAiB,IAAI,CAAA,CAAA,MAAM,EAAE,GAAG,CAAA,EAAG,EAAA,EAAIC,OAAA,EAAK,EAAE,CAAE,CAAA;AAAA,QAChD,aAAA;AAAA,QACA,cAAA;AAAA,QACA,CAAC,GAAG,0BAAA,EAA4B,GAAG,QAAQ;AAAA,OAC7C;AAEA,MAAA,OAAO,gBAAA;AAAA,IACT,CAAA;AAEA,IAAA,MAAM,SAAS,QAAA,EAAS;AAExB,IAAA,OAAO,MAAA,GAAS,EAAC,GAAI,MAAA;AAAA,EACvB,CAAA;AAEA,EAAA,MAAM,qBAAwD,YAAY;AACxE,IAAA,OAAO,MAAA,CAAO,GAAA,CAAI,CAAA,CAAA,KAAK,CAAA,CAAE,EAAE,CAAA;AAAA,EAC7B,CAAA;AAEA,EAAA,MAAM,eAAA,GAAkD,OACtD,OAAA,KACG;AACH,IAAA,MAAM,EAAE,cAAA,EAAgB,aAAA,EAAc,GAAI,OAAA;AAE1C,IAAA,MAAM,YAAA,GAAe,MAAMT,WAAA,CAAU,eAAA;AAAA,MACnC,cAAA;AAAA,MACA;AAAA,KACF;AAEA,IAAA,OAAO,YAAA;AAAA,EACT,CAAA;AAEA,EAAA,MAAM,mBAAoD,OAAO;AAAA,IAC/D;AAAA,GACF,KAA+B;AAC7B,IAAA,MAAM,aAAA,GAAgB,MAAMA,WAAA,CAAU,gBAAA,CAAiB,aAAa,CAAA;AAEpE,IAAA,OAAO,aAAA;AAAA,EACT,CAAA;AACA,EAAA,OAAO;AAAA,IACL,MAAA;AAAA,IACA,kBAAA;AAAA,IACA,eAAA;AAAA,IACA,gBAAA;AAAA,IACA;AAAA,GACF;AACF;AAEA,eAAe,OAAA,CACb,KAAA,EACA,aAAA,EACA,WAAA,EACA,OAAA,EACA;AACA,EAAA,MAAM,MAAA,GAAS,MAAM,KAAA,CAAM,GAAA,CAAI,aAAa,CAAA;AAE5C,EAAA,IAAI,MAAA,EAAQ;AACV,IAAA,OAAO,IAAA,CAAK,KAAA,CAAM,MAAA,CAAO,MAAM,CAAC,CAAA;AAAA,EAClC;AAEA,EAAA,MAAM,IAAA,GAAQ,MAAM,OAAA,CAAQ,cAAA,CAAe,aAAA,EAAe;AAAA,IACxD;AAAA,GACD,CAAA;AACD,EAAA,MAAM,MAAM,GAAA,CAAI,aAAA,EAAe,IAAA,CAAK,SAAA,CAAU,IAAI,CAAC,CAAA;AAEnD,EAAA,OAAO,IAAA;AACT;;;;"}

package/dist/services/ingestor.cjs.js

@@ -0,0 +1,89 @@
+'use strict';
+
+var backendPluginApi = require('@backstage/backend-plugin-api');
+var textsplitters = require('@langchain/textsplitters');
+
+const DEFAULT_DATA_INGESTION_SCHEDULE = {
+  frequency: {
+    hours: 24
+  },
+  timeout: {
+    hours: 3
+  }
+};
+const createDataIngestionPipeline = ({
+  config,
+  logger,
+  scheduler,
+  ingestors,
+  vectorStore
+}) => {
+  const schedule = config.has("aiAssistant.ingestion.schedule") ? backendPluginApi.readSchedulerServiceTaskScheduleDefinitionFromConfig(
+    config.getConfig("aiAssistant.ingestion.schedule")
+  ) : DEFAULT_DATA_INGESTION_SCHEDULE;
+  const taskRunner = scheduler.createScheduledTaskRunner(schedule);
+  const taskId = `ai-assistant.data-ingestion:start`;
+  const dataIngestion = async () => {
+    logger.info("Starting data ingestion...");
+    if (ingestors.length === 0) {
+      logger.warn("No ingestors available for data ingestion.");
+      return;
+    }
+    logger.info(`Ingestors available: ${ingestors.map((i) => i.id).join(", ")}`);
+    for await (const ingestor of ingestors) {
+      logger.info(`Running ingestor: ${ingestor.id}`);
+      const saveDocumentsBatch = async (documents2) => {
+        logger.info(
+          `Ingested documents for ${ingestor.id}: ${documents2.length}`
+        );
+        const splitter = new textsplitters.RecursiveCharacterTextSplitter({
+          chunkSize: 500,
+          // TODO: Make chunk size configurable
+          chunkOverlap: 50
+          // TODO: Make chunk overlap configurable
+        });
+        const docs = await Promise.all(
+          documents2.map(async (document) => {
+            logger.debug(
+              `Deleting existing documents with id: [${document.metadata.id}] and source: [${ingestor.id}]`
+            );
+            await vectorStore.deleteDocuments({
+              filter: { source: ingestor.id, id: document.metadata.id }
+            });
+            const chunks = await splitter.splitText(document.content);
+            const chunkDocs = chunks.flatMap(
+              (chunk, i) => ({
+                metadata: { ...document.metadata, chunk: String(i) },
+                content: chunk
+              })
+            );
+            return chunkDocs;
+          })
+        );
+        logger.info(`Adding documents to vector store...`);
+        await vectorStore.addDocuments(docs.flat());
+        logger.info(`Added documents to vector store for ${ingestor.id}`);
+      };
+      const documents = await ingestor.ingest({
+        saveDocumentsBatch
+      });
+      if (documents) {
+        saveDocumentsBatch(documents);
+      }
+      logger.info(`Finished processing ingestor: ${ingestor.id}`);
+    }
+    logger.info("Data ingestion completed.");
+  };
+  const start = async () => {
+    taskRunner.run({
+      id: taskId,
+      fn: dataIngestion
+    });
+  };
+  return {
+    start
+  };
+};
+
+exports.createDataIngestionPipeline = createDataIngestionPipeline;
+//# sourceMappingURL=ingestor.cjs.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ingestor.cjs.js","sources":["../../src/services/ingestor.ts"],"sourcesContent":["import {\n  DataIngestionPipeline,\n  DataIngestionPipelineOptions,\n  EmbeddingDocument,\n} from '@sweetoburrito/backstage-plugin-ai-assistant-node';\n\nimport {\n  SchedulerServiceTaskScheduleDefinition,\n  readSchedulerServiceTaskScheduleDefinitionFromConfig,\n} from '@backstage/backend-plugin-api';\n\nimport { RecursiveCharacterTextSplitter } from '@langchain/textsplitters';\n\nconst DEFAULT_DATA_INGESTION_SCHEDULE: SchedulerServiceTaskScheduleDefinition =\n  {\n    frequency: {\n      hours: 24,\n    },\n    timeout: {\n      hours: 3,\n    },\n  };\n\nexport const createDataIngestionPipeline = ({\n  config,\n  logger,\n  scheduler,\n  ingestors,\n  vectorStore,\n}: DataIngestionPipelineOptions): DataIngestionPipeline => {\n  const schedule = config.has('aiAssistant.ingestion.schedule')\n    ? readSchedulerServiceTaskScheduleDefinitionFromConfig(\n        config.getConfig('aiAssistant.ingestion.schedule'),\n      )\n    : DEFAULT_DATA_INGESTION_SCHEDULE;\n\n  const taskRunner = scheduler.createScheduledTaskRunner(schedule);\n\n  const taskId = `ai-assistant.data-ingestion:start`;\n\n  const dataIngestion = async () => {\n    logger.info('Starting data ingestion...');\n\n    if (ingestors.length === 0) {\n      logger.warn('No ingestors available for data ingestion.');\n      return;\n    }\n\n    logger.info(`Ingestors available: ${ingestors.map(i => i.id).join(', ')}`);\n\n    for await (const ingestor of ingestors) {\n      logger.info(`Running ingestor: ${ingestor.id}`);\n\n      const saveDocumentsBatch = async (documents: EmbeddingDocument[]) => {\n        logger.info(\n          `Ingested documents for ${ingestor.id}: ${documents.length}`,\n        );\n\n        const splitter = new RecursiveCharacterTextSplitter({\n          chunkSize: 500, // TODO: Make chunk size configurable\n          chunkOverlap: 50, // TODO: Make chunk overlap configurable\n        });\n\n        const docs = await Promise.all(\n          documents.map(async document => {\n            // Delete existing documents with this document id and ingestor source\n            logger.debug(\n              `Deleting existing documents with id: [${document.metadata.id}] and source: [${ingestor.id}]`,\n            );\n            await vectorStore.deleteDocuments({\n              filter: { source: ingestor.id, id: document.metadata.id },\n            });\n\n            const chunks = await splitter.splitText(document.content);\n\n            const chunkDocs: EmbeddingDocument[] = chunks.flatMap(\n              (chunk, i) => ({\n                metadata: { ...document.metadata, chunk: String(i) },\n                content: chunk,\n              }),\n            );\n\n            return chunkDocs;\n          }),\n        );\n\n        logger.info(`Adding documents to vector store...`);\n        await vectorStore.addDocuments(docs.flat());\n        logger.info(`Added documents to vector store for ${ingestor.id}`);\n      };\n\n      const documents = await ingestor.ingest({\n        saveDocumentsBatch,\n      });\n\n      if (documents) {\n        saveDocumentsBatch(documents);\n      }\n\n      logger.info(`Finished processing ingestor: ${ingestor.id}`);\n    }\n\n    logger.info('Data ingestion completed.');\n  };\n\n  const start = async () => {\n    taskRunner.run({\n      id: taskId,\n      fn: dataIngestion,\n    });\n  };\n\n  return {\n    start,\n  };\n};\n"],"names":["readSchedulerServiceTaskScheduleDefinitionFromConfig","documents","RecursiveCharacterTextSplitter"],"mappings":";;;;;AAaA,MAAM,+BAAA,GACJ;AAAA,EACE,SAAA,EAAW;AAAA,IACT,KAAA,EAAO;AAAA,GACT;AAAA,EACA,OAAA,EAAS;AAAA,IACP,KAAA,EAAO;AAAA;AAEX,CAAA;AAEK,MAAM,8BAA8B,CAAC;AAAA,EAC1C,MAAA;AAAA,EACA,MAAA;AAAA,EACA,SAAA;AAAA,EACA,SAAA;AAAA,EACA;AACF,CAAA,KAA2D;AACzD,EAAA,MAAM,QAAA,GAAW,MAAA,CAAO,GAAA,CAAI,gCAAgC,CAAA,GACxDA,qEAAA;AAAA,IACE,MAAA,CAAO,UAAU,gCAAgC;AAAA,GACnD,GACA,+BAAA;AAEJ,EAAA,MAAM,UAAA,GAAa,SAAA,CAAU,yBAAA,CAA0B,QAAQ,CAAA;AAE/D,EAAA,MAAM,MAAA,GAAS,CAAA,iCAAA,CAAA;AAEf,EAAA,MAAM,gBAAgB,YAAY;AAChC,IAAA,MAAA,CAAO,KAAK,4BAA4B,CAAA;AAExC,IAAA,IAAI,SAAA,CAAU,WAAW,CAAA,EAAG;AAC1B,MAAA,MAAA,CAAO,KAAK,4CAA4C,CAAA;AACxD,MAAA;AAAA,IACF;AAEA,IAAA,MAAA,CAAO,IAAA,CAAK,CAAA,qBAAA,EAAwB,SAAA,CAAU,GAAA,CAAI,CAAA,CAAA,KAAK,CAAA,CAAE,EAAE,CAAA,CAAE,IAAA,CAAK,IAAI,CAAC,CAAA,CAAE,CAAA;AAEzE,IAAA,WAAA,MAAiB,YAAY,SAAA,EAAW;AACtC,MAAA,MAAA,CAAO,IAAA,CAAK,CAAA,kBAAA,EAAqB,QAAA,CAAS,EAAE,CAAA,CAAE,CAAA;AAE9C,MAAA,MAAM,kBAAA,GAAqB,OAAOC,UAAAA,KAAmC;AACnE,QAAA,MAAA,CAAO,IAAA;AAAA,UACL,CAAA,uBAAA,EAA0B,QAAA,CAAS,EAAE,CAAA,EAAA,EAAKA,WAAU,MAAM,CAAA;AAAA,SAC5D;AAEA,QAAA,MAAM,QAAA,GAAW,IAAIC,4CAAA,CAA+B;AAAA,UAClD,SAAA,EAAW,GAAA;AAAA;AAAA,UACX,YAAA,EAAc;AAAA;AAAA,SACf,CAAA;AAED,QAAA,MAAM,IAAA,GAAO,MAAM,OAAA,CAAQ,GAAA;AAAA,UACzBD,UAAAA,CAAU,GAAA,CAAI,OAAM,QAAA,KAAY;AAE9B,YAAA,MAAA,CAAO,KAAA;AAAA,cACL,yCAAyC,QAAA,CAAS,QAAA,CAAS,EAAE,CAAA,eAAA,EAAkB,SAAS,EAAE,CAAA,CAAA;AAAA,aAC5F;AACA,YAAA,MAAM,YAAY,eAAA,CAAgB;AAAA,cAChC,MAAA,EAAQ,EAAE,MAAA,EAAQ,QAAA,CAAS,IAAI,EAAA,EAAI,QAAA,CAAS,SAAS,EAAA;AAAG,aACzD,CAAA;AAED,YAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,SAAA,CAAU,SAAS,OAAO,CAAA;AAExD,YAAA,MAAM,YAAiC,MAAA,CAAO,OAAA;AAAA,cAC5C,CAAC,OAAO,CAAA,MAAO;AAAA,gBACb,QAAA,EAAU,EAAE,GAAG,QAAA,CAAS,UAAU,KAAA,EAAO,MAAA,CAAO,CAAC,CAAA,EAAE;AAAA,gBACnD,OAAA,EAAS;AAAA,eACX;AAAA,aACF;AAEA,YAAA,OAAO,SAAA;AAAA,UACT,CAAC;AAAA,SACH;AAEA,QAAA,MAAA,CAAO,KAAK,CAAA,mCAAA,CAAqC,CAAA;AACjD,QAAA,MAAM,WAAA,CAAY,YAAA,CAAa,IAAA,CAAK,IAAA,EAAM,CAAA;AAC1C,QAAA,MAAA,CAAO,IAAA,CAAK,CAAA,oCAAA,EAAuC,QAAA,CAAS,EAAE,CAAA,CAAE,CAAA;AAAA,MAClE,CAAA;AAEA,MAAA,MAAM,SAAA,GAAY,MAAM,QAAA,CAAS,MAAA,CAAO;AAAA,QACtC;AAAA,OACD,CAAA;AAED,MAAA,IAAI,SAAA,EAAW;AACb,QAAA,kBAAA,CAAmB,SAAS,CAAA;AAAA,MAC9B;AAEA,MAAA,MAAA,CAAO,IAAA,CAAK,CAAA,8BAAA,EAAiC,QAAA,CAAS,EAAE,CAAA,CAAE,CAAA;AAAA,IAC5D;AAEA,IAAA,MAAA,CAAO,KAAK,2BAA2B,CAAA;AAAA,EACzC,CAAA;AAEA,EAAA,MAAM,QAAQ,YAAY;AACxB,IAAA,UAAA,CAAW,GAAA,CAAI;AAAA,MACb,EAAA,EAAI,MAAA;AAAA,MACJ,EAAA,EAAI;AAAA,KACL,CAAA;AAAA,EACH,CAAA;AAEA,EAAA,OAAO;AAAA,IACL;AAAA,GACF;AACF;;;;"}

package/dist/services/langfuse.cjs.js

@@ -0,0 +1,39 @@
+'use strict';
+
+var otel = require('@langfuse/otel');
+var sdkNode = require('@opentelemetry/sdk-node');
+
+function initLangfuse(config, logger) {
+  const langfuseSecret = config.getOptionalString(
+    "aiAssistant.langfuse.secretKey"
+  );
+  const langfusePublic = config.getOptionalString(
+    "aiAssistant.langfuse.publicKey"
+  );
+  const langfuseBaseUrl = config.getOptionalString(
+    "aiAssistant.langfuse.baseUrl"
+  );
+  let langfuseSpanProcessor = void 0;
+  if (langfuseSecret && langfusePublic && langfuseBaseUrl) {
+    langfuseSpanProcessor = new otel.LangfuseSpanProcessor({
+      secretKey: langfuseSecret,
+      publicKey: langfusePublic,
+      baseUrl: langfuseBaseUrl
+    });
+    const sdk = new sdkNode.NodeSDK({
+      spanProcessors: [langfuseSpanProcessor]
+    });
+    logger.info(
+      "Langfuse: Starting OpenTelemetry SDK with LangfuseSpanProcessor"
+    );
+    sdk.start();
+    return true;
+  }
+  logger.info(
+    "Langfuse: Skipping Langfuse initialization, credentials not found."
+  );
+  return false;
+}
+
+exports.initLangfuse = initLangfuse;
+//# sourceMappingURL=langfuse.cjs.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"langfuse.cjs.js","sources":["../../src/services/langfuse.ts"],"sourcesContent":["import {\n  RootConfigService,\n  RootLoggerService,\n} from '@backstage/backend-plugin-api';\nimport { LangfuseSpanProcessor } from '@langfuse/otel';\nimport { NodeSDK } from '@opentelemetry/sdk-node';\n\nexport function initLangfuse(\n  config: RootConfigService,\n  logger: RootLoggerService,\n): boolean {\n  const langfuseSecret = config.getOptionalString(\n    'aiAssistant.langfuse.secretKey',\n  );\n  const langfusePublic = config.getOptionalString(\n    'aiAssistant.langfuse.publicKey',\n  );\n  const langfuseBaseUrl = config.getOptionalString(\n    'aiAssistant.langfuse.baseUrl',\n  );\n\n  let langfuseSpanProcessor: LangfuseSpanProcessor | undefined = undefined;\n  if (langfuseSecret && langfusePublic && langfuseBaseUrl) {\n    langfuseSpanProcessor = new LangfuseSpanProcessor({\n      secretKey: langfuseSecret,\n      publicKey: langfusePublic,\n      baseUrl: langfuseBaseUrl,\n    });\n\n    const sdk = new NodeSDK({\n      spanProcessors: [langfuseSpanProcessor],\n    });\n\n    logger.info(\n      'Langfuse: Starting OpenTelemetry SDK with LangfuseSpanProcessor',\n    );\n    sdk.start();\n    return true;\n  }\n  logger.info(\n    'Langfuse: Skipping Langfuse initialization, credentials not found.',\n  );\n  return false;\n}\n"],"names":["LangfuseSpanProcessor","NodeSDK"],"mappings":";;;;;AAOO,SAAS,YAAA,CACd,QACA,MAAA,EACS;AACT,EAAA,MAAM,iBAAiB,MAAA,CAAO,iBAAA;AAAA,IAC5B;AAAA,GACF;AACA,EAAA,MAAM,iBAAiB,MAAA,CAAO,iBAAA;AAAA,IAC5B;AAAA,GACF;AACA,EAAA,MAAM,kBAAkB,MAAA,CAAO,iBAAA;AAAA,IAC7B;AAAA,GACF;AAEA,EAAA,IAAI,qBAAA,GAA2D,MAAA;AAC/D,EAAA,IAAI,cAAA,IAAkB,kBAAkB,eAAA,EAAiB;AACvD,IAAA,qBAAA,GAAwB,IAAIA,0BAAA,CAAsB;AAAA,MAChD,SAAA,EAAW,cAAA;AAAA,MACX,SAAA,EAAW,cAAA;AAAA,MACX,OAAA,EAAS;AAAA,KACV,CAAA;AAED,IAAA,MAAM,GAAA,GAAM,IAAIC,eAAA,CAAQ;AAAA,MACtB,cAAA,EAAgB,CAAC,qBAAqB;AAAA,KACvC,CAAA;AAED,IAAA,MAAA,CAAO,IAAA;AAAA,MACL;AAAA,KACF;AACA,IAAA,GAAA,CAAI,KAAA,EAAM;AACV,IAAA,OAAO,IAAA;AAAA,EACT;AACA,EAAA,MAAA,CAAO,IAAA;AAAA,IACL;AAAA,GACF;AACA,EAAA,OAAO,KAAA;AACT;;;;"}

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

@@ -0,0 +1,73 @@
+'use strict';
+
+var Router = require('express-promise-router');
+var z = require('zod');
+var validation = require('./middleware/validation.cjs.js');
+var uuid = require('uuid');
+
+function _interopDefaultCompat (e) { return e && typeof e === 'object' && 'default' in e ? e : { default: e }; }
+
+var Router__default = /*#__PURE__*/_interopDefaultCompat(Router);
+var z__default = /*#__PURE__*/_interopDefaultCompat(z);
+
+async function createChatRouter(options) {
+  const { chat, httpAuth, userInfo } = 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),
+        role: z__default.default.string(),
+        content: z__default.default.string()
+      })
+    ),
+    modelId: z__default.default.string(),
+    conversationId: z__default.default.string().uuid().optional().default(uuid.v4),
+    stream: z__default.default.boolean().optional()
+  });
+  router.post(
+    "/message",
+    validation.validation(messageSchema, "body"),
+    async (req, res) => {
+      const { messages, conversationId, modelId, stream } = req.body;
+      const credentials = await httpAuth.credentials(req);
+      const { userEntityRef } = await userInfo.getUserInfo(credentials);
+      const responseMessages = await chat.prompt({
+        modelId,
+        messages,
+        conversationId,
+        stream,
+        userEntityRef
+      });
+      res.json({
+        messages: responseMessages,
+        conversationId
+      });
+    }
+  );
+  const chatSchema = z__default.default.object({
+    id: z__default.default.string().uuid()
+  });
+  router.get("/conversations", async (req, res) => {
+    const credentials = await httpAuth.credentials(req);
+    const { userEntityRef } = await userInfo.getUserInfo(credentials);
+    const conversations = await chat.getConversations({
+      userEntityRef
+    });
+    res.json({ conversations });
+  });
+  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({
+      conversationId: id,
+      userEntityRef
+    });
+    res.json({ conversation });
+  });
+  return router;
+}
+
+exports.createChatRouter = createChatRouter;
+//# sourceMappingURL=chat.cjs.js.map

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

@@ -0,0 +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.string().uuid().optional().default(uuid),\n    stream: z.boolean().optional(),\n  });\n\n  router.post(\n    '/message',\n    validation(messageSchema, 'body'),\n    async (req, res) => {\n      const { messages, conversationId, modelId, stream } = req.body;\n\n      const credentials = await httpAuth.credentials(req);\n      const { userEntityRef } = await userInfo.getUserInfo(credentials);\n\n      const responseMessages = await chat.prompt({\n        modelId,\n        messages,\n        conversationId,\n        stream,\n        userEntityRef,\n      });\n\n      res.json({\n        messages: responseMessages,\n        conversationId,\n      });\n    },\n  );\n\n  const chatSchema = z.object({\n    id: z.string().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('/: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  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,cAAA,EAAgBA,mBAAE,MAAA,EAAO,CAAE,MAAK,CAAE,QAAA,EAAS,CAAE,OAAA,CAAQC,OAAI,CAAA;AAAA,IACzD,MAAA,EAAQD,kBAAA,CAAE,OAAA,EAAQ,CAAE,QAAA;AAAS,GAC9B,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,OAAA,EAAS,MAAA,KAAW,GAAA,CAAI,IAAA;AAE1D,MAAA,MAAM,WAAA,GAAc,MAAM,QAAA,CAAS,WAAA,CAAY,GAAG,CAAA;AAClD,MAAA,MAAM,EAAE,aAAA,EAAc,GAAI,MAAM,QAAA,CAAS,YAAY,WAAW,CAAA;AAEhE,MAAA,MAAM,gBAAA,GAAmB,MAAM,IAAA,CAAK,MAAA,CAAO;AAAA,QACzC,OAAA;AAAA,QACA,QAAA;AAAA,QACA,cAAA;AAAA,QACA,MAAA;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,kBAAA,CAAE,MAAA,EAAO,CAAE,IAAA;AAAK,GACrB,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,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,OAAO,MAAA;AACT;;;;"}

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

@@ -0,0 +1,25 @@
+'use strict';
+
+var express = require('express');
+var Router = require('express-promise-router');
+var chat = require('./chat.cjs.js');
+var models = require('./models.cjs.js');
+var rootHttpRouter = require('@backstage/backend-defaults/rootHttpRouter');
+
+function _interopDefaultCompat (e) { return e && typeof e === 'object' && 'default' in e ? e : { default: e }; }
+
+var express__default = /*#__PURE__*/_interopDefaultCompat(express);
+var Router__default = /*#__PURE__*/_interopDefaultCompat(Router);
+
+async function createRouter(options) {
+  const router = Router__default.default();
+  router.use(express__default.default.json());
+  router.use("/chat", await chat.createChatRouter(options));
+  router.use("/models", await models.createModelRouter(options));
+  const middleware = rootHttpRouter.MiddlewareFactory.create(options);
+  router.use(middleware.error());
+  return router;
+}
+
+exports.createRouter = createRouter;
+//# sourceMappingURL=index.cjs.js.map

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

@@ -0,0 +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 } from './models';\nimport {\n  LoggerService,\n  RootConfigService,\n} from '@backstage/backend-plugin-api';\nimport { MiddlewareFactory } from '@backstage/backend-defaults/rootHttpRouter';\n\nexport type RouterOptions = ChatRouterOptions & {\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\n  const middleware = MiddlewareFactory.create(options);\n\n  router.use(middleware.error());\n\n  return router;\n}\n"],"names":["Router","express","createChatRouter","createModelRouter","MiddlewareFactory"],"mappings":";;;;;;;;;;;;;AAeA,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;AAEtD,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/middleware/validation.cjs.js

@@ -0,0 +1,19 @@
+'use strict';
+
+const validation = (schema, key) => {
+  return (req, res, next) => {
+    const parsed = schema.safeParse(req[key]);
+    if (!parsed.success) {
+      const errors = parsed.error.issues.map(
+        (issue) => `Validation Error:Field ${issue.path.join(".")} - ${issue.message}`
+      );
+      res.status(400).send({ errors });
+      return;
+    }
+    req[key] = parsed.data;
+    next();
+  };
+};
+
+exports.validation = validation;
+//# sourceMappingURL=validation.cjs.js.map

package/dist/services/router/middleware/validation.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"validation.cjs.js","sources":["../../../../src/services/router/middleware/validation.ts"],"sourcesContent":["import express from 'express';\nimport { ZodObject } from 'zod';\n\ntype ValidationKey = 'body' | 'query' | 'params' | 'headers';\n\nexport const validation = (schema: ZodObject, key: ValidationKey) => {\n  return (\n    req: express.Request,\n    res: express.Response,\n    next: express.NextFunction,\n  ) => {\n    const parsed = schema.safeParse(req[key]);\n    if (!parsed.success) {\n      const errors = parsed.error.issues.map(\n        issue =>\n          `Validation Error:Field ${issue.path.join('.')} - ${issue.message}`,\n      );\n      res.status(400).send({ errors });\n      return;\n    }\n    req[key] = parsed.data;\n    next();\n  };\n};\n"],"names":[],"mappings":";;AAKO,MAAM,UAAA,GAAa,CAAC,MAAA,EAAmB,GAAA,KAAuB;AACnE,EAAA,OAAO,CACL,GAAA,EACA,GAAA,EACA,IAAA,KACG;AACH,IAAA,MAAM,MAAA,GAAS,MAAA,CAAO,SAAA,CAAU,GAAA,CAAI,GAAG,CAAC,CAAA;AACxC,IAAA,IAAI,CAAC,OAAO,OAAA,EAAS;AACnB,MAAA,MAAM,MAAA,GAAS,MAAA,CAAO,KAAA,CAAM,MAAA,CAAO,GAAA;AAAA,QACjC,CAAA,KAAA,KACE,0BAA0B,KAAA,CAAM,IAAA,CAAK,KAAK,GAAG,CAAC,CAAA,GAAA,EAAM,KAAA,CAAM,OAAO,CAAA;AAAA,OACrE;AACA,MAAA,GAAA,CAAI,OAAO,GAAG,CAAA,CAAE,IAAA,CAAK,EAAE,QAAQ,CAAA;AAC/B,MAAA;AAAA,IACF;AACA,IAAA,GAAA,CAAI,GAAG,IAAI,MAAA,CAAO,IAAA;AAClB,IAAA,IAAA,EAAK;AAAA,EACP,CAAA;AACF;;;;"}

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

@@ -0,0 +1,20 @@
+'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 createModelRouter(options) {
+  const { chat } = options;
+  const router = Router__default.default();
+  router.get("/", async (_req, res) => {
+    const models = await chat.getAvailableModels();
+    res.json({ models });
+  });
+  return router;
+}
+
+exports.createModelRouter = createModelRouter;
+//# sourceMappingURL=models.cjs.js.map

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

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