dac5be2c50
Now that the protocol package is merged into the mcp package, clean up the higher level 'Content' union types, in favor of types constructors. Change-Id: I37bedb0ff0a296eef799ba65c37ca7f94ee1dbbe Reviewed-on: https://go-review.googlesource.com/c/tools/+/672415 Reviewed-by: Jonathan Amsterdam <jba@google.com> TryBot-Bypass: Jonathan Amsterdam <jba@google.com>
97 lines
2.9 KiB
Go
97 lines
2.9 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 (
|
|
"encoding/json"
|
|
"fmt"
|
|
)
|
|
|
|
// The []byte fields below are marked omitzero, not omitempty:
|
|
// we want to marshal an empty byte slice.
|
|
// TODO(jba): figure out how to fix this for 1.23.
|
|
|
|
// Content is the wire format for content.
|
|
// It represents the protocol types TextContent, ImageContent, AudioContent
|
|
// and EmbeddedResource.
|
|
// The Type field distinguishes them. In the protocol, each type has a constant
|
|
// value for the field.
|
|
// At most one of Text, Data, and Resource is non-zero.
|
|
type Content struct {
|
|
Type string `json:"type"`
|
|
Text string `json:"text,omitempty"`
|
|
MIMEType string `json:"mimeType,omitempty"`
|
|
Data []byte `json:"data,omitzero"`
|
|
Resource *ResourceContents `json:"resource,omitempty"`
|
|
Annotations *Annotations `json:"annotations,omitempty"`
|
|
}
|
|
|
|
func (c *Content) UnmarshalJSON(data []byte) error {
|
|
type wireContent Content // for naive unmarshaling
|
|
var c2 wireContent
|
|
if err := json.Unmarshal(data, &c2); err != nil {
|
|
return err
|
|
}
|
|
switch c2.Type {
|
|
case "text", "image", "audio", "resource":
|
|
default:
|
|
return fmt.Errorf("unrecognized content type %s", c.Type)
|
|
}
|
|
*c = Content(c2)
|
|
return nil
|
|
}
|
|
|
|
func NewTextContent(text string) Content {
|
|
return Content{Type: "text", Text: text}
|
|
}
|
|
|
|
func NewImageContent(data []byte, mimeType string) Content {
|
|
return Content{Type: "image", Data: data, MIMEType: mimeType}
|
|
}
|
|
|
|
func NewAudioContent(data []byte, mimeType string) Content {
|
|
return Content{Type: "audio", Data: data, MIMEType: mimeType}
|
|
}
|
|
|
|
func NewResourceContent(resource ResourceContents) Content {
|
|
return Content{Type: "resource", Resource: &resource}
|
|
}
|
|
|
|
// A ResourceContents is either a TextResourceContents or a BlobResourceContents.
|
|
// See https://github.com/modelcontextprotocol/modelcontextprotocol/blob/main/schema/2025-03-26/schema.ts#L524-L551
|
|
// for the inheritance structure.
|
|
// If Blob is nil, this is a TextResourceContents; otherwise it's a BlobResourceContents.
|
|
//
|
|
// The URI field describes the resource location.
|
|
type ResourceContents struct {
|
|
URI string `json:"uri,"`
|
|
MIMEType string `json:"mimeType,omitempty"`
|
|
Text string `json:"text,omitempty"`
|
|
Blob []byte `json:"blob,omitzero"`
|
|
}
|
|
|
|
func NewTextResourceContents(uri, mimeType, text string) ResourceContents {
|
|
return ResourceContents{
|
|
URI: uri,
|
|
MIMEType: mimeType,
|
|
Text: 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 NewBlobResourceContents(uri, mimeType string, blob []byte) ResourceContents {
|
|
return ResourceContents{
|
|
URI: uri,
|
|
MIMEType: mimeType,
|
|
Blob: blob,
|
|
}
|
|
}
|