fix test

cheshire-cat-ai · Aug 7, 2024 · 079df52 · 079df52
1 parent 4df4bee
commit 079df52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)