yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-10-11 07:58:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Simplify subscription

Browse Source
This commit is contained in:
Kosta Petan 2024-02-28 15:44:13 +01:00
parent 125608e372
commit f074bbe37d
6 changed files with 9 additions and 128 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/libs/Microsoft.AI.DevTeam/Agents/Agent.cs
View File

@ -13,4 +13,13 @@ public abstract class Agent : Grain, IGrainWithStringKey
var stream = streamProvider.GetStream<Event>(streamId);
await stream.OnNextAsync(item);
}
public async override Task OnActivateAsync(CancellationToken cancellationToken)
{
var streamProvider = this.GetStreamProvider("StreamProvider");
var streamId = StreamId.Create(Consts.MainNamespace, this.GetPrimaryKeyString());
var stream = streamProvider.GetStream<Event>(streamId);
await stream.SubscribeAsync(HandleEvent);
}
}

9
src/libs/Microsoft.AI.DevTeam/Agents/AzureGenie.cs
View File

@ -13,15 +13,6 @@ public class AzureGenie : Agent
_azureService = azureService;
}
public async override Task OnActivateAsync(CancellationToken cancellationToken)
{
var streamProvider = this.GetStreamProvider("StreamProvider");
var streamId = StreamId.Create(Consts.MainNamespace, this.GetPrimaryKeyString());
var stream = streamProvider.GetStream<Event>(streamId);
await stream.SubscribeAsync(HandleEvent);
}
public override async Task HandleEvent(Event item, StreamSequenceToken? token)
{
switch (item.Type)

61
src/libs/Microsoft.AI.DevTeam/Agents/Developer.cs
View File

@ -9,7 +9,6 @@ using Orleans.Streams;
namespace Microsoft.AI.DevTeam;
//[RegexImplicitStreamSubscription("")]
[ImplicitStreamSubscription(Consts.MainNamespace)]
public class Dev : AiAgent
{
@ -24,15 +23,6 @@ public class Dev : AiAgent
_logger = logger;
}
public async override Task OnActivateAsync(CancellationToken cancellationToken)
{
var streamProvider = this.GetStreamProvider("StreamProvider");
var streamId = StreamId.Create(Consts.MainNamespace, this.GetPrimaryKeyString());
var stream = streamProvider.GetStream<Event>(streamId);
await stream.SubscribeAsync(HandleEvent);
}
public async override Task HandleEvent(Event item, StreamSequenceToken? token)
{
switch (item.Type)
@ -82,57 +72,6 @@ public class Dev : AiAgent
}
}
public Task<string> ReviewPlan(string plan)
{
throw new NotImplementedException();
}
public async Task CloseImplementation()
{
// var dev = _grains.GetGrain<IDevelopCode>(issueNumber, suffix);
// var code = await dev.GetLastMessage();
// var lookup = _grains.GetGrain<ILookupMetadata>(suffix);
// var parentIssue = await lookup.GetMetadata((int)issueNumber);
// await _azService.Store(new SaveOutputRequest
// {
// ParentIssueNumber = parentIssue.IssueNumber,
// IssueNumber = (int)issueNumber,
// Output = code,
// Extension = "sh",
// Directory = "output",
// FileName = "run",
// Org = org,
// Repo = repo
// });
// var sandboxRequest = new SandboxRequest
// {
// Org = org,
// Repo = repo,
// IssueNumber = (int)issueNumber,
// ParentIssueNumber = parentIssue.IssueNumber
// };
// await _azService.RunInSandbox(sandboxRequest);
// var commitRequest = new CommitRequest
// {
// Dir = "output",
// Org = org,
// Repo = repo,
// ParentNumber = parentIssue.IssueNumber,
// Number = (int)issueNumber,
// Branch = $"sk-{parentIssue.IssueNumber}"
// };
// var markTaskCompleteRequest = new MarkTaskCompleteRequest
// {
// Org = org,
// Repo = repo,
// CommentId = parentIssue.CommentId
// };
// var sandbox = _grains.GetGrain<IManageSandbox>(issueNumber, suffix);
// await sandbox.ScheduleCommitSandboxRun(commitRequest, markTaskCompleteRequest, sandboxRequest);
}
public async Task<UnderstandingResult> BuildUnderstanding(string content)
{
try

8
src/libs/Microsoft.AI.DevTeam/Agents/DeveloperLead.cs
View File

@ -23,14 +23,6 @@ public class DeveloperLead : AiAgent
_logger = logger;
_ghService = ghService;
}
public async override Task OnActivateAsync(CancellationToken cancellationToken)
{
var streamProvider = this.GetStreamProvider("StreamProvider");
var streamId = StreamId.Create(Consts.MainNamespace, this.GetPrimaryKeyString());
var stream = streamProvider.GetStream<Event>(streamId);
await stream.SubscribeAsync(HandleEvent);
}
public async override Task HandleEvent(Event item, StreamSequenceToken? token)
{

8
src/libs/Microsoft.AI.DevTeam/Agents/Hubber.cs
View File

@ -17,14 +17,6 @@ public class Hubber : Agent
_ghService = ghService;
}
public async override Task OnActivateAsync(CancellationToken cancellationToken)
{
var streamProvider = this.GetStreamProvider("StreamProvider");
var streamId = StreamId.Create(Consts.MainNamespace, this.GetPrimaryKeyString());
var stream = streamProvider.GetStream<Event>(streamId);
await stream.SubscribeAsync(HandleEvent);
}
public override async Task HandleEvent(Event item, StreamSequenceToken? token)
{
switch (item.Type)

42
src/libs/Microsoft.AI.DevTeam/Agents/ProductManager.cs
View File

@ -21,14 +21,6 @@ public class ProductManager : AiAgent
_logger = logger;
}
public async override Task OnActivateAsync(CancellationToken cancellationToken)
{
var streamProvider = this.GetStreamProvider("StreamProvider");
var streamId = StreamId.Create(Consts.MainNamespace, this.GetPrimaryKeyString());
var stream = streamProvider.GetStream<Event>(streamId);
await stream.SubscribeAsync(HandleEvent);
}
public async override Task HandleEvent(Event item, StreamSequenceToken? token)
{
switch (item.Type)
@ -77,38 +69,4 @@ public class ProductManager : AiAgent
return default;
}
}
// public async Task CloseReadme()
// {
// var pm = _grains.GetGrain<IManageProduct>(issueNumber, suffix);
// var readme = await pm.GetLastMessage();
// var lookup = _grains.GetGrain<ILookupMetadata>(suffix);
// var parentIssue = await lookup.GetMetadata((int)issueNumber);
// await _azService.Store(new SaveOutputRequest
// {
// ParentIssueNumber = parentIssue.IssueNumber,
// IssueNumber = (int)issueNumber,
// Output = readme,
// Extension = "md",
// Directory = "output",
// FileName = "readme",
// Org = org,
// Repo = repo
// });
// await _ghService.CommitToBranch(new CommitRequest
// {
// Dir = "output",
// Org = org,
// Repo = repo,
// ParentNumber = parentIssue.IssueNumber,
// Number = (int)issueNumber,
// Branch = $"sk-{parentIssue.IssueNumber}"
// });
// await _ghService.MarkTaskComplete(new MarkTaskCompleteRequest
// {
// Org = org,
// Repo = repo,
// CommentId = parentIssue.CommentId
// });
// }
}