mememechez commited on
Commit
7d8dfb1
·
1 Parent(s): 69185d3

Fix SSE JSON: avoid backslashes in f-string; use json.dumps

Browse files
Files changed (1) hide show
  1. home/chezy/golem_flask_server.py +3 -1
home/chezy/golem_flask_server.py CHANGED
@@ -2060,7 +2060,9 @@ Answer helpfully:"""
2060
 
2061
  if result and result.get('response'):
2062
  response_text = result.get('direct_response', result['response'])
2063
- yield f"data: {{\"status\": \"complete\", \"response\": \"{response_text.replace(chr(10), '\\\\n').replace(chr(13), '').replace('\"', '\\\\\"')}\", \"model_used\": \"{'gemini' if selected_model == 'gemini' else 'qwen2'}\"}}\n\n"
 
 
2064
  else:
2065
  yield "data: {\"error\": \"Failed to generate response\"}\n\n"
2066
 
 
2060
 
2061
  if result and result.get('response'):
2062
  response_text = result.get('direct_response', result['response'])
2063
+ model_used_value = "gemini" if selected_model == "gemini" else "qwen2"
2064
+ payload = {"status": "complete", "response": response_text, "model_used": model_used_value}
2065
+ yield "data: " + json.dumps(payload) + "\n\n"
2066
  else:
2067
  yield "data: {\"error\": \"Failed to generate response\"}\n\n"
2068