Tim Luka Horstmann
commited on
Commit
·
392cd96
1
Parent(s):
8583b57
Updated straeming
Browse files
app.py
CHANGED
@@ -144,11 +144,11 @@ def stream_response(query):
|
|
144 |
if not first_token_logged and time.time() - start_time > 0:
|
145 |
logger.info(f"First token time: {time.time() - start_time:.2f}s")
|
146 |
first_token_logged = True
|
147 |
-
# Yield
|
148 |
-
if
|
149 |
yield f"data: {buffer}\n\n"
|
150 |
buffer = ""
|
151 |
-
if buffer:
|
152 |
yield f"data: {buffer}\n\n"
|
153 |
yield "data: [DONE]\n\n"
|
154 |
|
|
|
144 |
if not first_token_logged and time.time() - start_time > 0:
|
145 |
logger.info(f"First token time: {time.time() - start_time:.2f}s")
|
146 |
first_token_logged = True
|
147 |
+
# Yield when buffer contains a word boundary (space, punctuation, or reasonable length)
|
148 |
+
if any(buffer.endswith(char) for char in [" ", ".", ",", "!", "?"]) or len(buffer) > 20:
|
149 |
yield f"data: {buffer}\n\n"
|
150 |
buffer = ""
|
151 |
+
if buffer: # Flush remaining buffer
|
152 |
yield f"data: {buffer}\n\n"
|
153 |
yield "data: [DONE]\n\n"
|
154 |
|