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

@ -1028,11 +1028,21 @@ class CompletionArgument(BaseModel):
model_config = ConfigDict(extra="allow")
class CompletionContext(BaseModel):
"""Additional, optional context for completions."""
arguments: dict[str, str] | None = None
"""Previously-resolved variables in a URI template or prompt."""
model_config = ConfigDict(extra="allow")
class CompleteRequestParams(RequestParams):
"""Parameters for completion requests."""
ref: ResourceTemplateReference | PromptReference
argument: CompletionArgument
context: CompletionContext | None = None
"""Additional, optional context for completions"""
model_config = ConfigDict(extra="allow")