File tree Expand file tree Collapse file tree 1 file changed +19
-0
lines changed
Expand file tree Collapse file tree 1 file changed +19
-0
lines changed Original file line number Diff line number Diff line change @@ -134,6 +134,25 @@ async def use_async_dep(value: str = Depends(get_async_value)) -> str:
134134 assert result .content [0 ].text == "async_value"
135135
136136
137+ @pytest .mark .anyio
138+ async def test_prompt_with_dependency ():
139+ """Test that prompts can receive dependencies via Depends()."""
140+
141+ def get_greeting () -> str :
142+ return "Hello"
143+
144+ server = MCPServer ("test-server" )
145+
146+ @server .prompt ()
147+ async def greeting_prompt (name : str , greeting : str = Depends (get_greeting )) -> str :
148+ return f"{ greeting } , { name } !"
149+
150+ async with Client (server ) as client :
151+ result = await client .get_prompt ("greeting_prompt" , arguments = {"name" : "World" })
152+ # Check that we got some result (the exact format may vary)
153+ assert len (result .messages ) > 0
154+
155+
137156@pytest .mark .anyio
138157async def test_dependency_caching_per_request ():
139158 """Test that dependencies are cached within a single request."""
You can’t perform that action at this time.
0 commit comments