Skip to content

Commit c6bc4a3

Browse files
committed
fix(tests): align mock Prompt attributes with Decompose.py usage
1 parent 2c4d16e commit c6bc4a3

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/unit/test_decompose.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,8 @@ class TestDecompose:
1010
def decomposer(self, mock_llm_client):
1111
# We also need to mock the prompt handler since it loads files
1212
mock_prompt = MagicMock(spec=BasePrompt)
13-
mock_prompt.decompose_prompt = "Mock Prompt Template {doc}"
13+
mock_prompt.sag_prompt = "Mock SAG Prompt {doc}"
14+
mock_prompt.restore_prompt = "Mock Restore {doc} {claims}"
1415

1516
return Decompose(llm_client=mock_llm_client, prompt=mock_prompt)
1617

0 commit comments

Comments
 (0)