@eventcatalog/core 3.0.0-beta.2 → 3.0.0-beta.21

package/eventcatalog/src/enterprise/eventcatalog-chat/components/default-prompts.ts

@@ -1,93 +0,0 @@
-import type { ChatPrompt } from '@enterprise/eventcatalog-chat/utils/chat-prompts';
-
-export const defaultPrompts: ChatPrompt[] = [
-  {
-    id: 'default-events',
-    collection: 'chatPrompts', // Required field
-    body: 'List all events.', // The actual prompt text
-    data: {
-      title: 'What events do we have in our architecture?', // Text displayed on the button
-      type: 'text', // Default type
-      category: { id: 'general', label: 'General', icon: 'HelpCircle' },
-    },
-  },
-  {
-    id: 'default-events-by-domain-feature',
-    collection: 'chatPrompts',
-    body: 'What events are relevant to this feature {{feature-description}}?',
-    data: {
-      title: 'Im building a new feature, what events are relevant to this feature?',
-      type: 'text',
-      inputs: [
-        {
-          id: 'feature-description',
-          label: 'Feature Description',
-          type: 'text-area',
-        },
-      ],
-      category: { id: 'general', label: 'General', icon: 'HelpCircle' },
-    },
-  },
-  {
-    id: 'default-events-by-domain-service',
-    collection: 'chatPrompts',
-    body: 'Review the given service {{service-name}}, and let me know how it works? You are an expert in the domain and architecture of the service.',
-    data: {
-      title: 'Review the given service, and let me know how it works?',
-      type: 'text',
-      inputs: [
-        {
-          id: 'service-name',
-          label: 'Service Name',
-          type: 'resource-list-services',
-        },
-      ],
-      category: { id: 'general', label: 'General', icon: 'HelpCircle' },
-    },
-  },
-  {
-    id: 'default-services',
-    collection: 'chatPrompts',
-    body: 'List all services.',
-    data: {
-      title: 'What services do we have in our architecture?',
-      type: 'text',
-      category: { id: 'general', label: 'General', icon: 'HelpCircle' },
-    },
-  },
-  // Example of another category
-  {
-    id: 'default-schema-json',
-    collection: 'chatPrompts',
-    body: 'Generate a JSON schema for {{event-name}}.',
-    data: {
-      title: 'Generate a JSON schema for the given event',
-      type: 'text',
-      inputs: [
-        {
-          id: 'event-name',
-          label: 'Event Name',
-          type: 'resource-list-events',
-        },
-      ],
-      category: { id: 'code', label: 'Code Generation', icon: 'Code' },
-    },
-  },
-  {
-    id: 'default-schema-avro',
-    collection: 'chatPrompts',
-    body: 'Generate a Avro schema for {{event-name}}.',
-    data: {
-      title: 'Generate a Avro schema for the given event',
-      type: 'text',
-      inputs: [
-        {
-          id: 'event-name',
-          label: 'Event Name',
-          type: 'resource-list-events',
-        },
-      ],
-      category: { id: 'code', label: 'Code Generation', icon: 'Code' },
-    },
-  },
-];

package/eventcatalog/src/enterprise/eventcatalog-chat/components/hooks/ChatProvider.tsx

@@ -1,143 +0,0 @@
-import type { UIMessage } from '@ai-sdk/react';
-import React, { createContext, useContext, useEffect, useState } from 'react';
-import type { ReactNode } from 'react';
-
-export interface Message {
-  content: string;
-  additionalContext?: string;
-  isUser: boolean;
-  timestamp: number;
-  resources?: Array<{
-    id: string;
-    type: string;
-    url: string;
-    title?: string;
-  }>;
-}
-
-export interface ChatSession {
-  id: string;
-  title: string;
-  messages: Message[];
-  lastUpdated: number;
-  createdAt: number;
-}
-
-interface ChatContextType {
-  sessions: ChatSession[];
-  currentSession: ChatSession | null;
-  createSession: () => void;
-  updateSession: (session: ChatSession) => void;
-  deleteSession: (id: string) => void;
-  setCurrentSession: (session: ChatSession | null) => void;
-  isStreaming: boolean;
-  setIsStreaming: (isStreaming: boolean) => void;
-  storeMessagesToSession: (sessionId: string, messages: UIMessage[]) => void;
-}
-
-const ChatContext = createContext<ChatContextType | undefined>(undefined);
-
-const STORAGE_KEY = 'chat_sessions';
-
-export function ChatProvider({ children }: { children: ReactNode }) {
-  const [sessions, setSessions] = useState<ChatSession[]>(() => {
-    if (typeof window !== 'undefined') {
-      const stored = localStorage.getItem(STORAGE_KEY);
-      return stored ? JSON.parse(stored) : [];
-    }
-    return [];
-  });
-  const [currentSession, setCurrentSession] = useState<ChatSession | null>(null);
-  const [isStreaming, setIsStreaming] = useState(false);
-
-  const createSession = () => {
-    const newSession: ChatSession = {
-      id: crypto.randomUUID(),
-      title: `Chat ${sessions.length + 1}`,
-      messages: [],
-      lastUpdated: Date.now(),
-      createdAt: Date.now(),
-    };
-
-    setSessions((prev) => {
-      const updated = [newSession, ...prev];
-      localStorage.setItem(STORAGE_KEY, JSON.stringify(updated));
-      return updated;
-    });
-    setCurrentSession(newSession);
-  };
-
-  const updateSession = (session: ChatSession) => {
-    setSessions((prev) => {
-      const index = prev.findIndex((s) => s.id === session.id);
-      if (index === -1) return prev;
-
-      const updated = [...prev];
-      updated[index] = {
-        ...session,
-        lastUpdated: Date.now(),
-      };
-      localStorage.setItem(STORAGE_KEY, JSON.stringify(updated));
-      return updated;
-    });
-  };
-
-  const deleteSession = (id: string) => {
-    setSessions((prev) => {
-      const updated = prev.filter((session) => session.id !== id);
-      localStorage.setItem(STORAGE_KEY, JSON.stringify(updated));
-      return updated;
-    });
-    if (currentSession?.id === id) {
-      setCurrentSession(null);
-    }
-  };
-
-  // Set all messages to the session
-  const storeMessagesToSession = (sessionId: string, messages: UIMessage[]) => {
-    setSessions((prev) => {
-      const index = prev.findIndex((s) => s.id === sessionId);
-      if (index === -1) return prev;
-
-      const updated = [...prev];
-      updated[index].messages = messages as unknown as Message[];
-      updated[index].lastUpdated = Date.now();
-      localStorage.setItem(STORAGE_KEY, JSON.stringify(updated));
-      return updated;
-    });
-  };
-
-  useEffect(() => {
-    if (sessions.length > 0) {
-      setCurrentSession(sessions[0]);
-    } else {
-      createSession();
-    }
-  }, []);
-
-  return (
-    <ChatContext.Provider
-      value={{
-        sessions,
-        currentSession,
-        createSession,
-        updateSession,
-        deleteSession,
-        setCurrentSession,
-        isStreaming,
-        setIsStreaming,
-        storeMessagesToSession,
-      }}
-    >
-      {children}
-    </ChatContext.Provider>
-  );
-}
-
-export function useChat() {
-  const context = useContext(ChatContext);
-  if (context === undefined) {
-    throw new Error('useChat must be used within a ChatProvider');
-  }
-  return context;
-}

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

