diff options
Diffstat (limited to 'continuedev')
| -rw-r--r-- | continuedev/src/continuedev/server/ide.py | 9 | ||||
| -rw-r--r-- | continuedev/src/continuedev/server/ide_protocol.py | 2 | 
2 files changed, 8 insertions, 3 deletions
| diff --git a/continuedev/src/continuedev/server/ide.py b/continuedev/src/continuedev/server/ide.py index e1f19447..1f790991 100644 --- a/continuedev/src/continuedev/server/ide.py +++ b/continuedev/src/continuedev/server/ide.py @@ -7,6 +7,7 @@ import uuid  from fastapi import WebSocket, Body, APIRouter  from uvicorn.main import Server +from ..libs.util.telemetry import capture_event  from ..libs.util.queue import AsyncSubscriptionQueue  from ..models.filesystem import FileSystem, RangeInFile, EditDiff, RealFileSystem  from ..models.filesystem_edit import AddDirectory, AddFile, DeleteDirectory, DeleteFile, FileSystemEdit, FileEdit, FileEditWithFullContents, RenameDirectory, RenameFile, SequentialFileSystemEdit @@ -142,6 +143,8 @@ class IdeProtocolServer(AbstractIdeProtocolServer):          elif message_type == "commandOutput":              output = data["output"]              self.onCommandOutput(output) +        elif message_type == "acceptRejectSuggestion": +            self.onAcceptRejectSuggestion(data["accepted"])          elif message_type in ["highlightedCode", "openFiles", "readFile", "editFile", "workspaceDirectory", "getUserSecret", "runCommand", "uniqueId"]:              self.sub_queue.post(message_type, data)          else: @@ -202,8 +205,10 @@ class IdeProtocolServer(AbstractIdeProtocolServer):      # This is where you might have triggers: plugins can subscribe to certian events      # like file changes, tracebacks, etc... -    def onAcceptRejectSuggestion(self, suggestionId: str, accepted: bool): -        pass +    def onAcceptRejectSuggestion(self, accepted: bool): +        capture_event(self.unique_id, "accept_reject_suggestion", { +            "accepted": accepted +        })      def onFileSystemUpdate(self, update: FileSystemEdit):          # Access to Autopilot (so SessionManager) diff --git a/continuedev/src/continuedev/server/ide_protocol.py b/continuedev/src/continuedev/server/ide_protocol.py index d2dafa9a..7faf5563 100644 --- a/continuedev/src/continuedev/server/ide_protocol.py +++ b/continuedev/src/continuedev/server/ide_protocol.py @@ -36,7 +36,7 @@ class AbstractIdeProtocolServer(ABC):          """Show suggestions to the user and wait for a response"""      @abstractmethod -    def onAcceptRejectSuggestion(self, suggestionId: str, accepted: bool): +    def onAcceptRejectSuggestion(self, accepted: bool):          """Called when the user accepts or rejects a suggestion"""      @abstractmethod | 
