Termination of event stream works

chriscarrollsmith · chriscarrollsmith · commit 35bd4ff67ae8 · 2025-02-04T08:20:10.000-05:00
diff --git a/routers/messages.py b/routers/messages.py
@@ -56,12 +56,16 @@ async def event_generator():
             assistant_id=assistant_id,
             thread_id=thread_id
         )
+
         async with stream as stream_manager:
             async for text in stream_manager.text_deltas:
+                logger.info(text)
                 yield f"data: {text}\n\n"
             
+            logger.info("Sending end message")
+
             # Send a done event when the stream is complete
-            yield f"event: EndMessage\n\n"
+            yield "event: EndMessage\ndata: DONE\n\n"
     
     return StreamingResponse(
         event_generator(),
diff --git a/static/styles.css b/static/styles.css
@@ -158,6 +158,10 @@ pre {
   height: 100%;
 }
 
+.assistantMessage {
+  white-space: pre-wrap;
+}
+
 .fileViewer {
   display: flex;
   flex-direction: column;
diff --git a/templates/layout.html b/templates/layout.html
@@ -7,8 +7,8 @@
     <link rel="icon" href="{{ url_for('static', path='openai.svg') }}">
     <link rel="stylesheet" href="{{ url_for('static', path='styles.css') }}">
     <link rel="favicon" href="{{ url_for('static', path='favicon.png') }}">
-    <script src="https://unpkg.com/htmx.org@1.9.10"></script>
-    <script src="https://unpkg.com/htmx.org/dist/ext/sse.js"></script>
+    <script src="https://unpkg.com/htmx.org@2.0.4" integrity="sha384-HGfztofotfshcF7+8n44JQL2oJmowVChPTg48S+jvZoztPfvwD79OC/LTtG6dMp+" crossorigin="anonymous"></script>
+    <script src="https://unpkg.com/htmx-ext-sse@2.2.2/sse.js"></script>
   </head>
   <body>
     {% block content %}
