quick text entry

1 files changed, 7 insertions, 0 deletions

diff --git a/continuedev/src/continuedev/server/ide.py b/continuedev/src/continuedev/server/ide.py
index d16bd449..1d51758e 100644
--- a/continuedev/src/continuedev/server/ide.py
+++ b/continuedev/src/continuedev/server/ide.py
@@ -150,6 +150,8 @@ class IdeProtocolServer(AbstractIdeProtocolServer):
             self.onAcceptRejectSuggestion(data["accepted"])
         elif message_type == "acceptRejectDiff":
             self.onAcceptRejectDiff(data["accepted"])
+        elif message_type == "mainUserInput":
+            self.onMainUserInput(data["input"])
         elif message_type in ["highlightedCode", "openFiles", "readFile", "editFile", "workspaceDirectory", "getUserSecret", "runCommand", "uniqueId"]:
             self.sub_queue.post(message_type, data)
         else:
@@ -255,6 +257,11 @@ class IdeProtocolServer(AbstractIdeProtocolServer):
             asyncio.create_task(
                 session.autopilot.handle_highlighted_code(range_in_files))
 
+    def onMainUserInput(self, input: str):
+        for _, session in self.session_manager.sessions.items():
+            asyncio.create_task(
+                session.autopilot.accept_user_input(input))
+
     # Request information. Session doesn't matter.
     async def getOpenFiles(self) -> List[str]:
         resp = await self._send_and_receive_json({}, OpenFilesResponse, "openFiles")