Tim Luka Horstmann commited on
Commit
392cd96
·
1 Parent(s): 8583b57

Updated straeming

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 on every token or small chunk for live streaming
148
- if len(buffer) >= 1: # Yield per character or small chunk
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