jupyterlab_ai_code_assistants_extension 0.6.15

package/src/providers/claude.ts

@@ -0,0 +1,102 @@
+// The Claude Code provider.
+//
+// Ported from `jupyterlab_claude_code_extension` v1.2.73, the architectural
+// base. Almost all of that extension's behaviour is now the shared core, and
+// what is left here is what only Claude has: the capability flags on the
+// descriptor, and the two places where "a live background agent holds this
+// conversation" changes what the user is offered.
+//
+// Nothing here touches the DOM outside its own badge element, and nothing here
+// reaches into core internals - the hooks are the whole seam.
+
+import {
+  IBranch,
+  IProviderDescriptor,
+  IProviderHooks,
+  ISession
+} from '../core/types';
+
+// Claude's own mark, drawn as a single path group so the theme's `jp-icon3`
+// fill drives it in both light and dark.
+const claudeSvgStr = `<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16" width="16" height="16">
+  <g class="jp-icon3" fill="#616161">
+    <path d="m14.375 6.48.49.28v.209l-.14.489-5.937 1.397-.558-1.387zm0 0"/>
+    <path d="m12.155 2.373.683.143.182.224.173.535-.072.342-3.983 5.447L7.81 7.737l3.673-4.82z"/>
+    <path d="m8.719 1.522.419-.28.349.14.349.49-.957 5.748-.65-.441-.279-.769.49-4.33z"/>
+    <path d="m4.239 1.614.43-.55L4.95 1l.558.081.275.216 2.004 4.442.724 2.11-.848.471-3.231-5.864z"/>
+    <path d="m2.154 4.665-.14-.56.42-.488.488.07h.14l2.933 2.165.908.698 1.257.978-.698 1.187-.629-.489-.419-.419-4.05-2.863z"/>
+    <path d="M1.316 8.296 1 7.946v-.31l.316-.108 3.562.21 3.491.279-.113.695-6.66-.346z"/>
+    <path d="M3.411 11.931h-.698l-.278-.32v-.382l1.186-.838 4.82-3.068.487.833z"/>
+    <path d="m4.738 13.883-.28.07-.418-.21.07-.35 4.12-5.446.558.768-3.072 4.05z"/>
+    <path d="m8.23 14.581-.21.28-.419.14-.349-.28-.21-.42L8.09 8.646l.629.07z"/>
+    <path d="M11.791 13.045v.558l-.07.21-.279.14-.489-.066-3.356-4.996 1.331-1.014 1.117 2.025.105.733z"/>
+    <path d="m13.398 12.207.07.349-.21.279-.21-.07-1.187-.838-1.815-1.606-1.397-.978.419-1.326.698.419.42.768z"/>
+    <path d="m12.49 8.645 1.746.14.419.28.279.418v.302l-.768.327-3.911-.978-1.606-.07.419-1.466 1.117.838z"/>
+  </g>
+</svg>`;
+
+export const descriptor: IProviderDescriptor = {
+  id: 'claude',
+  label: 'Claude Code',
+  panelTitle: 'Claude Code Sessions',
+  iconName: 'jupyterlab_ai_code_assistants_extension:claude',
+  iconSvg: claudeSvgStr,
+  cliBinary: 'claude',
+  // `claude --fork-session --session-id <uuid>`: the CLI forks in-process and
+  // takes the new id from us, so the fork is known before it exists - and its
+  // store entry is written lazily, on the first turn, which is what the core's
+  // branch watcher waits for.
+  forkStrategy: 'native-flag',
+  // `/color` is Claude's own surface for a conversation's colour, recorded in
+  // its transcript, so the assistant owns the tint and the extension's
+  // write-back store stays out of it.
+  colourSource: 'native',
+  // `-n <name>` at launch: Claude stamps the name itself and re-stamps it on
+  // every turn, which is the only way a fork's name survives - a title written
+  // after the fact loses to the parent title the fork inherits.
+  namingStrategy: 'launch-flag',
+  mintsNewSessionId: true,
+  launchModes: [
+    {
+      id: 'dangerouslySkipPermissions',
+      title: 'Dangerously skip permissions',
+      description:
+        'When enabled, newly spawned and resumed sessions are launched with `--dangerously-skip-permissions`. Bypasses every Claude Code permission prompt - only enable in sandboxes you trust.',
+      kind: 'boolean',
+      default: false,
+      menuLabel: 'Skip Permissions'
+    }
+  ],
+  hasRemoteControl: true,
+  hasBgAgents: true,
+  // Claude records the conversation a process is on, so a terminal is
+  // identified exactly; there is no per-project "something is running here"
+  // signal to render.
+  hasLiveProcess: false,
+  // Retired standalone extension this provider replaces.
+  legacyPluginId: 'jupyterlab_claude_code_extension'
+};
+
+/** The `bg` chip: this conversation belongs to a live background agent, so
+ * opening it joins the running agent instead of resuming a dormant transcript.
+ * Display only - the server picks the verb at launch, where it cannot be
+ * stale. */
+function bgChip(): HTMLElement {
+  const chip = document.createElement('span');
+  chip.className = 'jp-AiAssistantsPanel-bgBadge';
+  chip.textContent = 'bg';
+  return chip;
+}
+
+export const hooks: IProviderHooks = {
+  // Joining a live agent and resuming a dormant conversation are different
+  // verbs, so the item says which one this row gets before the click.
+  resumeLabel: (session: ISession | null) =>
+    session?.bg_id ? 'Attach to Background Agent' : 'Resume',
+
+  // In the manage-sessions popup the chip replaces the fork badge for an
+  // agent-held conversation: both are identity hints, and the one that changes
+  // what a click does is the one worth the column. Returning null elsewhere
+  // leaves the shared fork badge in place.
+  branchBadge: (branch: IBranch) => (branch.bg_id ? bgChip() : null)
+};

