|
@@ -53,7 +53,7 @@ class TestQuestionnaireSchema:
|
|
|
|
|
|
|
|
class TestGraph:
|
|
class TestGraph:
|
|
|
def test_graph_valid_output(self):
|
|
def test_graph_valid_output(self):
|
|
|
- with patch("src.llm.client.get_llm") as mock_get_llm:
|
|
|
|
|
|
|
+ with patch("src.graphs.questionnaire.get_llm") as mock_get_llm:
|
|
|
mock_llm = MagicMock()
|
|
mock_llm = MagicMock()
|
|
|
mock_llm.invoke.return_value = _mock_llm_response()
|
|
mock_llm.invoke.return_value = _mock_llm_response()
|
|
|
mock_get_llm.return_value = mock_llm
|
|
mock_get_llm.return_value = mock_llm
|
|
@@ -71,7 +71,7 @@ class TestGraph:
|
|
|
assert len(result["questionnaire"]["questions"]) == 1
|
|
assert len(result["questionnaire"]["questions"]) == 1
|
|
|
|
|
|
|
|
def test_graph_invalid_llm_response_rejected(self):
|
|
def test_graph_invalid_llm_response_rejected(self):
|
|
|
- with patch("src.llm.client.get_llm") as mock_get_llm:
|
|
|
|
|
|
|
+ with patch("src.graphs.questionnaire.get_llm") as mock_get_llm:
|
|
|
mock_llm = MagicMock()
|
|
mock_llm = MagicMock()
|
|
|
mock_llm.invoke.return_value = _mock_llm_invalid_response()
|
|
mock_llm.invoke.return_value = _mock_llm_invalid_response()
|
|
|
mock_get_llm.return_value = mock_llm
|
|
mock_get_llm.return_value = mock_llm
|