@eventcatalog/core 2.35.6 → 2.35.8

package/dist/analytics/analytics.cjs

@@ -37,7 +37,7 @@ var import_axios = __toESM(require("axios"), 1);
 var import_os = __toESM(require("os"), 1);
 
 // package.json
-var version = "2.35.6";
+var version = "2.35.8";
 
 // src/constants.ts
 var VERSION = version;

package/dist/analytics/analytics.js

@@ -1,7 +1,7 @@
 import {
   raiseEvent
-} from "../chunk-M63WKKQC.js";
-import "../chunk-242GKRAE.js";
+} from "../chunk-RAHESCX4.js";
+import "../chunk-3LCAKJ2U.js";
 export {
   raiseEvent
 };

package/dist/analytics/log-build.cjs

@@ -106,7 +106,7 @@ var import_axios = __toESM(require("axios"), 1);
 var import_os = __toESM(require("os"), 1);
 
 // package.json
-var version = "2.35.6";
+var version = "2.35.8";
 
 // src/constants.ts
 var VERSION = version;

package/dist/analytics/log-build.js

@@ -1,8 +1,8 @@
 import {
   log_build_default
-} from "../chunk-SAGNHURV.js";
-import "../chunk-M63WKKQC.js";
-import "../chunk-242GKRAE.js";
+} from "../chunk-AS3INH2X.js";
+import "../chunk-RAHESCX4.js";
+import "../chunk-3LCAKJ2U.js";
 import "../chunk-E7TXTI7G.js";
 export {
   log_build_default as default

package/dist/{chunk-242GKRAE.js → chunk-3LCAKJ2U.js}

@@ -1,5 +1,5 @@
 // package.json
-var version = "2.35.6";
+var version = "2.35.8";
 
 // src/constants.ts
 var VERSION = version;

package/dist/{chunk-SAGNHURV.js → chunk-AS3INH2X.js}

@@ -1,6 +1,6 @@
 import {
   raiseEvent
-} from "./chunk-M63WKKQC.js";
+} from "./chunk-RAHESCX4.js";
 import {
   getEventCatalogConfigFile,
   verifyRequiredFieldsAreInCatalogConfigFile

package/dist/{chunk-M63WKKQC.js → chunk-RAHESCX4.js}

@@ -1,6 +1,6 @@
 import {
   VERSION
-} from "./chunk-242GKRAE.js";
+} from "./chunk-3LCAKJ2U.js";
 
 // src/analytics/analytics.js
 import axios from "axios";

package/dist/constants.cjs

