BeWoPlaner/AiCoreUnitTest/OpenWebUIFacadeTest.cs

using AICore.Facade.LLM;
using Microsoft.VisualStudio.TestTools.UnitTesting;
using System;
using System.Net;

namespace AiCoreUnitTest
{
	[TestClass]
	public class OpenWebUIFacadeTest
	{
		public string OpenWebUIUrl { get; set; }
		public string OpenWebUIKey { get; set; }
		
		[TestInitialize]
		public void TestInitialize()
		{
			OpenWebUIUrl = "https://owui1.ownsoft.de";
			OpenWebUIKey = "sk-7f5d8c418630448dbe31ab1c7e8b100a";
		}

		public OpenWebApiClient GetClient()
			=> new OpenWebApiClient(OpenWebUIUrl, OpenWebUIKey);

		[TestMethod]
		public void GetModels()
		{
			var client = GetClient();

			var models = client.GetAiModelle();

			Assert.IsNotNull(models);
		}

		[TestMethod]
		public void SendMessage()
		{
			var client = GetClient();

			// Build Payload
			var payload = new
			{
				model = "llama3.2:latest",
				messages = new[] {
					new {
						role = "user",
						content = "Say Hi!"
					}
				}
			};

			var response = client.SendAiMessageRequest(payload, out string message, out string model, out int? duration);

			Assert.IsTrue(response?.Success);
		}
	}
}