From 079df525b5db3666cb58f2e23e322ea55361df4d Mon Sep 17 00:00:00 2001 From: Piero Savastano Date: Wed, 7 Aug 2024 16:59:50 +0200 Subject: [PATCH] fix test --- core/tests/routes/test_websocket.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/tests/routes/test_websocket.py b/core/tests/routes/test_websocket.py index 02d22c7e2..d1b52ce49 100644 --- a/core/tests/routes/test_websocket.py +++ b/core/tests/routes/test_websocket.py @@ -12,7 +12,7 @@ def check_correct_websocket_reply(reply): # why why = reply["why"] - assert {"input", "intermediate_steps", "memory", "model_interactions"} == set(why.keys()) + assert {"input", "intermediate_steps", "memory", "model_interactions", "agent_output"} == set(why.keys()) assert isinstance(why["input"], str) assert isinstance(why["intermediate_steps"], list) assert isinstance(why["memory"], dict)