feat: add content type

This commit is contained in:
Stevan Freeborn
2024-11-21 13:42:42 -06:00
parent 02bab6ae84
commit c4d4d8a15b
2 changed files with 14 additions and 19 deletions
@@ -24,4 +24,9 @@ public static class ContentType
/// Represents the tool result content type. /// Represents the tool result content type.
/// </summary> /// </summary>
public const string ToolResult = "tool_result"; public const string ToolResult = "tool_result";
/// <summary>
/// Represents the document content type.
/// </summary>
public const string Document = "document";
} }
@@ -5,40 +5,30 @@ public class ContentTypeTests
[Fact] [Fact]
public void Text_WhenCalled_ItShouldReturnText() public void Text_WhenCalled_ItShouldReturnText()
{ {
var expected = "text"; ContentType.Text.Should().Be("text");
var actual = ContentType.Text;
actual.Should().Be(expected);
} }
[Fact] [Fact]
public void Image_WhenCalled_ItShouldReturnImage() public void Image_WhenCalled_ItShouldReturnImage()
{ {
var expected = "image"; ContentType.Image.Should().Be("image");
var actual = ContentType.Image;
actual.Should().Be(expected);
} }
[Fact] [Fact]
public void ToolUse_WhenCalled_ItShouldReturnToolUse() public void ToolUse_WhenCalled_ItShouldReturnToolUse()
{ {
var expected = "tool_use"; ContentType.ToolUse.Should().Be("tool_use");
var actual = ContentType.ToolUse;
actual.Should().Be(expected);
} }
[Fact] [Fact]
public void ToolResult_WhenCalled_ItShouldReturnToolResult() public void ToolResult_WhenCalled_ItShouldReturnToolResult()
{ {
var expected = "tool_result"; ContentType.ToolResult.Should().Be("tool_result");
}
var actual = ContentType.ToolResult; [Fact]
public void Document_WhenCalled_ItShouldReturnDocument()
actual.Should().Be(expected); {
ContentType.Document.Should().Be("document");
} }
} }