package/src/providers/codex.ts

@@ -0,0 +1,89 @@
+// The Codex provider: OpenAI's `codex` CLI.
+//
+// Ported from `jupyterlab_codex_extension` v0.6.12. Almost everything that
+// extension did by hand is now shared panel behaviour, so what is left here is
+// the descriptor plus the two tooltip lines Codex has and the others do not.
+//
+// The three divergences that matter are capability flags, not code:
+//
+// * `forkStrategy: 'native-command'` - `codex fork <parent>` mints the fork's
+//   id itself, so the panel snapshots the branch list, launches, and diffs to
+//   discover the id (the shared 2s x 90 discovery watcher)
+// * `colourSource: 'none'` - Codex has no colour concept at all, so a tab tint
+//   only ever comes from the extension's own write-back store
+// * `hasLiveProcess: true` - a project row shows a dot when a `codex` process
+//   is running in it, which is the only liveness signal Codex exposes
+
+import {
+  IHookContext,
+  IProviderDescriptor,
+  IProviderHooks
+} from '../core/types';
+
+// The OpenAI "blossom" mark - Codex is OpenAI's CLI, so the panel wears its
+// vendor's logo. Single path, `jp-icon3` so it picks up the theme's foreground
+// in both light and dark. The box stays 16px like every other icon; the mark is
+// scaled down INSIDE it by padding the viewBox (24 units of art in a 30-unit
+// window, so ~80%), which keeps the sidebar tab and header alignment identical
+// while the dense logo reads less heavy than the simpler glyphs around it.
+const codexSvgStr = `<svg xmlns="http://www.w3.org/2000/svg" viewBox="-3 -3 30 30" width="16" height="16">
+  <path class="jp-icon3" fill="#616161" d="M22.282 9.821a5.985 5.985 0 0 0-.516-4.91 6.046 6.046 0 0 0-6.51-2.9A6.065 6.065 0 0 0 4.981 4.18a5.985 5.985 0 0 0-3.998 2.9 6.046 6.046 0 0 0 .743 7.097 5.98 5.98 0 0 0 .51 4.911 6.051 6.051 0 0 0 6.515 2.9A5.985 5.985 0 0 0 13.26 24a6.056 6.056 0 0 0 5.772-4.206 5.99 5.99 0 0 0 3.997-2.9 6.056 6.056 0 0 0-.747-7.073ZM13.26 22.43a4.476 4.476 0 0 1-2.876-1.04l.141-.081 4.779-2.758a.795.795 0 0 0 .392-.681v-6.737l2.02 1.168a.071.071 0 0 1 .038.052v5.583a4.504 4.504 0 0 1-4.494 4.494ZM3.6 18.304a4.47 4.47 0 0 1-.535-3.014l.142.085 4.783 2.759a.771.771 0 0 0 .78 0l5.843-3.369v2.332a.08.08 0 0 1-.033.062L9.74 19.95a4.5 4.5 0 0 1-6.14-1.646ZM2.34 7.896a4.485 4.485 0 0 1 2.366-1.973V11.6a.766.766 0 0 0 .388.676l5.815 3.355-2.02 1.168a.076.076 0 0 1-.071 0l-4.83-2.786A4.504 4.504 0 0 1 2.34 7.872Zm16.597 3.858-5.833-3.387L15.119 7.2a.076.076 0 0 1 .071 0l4.83 2.791a4.494 4.494 0 0 1-.676 8.105v-5.678a.79.79 0 0 0-.407-.667Zm2.01-3.024-.141-.085-4.774-2.782a.776.776 0 0 0-.785 0L9.409 9.23V6.897a.066.066 0 0 1 .028-.061l4.83-2.787a4.5 4.5 0 0 1 6.68 4.66Zm-12.64 4.135-2.02-1.164a.08.08 0 0 1-.038-.057V6.075a4.5 4.5 0 0 1 7.375-3.453l-.142.08L8.704 5.46a.795.795 0 0 0-.393.681ZM9.409 12l2.602-1.504L14.618 12v3.008l-2.597 1.5-2.607-1.5Z"/>
+</svg>`;
+
+// Codex's own name for its unsafe switch. Kept in the assistant's terminology
+// rather than normalised, because it is both the settings key under
+// `providers.codex.` and the mode token the server maps to the CLI flag
+// `--dangerously-bypass-approvals-and-sandbox`.
+const BYPASS_MODE = 'dangerouslyBypassApprovalsAndSandbox';
+
+export const descriptor: IProviderDescriptor = {
+  id: 'codex',
+  label: 'Codex',
+  panelTitle: 'Codex Sessions',
+  iconName: 'jupyterlab_ai_code_assistants_extension:codex',
+  iconSvg: codexSvgStr,
+  cliBinary: 'codex',
+  // `codex fork <parent>` owns both the verb and the new id.
+  forkStrategy: 'native-command',
+  // No `/color` equivalent, and nothing in the thread store records one.
+  colourSource: 'none',
+  // Forks are unnamed - Codex has no naming flag and no writable title field,
+  // so the panel never asks for a name it could not stamp anywhere.
+  namingStrategy: 'none',
+  // A bare `codex` mints its own thread id; there is no `--session-id`.
+  mintsNewSessionId: false,
+  launchModes: [
+    {
+      id: BYPASS_MODE,
+      title: 'Bypass Codex approvals and sandbox',
+      description:
+        'When enabled, newly spawned and resumed sessions are launched with `--dangerously-bypass-approvals-and-sandbox`. Bypasses every Codex approval prompt and sandbox - only enable in environments you trust.',
+      kind: 'boolean',
+      default: false,
+      menuLabel: 'Bypass Approvals'
+    }
+  ],
+  hasRemoteControl: false,
+  hasBgAgents: false,
+  // A `codex` process running in the project folder is the only liveness
+  // signal Codex exposes - it keeps no per-pid session record.
+  hasLiveProcess: true,
+  // Retired standalone extension this provider replaces.
+  legacyPluginId: 'jupyterlab_codex_extension'
+};
+
+export const hooks: IProviderHooks = {
+  // Codex's thread store records the model and the token spend of every
+  // conversation; no other assistant here does, so they are a provider line
+  // rather than a core one.
+  tooltipLines: ({ session }: IHookContext): string[] => {
+    const lines: string[] = [];
+    if (session.model) {
+      lines.push(`Model: ${session.model}`);
+    }
+    if ((session.tokens_used ?? 0) > 0) {
+      lines.push(`Tokens: ${session.tokens_used}`);
+    }
+    return lines;
+  }
+};

