From f9733f987002e6407fe57085fca258c3cd03fa31 Mon Sep 17 00:00:00 2001 From: Weves Date: Sun, 18 Feb 2024 15:13:36 -0800 Subject: [PATCH] Handle missing ri:userkey gracefully in Confluence connector --- backend/danswer/connectors/confluence/connector.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/backend/danswer/connectors/confluence/connector.py b/backend/danswer/connectors/confluence/connector.py index f2d091e4d..a11670bbe 100644 --- a/backend/danswer/connectors/confluence/connector.py +++ b/backend/danswer/connectors/confluence/connector.py @@ -127,8 +127,13 @@ def parse_html_page(text: str, confluence_client: Confluence) -> str: user_id = ( user.attrs["ri:account-id"] if "ri:account-id" in user.attrs - else user.attrs["ri:userkey"] + else user.get("ri:userkey") ) + if not user_id: + logger.warning( + "ri:userkey not found in ri:user element. " f"Found attrs: {user.attrs}" + ) + continue # Include @ sign for tagging, more clear for LLM user.replaceWith("@" + _get_user(user_id, confluence_client)) return format_document_soup(soup)