@@ -1,387 +0,0 @@
-import { useEffect, useState, useRef, useCallback, useMemo } from 'react';
-import { Send } from 'lucide-react';
-import { useChat, type Message } from '../hooks/ChatProvider';
-import React from 'react';
-import MentionInput from '../MentionInput';
-import InputModal from '../InputModal';
-import type { ChatPromptCategoryGroup, ChatPrompt } from '@enterprise/eventcatalog-chat/utils/chat-prompts';
-import WelcomePromptArea from '../WelcomePromptArea';
-import ChatMessage from '../ChatMessage'; // Import the new component
-import { useChat as useAiChat, type UIMessage } from '@ai-sdk/react';
-
-// Update Message type to include resources
-interface Resource {
-  id: string;
-  type: string;
-  url: string;
-  title?: string;
-  name?: string;
-}
-
-interface ChatWindowProps {
-  model?: string;
-  provider?: string;
-  embeddingModel?: string;
-  max_tokens?: number;
-  similarityResults?: number;
-  resources: Resource[];
-  chatPrompts: ChatPromptCategoryGroup[];
-}
-
-const ChatWindow = ({
-  model = 'o4-mini',
-  embeddingModel = 'text-embedding-3-large',
-  provider = 'openai',
-  max_tokens = 4096,
-  similarityResults = 50,
-  resources: mentionInputResources = [],
-  chatPrompts,
-}: ChatWindowProps) => {
-  // const [messages] = useState<Array<Message>>([]);
-  const [inputValue, setInputValue] = useState('');
-  const outputRef = useRef<HTMLDivElement>(null);
-  const inputRef = useRef<HTMLInputElement>(null);
-  const [activeCategory, setActiveCategory] = useState<string>(chatPrompts?.[0]?.label || '');
-
-  const { messages, sendMessage, stop, status, setMessages: setAiMessages } = useAiChat();
-  const isStreaming = status === 'streaming' && messages.length > 0;
-  const isThinking = status === 'submitted' || (status === 'streaming' && messages.length === 0);
-
-  // --- New state for input modal ---
-  const [promptForInput, setPromptForInput] = useState<ChatPrompt | null>(null);
-  const [isInputModalOpen, setIsInputModalOpen] = useState(false);
-  // --- End new state ---
-
-  const { currentSession, storeMessagesToSession, updateSession, createSession } = useChat();
-
-  // If the messages change add them to the session
-  useEffect(() => {
-    if (currentSession) {
-      storeMessagesToSession(currentSession.id, messages);
-    }
-  }, [messages]);
-
-  // Load messages when session changes
-  useEffect(() => {
-    if (currentSession) {
-      setAiMessages(currentSession.messages as unknown as UIMessage[]);
-    }
-  }, [currentSession]);
-
-  // Add effect to focus input when streaming stops
-  useEffect(() => {
-    if (!isStreaming && inputRef.current) {
-      inputRef.current.focus();
-    }
-  }, [isStreaming]);
-
-  // New function to handle submitting a question (user input or predefined)
-  const submitQuestion = useCallback(
-    async (question: string, additionalContext?: string) => {
-      if (!question.trim() || isStreaming || isThinking) return;
-
-      const userMessage: Message = { content: question, isUser: true, timestamp: Date.now(), additionalContext };
-      const isFirstMessage = messages.length === 0;
-
-      // setMessages((prev) => [...prev, userMessage]);
-      // setShowWelcome(false);
-      setInputValue('');
-
-      // Scroll to bottom immediately after adding user message and setting thinking state
-      requestAnimationFrame(() => scrollToBottom(false));
-
-      if (currentSession && isFirstMessage) {
-        updateSession({
-          ...currentSession,
-          // Use the submitted question (potentially the prompt title) for the session title
-          title: question.length > 25 ? `${question.substring(0, 22)}...` : question,
-        });
-      }
-
-      sendMessage({ text: additionalContext || question });
-
-      // mutation.mutate({ question, additionalContext });
-    },
-    [
-      currentSession,
-      updateSession,
-      // setMessages,
-      setInputValue,
-      messages.length,
-      isStreaming,
-      isThinking,
-      messages,
-    ]
-  );
-
-  // --- New handler for submitting from the modal ---
-  const handleSubmitWithInputs = useCallback(
-    (prompt: ChatPrompt, inputValues: Record<string, string>) => {
-      let finalBody = prompt.body || ''; // Start with the original body
-      // Ensure prompt and prompt.data exist before accessing properties
-      if (!prompt || !prompt.data) {
-        console.error('handleSubmitWithInputs called without a valid prompt.');
-        setIsInputModalOpen(false); // Close modal even on error
-        setPromptForInput(null);
-        return;
-      }
-
-      for (const [key, value] of Object.entries(inputValues)) {
-        const placeholder = `{{${key}}}`;
-        // Replace all occurrences of the placeholder in the body
-        finalBody = finalBody.replaceAll(placeholder, `"${value}"`);
-      }
-
-      // Submit the processed title and the processed body as additional context
-      submitQuestion(prompt.data.title, finalBody);
-
-      setIsInputModalOpen(false); // Close modal
-      setPromptForInput(null); // Clear stored prompt
-    },
-    [submitQuestion]
-  );
-
-  // --- Modified handler for clicking a predefined question ---
-  const handlePredefinedQuestionClick = useCallback(
-    (prompt: ChatPrompt) => {
-      // Ensure prompt and prompt.data exist
-      if (!prompt || !prompt.data) {
-        console.error('handlePredefinedQuestionClick called with invalid prompt:', prompt);
-        return;
-      }
-      // Check if prompt.data and prompt.data.inputs exist and have length > 0
-      if (prompt.data?.inputs && prompt.data.inputs.length > 0) {
-        setPromptForInput(prompt); // Store the prompt
-        setIsInputModalOpen(true); // Open the modal
-      } else {
-        // No inputs needed, submit directly using title and body
-        submitQuestion(prompt.data.title, prompt.body);
-      }
-    },
-    [submitQuestion]
-  );
-
-  // Handler for standard input submission
-  const handleSubmit = useCallback(
-    (e?: React.FormEvent) => {
-      e?.preventDefault();
-      // sendMessage({ text: inputValue });
-      submitQuestion(inputValue); // Use standard input value, no additional context here
-    },
-    [inputValue, sendMessage]
-  );
-
-  // Add new function to handle smooth scrolling
-  const scrollToBottom = useCallback((smooth = true) => {
-    if (outputRef.current) {
-      outputRef.current.scrollTo({
-        top: outputRef.current.scrollHeight,
-        behavior: smooth ? 'smooth' : 'auto',
-      });
-    }
-  }, []);
-
-  // Add effect to scroll when messages change or thinking state changes
-  useEffect(() => {
-    // Scroll immediately for new messages or when thinking starts/stops
-    requestAnimationFrame(() => scrollToBottom(messages.length > 0 && !isThinking));
-  }, [messages, isThinking, scrollToBottom]);
-
-  // Memoize the messages list with the new ChatMessage component
-  const messagesList = useMemo(
-    () => (
-      <div className="space-y-4 max-w-[900px] mx-auto pb-6">
-        {messages.map((message, index) => (
-          <ChatMessage
-            key={message.id}
-            message={{
-              isUser: message.role === 'user',
-              content: message.parts.map((part) => (part.type === 'text' ? part.text : '')).join(''),
-              timestamp: Date.now(),
-            }}
-          />
-        ))}
-        {isThinking && (
-          <div className="flex justify-start mb-4">
-            <div className="flex items-center space-x-2 max-w-[80%] rounded-lg p-3 bg-gray-100 text-gray-800 rounded-bl-none">
-              <div className="flex space-x-1">
-                <div className="w-2 h-2 bg-gray-500 rounded-full animate-bounce" style={{ animationDelay: '0ms' }}></div>
-                <div className="w-2 h-2 bg-gray-500 rounded-full animate-bounce" style={{ animationDelay: '150ms' }}></div>
-                <div className="w-2 h-2 bg-gray-500 rounded-full animate-bounce" style={{ animationDelay: '300ms' }}></div>
-              </div>
-            </div>
-          </div>
-        )}
-      </div>
-    ),
-    [messages, isThinking]
-  );
-
-  // Memoize the input change handler
-  const handleInputChange = useCallback((newValue: string) => {
-    setInputValue(newValue);
-  }, []);
-
-  // Memoize the key press handler for MentionInput
-  const handleInputKeyPress = useCallback(
-    (e: React.KeyboardEvent<HTMLInputElement>) => {
-      // Submit only if Enter is pressed WITHOUT Shift and suggestions are NOT shown
-      if (e.key === 'Enter' && !e.shiftKey) {
-        // The MentionInput's internal onKeyDown will handle preventDefault
-        // if suggestions are shown and Enter is pressed for selection.
-        // We check here if it *wasn't* handled for selection, meaning we should submit.
-        if (!e.defaultPrevented) {
-          handleSubmit();
-        }
-      }
-    },
-    [handleSubmit]
-  ); // Include handleSubmit in dependencies
-
-  // Effect to update activeCategory if chatPrompts load after initial render
-  useEffect(() => {
-    if (!activeCategory && chatPrompts && chatPrompts.length > 0) {
-      setActiveCategory(chatPrompts[0].label);
-    }
-  }, [chatPrompts, activeCategory]);
-
-  // Add Effect for clipboard copy functionality
-  useEffect(() => {
-    const outputElement = outputRef.current;
-    if (!outputElement) return;
-
-    const handleClick = async (event: MouseEvent) => {
-      const button = (event.target as Element).closest<HTMLButtonElement>('[data-copy-button="true"]');
-      if (!button) return;
-
-      const codeToCopy = button.dataset.code;
-      if (!codeToCopy) return;
-
-      try {
-        await navigator.clipboard.writeText(codeToCopy);
-        // Visual feedback: change icon to Check for a short time
-        const originalIcon = button.innerHTML;
-        button.innerHTML = `<svg xmlns="http://www.w3.org/2000/svg" width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="lucide lucide-check"><polyline points="20 6 9 17 4 12"></polyline></svg>`;
-        button.disabled = true;
-
-        setTimeout(() => {
-          button.innerHTML = originalIcon;
-          button.disabled = false;
-        }, 1500); // Reset after 1.5 seconds
-      } catch (err) {
-        console.error('Failed to copy code: ', err);
-        // Optional: Provide error feedback to the user
-        const originalTitle = button.title;
-        button.title = 'Failed to copy!';
-        setTimeout(() => {
-          button.title = originalTitle;
-        }, 1500);
-      }
-    };
-
-    outputElement.addEventListener('click', handleClick);
-
-    // Cleanup listener on component unmount
-    return () => {
-      outputElement.removeEventListener('click', handleClick);
-    };
-  }, []); // Empty dependency array ensures this runs only once on mount
-
-  // Handle the query from the url
-  useEffect(() => {
-    const urlParams = new URLSearchParams(window.location.search);
-    const query = urlParams.get('query');
-    if (query) {
-      setTimeout(() => {
-        createSession();
-        setInputValue(query);
-      }, 250);
-    }
-  }, []);
-
-  return (
-    <div className="flex-1 flex flex-col overflow-hidden h-[calc(100vh-60px)] w-full bg-white">
-      {messages.length === 0 && (
-        <WelcomePromptArea
-          chatPrompts={chatPrompts}
-          activeCategory={activeCategory}
-          setActiveCategory={setActiveCategory}
-          onPromptClick={handlePredefinedQuestionClick}
-          isProcessing={isStreaming || isThinking}
-        />
-      )}
-
-      {messages.length > 0 && (
-        <div ref={outputRef} className="flex-1 overflow-y-auto">
-          {messages.length > 0 && (
-            <div id="output" className="p-4 space-y-4 w-full max-w-[900px] mx-auto h-full pb-16">
-              {messagesList}
-            </div>
-          )}
-          ;
-        </div>
-      )}
-
-      {/* Input Area (remains at the bottom) */}
-      <div className="border-t border-gray-200 p-4 bg-white">
-        <div className="max-w-[900px] mx-auto relative">
-          {/* Replace standard input with MentionInput */}
-          <MentionInput
-            suggestions={mentionInputResources.map((resource) => ({
-              id: resource.id,
-              name: resource.name || '',
-              type: resource.type,
-            }))}
-            trigger="@"
-            type="text"
-            value={inputValue}
-            onChange={handleInputChange}
-            onKeyDown={handleInputKeyPress}
-            placeholder="Type your message or '@' for events..."
-            className="w-full px-4 py-3 bg-white text-gray-800 rounded-lg border border-gray-200 focus:outline-none focus:border-purple-500 focus:ring-1 focus:ring-purple-500 disabled:bg-gray-50 disabled:cursor-not-allowed pr-24"
-            disabled={isStreaming || isThinking} // Disable input while streaming/thinking
-          />
-          <div className="absolute right-3 top-1/2 -translate-y-1/2">
-            {isStreaming ? (
-              <button
-                onClick={() => stop()}
-                className="px-4 py-2 bg-red-600 text-white rounded-lg hover:bg-red-700 text-sm font-medium"
-              >
-                Stop
-              </button>
-            ) : (
-              <button
-                onClick={handleSubmit}
-                disabled={!inputValue.trim() || isThinking} // Disable send if input empty or thinking
-                className="px-4 py-2 flex items-center bg-purple-600 text-white rounded-lg hover:bg-purple-700 disabled:bg-gray-200 disabled:cursor-not-allowed text-sm font-medium"
-              >
-                {/* Add icon */}
-                <Send size={16} strokeWidth={1.5} className="mr-2" />
-                Send
-              </button>
-            )}
-          </div>
-        </div>
-        <div className="max-w-[900px] mx-auto flex justify-between">
-          {/* show what model is loaded */}
-          <p className="text-[10px] text-gray-400 mt-2">
-            Provider: {provider.charAt(0).toUpperCase() + provider.slice(1)} | 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>
-
-      {/* --- Render Input Modal --- */}
-      <InputModal
-        isOpen={isInputModalOpen}
-        onClose={() => setIsInputModalOpen(false)} // Allow closing the modal
-        prompt={promptForInput}
-        onSubmit={handleSubmitWithInputs}
-        resources={mentionInputResources} // Pass resources here
-      />
-    </div>
-  );
-};
-
-export default ChatWindow;

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

