mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-07 09:01:29 +00:00
698 lines
30 KiB
C#
698 lines
30 KiB
C#
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// OpenAIMessageTests.cs
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Reflection;
|
|
using System.Text.Json;
|
|
using System.Threading.Tasks;
|
|
using ApprovalTests;
|
|
using ApprovalTests.Namers;
|
|
using ApprovalTests.Reporters;
|
|
using AutoGen.Tests;
|
|
using FluentAssertions;
|
|
using OpenAI.Chat;
|
|
using Xunit;
|
|
|
|
namespace AutoGen.OpenAI.Tests;
|
|
|
|
[Trait("Category", "UnitV1")]
|
|
public class OpenAIMessageTests
|
|
{
|
|
private readonly JsonSerializerOptions jsonSerializerOptions = new JsonSerializerOptions
|
|
{
|
|
WriteIndented = true,
|
|
IgnoreReadOnlyProperties = false,
|
|
};
|
|
|
|
[Fact]
|
|
[UseReporter(typeof(DiffReporter))]
|
|
[UseApprovalSubdirectory("ApprovalTests")]
|
|
public void BasicMessageTest()
|
|
{
|
|
IMessage[] messages = [
|
|
new TextMessage(Role.System, "You are a helpful AI assistant"),
|
|
new TextMessage(Role.User, "Hello", "user"),
|
|
new TextMessage(Role.Assistant, "How can I help you?", from: "assistant"),
|
|
new ImageMessage(Role.User, "https://example.com/image.png", "user"),
|
|
new MultiModalMessage(Role.Assistant,
|
|
[
|
|
new TextMessage(Role.User, "Hello", "user"),
|
|
new ImageMessage(Role.User, "https://example.com/image.png", "user"),
|
|
], "user"),
|
|
new ToolCallMessage("test", "test", "assistant"),
|
|
new ToolCallResultMessage("result", "test", "test", "user"),
|
|
new ToolCallResultMessage(
|
|
[
|
|
new ToolCall("result", "test", "test"),
|
|
new ToolCall("result", "test", "test"),
|
|
], "user"),
|
|
new ToolCallMessage(
|
|
[
|
|
new ToolCall("test", "test"),
|
|
new ToolCall("test", "test"),
|
|
], "assistant"),
|
|
new AggregateMessage<ToolCallMessage, ToolCallResultMessage>(
|
|
message1: new ToolCallMessage("test", "test", "assistant"),
|
|
message2: new ToolCallResultMessage("result", "test", "test", "assistant"), "assistant"),
|
|
];
|
|
var openaiMessageConnectorMiddleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant");
|
|
|
|
var oaiMessages = messages.Select(m => (m, openaiMessageConnectorMiddleware.ProcessIncomingMessages(agent, [m])));
|
|
VerifyOAIMessages(oaiMessages);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItProcessUserTextMessageAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(async (msgs, _, innerAgent, _) =>
|
|
{
|
|
var innerMessage = msgs.Last();
|
|
innerMessage!.Should().BeOfType<MessageEnvelope<ChatMessage>>();
|
|
var chatRequestMessage = (UserChatMessage)((MessageEnvelope<ChatMessage>)innerMessage!).Content;
|
|
chatRequestMessage.Content.First().Text.Should().Be("Hello");
|
|
chatRequestMessage.ParticipantName.Should().Be("user");
|
|
return await innerAgent.GenerateReplyAsync(msgs);
|
|
})
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// user message
|
|
IMessage message = new TextMessage(Role.User, "Hello", "user");
|
|
await agent.GenerateReplyAsync([message]);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItShortcutChatRequestMessageAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(async (msgs, _, innerAgent, _) =>
|
|
{
|
|
var innerMessage = msgs.Last();
|
|
innerMessage!.Should().BeOfType<MessageEnvelope<UserChatMessage>>();
|
|
|
|
var chatRequestMessage = (UserChatMessage)((MessageEnvelope<UserChatMessage>)innerMessage!).Content;
|
|
chatRequestMessage.Content.First().Text.Should().Be("hello");
|
|
return await innerAgent.GenerateReplyAsync(msgs);
|
|
})
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// user message
|
|
var userMessage = new UserChatMessage("hello");
|
|
var chatRequestMessage = MessageEnvelope.Create(userMessage);
|
|
await agent.GenerateReplyAsync([chatRequestMessage]);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItShortcutMessageWhenStrictModelIsFalseAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(async (msgs, _, innerAgent, _) =>
|
|
{
|
|
var innerMessage = msgs.Last();
|
|
innerMessage!.Should().BeOfType<MessageEnvelope<string>>();
|
|
|
|
var chatRequestMessage = ((MessageEnvelope<string>)innerMessage!).Content;
|
|
chatRequestMessage.Should().Be("hello");
|
|
return await innerAgent.GenerateReplyAsync(msgs);
|
|
})
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// user message
|
|
var userMessage = "hello";
|
|
var chatRequestMessage = MessageEnvelope.Create(userMessage);
|
|
await agent.GenerateReplyAsync([chatRequestMessage]);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItThrowExceptionWhenStrictModeIsTrueAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector(true);
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// user message
|
|
var userMessage = "hello";
|
|
var chatRequestMessage = MessageEnvelope.Create(userMessage);
|
|
Func<Task> action = async () => await agent.GenerateReplyAsync([chatRequestMessage]);
|
|
|
|
await action.Should().ThrowAsync<InvalidOperationException>().WithMessage("Invalid message type: MessageEnvelope`1");
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItProcessAssistantTextMessageAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(async (msgs, _, innerAgent, _) =>
|
|
{
|
|
var innerMessage = msgs.Last();
|
|
innerMessage!.Should().BeOfType<MessageEnvelope<ChatMessage>>();
|
|
var chatRequestMessage = (AssistantChatMessage)((MessageEnvelope<ChatMessage>)innerMessage!).Content;
|
|
chatRequestMessage.Content.First().Text.Should().Be("How can I help you?");
|
|
chatRequestMessage.ParticipantName.Should().Be("assistant");
|
|
return await innerAgent.GenerateReplyAsync(msgs);
|
|
})
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// assistant message
|
|
IMessage message = new TextMessage(Role.Assistant, "How can I help you?", "assistant");
|
|
await agent.GenerateReplyAsync([message]);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItProcessSystemTextMessageAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(async (msgs, _, innerAgent, _) =>
|
|
{
|
|
var innerMessage = msgs.Last();
|
|
innerMessage!.Should().BeOfType<MessageEnvelope<ChatMessage>>();
|
|
var chatRequestMessage = (SystemChatMessage)((MessageEnvelope<ChatMessage>)innerMessage!).Content;
|
|
chatRequestMessage.Content.First().Text.Should().Be("You are a helpful AI assistant");
|
|
return await innerAgent.GenerateReplyAsync(msgs);
|
|
})
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// system message
|
|
IMessage message = new TextMessage(Role.System, "You are a helpful AI assistant");
|
|
await agent.GenerateReplyAsync([message]);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItProcessImageMessageAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(async (msgs, _, innerAgent, _) =>
|
|
{
|
|
var innerMessage = msgs.Last();
|
|
innerMessage!.Should().BeOfType<MessageEnvelope<ChatMessage>>();
|
|
var chatRequestMessage = (UserChatMessage)((MessageEnvelope<ChatMessage>)innerMessage!).Content;
|
|
chatRequestMessage.ParticipantName.Should().Be("user");
|
|
chatRequestMessage.Content.Count().Should().Be(1);
|
|
chatRequestMessage.Content.First().Kind.Should().Be(ChatMessageContentPartKind.Image);
|
|
return await innerAgent.GenerateReplyAsync(msgs);
|
|
})
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// user message
|
|
IMessage message = new ImageMessage(Role.User, "https://example.com/image.png", "user");
|
|
await agent.GenerateReplyAsync([message]);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItThrowExceptionWhenProcessingImageMessageFromSelfAndStrictModeIsTrueAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector(true);
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(middleware);
|
|
|
|
var imageMessage = new ImageMessage(Role.Assistant, "https://example.com/image.png", "assistant");
|
|
Func<Task> action = async () => await agent.GenerateReplyAsync([imageMessage]);
|
|
|
|
await action.Should().ThrowAsync<InvalidOperationException>().WithMessage("Invalid message type: ImageMessage");
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItProcessMultiModalMessageAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(async (msgs, _, innerAgent, _) =>
|
|
{
|
|
var innerMessage = msgs.Last();
|
|
innerMessage!.Should().BeOfType<MessageEnvelope<ChatMessage>>();
|
|
var chatRequestMessage = (UserChatMessage)((MessageEnvelope<ChatMessage>)innerMessage!).Content;
|
|
chatRequestMessage.ParticipantName.Should().Be("user");
|
|
chatRequestMessage.Content.Count().Should().Be(2);
|
|
chatRequestMessage.Content.First().Kind.Should().Be(ChatMessageContentPartKind.Text);
|
|
chatRequestMessage.Content.Last().Kind.Should().Be(ChatMessageContentPartKind.Image);
|
|
return await innerAgent.GenerateReplyAsync(msgs);
|
|
})
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// user message
|
|
IMessage message = new MultiModalMessage(
|
|
Role.User,
|
|
[
|
|
new TextMessage(Role.User, "Hello", "user"),
|
|
new ImageMessage(Role.User, "https://example.com/image.png", "user"),
|
|
], "user");
|
|
await agent.GenerateReplyAsync([message]);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItThrowExceptionWhenProcessingMultiModalMessageFromSelfAndStrictModeIsTrueAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector(true);
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(middleware);
|
|
|
|
var multiModalMessage = new MultiModalMessage(
|
|
Role.Assistant,
|
|
[
|
|
new TextMessage(Role.User, "Hello", "assistant"),
|
|
new ImageMessage(Role.User, "https://example.com/image.png", "assistant"),
|
|
], "assistant");
|
|
|
|
Func<Task> action = async () => await agent.GenerateReplyAsync([multiModalMessage]);
|
|
|
|
await action.Should().ThrowAsync<InvalidOperationException>().WithMessage("Invalid message type: MultiModalMessage");
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItProcessToolCallMessageAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(async (msgs, _, innerAgent, _) =>
|
|
{
|
|
var innerMessage = msgs.Last();
|
|
innerMessage!.Should().BeOfType<MessageEnvelope<ChatMessage>>();
|
|
var chatRequestMessage = (AssistantChatMessage)((MessageEnvelope<ChatMessage>)innerMessage!).Content;
|
|
// when the message is a tool call message
|
|
// the name field should not be set
|
|
// please visit OpenAIChatRequestMessageConnector class for more information
|
|
chatRequestMessage.ParticipantName.Should().BeNullOrEmpty();
|
|
chatRequestMessage.ToolCalls.Count().Should().Be(1);
|
|
chatRequestMessage.Content.First().Text.Should().Be("textContent");
|
|
chatRequestMessage.ToolCalls.First().Should().BeOfType<ChatToolCall>();
|
|
var functionToolCall = (ChatToolCall)chatRequestMessage.ToolCalls.First();
|
|
functionToolCall.FunctionName.Should().Be("test");
|
|
functionToolCall.Id.Should().Be("test");
|
|
functionToolCall.FunctionArguments.ToString().Should().Be("test");
|
|
return await innerAgent.GenerateReplyAsync(msgs);
|
|
})
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// user message
|
|
IMessage message = new ToolCallMessage("test", "test", "assistant")
|
|
{
|
|
Content = "textContent",
|
|
};
|
|
await agent.GenerateReplyAsync([message]);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItProcessParallelToolCallMessageAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(async (msgs, _, innerAgent, _) =>
|
|
{
|
|
var innerMessage = msgs.Last();
|
|
innerMessage!.Should().BeOfType<MessageEnvelope<ChatMessage>>();
|
|
var chatRequestMessage = (AssistantChatMessage)((MessageEnvelope<ChatMessage>)innerMessage!).Content;
|
|
chatRequestMessage.Content.Should().BeNullOrEmpty();
|
|
// when the message is a tool call message
|
|
// the name field should not be set
|
|
// please visit OpenAIChatRequestMessageConnector class for more information
|
|
chatRequestMessage.ParticipantName.Should().BeNullOrEmpty();
|
|
chatRequestMessage.ToolCalls.Count().Should().Be(2);
|
|
for (int i = 0; i < chatRequestMessage.ToolCalls.Count(); i++)
|
|
{
|
|
chatRequestMessage.ToolCalls.ElementAt(i).Should().BeOfType<ChatToolCall>();
|
|
var functionToolCall = (ChatToolCall)chatRequestMessage.ToolCalls.ElementAt(i);
|
|
functionToolCall.FunctionName.Should().Be("test");
|
|
functionToolCall.Id.Should().Be($"test_{i}");
|
|
functionToolCall.FunctionArguments.ToString().Should().Be("test");
|
|
}
|
|
return await innerAgent.GenerateReplyAsync(msgs);
|
|
})
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// user message
|
|
var toolCalls = new[]
|
|
{
|
|
new ToolCall("test", "test"),
|
|
new ToolCall("test", "test"),
|
|
};
|
|
IMessage message = new ToolCallMessage(toolCalls, "assistant");
|
|
await agent.GenerateReplyAsync([message]);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItThrowExceptionWhenProcessingToolCallMessageFromUserAndStrictModeIsTrueAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector(strictMode: true);
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(middleware);
|
|
|
|
var toolCallMessage = new ToolCallMessage("test", "test", "user");
|
|
Func<Task> action = async () => await agent.GenerateReplyAsync([toolCallMessage]);
|
|
await action.Should().ThrowAsync<InvalidOperationException>().WithMessage("Invalid message type: ToolCallMessage");
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItProcessToolCallResultMessageAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(async (msgs, _, innerAgent, _) =>
|
|
{
|
|
var innerMessage = msgs.Last();
|
|
innerMessage!.Should().BeOfType<MessageEnvelope<ChatMessage>>();
|
|
var chatRequestMessage = (ToolChatMessage)((MessageEnvelope<ChatMessage>)innerMessage!).Content;
|
|
chatRequestMessage.Content.First().Text.Should().Be("result");
|
|
chatRequestMessage.ToolCallId.Should().Be("test");
|
|
|
|
return await innerAgent.GenerateReplyAsync(msgs);
|
|
})
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// user message
|
|
IMessage message = new ToolCallResultMessage("result", "test", "test", "user");
|
|
await agent.GenerateReplyAsync([message]);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItProcessParallelToolCallResultMessageAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(async (msgs, _, innerAgent, _) =>
|
|
{
|
|
msgs.Count().Should().Be(2);
|
|
|
|
for (int i = 0; i < msgs.Count(); i++)
|
|
{
|
|
var innerMessage = msgs.ElementAt(i);
|
|
innerMessage!.Should().BeOfType<MessageEnvelope<ChatMessage>>();
|
|
var chatRequestMessage = (ToolChatMessage)((MessageEnvelope<ChatMessage>)innerMessage!).Content;
|
|
chatRequestMessage.Content.First().Text.Should().Be("result");
|
|
chatRequestMessage.ToolCallId.Should().Be($"test_{i}");
|
|
}
|
|
return await innerAgent.GenerateReplyAsync(msgs);
|
|
})
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// user message
|
|
var toolCalls = new[]
|
|
{
|
|
new ToolCall("test", "test", "result"),
|
|
new ToolCall("test", "test", "result"),
|
|
};
|
|
IMessage message = new ToolCallResultMessage(toolCalls, "user");
|
|
await agent.GenerateReplyAsync([message]);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItProcessFunctionCallMiddlewareMessageFromUserAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(async (msgs, _, innerAgent, _) =>
|
|
{
|
|
msgs.Count().Should().Be(1);
|
|
var innerMessage = msgs.Last();
|
|
innerMessage!.Should().BeOfType<MessageEnvelope<ChatMessage>>();
|
|
var chatRequestMessage = (UserChatMessage)((MessageEnvelope<ChatMessage>)innerMessage!).Content;
|
|
chatRequestMessage.Content.First().Text.Should().Be("result");
|
|
chatRequestMessage.ParticipantName.Should().Be("user");
|
|
return await innerAgent.GenerateReplyAsync(msgs);
|
|
})
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// user message
|
|
var toolCallMessage = new ToolCallMessage("test", "test", "user");
|
|
var toolCallResultMessage = new ToolCallResultMessage("result", "test", "test", "user");
|
|
var aggregateMessage = new AggregateMessage<ToolCallMessage, ToolCallResultMessage>(toolCallMessage, toolCallResultMessage, "user");
|
|
await agent.GenerateReplyAsync([aggregateMessage]);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItProcessFunctionCallMiddlewareMessageFromAssistantAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(async (msgs, _, innerAgent, _) =>
|
|
{
|
|
msgs.Count().Should().Be(2);
|
|
var innerMessage = msgs.Last();
|
|
innerMessage!.Should().BeOfType<MessageEnvelope<ChatMessage>>();
|
|
var chatRequestMessage = (ToolChatMessage)((MessageEnvelope<ChatMessage>)innerMessage!).Content;
|
|
chatRequestMessage.Content.First().Text.Should().Be("result");
|
|
chatRequestMessage.ToolCallId.Should().Be("test");
|
|
|
|
var toolCallMessage = msgs.First();
|
|
toolCallMessage!.Should().BeOfType<MessageEnvelope<ChatMessage>>();
|
|
var toolCallRequestMessage = (AssistantChatMessage)((MessageEnvelope<ChatMessage>)toolCallMessage!).Content;
|
|
toolCallRequestMessage.Content.Should().BeNullOrEmpty();
|
|
toolCallRequestMessage.ToolCalls.Count().Should().Be(1);
|
|
toolCallRequestMessage.ToolCalls.First().Should().BeOfType<ChatToolCall>();
|
|
var functionToolCall = (ChatToolCall)toolCallRequestMessage.ToolCalls.First();
|
|
functionToolCall.FunctionName.Should().Be("test");
|
|
functionToolCall.Id.Should().Be("test");
|
|
functionToolCall.FunctionArguments.ToString().Should().Be("test");
|
|
return await innerAgent.GenerateReplyAsync(msgs);
|
|
})
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// user message
|
|
var toolCallMessage = new ToolCallMessage("test", "test", "assistant");
|
|
var toolCallResultMessage = new ToolCallResultMessage("result", "test", "test", "assistant");
|
|
var aggregateMessage = new ToolCallAggregateMessage(toolCallMessage, toolCallResultMessage, "assistant");
|
|
await agent.GenerateReplyAsync([aggregateMessage]);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItProcessParallelFunctionCallMiddlewareMessageFromAssistantAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(async (msgs, _, innerAgent, _) =>
|
|
{
|
|
msgs.Count().Should().Be(3);
|
|
var toolCallMessage = msgs.First();
|
|
toolCallMessage!.Should().BeOfType<MessageEnvelope<ChatMessage>>();
|
|
var toolCallRequestMessage = (AssistantChatMessage)((MessageEnvelope<ChatMessage>)toolCallMessage!).Content;
|
|
toolCallRequestMessage.Content.Should().BeNullOrEmpty();
|
|
toolCallRequestMessage.ToolCalls.Count().Should().Be(2);
|
|
|
|
for (int i = 0; i < toolCallRequestMessage.ToolCalls.Count(); i++)
|
|
{
|
|
toolCallRequestMessage.ToolCalls.ElementAt(i).Should().BeOfType<ChatToolCall>();
|
|
var functionToolCall = (ChatToolCall)toolCallRequestMessage.ToolCalls.ElementAt(i);
|
|
functionToolCall.FunctionName.Should().Be("test");
|
|
functionToolCall.Id.Should().Be($"test_{i}");
|
|
functionToolCall.FunctionArguments.ToString().Should().Be("test");
|
|
}
|
|
|
|
for (int i = 1; i < msgs.Count(); i++)
|
|
{
|
|
var toolCallResultMessage = msgs.ElementAt(i);
|
|
toolCallResultMessage!.Should().BeOfType<MessageEnvelope<ChatMessage>>();
|
|
var toolCallResultRequestMessage = (ToolChatMessage)((MessageEnvelope<ChatMessage>)toolCallResultMessage!).Content;
|
|
toolCallResultRequestMessage.Content.First().Text.Should().Be("result");
|
|
toolCallResultRequestMessage.ToolCallId.Should().Be($"test_{i - 1}");
|
|
}
|
|
|
|
return await innerAgent.GenerateReplyAsync(msgs);
|
|
})
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// user message
|
|
var toolCalls = new[]
|
|
{
|
|
new ToolCall("test", "test", "result"),
|
|
new ToolCall("test", "test", "result"),
|
|
};
|
|
var toolCallMessage = new ToolCallMessage(toolCalls, "assistant");
|
|
var toolCallResultMessage = new ToolCallResultMessage(toolCalls, "assistant");
|
|
var aggregateMessage = new AggregateMessage<ToolCallMessage, ToolCallResultMessage>(toolCallMessage, toolCallResultMessage, "assistant");
|
|
await agent.GenerateReplyAsync([aggregateMessage]);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItReturnOriginalMessageWhenStrictModeIsFalseAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// text message
|
|
var textMessage = "hello";
|
|
var messageToSend = MessageEnvelope.Create(textMessage);
|
|
|
|
var message = await agent.GenerateReplyAsync([messageToSend]);
|
|
message.Should().BeOfType<MessageEnvelope<string>>();
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ItThrowInvalidOperationExceptionWhenStrictModeIsTrueAsync()
|
|
{
|
|
var middleware = new OpenAIChatRequestMessageConnector(true);
|
|
var agent = new EchoAgent("assistant")
|
|
.RegisterMiddleware(middleware);
|
|
|
|
// text message
|
|
var textMessage = new UserChatMessage("hello");
|
|
var messageToSend = MessageEnvelope.Create(textMessage);
|
|
Func<Task> action = async () => await agent.GenerateReplyAsync([messageToSend]);
|
|
|
|
await action.Should().ThrowAsync<InvalidOperationException>().WithMessage("Invalid return message type MessageEnvelope`1");
|
|
}
|
|
|
|
[Fact]
|
|
public void ToOpenAIChatRequestMessageShortCircuitTest()
|
|
{
|
|
var agent = new EchoAgent("assistant");
|
|
var middleware = new OpenAIChatRequestMessageConnector();
|
|
#pragma warning disable CS0618 // Type or member is obsolete
|
|
ChatMessage[] messages =
|
|
[
|
|
new UserChatMessage("Hello"),
|
|
new AssistantChatMessage("How can I help you?"),
|
|
new SystemChatMessage("You are a helpful AI assistant"),
|
|
new FunctionChatMessage("functionName", "result"),
|
|
new ToolChatMessage("test", "test"),
|
|
];
|
|
#pragma warning restore CS0618 // Type or member is obsolete
|
|
|
|
foreach (var oaiMessage in messages)
|
|
{
|
|
IMessage message = new MessageEnvelope<ChatMessage>(oaiMessage);
|
|
var oaiMessages = middleware.ProcessIncomingMessages(agent, [message]);
|
|
oaiMessages.Count().Should().Be(1);
|
|
//oaiMessages.First().Should().BeOfType<IMessage<ChatRequestMessage>>();
|
|
if (oaiMessages.First() is IMessage<ChatMessage> chatRequestMessage)
|
|
{
|
|
chatRequestMessage.Content.Should().Be(oaiMessage);
|
|
}
|
|
else
|
|
{
|
|
// fail the test
|
|
Assert.True(false);
|
|
}
|
|
}
|
|
}
|
|
private void VerifyOAIMessages(IEnumerable<(IMessage, IEnumerable<IMessage>)> messages)
|
|
{
|
|
var jsonObjects = messages.Select(pair =>
|
|
{
|
|
var (originalMessage, ms) = pair;
|
|
var objs = new List<object>();
|
|
foreach (var m in ms)
|
|
{
|
|
object? obj = null;
|
|
var chatRequestMessage = (m as IMessage<ChatMessage>)?.Content;
|
|
if (chatRequestMessage is UserChatMessage userMessage)
|
|
{
|
|
obj = new
|
|
{
|
|
Role = "user",
|
|
Content = userMessage.Content,
|
|
Name = userMessage.ParticipantName,
|
|
MultiModaItem = userMessage.Content?.Select(item =>
|
|
{
|
|
return item switch
|
|
{
|
|
_ when item.Kind == ChatMessageContentPartKind.Image => new
|
|
{
|
|
Type = "Image",
|
|
ImageUrl = GetImageUrlFromContent(item),
|
|
} as object,
|
|
_ when item.Kind == ChatMessageContentPartKind.Text => new
|
|
{
|
|
Type = "Text",
|
|
Text = item.Text,
|
|
} as object,
|
|
_ => throw new System.NotImplementedException(),
|
|
};
|
|
}),
|
|
};
|
|
}
|
|
|
|
if (chatRequestMessage is AssistantChatMessage assistantMessage)
|
|
{
|
|
obj = new
|
|
{
|
|
Role = "assistant",
|
|
Content = assistantMessage.Content,
|
|
Name = assistantMessage.ParticipantName,
|
|
TooCall = assistantMessage.ToolCalls.Select(tc =>
|
|
{
|
|
return tc switch
|
|
{
|
|
ChatToolCall functionToolCall => new
|
|
{
|
|
Type = "Function",
|
|
Name = functionToolCall.FunctionName,
|
|
Arguments = functionToolCall.FunctionArguments,
|
|
Id = functionToolCall.Id,
|
|
} as object,
|
|
_ => throw new System.NotImplementedException(),
|
|
};
|
|
}),
|
|
};
|
|
}
|
|
|
|
if (chatRequestMessage is SystemChatMessage systemMessage)
|
|
{
|
|
obj = new
|
|
{
|
|
Name = systemMessage.ParticipantName,
|
|
Role = "system",
|
|
Content = systemMessage.Content,
|
|
};
|
|
}
|
|
|
|
#pragma warning disable CS0618 // Type or member is obsolete
|
|
if (chatRequestMessage is FunctionChatMessage functionMessage)
|
|
{
|
|
obj = new
|
|
{
|
|
Role = "function",
|
|
Content = functionMessage.Content,
|
|
Name = functionMessage.FunctionName,
|
|
};
|
|
}
|
|
#pragma warning restore CS0618 // Type or member is obsolete
|
|
|
|
if (chatRequestMessage is ToolChatMessage toolCallMessage)
|
|
{
|
|
obj = new
|
|
{
|
|
Role = "tool",
|
|
Content = toolCallMessage.Content.First().Text,
|
|
ToolCallId = toolCallMessage.ToolCallId,
|
|
};
|
|
}
|
|
|
|
objs.Add(obj ?? throw new System.NotImplementedException());
|
|
}
|
|
|
|
return new
|
|
{
|
|
OriginalMessage = originalMessage.ToString(),
|
|
ConvertedMessages = objs,
|
|
};
|
|
});
|
|
|
|
var json = JsonSerializer.Serialize(jsonObjects, this.jsonSerializerOptions);
|
|
Approvals.Verify(json);
|
|
}
|
|
|
|
private object? GetImageUrlFromContent(ChatMessageContentPart content)
|
|
{
|
|
return content.ImageUri;
|
|
}
|
|
|
|
private static T CreateInstance<T>(params object[] args)
|
|
{
|
|
var type = typeof(T);
|
|
var instance = type.Assembly.CreateInstance(
|
|
type.FullName!, false,
|
|
BindingFlags.Instance | BindingFlags.NonPublic,
|
|
null, args, null, null);
|
|
return (T)instance!;
|
|
}
|
|
}
|