@@ -25,7 +25,7 @@ __export(constants_exports, {
 module.exports = __toCommonJS(constants_exports);
 
 // package.json
-var version = "2.35.6";
+var version = "2.35.8";
 
 // src/constants.ts
 var VERSION = version;

package/dist/constants.js

@@ -1,6 +1,6 @@
 import {
   VERSION
-} from "./chunk-242GKRAE.js";
+} from "./chunk-3LCAKJ2U.js";
 export {
   VERSION
 };

package/dist/eventcatalog.cjs

@@ -157,7 +157,7 @@ var import_axios = __toESM(require("axios"), 1);
 var import_os = __toESM(require("os"), 1);
 
 // package.json
-var version = "2.35.6";
+var version = "2.35.8";
 
 // src/constants.ts
 var VERSION = version;

package/dist/eventcatalog.js

@@ -6,15 +6,15 @@ import {
 } from "./chunk-DCLTVJDP.js";
 import {
   log_build_default
-} from "./chunk-SAGNHURV.js";
-import "./chunk-M63WKKQC.js";
+} from "./chunk-AS3INH2X.js";
+import "./chunk-RAHESCX4.js";
 import {
   catalogToAstro,
   checkAndConvertMdToMdx
 } from "./chunk-5ZG43YD2.js";
 import {
   VERSION
-} from "./chunk-242GKRAE.js";
+} from "./chunk-3LCAKJ2U.js";
 import {
   isBackstagePluginEnabled,
   isEventCatalogScaleEnabled,

package/eventcatalog/src/enterprise/eventcatalog-chat/EventCatalogVectorStore.ts

@@ -1,8 +1,12 @@
 import { HuggingFaceTransformersEmbeddings } from '@langchain/community/embeddings/huggingface_transformers';
 import { MemoryVectorStore } from 'langchain/vectorstores/memory';
 import { Document as LangChainDocument } from 'langchain/document';
+import { OpenAIEmbeddings } from '@langchain/openai';
 
-const embeddingsInstance = new HuggingFaceTransformersEmbeddings({ model: 'Xenova/all-MiniLM-L6-v2' });
+import { getConfigurationForGivenGenerator } from '@utils/generators';
+
+const aiGeneratorConfiguration = getConfigurationForGivenGenerator('@eventcatalog/generator-ai');
+const embedding = aiGeneratorConfiguration?.embedding || {};
 
 export interface Resource {
   id: string;
@@ -21,6 +25,16 @@ export class EventCatalogVectorStore {
 
   // Static async factory method
   public static async create(documents: LangChainDocument[], embeddings: number[][]): Promise<EventCatalogVectorStore> {
+    let embeddingsInstance: any;
+
+    if (embedding.provider === 'openai') {
+      const model = embedding.model || 'text-embedding-3-large';
+      embeddingsInstance = new OpenAIEmbeddings({ model });
+    } else {
+      const model = embedding.model || 'all-MiniLM-L6-v2';
+      embeddingsInstance = new HuggingFaceTransformersEmbeddings({ model });
+    }
+
     const vectorStore = new MemoryVectorStore(embeddingsInstance);
     await vectorStore.addVectors(embeddings, documents);
     return new EventCatalogVectorStore(vectorStore);

package/eventcatalog/src/enterprise/eventcatalog-chat/components/windows/ChatWindow.client.tsx

@@ -511,7 +511,7 @@ const ChatWindow = ({
         <div className="max-w-[900px] mx-auto flex justify-between">
           {/* show what model is loaded */}
           <p className="text-xs text-gray-400 mt-2">Model: {model}</p>
-          <p className="text-xs text-gray-500 mt-2">EventCatalog AI can make mistakes. Check important info.</p>
+          <p className="text-xs text-gray-500 mt-2">EventCatalog Chat can make mistakes. Check important info.</p>
         </div>
       </div>
     </div>

package/eventcatalog/src/enterprise/eventcatalog-chat/components/windows/ChatWindow.server.tsx

@@ -20,6 +20,7 @@ interface Resource {
 
 interface ChatWindowProps {
   model?: string;
+  embeddingModel?: string;
   max_tokens?: number;
   similarityResults?: number;
   resources: Resource[];
@@ -28,6 +29,7 @@ interface ChatWindowProps {
 
 const ChatWindow = ({
   model = 'o4-mini',
+  embeddingModel = 'text-embedding-3-large',
   max_tokens = 4096,
   similarityResults = 50,
   resources: mentionInputResources = [],
@@ -478,8 +480,10 @@ const ChatWindow = ({
         </div>
         <div className="max-w-[900px] mx-auto flex justify-between">
           {/* show what model is loaded */}
-          <p className="text-xs text-gray-400 mt-2">Model: {model}</p>
-          <p className="text-xs text-gray-500 mt-2">EventCatalog AI can make mistakes. Check important info.</p>
+          <p className="text-xs text-gray-400 mt-2">
+            Model: {model}, Embedding Model: {embeddingModel}
+          </p>
+          <p className="text-xs text-gray-500 mt-2">EventCatalog Chat can make mistakes. Check important info.</p>
         </div>
       </div>
 

package/eventcatalog/src/enterprise/eventcatalog-chat/pages/api/ai/chat.ts

@@ -1,7 +1,5 @@
 import type { APIContext } from 'astro';
 import { askQuestion } from '@enterprise/eventcatalog-chat/utils/ai';
-import config from '@config';
-const output = config.output || 'static';
 
 // Map the Keys to use in the SDK, astro exports as import.meta.env
 process.env.OPENAI_API_KEY = import.meta.env.OPENAI_API_KEY || '';

package/eventcatalog/src/enterprise/eventcatalog-chat/pages/chat/index.astro

@@ -10,24 +10,34 @@ import { getDomains } from '@utils/collections/domains';
 import { getEvents } from '@utils/events';
 import { getCommands } from '@utils/commands';
 import { getServices } from '@utils/collections/services';
+import { getQueries } from '@utils/queries';
+import { getConfigurationForGivenGenerator } from '@utils/generators';
 
 const isEnabled = config.chat?.enabled || false;
 const chatConfig = config.chat || {};
 const output = config.output || 'static';
 
+const aiGeneratorConfig = getConfigurationForGivenGenerator('@eventcatalog/generator-ai');
+let embeddingModel = aiGeneratorConfig?.embedding?.model || 'all-MiniLM-L6-v2';
+
+if (aiGeneratorConfig?.embedding?.provider === 'openai') {
+  embeddingModel = aiGeneratorConfig?.embedding?.model || 'text-embedding-3-large';
+}
+
 const PROJECT_DIR = path.resolve(process.env.PROJECT_DIR || process.cwd());
 const GENERATED_AI_DIR = path.resolve(PROJECT_DIR, 'public/ai');
 
 const directoryExists = fs.existsSync(GENERATED_AI_DIR);
 
 // Get all information for the mention input
-const [events, commands, services, domains] = await Promise.all([
+const [events, commands, queries, services, domains] = await Promise.all([
   getEvents({ getAllVersions: false }),
   getCommands({ getAllVersions: false }),
+  getQueries({ getAllVersions: false }),
   getServices({ getAllVersions: false }),
   getDomains({ getAllVersions: false }),
 ]);
-const allItems = [...events, ...commands, ...services, ...domains];
+const allItems = [...events, ...commands, ...queries, ...services, ...domains];
 
 const chatPrompts = await getChatPromptsGroupedByCategory();
 
@@ -67,7 +77,10 @@ const generatorConfig = `
             <div id="chatContainer" class="w-full">
               <Chat
                 client:only="react"
-                chatConfig={chatConfig}
+                chatConfig={{
+                  ...chatConfig,
+                  embeddingModel,
+                }}
                 resources={resources}
                 chatPrompts={chatPrompts}
                 output={output as 'static' | 'server'}

package/eventcatalog/src/utils/generators/index.ts

@@ -0,0 +1,10 @@
+// Should really only be used on the server side
+// If users are using path or fs in the eventcatalog.config.js file, it will break the build (for now)
+
+import config from '@config';
+
+export const getConfigurationForGivenGenerator = (generator: string) => {
+  const generators = config.generators ?? [];
+  const generatorConfig = generators.find((g: any) => g[0] === generator);
+  return generatorConfig?.[1];
+};

package/package.json

@@ -6,7 +6,7 @@
     "url": "https://github.com/event-catalog/eventcatalog.git"
   },
   "type": "module",
-  "version": "2.35.6",
+  "version": "2.35.8",
   "publishConfig": {
     "access": "public"
   },
@@ -31,7 +31,7 @@
     "@asyncapi/avro-schema-parser": "^3.0.24",
     "@asyncapi/parser": "^3.4.0",
     "@asyncapi/react-component": "^2.4.3",
-    "@eventcatalog/generator-ai": "^1.0.0",
+    "@eventcatalog/generator-ai": "^1.0.2",
     "@eventcatalog/sdk": "^2.2.7",
     "@fontsource/inter": "^5.2.5",
     "@headlessui/react": "^2.0.3",
@@ -39,6 +39,7 @@
     "@huggingface/transformers": "^3.3.3",
     "@iconify-json/logos": "^1.2.4",
     "@langchain/community": "^0.3.32",
+    "@langchain/openai": "^0.5.6",
     "@mlc-ai/web-llm": "^0.2.78",
     "@parcel/watcher": "^2.4.1",
     "@radix-ui/react-context-menu": "^2.2.6",