Github/screenshot-to-code
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

when all generations fail, print the all the underlying exceptions for debugging

Browse Source
This commit is contained in:
Abi Raja 2024-09-20 13:56:20 +02:00
parent 9199fee21d
commit 6899c7792e
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
backend/routes/generate_code.py
View File

@ -1,5 +1,6 @@
import asyncio import asyncio
from dataclasses import dataclass from dataclasses import dataclass
import traceback
from fastapi import APIRouter, WebSocket from fastapi import APIRouter, WebSocket
import openai import openai
from codegen.utils import extract_html_content from codegen.utils import extract_html_content
@ -321,6 +322,12 @@ async def stream_code(websocket: WebSocket):
) )
if all_generations_failed: if all_generations_failed:
await throw_error("Error generating code. Please contact support.") await throw_error("Error generating code. Please contact support.")
# Print the all the underlying exceptions for debugging
for completion in completions:
traceback.print_exception(
type(completion), completion, completion.__traceback__
)
raise Exception("All generations failed") raise Exception("All generations failed")
# If some completions failed, replace them with empty strings # If some completions failed, replace them with empty strings