feat: :loud_sound: light telemetry or context providers

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)):