@gravity-ui/aikit 1.3.5 → 1.4.0

package/dist/components/organisms/MessageList/MessageList.js

@@ -9,8 +9,8 @@ import { UserMessage } from '../UserMessage';
 import { ErrorAlert } from './ErrorAlert';
 import './MessageList.scss';
 const b = block('message-list');
-export function MessageList({ messages, messageRendererRegistry, transformOptions, shouldParseIncompleteMarkdown, showActionsOnHover, showTimestamp, showAvatar, userActions, assistantActions, loaderStatuses = ['submitted'], className, qa, status, errorMessage, onRetry, hasPreviousMessages = false, onLoadPreviousMessages, }) {
-    const isStreaming = status === 'streaming';
+export function MessageList({ messages, messageRendererRegistry, transformOptions, shouldParseIncompleteMarkdown, showActionsOnHover, showTimestamp, showAvatar, userActions, assistantActions, loaderStatuses = ['submitted', 'streaming_loading'], className, qa, status, errorMessage, onRetry, hasPreviousMessages = false, onLoadPreviousMessages, }) {
+    const isStreaming = status === 'streaming' || status === 'streaming_loading';
     const isSubmitted = status === 'submitted';
     const showLoader = status && loaderStatuses.includes(status);
     const { containerRef } = useSmartScroll({

package/dist/components/organisms/PromptInput/usePromptInput.js

@@ -16,6 +16,7 @@ export function usePromptInput(props) {
     // ChatStatus.ready → submitButtonState.enabled
     // ChatStatus.error → submitButtonState.enabled
     // ChatStatus.streaming → submitButtonState.cancelable
+    // ChatStatus.streaming_loading → submitButtonState.cancelable (same as streaming)
     // ChatStatus.submitted → submitButtonState.loading
     let submitButtonState = 'disabled';
     // disabled by props or empty value and status is ready
@@ -32,6 +33,7 @@ export function usePromptInput(props) {
                 submitButtonState = 'enabled';
                 break;
             case 'streaming':
+            case 'streaming_loading':
                 submitButtonState = onCancel ? 'cancelable' : 'enabled';
                 break;
             case 'submitted':

package/dist/components/pages/ChatContainer/__stories__/ChatContainer.stories.js

@@ -359,9 +359,16 @@ export const WithStreaming = {
                     actions: createMessageActions(assistantMessageId, 'assistant'),
                 },
             ]);
-            // Simulate word-by-word streaming
+            // Simulate word-by-word streaming with a streaming_loading pause in the middle
             const words = fullResponse.split(' ');
+            const pauseIndex = Math.floor(words.length / 2);
             for (let i = 0; i < words.length; i++) {
+                // Pause streaming at the midpoint: switch to streaming_loading for 5 seconds
+                if (i === pauseIndex) {
+                    setStatus('streaming_loading');
+                    await new Promise((resolve) => setTimeout(resolve, 5000));
+                    setStatus('streaming');
+                }
                 await new Promise((resolve) => setTimeout(resolve, 100));
                 const currentText = words.slice(0, i + 1).join(' ');
                 setMessages((prev) => prev.map((msg) => msg.id === assistantMessageId
@@ -747,7 +754,10 @@ export const FullStreamingExample = {
         const [controller, setController] = useState(null);
         const isProcessingRef = React.useRef(false);
         const handleSendMessage = async (data) => {
-            if (isProcessingRef.current || status === 'streaming' || status === 'submitted') {
+            if (isProcessingRef.current ||
+                status === 'streaming' ||
+                status === 'streaming_loading' ||
+                status === 'submitted') {
                 return;
             }
             isProcessingRef.current = true;

package/dist/types/chat.d.ts

@@ -5,7 +5,7 @@ export type ChatType = {
     lastMessage?: string;
     metadata?: Record<string, unknown>;
 };
-export type ChatStatus = 'submitted' | 'streaming' | 'ready' | 'error';
+export type ChatStatus = 'submitted' | 'streaming' | 'streaming_loading' | 'ready' | 'error';
 /**
  * List item type for chat history that can be either a chat or a date header
  */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gravity-ui/aikit",
-  "version": "1.3.5",
+  "version": "1.4.0",
   "description": "Gravity UI base kit for building ai assistant chats",
   "license": "MIT",
   "main": "./dist/index.js",