package/src/providers/gemini.ts

@@ -0,0 +1,80 @@
+// The Gemini provider - Google's `gemini` CLI.
+//
+// New here: no standalone extension preceded it, so there is nothing to port
+// and nothing to migrate. Everything that makes this assistant different from
+// the base one is descriptor data, so no core file knows it exists:
+//
+// * `forkStrategy: 'server-copy'` - gemini has no fork verb at all, so the
+//   server copies the chat file under a fresh id and answers with it
+// * `colourSource: 'none'` - no colour command, and nothing in the chat store
+//   records one, so a tab tint only ever comes from the extension's own
+//   write-back store
+// * `namingStrategy: 'server-side'` - a fork's name is stamped into the copied
+//   chat file's `summary`, which is the field the CLI itself displays a
+//   conversation under; there is no naming flag to pass it to
+// * `mintsNewSessionId: true` - `--session-id <uuid>` starts a fresh
+//   conversation under an id we chose, so a new session is identifiable from
+//   its argv on the first poll instead of being an unknown terminal
+// * two launch modes, gemini's `--yolo` switch and its four-step approval
+//   ladder
+//
+// No hooks: gemini rows carry no badge, no extra menu item and no derived
+// colour, so the shared panel renders them unchanged.
+
+import { IProviderDescriptor } from '../core/types';
+
+// The four-pointed spark of the Gemini mark: four tips on the box edges, each
+// pair joined by a quadratic whose control point is the centre, which is what
+// pinches the sides inwards. Single path, `jp-icon3` so the theme drives the
+// fill in both light and dark; the 16px box matches every other icon in the
+// sidebar.
+const geminiSvgStr = `<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16" width="16" height="16">
+  <path class="jp-icon3" fill="#616161" d="M8 0Q8 8 16 8Q8 8 8 16Q8 8 0 8Q8 8 8 0Z"/>
+</svg>`;
+
+// Gemini's own names for its two launch surfaces. Kept in the assistant's
+// terminology rather than normalised, because each is both the settings key
+// under `providers.gemini.` and the mode token the server maps to a CLI flag.
+const YOLO_MODE = 'yoloMode';
+const APPROVAL_MODE = 'approvalMode';
+
+export const descriptor: IProviderDescriptor = {
+  id: 'gemini',
+  label: 'Gemini',
+  panelTitle: 'Gemini Sessions',
+  iconName: 'jupyterlab_ai_code_assistants_extension:gemini',
+  iconSvg: geminiSvgStr,
+  cliBinary: 'gemini',
+  // No fork flag exists, so the server copies the chat file under a fresh id
+  // and the launch resumes that id like any other conversation.
+  forkStrategy: 'server-copy',
+  colourSource: 'none',
+  // The fork's name is written into the copy by the server; the CLI has no
+  // naming flag to take it.
+  namingStrategy: 'server-side',
+  // `--session-id <uuid>` starts a conversation under an id we mint.
+  mintsNewSessionId: true,
+  launchModes: [
+    {
+      id: YOLO_MODE,
+      title: 'YOLO mode',
+      description:
+        'When enabled, newly spawned and resumed sessions are launched with `--yolo`. Gemini auto-approves every tool call without asking - only enable in workspaces you trust.',
+      kind: 'boolean',
+      default: false,
+      menuLabel: 'YOLO'
+    },
+    {
+      id: APPROVAL_MODE,
+      title: 'Approval mode',
+      description:
+        'Approval ladder passed as `--approval-mode` when set to a value other than the default. A non-default value here takes precedence over the YOLO switch; only the forced YOLO menu action overrides it. `default` prompts for approval, `auto_edit` auto-approves edit tools, `yolo` auto-approves every tool, `plan` is read-only.',
+      kind: 'enum',
+      values: ['default', 'auto_edit', 'yolo', 'plan'],
+      default: 'default'
+    }
+  ],
+  hasRemoteControl: false,
+  hasBgAgents: false,
+  hasLiveProcess: false
+};

