microsoft/teams.net
Publicmirrored from https://github.com/microsoft/teams.netAvailable
Libraries/Microsoft.Teams.AI.Models.OpenAI/Extensions/ChatMessage.cs
30lines · modeblame
82a4e3c3Rajan1 years ago | 1 | // Copyright (c) Microsoft Corporation. All rights reserved. |
| 2 | // Licensed under the MIT License. | |
| 3 | | |
73e7847aAlex Acebo1 years ago | 4 | using Microsoft.Teams.AI.Messages; |
| 5 | | |
| 6 | using OpenAI.Chat; | |
| 7 | | |
| 8 | namespace Microsoft.Teams.AI.Models.OpenAI; | |
| 9 | | |
| 10 | public static partial class MessageExtensions | |
| 11 | { | |
| 12 | public static IMessage ToTeams(this ChatMessage message) | |
| 13 | { | |
| 14 | if (message is SystemChatMessage system) return system.ToTeams(); | |
| 15 | if (message is AssistantChatMessage assistant) return assistant.ToTeams(); | |
| 16 | if (message is ToolChatMessage tool) return tool.ToTeams(); | |
| 17 | if (message is UserChatMessage user) return user.ToTeams(); | |
| 18 | throw new Exception("OpenAI ChatMessage type not supported"); | |
| 19 | } | |
| 20 | | |
| 21 | public static ChatMessage ToOpenAI(this IMessage message) | |
| 22 | { | |
| 23 | if (message is DeveloperMessage developer) return developer.ToOpenAI(); | |
| 24 | if (message is ModelMessage<string> model) return model.ToOpenAI(); | |
| 25 | if (message is FunctionMessage function) return function.ToOpenAI(); | |
| 26 | if (message is UserMessage<string> userText) return userText.ToOpenAI(); | |
| 27 | if (message is UserMessage<IEnumerable<IContent>> userParts) return userParts.ToOpenAI(); | |
| 28 | throw new Exception("Teams Message type not supported"); | |
| 29 | } | |
| 30 | } |