@jupyter-ai/persona-manager 0.2.0-a3 → 0.2.0-a4

package/lib/index.js

@@ -76,9 +76,17 @@ const toolbarPlugin = {
     provides: IInputToolbarRegistryFactory,
     requires: [IPersonaControlRegistry, IPersonaSessionRegistry],
     activate: (app, controlRegistry, sessionRegistry) => {
-        // Wrap the persona controls to inject the control + session registries,
-        // which the generic toolbar-item props don't carry.
-        const PersonaControlsItem = (itemProps) => PersonaControls({ ...itemProps, controlRegistry, sessionRegistry });
+        // The event bus front-end; `PersonaControls` uses it to emit
+        // `persona_selected` on selection.
+        const events = app.serviceManager.events;
+        // Wrap the persona controls to inject the control + session registries and
+        // the event manager, which the generic toolbar-item props don't carry.
+        const PersonaControlsItem = (itemProps) => PersonaControls({
+            ...itemProps,
+            controlRegistry,
+            sessionRegistry,
+            events
+        });
         return {
             create: () => {
                 // Start with the default toolbar (Send, Attach, Cancel, SaveEdit)

package/lib/persona-controls.d.ts

@@ -1,3 +1,4 @@
+import { Event } from '@jupyterlab/services';
 import { InputToolbarRegistry } from '@jupyter/chat';
 import { PersonaOption, Usage } from './awareness';
 import { PersonaSessionRegistry, PersonaSessionState } from './persona-events';
@@ -177,5 +178,10 @@ export declare function PersonaControls(props: InputToolbarRegistry.IToolbarItem
      * Optional so the component still works without it (renders nothing).
      */
     sessionRegistry?: PersonaSessionRegistry;
+    /**
+     * The event bus front-end, used to emit a `persona_selected` event when the
+     * user picks a persona so the server prepares it eagerly.
+     */
+    events?: Event.IManager;
 }): JSX.Element | null;
 export {};

package/lib/persona-controls.js

@@ -6,6 +6,7 @@ import MoreHorizIcon from '@mui/icons-material/MoreHoriz';
 import { PageConfig } from '@jupyterlab/coreutils';
 import { EMPTY_USAGE } from './awareness';
 import { buildMessageMetadata, emptyPersonaSettings } from './metadata';
+import { PERSONA_SELECTED_EVENT_SCHEMA_ID } from './persona-events';
 const SELECTOR_CLASS = 'jp-jai-personaControls';
 const MENU_CLASS = 'jp-jai-controlMenu';
 const USAGE_CLASS = 'jp-jai-usage';
@@ -635,7 +636,7 @@ export function UsageChip(props) {
  */
 export function PersonaControls(props) {
     var _a, _b, _c, _d, _e, _f;
-    const { chatModel, model, controlRegistry, sessionRegistry } = props;
+    const { chatModel, model, controlRegistry, sessionRegistry, events } = props;
     // The chat's stable id scopes persona events to this chat. It is assigned
     // asynchronously (once the model is `ready`: the WS connection frame arrives,
     // or the RTC document syncs), so track it in state and update it when the
@@ -748,6 +749,21 @@ export function PersonaControls(props) {
         model.clearMetadata();
         model.updateMetadata(buildMessageMetadata(selectedId, settings));
     }, [model, metadataSignature]);
+    // On selection, tell the server to prepare the persona by
+    // emitting `persona_selected` over the event bus, so its controls appear
+    // before the first message. Skips "No one"; idempotent server-side.
+    useEffect(() => {
+        if (!events || !chatId || !selectedId) {
+            return;
+        }
+        void events
+            .emit({
+            schema_id: PERSONA_SELECTED_EVENT_SCHEMA_ID,
+            version: '1',
+            data: { chat_id: chatId, persona_id: selectedId }
+        })
+            .catch(e => console.warn('Error emitting persona_selected event: ', e));
+    }, [events, chatId, selectedId]);
     // No personas yet. While the manager's slot or its first list read is still
     // pending, show a loading placeholder (on slow networks this takes seconds);
     // once ready with no personas, show nothing.

package/lib/persona-events.d.ts

@@ -5,6 +5,7 @@ import { ISignal } from '@lumino/signaling';
 import { CommandOption, ModelConfiguration, PersonaOption, SettingConfiguration, Usage } from './awareness';
 export declare const PERSONAS_EVENT_SCHEMA_ID = "https://schema.jupyter.org/jupyter_ai_persona_manager/personas/v1";
 export declare const PERSONA_STATE_EVENT_SCHEMA_ID = "https://schema.jupyter.org/jupyter_ai_persona_manager/persona_state/v1";
+export declare const PERSONA_SELECTED_EVENT_SCHEMA_ID = "https://schema.jupyter.org/jupyter_ai_persona_manager/persona_selected/v1";
 /** The wire shape of a `persona_state` event. */
 type PersonaStatePayload = {
     chat_id?: string;

package/lib/persona-events.js

@@ -3,6 +3,7 @@ import { Signal } from '@lumino/signaling';
 import { EMPTY_MODEL_CONFIGURATION, EMPTY_USAGE } from './awareness';
 export const PERSONAS_EVENT_SCHEMA_ID = 'https://schema.jupyter.org/jupyter_ai_persona_manager/personas/v1';
 export const PERSONA_STATE_EVENT_SCHEMA_ID = 'https://schema.jupyter.org/jupyter_ai_persona_manager/persona_state/v1';
+export const PERSONA_SELECTED_EVENT_SCHEMA_ID = 'https://schema.jupyter.org/jupyter_ai_persona_manager/persona_selected/v1';
 /**
  * One persona's live session state, built from a `persona_state` event.
  * Immutable: each update produces a new instance, so React consumers see a new

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@jupyter-ai/persona-manager",
-    "version": "0.2.0-a3",
+    "version": "0.2.0-a4",
     "description": "The core manager & registry for AI personas in Jupyter AI",
     "keywords": [
         "jupyter",

package/src/index.ts

@@ -119,11 +119,20 @@ const toolbarPlugin: JupyterFrontEndPlugin<IInputToolbarRegistryFactory> = {
     controlRegistry: IPersonaControlRegistry,
     sessionRegistry: PersonaSessionRegistry
   ): IInputToolbarRegistryFactory => {
-    // Wrap the persona controls to inject the control + session registries,
-    // which the generic toolbar-item props don't carry.
+    // The event bus front-end; `PersonaControls` uses it to emit
+    // `persona_selected` on selection.
+    const events = app.serviceManager.events;
+    // Wrap the persona controls to inject the control + session registries and
+    // the event manager, which the generic toolbar-item props don't carry.
     const PersonaControlsItem = (
       itemProps: InputToolbarRegistry.IToolbarItemProps
-    ) => PersonaControls({ ...itemProps, controlRegistry, sessionRegistry });
+    ) =>
+      PersonaControls({
+        ...itemProps,
+        controlRegistry,
+        sessionRegistry,
+        events
+      });
     return {
       create: () => {
         // Start with the default toolbar (Send, Attach, Cancel, SaveEdit)

package/src/persona-controls.tsx

@@ -21,6 +21,7 @@ import ArrowDropDownIcon from '@mui/icons-material/ArrowDropDown';
 import CheckIcon from '@mui/icons-material/Check';
 import MoreHorizIcon from '@mui/icons-material/MoreHoriz';
 import { PageConfig } from '@jupyterlab/coreutils';
+import { Event } from '@jupyterlab/services';
 import { InputToolbarRegistry } from '@jupyter/chat';
 import {
   EMPTY_USAGE,
@@ -35,6 +36,7 @@ import {
   emptyPersonaSettings
 } from './metadata';
 import { IPersonaControlRegistry } from './persona-control-registry';
+import { PERSONA_SELECTED_EVENT_SCHEMA_ID } from './persona-events';
 
 const SELECTOR_CLASS = 'jp-jai-personaControls';
 const MENU_CLASS = 'jp-jai-controlMenu';
@@ -1072,9 +1074,14 @@ export function PersonaControls(
      * Optional so the component still works without it (renders nothing).
      */
     sessionRegistry?: PersonaSessionRegistry;
+    /**
+     * The event bus front-end, used to emit a `persona_selected` event when the
+     * user picks a persona so the server prepares it eagerly.
+     */
+    events?: Event.IManager;
   }
 ): JSX.Element | null {
-  const { chatModel, model, controlRegistry, sessionRegistry } = props;
+  const { chatModel, model, controlRegistry, sessionRegistry, events } = props;
   // The chat's stable id scopes persona events to this chat. It is assigned
   // asynchronously (once the model is `ready`: the WS connection frame arrives,
   // or the RTC document syncs), so track it in state and update it when the
@@ -1207,6 +1214,22 @@ export function PersonaControls(
     model.updateMetadata(buildMessageMetadata(selectedId, settings));
   }, [model, metadataSignature]);
 
+  // On selection, tell the server to prepare the persona by
+  // emitting `persona_selected` over the event bus, so its controls appear
+  // before the first message. Skips "No one"; idempotent server-side.
+  useEffect(() => {
+    if (!events || !chatId || !selectedId) {
+      return;
+    }
+    void events
+      .emit({
+        schema_id: PERSONA_SELECTED_EVENT_SCHEMA_ID,
+        version: '1',
+        data: { chat_id: chatId, persona_id: selectedId }
+      })
+      .catch(e => console.warn('Error emitting persona_selected event: ', e));
+  }, [events, chatId, selectedId]);
+
   // No personas yet. While the manager's slot or its first list read is still
   // pending, show a loading placeholder (on slow networks this takes seconds);
   // once ready with no personas, show nothing.

package/src/persona-events.ts

@@ -31,6 +31,8 @@ export const PERSONAS_EVENT_SCHEMA_ID =
   'https://schema.jupyter.org/jupyter_ai_persona_manager/personas/v1';
 export const PERSONA_STATE_EVENT_SCHEMA_ID =
   'https://schema.jupyter.org/jupyter_ai_persona_manager/persona_state/v1';
+export const PERSONA_SELECTED_EVENT_SCHEMA_ID =
+  'https://schema.jupyter.org/jupyter_ai_persona_manager/persona_selected/v1';
 
 /** The wire shape of a `persona_state` event. */
 type PersonaStatePayload = {