mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-03-29 03:01:59 +01:00
f6a75c86c6
* print the test name when it runs * type hints * can't reuse session after an exception * better logging --------- Co-authored-by: Richard Kuo (Danswer) <rkuo@onyx.app>
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
from sqlalchemy.exc import IntegrityError
|
|
|
|
from onyx.db.background_error import create_background_error
|
|
from onyx.db.engine import get_session_with_current_tenant
|
|
|
|
|
|
def emit_background_error(
|
|
message: str,
|
|
cc_pair_id: int | None = None,
|
|
) -> None:
|
|
"""Currently just saves a row in the background_errors table.
|
|
|
|
In the future, could create notifications based on the severity."""
|
|
error_message = ""
|
|
|
|
# try to write to the db, but handle IntegrityError specifically
|
|
try:
|
|
with get_session_with_current_tenant() as db_session:
|
|
create_background_error(db_session, message, cc_pair_id)
|
|
except IntegrityError as e:
|
|
# Log an error if the cc_pair_id was deleted or any other exception occurs
|
|
error_message = (
|
|
f"Failed to create background error: {str(e)}. Original message: {message}"
|
|
)
|
|
except Exception:
|
|
pass
|
|
|
|
if not error_message:
|
|
return
|
|
|
|
# if we get here from an IntegrityError, try to write the error message to the db
|
|
# we need a new session because the first session is now invalid
|
|
try:
|
|
with get_session_with_current_tenant() as db_session:
|
|
create_background_error(db_session, error_message, None)
|
|
except Exception:
|
|
pass
|