Include context into completions (#966)
This commit is contained in:
parent
7b420656de
commit
a3bcabdce2
1 changed files with 363 additions and 7 deletions
|
|
@ -433,6 +433,7 @@ class Server(Generic[LifespanResultT, RequestT]):
|
|||
[
|
||||
types.PromptReference | types.ResourceTemplateReference,
|
||||
types.CompletionArgument,
|
||||
types.CompletionContext | None,
|
||||
],
|
||||
Awaitable[types.Completion | None],
|
||||
],
|
||||
|
|
@ -440,7 +441,7 @@ class Server(Generic[LifespanResultT, RequestT]):
|
|||
logger.debug("Registering handler for CompleteRequest")
|
||||
|
||||
async def handler(req: types.CompleteRequest):
|
||||
completion = await func(req.params.ref, req.params.argument)
|
||||
completion = await func(req.params.ref, req.params.argument, req.params.context)
|
||||
return types.ServerResult(
|
||||
types.CompleteResult(
|
||||
completion=completion
|
||||
|
|
|
|||
Loading…
Reference in a new issue