Stevan/anthropic-client
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
900a7a354941f6b5a1357cacbafb64cd8220d918
anthropic-client/tests/AnthropicClient.Tests/Unit/Models/MessageRequestTests.cs
T

501 lines
16 KiB
C#
Raw Blame History

namespace AnthropicClient.Tests.Unit.Models;
public class MessageRequestTests : SerializationTest
{
private readonly string _testJson = @"{
""model"": ""claude-3-sonnet-20240229"",
""system"": [{
""type"": ""text"",
""text"": ""test-system""
}],
""messages"": [
{ ""role"": ""user"", ""content"": [{ ""text"": ""Hello!"", ""type"": ""text"" }] }
],
""max_tokens"": 512,
""metadata"": { ""test"":""test"" },
""stop_sequences"": [],
""temperature"": 0.5,
""topK"": 10,
""topP"": 0.5,
""tool_choice"": { ""type"":""auto"" },
""tools"": [],
""stream"": false
}";
private readonly string _testJsonWithAnyToolChoice = @"{
""model"": ""claude-3-sonnet-20240229"",
""system"": [{
""type"": ""text"",
""text"": ""test-system""
}],
""messages"":[
{ ""role"": ""user"", ""content"": [{ ""text"": ""Hello!"", ""type"":""text"" }] }
],
""max_tokens"": 512,
""metadata"": {""test"":""test""},
""stop_sequences"": [],
""temperature"": 0.5,
""topK"": 10,
""topP"": 0.5,
""tool_choice"": {""type"":""any""},
""tools"": [],
""stream"":false
}";
private readonly string _testJsonWithSpecificToolChoice = @"{
""model"": ""claude-3-sonnet-20240229"",
""system"": [{
""type"": ""text"",
""text"": ""test-system""
}],
""messages"": [
{ ""role"": ""user"", ""content"": [{ ""text"": ""Hello!"", ""type"": ""text"" }] }
],
""max_tokens"": 512,
""metadata"": { ""test"": ""test"" },
""stop_sequences"": [],
""temperature"": 0.5,
""topK"": 10,
""topP"": 0.5,
""tool_choice"": { ""type"": ""tool"", ""name"": ""test-tool"" },
""tools"": [],
""stream"": false
}";
private readonly string _testJsonWithImageContent = @"{
""model"": ""claude-3-sonnet-20240229"",
""system"": [{
""type"": ""text"",
""text"": ""test-system""
}],
""messages"":[
{
""role"": ""user"",
""content"": [
{
""type"": ""image"",
""source"": { ""media_type"": ""image/jpeg"", ""data"": ""data"" }
}
]
}
],
""max_tokens"": 512,
""metadata"": { ""test"": ""test"" },
""stop_sequences"": [],
""temperature"": 0.5,
""topK"": 10,
""topP"": 0.5,
""tool_choice"": { ""type"": ""auto"" },
""tools"":[],
""stream"": false
}";
private readonly string _testJsonWithUnknownContent = @"{
""model"": ""claude-3-sonnet-20240229"",
""system"": [{
""type"": ""text"",
""text"": ""test-system""
}],
""messages"": [{ ""role"": ""user"", ""content"": [{ ""type"": ""unknown"", ""text"": ""text"" }] }],
""max_tokens"": 512,
""metadata"": { ""test"": ""test"" },
""stop_sequences"": [],
""temperature"": 0.5,
""topK"": 10,
""topP"": 0.5,
""tool_choice"": { ""type"":""auto"" },
""tools"": [],
""stream"": false
}";
private readonly string _testJsonWithToolUseContent = @"{
""model"": ""claude-3-sonnet-20240229"",
""system"": [{
""type"": ""text"",
""text"": ""test-system""
}],
""messages"": [
{
""role"": ""assistant"",
""content"": [
{
""type"": ""tool_use"",
""name"": ""test-tool"",
""id"": ""test-tool-id"",
""input"": {
""test-property"": ""test-value""
}
}
]
}
],
""max_tokens"": 512,
""metadata"": { ""test"": ""test"" },
""stop_sequences"": [],
""temperature"": 0.5,
""topK"": 10,
""topP"": 0.5,
""tool_choice"": { ""type"": ""auto"" },
""tools"": [],
""stream"": false
}";
private readonly string _testJsonWithToolResultContent = @"{
""model"": ""claude-3-sonnet-20240229"",
""system"": [{
""type"": ""text"",
""text"": ""test-system""
}],
""messages"": [
{
""role"": ""assistant"",
""content"": [
{
""type"": ""tool_result"",
""tool_use_id"": ""test-tool"",
""content"": ""test-value""
}
]
}
],
""max_tokens"": 512,
""metadata"": { ""test"": ""test"" },
""stop_sequences"": [],
""temperature"": 0.5,
""topK"": 10,
""topP"": 0.5,
""tool_choice"": { ""type"": ""auto"" },
""tools"": [],
""stream"":false
}";
[Fact]
public void Constructor_WhenCalled_ItShouldInitializeProperties()
{
var model = AnthropicModels.Claude3Sonnet;
var messages = new List<Message> { new() };
var maxTokens = 512;
var system = "test-system";
var metadata = new Dictionary<string, object> { ["test"] = "test" };
var temperature = 0.5m;
var topK = 10;
var topP = 0.5m;
var toolChoice = new AutoToolChoice();
var tools = new List<Tool>();
var messageRequest = new MessageRequest(
model: model,
messages: messages,
maxTokens: maxTokens,
system: system,
metadata: metadata,
temperature: temperature,
topK: topK,
topP: topP,
toolChoice: toolChoice,
tools: tools
);
messageRequest.Model.Should().Be(model);
messageRequest.Messages.Should().BeSameAs(messages);
messageRequest.MaxTokens.Should().Be(maxTokens);
messageRequest.System.Should().Be(system);
messageRequest.Metadata.Should().BeSameAs(metadata);
messageRequest.Temperature.Should().Be(temperature);
messageRequest.TopK.Should().Be(topK);
messageRequest.TopP.Should().Be(topP);
messageRequest.ToolChoice.Should().Be(toolChoice);
messageRequest.Tools.Should().BeSameAs(tools);
messageRequest.Stream.Should().BeFalse();
}
[Fact]
public void Constructor_WhenCalledAndModelIsNull_ItShouldThrowArgumentNullException()
{
var action = () => new MessageRequest(
model: null!,
messages: [new()]
);
action.Should().Throw<ArgumentNullException>();
}
[Fact]
public void Constructor_WhenCalledAndMessagesIsNull_ItShouldThrowArgumentNullException()
{
var action = () => new MessageRequest(
model: AnthropicModels.Claude3Sonnet,
messages: null!
);
action.Should().Throw<ArgumentNullException>();
}
[Fact]
public void Constructor_WhenCalledAndModelIsInvalid_ItShouldThrowArgumentException()
{
var action = () => new MessageRequest(
model: "invalid-model",
messages: [new()]
);
action.Should().Throw<ArgumentException>();
}
[Fact]
public void Constructor_WhenCalledAndMessagesIsEmpty_ItShouldThrowArgumentException()
{
var action = () => new MessageRequest(
model: AnthropicModels.Claude3Sonnet,
messages: []
);
action.Should().Throw<ArgumentException>();
}
[Fact]
public void Constructor_WhenCalledAndMaxTokensIsInvalid_ItShouldThrowArgumentException()
{
var action = () => new MessageRequest(
model: AnthropicModels.Claude3Sonnet,
messages: [new()],
maxTokens: 0
);
action.Should().Throw<ArgumentException>();
}
[Theory]
[InlineData(-1)]
[InlineData(2)]
public void Constructor_WhenCalledAndTemperatureIsInvalid_ItShouldThrowArgumentException(decimal temperature)
{
var action = () => new MessageRequest(
model: AnthropicModels.Claude3Sonnet,
messages: [new()],
temperature: temperature
);
action.Should().Throw<ArgumentException>();
}
[Fact]
public void JsonSerialization_WhenSerialized_ItShouldHaveExpectedShape()
{
var messages = new List<Message>()
{
new()
{
Role = MessageRole.User,
Content = [new TextContent("Hello!")]
}
};
var model = AnthropicModels.Claude3Sonnet;
var maxTokens = 512;
var system = "test-system";
var metadata = new Dictionary<string, object>
{
["test"] = "test"
};
var temperature = 0.5m;
var topK = 10;
var topP = 0.5m;
var toolChoice = new AutoToolChoice();
var tools = new List<Tool>();
var messageRequest = new MessageRequest(
model: model,
messages: messages,
maxTokens: maxTokens,
system: system,
metadata: metadata,
temperature: temperature,
topK: topK,
topP: topP,
toolChoice: toolChoice,
tools: tools
);
var actual = Serialize(messageRequest);
JsonAssert.Equal(_testJson, actual);
}
[Fact]
public void JsonDeserialization_WhenDeserialized_ItShouldHaveExpectedShape()
{
var messageRequest = Deserialize<MessageRequest>(_testJson);
messageRequest!.Model.Should().Be(AnthropicModels.Claude3Sonnet);
messageRequest.System.Should().BeNull();
messageRequest.Messages.Should().HaveCount(1);
messageRequest.MaxTokens.Should().Be(512);
messageRequest.Metadata.Should().HaveCount(1);
var testValue = messageRequest.Metadata!.GetValueOrDefault("test")!.ToString();
testValue.Should().Be("test");
messageRequest.Temperature.Should().Be(0.5m);
messageRequest.TopK.Should().Be(10);
messageRequest.TopP.Should().Be(0.5m);
messageRequest.ToolChoice.Should().BeOfType<AutoToolChoice>();
messageRequest.ToolChoice!.Type.Should().Be("auto");
messageRequest.Tools.Should().HaveCount(0);
messageRequest.Stream.Should().BeFalse();
}
[Fact]
public void JsonDeserialization_WhenDeserializedWithAnyToolChoice_ItShouldHaveExpectedShape()
{
var messageRequest = Deserialize<MessageRequest>(_testJsonWithAnyToolChoice);
messageRequest!.Model.Should().Be(AnthropicModels.Claude3Sonnet);
messageRequest.System.Should().BeNull();
messageRequest.Messages.Should().HaveCount(1);
messageRequest.MaxTokens.Should().Be(512);
messageRequest.Metadata.Should().HaveCount(1);
var testValue = messageRequest.Metadata!.GetValueOrDefault("test")!.ToString();
testValue.Should().Be("test");
messageRequest.Temperature.Should().Be(0.5m);
messageRequest.TopK.Should().Be(10);
messageRequest.TopP.Should().Be(0.5m);
messageRequest.ToolChoice.Should().BeOfType<AnyToolChoice>();
messageRequest.ToolChoice!.Type.Should().Be("any");
messageRequest.Tools.Should().HaveCount(0);
}
[Fact]
public void JsonDeserialization_WhenDeserializedWithSpecificToolChoice_ItShouldHaveExpectedShape()
{
var messageRequest = Deserialize<MessageRequest>(_testJsonWithSpecificToolChoice);
messageRequest!.Model.Should().Be(AnthropicModels.Claude3Sonnet);
messageRequest.System.Should().BeNull();
messageRequest.Messages.Should().HaveCount(1);
messageRequest.MaxTokens.Should().Be(512);
messageRequest.Metadata.Should().HaveCount(1);
var testValue = messageRequest.Metadata!.GetValueOrDefault("test")!.ToString();
testValue.Should().Be("test");
messageRequest.Temperature.Should().Be(0.5m);
messageRequest.TopK.Should().Be(10);
messageRequest.TopP.Should().Be(0.5m);
messageRequest.ToolChoice.Should().BeOfType<SpecificToolChoice>();
var specificToolChoice = messageRequest.ToolChoice as SpecificToolChoice;
specificToolChoice!.Type.Should().Be("tool");
specificToolChoice.Name.Should().Be("test-tool");
messageRequest.Tools.Should().HaveCount(0);
}
[Fact]
public void JsonDeserialization_WhenDeserializedWithUnknownToolChoice_ItShouldThrowJsonException()
{
var json = @"{""model"":""claude-3-sonnet-20240229"",""system"":""test-system"",""messages"":[{""role"":""user"",""content"":[{""text"":""Hello!"",""type"":""text""}]}],""max_tokens"":512,""metadata"":{""test"":""test""},""stop_sequences"":[],""temperature"":0.5,""topK"":10,""topP"":0.5,""tool_choice"":{""type"":""unknown""},""tools"":[{""name"":""test-tool"",""description"":""test-description"",""input_schema"":{""type"":""object"",""properties"":{""test-property"":{""type"":""string"",""description"":""test-description""}},""required"":[""test-property""]}}],""stream"":false}";
var action = () => Deserialize<MessageRequest>(json);
action.Should().Throw<JsonException>();
}
[Fact]
public void JsonDeserialization_WhenDeserializedWithImageContent_ItShouldHaveExpectedShape()
{
var messageRequest = Deserialize<MessageRequest>(_testJsonWithImageContent);
messageRequest!.Model.Should().Be(AnthropicModels.Claude3Sonnet);
messageRequest.System.Should().BeNull();
messageRequest.Messages.Should().HaveCount(1);
messageRequest.MaxTokens.Should().Be(512);
messageRequest.Metadata.Should().HaveCount(1);
var testValue = messageRequest.Metadata!.GetValueOrDefault("test")!.ToString();
testValue.Should().Be("test");
messageRequest.Temperature.Should().Be(0.5m);
messageRequest.TopK.Should().Be(10);
messageRequest.TopP.Should().Be(0.5m);
messageRequest.ToolChoice.Should().BeOfType<AutoToolChoice>();
messageRequest.ToolChoice!.Type.Should().Be("auto");
messageRequest.Tools.Should().HaveCount(0);
messageRequest.Messages[0].Content.Should().HaveCount(1);
messageRequest.Messages[0].Content[0].Should().BeOfType<ImageContent>();
var imageContent = messageRequest.Messages[0].Content[0] as ImageContent;
imageContent!.Type.Should().Be("image");
imageContent.Source.MediaType.Should().Be("image/jpeg");
imageContent.Source.Data.Should().Be("data");
}
[Fact]
public void JsonDeserialization_WhenDeserializedWithToolUseContent_ItShouldHaveExpectedShape()
{
var messageRequest = Deserialize<MessageRequest>(_testJsonWithToolUseContent);
messageRequest!.Model.Should().Be(AnthropicModels.Claude3Sonnet);
messageRequest.System.Should().BeNull();
messageRequest.Messages.Should().HaveCount(1);
messageRequest.MaxTokens.Should().Be(512);
messageRequest.Metadata.Should().HaveCount(1);
var testValue = messageRequest.Metadata!.GetValueOrDefault("test")!.ToString();
testValue.Should().Be("test");
messageRequest.Temperature.Should().Be(0.5m);
messageRequest.TopK.Should().Be(10);
messageRequest.TopP.Should().Be(0.5m);
messageRequest.ToolChoice.Should().BeOfType<AutoToolChoice>();
messageRequest.ToolChoice!.Type.Should().Be("auto");
messageRequest.Tools.Should().HaveCount(0);
messageRequest.Messages[0].Content.Should().HaveCount(1);
messageRequest.Messages[0].Content[0].Should().BeOfType<ToolUseContent>();
var toolUseContent = messageRequest.Messages[0].Content[0] as ToolUseContent;
toolUseContent!.Type.Should().Be("tool_use");
toolUseContent.Name.Should().Be("test-tool");
toolUseContent.Id.Should().Be("test-tool-id");
toolUseContent.Input.Should().HaveCount(1);
toolUseContent.Input["test-property"]!.ToString().Should().Be("test-value");
}
[Fact]
public void JsonDeserialization_WhenDeserializedWithToolResultContent_ItShouldHaveExpectedShape()
{
var messageRequest = Deserialize<MessageRequest>(_testJsonWithToolResultContent);
messageRequest!.Model.Should().Be(AnthropicModels.Claude3Sonnet);
messageRequest.System.Should().BeNull();
messageRequest.Messages.Should().HaveCount(1);
messageRequest.MaxTokens.Should().Be(512);
messageRequest.Metadata.Should().HaveCount(1);
var testValue = messageRequest.Metadata!.GetValueOrDefault("test")!.ToString();
testValue.Should().Be("test");
messageRequest.Temperature.Should().Be(0.5m);
messageRequest.TopK.Should().Be(10);
messageRequest.TopP.Should().Be(0.5m);
messageRequest.ToolChoice.Should().BeOfType<AutoToolChoice>();
messageRequest.ToolChoice!.Type.Should().Be("auto");
messageRequest.Tools.Should().HaveCount(0);
messageRequest.Messages[0].Content.Should().HaveCount(1);
messageRequest.Messages[0].Content[0].Should().BeOfType<ToolResultContent>();
var toolResultContent = messageRequest.Messages[0].Content[0] as ToolResultContent;
toolResultContent!.Type.Should().Be("tool_result");
toolResultContent.ToolUseId.Should().Be("test-tool");
toolResultContent.Content.Should().Be("test-value");
}
[Fact]
public void JsonDeserialization_WhenDeserializedWithUnknownContent_ItShouldThrowJsonException()
{
var action = () => Deserialize<MessageRequest>(_testJsonWithUnknownContent);
action.Should().Throw<JsonException>();
}
}
Reference in New Issue View Git Blame Copy Permalink