diff options
Diffstat (limited to 'continuedev/src')
| -rw-r--r-- | continuedev/src/continuedev/core/context.py | 6 | ||||
| -rw-r--r-- | continuedev/src/continuedev/server/gui.py | 6 | 
2 files changed, 12 insertions, 0 deletions
diff --git a/continuedev/src/continuedev/core/context.py b/continuedev/src/continuedev/core/context.py index f83cbd34..b1f68b50 100644 --- a/continuedev/src/continuedev/core/context.py +++ b/continuedev/src/continuedev/core/context.py @@ -10,6 +10,7 @@ from pydantic import BaseModel  from .main import ChatMessage, ContextItem, ContextItemDescription, ContextItemId  from ..server.meilisearch_server import check_meilisearch_running  from ..libs.util.logging import logger +from ..libs.util.telemetry import posthog_logger  SEARCH_INDEX_NAME = "continue_context_items" @@ -199,6 +200,11 @@ class ContextManager:              raise ValueError(                  f"Context provider with title {id.provider_title} not found") +        posthog_logger.capture_event("select_context_item", { +            "provider_title": id.provider_title, +            "item_id": id.item_id, +            "query": query +        })          await self.context_providers[id.provider_title].add_context_item(id, query)      async def delete_context_with_ids(self, ids: List[str]): diff --git a/continuedev/src/continuedev/server/gui.py b/continuedev/src/continuedev/server/gui.py index 4470999a..49d46be3 100644 --- a/continuedev/src/continuedev/server/gui.py +++ b/continuedev/src/continuedev/server/gui.py @@ -139,6 +139,7 @@ class GUIProtocolServer(AbstractGUIProtocolServer):      def on_toggle_adding_highlighted_code(self):          create_async_task(              self.session.autopilot.toggle_adding_highlighted_code(), self.on_error) +        posthog_logger.capture_event("toggle_adding_highlighted_code", {})      def on_set_editing_at_ids(self, ids: List[str]):          create_async_task( @@ -150,6 +151,7 @@ class GUIProtocolServer(AbstractGUIProtocolServer):              ["This is a log of the exact prompt/completion pairs sent/received from the LLM during this step"] + self.session.autopilot.continue_sdk.history.timeline[index].logs)          create_async_task(              self.session.autopilot.ide.showVirtualFile(name, logs), self.on_error) +        posthog_logger.capture_event("show_logs_at_index", {})      def select_context_item(self, id: str, query: str):          """Called when user selects an item from the dropdown""" @@ -164,6 +166,10 @@ class GUIProtocolServer(AbstractGUIProtocolServer):          create_async_task(              load_and_tell_to_reconnect(), self.on_error) +        posthog_logger.capture_event("load_session", { +            "session_id": session_id +        }) +  @router.websocket("/ws")  async def websocket_endpoint(websocket: WebSocket, session: Session = Depends(websocket_session)):  | 
