Update test to use FIND_PROMPT

Author: daavoo

tests/unit/test_workflow.py

@@ -1,11 +1,11 @@
-from structured_qa.workflow import find_retrieve_answer
+from structured_qa.workflow import find_retrieve_answer, FIND_PROMPT
 
 
 def test_find_retrieve_answer_multi_sections(tmp_path, mocker):
     model = mocker.MagicMock()
 
     def side_effect(messages):
-        if "You are given two pieces" in messages[0]["content"]:
+        if FIND_PROMPT[:10] in messages[0]["content"]:
             if "section_1" in messages[0]["content"]:
                 return {"choices": [{"message": {"content": "section_1"}}]}
             else:
@@ -35,7 +35,7 @@ def test_find_retrieve_answer_unkown_section(tmp_path, mocker):
     model = mocker.MagicMock()
 
     def side_effect(messages):
-        if "Given an input question" in messages[0]["content"]:
+        if FIND_PROMPT[:10] in messages[0]["content"]:
             return {"choices": [{"message": {"content": "section_x"}}]}
 
     model.create_chat_completion.side_effect = side_effect