1010class TestGraphCreation :
1111 """Test graph creation and configuration."""
1212
13- @patch ('sample_deep_agent.graph.async_create_deep_agent ' )
13+ @patch ('sample_deep_agent.graph.create_deep_agent ' )
1414 @patch ('sample_deep_agent.graph.load_chat_model' )
1515 def test_make_graph_with_default_config (self , mock_load_model , mock_create_deep_agent ):
1616 """Test graph creation with default configuration."""
@@ -25,7 +25,7 @@ def test_make_graph_with_default_config(self, mock_load_model, mock_create_deep_
2525
2626 result = make_graph ()
2727
28- mock_load_model .assert_called_once_with ("siliconflow:zai-org/GLM-4.5 " )
28+ mock_load_model .assert_called_once_with ("siliconflow:deepseek-ai/DeepSeek-V3.2-Exp " )
2929 mock_create_deep_agent .assert_called_once ()
3030
3131 call_args = mock_create_deep_agent .call_args
@@ -40,7 +40,7 @@ def test_make_graph_with_default_config(self, mock_load_model, mock_create_deep_
4040
4141 assert result == mock_agent_with_config
4242
43- @patch ('sample_deep_agent.graph.async_create_deep_agent ' )
43+ @patch ('sample_deep_agent.graph.create_deep_agent ' )
4444 @patch ('sample_deep_agent.graph.load_chat_model' )
4545 def test_make_graph_with_custom_config (self , mock_load_model , mock_create_deep_agent ):
4646 """Test graph creation with custom configuration."""
@@ -67,14 +67,14 @@ def test_make_graph_with_custom_config(self, mock_load_model, mock_create_deep_a
6767
6868 call_args = mock_create_deep_agent .call_args
6969
70- # Verify instructions are present (runtime context will be used during actual execution)
71- instructions = call_args [1 ]['instructions ' ]
72- assert "expert research coordinator" in instructions .lower ()
73- assert "TODOs" in instructions
70+ # Verify system_prompt is present (runtime context will be used during actual execution)
71+ system_prompt = call_args [1 ]['system_prompt ' ]
72+ assert "expert research coordinator" in system_prompt .lower ()
73+ assert "TODOs" in system_prompt
7474
7575 assert result == mock_agent_with_config
7676
77- @patch ('sample_deep_agent.graph.async_create_deep_agent ' )
77+ @patch ('sample_deep_agent.graph.create_deep_agent ' )
7878 @patch ('sample_deep_agent.graph.load_chat_model' )
7979 def test_make_graph_with_none_config (self , mock_load_model , mock_create_deep_agent ):
8080 """Test graph creation with None config."""
@@ -90,12 +90,12 @@ def test_make_graph_with_none_config(self, mock_load_model, mock_create_deep_age
9090 result = make_graph (None )
9191
9292 # Should use default configuration
93- mock_load_model .assert_called_once_with ("siliconflow:zai-org/GLM-4.5 " )
93+ mock_load_model .assert_called_once_with ("siliconflow:deepseek-ai/DeepSeek-V3.2-Exp " )
9494 mock_create_deep_agent .assert_called_once ()
9595
9696 assert result == mock_agent_with_config
9797
98- @patch ('sample_deep_agent.graph.async_create_deep_agent ' )
98+ @patch ('sample_deep_agent.graph.create_deep_agent ' )
9999 @patch ('sample_deep_agent.graph.load_chat_model' )
100100 def test_tools_are_included (self , mock_load_model , mock_create_deep_agent ):
101101 """Test that required tools are included."""
@@ -156,36 +156,36 @@ class TestSubAgentConfiguration:
156156 def test_subagent_configuration_valid (self ):
157157 """Test that sub-agent configuration is valid for deepagents."""
158158 # Verify all required fields are present
159- required_fields = ["name" , "description" , "prompt " ]
159+ required_fields = ["name" , "description" , "system_prompt " ]
160160 for field in required_fields :
161161 assert field in research_sub_agent
162162
163163 # Research agent no longer has explicit tools - they're passed from main agent
164- # Check that the prompt contains TODO constraints instead
164+ # Check that the system_prompt contains TODO constraints instead
165165
166166 # Verify name and description
167167 assert research_sub_agent ["name" ] == "research-agent"
168168 assert "research" in research_sub_agent ["description" ].lower ()
169169
170- # Verify prompt contains expected elements
171- prompt = research_sub_agent ["prompt " ]
172- assert "researcher" in prompt .lower ()
173- assert "TODO CONSTRAINTS" in prompt
174- assert "GLOBAL TODO LIMIT" in prompt
175- assert str (3 ) in prompt # MAX_TODOS value
170+ # Verify system_prompt contains expected elements
171+ system_prompt = research_sub_agent ["system_prompt " ]
172+ assert "researcher" in system_prompt .lower ()
173+ assert "TODO CONSTRAINTS" in system_prompt
174+ assert "GLOBAL TODO LIMIT" in system_prompt
175+ assert str (3 ) in system_prompt # MAX_TODOS value
176176
177177 def test_critique_agent_configuration (self ):
178178 """Test critique agent configuration."""
179179 assert critique_sub_agent ["name" ] == "critique-agent"
180180 assert "critique" in critique_sub_agent ["description" ].lower ()
181- assert "prompt " in critique_sub_agent
181+ assert "system_prompt " in critique_sub_agent
182182
183183 # Critique agent no longer has explicit tools - they're passed from main agent
184- # Check that the prompt contains TODO constraints
185- prompt = critique_sub_agent ["prompt " ]
186- assert "TODO CONSTRAINTS" in prompt
187- assert "GLOBAL TODO LIMIT" in prompt
188- assert str (3 ) in prompt # MAX_TODOS value
184+ # Check that the system_prompt contains TODO constraints
185+ system_prompt = critique_sub_agent ["system_prompt " ]
186+ assert "TODO CONSTRAINTS" in system_prompt
187+ assert "GLOBAL TODO LIMIT" in system_prompt
188+ assert str (3 ) in system_prompt # MAX_TODOS value
189189
190190
191191class TestAppExport :
@@ -228,9 +228,9 @@ def test_tool_integration_in_graph(self):
228228 assert hasattr (deep_web_search , "invoke" ) or callable (deep_web_search )
229229 assert hasattr (think_tool , "invoke" ) or callable (think_tool )
230230
231- @patch ('sample_deep_agent.graph.async_create_deep_agent ' )
231+ @patch ('sample_deep_agent.graph.create_deep_agent ' )
232232 def test_tools_passed_to_deep_agent (self , mock_create_deep_agent ):
233- """Test that tools are properly passed to async_create_deep_agent ."""
233+ """Test that tools are properly passed to create_deep_agent ."""
234234 from sample_deep_agent .graph import make_graph
235235
236236 mock_agent = Mock ()
0 commit comments