@@ -317,21 +317,33 @@ async def test_basic_prompts(server_transport: str, server_url: str) -> None:
317317 assert result .serverInfo .name == "Prompt Example"
318318 assert result .capabilities .prompts is not None
319319
320- # Test summarize prompt
320+ # Test review_code prompt
321321 prompts = await session .list_prompts ()
322- summarize_prompt = next ((p for p in prompts .prompts if p .name == "summarize " ), None )
323- assert summarize_prompt is not None
322+ review_prompt = next ((p for p in prompts .prompts if p .name == "review_code " ), None )
323+ assert review_prompt is not None
324324
325- prompt_result = await session .get_prompt ("summarize " , {"text " : "Long text here" , "max_words" : "50 " })
325+ prompt_result = await session .get_prompt ("review_code " , {"code " : "def hello(): \n print('Hello') " })
326326 assert isinstance (prompt_result , GetPromptResult )
327- assert len (prompt_result .messages ) >= 1
328-
329- # Test explain prompt
330- explain_result = await session .get_prompt (
331- "explain" , {"concept" : "machine learning" , "audience" : "beginner" }
327+ assert len (prompt_result .messages ) == 1
328+ assert isinstance (prompt_result .messages [0 ].content , TextContent )
329+ assert "Please review this code:" in prompt_result .messages [0 ].content .text
330+ assert "def hello():" in prompt_result .messages [0 ].content .text
331+
332+ # Test debug_error prompt
333+ debug_result = await session .get_prompt (
334+ "debug_error" , {"error" : "TypeError: 'NoneType' object is not subscriptable" }
332335 )
333- assert isinstance (explain_result , GetPromptResult )
334- assert len (explain_result .messages ) >= 1
336+ assert isinstance (debug_result , GetPromptResult )
337+ assert len (debug_result .messages ) == 3
338+ assert debug_result .messages [0 ].role == "user"
339+ assert isinstance (debug_result .messages [0 ].content , TextContent )
340+ assert "I'm seeing this error:" in debug_result .messages [0 ].content .text
341+ assert debug_result .messages [1 ].role == "user"
342+ assert isinstance (debug_result .messages [1 ].content , TextContent )
343+ assert "TypeError" in debug_result .messages [1 ].content .text
344+ assert debug_result .messages [2 ].role == "assistant"
345+ assert isinstance (debug_result .messages [2 ].content , TextContent )
346+ assert "I'll help debug that" in debug_result .messages [2 ].content .text
335347
336348
337349# Test progress reporting
0 commit comments