yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-09-02 12:57:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

improve subscriptions (#4304)

Browse Source
This commit is contained in:
Ryan Sweet 2024-11-22 05:57:11 -08:00 committed by GitHub
parent eb67e4ac93
commit 97fd6cc1e0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
dotnet/src/Microsoft.AutoGen/Agents/Services/Orleans/SubscriptionsGrain.cs
View File

@ -6,8 +6,13 @@ namespace Microsoft.AutoGen.Agents;
internal sealed class SubscriptionsGrain([PersistentState("state", "PubSubStore")] IPersistentState<SubscriptionsState> state) : Grain, ISubscriptionsGrain internal sealed class SubscriptionsGrain([PersistentState("state", "PubSubStore")] IPersistentState<SubscriptionsState> state) : Grain, ISubscriptionsGrain
{ {
private readonly Dictionary<string, List<string>> _subscriptions = new(); private readonly Dictionary<string, List<string>> _subscriptions = new();
public ValueTask<Dictionary<string, List<string>>> GetSubscriptions(string agentType) public ValueTask<Dictionary<string, List<string>>> GetSubscriptions(string? agentType = null)
{ {
//if agentType is null, return all subscriptions else filter on agentType
if (agentType != null)
{
return new ValueTask<Dictionary<string, List<string>>>(_subscriptions.Where(x => x.Value.Contains(agentType)).ToDictionary(x => x.Key, x => x.Value));
}
return new ValueTask<Dictionary<string, List<string>>>(_subscriptions); return new ValueTask<Dictionary<string, List<string>>>(_subscriptions);
} }
public ValueTask Subscribe(string agentType, string topic) public ValueTask Subscribe(string agentType, string topic)