fix

i-dot-ai · Oct 15, 2024 · 2cd9f53 · 2cd9f53
1 parent c2ce0ca
commit 2cd9f53
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/django_app/redbox_app/redbox_core/views/chat_views.py b/django_app/redbox_app/redbox_core/views/chat_views.py
@@ -34,13 +34,17 @@ def get(self, request: HttpRequest, chat_id: uuid.UUID | None = None) -> HttpRes
                 return redirect(reverse("chats"))
             messages = ChatMessage.get_messages_ordered_by_citation_priority(chat_id)
 
+        logger.info("SERVER_NAME=%s", request.META["SERVER_NAME"])
         logger.info("HOST=%s", request.get_host())
 
-        host, port = request.get_host().split(":")
+        host = request.get_host()
+        if ":" in host:
+            host = host.split(":")[0]
+
         endpoint = URL.build(
             scheme=settings.WEBSOCKET_SCHEME,
             host=host,
-            port=int(port),
+            port=int(request.META["SERVER_PORT"]),
             path=r"/ws/chat/",
         )
         logger.info("endpoint=%s", endpoint)