Skip to content

Commit 98e6e4b

Browse files
committed
Add possibility for custom kernels
1 parent 5cd48bc commit 98e6e4b

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

python/e2b_code_interpreter/main.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,8 +79,11 @@ def default_kernel_id(self) -> str:
7979

8080
return self._default_kernel_id
8181

82-
def create_kernel(self, cwd: Optional[str] = None,timeout: Optional[float] = TIMEOUT) -> str:
83-
data = {"cwd": cwd} if cwd else None
82+
def create_kernel(self, cwd: str = "/home/user", kernel_name: Optional[str] = None, timeout: Optional[float] = TIMEOUT) -> str:
83+
data = {"cwd": cwd}
84+
if kernel_name:
85+
data["kernel_name"] = kernel_name
86+
8487
response = requests.post(
8588
f"{self._sandbox.get_protocol()}://{self._sandbox.get_hostname(8888)}/api/kernels",
8689
json=data,

0 commit comments

Comments
 (0)