now updating

django_app/redbox_app/redbox_core/admin.py

@@ -143,7 +143,7 @@ def export_as_csv(self, request, queryset: QuerySet):  # noqa:ARG002
         return response
 
     export_as_csv.short_description = "Export Selected"
-    fields = ["name", "user"]
+    fields = ["name", "user", "chat_backend"]
     inlines = [ChatMessageInline]
     list_display = ["name", "user", "created_at"]
     list_filter = ["user"]

django_app/redbox_app/redbox_core/consumers.py

@@ -65,13 +65,13 @@ async def receive(self, text_data=None, bytes_data=None):
         user_message_text: str = data.get("message", "")
         selected_file_uuids: Sequence[UUID] = [UUID(u) for u in data.get("selectedFiles", [])]
         user: User = self.scope.get("user")
-        chat_backend = self.scope.get("llm")
+        chat_backend = data.get("llm")
 
         if session_id := data.get("sessionId"):
             session = await Chat.objects.aget(id=session_id)
-            if chat_backend and session.chat_backend != chat_backend:
-                session.chat_backend = chat_backend
-                await session.asave()
+            logger.info("updating: chat_backend=%s -> ai_settings=%s", session.chat_backend, chat_backend)
+            session.chat_backend = chat_backend
+            await session.asave()
         else:
             session = await Chat.objects.acreate(
                 name=user_message_text[: settings.CHAT_TITLE_LENGTH], user=user, chat_backend=chat_backend

django_app/redbox_app/redbox_core/models.py

@@ -420,7 +420,14 @@ def __str__(self) -> str:  # pragma: no cover
     @override
     def save(self, force_insert=False, force_update=False, using=None, update_fields=None):
         self.name = sanitise_string(self.name)
-        self.chat_backend = self.user.ai_settings.chat_backend
+
+        logger.info(
+            "saving chat with: chat_backend=%s, ai_settings=%s",
+            self.chat_backend,
+            self.user.ai_settings.chat_backend,
+        )
+        if self.chat_backend is None:
+            self.chat_backend = self.user.ai_settings.chat_backend
         super().save(force_insert, force_update, using, update_fields)
 
     @classmethod