| | | 1 | | using System.Text.Json; |
| | | 2 | | using Elsa.AI.Abstractions.Models; |
| | | 3 | | using GitHub.Copilot; |
| | | 4 | | |
| | | 5 | | namespace Elsa.AI.Copilot.Adapters; |
| | | 6 | | |
| | | 7 | | public class CopilotSessionEventMapper |
| | | 8 | | { |
| | | 9 | | public IEnumerable<AIProviderEvent> Map(SessionEvent sessionEvent) |
| | | 10 | | { |
| | 0 | 11 | | var timestamp = sessionEvent.Timestamp == default ? DateTimeOffset.UtcNow : sessionEvent.Timestamp; |
| | 0 | 12 | | var sequence = 0L; |
| | | 13 | | |
| | 0 | 14 | | switch (sessionEvent) |
| | | 15 | | { |
| | 0 | 16 | | case AssistantMessageDeltaEvent assistantDelta when !string.IsNullOrEmpty(assistantDelta.Data?.DeltaContent) |
| | 0 | 17 | | yield return Create("assistant.delta", sequence++, timestamp, new JsonObject |
| | 0 | 18 | | { |
| | 0 | 19 | | ["content"] = assistantDelta.Data.DeltaContent, |
| | 0 | 20 | | ["messageId"] = assistantDelta.Data.MessageId, |
| | 0 | 21 | | ["parentToolCallId"] = assistantDelta.Data.ParentToolCallId |
| | 0 | 22 | | }); |
| | 0 | 23 | | break; |
| | 0 | 24 | | case AssistantMessageEvent assistantMessage when !string.IsNullOrEmpty(assistantMessage.Data?.Content): |
| | 0 | 25 | | yield return Create("assistant.message", sequence++, timestamp, new JsonObject |
| | 0 | 26 | | { |
| | 0 | 27 | | ["content"] = assistantMessage.Data.Content, |
| | 0 | 28 | | ["messageId"] = assistantMessage.Data.MessageId, |
| | 0 | 29 | | ["model"] = assistantMessage.Data.Model, |
| | 0 | 30 | | ["turnId"] = assistantMessage.Data.TurnId |
| | 0 | 31 | | }); |
| | 0 | 32 | | break; |
| | 0 | 33 | | case AssistantReasoningDeltaEvent reasoningDelta when !string.IsNullOrEmpty(reasoningDelta.Data?.DeltaConten |
| | 0 | 34 | | yield return Create("assistant.reasoning.delta", sequence++, timestamp, new JsonObject |
| | 0 | 35 | | { |
| | 0 | 36 | | ["content"] = reasoningDelta.Data.DeltaContent, |
| | 0 | 37 | | ["reasoningId"] = reasoningDelta.Data.ReasoningId |
| | 0 | 38 | | }); |
| | 0 | 39 | | break; |
| | 0 | 40 | | case AssistantReasoningEvent reasoning when !string.IsNullOrEmpty(reasoning.Data?.Content): |
| | 0 | 41 | | yield return Create("assistant.reasoning", sequence++, timestamp, new JsonObject |
| | 0 | 42 | | { |
| | 0 | 43 | | ["content"] = reasoning.Data.Content, |
| | 0 | 44 | | ["reasoningId"] = reasoning.Data.ReasoningId |
| | 0 | 45 | | }); |
| | 0 | 46 | | break; |
| | | 47 | | case ToolExecutionStartEvent toolStart: |
| | 0 | 48 | | yield return Create("tool.started", sequence++, timestamp, new JsonObject |
| | 0 | 49 | | { |
| | 0 | 50 | | ["toolCallId"] = toolStart.Data?.ToolCallId, |
| | 0 | 51 | | ["toolName"] = ReadToolName(toolStart.Data?.ToolName, toolStart.Data?.McpToolName), |
| | 0 | 52 | | ["mcpServerName"] = toolStart.Data?.McpServerName, |
| | 0 | 53 | | ["arguments"] = toolStart.Data?.Arguments is { } arguments ? JsonNode.Parse(arguments.GetRawText()) |
| | 0 | 54 | | }); |
| | 0 | 55 | | break; |
| | | 56 | | case ToolExecutionCompleteEvent toolComplete: |
| | 0 | 57 | | yield return Create("tool.result", sequence++, timestamp, new JsonObject |
| | 0 | 58 | | { |
| | 0 | 59 | | ["toolCallId"] = toolComplete.Data?.ToolCallId, |
| | 0 | 60 | | ["toolName"] = ReadToolName(toolComplete.Data?.ToolDescription?.Name, null), |
| | 0 | 61 | | ["status"] = toolComplete.Data?.Success == false ? AIToolInvocationStatus.Failed.ToString() : AITool |
| | 0 | 62 | | ["summary"] = toolComplete.Data?.Result?.Content ?? "", |
| | 0 | 63 | | ["error"] = toolComplete.Data?.Error?.Message |
| | 0 | 64 | | }); |
| | 0 | 65 | | break; |
| | | 66 | | case SessionErrorEvent error: |
| | 0 | 67 | | yield return Create("conversation.error", sequence++, timestamp, new JsonObject |
| | 0 | 68 | | { |
| | 0 | 69 | | ["content"] = error.Data?.Message ?? "Copilot session error.", |
| | 0 | 70 | | ["errorCode"] = error.Data?.ErrorCode, |
| | 0 | 71 | | ["errorType"] = error.Data?.ErrorType, |
| | 0 | 72 | | ["statusCode"] = error.Data?.StatusCode |
| | 0 | 73 | | }); |
| | | 74 | | break; |
| | | 75 | | } |
| | 0 | 76 | | } |
| | | 77 | | |
| | | 78 | | private static AIProviderEvent Create(string type, long sequence, DateTimeOffset timestamp, JsonObject data) => |
| | 0 | 79 | | new() |
| | 0 | 80 | | { |
| | 0 | 81 | | Type = type, |
| | 0 | 82 | | Sequence = sequence, |
| | 0 | 83 | | Timestamp = timestamp, |
| | 0 | 84 | | Data = data |
| | 0 | 85 | | }; |
| | | 86 | | |
| | | 87 | | private static string? ReadToolName(string? toolName, string? mcpToolName) => |
| | 0 | 88 | | !string.IsNullOrWhiteSpace(toolName) ? toolName : mcpToolName; |
| | | 89 | | } |