package/src/providers/index.ts

@@ -0,0 +1,21 @@
+// The single registration point for assistants.
+//
+// One line per provider. Adding an assistant means adding its module and its
+// line here; removing one means deleting both. No core file is edited either
+// way, and nothing outside this directory names an assistant.
+//
+// A provider module exports a `descriptor` (pure data - the settings-schema
+// generator reads it under plain Node) and optionally `hooks` (the DOM and
+// JupyterLab-aware parts). Keep module-scope side effects out of both, so the
+// generator can import this barrel without a browser.
+
+import { IProviderModule } from '../core/types';
+
+import * as claude from './claude';
+import * as codex from './codex';
+import * as kimi from './kimi';
+import * as gemini from './gemini';
+
+export const PROVIDERS: IProviderModule[] = [claude, codex, kimi, gemini];
+
+export default PROVIDERS;

package/src/providers/kimi.ts

@@ -0,0 +1,155 @@
+// The Kimi provider - Moonshot AI's `kimi` CLI.
+//
+// Ported from jupyterlab_kimi_code_extension v0.7.8. Everything that made that
+// extension different from the base one is declared here as descriptor data,
+// so no core file knows this assistant exists:
+//
+// * `forkStrategy: 'server-copy'` - kimi has no fork verb at all, so the
+//   server copies the session directory and answers with the new id
+// * `colourSource: 'derived'` - kimi has no `/color`, so the default tint is a
+//   stable hash of the conversation id (a user-set colour still wins, per the
+//   Colour section of the acceptance criteria)
+// * `namingStrategy: 'server-side'` - the name is stamped into the copied
+//   `state.json`, not passed to the CLI
+// * one launch mode, `--yolo`, under kimi's own name
+//
+// The only hook is the branch label: kimi's conversation titles are
+// auto-generated by a Chinese-first model, and both the session id shape and
+// the glyph widths of those titles need handling the shared default cannot do.
+
+import { IBranch, IProviderDescriptor, IProviderHooks } from '../core/types';
+
+// Crescent moon (Kimi is Moonshot AI's CLI) - an outer semicircle on the left,
+// bitten from the right by a wider inner arc.
+const kimiSvgStr = `<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16" width="16" height="16">
+  <g class="jp-icon3" fill="#616161">
+    <path d="M8 1.5A6.5 6.5 0 1 0 8 14.5 8 8 0 0 1 8 1.5Z"/>
+  </g>
+</svg>`;
+
+// ------------------------------------------------------------------- labels
+
+// Budget is in COLUMNS, not characters. Han, Kana, Hangul and emoji render at
+// roughly one em against Latin's half, so counting characters bounds nothing:
+// 60 Han glyphs measured 851-862px in a real Lumino submenu, at or above the
+// 850px that filed the defect in the first place. Kimi is Moonshot AI's CLI,
+// so a Chinese auto-generated title is the expected case, not an exotic one.
+export const MENU_TITLE_COLUMNS = 60;
+
+// East Asian Wide and Fullwidth ranges, plus emoji. Deliberately coarse: this
+// bounds a menu width, so a handful of mis-classified glyphs cost pixels, not
+// correctness. Anything outside these ranges counts as one column.
+const WIDE_RANGES: ReadonlyArray<readonly [number, number]> = [
+  [0x1100, 0x115f], // Hangul Jamo
+  [0x2e80, 0x303e], // CJK radicals, Kangxi, CJK symbols and punctuation
+  [0x3041, 0x33ff], // Hiragana, Katakana, Bopomofo, Hangul Compat, CJK Compat
+  [0x3400, 0x4dbf], // CJK Unified Extension A
+  [0x4e00, 0x9fff], // CJK Unified Ideographs
+  [0xa000, 0xa4cf], // Yi
+  [0xac00, 0xd7a3], // Hangul syllables
+  [0xf900, 0xfaff], // CJK Compatibility Ideographs
+  [0xfe30, 0xfe6f], // CJK Compatibility Forms
+  [0xff00, 0xff60], // Fullwidth forms
+  [0xffe0, 0xffe6], // Fullwidth signs
+  [0x1f300, 0x1faff], // Emoji and pictographs
+  [0x20000, 0x3fffd] // CJK Unified Extensions B and beyond
+];
+
+function isWide(codePoint: number): boolean {
+  return WIDE_RANGES.some(([lo, hi]) => codePoint >= lo && codePoint <= hi);
+}
+
+/** Trim `text` to `maxColumns` display columns, appending an ellipsis when
+ * anything was cut. Iterates by CODE POINT, so an astral character is never
+ * split into a lone surrogate (which chromium paints as the replacement
+ * glyph). Returns `text` unchanged when it already fits. */
+export function truncateToColumns(text: string, maxColumns: number): string {
+  let columns = 0;
+  let kept = '';
+  for (const ch of text) {
+    const width = isWide(ch.codePointAt(0) as number) ? 2 : 1;
+    if (columns + width > maxColumns) {
+      // The cut can land just after a space; drop it so the ellipsis hugs the
+      // last word. `trimEnd` is ES2019 and this project targets ES2018.
+      return `${kept.replace(/\s+$/, '')}…`;
+    }
+    columns += width;
+    kept += ch;
+  }
+  return text;
+}
+
+/** The distinguishing part of a kimi session id.
+ *
+ * Kimi ids are `session_<uuid>`, so the core's front-slice would render the
+ * constant `session_` for every conversation of a project - the exact thing
+ * the short id exists to tell apart. Slice past the prefix instead.
+ */
+export function shortSessionId(sessionId: string): string {
+  return sessionId.startsWith('session_')
+    ? sessionId.slice(8, 16)
+    : sessionId.slice(0, 8);
+}
+
+/** Menu-item label for one branch: the conversation title trimmed to the
+ * column budget, followed by the short session id. Only the TITLE is trimmed -
+ * branches of one project share a path, so the id is what tells them apart and
+ * it must survive. The bracketed id is dropped when the title already IS the
+ * id (the server's last-resort fallback), which can never be truncated since a
+ * short id is at most 8 characters. */
+export function branchMenuLabel(label: string, shortId: string): string {
+  const title = truncateToColumns(label, MENU_TITLE_COLUMNS);
+  return label === shortId ? title : `${title} (${shortId})`;
+}
+
+// -------------------------------------------------------------- descriptor
+
+export const descriptor: IProviderDescriptor = {
+  id: 'kimi',
+  label: 'Kimi',
+  panelTitle: 'Kimi Code Sessions',
+  iconName: 'jupyterlab_ai_code_assistants_extension:kimi',
+  iconSvg: kimiSvgStr,
+  cliBinary: 'kimi',
+  // No fork flag exists, so the server copies the session directory under a
+  // fresh id and the launch resumes that id like any other conversation.
+  forkStrategy: 'server-copy',
+  // No `/color` command either: the tint is derived from the conversation id.
+  // The core's own FNV-1a is the same hash over the same six ids the retired
+  // extension used, so every conversation keeps the colour it already had.
+  colourSource: 'derived',
+  // The fork's name is written into the copy's `state.json` by the server; the
+  // CLI has no naming flag to take it.
+  namingStrategy: 'server-side',
+  // A bare `kimi` mints its own id and writes it on the first turn, and `-S`
+  // only ever resumes - so a new conversation cannot be launched under an id
+  // we chose, and is identified from the store on the next poll instead.
+  mintsNewSessionId: false,
+  launchModes: [
+    {
+      id: 'yoloMode',
+      title: 'YOLO mode',
+      description:
+        'When enabled, newly spawned and resumed sessions are launched with `--yolo`. Kimi auto-approves regular tool calls without asking - only enable in workspaces you trust.',
+      kind: 'boolean',
+      default: false,
+      menuLabel: 'YOLO'
+    }
+  ],
+  hasRemoteControl: false,
+  hasBgAgents: false,
+  hasLiveProcess: false,
+  // Retired standalone extension this provider replaces.
+  legacyPluginId: 'jupyterlab_kimi_code_extension'
+};
+
+// ------------------------------------------------------------------- hooks
+
+export const hooks: IProviderHooks = {
+  /** Lumino sets no `max-width` on `.lm-Menu-itemLabel`, so an uncapped
+   * auto-generated title stretches the submenu across the window. The core's
+   * `shortId` argument is a front-slice, which is the constant `session_` for
+   * every kimi id, so the id is re-derived here. */
+  branchLabel: (branch: IBranch, _shortId: string): string =>
+    branchMenuLabel(branch.label, shortSessionId(branch.session_id))
+};