From fd2409f5e1d75b5a5fc0e0511fa772f3407dbd0f Mon Sep 17 00:00:00 2001
From: Jacob Alber <jaalber@microsoft.com>
Date: Mon, 27 Jan 2025 12:19:37 -0500
Subject: [PATCH] fix: Add missing AgentId in IAgentRuntime methods

* also formatting fix
---
 .../Microsoft.AutoGen/Contracts/PythonEquiv/IAgentRuntime.cs  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dotnet/src/Microsoft.AutoGen/Contracts/PythonEquiv/IAgentRuntime.cs b/dotnet/src/Microsoft.AutoGen/Contracts/PythonEquiv/IAgentRuntime.cs
index 107e193b8..a9d958fa4 100644
--- a/dotnet/src/Microsoft.AutoGen/Contracts/PythonEquiv/IAgentRuntime.cs
+++ b/dotnet/src/Microsoft.AutoGen/Contracts/PythonEquiv/IAgentRuntime.cs
@@ -15,8 +15,8 @@ public interface IAgentRuntime : ISaveState<IAgentRuntime>
     public ValueTask<AgentId> GetAgentAsync(AgentType agentType, string key = "default", bool lazy = true/*, CancellationToken? = default*/);
     public ValueTask<AgentId> GetAgentAsync(string agent, string key = "default", bool lazy = true/*, CancellationToken? = default*/);
 
-    public ValueTask<StateDict> SaveAgentStateAsync(/*CancellationToken? cancellationToken = default*/);
-    public ValueTask LoadAgentStateAsync(StateDict state/*, CancellationToken? cancellationToken = default*/);
+    public ValueTask<StateDict> SaveAgentStateAsync(AgentId agentId, /*CancellationToken? cancellationToken = default*/);
+    public ValueTask LoadAgentStateAsync(AgentId agentId, StateDict state/*, CancellationToken? cancellationToken = default*/);
 
     public ValueTask<AgentMetadata> GetAgentMetadataAsync(AgentId agentId/*, CancellationToken? cancellationToken = default*/);