Handle wrong language

Author: jakubno

python/tests/sync/test_default_kernels.py

@@ -1,3 +1,5 @@
+import pytest
+
 from e2b_code_interpreter.code_interpreter_sync import Sandbox
 
 
@@ -6,11 +8,13 @@ def test_js_kernel(sandbox: Sandbox):
     assert execution.logs.stdout == ["Hello, World!\n"]
 
 
+@pytest.mark.skip_debug()
 def test_r_kernel(sandbox: Sandbox):
     execution = sandbox.run_code('print("Hello, World!")', language="r")
     assert execution.logs.stdout == ['[1] "Hello, World!"\n']
 
 
+@pytest.mark.skip_debug()
 def test_java_kernel(sandbox: Sandbox):
     execution = sandbox.run_code('System.out.println("Hello, World!")', language="java")
     assert execution.logs.stdout[0] == "Hello, World!"

template/server/contexts.py

@@ -36,9 +36,8 @@ async def create_context(client, websockets: dict, language: str, cwd: str) -> C
     response = await client.post(f"{JUPYTER_BASE_URL}/api/sessions", json=data)
 
     if not response.is_success:
-        return PlainTextResponse(
+        raise Exception(
             f"Failed to create context: {response.text}",
-            status_code=500,
         )
 
     session_data = response.json()

template/server/main.py

@@ -89,9 +89,13 @@ async def post_execute(request: ExecutionRequest):
             context_id = default_websockets.get(language)
 
             if not context_id:
-                context = await create_context(
-                    client, websockets, language, "/home/user"
-                )
+                try:
+                    context = await create_context(
+                        client, websockets, language, "/home/user"
+                    )
+                except Exception as e:
+                    return PlainTextResponse(str(e), status_code=500)
+
                 context_id = context.id
                 default_websockets[language] = context_id
 
@@ -124,7 +128,10 @@ async def post_contexts(request: CreateContext) -> Context:
     language = normalize_language(request.language)
     cwd = request.cwd or "/home/user"
 
-    return await create_context(client, websockets, language, cwd)
+    try:
+        return await create_context(client, websockets, language, cwd)
+    except Exception as e:
+        return PlainTextResponse(str(e), status_code=500)
 
 
 @app.get("/contexts")