autogen/dotnet/test/AutoGen.Tests/ImageMessageTests.cs
Xiaoyun Zhang a16b307dc0
[.Net] Add Goolge gemini (#2868)
* update

* add vertex gemini test

* remove DTO

* add test for vertexGeminiAgent

* update test name

* update IGeminiClient interface

* add test for streaming

* add message connector

* add gemini message extension

* add tests

* update

* add gemnini sample

* update examples

* add test for iamge

* fix test

* add more tests

* add streaming message test

* add comment

* remove unused json

* implement google gemini client

* update

* fix comment
2024-06-10 17:31:45 +00:00

39 lines
1.2 KiB
C#

// Copyright (c) Microsoft Corporation. All rights reserved.
// ImageMessageTests.cs
using System;
using System.IO;
using System.Threading.Tasks;
using FluentAssertions;
using Xunit;
namespace AutoGen.Tests;
public class ImageMessageTests
{
[Fact]
public async Task ItCreateFromLocalImage()
{
var image = Path.Combine("testData", "images", "background.png");
var binary = File.ReadAllBytes(image);
var base64 = Convert.ToBase64String(binary);
var imageMessage = new ImageMessage(Role.User, BinaryData.FromBytes(binary, "image/png"));
imageMessage.MimeType.Should().Be("image/png");
imageMessage.BuildDataUri().Should().Be($"data:image/png;base64,{base64}");
}
[Fact]
public async Task ItCreateFromUrl()
{
var image = Path.Combine("testData", "images", "background.png");
var fullPath = Path.GetFullPath(image);
var localUrl = new Uri(fullPath).AbsoluteUri;
var imageMessage = new ImageMessage(Role.User, localUrl);
imageMessage.Url.Should().Be(localUrl);
imageMessage.MimeType.Should().Be("image/png");
imageMessage.Data.Should().BeNull();
}
}