Change RawRepresentation for AgentRunResponse/Update to ChatResponse/Update instead of RawRepresentation from that. (#278)

Author: westey-m

dotnet/src/Microsoft.Extensions.AI.Agents.Abstractions/AgentRunResponse.cs

@@ -54,7 +54,7 @@ public AgentRunResponse(ChatResponse response)
         this.AdditionalProperties = response.AdditionalProperties;
         this.CreatedAt = response.CreatedAt;
         this.Messages = response.Messages;
-        this.RawRepresentation = response.RawRepresentation;
+        this.RawRepresentation = response;
         this.ResponseId = response.ResponseId;
         this.Usage = response.Usage;
     }

dotnet/src/Microsoft.Extensions.AI.Agents.Abstractions/AgentRunResponseUpdate.cs

@@ -70,7 +70,7 @@ public AgentRunResponseUpdate(ChatResponseUpdate chatResponseUpdate)
         this.Contents = chatResponseUpdate.Contents;
         this.CreatedAt = chatResponseUpdate.CreatedAt;
         this.MessageId = chatResponseUpdate.MessageId;
-        this.RawRepresentation = chatResponseUpdate.RawRepresentation;
+        this.RawRepresentation = chatResponseUpdate;
         this.ResponseId = chatResponseUpdate.ResponseId;
         this.Role = chatResponseUpdate.Role;
     }

dotnet/tests/Microsoft.Extensions.AI.Agents.Abstractions.UnitTests/AgentRunResponseTests.cs

@@ -61,7 +61,7 @@ public void ConstructorWithChatResponseRoundtrips()
         Assert.Equal(chatResponse.CreatedAt, response.CreatedAt);
         Assert.Same(chatResponse.Messages, response.Messages);
         Assert.Equal(chatResponse.ResponseId, response.ResponseId);
-        Assert.Same(chatResponse.RawRepresentation, response.RawRepresentation);
+        Assert.Same(chatResponse, response.RawRepresentation as ChatResponse);
         Assert.Same(chatResponse.Usage, response.Usage);
     }
 

dotnet/tests/Microsoft.Extensions.AI.Agents.Abstractions.UnitTests/AgentRunResponseUpdatesTests.cs

@@ -48,7 +48,7 @@ public void ConstructorWithChatResponseUpdateRoundtrips()
         Assert.Same(chatResponseUpdate.Contents, response.Contents);
         Assert.Equal(chatResponseUpdate.CreatedAt, response.CreatedAt);
         Assert.Equal(chatResponseUpdate.MessageId, response.MessageId);
-        Assert.Same(chatResponseUpdate.RawRepresentation, response.RawRepresentation);
+        Assert.Same(chatResponseUpdate, response.RawRepresentation as ChatResponseUpdate);
         Assert.Equal(chatResponseUpdate.ResponseId, response.ResponseId);
         Assert.Equal(chatResponseUpdate.Role, response.Role);
     }