diff --git a/dotnet/src/Microsoft.Extensions.AI.Agents.Abstractions/AgentRunResponse.cs b/dotnet/src/Microsoft.Extensions.AI.Agents.Abstractions/AgentRunResponse.cs index 28e964fbce..a50448c8a6 100644 --- a/dotnet/src/Microsoft.Extensions.AI.Agents.Abstractions/AgentRunResponse.cs +++ b/dotnet/src/Microsoft.Extensions.AI.Agents.Abstractions/AgentRunResponse.cs @@ -54,7 +54,7 @@ public class AgentRunResponse 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; } diff --git a/dotnet/src/Microsoft.Extensions.AI.Agents.Abstractions/AgentRunResponseUpdate.cs b/dotnet/src/Microsoft.Extensions.AI.Agents.Abstractions/AgentRunResponseUpdate.cs index 642216c6f0..4e6ce5f22e 100644 --- a/dotnet/src/Microsoft.Extensions.AI.Agents.Abstractions/AgentRunResponseUpdate.cs +++ b/dotnet/src/Microsoft.Extensions.AI.Agents.Abstractions/AgentRunResponseUpdate.cs @@ -70,7 +70,7 @@ public class AgentRunResponseUpdate 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; } diff --git a/dotnet/tests/Microsoft.Extensions.AI.Agents.Abstractions.UnitTests/AgentRunResponseTests.cs b/dotnet/tests/Microsoft.Extensions.AI.Agents.Abstractions.UnitTests/AgentRunResponseTests.cs index ec95bee39b..1ef375901b 100644 --- a/dotnet/tests/Microsoft.Extensions.AI.Agents.Abstractions.UnitTests/AgentRunResponseTests.cs +++ b/dotnet/tests/Microsoft.Extensions.AI.Agents.Abstractions.UnitTests/AgentRunResponseTests.cs @@ -61,7 +61,7 @@ public class AgentRunResponseTests 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); } diff --git a/dotnet/tests/Microsoft.Extensions.AI.Agents.Abstractions.UnitTests/AgentRunResponseUpdatesTests.cs b/dotnet/tests/Microsoft.Extensions.AI.Agents.Abstractions.UnitTests/AgentRunResponseUpdatesTests.cs index 9cac5b6151..55596ffe72 100644 --- a/dotnet/tests/Microsoft.Extensions.AI.Agents.Abstractions.UnitTests/AgentRunResponseUpdatesTests.cs +++ b/dotnet/tests/Microsoft.Extensions.AI.Agents.Abstractions.UnitTests/AgentRunResponseUpdatesTests.cs @@ -48,7 +48,7 @@ public class AgentRunResponseUpdateTests 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); }