move publishEventAsync override to agentBaseExtension and keep IAgent…

…Base simple
microsoft · Nov 26, 2024 · 50dea76 · 50dea76
1 parent 100f893
commit 50dea76
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/dotnet/src/Microsoft.AutoGen/Abstractions/IAgentBase.cs b/dotnet/src/Microsoft.AutoGen/Abstractions/IAgentBase.cs
@@ -16,6 +16,5 @@ public interface IAgentBase
     Task StoreAsync(AgentState state, CancellationToken cancellationToken = default);
     Task<T> ReadAsync<T>(AgentId agentId, CancellationToken cancellationToken = default) where T : IMessage, new();
     ValueTask PublishEventAsync(CloudEvent item, CancellationToken cancellationToken = default);
-    ValueTask PublishEventAsync(string topic, IMessage evt, CancellationToken cancellationToken = default);
     List<string> Subscribe(string topic);
 }
diff --git a/dotnet/src/Microsoft.AutoGen/Agents/AgentBase.cs b/dotnet/src/Microsoft.AutoGen/Agents/AgentBase.cs
@@ -265,10 +265,6 @@ public virtual Task HandleObject(object item)
         // otherwise, complain
         throw new InvalidOperationException($"No handler found for type {item.GetType().FullName}");
     }
-    public async ValueTask PublishEventAsync(string topic, IMessage evt, CancellationToken cancellationToken = default)
-    {
-        await PublishEventAsync(evt.ToCloudEvent(topic), cancellationToken).ConfigureAwait(false);
-    }
 
     public virtual Task Handle(CloudEvent item)
     {

diff --git a/dotnet/src/Microsoft.AutoGen/Agents/AgentBaseExtensions.cs b/dotnet/src/Microsoft.AutoGen/Agents/AgentBaseExtensions.cs
@@ -2,6 +2,8 @@
 // AgentBaseExtensions.cs
 
 using System.Diagnostics;
+using Google.Protobuf;
+using Microsoft.AutoGen.Abstractions;
 
 namespace Microsoft.AutoGen.Agents;
 
@@ -112,4 +114,9 @@ public static async Task InvokeWithActivityAsync<TState>(this AgentBase agent, F
             activity?.Stop();
         }
     }
+
+    public static async ValueTask PublishEventAsync(this AgentBase agent, string topic, IMessage evt, CancellationToken cancellationToken = default)
+    {
+        await agent.PublishEventAsync(evt.ToCloudEvent(topic), cancellationToken).ConfigureAwait(false);
+    }
 }