bugfix for rank

Author: terapyon

src/index.ts

@@ -6,7 +6,7 @@ import {
   JupyterFrontEndPlugin,
 } from '@jupyterlab/application';
 
-import { IRenderMimeRegistry } from '@jupyterlab/rendermime';
+import { IRenderMimeRegistry, IRenderMime } from '@jupyterlab/rendermime';
 import { NetVisRenderer, MIME_TYPE } from './renderer';
 
 // Export other modules for backward compatibility
@@ -25,16 +25,29 @@ const mimeExtension: JupyterFrontEndPlugin<void> = {
     rendermime: IRenderMimeRegistry,
   ): void => {
     // Register MIME renderer for _repr_mimebundle_
-    rendermime.addFactory(
-      {
-        safe: true,
-        mimeTypes: [MIME_TYPE],
-        createRenderer: (options) => new NetVisRenderer(options),
-      },
-      0, // rank: 0 means highest priority
-    );
+    // JupyterLab 4.x compatible registration
+    const rendererFactory: IRenderMime.IRendererFactory = {
+      safe: true,
+      mimeTypes: [MIME_TYPE],
+      createRenderer: (options: IRenderMime.IRendererOptions) => new NetVisRenderer(options),
+    };
 
-    console.log('NetVis MIME renderer registered');
+    try {
+      // Try to add factory with default rank
+      // JupyterLab 4.x doesn't use rank parameter directly
+      rendermime.addFactory(rendererFactory);
+      console.log('NetVis MIME renderer registered successfully');
+    } catch (error) {
+      console.error('Failed to register NetVis MIME renderer:', error);
+      // Fallback: try alternative registration if the first attempt fails
+      try {
+        // @ts-ignore - For backward compatibility
+        rendermime.addFactory(rendererFactory, 0);
+        console.log('NetVis MIME renderer registered with rank 0');
+      } catch (fallbackError) {
+        console.error('Fallback registration also failed:', fallbackError);
+      }
+    }
   },
 };