Change response to json

e2b-dev · Jul 4, 2024 · 868d600 · 868d600
1 parent 5a730ab
commit 868d600
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/template/server/main.py b/template/server/main.py
@@ -1,6 +1,7 @@
 import uuid
 
 from fastapi import FastAPI
+from starlette.responses import JSONResponse
 
 from messaging import JupyterKernelWebSocket
 from api.models.execution import Execution
@@ -26,5 +27,6 @@ def health():
 
 
 @app.post("/execute", response_model=Execution)
-def execute(request: ExecutionRequest):
-    return ws.execute(code=request.code)
+def execute(request: ExecutionRequest) -> JSONResponse:
+    result = ws.execute(code=request.code)
+    return JSONResponse(content=result)
diff --git a/template/server/messaging.py b/template/server/messaging.py
@@ -55,7 +55,7 @@ def __init__(
         on_stderr: Optional[Callable[[CellMessage], Any]] = None,
         on_result: Optional[Callable[[Result], Any]] = None,
     ):
-        self.partial_result = Execution()
+        self.partial_result = Execution(results=[])
         self.execution = Future()
         self.on_stdout = on_stdout
         self.on_stderr = on_stderr