Fix Choice.Content => Choice.Message

dotnet · Oct 17, 2024 · 4594124 · 4594124
1 parent 41e7383
commit 4594124
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/Libraries/Microsoft.Extensions.AI/ChatCompletion/OpenTelemetryChatClient.cs b/src/Libraries/Microsoft.Extensions.AI/ChatCompletion/OpenTelemetryChatClient.cs
@@ -499,7 +499,7 @@ private void LogChatCompletion(ChatCompletion completion)
             {
                 FinishReason = completion.FinishReason?.Value ?? "error",
                 Index = choiceIndex,
-                Content = CreateAssistantEvent(completion.Choices[choiceIndex]),
+                Message = CreateAssistantEvent(completion.Choices[choiceIndex]),
             }, OtelContext.Default.ChoiceEvent));
         }
     }
@@ -577,7 +577,7 @@ private sealed class ChoiceEvent
     {
         public string? FinishReason { get; set; }
         public int Index { get; set; }
-        public AssistantEvent? Content { get; set; }
+        public AssistantEvent? Message { get; set; }
     }
 
     private sealed class ToolCall

diff --git a/test/Libraries/Microsoft.Extensions.AI.Tests/ChatCompletion/OpenTelemetryChatClientTests.cs b/test/Libraries/Microsoft.Extensions.AI.Tests/ChatCompletion/OpenTelemetryChatClientTests.cs
@@ -167,7 +167,7 @@ async static IAsyncEnumerable<StreamingChatCompletionUpdate> CallbackAsync(
                 log => Assert.Equal("""{"id":"12345","content":"John"}""", log.Message),
                 log => Assert.Equal("""{"content":"Hey John, what\u0027s up?"}""", log.Message),
                 log => Assert.Equal("""{"content":"What\u0027s the biggest animal?"}""", log.Message),
-                log => Assert.Equal("""{"finish_reason":"stop","index":0,"content":{"content":"The blue whale, I think."}}""", log.Message));
+                log => Assert.Equal("""{"finish_reason":"stop","index":0,"message":{"content":"The blue whale, I think."}}""", log.Message));
         }
         else
         {
@@ -178,7 +178,7 @@ async static IAsyncEnumerable<StreamingChatCompletionUpdate> CallbackAsync(
                 log => Assert.Equal("""{"id":"12345"}""", log.Message),
                 log => Assert.Equal("""{}""", log.Message),
                 log => Assert.Equal("""{}""", log.Message),
-                log => Assert.Equal("""{"finish_reason":"stop","index":0,"content":{}}""", log.Message));
+                log => Assert.Equal("""{"finish_reason":"stop","index":0,"message":{}}""", log.Message));
         }
 
         Assert.Collection(logs,