@@ -1,59 +0,0 @@
-import type { APIContext } from 'astro';
-import { type UIMessage } from 'ai';
-import config from '@config';
-import { getProvider } from '@enterprise/eventcatalog-chat/providers';
-
-// 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 || '';
-
-const baseSystemPrompt = `You are an expert in event-driven architecture and domain-driven design, specializing in documentation for EventCatalog.
-
-You assist developers, architects, and business stakeholders who need information about their event-driven system catalog. You help with questions about:
-- Events (asynchronous messages that notify about something that has happened)
-- Commands (requests to perform an action)
-- Queries (requests for information)
-- Services (bounded contexts or applications that produce/consume events)
-- Domains (business capabilities or functional areas)
-
-Your primary goal is to help users understand their event-driven system through accurate documentation interpretation.
-
-When responding:
-2. Explain connections between resources when relevant.
-3. Use appropriate technical terminology.
-4. Use clear formatting with headings and bullet points when helpful.
-5. State clearly when information is missing rather than making assumptions.
-6. Don't provide code examples unless specifically requested.
-
-Your primary goal is to help users understand their event-driven system through accurate documentation interpretation.
-
-If you have additional context, use it to answer the question.`;
-
-interface Message {
-  content: string;
-}
-
-export const GET = async ({ request }: APIContext<{ question: string; messages: Message[]; additionalContext?: string }>) => {
-  // return 404
-  return new Response(JSON.stringify({ error: 'Not found' }), {
-    status: 404,
-    headers: { 'Content-Type': 'application/json' },
-  });
-};
-
-export const POST = async ({ request }: APIContext<{ question: string; messages: Message[]; additionalContext?: string }>) => {
-  const { messages }: { messages: UIMessage[] } = await request.json();
-
-  const provider = getProvider(config?.chat?.provider || 'openai', {
-    modelId: config?.chat?.model || 'gpt-3.5-turbo-0125',
-    temperature: config?.chat?.temperature,
-    topP: config?.chat?.topP,
-    topK: config?.chat?.topK,
-    frequencyPenalty: config?.chat?.frequencyPenalty,
-    presencePenalty: config?.chat?.presencePenalty,
-  });
-
-  const result = await provider.streamText(baseSystemPrompt, messages);
-  return result.toUIMessageStreamResponse();
-};
-
-export const prerender = false;