fix: :bug: change for/backwardslash decoding scheme

2 files changed, 16 insertions, 1 deletions

diff --git a/continuedev/src/continuedev/server/gui.py b/continuedev/src/continuedev/server/gui.py
index cc6bc911..26fcbd42 100644
--- a/continuedev/src/continuedev/server/gui.py
+++ b/continuedev/src/continuedev/server/gui.py
@@ -306,6 +306,17 @@ class GUIProtocolServer:
 
     def add_model_for_role(self, role: str, model_class: str, model: Any):
         models = self.session.autopilot.continue_sdk.config.models
+
+        model_copy = model.copy()
+        if "api_key" in model_copy:
+            del model_copy["api_key"]
+        if "hf_token" in model_copy:
+            del model_copy["hf_token"]
+        posthog_logger.capture_event(
+            "select_model_for_role",
+            {"role": role, "model_class": model_class, "model": model_copy},
+        )
+
         if role == "*":
 
             async def async_stuff():
diff --git a/continuedev/src/continuedev/server/ide.py b/continuedev/src/continuedev/server/ide.py
index 579c8f1d..1709503b 100644
--- a/continuedev/src/continuedev/server/ide.py
+++ b/continuedev/src/continuedev/server/ide.py
@@ -602,7 +602,11 @@ async def websocket_endpoint(websocket: WebSocket, session_id: str = None):
 
         # Message handler
         def handle_msg(msg):
-            message = json.loads(msg)
+            try:
+                message = json.loads(msg)
+            except json.JSONDecodeError:
+                logger.critical(f"Error decoding json: {msg}")
+                return
 
             if "messageType" not in message or "data" not in message:
                 return