microsoft/teams.net
Publicmirrored fromhttps://github.com/microsoft/teams.netAvailable
Tests/Microsoft.Teams.AI.Tests/Utils/TestModel.cs
41lines · modecode
| 1 | using Microsoft.Teams.AI.Messages; |
| 2 | using Microsoft.Teams.AI.Models; |
| 3 | |
| 4 | namespace Microsoft.Teams.AI.Tests.Utils; |
| 5 | |
| 6 | internal class TestModel : IChatModel<TestModelOptions> |
| 7 | { |
| 8 | string IModel<TestModelOptions>.Name => "TestModel"; |
| 9 | |
| 10 | Task<ModelMessage<string>> IChatModel<TestModelOptions>.Send(IMessage message, ChatModelOptions<TestModelOptions> options, CancellationToken cancellationToken) |
| 11 | { |
| 12 | foreach (var function in options.Functions) |
| 13 | { |
| 14 | options.Invoke(new FunctionCall() { Name = function.Name, Id = "testId" }, cancellationToken); |
| 15 | } |
| 16 | |
| 17 | return _Send(message, options, cancellationToken); |
| 18 | } |
| 19 | |
| 20 | Task<ModelMessage<string>> IChatModel<TestModelOptions>.Send(IMessage message, ChatModelOptions<TestModelOptions> options, IStream stream, CancellationToken cancellationToken) |
| 21 | { |
| 22 | foreach (var function in options.Functions) |
| 23 | { |
| 24 | options.Invoke(new FunctionCall() { Name = function.Name, Id = "testId" }, cancellationToken); |
| 25 | } |
| 26 | |
| 27 | return _Send(message, options, cancellationToken); |
| 28 | } |
| 29 | |
| 30 | Task<IMessage> IModel<TestModelOptions>.Send(IMessage message, TestModelOptions? options, CancellationToken cancellationToken) |
| 31 | { |
| 32 | return Task.FromResult((IMessage)new ModelMessage<string>("test")); |
| 33 | } |
| 34 | |
| 35 | private Task<ModelMessage<string>> _Send(IMessage message, ChatModelOptions<TestModelOptions> options, CancellationToken cancellationToken) |
| 36 | { |
| 37 | return Task.FromResult(new ModelMessage<string>("test")); |
| 38 | } |
| 39 | } |
| 40 | |
| 41 | internal class TestModelOptions { } |