File tree Expand file tree Collapse file tree 2 files changed +16
-0
lines changed Expand file tree Collapse file tree 2 files changed +16
-0
lines changed Original file line number Diff line number Diff line change @@ -186,6 +186,7 @@ def get_capabilities(
186
186
resources_capability = None
187
187
tools_capability = None
188
188
logging_capability = None
189
+ completion_capability = None
189
190
190
191
# Set prompt capabilities if handler exists
191
192
if types .ListPromptsRequest in self .request_handlers :
@@ -209,12 +210,17 @@ def get_capabilities(
209
210
if types .SetLevelRequest in self .request_handlers :
210
211
logging_capability = types .LoggingCapability ()
211
212
213
+ # Set completions capabilities if handler exists
214
+ if types .CompleteRequest in self .request_handlers :
215
+ completion_capability = types .CompletionsCapability ()
216
+
212
217
return types .ServerCapabilities (
213
218
prompts = prompts_capability ,
214
219
resources = resources_capability ,
215
220
tools = tools_capability ,
216
221
logging = logging_capability ,
217
222
experimental = experimental_capabilities ,
223
+ completions = completion_capability ,
218
224
)
219
225
220
226
@property
Original file line number Diff line number Diff line change @@ -267,6 +267,12 @@ class LoggingCapability(BaseModel):
267
267
model_config = ConfigDict (extra = "allow" )
268
268
269
269
270
+ class CompletionsCapability (BaseModel ):
271
+ """Capability for completions operations."""
272
+
273
+ model_config = ConfigDict (extra = "allow" )
274
+
275
+
270
276
class ServerCapabilities (BaseModel ):
271
277
"""Capabilities that a server may support."""
272
278
@@ -280,6 +286,10 @@ class ServerCapabilities(BaseModel):
280
286
"""Present if the server offers any resources to read."""
281
287
tools : ToolsCapability | None = None
282
288
"""Present if the server offers any tools to call."""
289
+ completions : CompletionsCapability | None = None
290
+ """
291
+ Present if the server offers autocompletion suggestions for prompts and resources.
292
+ """
283
293
model_config = ConfigDict (extra = "allow" )
284
294
285
295
You can’t perform that action at this time.
0 commit comments