File tree 2 files changed +12
-7
lines changed
backend/tests/integration
multitenant_tests/syncing
2 files changed +12
-7
lines changed Original file line number Diff line number Diff line change @@ -75,17 +75,20 @@ def send_message(
75
75
alternate_assistant_id = alternate_assistant_id ,
76
76
use_existing_user_message = use_existing_user_message ,
77
77
)
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
80
85
81
86
response = requests .post (
82
87
f"{ API_SERVER_URL } /chat/send-message" ,
83
88
json = chat_message_req .model_dump (),
84
- headers = user_performing_action .headers
85
- if user_performing_action
86
- else GENERAL_HEADERS ,
89
+ headers = headers ,
87
90
stream = True ,
88
- cookies = user_performing_action . cookies ,
91
+ cookies = cookies ,
89
92
)
90
93
91
94
return ChatSessionManager .analyze_response (response )
Original file line number Diff line number Diff line change
1
+ from typing import Any
2
+
1
3
from onyx .db .models import UserRole
2
4
from tests .integration .common_utils .managers .api_key import APIKeyManager
3
5
from tests .integration .common_utils .managers .cc_pair import CCPairManager
11
13
from tests .integration .common_utils .test_models import DATestUser
12
14
13
15
14
- def setup_test_tenants (reset_multitenant : None ):
16
+ def setup_test_tenants (reset_multitenant : None ) -> dict [ str , Any ] :
15
17
"""Helper function to set up test tenants with documents and users."""
16
18
# Creating an admin user for Tenant 1
17
19
admin_user1 : DATestUser = UserManager .create (
You can’t perform that action at this time.
0 commit comments