feat: add support for call tool result annotations (#547)

* fix: tool call result do not have annotations

* fix: annotations parse

* fix: ut

* fix: cr

---------

Co-authored-by: Ed Zynda <[email protected]>

Author: Crescent617

mcp/utils.go

@@ -536,6 +536,27 @@ func ExtractString(data map[string]any, key string) string {
 	return ""
 }
 
+func ParseAnnotations(data map[string]any) *Annotations {
+	if data == nil {
+		return nil
+	}
+	annotations := &Annotations{}
+	if value, ok := data["priority"]; ok {
+		annotations.Priority = cast.ToFloat64(value)
+	}
+
+	if value, ok := data["audience"]; ok {
+		for _, a := range cast.ToStringSlice(value) {
+			a := Role(a)
+			if a == RoleUser || a == RoleAssistant {
+				annotations.Audience = append(annotations.Audience, a)
+			}
+		}
+	}
+	return annotations
+
+}
+
 func ExtractMap(data map[string]any, key string) map[string]any {
 	if value, ok := data[key]; ok {
 		if m, ok := value.(map[string]any); ok {
@@ -548,26 +569,37 @@ func ExtractMap(data map[string]any, key string) map[string]any {
 func ParseContent(contentMap map[string]any) (Content, error) {
 	contentType := ExtractString(contentMap, "type")
 
+	var annotations *Annotations
+	if annotationsMap := ExtractMap(contentMap, "annotations"); annotationsMap != nil {
+		annotations = ParseAnnotations(annotationsMap)
+	}
+
 	switch contentType {
 	case ContentTypeText:
 		text := ExtractString(contentMap, "text")
-		return NewTextContent(text), nil
+		c := NewTextContent(text)
+		c.Annotations = annotations
+		return c, nil
 
 	case ContentTypeImage:
 		data := ExtractString(contentMap, "data")
 		mimeType := ExtractString(contentMap, "mimeType")
 		if data == "" || mimeType == "" {
 			return nil, fmt.Errorf("image data or mimeType is missing")
 		}
-		return NewImageContent(data, mimeType), nil
+		c := NewImageContent(data, mimeType)
+		c.Annotations = annotations
+		return c, nil
 
 	case ContentTypeAudio:
 		data := ExtractString(contentMap, "data")
 		mimeType := ExtractString(contentMap, "mimeType")
 		if data == "" || mimeType == "" {
 			return nil, fmt.Errorf("audio data or mimeType is missing")
 		}
-		return NewAudioContent(data, mimeType), nil
+		c := NewAudioContent(data, mimeType)
+		c.Annotations = annotations
+		return c, nil
 
 	case ContentTypeLink:
 		uri := ExtractString(contentMap, "uri")
@@ -577,7 +609,9 @@ func ParseContent(contentMap map[string]any) (Content, error) {
 		if uri == "" || name == "" {
 			return nil, fmt.Errorf("resource_link uri or name is missing")
 		}
-		return NewResourceLink(uri, name, description, mimeType), nil
+		c := NewResourceLink(uri, name, description, mimeType)
+		c.Annotations = annotations
+		return c, nil
 
 	case ContentTypeResource:
 		resourceMap := ExtractMap(contentMap, "resource")
@@ -590,7 +624,9 @@ func ParseContent(contentMap map[string]any) (Content, error) {
 			return nil, err
 		}
 
-		return NewEmbeddedResource(resourceContents), nil
+		c := NewEmbeddedResource(resourceContents)
+		c.Annotations = annotations
+		return c, nil
 	}
 
 	return nil, fmt.Errorf("unsupported content type: %s", contentType)