fix: Expose completions capability (#865)

This commit is contained in:
Frédéric Barthelet 2025-07-07 10:26:22 +02:00 • committed by GitHub
commit 906ceea388
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -190,6 +190,7 @@ class Server(Generic[LifespanResultT, RequestT]):
resources_capability = None
tools_capability = None
logging_capability = None
completions_capability = None
# Set prompt capabilities if handler exists
if types.ListPromptsRequest in self.request_handlers:
@ -209,12 +210,17 @@ class Server(Generic[LifespanResultT, RequestT]):
if types.SetLevelRequest in self.request_handlers:
logging_capability = types.LoggingCapability()
# Set completions capabilities if handler exists
if types.CompleteRequest in self.request_handlers:
completions_capability = types.CompletionsCapability()
return types.ServerCapabilities(
prompts=prompts_capability,
resources=resources_capability,
tools=tools_capability,
logging=logging_capability,
experimental=experimental_capabilities,
completions=completions_capability,
)
@property