pi-orchestrator-config 4.1.5

package/extensions/shared/stream-builder.ts

@@ -0,0 +1,275 @@
+/**
+ * Shared stream builder — assembles DriverStreamEvents into pi's
+ * AssistantMessageEventStream.
+ *
+ * Both cli-provider and acpx-provider had near-identical ~100-line blocks
+ * for mapping text_delta/thinking_delta events into the pi event stream
+ * format. This module unifies that into a single, testable function.
+ *
+ * @module shared/stream-builder
+ */
+
+import type {
+  AssistantMessage,
+  AssistantMessageEventStream,
+} from "@earendil-works/pi-ai";
+import type { DriverStreamEvent } from "./provider-driver.js";
+
+/**
+ * State tracker for building AssistantMessage content blocks from
+ * a stream of DriverStreamEvents. Manages thinking/text block indices
+ * and emits the correct pi event stream events.
+ */
+export class StreamAssembler {
+  private thinkingIndex = -1;
+  private textIndex = -1;
+  private thinkingClosed = false;
+
+  constructor(
+    private readonly output: AssistantMessage,
+    private readonly stream: AssistantMessageEventStream,
+  ) {}
+
+  /** Process a single driver event and emit corresponding pi stream events. */
+  handleEvent(event: DriverStreamEvent): void {
+    switch (event.kind) {
+      case "thinking_delta":
+        this.handleThinkingDelta(event.text);
+        break;
+      case "text_delta":
+        this.handleTextDelta(event.text);
+        break;
+      case "session":
+        // Session events are handled by the driver adapter, not the stream builder.
+        break;
+      case "turn_complete":
+        // Handled by finalize().
+        break;
+    }
+  }
+
+  /**
+   * Finalize the stream — close any open content blocks and emit done.
+   * Call after all events have been processed or when the turn completes.
+   *
+   * @param finalText - Authoritative final text from the parser (may differ from accumulated deltas).
+   * @param finalThinking - Authoritative final thinking from the parser.
+   * @param stopReason - Stop reason for the done event.
+   */
+  finalize(opts?: {
+    finalText?: string;
+    finalThinking?: string;
+    stopReason?: string;
+    usage?: import("./provider-driver.js").DriverUsage;
+  }): void {
+    const finalText = opts?.finalText;
+    const finalThinking = opts?.finalThinking;
+
+    // Close thinking block if still open
+    if (this.thinkingIndex >= 0 && !this.thinkingClosed) {
+      const block = this.output.content[this.thinkingIndex];
+      if (block.type === "thinking") {
+        // Apply authoritative thinking if available and block was empty
+        if (!block.thinking && finalThinking) {
+          block.thinking = finalThinking;
+          this.stream.push({
+            type: "thinking_delta",
+            contentIndex: this.thinkingIndex,
+            delta: finalThinking,
+            partial: this.output,
+          });
+        }
+        this.stream.push({
+          type: "thinking_end",
+          contentIndex: this.thinkingIndex,
+          content: block.thinking,
+          partial: this.output,
+        });
+      }
+      this.thinkingClosed = true;
+    }
+
+    // Handle text: either apply authoritative final text or close existing block
+    if (this.textIndex < 0 && finalText) {
+      // No text block yet but parser returned text — create and emit it
+      this.output.content.push({ type: "text", text: finalText });
+      this.textIndex = this.output.content.length - 1;
+      this.stream.push({
+        type: "text_start",
+        contentIndex: this.textIndex,
+        partial: this.output,
+      });
+      this.stream.push({
+        type: "text_delta",
+        contentIndex: this.textIndex,
+        delta: finalText,
+        partial: this.output,
+      });
+    } else if (this.textIndex >= 0 && finalText) {
+      const block = this.output.content[this.textIndex];
+      if (block.type === "text") {
+        // Prefer authoritative final text if it differs
+        if (finalText !== block.text) {
+          const missing = finalText.startsWith(block.text)
+            ? finalText.slice(block.text.length)
+            : "";
+          if (missing) {
+            block.text = finalText;
+            this.stream.push({
+              type: "text_delta",
+              contentIndex: this.textIndex,
+              delta: missing,
+              partial: this.output,
+            });
+          } else {
+            block.text = finalText;
+          }
+        }
+      }
+    }
+
+    // Close text block
+    if (this.textIndex >= 0) {
+      const block = this.output.content[this.textIndex];
+      if (block.type === "text") {
+        this.stream.push({
+          type: "text_end",
+          contentIndex: this.textIndex,
+          content: block.text,
+          partial: this.output,
+        });
+      }
+    }
+
+    // Map driver usage to pi's usage format
+    if (opts?.usage) {
+      const u = opts.usage;
+      this.output.usage = {
+        input: u.inputTokens ?? 0,
+        output: u.outputTokens ?? 0,
+        cacheRead: u.cachedReadTokens ?? 0,
+        cacheWrite: u.cachedWriteTokens ?? 0,
+        totalTokens: u.totalTokens ?? ((u.inputTokens ?? 0) + (u.outputTokens ?? 0)),
+        cost: {
+          input: 0,
+          output: 0,
+          cacheRead: 0,
+          cacheWrite: 0,
+          total: u.costUsd ?? 0,
+        },
+      };
+    }
+
+    // AssistantMessage.stopReason keeps the full driver value (length, toolUse, …).
+    // pi's done-event reason is only "stop" | "aborted" — map everything else to "stop".
+    const stopReason = this.output.stopReason = opts?.stopReason || "stop";
+    const reason = (stopReason === "aborted" ? "aborted" : "stop") as "stop" | "aborted";
+    this.stream.push({ type: "done", reason, message: this.output });
+    this.stream.end();
+  }
+
+  /**
+   * Emit an error event and end the stream.
+   */
+  emitError(error: unknown, aborted?: boolean): void {
+    this.output.stopReason = aborted ? "aborted" : "error";
+    this.output.errorMessage =
+      error instanceof Error ? error.message : String(error);
+    this.stream.push({
+      type: "error",
+      reason: this.output.stopReason as "aborted" | "error",
+      error: this.output,
+    });
+    this.stream.end();
+  }
+
+  // -------------------------------------------------------------------------
+  // Private
+  // -------------------------------------------------------------------------
+
+  private handleThinkingDelta(text: string): void {
+    if (this.thinkingIndex < 0) {
+      this.output.content.push({ type: "thinking", thinking: "" });
+      this.thinkingIndex = this.output.content.length - 1;
+      this.stream.push({
+        type: "thinking_start",
+        contentIndex: this.thinkingIndex,
+        partial: this.output,
+      });
+    }
+    const block = this.output.content[this.thinkingIndex];
+    if (block.type === "thinking") {
+      block.thinking += text;
+      this.stream.push({
+        type: "thinking_delta",
+        contentIndex: this.thinkingIndex,
+        delta: text,
+        partial: this.output,
+      });
+    }
+  }
+
+  private handleTextDelta(text: string): void {
+    // Close thinking block on first text delta
+    if (this.thinkingIndex >= 0 && !this.thinkingClosed) {
+      this.thinkingClosed = true;
+      const thinkBlock = this.output.content[this.thinkingIndex];
+      if (thinkBlock.type === "thinking") {
+        this.stream.push({
+          type: "thinking_end",
+          contentIndex: this.thinkingIndex,
+          content: thinkBlock.thinking,
+          partial: this.output,
+        });
+      }
+    }
+
+    if (this.textIndex < 0) {
+      this.output.content.push({ type: "text", text: "" });
+      this.textIndex = this.output.content.length - 1;
+      this.stream.push({
+        type: "text_start",
+        contentIndex: this.textIndex,
+        partial: this.output,
+      });
+    }
+    const block = this.output.content[this.textIndex];
+    if (block.type === "text") {
+      block.text += text;
+      this.stream.push({
+        type: "text_delta",
+        contentIndex: this.textIndex,
+        delta: text,
+        partial: this.output,
+      });
+    }
+  }
+}
+
+/**
+ * Create a fresh AssistantMessage output object for stream building.
+ * Reusable across all drivers.
+ */
+export function createAssistantMessageOutput(model: {
+  api: string;
+  provider: string;
+  id: string;
+}): AssistantMessage {
+  return {
+    role: "assistant",
+    content: [],
+    api: model.api,
+    provider: model.provider,
+    model: model.id,
+    usage: {
+      input: 0,
+      output: 0,
+      cacheRead: 0,
+      cacheWrite: 0,
+      totalTokens: 0,
+      cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+    },
+    stopReason: "stop",
+    timestamp: Date.now(),
+  } as AssistantMessage;
+}

