@promptbook/components 0.104.0-2 → 0.104.0-4

package/esm/typings/src/_packages/types.index.d.ts

@@ -185,6 +185,7 @@ import type { BookTranspiler } from '../transpilers/_common/BookTranspiler';
 import type { BookTranspilerOptions } from '../transpilers/_common/BookTranspilerOptions';
 import type { IntermediateFilesStrategy } from '../types/IntermediateFilesStrategy';
 import type { LlmCall } from '../types/LlmCall';
+import type { Message } from '../types/Message';
 import type { ModelRequirements } from '../types/ModelRequirements';
 import type { CompletionModelRequirements } from '../types/ModelRequirements';
 import type { ChatModelRequirements } from '../types/ModelRequirements';
@@ -551,6 +552,7 @@ export type { BookTranspiler };
 export type { BookTranspilerOptions };
 export type { IntermediateFilesStrategy };
 export type { LlmCall };
+export type { Message };
 export type { ModelRequirements };
 export type { CompletionModelRequirements };
 export type { ChatModelRequirements };

package/esm/typings/src/book-components/Chat/types/ChatMessage.d.ts

@@ -1,22 +1,17 @@
+import { Message } from '../../../types/Message';
 import type { id, string_markdown } from '../../../types/typeAliases';
 /**
  * A message in the chat
  *
  * @public exported from `@promptbook/components`
  */
