microsoft/teams.net
Publicmirrored from https://github.com/microsoft/teams.netAvailable
Tests/Microsoft.Teams.AI.Tests/Utils/TestModel.cs
41lines · modeblame
cc0afc37Kavin9 months ago | 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 | | |
2e8d526cAlex Acebo9 months ago | 41 | internal class TestModelOptions { } |