From a42f2cd3ed633bee2f03ea4195b1b89a5ba15cbf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andreas=20K=C3=B6pf?= Date: Mon, 16 Jan 2023 08:46:45 +0100 Subject: [PATCH] add new debug option 'DEBUG_DATABASE_ECHO' (default False) --- backend/oasst_backend/config.py | 1 + backend/oasst_backend/database.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/backend/oasst_backend/config.py b/backend/oasst_backend/config.py index b0c6b7f5..2a8a3a7c 100644 --- a/backend/oasst_backend/config.py +++ b/backend/oasst_backend/config.py @@ -79,6 +79,7 @@ class Settings(BaseSettings): DEBUG_ALLOW_SELF_LABELING: bool = False # allow users to label their own messages DEBUG_SKIP_EMBEDDING_COMPUTATION: bool = False DEBUG_SKIP_TOXICITY_CALCULATION: bool = False + DEBUG_DATABASE_ECHO: bool = False HUGGING_FACE_API_KEY: str = "" diff --git a/backend/oasst_backend/database.py b/backend/oasst_backend/database.py index 9632b9da..1d0e19b2 100644 --- a/backend/oasst_backend/database.py +++ b/backend/oasst_backend/database.py @@ -5,4 +5,4 @@ from sqlmodel import create_engine if settings.DATABASE_URI is None: raise OasstError("DATABASE_URI is not set", error_code=OasstErrorCode.DATABASE_URI_NOT_SET) -engine = create_engine(settings.DATABASE_URI, echo=True, isolation_level="REPEATABLE READ") +engine = create_engine(settings.DATABASE_URI, echo=settings.DEBUG_DATABASE_ECHO, isolation_level="REPEATABLE READ")