wehub-resource-sync
63 lines
2.3 KiB
C#
63 lines
2.3 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
using System;
|
|
using System.Linq;
|
|
using Microsoft.SemanticKernel;
|
|
using ModelContextProtocol.Protocol;
|
|
|
|
namespace MCPClient;
|
|
|
|
/// <summary>
|
|
/// Extension methods for <see cref="ChatMessageContent"/>.
|
|
/// </summary>
|
|
public static class ChatMessageContentExtensions
|
|
{
|
|
/// <summary>
|
|
/// Converts a <see cref="ChatMessageContent"/> to a <see cref="CreateMessageResult"/>.
|
|
/// </summary>
|
|
/// <param name="chatMessageContent">The <see cref="ChatMessageContent"/> to convert.</param>
|
|
/// <returns>The corresponding <see cref="CreateMessageResult"/>.</returns>
|
|
public static CreateMessageResult ToCreateMessageResult(this ChatMessageContent chatMessageContent)
|
|
{
|
|
// Using the same heuristic as in the original MCP SDK code: McpClientExtensions.ToCreateMessageResult for consistency.
|
|
// ChatMessageContent can contain multiple items of different modalities, while the CreateMessageResult
|
|
// can only have a single content type: text, image, or audio. First, look for image or audio content,
|
|
// and if not found, fall back to the text content type by concatenating the text of all text contents.
|
|
ContentBlock? content = null;
|
|
|
|
foreach (KernelContent item in chatMessageContent.Items)
|
|
{
|
|
if (item is ImageContent image)
|
|
{
|
|
content = new ImageContentBlock
|
|
{
|
|
Data = Convert.ToBase64String(image.Data!.Value.Span),
|
|
MimeType = image.MimeType ?? "image/jpeg"
|
|
};
|
|
break;
|
|
}
|
|
else if (item is AudioContent audio)
|
|
{
|
|
content = new AudioContentBlock
|
|
{
|
|
Data = Convert.ToBase64String(audio.Data!.Value.Span),
|
|
MimeType = audio.MimeType ?? "audio/mpeg"
|
|
};
|
|
break;
|
|
}
|
|
}
|
|
|
|
content ??= new TextContentBlock
|
|
{
|
|
Text = string.Concat(chatMessageContent.Items.OfType<TextContent>()),
|
|
};
|
|
|
|
return new CreateMessageResult
|
|
{
|
|
Role = chatMessageContent.Role.ToMCPRole(),
|
|
Model = chatMessageContent.ModelId ?? "unknown",
|
|
Content = content
|
|
};
|
|
}
|
|
}
|