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

add sentry log for text generation

Browse Source
This commit is contained in:
Abi Raja 2024-07-09 13:17:00 -04:00
parent 7c132fd1bd
commit 8c11366bd0
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
backend/routes/generate_code.py
View File

@ -238,6 +238,7 @@ async def stream_code(websocket: WebSocket):
else: else:
prompt_messages = assemble_prompt(params["image"], valid_stack) prompt_messages = assemble_prompt(params["image"], valid_stack)
elif validated_input_mode == "text": elif validated_input_mode == "text":
sentry_sdk.capture_exception(Exception("Text generation used"))
prompt_messages = assemble_text_prompt(params["image"], valid_stack) prompt_messages = assemble_text_prompt(params["image"], valid_stack)
else: else:
await throw_error("Invalid input mode") await throw_error("Invalid input mode")