Log entries for a GenAI span now link to GenAI visualizer

src/Aspire.Dashboard/Components/Controls/SpanDetails.razor.cs

@@ -116,7 +116,7 @@ private void UpdateSpanActionsMenu()
             }
         });
 
-        if (GenAIHelpers.IsGenAISpan(ViewModel.Span.Attributes))
+        if (GenAIHelpers.HasGenAIAttribute(ViewModel.Span.Attributes))
         {
             _spanActionsMenuItems.Add(new MenuButtonItem
             {

src/Aspire.Dashboard/Components/Pages/StructuredLogs.razor

@@ -148,7 +148,7 @@
                                         </AspireTemplateColumn>
                                         <AspireTemplateColumn ColumnId="@MessageColumn" ColumnManager="@_manager" Title="@Loc[nameof(Dashboard.Resources.StructuredLogs.StructuredLogsMessageColumnHeader)]">
                                             @* Tooltip is displayed by the message GridValue instance *@
-                                            <LogMessageColumnDisplay FilterText="@(ViewModel.FilterText)" LogEntry="@context" LaunchGenAIVisualizerCallback="@LaunchGenAIVisualizerAsync" />
+                                            <LogMessageColumnDisplay FilterText="@(ViewModel.FilterText)" LogEntry="@context" IsGenAILogCallback="@IsGenAILogEntry" LaunchGenAIVisualizerCallback="@LaunchGenAIVisualizerAsync" />
                                         </AspireTemplateColumn>
                                         <AspireTemplateColumn ColumnId="@TraceColumn" ColumnManager="@_manager" Title="@Loc[nameof(Dashboard.Resources.StructuredLogs.StructuredLogsTraceColumnHeader)]">
                                             @if (!string.IsNullOrEmpty(context.TraceId))

src/Aspire.Dashboard/Components/Pages/StructuredLogs.razor.cs

@@ -541,6 +541,34 @@ public async Task UpdateViewModelFromQueryAsync(StructuredLogsPageViewModel view
         await InvokeAsync(_dataGrid.SafeRefreshDataAsync);
     }
 
+    private bool IsGenAILogEntry(OtlpLogEntry logEntry)
+    {
+        if (string.IsNullOrEmpty(logEntry.SpanId) || string.IsNullOrEmpty(logEntry.TraceId))
+        {
+            return false;
+        }
+
+        if (GenAIHelpers.HasGenAIAttribute(logEntry.Attributes))
+        {
+            // GenAI telemetry is on the log entry.
+            return true;
+        }
+
+        var span = TelemetryRepository.GetSpan(logEntry.TraceId, logEntry.SpanId);
+        if (span == null)
+        {
+            return false;
+        }
+
+        if (GenAIHelpers.HasGenAIAttribute(span.Attributes))
+        {
+            // Log entry belongs to a span that has GenAI telemetry.
+            return true;
+        }
+
+        return false;
+    }
+
     private async Task LaunchGenAIVisualizerAsync(OtlpLogEntry logEntry)
     {
         var available = await TraceLinkHelpers.WaitForSpanToBeAvailableAsync(
@@ -571,7 +599,13 @@ await GenAIVisualizerDialog.OpenDialogAsync(
                     var filters = ViewModel.GetFilters();
                     filters.Add(new FieldTelemetryFilter
                     {
-                        Field = GenAIHelpers.GenAISystem,
+                        Field = KnownStructuredLogFields.SpanIdField,
+                        Condition = FilterCondition.NotEqual,
+                        Value = string.Empty
+                    });
+                    filters.Add(new FieldTelemetryFilter
+                    {
+                        Field = KnownStructuredLogFields.TraceIdField,
                         Condition = FilterCondition.NotEqual,
                         Value = string.Empty
                     });
@@ -584,10 +618,22 @@ await GenAIVisualizerDialog.OpenDialogAsync(
                         Filters = filters
                     });
 
-                    return logs.Items
-                        .DistinctBy(l => (l.SpanId, l.TraceId))
-                        .Select(l => TelemetryRepository.GetSpan(l.TraceId, l.SpanId)!)
-                        .ToList();
+                    var genAISpans = new List<OtlpSpan>();
+                    foreach (var l in logs.Items.DistinctBy(l => (l.SpanId, l.TraceId)))
+                    {
+                        var span = TelemetryRepository.GetSpan(l.TraceId, l.SpanId);
+                        if (span == null)
+                        {
+                            continue;
+                        }
+
+                        if (GenAIHelpers.HasGenAIAttribute(l.Attributes) || GenAIHelpers.HasGenAIAttribute(span.Attributes))
+                        {
+                            genAISpans.Add(span);
+                        }
+                    }
+
+                    return genAISpans;
                 });
         }
     }

src/Aspire.Dashboard/Components/Pages/TraceDetail.razor.cs

@@ -544,7 +544,7 @@ private async Task ToggleSpanLogsAsync(OtlpLogEntry logEntry)
 
     private static bool IsGenAISpan(SpanWaterfallViewModel spanViewModel)
     {
-        return GenAIHelpers.IsGenAISpan(spanViewModel.Span.Attributes);
+        return GenAIHelpers.HasGenAIAttribute(spanViewModel.Span.Attributes);
     }
 
     private async Task OnGenAIClickedAsync(OtlpSpan span)

src/Aspire.Dashboard/Components/ResourcesGridColumns/LogMessageColumnDisplay.razor

@@ -17,9 +17,7 @@
            HighlightText="@FilterText"
            StopClickPropagation="true">
     <ContentInButtonArea>
-        @if (!string.IsNullOrEmpty(LogEntry.TraceId) &&
-            !string.IsNullOrEmpty(LogEntry.SpanId) &&
-            GenAIHelpers.IsGenAISpan(LogEntry.Attributes))
+        @if (IsGenAILogCallback(LogEntry))
         {
             <FluentButton Appearance="Appearance.Lightweight"
                           Title="@ControlStringsLoc[nameof(ControlsStrings.GenAIDetailsTitle)]"

src/Aspire.Dashboard/Components/ResourcesGridColumns/LogMessageColumnDisplay.razor.cs

@@ -17,6 +17,9 @@ public partial class LogMessageColumnDisplay
     [Parameter, EditorRequired]
     public required EventCallback<OtlpLogEntry> LaunchGenAIVisualizerCallback { get; set; }
 
+    [Parameter, EditorRequired]
+    public required Func<OtlpLogEntry, bool> IsGenAILogCallback { get; set; }
+
     private string? _exceptionText;
 
     protected override void OnInitialized()

src/Aspire.Dashboard/Model/GenAI/GenAIHelpers.cs

@@ -34,7 +34,7 @@ public static class GenAIHelpers
 
     public const string ErrorType = "error.type";
 
-    public static bool IsGenAISpan(KeyValuePair<string, string>[] attributes)
+    public static bool HasGenAIAttribute(KeyValuePair<string, string>[] attributes)
     {
         return attributes.GetValueWithFallback(GenAISystem, GenAIProviderName) is { Length: > 0 };
     }