feat(chat_manager.py): add support for sending file responses

fix(schemas.py): add validation for file response type and data type
test(test_websocket.py): remove data and data_type fields from ChatResponse messages in tests
This commit is contained in:
Gabriel Almeida 2023-04-20 11:09:42 -03:00
commit 5169c0bc27
3 changed files with 77 additions and 16 deletions

View file

@ -32,8 +32,6 @@ def test_chat_history(client: TestClient):
"message": None,
"intermediate_steps": "",
"type": "start",
"data": None,
"data_type": "",
}
# Send another message
payload = {"message": "How are you?"}
@ -46,6 +44,4 @@ def test_chat_history(client: TestClient):
"message": "Hello, I'm a mock response!",
"intermediate_steps": "",
"type": "end",
"data": None,
"data_type": "",
}