modelcontextprotocol/python-sdk
Watch
1
0
Fork
You've already forked python-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix completions (#1107)

Browse source
This commit is contained in:
Inna Harper 2025-07-09 11:15:49 +01:00 • committed by GitHub
commit c067ef291e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/server/test_session.py
View file

@ -13,12 +13,13 @@ from mcp.types import (
ClientNotification, ClientNotification,
Completion, Completion,
CompletionArgument, CompletionArgument,
CompletionContext,
CompletionsCapability, CompletionsCapability,
InitializedNotification, InitializedNotification,
PromptReference, PromptReference,
PromptsCapability, PromptsCapability,
ResourceReference,
ResourcesCapability, ResourcesCapability,
ResourceTemplateReference,
ServerCapabilities, ServerCapabilities,
) )
@ -109,7 +110,11 @@ async def test_server_capabilities():
# Add a complete handler # Add a complete handler
@server.completion() @server.completion()
async def complete(ref: PromptReference | ResourceReference, argument: CompletionArgument): async def complete(
ref: PromptReference | ResourceTemplateReference,
argument: CompletionArgument,
context: CompletionContext | None,
) -> Completion | None:
return Completion( return Completion(
values=["completion1", "completion2"], values=["completion1", "completion2"],
) )