reorder citations so it doesnt conflict with tools

microsoft · Dec 27, 2024 · d8a185d · d8a185d
1 parent 3c9e4fb
commit d8a185d
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/dotnet/packages/Microsoft.TeamsAI/Microsoft.TeamsAI/AI/Clients/LLMClient.cs b/dotnet/packages/Microsoft.TeamsAI/Microsoft.TeamsAI/AI/Clients/LLMClient.cs
@@ -199,6 +199,12 @@ public async Task<PromptResponse> CompletePromptAsync(
                     return;
                 }
 
+                IList<Citation>? citations = args.Chunk.delta?.Context?.Citations ?? null;
+
+                if (citations != null)
+                {
+                    streamer.SetCitations(citations);
+                }
 
                 // Ignore content without text
                 // - The chunk is likely for a Tool Call.
@@ -210,13 +216,6 @@ public async Task<PromptResponse> CompletePromptAsync(
 
                 // Send chunk to client
                 string text = args.Chunk.delta?.GetContent<string>() ?? "";
-                IList<Citation>? citations = args.Chunk.delta?.Context?.Citations ?? null;
-
-                if (citations != null)
-                {
-                    streamer.SetCitations(citations);
-                }
-
 
                 if (text.Length > 0)
                 {