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,
Completion,
CompletionArgument,
CompletionContext,
CompletionsCapability,
InitializedNotification,
PromptReference,
PromptsCapability,
ResourceReference,
ResourcesCapability,
ResourceTemplateReference,
ServerCapabilities,
)
@ -109,7 +110,11 @@ async def test_server_capabilities():
# Add a complete handler
@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(
values=["completion1", "completion2"],
)