added temperature to consumer

i-dot-ai · Sep 9, 2024 · 3028767 · 3028767
1 parent 2a5e0f1
commit 3028767
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/django_app/redbox_app/redbox_core/consumers.py b/django_app/redbox_app/redbox_core/consumers.py
@@ -66,11 +66,13 @@ async def receive(self, text_data=None, bytes_data=None):
         selected_file_uuids: Sequence[UUID] = [UUID(u) for u in data.get("selectedFiles", [])]
         user: User = self.scope.get("user")
         chat_backend = data.get("llm")
+        temperature = data.get("temperature")
 
         if session_id := data.get("sessionId"):
             session = await Chat.objects.aget(id=session_id)
             logger.info("updating: chat_backend=%s -> ai_settings=%s", session.chat_backend, chat_backend)
             session.chat_backend = chat_backend
+            session.temperature = temperature
             await session.asave()
         else:
             session = await Chat.objects.acreate(