Add cache invalidation when Blob/Data properties are set

Co-authored-by: stephentoub <[email protected]>

Author: 2 people

src/ModelContextProtocol.Core/Protocol/BlobResourceContents.cs

@@ -25,6 +25,7 @@ namespace ModelContextProtocol.Protocol;
 public sealed class BlobResourceContents : ResourceContents
 {
     private byte[]? _decodedData;
+    private ReadOnlyMemory<byte> _blob;
 
     /// <summary>
     /// Gets or sets the base64-encoded UTF-8 bytes representing the binary data of the item.
@@ -33,7 +34,15 @@ public sealed class BlobResourceContents : ResourceContents
     /// This is a zero-copy representation of the wire payload of this item. Setting this value will invalidate any cached value of <see cref="Data"/>.
     /// </remarks>
     [JsonPropertyName("blob")]
-    public required ReadOnlyMemory<byte> Blob { get; set; }
+    public required ReadOnlyMemory<byte> Blob
+    {
+        get => _blob;
+        set
+        {
+            _blob = value;
+            _decodedData = null; // Invalidate cache
+        }
+    }
 
     /// <summary>
     /// Gets the decoded data represented by <see cref="Blob"/>.

src/ModelContextProtocol.Core/Protocol/ContentBlock.cs

@@ -388,6 +388,7 @@ public sealed class TextContentBlock : ContentBlock
 public sealed class ImageContentBlock : ContentBlock
 {
     private byte[]? _decodedData;
+    private ReadOnlyMemory<byte> _data;
 
     /// <inheritdoc/>
     public override string Type => "image";
@@ -399,7 +400,15 @@ public sealed class ImageContentBlock : ContentBlock
     /// This is a zero-copy representation of the wire payload of this item. Setting this value will invalidate any cached value of <see cref="DecodedData"/>.
     /// </remarks>
     [JsonPropertyName("data")]
-    public required ReadOnlyMemory<byte> Data { get; set; }
+    public required ReadOnlyMemory<byte> Data
+    {
+        get => _data;
+        set
+        {
+            _data = value;
+            _decodedData = null; // Invalidate cache
+        }
+    }
 
     /// <summary>
     /// Gets the decoded image data represented by <see cref="Data"/>.
@@ -443,6 +452,7 @@ public ReadOnlyMemory<byte> DecodedData
 public sealed class AudioContentBlock : ContentBlock
 {
     private byte[]? _decodedData;
+    private ReadOnlyMemory<byte> _data;
 
     /// <inheritdoc/>
     public override string Type => "audio";
@@ -454,7 +464,15 @@ public sealed class AudioContentBlock : ContentBlock
     /// This is a zero-copy representation of the wire payload of this item. Setting this value will invalidate any cached value of <see cref="DecodedData"/>.
     /// </remarks>
     [JsonPropertyName("data")]
-    public required ReadOnlyMemory<byte> Data { get; set; }
+    public required ReadOnlyMemory<byte> Data
+    {
+        get => _data;
+        set
+        {
+            _data = value;
+            _decodedData = null; // Invalidate cache
+        }
+    }
 
     /// <summary>
     /// Gets the decoded audio data represented by <see cref="Data"/>.