🪓 fix localbox

Author: shroominic

codeboxapi/box/codebox.py

@@ -51,10 +51,15 @@ class CodeBox(BaseBox):
     """
 
     def __new__(cls, *args, **kwargs):
-        if settings.CODEBOX_API_KEY is None:
+        if (
+            settings.CODEBOX_API_KEY is None
+            or settings.CODEBOX_API_KEY == "local"
+            or kwargs.pop("local", False)
+        ):
             from .localbox import LocalBox
 
             return LocalBox(*args, **kwargs)
+
         return super().__new__(cls, *args, **kwargs)
 
     def __init__(self, *args, **kwargs) -> None:

codeboxapi/box/localbox.py

@@ -9,8 +9,10 @@
 import json
 import os
 import subprocess
+import sys
 import time
 from asyncio.subprocess import Process
+from pathlib import Path
 from typing import List, Optional, Union
 from uuid import uuid4
 
@@ -67,8 +69,11 @@ def start(self) -> CodeBoxStatus:
         else:
             out = subprocess.PIPE
         try:
+            python = Path(sys.executable).absolute()
             self.jupyter = subprocess.Popen(
                 [
+                    python,
+                    "-m",
                     "jupyter",
                     "kernelgateway",
                     "--KernelGatewayApp.ip='0.0.0.0'",
@@ -127,7 +132,10 @@ async def astart(self) -> CodeBoxStatus:
             out = None
         else:
             out = asyncio.subprocess.PIPE
+        python = Path(sys.executable).absolute()
         self.jupyter = await asyncio.create_subprocess_exec(
+            python,
+            "-m",
             "jupyter",
             "kernelgateway",
             "--KernelGatewayApp.ip='0.0.0.0'",
@@ -307,11 +315,6 @@ def run(
                     print("Error:\n", error)
                 return CodeBoxOutput(type="error", content=error)
 
-            return CodeBoxOutput(
-                type="error",
-                content="Unknown message",
-            )
-
     async def arun(
         self,
         code: str,