@nteract/pi 0.1.8 → 0.2.0

package/extensions/repl.ts

@@ -89,8 +89,29 @@ type QueuedExecution = {
   executionId: string;
 };
 
+type ExecutionViewSnapshot = {
+  execution_count: number | null;
+  status: string;
+  success: boolean | null;
+  output_ids: string[];
+};
+
+type ExecutionViewChangeset = {
+  execution_upserts?: Array<[executionId: string, snapshot: ExecutionViewSnapshot]>;
+};
+
+type SubscriptionLike = {
+  unsubscribe?: () => void;
+  dispose?: () => void;
+};
+
+type ObservableLike<T> = {
+  subscribe(next: (value: T) => void): SubscriptionLike;
+};
+
 type Session = {
   readonly notebookId: string;
+  readonly executionViewChanges$?: ObservableLike<ExecutionViewChangeset>;
   runCell(
     source: string,
     opts?: { timeoutMs?: number; cellType?: string; onUpdate?: (progress: CellResult) => void },
@@ -649,6 +670,27 @@ export default function nteractReplExtension(pi: ExtensionAPI) {
   let session: Session | null = null;
   let opening: Promise<Session> | null = null;
   let nextExecCount: number | null = 1;
+  let executionViewSubscription: SubscriptionLike | null = null;
+
+  function disposeExecutionViewTracking(): void {
+    const sub = executionViewSubscription;
+    executionViewSubscription = null;
+    sub?.unsubscribe?.();
+    sub?.dispose?.();
+  }
+
+  function trackExecutionView(sess: Session): void {
+    disposeExecutionViewTracking();
+    executionViewSubscription =
+      sess.executionViewChanges$?.subscribe((changeset) => {
+        for (const [, snapshot] of changeset.execution_upserts ?? []) {
+          const count = snapshot.execution_count;
+          if (typeof count === "number") {
+            nextExecCount = Math.max(nextExecCount ?? 1, count + 1);
+          }
+        }
+      }) ?? null;
+  }
 
   async function addDependenciesAndSync(sess: Session, packages: string[]): Promise<void> {
     const unique = Array.from(new Set(packages.map((pkg) => pkg.trim()).filter(Boolean)));
@@ -689,6 +731,7 @@ export default function nteractReplExtension(pi: ExtensionAPI) {
         description: "pi Python REPL",
         dependencies,
       });
+      trackExecutionView(session);
       return session;
     })();
     try {
@@ -951,6 +994,7 @@ export default function nteractReplExtension(pi: ExtensionAPI) {
       const old = session;
       session = null;
       nextExecCount = 1;
+      disposeExecutionViewTracking();
       if (old) {
         try {
           if (old.shutdownNotebook) {
@@ -968,6 +1012,7 @@ export default function nteractReplExtension(pi: ExtensionAPI) {
   });
 
   pi.on("session_shutdown", async () => {
+    disposeExecutionViewTracking();
     if (session) {
       try {
         if (session.shutdownNotebook) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nteract/pi",
-  "version": "0.1.8",
+  "version": "0.2.0",
   "description": "Persistent notebook-backed Python REPL for Pi coding agents. Stateful execution, hot dependency sync, zero cold starts.",
   "author": "nteract contributors",
   "icon": "icon.png",
@@ -8,12 +8,12 @@
   "license": "BSD-3-Clause",
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/nteract/desktop.git",
+    "url": "git+https://github.com/nteract/nteract.git",
     "directory": "plugins/nteract/pi"
   },
-  "homepage": "https://github.com/nteract/desktop/tree/main/plugins/nteract/pi#readme",
+  "homepage": "https://github.com/nteract/nteract/tree/main/plugins/nteract/pi#readme",
   "bugs": {
-    "url": "https://github.com/nteract/desktop/issues"
+    "url": "https://github.com/nteract/nteract/issues"
   },
   "keywords": [
     "pi-package",
@@ -36,7 +36,7 @@
     "extensions"
   ],
   "dependencies": {
-    "@runtimed/node": "0.2.5"
+    "@runtimed/node": "0.3.0"
   },
   "peerDependencies": {
     "@earendil-works/pi-ai": "*",