diff options
| author | Nate Sesti <sestinj@gmail.com> | 2023-08-08 20:18:40 -0700 |
|---|---|---|
| committer | Nate Sesti <sestinj@gmail.com> | 2023-08-08 20:18:40 -0700 |
| commit | 6112f26888086ccd47ca6bcfefdbc5b82ea86879 (patch) | |
| tree | d09c392ae811b9287c78091920aec3992307bd7b /continuedev/src/continuedev/server | |
| parent | adae79d98dae05904bc416ab99a09b6354b6df28 (diff) | |
| download | sncontinue-6112f26888086ccd47ca6bcfefdbc5b82ea86879.tar.gz sncontinue-6112f26888086ccd47ca6bcfefdbc5b82ea86879.tar.bz2 sncontinue-6112f26888086ccd47ca6bcfefdbc5b82ea86879.zip | |
feat: :sparkles: testing improved prompting for stablecode
Diffstat (limited to 'continuedev/src/continuedev/server')
| -rw-r--r-- | continuedev/src/continuedev/server/gui.py | 4 | ||||
| -rw-r--r-- | continuedev/src/continuedev/server/ide.py | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/continuedev/src/continuedev/server/gui.py b/continuedev/src/continuedev/server/gui.py index 49d46be3..35c52a47 100644 --- a/continuedev/src/continuedev/server/gui.py +++ b/continuedev/src/continuedev/server/gui.py @@ -59,12 +59,12 @@ class GUIProtocolServer(AbstractGUIProtocolServer): "data": data }) - async def _receive_json(self, message_type: str, timeout: int = 20) -> Any: + async def _receive_json(self, message_type: str, timeout: int = 10) -> Any: try: return await asyncio.wait_for(self.sub_queue.get(message_type), timeout=timeout) except asyncio.TimeoutError: raise Exception( - "GUI Protocol _receive_json timed out after 20 seconds") + "GUI Protocol _receive_json timed out after 10 seconds") async def _send_and_receive_json(self, data: Any, resp_model: Type[T], message_type: str) -> T: await self._send_json(message_type, data) diff --git a/continuedev/src/continuedev/server/ide.py b/continuedev/src/continuedev/server/ide.py index e523d3aa..b0dbc552 100644 --- a/continuedev/src/continuedev/server/ide.py +++ b/continuedev/src/continuedev/server/ide.py @@ -165,12 +165,12 @@ class IdeProtocolServer(AbstractIdeProtocolServer): "data": data }) - async def _receive_json(self, message_type: str, timeout: int = 20) -> Any: + async def _receive_json(self, message_type: str, timeout: int = 10) -> Any: try: return await asyncio.wait_for(self.sub_queue.get(message_type), timeout=timeout) except asyncio.TimeoutError: raise Exception( - f"IDE Protocol _receive_json timed out after 20 seconds: {message_type}") + f"IDE Protocol _receive_json timed out after 10 seconds: {message_type}") async def _send_and_receive_json(self, data: Any, resp_model: Type[T], message_type: str) -> T: await self._send_json(message_type, data) |