From 9e7dcab50ca49365f55a07c863407de9cd51e02e Mon Sep 17 00:00:00 2001
From: Peter Ibekwe <peibekwe@microsoft.com>
Date: Mon, 20 Apr 2026 18:28:04 -0700
Subject: [PATCH] Make relevant object disposable and improve exception
 handling.

---
 .../ObjectModel/InvokeAzureAgentExecutor.cs                   | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dotnet/src/Microsoft.Agents.AI.Workflows.Declarative/ObjectModel/InvokeAzureAgentExecutor.cs b/dotnet/src/Microsoft.Agents.AI.Workflows.Declarative/ObjectModel/InvokeAzureAgentExecutor.cs
index 13dfc30066..322c460ee3 100644
--- a/dotnet/src/Microsoft.Agents.AI.Workflows.Declarative/ObjectModel/InvokeAzureAgentExecutor.cs
+++ b/dotnet/src/Microsoft.Agents.AI.Workflows.Declarative/ObjectModel/InvokeAzureAgentExecutor.cs
@@ -94,11 +94,11 @@ internal sealed class InvokeAzureAgentExecutor(InvokeAzureAgent model, ResponseA
         {
             try
             {
-                JsonDocument jsonDocument = JsonDocument.Parse(lastMessageText);
+                using JsonDocument jsonDocument = JsonDocument.Parse(lastMessageText);
                 Dictionary<string, object?> objectProperties = jsonDocument.ParseRecord(VariableType.RecordType);
                 await this.AssignAsync(this.AgentOutput?.ResponseObject?.Path, objectProperties.ToFormula(), context).ConfigureAwait(false);
             }
-            catch
+            catch (JsonException)
             {
                 // Not valid json, skip assignment.
             }