Skip to content

Commit d303913

Browse files
committed
finalize`
1 parent 5e82115 commit d303913

File tree

2 files changed

+12
-7
lines changed

2 files changed

+12
-7
lines changed

Diff for: backend/tests/integration/common_utils/managers/chat.py

+9-6
Original file line numberDiff line numberDiff line change
@@ -75,17 +75,20 @@ def send_message(
7575
alternate_assistant_id=alternate_assistant_id,
7676
use_existing_user_message=use_existing_user_message,
7777
)
78-
print(user_performing_action.headers)
79-
print(user_performing_action.cookies)
78+
79+
headers = (
80+
user_performing_action.headers
81+
if user_performing_action
82+
else GENERAL_HEADERS
83+
)
84+
cookies = user_performing_action.cookies if user_performing_action else None
8085

8186
response = requests.post(
8287
f"{API_SERVER_URL}/chat/send-message",
8388
json=chat_message_req.model_dump(),
84-
headers=user_performing_action.headers
85-
if user_performing_action
86-
else GENERAL_HEADERS,
89+
headers=headers,
8790
stream=True,
88-
cookies=user_performing_action.cookies,
91+
cookies=cookies,
8992
)
9093

9194
return ChatSessionManager.analyze_response(response)

Diff for: backend/tests/integration/multitenant_tests/syncing/test_search_permissions.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
from typing import Any
2+
13
from onyx.db.models import UserRole
24
from tests.integration.common_utils.managers.api_key import APIKeyManager
35
from tests.integration.common_utils.managers.cc_pair import CCPairManager
@@ -11,7 +13,7 @@
1113
from tests.integration.common_utils.test_models import DATestUser
1214

1315

14-
def setup_test_tenants(reset_multitenant: None):
16+
def setup_test_tenants(reset_multitenant: None) -> dict[str, Any]:
1517
"""Helper function to set up test tenants with documents and users."""
1618
# Creating an admin user for Tenant 1
1719
admin_user1: DATestUser = UserManager.create(

0 commit comments

Comments
 (0)