Include context into completions (#966)

This commit is contained in:
Inna Harper 2025-06-17 09:33:25 +01:00 • committed by GitHub
commit a3bcabdce2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -304,8 +304,13 @@ class ClientSession(
self,
ref: types.ResourceTemplateReference | types.PromptReference,
argument: dict[str, str],
context_arguments: dict[str, str] | None = None,
) -> types.CompleteResult:
"""Send a completion/complete request."""
context = None
if context_arguments is not None:
context = types.CompletionContext(arguments=context_arguments)
return await self.send_request(
types.ClientRequest(
types.CompleteRequest(
@ -313,6 +318,7 @@ class ClientSession(
params=types.CompleteRequestParams(
ref=ref,
argument=types.CompletionArgument(**argument),
context=context,
),
)
),