Update test

daavoo · daavoo · commit 1ba3420a382d · 2025-01-13T15:05:21.000+01:00
diff --git a/tests/unit/test_workflow.py b/tests/unit/test_workflow.py
@@ -5,7 +5,7 @@ def test_find_retrieve_answer_multi_sections(tmp_path, mocker):
     model = mocker.MagicMock()
 
     def side_effect(messages):
-        if "Given an input question" in messages[0]["content"]:
+        if "You are given two pieces" in messages[0]["content"]:
             if "section_1" in messages[0]["content"]:
                 return {"choices": [{"message": {"content": "section_1"}}]}
             else:
