yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-08-13 03:01:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

added retry logic if throttling eventually kills a task.

Browse Source
This commit is contained in:
Ryan Sweet 2023-06-17 06:35:23 +00:00
parent 542e5634fb
commit 0ed02dfb35
1 changed files with 29 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
cli/Program.cs
View File

@ -65,25 +65,47 @@ class Program
var sandboxSkill = new SandboxSkill(); var sandboxSkill = new SandboxSkill();
var outputPath = Directory.CreateDirectory("output"); var outputPath = Directory.CreateDirectory("output");
Console.WriteLine($"Using output directory: {outputPath}");
var readme = await CallWithFile<string>(nameof(PM), PM.Readme , file, maxRetry); var readme = await CallWithFile<string>(nameof(PM), PM.Readme , file, maxRetry);
string readmeFile = Path.Combine(outputPath.FullName, "README.md"); string readmeFile = Path.Combine(outputPath.FullName, "README.md");
await SaveToFile(readmeFile, readme); await SaveToFile(readmeFile, readme);
Console.WriteLine($"Saved README to {readmeFile}");
var script = await CallWithFile<string>(nameof(PM), PM.BootstrapProject, file, maxRetry); var script = await CallWithFile<string>(nameof(PM), PM.BootstrapProject, file, maxRetry);
await sandboxSkill.RunInDotnetAlpineAsync(script); await sandboxSkill.RunInDotnetAlpineAsync(script);
await SaveToFile(Path.Combine(outputPath.FullName, "bootstrap.sh"), script); await SaveToFile(Path.Combine(outputPath.FullName, "bootstrap.sh"), script);
Console.WriteLine($"Saved bootstrap script to {outputPath.FullName}bootstrap.sh");
var plan = await CallWithFile<DevLeadPlanResponse>(nameof(DevLead), DevLead.Plan, readmeFile, maxRetry); var plan = await CallWithFile<DevLeadPlanResponse>(nameof(DevLead), DevLead.Plan, readmeFile, maxRetry);
await SaveToFile(Path.Combine(outputPath.FullName, "plan.json"), JsonSerializer.Serialize(plan)); await SaveToFile(Path.Combine(outputPath.FullName, "plan.json"), JsonSerializer.Serialize(plan));
Console.WriteLine($"Using Plan: \n {plan}");
var implementationTasks = plan.steps.SelectMany( var implementationTasks = plan.steps.SelectMany(
(step) => step.subtasks.Select( (step) => step.subtasks.Select(
async (subtask) => { async (subtask) => {
var implementationResult = await CallFunction<string>(nameof(Developer), Developer.Implement, subtask.LLM_prompt, maxRetry); Console.WriteLine($"Implementing {step.step}-{subtask.subtask}");
//var improvementResult = await CallFunction<string>(nameof(Developer), Developer.Improve, subtask.LLM_prompt, maxRetry); var implementationResult = string.Empty;
await sandboxSkill.RunInDotnetAlpineAsync(implementationResult); while (true)
await SaveToFile(Path.Combine(outputPath.FullName, $"{step.step}-{subtask.subtask}.sh"), implementationResult); {
return implementationResult; })); try
{
implementationResult = await CallFunction<string>(nameof(Developer), Developer.Implement, subtask.LLM_prompt, maxRetry);
break;
}
catch (Exception ex)
{
if (ex.Message.Contains("TooMany"))
{
Console.WriteLine("Throttled, retrying...");
continue;
}
throw;
}
}
await sandboxSkill.RunInDotnetAlpineAsync(implementationResult);
await SaveToFile(Path.Combine(outputPath.FullName, $"{step.step}-{subtask.subtask}.sh"), implementationResult);
return implementationResult; }));
await Task.WhenAll(implementationTasks); await Task.WhenAll(implementationTasks);
} }
@ -132,7 +154,7 @@ class Program
// RetryableExceptions = new[] { typeof(HttpRequestException) } // RetryableExceptions = new[] { typeof(HttpRequestException) }
})) }))
.Build(); .Build();
Console.WriteLine($"Calling skill '{skillName}' function '{functionName}' with input '{input}'"); //Console.WriteLine($"Calling skill '{skillName}' function '{functionName}' with input '{input}'");
var interestingMemories = kernel.Memory.SearchAsync("waf-pages", input, 2); var interestingMemories = kernel.Memory.SearchAsync("waf-pages", input, 2);
var wafContext = "Consider the following architectural guidelines:"; var wafContext = "Consider the following architectural guidelines:";
await foreach (var memory in interestingMemories) await foreach (var memory in interestingMemories)
@ -150,7 +172,7 @@ class Program
var answer = await kernel.RunAsync(context, function).ConfigureAwait(false); var answer = await kernel.RunAsync(context, function).ConfigureAwait(false);
var result = typeof(T) != typeof(string) ? JsonSerializer.Deserialize<T>(answer.ToString()) : (T)(object)answer.ToString(); var result = typeof(T) != typeof(string) ? JsonSerializer.Deserialize<T>(answer.ToString()) : (T)(object)answer.ToString();
Console.WriteLine(answer); //Console.WriteLine(answer);
return result; return result;
} }
} }