add local kernel support

Author: theomonnom

livekit-rtc/jupyter-html/src/App.tsx

@@ -20,16 +20,38 @@ export async function fetchJoinInfo(): Promise<{ url: string; token: string }> {
   if (invoke) {
     const res = await invoke("get_join_token", []);
     return res.data["application/json"];
-  } else if ((window as any).jupyterFetchJoinToken) {
-    return await (window as any).jupyterFetchJoinToken();
-  } else if (import.meta.env.MODE === "development") {
-    // use env variables
+  }
+
+
+  // This requires that JupyterLab was started with --LabApp.expose_app_in_browser,
+  if ((window as any).jupyterapp && (window as any).jupyterapp.shell) {
+    const currentWidget = (window as any).jupyterapp.shell.currentWidget;
+    if (currentWidget && currentWidget.context?.sessionContext) {
+      const session = currentWidget.context.sessionContext.session;
+      if (session && session.kernel) {
+        try {
+          const comm = session.kernel.createComm("get_join_token_comm");
+          comm.open();
+          comm.send({ request: "token" });
+          return new Promise((resolve) => {
+            comm.onMsg = (msg: any) => {
+              resolve(msg.content.data);
+            };
+          });
+        } catch (error) {
+          throw new Error("Error creating comm channel: " + error);
+        }
+      }
+    }
+  }
+
+  if (import.meta.env.MODE === "development") {
     const url = import.meta.env.VITE_LIVEKIT_URL;
     const token = import.meta.env.VITE_LIVEKIT_TOKEN;
-    return { url: url, token: token };
-  } else {
-    throw new Error("No Colab or Jupyter kernel function available");
+    return { url, token };
   }
+
+  throw new Error("No suitable kernel connection available");
 }
 
 

livekit-rtc/livekit/rtc/jupyter.py

@@ -1,3 +1,4 @@
+# type: ignore
 from __future__ import annotations
 
 import atexit
@@ -13,6 +14,7 @@
 
 def room_html(url: str, token: str) -> HTML:
     IN_COLAB = "google.colab" in sys.modules
+    IN_JUPYTER = "ipykernel" in sys.modules
 
     if IN_COLAB:
         from google.colab import output
@@ -21,6 +23,18 @@ def get_join_token():
             return JSON({"url": url, "token": token})
 
         output.register_callback("get_join_token", get_join_token)
+    elif IN_JUPYTER:
+        from IPython import get_ipython
+
+        ip = get_ipython()
+        if ip and hasattr(ip, "kernel"):
+
+            def token_comm_target(comm, open_msg):
+                @comm.on_msg
+                def handle_message(msg):
+                    comm.send({"url": url, "token": token})
+
+            ip.kernel.comm_manager.register_target("get_join_token_comm", token_comm_target)
 
     index_path = files("livekit.rtc.resources") / "jupyter-html" / "index.html"
     index_path = _resource_stack.enter_context(as_file(index_path))