tightening

2026-06-21 22:12:02 -05:00
parent 3c7f7e61b7
commit 123c491413
29 changed files with 150 additions and 357 deletions
@@ -28,9 +28,13 @@ class TestMcpPromptSurface:
            """Ensures prompts/get resolves a listed prompt into structured message objects."""
            async with mcp_session_factory() as mcp_session:
                listed_prompts = await mcp_session.list_prompts()
-                prompt_name = listed_prompts.prompts[0].name
+                prompt = listed_prompts.prompts[0]
+                arguments = {arg.name: "test" for arg in (prompt.arguments or []) if arg.required}

-                resolved_prompt = await mcp_session.get_prompt(name=prompt_name)
+                resolved_prompt = await mcp_session.get_prompt(
+                    name=prompt.name,
+                    arguments=arguments or None,
+                )

            assert resolved_prompt.messages
            assert all(message.content for message in resolved_prompt.messages)