starred/aryx
1
0
Fork 0
mirror of https://github.com/davidkaya/aryx.git synced 2026-08-07 04:08:45 +02:00
Code Issues 3 Packages Projects Releases 10 Wiki Activity

fix: finalize streamed agent responses

Browse Source 
Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
This commit is contained in:
David Kaya
2026-03-21 17:37:21 +01:00
co-authored by Copilot
parent 9ba0174f68
commit dd40ebbfca
2 changed files with 120 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sidecar/src/Kopaya.AgentHost/Services/CopilotWorkflowRunner.cs
+18 -6
View File
@@ -138,10 +138,21 @@ public sealed class CopilotWorkflowRunner : ITurnWorkflowRunner
{
List<ChatMessage> allMessages = outputEvent.As<List<ChatMessage>>() ?? [];
List<ChatMessage> newMessages = allMessages.Skip(inputMessages.Count).ToList();
completedMessages = ConvertOutputMessages(command, newMessages, segments);
completedMessages = ProjectCompletedMessages(
command,
newMessages,
segments.Select(segment => (segment.MessageId, segment.AuthorName, segment.Content.ToString())).ToList());
}
}
if (completedMessages.Count == 0 && segments.Count > 0)
{
completedMessages = ProjectCompletedMessages(
command,
[],
segments.Select(segment => (segment.MessageId, segment.AuthorName, segment.Content.ToString())).ToList());
}
return completedMessages;
}
@@ -291,17 +302,18 @@ public sealed class CopilotWorkflowRunner : ITurnWorkflowRunner
return created;
}
private static List<ChatMessageDto> ConvertOutputMessages(
internal static List<ChatMessageDto> ProjectCompletedMessages(
RunTurnCommandDto command,
IReadOnlyList<ChatMessage> newMessages,
IReadOnlyList<StreamingSegment> segments)
IReadOnlyList<(string MessageId, string AuthorName, string Content)> segments)
{
List<ChatMessageDto> mapped = [];
int segmentIndex = 0;
foreach (ChatMessage message in newMessages.Where(message => message.Role != ChatRole.User))
{
StreamingSegment? segment = segmentIndex < segments.Count ? segments[segmentIndex] : null;
(string MessageId, string AuthorName, string Content)? segment =
segmentIndex < segments.Count ? segments[segmentIndex] : null;
segmentIndex++;
mapped.Add(new ChatMessageDto
@@ -312,7 +324,7 @@ public sealed class CopilotWorkflowRunner : ITurnWorkflowRunner
command.Pattern,
message.AuthorName,
segment?.AuthorName),
Content = message.Text ?? segment?.Content.ToString() ?? string.Empty,
Content = message.Text ?? segment?.Content ?? string.Empty,
CreatedAt = DateTimeOffset.UtcNow.ToString("O"),
});
}
@@ -324,7 +336,7 @@ public sealed class CopilotWorkflowRunner : ITurnWorkflowRunner
Id = segment.MessageId,
Role = "assistant",
AuthorName = AgentIdentityResolver.ResolveDisplayAuthorName(command.Pattern, segment.AuthorName),
Content = segment.Content.ToString(),
Content = segment.Content,
CreatedAt = DateTimeOffset.UtcNow.ToString("O"),
}));
}