sugiv commited on
Commit
8d76def
·
1 Parent(s): 6f11e4e

Debugging shit

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -139,11 +139,14 @@ Here's the complete Python function implementation:
139
  for chunk in llm(full_prompt, stream=True, **generation_kwargs):
140
  token = chunk["choices"][0]["text"]
141
  generated_text += token
 
142
  yield json.dumps({"token": token, "generated_text": generated_text}) + "\n"
143
 
144
  formatted_code = extract_and_format_code(generated_text)
 
145
  yield json.dumps({"complete": True, "formatted_code": formatted_code}) + "\n"
146
 
 
147
  generated_response = StreamingResponse(generate(), media_type="application/x-ndjson")
148
  logger.info(f"Streaming response {generated_response}")
149
  return StreamingResponse(generate(), media_type="application/text")
 
139
  for chunk in llm(full_prompt, stream=True, **generation_kwargs):
140
  token = chunk["choices"][0]["text"]
141
  generated_text += token
142
+ logger.info(f"Generated text: {generated_text}")
143
  yield json.dumps({"token": token, "generated_text": generated_text}) + "\n"
144
 
145
  formatted_code = extract_and_format_code(generated_text)
146
+ logger.info(f"Formatted code: {formatted_code}")
147
  yield json.dumps({"complete": True, "formatted_code": formatted_code}) + "\n"
148
 
149
+
150
  generated_response = StreamingResponse(generate(), media_type="application/x-ndjson")
151
  logger.info(f"Streaming response {generated_response}")
152
  return StreamingResponse(generate(), media_type="application/text")