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

throw exceptions with stack traces

Browse Source
This commit is contained in:
Abi Raja 2024-07-16 12:00:53 -04:00
parent f7d0dbb7ce
commit b733400e91
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
backend/llm.py
View File

@ -1,4 +1,3 @@
import base64
from enum import Enum from enum import Enum
from typing import Any, Awaitable, Callable, List, cast from typing import Any, Awaitable, Callable, List, cast
from anthropic import AsyncAnthropic from anthropic import AsyncAnthropic
@ -76,7 +75,10 @@ async def stream_openai_response(
finish_reason = chunk.choices[0].finish_reason finish_reason = chunk.choices[0].finish_reason
print("[STOP REASON] OpenAI " + finish_reason) print("[STOP REASON] OpenAI " + finish_reason)
if finish_reason == "length": if finish_reason == "length":
sentry_sdk.capture_exception(Exception("OpenAI response too long")) try:
raise Exception("OpenAI response too long")
except Exception as e:
sentry_sdk.capture_exception()
except Exception as e: except Exception as e:
sentry_sdk.capture_exception(e) sentry_sdk.capture_exception(e)
@ -155,7 +157,10 @@ async def stream_claude_response(
# Log stop reason # Log stop reason
print("[STOP REASON] " + str(response.stop_reason)) print("[STOP REASON] " + str(response.stop_reason))
if response.stop_reason == "max_tokens": if response.stop_reason == "max_tokens":
sentry_sdk.capture_exception(Exception("Claude response too long")) try:
raise Exception("Claude response too long")
except Exception:
sentry_sdk.capture_exception()
# Close the Anthropic client # Close the Anthropic client
await client.close() await client.close()