package/extensions/shared/types.ts

@@ -0,0 +1,20 @@
+/** Session info shared between pidash-server and pidash-ui. */
+export interface SessionInfo {
+  sessionId: string;
+  pid: number;
+  cwd: string;
+  branch: string;
+  model: string;
+  startedAt: string;
+  lastActivity: number;
+  active: boolean;
+  sessionFile?: string;
+  gitDirty?: boolean;
+  gitChanges?: number;
+  container?: boolean;
+  contextWindow?: number;
+  thinkingLevel?: string;
+  diffPort?: number | null;
+  working?: boolean;
+  name?: string;
+}

package/extensions/shared/ui/README.md

@@ -0,0 +1,10 @@
+# Shared UI Components
+
+shadcn/ui components shared between pidash-ui and pidiff-ui.
+Both UIs import these via the `@ui` Vite alias.
+
+To add a new shared component, place it here and it's automatically
+available in both UIs via `import { X } from "@ui/component-name"`.
+
+Components unique to one UI (e.g. pidiff's `switch.tsx`) stay in that UI's
+local `components/ui/` directory.

package/extensions/shared/ui/badge.tsx

@@ -0,0 +1,52 @@
+import { mergeProps } from "@base-ui/react/merge-props"
+import { useRender } from "@base-ui/react/use-render"
+import { cva, type VariantProps } from "class-variance-authority"
+
+import { cn } from "@/lib/utils"
+
+const badgeVariants = cva(
+  "group/badge inline-flex h-5 w-fit shrink-0 items-center justify-center gap-1 overflow-hidden rounded-4xl border border-transparent px-2 py-0.5 text-xs font-medium whitespace-nowrap transition-all focus-visible:border-ring focus-visible:ring-[3px] focus-visible:ring-ring/50 has-data-[icon=inline-end]:pr-1.5 has-data-[icon=inline-start]:pl-1.5 aria-invalid:border-destructive aria-invalid:ring-destructive/20 dark:aria-invalid:ring-destructive/40 [&>svg]:pointer-events-none [&>svg]:size-3!",
+  {
+    variants: {
+      variant: {
+        default: "bg-primary text-primary-foreground [a]:hover:bg-primary/80",
+        secondary:
+          "bg-secondary text-secondary-foreground [a]:hover:bg-secondary/80",
+        destructive:
+          "bg-destructive/10 text-destructive focus-visible:ring-destructive/20 dark:bg-destructive/20 dark:focus-visible:ring-destructive/40 [a]:hover:bg-destructive/20",
+        outline:
+          "border-border text-foreground [a]:hover:bg-muted [a]:hover:text-muted-foreground",
+        ghost:
+          "hover:bg-muted hover:text-muted-foreground dark:hover:bg-muted/50",
+        link: "text-primary underline-offset-4 hover:underline",
+      },
+    },
+    defaultVariants: {
+      variant: "default",
+    },
+  }
+)
+
+function Badge({
+  className,
+  variant = "default",
+  render,
+  ...props
+}: useRender.ComponentProps<"span"> & VariantProps<typeof badgeVariants>) {
+  return useRender({
+    defaultTagName: "span",
+    props: mergeProps<"span">(
+      {
+        className: cn(badgeVariants({ variant }), className),
+      },
+      props
+    ),
+    render,
+    state: {
+      slot: "badge",
+      variant,
+    },
+  })
+}
+
+export { Badge, badgeVariants }

package/extensions/shared/ui/button.tsx

@@ -0,0 +1,58 @@
+import { Button as ButtonPrimitive } from "@base-ui/react/button"
+import { cva, type VariantProps } from "class-variance-authority"
+
+import { cn } from "@/lib/utils"
+
+const buttonVariants = cva(
+  "group/button inline-flex shrink-0 items-center justify-center rounded-lg border border-transparent bg-clip-padding text-sm font-medium whitespace-nowrap transition-all outline-none select-none focus-visible:border-ring focus-visible:ring-3 focus-visible:ring-ring/50 active:not-aria-[haspopup]:translate-y-px disabled:pointer-events-none disabled:opacity-50 aria-invalid:border-destructive aria-invalid:ring-3 aria-invalid:ring-destructive/20 dark:aria-invalid:border-destructive/50 dark:aria-invalid:ring-destructive/40 [&_svg]:pointer-events-none [&_svg]:shrink-0 [&_svg:not([class*='size-'])]:size-4",
+  {
+    variants: {
+      variant: {
+        default: "bg-primary text-primary-foreground [a]:hover:bg-primary/80",
+        outline:
+          "border-border bg-background hover:bg-muted hover:text-foreground aria-expanded:bg-muted aria-expanded:text-foreground dark:border-input dark:bg-input/30 dark:hover:bg-input/50",
+        secondary:
+          "bg-secondary text-secondary-foreground hover:bg-secondary/80 aria-expanded:bg-secondary aria-expanded:text-secondary-foreground",
+        ghost:
+          "hover:bg-muted hover:text-foreground aria-expanded:bg-muted aria-expanded:text-foreground dark:hover:bg-muted/50",
+        destructive:
+          "bg-destructive/10 text-destructive hover:bg-destructive/20 focus-visible:border-destructive/40 focus-visible:ring-destructive/20 dark:bg-destructive/20 dark:hover:bg-destructive/30 dark:focus-visible:ring-destructive/40",
+        link: "text-primary underline-offset-4 hover:underline",
+      },
+      size: {
+        default:
+          "h-8 gap-1.5 px-2.5 has-data-[icon=inline-end]:pr-2 has-data-[icon=inline-start]:pl-2",
+        xs: "h-6 gap-1 rounded-[min(var(--radius-md),10px)] px-2 text-xs in-data-[slot=button-group]:rounded-lg has-data-[icon=inline-end]:pr-1.5 has-data-[icon=inline-start]:pl-1.5 [&_svg:not([class*='size-'])]:size-3",
+        sm: "h-7 gap-1 rounded-[min(var(--radius-md),12px)] px-2.5 text-[0.8rem] in-data-[slot=button-group]:rounded-lg has-data-[icon=inline-end]:pr-1.5 has-data-[icon=inline-start]:pl-1.5 [&_svg:not([class*='size-'])]:size-3.5",
+        lg: "h-9 gap-1.5 px-2.5 has-data-[icon=inline-end]:pr-2 has-data-[icon=inline-start]:pl-2",
+        icon: "size-8",
+        "icon-xs":
+          "size-6 rounded-[min(var(--radius-md),10px)] in-data-[slot=button-group]:rounded-lg [&_svg:not([class*='size-'])]:size-3",
+        "icon-sm":
+          "size-7 rounded-[min(var(--radius-md),12px)] in-data-[slot=button-group]:rounded-lg",
+        "icon-lg": "size-9",
+      },
+    },
+    defaultVariants: {
+      variant: "default",
+      size: "default",
+    },
+  }
+)
+
+function Button({
+  className,
+  variant = "default",
+  size = "default",
+  ...props
+}: ButtonPrimitive.Props & VariantProps<typeof buttonVariants>) {
+  return (
+    <ButtonPrimitive
+      data-slot="button"
+      className={cn(buttonVariants({ variant, size, className }))}
+      {...props}
+    />
+  )
+}
+
+export { Button, buttonVariants }

package/extensions/shared/ui/collapsible.tsx

@@ -0,0 +1,21 @@
+"use client"
+
+import { Collapsible as CollapsiblePrimitive } from "@base-ui/react/collapsible"
+
+function Collapsible({ ...props }: CollapsiblePrimitive.Root.Props) {
+  return <CollapsiblePrimitive.Root data-slot="collapsible" {...props} />
+}
+
+function CollapsibleTrigger({ ...props }: CollapsiblePrimitive.Trigger.Props) {
+  return (
+    <CollapsiblePrimitive.Trigger data-slot="collapsible-trigger" {...props} />
+  )
+}
+
+function CollapsibleContent({ ...props }: CollapsiblePrimitive.Panel.Props) {
+  return (
+    <CollapsiblePrimitive.Panel data-slot="collapsible-content" {...props} />
+  )
+}
+
+export { Collapsible, CollapsibleTrigger, CollapsibleContent }