lowlevel Server: widen on_* return types for InputRequiredResult; add subscriptions/listen slot (#2967)
This commit is contained in:
parent
a527142312
commit
ae13ede143
1 changed files with 29 additions and 6 deletions
|
|
@ -148,7 +148,7 @@ class Server(Generic[LifespanResultT]):
|
|||
| None = None,
|
||||
on_call_tool: Callable[
|
||||
[ServerRequestContext[LifespanResultT], types.CallToolRequestParams],
|
||||
Awaitable[types.CallToolResult],
|
||||
Awaitable[types.CallToolResult | types.InputRequiredResult],
|
||||
]
|
||||
| None = None,
|
||||
on_list_resources: Callable[
|
||||
|
|
@ -163,7 +163,7 @@ class Server(Generic[LifespanResultT]):
|
|||
| None = None,
|
||||
on_read_resource: Callable[
|
||||
[ServerRequestContext[LifespanResultT], types.ReadResourceRequestParams],
|
||||
Awaitable[types.ReadResourceResult],
|
||||
Awaitable[types.ReadResourceResult | types.InputRequiredResult],
|
||||
]
|
||||
| None = None,
|
||||
on_subscribe_resource: Callable[
|
||||
|
|
@ -176,6 +176,11 @@ class Server(Generic[LifespanResultT]):
|
|||
Awaitable[types.EmptyResult],
|
||||
]
|
||||
| None = None,
|
||||
on_subscriptions_listen: Callable[
|
||||
[ServerRequestContext[LifespanResultT], types.SubscriptionsListenRequestParams],
|
||||
Awaitable[types.EmptyResult],
|
||||
]
|
||||
| None = None,
|
||||
on_list_prompts: Callable[
|
||||
[ServerRequestContext[LifespanResultT], types.PaginatedRequestParams | None],
|
||||
Awaitable[types.ListPromptsResult],
|
||||
|
|
@ -183,7 +188,7 @@ class Server(Generic[LifespanResultT]):
|
|||
| None = None,
|
||||
on_get_prompt: Callable[
|
||||
[ServerRequestContext[LifespanResultT], types.GetPromptRequestParams],
|
||||
Awaitable[types.GetPromptResult],
|
||||
Awaitable[types.GetPromptResult | types.InputRequiredResult],
|
||||
]
|
||||
| None = None,
|
||||
on_completion: Callable[
|
||||
|
|
@ -242,6 +247,7 @@ class Server(Generic[LifespanResultT]):
|
|||
("resources/read", types.ReadResourceRequestParams, on_read_resource),
|
||||
("resources/subscribe", types.SubscribeRequestParams, on_subscribe_resource),
|
||||
("resources/unsubscribe", types.UnsubscribeRequestParams, on_unsubscribe_resource),
|
||||
("subscriptions/listen", types.SubscriptionsListenRequestParams, on_subscriptions_listen),
|
||||
("tools/list", types.PaginatedRequestParams, on_list_tools),
|
||||
("tools/call", types.CallToolRequestParams, on_call_tool),
|
||||
("logging/setLevel", types.SetLevelRequestParams, on_set_logging_level),
|
||||
|
|
|
|||
Loading…
Reference in a new issue