diff --git a/integrations/github/tests/test_file_editor_tool.py b/integrations/github/tests/test_file_editor_tool.py index 4b7a98ce1d..eeb0e49aef 100644 --- a/integrations/github/tests/test_file_editor_tool.py +++ b/integrations/github/tests/test_file_editor_tool.py @@ -204,10 +204,13 @@ def test_pipeline_serialization(self, monkeypatch): "connections": [], "connection_type_validation": True, } - if "confirmation_strategies" in pipeline_dict["components"]["agent"]["init_parameters"]: - expected_dict["components"]["agent"]["init_parameters"]["confirmation_strategies"] = pipeline_dict[ - "components" - ]["agent"]["init_parameters"]["confirmation_strategies"] + + # Compatibility with newer versions of Haystack that include these parameters + for key in ["confirmation_strategies", "required_variables", "user_prompt"]: + if key in pipeline_dict["components"]["agent"]["init_parameters"]: + expected_dict["components"]["agent"]["init_parameters"][key] = pipeline_dict["components"]["agent"][ + "init_parameters" + ][key] assert pipeline_dict == expected_dict