From dc72f2e52fd6e41de17d078d2bdfd77ce81056ea Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Tue, 25 Apr 2023 21:06:29 -0300 Subject: [PATCH] refactor(test_websocket.py): change sender field to is_bot field in chat history tests --- tests/test_websocket.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_websocket.py b/tests/test_websocket.py index 74e147075..9ec128bcf 100644 --- a/tests/test_websocket.py +++ b/tests/test_websocket.py @@ -28,7 +28,7 @@ def test_chat_history(client: TestClient): # Receive the response from the server response = websocket.receive_json() assert json.loads(response) == { - "sender": "bot", + "is_bot": True, "message": None, "intermediate_steps": "", "type": "start", @@ -40,7 +40,7 @@ def test_chat_history(client: TestClient): # Receive the response from the server response = websocket.receive_json() assert json.loads(response) == { - "sender": "bot", + "is_bot": True, "message": "Hello, I'm a mock response!", "intermediate_steps": "", "type": "end",