[RELEASE] v0.11.0

ci/release/update-version.sh

@@ -29,5 +29,5 @@ function sed_runner() {
     sed -i.bak ''"$1"'' $2 && rm -f ${2}.bak
 }
 
-jq -e --arg tag "$NEXT_FULL_TAG" '.version=$tag' package.json > package.json.tmp
+jq --indent 4 -e --arg tag "$NEXT_FULL_TAG" '.version=$tag' package.json > package.json.tmp
 mv package.json.tmp package.json

jupyterlab_nvdashboard/apps/utils.py

@@ -1,10 +1,15 @@
 from tornado.websocket import WebSocketHandler
+from jupyter_server.base.handlers import JupyterHandler
 import tornado
 import json
 
 
-class CustomWebSocketHandler(WebSocketHandler):
+class CustomWebSocketHandler(JupyterHandler, WebSocketHandler):
     def open(self):
+        if not self.current_user:
+            self.write_message(json.dumps({"error": "Unauthorized access"}))
+            self.close()
+            return
         self.write_message(json.dumps({"status": "connected"}))
         self.set_nodelay(True)
         # Start a periodic callback to send data every 50ms

jupyterlab_nvdashboard/tests/test_cpu_handlers.py

@@ -20,6 +20,11 @@ def handler_args():
     with patch("tornado.web.Application") as mock_application, patch(
         "tornado.httputil.HTTPServerRequest"
     ) as mock_request:
+        # Mock the settings to return appropriate values
+        mock_settings = {
+            "base_url": "/",
+        }
+        mock_application.settings = mock_settings
         yield mock_application, mock_request
 
 

jupyterlab_nvdashboard/tests/test_gpu_handlers.py

@@ -26,6 +26,11 @@ def handler_args():
     with patch("tornado.web.Application") as mock_application, patch(
         "tornado.httputil.HTTPServerRequest"
     ) as mock_request:
+        # Mock the settings to return appropriate values
+        mock_settings = {
+            "base_url": "/",
+        }
+        mock_application.settings = mock_settings
         yield mock_application, mock_request
 
 

package.json

@@ -1,6 +1,6 @@
 {
     "name": "jupyterlab-nvdashboard",
-    "version": "0.10.0",
+    "version": "0.11.0",
     "description": "A JupyterLab extension for displaying GPU usage dashboards",
     "keywords": [
         "jupyter",

src/charts/PciThroughputChart.tsx

@@ -24,7 +24,7 @@ const PciThroughputChart: React.FC<IChartProps> = ({ settingRegistry }) => {
   // Load settings and initialize WebSocket connection
   loadSettingRegistry(settingRegistry, setUpdateFrequency, setIsSettingsLoaded);
   useWebSocket<IPCIThroughputProps>(
-    'nvlink_throughput',
+    'pci_stats',
     false,
     updateFrequency,
     setPciStats,

src/handler.ts

@@ -1,14 +1,18 @@
+import { URLExt } from '@jupyterlab/coreutils';
+import { ServerConnection } from '@jupyterlab/services';
+
 /**
  * Connect to a WebSocket API endpoint
  *
  * @param endPoint WebSocket endpoint for the extension
  * @returns A WebSocket object connected to the endpoint
  */
 export function connectToWebSocket(endPoint = '') {
-  const baseUrl = document.location.origin.replace(/^http/, 'ws');
-  const wsUrl = new URL(`nvdashboard/${endPoint}`, baseUrl);
+  const settings = ServerConnection.makeSettings();
+  const baseUrl = settings.baseUrl.replace(/^http/, 'ws');
+  const wsUrl = URLExt.join(baseUrl, 'nvdashboard', endPoint);
 
-  const ws = new WebSocket(wsUrl.href);
+  const ws = new WebSocket(wsUrl);
 
   ws.onopen = () => {
     console.log(`WebSocket connected to ${endPoint}`);