45 lines
1.0 KiB
C#
45 lines
1.0 KiB
C#
using AICore.Facade.LLM;
|
|
using BeWo.Data.Security;
|
|
using BeWo.Service.Core;
|
|
using BS.Shared.Tests.Objects;
|
|
using Microsoft.VisualStudio.TestTools.UnitTesting;
|
|
|
|
namespace AiCoreUnitTest
|
|
{
|
|
[TestClass]
|
|
public class Ollama2FacadeTest
|
|
{
|
|
public OllamaApiClient Client { get; set; }
|
|
|
|
[TestInitialize]
|
|
public void TestInitialize()
|
|
{
|
|
var url = "https://ai3.ownsoft.de";
|
|
var key = "mu9ZPuUXJ4aVGLU1WL2RfPCeSMRmJREwyjVJBP7bHfSz8quZRqN7e7UwnwCDi8GZ";
|
|
|
|
Client = new OllamaApiClient(url, key, BS.Shared.AiModelSource.ollama2);
|
|
}
|
|
|
|
[TestMethod]
|
|
public void GetModels()
|
|
{
|
|
var response = Client.GetAiModelle();
|
|
|
|
Assert.IsTrue(response?.Success);
|
|
}
|
|
|
|
[TestMethod]
|
|
public void SendMessage()
|
|
{
|
|
var logger = ServiceLogger.GetRequestLogger(out string id);
|
|
|
|
// Build Payload
|
|
var payload = AiTestObjects.GetExampleSendRequest(UserRightHelper.GetTenant(), id);
|
|
|
|
var response = Client.SendAiMessageRequest(payload, out LLMResponse llm_response, logger);
|
|
|
|
Assert.IsTrue(response?.Success);
|
|
}
|
|
}
|
|
}
|