fix comment and agentProxy -- build succeeds

vovanduc · Jan 28, 2025 · e006240 · e006240
1 parent fd2409f
commit e006240
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/dotnet/src/Microsoft.AutoGen/Contracts/PythonEquiv/AgentProxy.cs b/dotnet/src/Microsoft.AutoGen/Contracts/PythonEquiv/AgentProxy.cs
@@ -57,7 +57,7 @@ public ValueTask<object> SendMessageAsync(object message, AgentId sender, string
     /// <returns>A task representing the asynchronous operation.</returns>
     public ValueTask LoadStateAsync(IDictionary<string, object> state)
     {
-        return this.runtime.LoadAgentStateAsync(state);
+        return this.runtime.LoadAgentStateAsync(this.Id, state);
     }
 
     /// <summary>
@@ -66,6 +66,6 @@ public ValueTask LoadStateAsync(IDictionary<string, object> state)
     /// <returns>A task representing the asynchronous operation, returning a dictionary containing the saved state.</returns>
     public ValueTask<IDictionary<string, object>> SaveStateAsync()
     {
-        return this.runtime.SaveAgentStateAsync();
+        return this.runtime.SaveAgentStateAsync(this.Id);
     }
 }
diff --git a/dotnet/src/Microsoft.AutoGen/Contracts/PythonEquiv/IAgentRuntime.cs b/dotnet/src/Microsoft.AutoGen/Contracts/PythonEquiv/IAgentRuntime.cs
@@ -15,7 +15,7 @@ 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(AgentId agentId, /*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*/);