-export type ChatMessage = {
+export type ChatMessage = Omit<Message<id>, 'direction' | 'recipients' | 'threadId' | 'metadata'> & {
     /**
-     * Unique identifier of the message
-     */
-    id?: id;
-    /**
-     * Date when the message was created
-     */
-    date?: Date;
-    /**
-     * The name of the participant who sent the message
+     * Force the channel to be 'PROMPTBOOK_CHAT'
+     *
+     * @default 'PROMPTBOOK_CHAT'
      */
-    from: id;
+    channel?: 'PROMPTBOOK_CHAT';
     /**
      * The content of the message with optional markdown formatting
      */
@@ -37,6 +32,7 @@ export type ChatMessage = {
     isVoiceCall?: boolean;
 };
 /**
+ * TODO: Make all fields readonly
  * TODO: Delete `expectedAnswer` from ChatMessage
  * TODO: Rename `date` into `created`+`modified`
  */

package/esm/typings/src/llm-providers/_multiple/MultipleLlmExecutionTools.d.ts

@@ -1,8 +1,8 @@
 import type { ChatParticipant } from '../../book-components/Chat/types/ChatParticipant';
 import type { AvailableModel } from '../../execution/AvailableModel';
 import type { LlmExecutionTools } from '../../execution/LlmExecutionTools';
-import type { ChatPromptResult, CompletionPromptResult, EmbeddingPromptResult, PromptResult } from '../../execution/PromptResult';
-import type { ChatPrompt, CompletionPrompt, EmbeddingPrompt, Prompt } from '../../types/Prompt';
+import type { ChatPromptResult, CompletionPromptResult, EmbeddingPromptResult, ImagePromptResult, PromptResult } from '../../execution/PromptResult';
+import type { ChatPrompt, CompletionPrompt, EmbeddingPrompt, ImagePrompt, Prompt } from '../../types/Prompt';
 import type { string_markdown, string_markdown_text, string_title } from '../../types/typeAliases';
 /**
  * Multiple LLM Execution Tools is a proxy server that uses multiple execution tools internally and exposes the executor interface externally.
@@ -43,6 +43,10 @@ export declare class MultipleLlmExecutionTools implements LlmExecutionTools {
      * Calls the best available embedding model
      */
     callEmbeddingModel(prompt: EmbeddingPrompt): Promise<EmbeddingPromptResult>;
+    /**
+     * Calls the best available embedding model
+     */
+    callImageGenerationModel(prompt: ImagePrompt): Promise<ImagePromptResult>;
     /**
      * Calls the best available model
      *

package/esm/typings/src/llm-providers/remote/RemoteLlmExecutionTools.d.ts

@@ -46,6 +46,7 @@ export declare class RemoteLlmExecutionTools<TCustomOptions = undefined> impleme
     private callCommonModel;
 }
 /**
+ * TODO: !!!! Deprecate pipeline server and all of its components
  * TODO: Maybe use `$exportJson`
  * TODO: [🧠][🛍] Maybe not `isAnonymous: boolean` BUT `mode: 'ANONYMOUS'|'COLLECTION'`
  * TODO: [🍓] Allow to list compatible models with each variant

package/esm/typings/src/types/Message.d.ts

@@ -0,0 +1,49 @@
+import { Arrayable } from 'type-fest';
+import { really_any } from '../_packages/types.index';
+import { id, string_date_iso8601, string_markdown } from './typeAliases';
+/**
+ * A generic message structure for various communication channels
+ */
+export type Message<TParticipant> = {
+    /**
+     * Unique identifier of the message
+     */
+    readonly id?: id;
+    /**
+     * Date when the message was created
+     */
+    readonly createdAt?: Date | string_date_iso8601;
+    /**
+     * The communication channel of the message
+     */
+    readonly channel?: 'PROMPTBOOK_CHAT' | 'EMAIL' | 'SMS' | 'WHATSAPP' | 'TELEGRAM' | 'SIGNAL' | string | 'UNKNOWN';
+    /**
+     * Is the message send from the Promptbook or to the Promptbook
+     */
+    readonly direction?: 'INBOUND' | 'OUTBOUND' | 'INTERNAL' | 'INITIAL';
+    /**
+     * Who sent the message
+     */
+    readonly sender: TParticipant;
+    /**
+     * Who are the recipients of the message
+     */
+    readonly recipients?: Readonly<Arrayable<TParticipant>>;
+    /**
+     * The content of the message as markdown
+     *
+     * Note: We are converting all message content to markdown for consistency
+     */
+    readonly content: string_markdown;
+    /**
+     * The thread identifier the message belongs to
+     *
+     * - `null` means the message is not part of any thread
+     * - `undefined` means that we don't know if the message is part of a thread or not
+     */
+    readonly threadId?: id | null;
+    /**
+     * Arbitrary metadata associated with the message
+     */
+    readonly metadata?: Readonly<Record<string, really_any>>;
+};

package/esm/typings/src/version.d.ts

@@ -15,7 +15,7 @@ export declare const BOOK_LANGUAGE_VERSION: string_semantic_version;
 export declare const PROMPTBOOK_ENGINE_VERSION: string_promptbook_version;
 /**
  * Represents the version string of the Promptbook engine.
- * It follows semantic versioning (e.g., `0.104.0-1`).
+ * It follows semantic versioning (e.g., `0.104.0-3`).
  *
  * @generated
  */

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@promptbook/components",
-    "version": "0.104.0-2",
+    "version": "0.104.0-4",
     "description": "Promptbook: Turn your company's scattered knowledge into AI ready books",
     "private": false,
     "sideEffects": false,

package/umd/index.umd.js

@@ -30,7 +30,7 @@
      * @generated
      * @see https://github.com/webgptorg/promptbook
      */
-    const PROMPTBOOK_ENGINE_VERSION = '0.104.0-2';
+    const PROMPTBOOK_ENGINE_VERSION = '0.104.0-4';
     /**
      * TODO: string_promptbook_version should be constrained to the all versions of Promptbook engine
      * Note: [💞] Ignore a discrepancy between file name and entity name
@@ -7114,17 +7114,64 @@
             };
         }
         const lines = agentSource.split('\n');
-        const agentName = (((_a = lines[0]) === null || _a === void 0 ? void 0 : _a.trim()) || null);
+        let agentName = null;
+        let agentNameLineIndex = -1;
+        // Find the agent name: first non-empty line that is not a commitment and not a horizontal line
+        for (let i = 0; i < lines.length; i++) {
+            const line = lines[i];
+            if (line === undefined) {
+                continue;
+            }
+            const trimmed = line.trim();
+            if (!trimmed) {
+                continue;
+            }
+            const isHorizontal = HORIZONTAL_LINE_PATTERN.test(line);
+            if (isHorizontal) {
+                continue;
+            }
+            let isCommitment = false;
+            for (const definition of COMMITMENT_REGISTRY) {
+                const typeRegex = definition.createTypeRegex();
+                const match = typeRegex.exec(trimmed);
+                if (match && ((_a = match.groups) === null || _a === void 0 ? void 0 : _a.type)) {
+                    isCommitment = true;
+                    break;
+                }
+            }
+            if (!isCommitment) {
+                agentName = trimmed;
+                agentNameLineIndex = i;
+                break;
+            }
+        }
         const commitments = [];
         const nonCommitmentLines = [];
-        // Always add the first line (agent name) to non-commitment lines
-        if (lines[0] !== undefined) {
-            nonCommitmentLines.push(lines[0]);
+        // Add lines before agentName that are horizontal lines (they are non-commitment)
+        for (let i = 0; i < agentNameLineIndex; i++) {
+            const line = lines[i];
+            if (line === undefined) {
+                continue;
+            }
+            const trimmed = line.trim();
+            if (!trimmed) {
+                continue;
+            }
+            const isHorizontal = HORIZONTAL_LINE_PATTERN.test(line);
+            if (isHorizontal) {
+                nonCommitmentLines.push(line);
+            }
+            // Note: Commitments before agentName are not added to nonCommitmentLines
+        }
+        // Add the agent name line to non-commitment lines
+        if (agentNameLineIndex >= 0) {
+            nonCommitmentLines.push(lines[agentNameLineIndex]);
         }
         // Parse commitments with multiline support
         let currentCommitment = null;
-        // Process lines starting from the second line (skip agent name)
-        for (let i = 1; i < lines.length; i++) {
+        // Process lines starting from after the agent name line
+        const startIndex = agentNameLineIndex >= 0 ? agentNameLineIndex + 1 : 0;
+        for (let i = startIndex; i < lines.length; i++) {
             const line = lines[i];
             if (line === undefined) {
                 continue;
@@ -8316,6 +8363,7 @@
             const parameterRegex = /@([a-zA-Z0-9_á-žÁ-Žč-řČ-Řš-žŠ-Žа-яА-ЯёЁ]+)/;
             // eslint-disable-next-line @typescript-eslint/no-explicit-any
             const bookRules = [
+                [/^---[-]*$/, ''],
                 [parameterRegex, 'parameter'],
                 [/\{[^}]+\}/, 'parameter'],
                 [commitmentRegex, 'commitment'],
@@ -8324,7 +8372,12 @@
             const tokenProvider = monaco.languages.setMonarchTokensProvider(BOOK_LANGUAGE_ID, {
                 ignoreCase: true,
                 tokenizer: {
-                    root: [[/^.*$/, 'title', '@body']],
+                    root: [
+                        [/^\s*$/, 'empty'],
+                        [/^-*$/, 'line'],
+                        [/^.*$/, 'title', '@body'],
+                        [commitmentRegex, 'commitment'],
+                    ],
                     body: bookRules,
                 },
             });
@@ -9250,15 +9303,15 @@
             ASSISTANT: '#ffb300',
             SYSTEM: '#888',
         };
-        const bgColor = participantColors[String(message.from)] || '#2b7cff';
+        const bgColor = participantColors[String(message.sender)] || '#2b7cff';
         const textColor = getTextColor(bgColor);
         return spaceTrim__default["default"](`
                             <div class="chat-message">
                                 <div class="avatar" style="background:${bgColor};color:${getTextColor(bgColor)};">
-                                    ${String(message.from)[0] || '?'}
+                                    ${String(message.sender)[0] || '?'}
                                 </div>
                                 <div class="bubble" style="background:${bgColor};color:${textColor};">
-                                    <span class="from-label">${String(message.from)}:</span>
+                                    <span class="from-label">${String(message.sender)}:</span>
                                     ${message.content}
                                 </div>
                             </div>
@@ -9304,7 +9357,7 @@
         getContent: ({ messages }) => spaceTrim$1.spaceTrim(`
             ${messages
         .map((message) => spaceTrim$1.spaceTrim(`
-                        **${message.from}:**
+                        **${message.sender}:**
 
                         > ${message.content.replace(/\n/g, '\n> ')}
                     `))
@@ -9586,7 +9639,7 @@
                 console.info('participant avatarSrc', avatarSrc);
                 console.info('participant color', { color, colorOfText });
                 console.groupEnd();
-            }, children: [avatarSrc && (jsxRuntime.jsxs("div", { ref: avatarRef, className: chatStyles.avatar, onMouseEnter: handleMouseEnter, onMouseLeave: handleMouseLeave, onClick: showTooltip, children: [jsxRuntime.jsx("img", { width: AVATAR_SIZE, src: avatarSrc, alt: `Avatar of ${message.from.toString().toLocaleLowerCase()}`, style: {
+            }, children: [avatarSrc && (jsxRuntime.jsxs("div", { ref: avatarRef, className: chatStyles.avatar, onMouseEnter: handleMouseEnter, onMouseLeave: handleMouseLeave, onClick: showTooltip, children: [jsxRuntime.jsx("img", { width: AVATAR_SIZE, src: avatarSrc, alt: `Avatar of ${message.sender.toString().toLocaleLowerCase()}`, style: {
                                 '--avatar-bg-color': color.toHex(),
                                 width: AVATAR_SIZE,
                             } }), isAvatarTooltipVisible && (participant === null || participant === void 0 ? void 0 : participant.agentSource) && avatarTooltipPosition && (jsxRuntime.jsx(AvatarProfileTooltip, { ref: tooltipRef, agentSource: participant.agentSource, position: avatarTooltipPosition }))] })), jsxRuntime.jsxs("div", { className: chatStyles.messageText, style: {
@@ -9935,7 +9988,7 @@
             }
         }, [ratingModalOpen, isMobile]);
         // Determine alignment for actions (Reset button) based on the first message
-        const firstMessageFromUser = ((_a = messages[0]) === null || _a === void 0 ? void 0 : _a.from) === 'USER';
+        const firstMessageFromUser = ((_a = messages[0]) === null || _a === void 0 ? void 0 : _a.sender) === 'USER';
         const actionsAlignmentClass = firstMessageFromUser
             ? chatStyles.actions + ' ' + chatStyles.left
             : chatStyles.actions + ' ' + chatStyles.right;
@@ -10012,7 +10065,7 @@
                                     }
                                     return true;
                                 })() && chatStyles.hasActionsAndFirstMessageIsLong), ref: chatMessagesRef, onScroll: handleScroll, children: [postprocessedMessages.map((message, i) => {
-                                        const participant = participants.find((participant) => participant.name === message.from);
+                                        const participant = participants.find((participant) => participant.name === message.sender);
                                         const isLastMessage = i === postprocessedMessages.length - 1;
                                         const isExpanded = expandedMessageId === message.id;
                                         const currentRating = messageRatings.get(message.id || message.content /* <-[💃] */) || 0;
@@ -10104,13 +10157,13 @@
                                     const idx = postprocessedMessages.findIndex((m) => m.id === selectedMessage.id);
                                     if (idx > 0) {
                                         const prev = postprocessedMessages[idx - 1];
-                                        if (prev.from === 'USER') {
+                                        if (prev.sender === 'USER') {
                                             return prev.content;
                                         }
                                     }
                                     // fallback: find last USER message before selectedMessage
                                     for (let i = messages.findIndex((m) => m.id === selectedMessage.id) - 1; i >= 0; i--) {
-                                        if (messages[i].from === 'USER') {
+                                        if (messages[i].sender === 'USER') {
                                             return messages[i].content;
                                         }
                                     }
@@ -10137,7 +10190,9 @@
             try {
                 const serializableMessages = messages.map((message) => ({
                     ...message,
-                    date: (message.date || new Date()).toISOString(),
+                    createdAt: (typeof message.createdAt === 'string'
+                        ? new Date(message.createdAt)
+                        : message.createdAt || new Date()).toISOString(),
                 }));
                 const storageKey = this.STORAGE_PREFIX + persistenceKey;
                 localStorage.setItem(storageKey, JSON.stringify(serializableMessages));
@@ -10160,7 +10215,7 @@
                 // Convert date strings back to Date objects
                 return serializableMessages.map((message) => ({
                     ...message,
-                    date: new Date(message.date),
+                    createdAt: new Date(message.createdAt),
                 }));
             }
             catch (error) {
@@ -10213,7 +10268,7 @@
         const { llmTools, persistenceKey, onChange, onReset, initialMessages, sendMessage, userParticipantName = 'USER', llmParticipantName = 'ASSISTANT', autoExecuteMessage, buttonColor, ...restProps } = props;
         // Internal state management
         // DRY: Single factory for seeding initial messages (used on mount and after reset)
-        const buildInitialMessages = react.useCallback(() => (initialMessages ? [...initialMessages] : []), [initialMessages]);
+        const buildInitialMessages = react.useCallback(() => initialMessages ? ([...initialMessages]) : ([]), [initialMessages]);
         const [messages, setMessages] = react.useState(() => buildInitialMessages());
         const [tasksProgress, setTasksProgress] = react.useState([]);
         const [isVoiceCalling, setIsVoiceCalling] = react.useState(false);
@@ -10306,17 +10361,19 @@
                 setTasksProgress([{ id: taskId, name: 'Playing response...', progress: 100 }]);
                 const now = Date.now();
                 const userMessage = {
+                    // channel: 'PROMPTBOOK_CHAT',
                     id: `user_${now}`,
-                    date: new Date(),
-                    from: userParticipantName,
+                    createdAt: new Date(),
+                    sender: userParticipantName,
                     content: (result.userMessage || '(Voice message)'),
                     isComplete: true,
                     isVoiceCall: true,
                 };
                 const agentMessage = {
+                    // channel: 'PROMPTBOOK_CHAT',
                     id: `agent_${now}`,
-                    date: new Date(),
-                    from: llmParticipantName,
+                    createdAt: new Date(),
+                    sender: llmParticipantName,
                     content: (result.agentMessage || result.text),
                     isComplete: true,
                     isVoiceCall: true,
@@ -10413,9 +10470,10 @@
             hasUserInteractedRef.current = true;
             // Add user message
             const userMessage = {
+                // channel: 'PROMPTBOOK_CHAT',
                 id: `user_${Date.now()}`,
-                date: new Date(),
-                from: userParticipantName,
+                createdAt: new Date(),
+                sender: userParticipantName,
                 content: messageContent,
                 isComplete: true,
             };
@@ -10427,9 +10485,10 @@
             }
             // Add loading message for assistant
             const loadingMessage = {
+                // channel: 'PROMPTBOOK_CHAT',
                 id: `assistant_${Date.now()}`,
-                date: new Date(),
-                from: llmParticipantName,
+                createdAt: new Date(),
+                sender: llmParticipantName,
                 content: 'Thinking...',
                 isComplete: false,
             };
@@ -10460,9 +10519,10 @@
                 if (llmTools.callChatModelStream) {
                     result = await llmTools.callChatModelStream(prompt, (chunk) => {
                         const assistantMessage = {
+                            // channel: 'PROMPTBOOK_CHAT',
                             id: loadingMessage.id,
-                            date: new Date(),
-                            from: llmParticipantName,
+                            createdAt: new Date(),
+                            sender: llmParticipantName,
                             content: chunk.content,
                             isComplete: false,
                         };
@@ -10483,9 +10543,10 @@
                 setTasksProgress([{ id: taskId, name: 'Response generated', progress: 100 }]);
                 // Replace loading message with actual response
                 const assistantMessage = {
+                    // channel: 'PROMPTBOOK_CHAT',
                     id: loadingMessage.id,
-                    date: new Date(),
-                    from: llmParticipantName,
+                    createdAt: new Date(),
+                    sender: llmParticipantName,
                     content: result.content,
                     isComplete: true,
                 };
@@ -10504,9 +10565,10 @@
                 console.error('Error calling LLM:', error);
                 // Replace loading message with error message
                 const errorMessage = {
+                    // channel: 'PROMPTBOOK_CHAT',
                     id: loadingMessage.id,
-                    date: new Date(),
-                    from: llmParticipantName,
+                    createdAt: new Date(),
+                    sender: llmParticipantName,
                     content: `Sorry, I encountered an error: ${error instanceof Error ? error.message : 'Unknown error'}`,
                     isComplete: true,
                 };
@@ -10573,7 +10635,8 @@
         return (jsxRuntime.jsx(jsxRuntime.Fragment, { children: jsxRuntime.jsx(LlmChat, { title: title || `Chat with ${agent.meta.fullname || agent.agentName || 'Agent'}`, persistenceKey: persistenceKey || `agent-chat-${agent.agentName}`, userParticipantName: "USER", llmParticipantName: "AGENT" // <- TODO: [🧠] Maybe dynamic agent id
                 , initialMessages: [
                     {
-                        from: 'AGENT',
+                        // channel: 'PROMPTBOOK_CHAT',
+                        sender: 'AGENT',
                         content: agent.initialMessage ||
                             spaceTrim__default["default"](`
                                 
@@ -10851,7 +10914,7 @@
                         if (delays.longPauseChance &&
                             Math.random() < delays.longPauseChance &&
                             i > 0 &&
-                            originalMessages[i].from !== originalMessages[i - 1].from) {
+                            originalMessages[i].sender !== originalMessages[i - 1].sender) {
                             await waitasecond.forTime(getDelay(delays.longPauseDuration, 2000));
                             didLongPause = true;
                             if (isCancelled)
@@ -10872,9 +10935,10 @@
                     }
                     // Show incomplete message first (for typing effect)
                     const incompleteMessage = {
+                        // channel: 'PROMPTBOOK_CHAT',
                         id: currentMessage.id,
-                        date: currentMessage.date,
-                        from: currentMessage.from,
+                        createdAt: currentMessage.createdAt,
+                        sender: currentMessage.sender,
                         content: '',
                         isComplete: false,
                         expectedAnswer: currentMessage.expectedAnswer,
@@ -10892,9 +10956,10 @@
                         currentContent += (wordIndex > 0 ? ' ' : '') + word;
                         // Update the message with current content
                         const updatingMessage = {
+                            // channel: 'PROMPTBOOK_CHAT',
                             id: currentMessage.id,
-                            date: currentMessage.date,
-                            from: currentMessage.from,
+                            createdAt: currentMessage.createdAt,
+                            sender: currentMessage.sender,
                             content: currentContent,
                             isComplete: false,
                             expectedAnswer: currentMessage.expectedAnswer,
@@ -10912,9 +10977,10 @@
                     }
                     // Mark message as complete
                     const completeMessage = {
+                        // channel: 'PROMPTBOOK_CHAT',
                         id: currentMessage.id,
-                        date: currentMessage.date,
-                        from: currentMessage.from,
+                        createdAt: currentMessage.createdAt,
+                        sender: currentMessage.sender,
                         content: currentMessage.content,
                         isComplete: true,
                         expectedAnswer: currentMessage.expectedAnswer,
@@ -11166,6 +11232,12 @@
         callEmbeddingModel(prompt) {
             return this.callCommonModel(prompt);
         }
+        /**
+         * Calls the best available embedding model
+         */
+        callImageGenerationModel(prompt) {
+            return this.callCommonModel(prompt);
+        }
         // <- Note: [🤖]
         /**
          * Calls the best available model
@@ -11192,6 +11264,11 @@
                                 continue llm;
                             }
                             return await llmExecutionTools.callEmbeddingModel(prompt);
+                        case 'IMAGE_GENERATION':
+                            if (llmExecutionTools.callImageGenerationModel === undefined) {
+                                continue llm;
+                            }
+                            return await llmExecutionTools.callImageGenerationModel(prompt);
                         // <- case [🤖]:
                         default:
                             throw new UnexpectedError(`Unknown model variant "${prompt.modelRequirements.modelVariant}" in ${llmExecutionTools.title}`);
@@ -12366,6 +12443,15 @@
                 return promptResult;
             };
         }
+        if (llmTools.callImageGenerationModel !== undefined) {
+            proxyTools.callImageGenerationModel = async (prompt) => {
+                // console.info('[🚕] callImageGenerationModel through countTotalUsage');
+                const promptResult = await llmTools.callImageGenerationModel(prompt);
+                totalUsage = addUsage(totalUsage, promptResult.usage);
+                spending.next(promptResult.usage);
+                return promptResult;
+            };
+        }
         // <- Note: [🤖]
         return proxyTools;
     }
@@ -13909,8 +13995,9 @@
                                         $ongoingTaskResult.$resultString = $ongoingTaskResult.$completionResult.content;
                                         break variant;
                                     case 'EMBEDDING':
+                                    case 'IMAGE_GENERATION':
                                         throw new PipelineExecutionError(spaceTrim$1.spaceTrim((block) => `
-                                                Embedding model can not be used in pipeline
+                                                ${modelRequirements.modelVariant} model can not be used in pipeline
 
                                                 This should be catched during parsing
 
@@ -16241,7 +16328,7 @@
             let threadMessages = [];
             if ('thread' in prompt && Array.isArray(prompt.thread)) {
                 threadMessages = prompt.thread.map((msg) => ({
-                    role: msg.role === 'assistant' ? 'assistant' : 'user',
+                    role: msg.sender === 'assistant' ? 'assistant' : 'user',
                     content: msg.content,
                 }));
             }