From 29e00e9a961d0fa7b88895ff43392ec93b657d02 Mon Sep 17 00:00:00 2001
From: lilydu <lilydu+odspmdb@microsoft.com>
Date: Thu, 2 Jan 2025 09:54:48 -0800
Subject: [PATCH] add extra check

---
 .../Microsoft.TeamsAI/Application/StreamingResponse.cs         | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/dotnet/packages/Microsoft.TeamsAI/Microsoft.TeamsAI/Application/StreamingResponse.cs b/dotnet/packages/Microsoft.TeamsAI/Microsoft.TeamsAI/Application/StreamingResponse.cs
index 631b9e167..5d44f9675 100644
--- a/dotnet/packages/Microsoft.TeamsAI/Microsoft.TeamsAI/Application/StreamingResponse.cs
+++ b/dotnet/packages/Microsoft.TeamsAI/Microsoft.TeamsAI/Application/StreamingResponse.cs
@@ -348,10 +348,11 @@ private async Task SendActivity(Activity activity)
                 // Add in Generated by AI
                 if (this.EnableGeneratedByAILabel == true)
                 {
+                    List<ClientCitation>? currCitations = CitationUtils.GetUsedCitations(this.Message, this.Citations);
                     AIEntity entity = new AIEntity();
                     if (this.Citations != null && this.Citations.Count > 0)
                     {
-                        entity.Citation = this.Citations;
+                        entity.Citation = currCitations;
                     }
 
                     entity.UsageInfo = this.SensitivityLabel;