highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-09-19 12:03:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

Logging Level Update (#2165)

Browse Source
This commit is contained in:
Yuhong Sun
2024-08-18 21:53:40 -07:00
committed by GitHub
parent 119aefba88
commit 5ab4d94d94
57 changed files with 183 additions and 201 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/ee/danswer/auth/users.py
View File

@@ -19,7 +19,7 @@ logger = setup_logger()
def verify_auth_setting() -> None:
# All the Auth flows are valid for EE version
logger.info(f"Using Auth Type: {AUTH_TYPE.value}")
logger.notice(f"Using Auth Type: {AUTH_TYPE.value}")
async def optional_user_(

2
backend/ee/danswer/background/permission_sync.py
View File

@@ -215,7 +215,7 @@ def permission_loop(delay: int = 60, num_workers: int = NUM_PERMISSION_WORKERS)
def update__main() -> None:
logger.info("Starting Permission Syncing Loop")
logger.notice("Starting Permission Syncing Loop")
init_sqlalchemy_engine(POSTGRES_PERMISSIONS_APP_NAME)
permission_loop()

4
backend/ee/danswer/server/enterprise_settings/store.py
View File

@@ -85,7 +85,7 @@ def upload_logo(
content: IO[Any]
if isinstance(file, str):
logger.info(f"Uploading logo from local path {file}")
logger.notice(f"Uploading logo from local path {file}")
if not os.path.isfile(file) or not is_valid_file_type(file):
logger.error(
"Invalid file type- only .png, .jpg, and .jpeg files are allowed"
@@ -99,7 +99,7 @@ def upload_logo(
file_type = guess_file_type(file)
else:
logger.info("Uploading logo from uploaded file")
logger.notice("Uploading logo from uploaded file")
if not file.filename or not is_valid_file_type(file.filename):
raise HTTPException(
status_code=400,

4
backend/ee/danswer/server/query_and_chat/chat_backend.py
View File

@@ -72,7 +72,7 @@ def handle_simplified_chat_message(
db_session: Session = Depends(get_session),
) -> ChatBasicResponse:
"""This is a Non-Streaming version that only gives back a minimal set of information"""
logger.info(f"Received new simple api chat message: {chat_message_req.message}")
logger.notice(f"Received new simple api chat message: {chat_message_req.message}")
if not chat_message_req.message:
raise HTTPException(status_code=400, detail="Empty chat message is invalid")
@@ -170,7 +170,7 @@ def handle_send_message_simple_with_history(
query = req.messages[-1].message
msg_history = req.messages[:-1]
logger.info(f"Received new simple with history chat message: {query}")
logger.notice(f"Received new simple with history chat message: {query}")
user_id = user.id if user is not None else None
chat_session = create_chat_session(

4
backend/ee/danswer/server/query_and_chat/query_backend.py
View File

@@ -51,7 +51,7 @@ def handle_search_request(
) -> DocumentSearchResponse:
"""Simple search endpoint, does not create a new message or records in the DB"""
query = search_request.message
logger.info(f"Received document search query: {query}")
logger.notice(f"Received document search query: {query}")
llm, fast_llm = get_default_llms()
search_pipeline = SearchPipeline(
@@ -130,7 +130,7 @@ def get_answer_with_quote(
db_session: Session = Depends(get_session),
) -> OneShotQAResponse:
query = query_request.messages[0].message
logger.info(f"Received query for one shot answer API with quotes: {query}")
logger.notice(f"Received query for one shot answer API with quotes: {query}")
persona = get_persona_by_id(
persona_id=query_request.persona_id,

2
backend/ee/danswer/server/saml.py
View File

@@ -50,7 +50,7 @@ async def upsert_saml_user(email: str) -> User:
try:
return await user_manager.get_by_email(email)
except exceptions.UserNotExists:
logger.info("Creating user from SAML login")
logger.notice("Creating user from SAML login")
user_count = await get_user_count()
role = UserRole.ADMIN if user_count == 0 else UserRole.BASIC

16
backend/ee/danswer/server/seeding.py
View File

@@ -49,7 +49,7 @@ def _seed_llms(
db_session: Session, llm_upsert_requests: list[LLMProviderUpsertRequest]
) -> None:
if llm_upsert_requests:
logger.info("Seeding LLMs")
logger.notice("Seeding LLMs")
seeded_providers = [
upsert_llm_provider(db_session, llm_upsert_request)
for llm_upsert_request in llm_upsert_requests
@@ -59,7 +59,7 @@ def _seed_llms(
def _seed_personas(db_session: Session, personas: list[CreatePersonaRequest]) -> None:
if personas:
logger.info("Seeding Personas")
logger.notice("Seeding Personas")
for persona in personas:
upsert_persona(
user=None, # Seeding is done as admin
@@ -83,31 +83,31 @@ def _seed_personas(db_session: Session, personas: list[CreatePersonaRequest]) ->
def _seed_settings(settings: Settings) -> None:
logger.info("Seeding Settings")
logger.notice("Seeding Settings")
try:
settings.check_validity()
store_base_settings(settings)
logger.info("Successfully seeded Settings")
logger.notice("Successfully seeded Settings")
except ValueError as e:
logger.error(f"Failed to seed Settings: {str(e)}")
def _seed_enterprise_settings(seed_config: SeedConfiguration) -> None:
if seed_config.enterprise_settings is not None:
logger.info("Seeding enterprise settings")
logger.notice("Seeding enterprise settings")
store_ee_settings(seed_config.enterprise_settings)
def _seed_logo(db_session: Session, logo_path: str | None) -> None:
if logo_path:
logger.info("Uploading logo")
logger.notice("Uploading logo")
upload_logo(db_session=db_session, file=logo_path)
def _seed_analytics_script(seed_config: SeedConfiguration) -> None:
custom_analytics_secret_key = os.environ.get("CUSTOM_ANALYTICS_SECRET_KEY")
if seed_config.analytics_script_path and custom_analytics_secret_key:
logger.info("Seeding analytics script")
logger.notice("Seeding analytics script")
try:
with open(seed_config.analytics_script_path, "r") as file:
script_content = file.read()
@@ -130,7 +130,7 @@ def get_seed_config() -> SeedConfiguration | None:
def seed_db() -> None:
seed_config = _parse_env()
if seed_config is None:
logger.info("No seeding configuration file passed")
logger.debug("No seeding configuration file passed")
return
with get_session_context_manager() as db_session: