Shroominic commited on
Commit
65d2933
·
1 Parent(s): ff1efb6

async context manager support

Browse files
Files changed (1) hide show
  1. codeinterpreterapi/session.py +7 -0
codeinterpreterapi/session.py CHANGED
@@ -174,3 +174,10 @@ class CodeInterpreterSession():
174
  "Sorry, something went while generating your response."
175
  "Please try again or restart the session."
176
  )
 
 
 
 
 
 
 
 
174
  "Sorry, something went while generating your response."
175
  "Please try again or restart the session."
176
  )
177
+
178
+ async def __aenter__(self) -> "CodeInterpreterSession":
179
+ await self._init()
180
+ return self
181
+
182
+ async def __aexit__(self, exc_type, exc_value, traceback) -> None:
183
+ await self._close()