@jupyter-ai/jupyternaut 0.0.9 → 0.0.11

package/lib/index.js

@@ -1,9 +1,7 @@
 import { INotebookShell } from '@jupyter-notebook/application';
 import { IThemeManager, MainAreaWidget, ICommandPalette } from '@jupyterlab/apputils';
-import { IMessageFooterRegistry } from '@jupyter/chat';
 import { IRenderMimeRegistry } from '@jupyterlab/rendermime';
 import { SingletonLayout, Widget } from '@lumino/widgets';
-import { StopButton } from './components/message-footer/stop-button';
 //import { completionPlugin } from './completions';
 import { buildErrorWidget } from './widgets/chat-error';
 import { buildAiSettings } from './widgets/settings-widget';
@@ -95,22 +93,10 @@ const jupyternautSettingsPlugin = {
         }
     }
 };
-const stopButtonPlugin = {
-    id: '@jupyter-ai/jupyternaut:stop-button',
-    autoStart: true,
-    requires: [IMessageFooterRegistry],
-    activate: (app, registry) => {
-        registry.addSection({
-            component: StopButton,
-            position: 'center'
-        });
-    }
-};
 export default [
     plugin,
     jupyternautSettingsPlugin,
     // webComponentsPlugin,
-    stopButtonPlugin,
     // completionPlugin,
     statusItemPlugin
 ];

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@jupyter-ai/jupyternaut",
-    "version": "0.0.9",
+    "version": "0.0.11",
     "description": "Package providing the default AI persona, Jupyternaut, in Jupyter AI.",
     "keywords": [
         "jupyter",

package/lib/components/message-footer/stop-button.d.ts

@@ -1,5 +0,0 @@
-import { MessageFooterSectionProps } from '@jupyter/chat';
-/**
- * The stop button.
- */
-export declare function StopButton(props: MessageFooterSectionProps): JSX.Element;

package/lib/components/message-footer/stop-button.js

@@ -1,53 +0,0 @@
-import { TooltippedButton } from '@jupyter/chat';
-import StopIcon from '@mui/icons-material/Stop';
-import React, { useEffect, useState } from 'react';
-import { requestAPI } from '../../handler';
-/**
- * The stop button.
- */
-export function StopButton(props) {
-    const { message, model } = props;
-    const [visible, setVisible] = useState(false);
-    const tooltip = 'Stop streaming';
-    useEffect(() => {
-        var _a, _b;
-        const writerChanged = (_, writers) => {
-            const w = writers.filter(w => w.messageID === message.id);
-            if (w.length > 0) {
-                setVisible(true);
-            }
-            else {
-                setVisible(false);
-            }
-        };
-        // Listen only the messages that are from a bot.
-        if (message.sender.username !== ((_a = model.user) === null || _a === void 0 ? void 0 : _a.username) &&
-            message.sender.bot) {
-            (_b = model.writersChanged) === null || _b === void 0 ? void 0 : _b.connect(writerChanged);
-            // Check if the message is currently being edited.
-            writerChanged(model, model.writers);
-        }
-        return () => {
-            var _a;
-            (_a = model.writersChanged) === null || _a === void 0 ? void 0 : _a.disconnect(writerChanged);
-        };
-    }, [model]);
-    const onClick = () => {
-        // Post request to the stop streaming handler.
-        requestAPI('api/jupyternaut/chats/stop_streaming', {
-            method: 'POST',
-            body: JSON.stringify({
-                message_id: message.id
-            }),
-            headers: {
-                'Content-Type': 'application/json'
-            }
-        });
-    };
-    return visible ? (React.createElement(TooltippedButton, { onClick: onClick, tooltip: tooltip, buttonProps: {
-            size: 'small',
-            variant: 'contained',
-            title: tooltip
-        }, sx: { display: visible ? 'inline-flex' : 'none' } },
-        React.createElement(StopIcon, null))) : (React.createElement(React.Fragment, null));
-}