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

fix: restore final assistant responses

Browse Source 
Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
This commit is contained in:
David Kaya
2026-03-28 12:44:14 +01:00
co-authored by Copilot
parent f1fa52f9c3
commit e13e3b818a
2 changed files with 124 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sidecar/src/Aryx.AgentHost/Services/WorkflowTranscriptProjector.cs
+80 -2
View File
@@ -1,5 +1,6 @@
using System.Text;
using Aryx.AgentHost.Contracts;
using GitHub.Copilot.SDK;
using Microsoft.Extensions.AI;
namespace Aryx.AgentHost.Services;
@@ -85,7 +86,7 @@ internal static class WorkflowTranscriptProjector
assistantMessages.Count - messageIndex,
command.Pattern,
fallbackAgent);
string content = message.Text ?? matchedSegment?.Content ?? string.Empty;
string content = ResolveProjectedContent(message, matchedSegment);
if (string.IsNullOrWhiteSpace(content))
{
continue;
@@ -127,7 +128,9 @@ internal static class WorkflowTranscriptProjector
{
return new ChatMessageDto
{
Id = matchedSegment?.MessageId ?? $"{command.RequestId}-final-{fallbackOutputIndex}",
Id = matchedSegment?.MessageId
?? message.MessageId
?? $"{command.RequestId}-final-{fallbackOutputIndex}",
Role = message.Role == ChatRole.System ? "system" : "assistant",
AuthorName = ResolveProjectedAuthorName(
command.Pattern,
@@ -139,6 +142,35 @@ internal static class WorkflowTranscriptProjector
};
}
private static string ResolveProjectedContent(
ChatMessage message,
TranscriptSegment? matchedSegment)
{
return FirstNonBlank(
message.Text,
matchedSegment?.Content,
TryGetAssistantMessageContent(message))
?? string.Empty;
}
private static string? TryGetAssistantMessageContent(ChatMessage message)
{
if (TryGetAssistantMessageData(message.RawRepresentation, out AssistantMessageData? assistantMessageData))
{
return assistantMessageData?.Content;
}
foreach (AIContent content in message.Contents)
{
if (TryGetAssistantMessageData(content.RawRepresentation, out assistantMessageData))
{
return assistantMessageData?.Content;
}
}
return null;
}
private static ChatMessageDto CreateProjectedMessageFromSegment(
RunTurnCommandDto command,
TranscriptSegment segment,
@@ -360,11 +392,57 @@ internal static class WorkflowTranscriptProjector
return fallbackAgent.Value.AgentName;
}
if (fallbackAgent.HasValue
&& string.IsNullOrWhiteSpace(primaryIdentifier)
&& string.IsNullOrWhiteSpace(fallbackIdentifier))
{
return fallbackAgent.Value.AgentName;
}
if (pattern.Agents.Count == 1
&& string.IsNullOrWhiteSpace(primaryIdentifier)
&& string.IsNullOrWhiteSpace(fallbackIdentifier))
{
PatternAgentDefinitionDto singleAgent = pattern.Agents[0];
return AgentIdentityResolver.ResolveDisplayAuthorName(pattern, singleAgent.Id, singleAgent.Name);
}
return AgentIdentityResolver.ResolveDisplayAuthorName(
pattern,
primaryIdentifier,
fallbackIdentifier);
}
private static bool TryGetAssistantMessageData(
object? rawRepresentation,
out AssistantMessageData? assistantMessageData)
{
switch (rawRepresentation)
{
case AssistantMessageEvent assistantMessage when !string.IsNullOrWhiteSpace(assistantMessage.Data.Content):
assistantMessageData = assistantMessage.Data;
return true;
case AssistantMessageData data when !string.IsNullOrWhiteSpace(data.Content):
assistantMessageData = data;
return true;
default:
assistantMessageData = null;
return false;
}
}
private static string? FirstNonBlank(params string?[] values)
{
foreach (string? value in values)
{
if (!string.IsNullOrWhiteSpace(value))
{
return value;
}
}
return null;
}
}
internal sealed class StreamingTranscriptBuffer