71265ba130
- Change the names of some content-related types to better match the spec. Most notably, rename Resource to EmbeddedResource, because the former has a different meaning in the spec. - Use []byte intead of string where the spec says "base64-encoded data." - Unexport EmbeddedResource.ToWire. Change-Id: I9f65efaadadda1b3abc0e58c64a743af1a72852c Reviewed-on: https://go-review.googlesource.com/c/tools/+/671362 Reviewed-by: Robert Findley <rfindley@google.com> TryBot-Bypass: Jonathan Amsterdam <jba@google.com> Commit-Queue: Jonathan Amsterdam <jba@google.com>
130 lines
3.2 KiB
Go
130 lines
3.2 KiB
Go
// Copyright 2025 The Go MCP SDK Authors. All rights reserved.
|
|
// Use of this source code is governed by an MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package mcp
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/modelcontextprotocol/go-sdk/mcp/protocol"
|
|
)
|
|
|
|
// Content is the union of supported content types: [TextContent],
|
|
// [ImageContent], [AudioContent], and [ResourceContent].
|
|
//
|
|
// ToWire converts content to its jsonrpc2 wire format.
|
|
type Content interface {
|
|
// TODO: unexport this, and move the tests that use it to this package.
|
|
ToWire() protocol.Content
|
|
}
|
|
|
|
// TextContent is a textual content.
|
|
type TextContent struct {
|
|
Text string
|
|
}
|
|
|
|
func (c TextContent) ToWire() protocol.Content {
|
|
return protocol.Content{Type: "text", Text: c.Text}
|
|
}
|
|
|
|
// ImageContent contains base64-encoded image data.
|
|
type ImageContent struct {
|
|
Data []byte // base64-encoded
|
|
MIMEType string
|
|
}
|
|
|
|
func (c ImageContent) ToWire() protocol.Content {
|
|
return protocol.Content{Type: "image", MIMEType: c.MIMEType, Data: c.Data}
|
|
}
|
|
|
|
// AudioContent contains base64-encoded audio data.
|
|
type AudioContent struct {
|
|
Data []byte
|
|
MIMEType string
|
|
}
|
|
|
|
func (c AudioContent) ToWire() protocol.Content {
|
|
return protocol.Content{Type: "audio", MIMEType: c.MIMEType, Data: c.Data}
|
|
}
|
|
|
|
// ResourceContent contains embedded resources.
|
|
type ResourceContent struct {
|
|
Resource EmbeddedResource
|
|
}
|
|
|
|
func (r ResourceContent) ToWire() protocol.Content {
|
|
res := r.Resource.toWire()
|
|
return protocol.Content{Type: "resource", Resource: &res}
|
|
}
|
|
|
|
type EmbeddedResource interface {
|
|
toWire() protocol.ResourceContents
|
|
}
|
|
|
|
// The {Text,Blob}ResourceContents types match the protocol definitions,
|
|
// but we represent both as a single type on the wire.
|
|
|
|
// A TextResourceContents is the contents of a text resource.
|
|
type TextResourceContents struct {
|
|
URI string
|
|
MIMEType string
|
|
Text string
|
|
}
|
|
|
|
func (r TextResourceContents) toWire() protocol.ResourceContents {
|
|
return protocol.ResourceContents{
|
|
URI: r.URI,
|
|
MIMEType: r.MIMEType,
|
|
Text: r.Text,
|
|
// Blob is nil, indicating this is a TextResourceContents.
|
|
}
|
|
}
|
|
|
|
// A BlobResourceContents is the contents of a blob resource.
|
|
type BlobResourceContents struct {
|
|
URI string
|
|
MIMEType string
|
|
Blob []byte
|
|
}
|
|
|
|
func (r BlobResourceContents) toWire() protocol.ResourceContents {
|
|
return protocol.ResourceContents{
|
|
URI: r.URI,
|
|
MIMEType: r.MIMEType,
|
|
Blob: r.Blob,
|
|
}
|
|
}
|
|
|
|
// ContentFromWireContent converts content from the jsonrpc2 wire format to a
|
|
// typed Content value.
|
|
func ContentFromWireContent(c protocol.Content) Content {
|
|
switch c.Type {
|
|
case "text":
|
|
return TextContent{Text: c.Text}
|
|
case "image":
|
|
return ImageContent{Data: c.Data, MIMEType: c.MIMEType}
|
|
case "audio":
|
|
return AudioContent{Data: c.Data, MIMEType: c.MIMEType}
|
|
case "resource":
|
|
r := ResourceContent{}
|
|
if c.Resource != nil {
|
|
if c.Resource.Blob != nil {
|
|
r.Resource = BlobResourceContents{
|
|
URI: c.Resource.URI,
|
|
MIMEType: c.Resource.MIMEType,
|
|
Blob: c.Resource.Blob,
|
|
}
|
|
} else {
|
|
r.Resource = TextResourceContents{
|
|
URI: c.Resource.URI,
|
|
MIMEType: c.Resource.MIMEType,
|
|
Text: c.Resource.Text,
|
|
}
|
|
}
|
|
}
|
|
return r
|
|
default:
|
|
panic(fmt.Sprintf("unrecognized wire content type %q", c.Type))
|
|
}
|
|
}
|