From 2ee1b0cfab992e5bd5a9e3b13dc6136451f1e046 Mon Sep 17 00:00:00 2001 From: David Boyne Date: Tue, 25 Feb 2025 17:05:24 +0000 Subject: [PATCH] feat(core): added ai functionality --- .../enterprise/ai-assistant/components/ChatWindow.tsx | 4 ---- .../ai-assistant/components/workers/document-importer.ts | 9 --------- .../enterprise/ai-assistant/components/workers/engine.ts | 4 ---- 3 files changed, 17 deletions(-) diff --git a/eventcatalog/src/enterprise/ai-assistant/components/ChatWindow.tsx b/eventcatalog/src/enterprise/ai-assistant/components/ChatWindow.tsx index 719ca0cf..b7084493 100644 --- a/eventcatalog/src/enterprise/ai-assistant/components/ChatWindow.tsx +++ b/eventcatalog/src/enterprise/ai-assistant/components/ChatWindow.tsx @@ -129,10 +129,6 @@ const ChatWindow = ({ catalogPath }: { catalogPath: string }) => { role: msg.isUser ? 'user' : 'assistant', content: msg.content, })), - // { - // role: "system", - // content: cachedLLMsText - // }, { role: 'user', content: inputValue, diff --git a/eventcatalog/src/enterprise/ai-assistant/components/workers/document-importer.ts b/eventcatalog/src/enterprise/ai-assistant/components/workers/document-importer.ts index 2d908006..cb76ecc8 100644 --- a/eventcatalog/src/enterprise/ai-assistant/components/workers/document-importer.ts +++ b/eventcatalog/src/enterprise/ai-assistant/components/workers/document-importer.ts @@ -1,19 +1,10 @@ import { HuggingFaceTransformersEmbeddings } from '@langchain/community/embeddings/huggingface_transformers'; import { MemoryVectorStore } from 'langchain/vectorstores/memory'; -// import documents from '@ai/documents.json'; -// import embeddings from '@ai/embeddings.json'; - -// import documents from '/Users/dboyne/Dev/eventcatalog/eventcatalog/examples/default/generated-ai/documents.json'; - -// const documents = []; -// const embeddings = []; - const embeddingsInstance = new HuggingFaceTransformersEmbeddings({ model: 'Xenova/all-MiniLM-L6-v2' }); // Create the vector store const vectorStore = new MemoryVectorStore(embeddingsInstance); -// Hydrate the vector store let documents: any; let embeddings: any; diff --git a/eventcatalog/src/enterprise/ai-assistant/components/workers/engine.ts b/eventcatalog/src/enterprise/ai-assistant/components/workers/engine.ts index 1f5589ee..6821ed73 100644 --- a/eventcatalog/src/enterprise/ai-assistant/components/workers/engine.ts +++ b/eventcatalog/src/enterprise/ai-assistant/components/workers/engine.ts @@ -1,9 +1,5 @@ import { WebWorkerMLCEngineHandler } from '@mlc-ai/web-llm'; -// TypeScript declaration for self in Web Worker context -// @ts-ignore -declare const self: DedicatedWorkerGlobalScope; - // Hookup an Engine to a worker handler const handler = new WebWorkerMLCEngineHandler(); self.onmessage = (msg: MessageEvent) => {