Sfoglia il codice sorgente

fix(langgraph): GraphState 补 _prompt 字段 + 测试 mock 目标修正为模块内引用

iwt 1 mese fa
parent
commit
f001f5e470

+ 1 - 0
cfc-langgraph/src/graphs/questionnaire.py

@@ -10,6 +10,7 @@ from ..schemas.questionnaire import Questionnaire, GenerateRequest
 
 class GraphState(TypedDict):
     request: GenerateRequest
+    _prompt: Optional[str]
     raw_response: str
     questionnaire: Optional[dict]
     error: Optional[str]

+ 2 - 2
cfc-langgraph/tests/test_graph.py

@@ -53,7 +53,7 @@ class TestQuestionnaireSchema:
 
 class TestGraph:
     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.invoke.return_value = _mock_llm_response()
             mock_get_llm.return_value = mock_llm
@@ -71,7 +71,7 @@ class TestGraph:
             assert len(result["questionnaire"]["questions"]) == 1
 
     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.invoke.return_value = _mock_llm_invalid_response()
             mock_get_llm.return_value = mock_llm