mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
32e054f1fe
* Rename AI Agent packages to use Microsoft.Agents.AI * Fix for build * Fix formatting * Fix formatting * Ignore in VSTHRD200 in migration samples * Ignore in VSTHRD200 in migration samples * Add some missing projects and run format * Fix build errors * Address code review feedback * Fix merge issues --------- Co-authored-by: Mark Wallace <markwallace@microsoft.com>
92 lines
3.2 KiB
C#
92 lines
3.2 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
using System.Collections.Generic;
|
|
using A2A;
|
|
using Microsoft.Extensions.AI;
|
|
|
|
namespace Microsoft.Agents.AI.A2A.UnitTests;
|
|
|
|
/// <summary>
|
|
/// Unit tests for the <see cref="ChatMessageExtensions"/> class.
|
|
/// </summary>
|
|
public sealed class ChatMessageExtensionsTests
|
|
{
|
|
[Fact]
|
|
public void ToA2AMessage_WithMessageContainingMultipleContents_AddsAllContentsAsParts()
|
|
{
|
|
// Arrange
|
|
var contents = new List<AIContent>
|
|
{
|
|
new HostedFileContent("https://example.com/report.pdf"),
|
|
new TextContent("please summarize the file content"),
|
|
new TextContent("and send it to me over email")
|
|
};
|
|
var chatMessage = new ChatMessage(ChatRole.User, contents);
|
|
var messages = new List<ChatMessage> { chatMessage };
|
|
|
|
// Act
|
|
var a2aMessage = messages.ToA2AMessage();
|
|
|
|
// Assert
|
|
Assert.NotNull(a2aMessage);
|
|
Assert.NotNull(a2aMessage.MessageId);
|
|
Assert.NotEmpty(a2aMessage.MessageId);
|
|
|
|
Assert.Equal(MessageRole.User, a2aMessage.Role);
|
|
|
|
Assert.NotNull(a2aMessage.Parts);
|
|
Assert.Equal(3, a2aMessage.Parts.Count);
|
|
|
|
var filePart = Assert.IsType<FilePart>(a2aMessage.Parts[0]);
|
|
Assert.NotNull(filePart.File);
|
|
var fileWithUri = Assert.IsType<FileWithUri>(filePart.File);
|
|
Assert.Equal("https://example.com/report.pdf", fileWithUri.Uri);
|
|
|
|
var secondTextPart = Assert.IsType<TextPart>(a2aMessage.Parts[1]);
|
|
Assert.Equal("please summarize the file content", secondTextPart.Text);
|
|
|
|
var thirdTextPart = Assert.IsType<TextPart>(a2aMessage.Parts[2]);
|
|
Assert.Equal("and send it to me over email", thirdTextPart.Text);
|
|
}
|
|
|
|
[Fact]
|
|
public void ToA2AMessage_WithMixedMessages_AddsAllContentsAsParts()
|
|
{
|
|
// Arrange
|
|
var firstMessage = new ChatMessage(ChatRole.User, [
|
|
new HostedFileContent("https://example.com/report.pdf")
|
|
]);
|
|
var secondMessage = new ChatMessage(ChatRole.User, [
|
|
new TextContent("please summarize the file content")
|
|
]);
|
|
var thirdMessage = new ChatMessage(ChatRole.User, [
|
|
new TextContent("and send it to me over email")
|
|
]);
|
|
var messages = new List<ChatMessage> { firstMessage, secondMessage, thirdMessage };
|
|
|
|
// Act
|
|
var a2aMessage = messages.ToA2AMessage();
|
|
|
|
// Assert
|
|
Assert.NotNull(a2aMessage);
|
|
Assert.NotNull(a2aMessage.MessageId);
|
|
Assert.NotEmpty(a2aMessage.MessageId);
|
|
|
|
Assert.Equal(MessageRole.User, a2aMessage.Role);
|
|
|
|
Assert.NotNull(a2aMessage.Parts);
|
|
Assert.Equal(3, a2aMessage.Parts.Count);
|
|
|
|
var filePart = Assert.IsType<FilePart>(a2aMessage.Parts[0]);
|
|
Assert.NotNull(filePart.File);
|
|
var fileWithUri = Assert.IsType<FileWithUri>(filePart.File);
|
|
Assert.Equal("https://example.com/report.pdf", fileWithUri.Uri);
|
|
|
|
var secondTextPart = Assert.IsType<TextPart>(a2aMessage.Parts[1]);
|
|
Assert.Equal("please summarize the file content", secondTextPart.Text);
|
|
|
|
var thirdTextPart = Assert.IsType<TextPart>(a2aMessage.Parts[2]);
|
|
Assert.Equal("and send it to me over email", thirdTextPart.Text);
|
|
}
|
|
}
|