fix chatpromptbuilder test

deepset-ai · Dec 17, 2024 · b88daae · b88daae
1 parent 328cebd
commit b88daae
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/test/components/builders/test_chat_prompt_builder.py b/test/components/builders/test_chat_prompt_builder.py
@@ -531,8 +531,13 @@ def test_to_dict(self):
             "type": "haystack.components.builders.chat_prompt_builder.ChatPromptBuilder",
             "init_parameters": {
                 "template": [
-                    {"_content": [{"text": "text and {var}"}], "_role": "user", "_meta": {}},
-                    {"_content": [{"text": "content {required_var}"}], "_role": "assistant", "_meta": {}},
+                    {"_content": [{"text": "text and {var}"}], "_role": "user", "_meta": {}, "_name": None},
+                    {
+                        "_content": [{"text": "content {required_var}"}],
+                        "_role": "assistant",
+                        "_meta": {},
+                        "_name": None,
+                    },
                 ],
                 "variables": ["var", "required_var"],
                 "required_variables": ["required_var"],
@@ -545,8 +550,13 @@ def test_from_dict(self):
                 "type": "haystack.components.builders.chat_prompt_builder.ChatPromptBuilder",
                 "init_parameters": {
                     "template": [
-                        {"_content": [{"text": "text and {var}"}], "_role": "user", "_meta": {}},
-                        {"_content": [{"text": "content {required_var}"}], "_role": "assistant", "_meta": {}},
+                        {"_content": [{"text": "text and {var}"}], "_role": "user", "_meta": {}, "_name": None},
+                        {
+                            "_content": [{"text": "content {required_var}"}],
+                            "_role": "assistant",
+                            "_meta": {},
+                            "_name": None,
+                        },
                     ],
                     "variables": ["var", "required_var"],
                     "required_variables": ["required_var"],