💬 return status on start/end

ishaan-jaff · Aug 8, 2023 · 0a41406 · 0a41406
1 parent 45bf730
commit 0a41406
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/codeinterpreterapi/schema/__init__.py b/codeinterpreterapi/schema/__init__.py
@@ -1,11 +1,13 @@
 from .file import File
 from .input import CodeInput, FileInput
 from .response import CodeInterpreterResponse, UserRequest
+from .status import SessionStatus
 
 __all__ = [
     "CodeInterpreterResponse",
     "CodeInput",
     "File",
     "FileInput",
     "UserRequest",
+    "SessionStatus",
 ]
diff --git a/codeinterpreterapi/schema/status.py b/codeinterpreterapi/schema/status.py
@@ -0,0 +1,10 @@
+from codeboxapi.schema import CodeBoxStatus  # type: ignore
+
+
+class SessionStatus(CodeBoxStatus):
+    @classmethod
+    def from_codebox_status(cls, cbs: CodeBoxStatus) -> "SessionStatus":
+        return cls(status=cbs.status)
+
+    def __repr__(self):
+        return f"<SessionStatus status={self.status}>"