Debugging shit
Browse files
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")
|