From 28ad01a51aa28e0d03f5f4dccddebe926426f514 Mon Sep 17 00:00:00 2001 From: hagen-danswer Date: Thu, 17 Oct 2024 12:37:34 -0700 Subject: [PATCH] py --- .../integration/common_utils/managers/document_search.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/backend/tests/integration/common_utils/managers/document_search.py b/backend/tests/integration/common_utils/managers/document_search.py index 6ac012e4d..4fe2442b6 100644 --- a/backend/tests/integration/common_utils/managers/document_search.py +++ b/backend/tests/integration/common_utils/managers/document_search.py @@ -6,6 +6,7 @@ from danswer.search.models import RetrievalDetails from danswer.search.models import SavedSearchDocWithContent from ee.danswer.server.query_and_chat.models import DocumentSearchRequest from tests.integration.common_utils.constants import API_SERVER_URL +from tests.integration.common_utils.constants import GENERAL_HEADERS from tests.integration.common_utils.test_models import DATestUser @@ -25,7 +26,9 @@ class DocumentSearchManager: result = requests.post( url=f"{API_SERVER_URL}/query/document-search", json=search_request.model_dump(), - headers=user_performing_action.headers, + headers=user_performing_action.headers + if user_performing_action + else GENERAL_HEADERS, ) result.raise_for_status() result_json = result.json()