diff options
| author | Nate Sesti <sestinj@gmail.com> | 2023-08-04 21:21:55 -0700 |
|---|---|---|
| committer | Nate Sesti <sestinj@gmail.com> | 2023-08-04 21:21:55 -0700 |
| commit | 345b7734d8c887d699d5038416d2a1f8193a33e9 (patch) | |
| tree | e2a7151c3266fa447cbff03d730106049f3187c0 /continuedev/src/continuedev/server | |
| parent | 51abe8eb79f56b2a32dbbef19e8d352eeca56abe (diff) | |
| download | sncontinue-345b7734d8c887d699d5038416d2a1f8193a33e9.tar.gz sncontinue-345b7734d8c887d699d5038416d2a1f8193a33e9.tar.bz2 sncontinue-345b7734d8c887d699d5038416d2a1f8193a33e9.zip | |
fix: :bug: timeout on blocking processes
Diffstat (limited to 'continuedev/src/continuedev/server')
| -rw-r--r-- | continuedev/src/continuedev/server/ide.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/continuedev/src/continuedev/server/ide.py b/continuedev/src/continuedev/server/ide.py index 9797a8b7..e523d3aa 100644 --- a/continuedev/src/continuedev/server/ide.py +++ b/continuedev/src/continuedev/server/ide.py @@ -249,8 +249,10 @@ class IdeProtocolServer(AbstractIdeProtocolServer): }) async def getSessionId(self): - session_id = (await self.session_manager.new_session( - self, self.session_id)).session_id + new_session = await asyncio.wait_for(self.session_manager.new_session( + self, self.session_id), timeout=5) + session_id = new_session.session_id + logger.debug(f"Sending session id: {session_id}") await self._send_json("getSessionId", { "sessionId": session_id }) @@ -279,11 +281,9 @@ class IdeProtocolServer(AbstractIdeProtocolServer): return any([r.accepted for r in responses]) def on_error(self, e: Exception) -> Coroutine: - try: - return self.session_manager.sessions[self.session_id].autopilot.continue_sdk.run_step(DisplayErrorStep(e=e)) - except: - err_msg = '\n'.join(traceback.format_exception(e)) - return self.showMessage(f"Error in Continue server: {err_msg}") + err_msg = '\n'.join(traceback.format_exception(e)) + e_title = e.__str__() or e.__repr__() + return self.showMessage(f"Error in Continue server: {e_title}\n {err_msg}") def onAcceptRejectSuggestion(self, accepted: bool): posthog_logger.capture_event("accept_reject_suggestion", { |