11 KiB
Support for MCP server features
%toc
Prompts
MCP servers can provide LLM prompt templates (called simply prompts) to clients. Every prompt has a required name which identifies it, and a set of named arguments, which are strings.
Client-side: To list the server's prompts, use the
ClientSession.Prompts
iterator, or the lower-level
ClientSession.ListPrompts
(see pagination below). Set
ClientOptions.PromptListChangedHandler
to be notified of changes in the list of prompts.
Call
ClientSession.GetPrompt
to retrieve a prompt by name, providing arguments for expansion.
Server-side: Use
Server.AddPrompt
to add a prompt to the server along with its handler.
The server will have the prompts capability if any prompt is added before the
server is connected to a client, or if
ServerOptions.HasPrompts
is explicitly set. When a prompt is added, any clients already connected to the
server will be notified via a notifications/prompts/list_changed
notification.
%include ../../mcp/server_example_test.go prompts -
Resources
In MCP terms, a resource is some data referenced by a URI. MCP servers can serve resources to clients. They can register resources individually, or register a resource template that uses a URI pattern to describe a collection of resources.
Client-side:
Call ClientSession.ReadResource
to read a resource.
The SDK ensures that a read succeeds only if the URI matches a registered resource exactly,
or matches the URI pattern of a resource template.
To list a server's resources and resource templates, use the
ClientSession.Resources
and
ClientSession.ResourceTemplates
iterators, or the lower-level ListXXX calls (see pagination).
Set
ClientOptions.ResourceListChangedHandler
to be notified of changes in the lists of resources or resource templates.
Clients can be notified when the contents of a resource changes by subscribing to the resource's URI.
Call
ClientSession.Subscribe
to subscribe to a resource
and
ClientSession.Unsubscribe
to unsubscribe.
Set
ClientOptions.ResourceUpdatedHandler
to be notified of changes to subscribed resources.
Server-side:
Use
Server.AddResource
or
Server.AddResourceTemplate
to add a resource or resource template to the server along with its handler.
A
ResourceHandler
maps a URI to the contents of a resource, which can include text, binary data,
or both.
If AddResource or AddResourceTemplate is called before a server is connected, the server will have the
resources capability.
The server will have the resources capability if any resource or resource template is added before the
server is connected to a client, or if
ServerOptions.HasResources
is explicitly set. When a prompt is added, any clients already connected to the
server will be notified via a notifications/resources/list_changed
notification.
%include ../../mcp/server_example_test.go resources -
Tools
MCP servers can provide tools to allow clients to interact with external systems or functionality. Tools are effectively remote function calls, and the Go SDK provides mechanisms to bind them to ordinary Go functions.
Client-side: To list the server's tools, use the
ClientSession.Tools
iterator, or the lower-level
ClientSession.ListTools
(see pagination). Set
ClientOptions.ToolListChangedHandler
to be notified of changes in the list of tools.
To call a tool, use
ClientSession.CallTool
with CallToolParams holding the name and arguments of the tool to call.
res, err := session.CallTool(ctx, &mcp.CallToolParams{
Name: "my_tool",
Arguments: map[string]any{"name": "user"},
})
Arguments may be any value that can be marshaled to JSON.
Server-side: the basic API for adding a tool is symmetrical with the API
for prompts or resources:
Server.AddTool
adds a
Tool to
the server along with its
ToolHandler
to handle it. The server will have the tools capability if any tool is added
before the server is connected to a client, or if
ServerOptions.HasTools
is explicitly set. When a tool is added, any clients already connected to the
server will be notified via a notifications/tools/list_changed notification.
However, the Server.AddTool API leaves it to the user to implement the tool
handler correctly according to the spec, providing very little out of the box.
In order to implement a tool, the user must do all of the following:
- Provide a tool input and output schema.
- Validate the tool arguments against its input schema.
- Unmarshal the input schema into a Go value
- Execute the tool logic.
- Marshal the tool's structured output (if any) to JSON, and store it in the
result's
StructuredOutputfield as well as the unstructuredContentfield. - Validate that output JSON against the tool's output schema.
- If any tool errors occurred, pack them into the unstructured content and set
IsErrortotrue.
For this reason, the SDK provides a generic
AddTool
function that handles this for you. It can bind a tool to any function with the
following shape:
func(_ context.Context, request *CallToolRequest, input In) (result *CallToolResult, output Out, _ error)
This is like a ToolHandler, but with an extra arbitrary In input parameter,
and Out output parameter.
Such a function can then be bound to the server using AddTool:
mcp.AddTool(server, &mcp.Tool{Name: "my_tool"}, handler)
This does the following automatically:
- If
Tool.InputSchemaorTool.OutputSchemaare unset, the input and output schemas are inferred from theIntype, which must be a struct or map. Optionaljsonschemastruct tags provide argument descriptions. - Tool arguments are validated against the input schema.
- Tool arguments are marshaled into the
Invalue. - Tool output (the
Outvalue) is marshaled into the result'sStructuredOutput, as well as the unstructuredContent. - Output is validated against the tool's output schema.
- If an ordinary error is returned, it is stored int the
CallToolResultandIsErroris set totrue.
In fact, under ordinary circumstances, the user can ignore CallToolRequest
and CallToolResult.
For a more realistic example, consider a tool that retrieves the weather:
%include ../../mcp/tool_example_test.go weathertool -
In this case, we want to customize part of the inferred schema, though we can
still infer the rest. Since we want to control the inference ourselves, we set
the Tool.InputSchema explicitly:
%include ../../mcp/tool_example_test.go customschemas -
See mcp/tool_example_test.go for the full example, or examples/server/toolschemas for more examples of customizing tool schemas.
Utilities
Completion
To support the completion capability, the server needs a completion handler.
Client-side: completion is called using the
ClientSession.Complete
method.
Server-side: completion is enabled by setting
ServerOptions.CompletionHandler.
If this field is set to a non-nil value, the server will advertise the
completions server capability, and use this handler to respond to completion
requests.
%include ../../examples/server/completion/main.go completionhandler -
Logging
Pagination
Server-side feature lists may be paginated, using cursors. The SDK supports this by default.
Client-side: The ClientSession provides methods returning
iterators for each feature type.
These iterators are an iter.Seq2[Feature, error], where the error value
indicates whether page retrieval failed.
ClientSession.Promptsiterates prompts.ClientSession.Resourceiterates resources.ClientSession.ResourceTemplatesiterates resource templates.ClientSession.Toolsiterates tools.
The ClientSession also exposes ListXXX methods for fine-grained control
over pagination.
Server-side: pagination is on by default, so in general nothing is required
server-side. However, you may use
ServerOptions.PageSize
to customize the page size.