local client by default

microsoft · Oct 8, 2024 · 8b12b77 · 8b12b77
1 parent 08ee08f
commit 8b12b77
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 9 deletions.
diff --git a/dotnet/src/Microsoft.AutoGen.Agents.Client/App.cs b/dotnet/src/Microsoft.AutoGen.Agents.Client/App.cs
@@ -7,7 +7,7 @@ public static class App
     public static async Task<WebApplication> StartAsync<T>(string name) where T : AgentBase
     {
         var clientBuilder = WebApplication.CreateBuilder();
-        clientBuilder.AddLocalAgentWorker().AddAgent<T>(name);
+        clientBuilder.AddAgentWorker().AddAgent<T>(name);
         var clientApp = clientBuilder.Build();
         await clientApp.StartAsync().ConfigureAwait(false);
         return clientApp;

diff --git a/dotnet/src/Microsoft.AutoGen.Agents.Client/HostBuilderExtensions.cs b/dotnet/src/Microsoft.AutoGen.Agents.Client/HostBuilderExtensions.cs
@@ -14,14 +14,8 @@ namespace Microsoft.AutoGen.Agents.Client;
 
 public static class HostBuilderExtensions
 {
-    public static AgentApplicationBuilder AddLocalAgentWorker(this IHostApplicationBuilder clientBuilder)
-    {
-        clientBuilder.Services.AddHostedService<AgentWorkerRuntime>();
-        clientBuilder.Services.AddSingleton<AgentClient>();
-        clientBuilder.AddAgentWorker("https://localhost:5001");
-        return new AgentApplicationBuilder(clientBuilder);
-    }
-    public static AgentApplicationBuilder AddAgentWorker(this IHostApplicationBuilder builder, string agentServiceAddress)
+    private const string _defaultAgentServiceAddress = "https://localhost:5001";
+    public static AgentApplicationBuilder AddAgentWorker(this IHostApplicationBuilder builder, string agentServiceAddress =_defaultAgentServiceAddress, bool local = false)
     {
         builder.Services.AddGrpcClient<AgentRpc.AgentRpcClient>(options =>
         {
@@ -55,6 +49,7 @@ public static AgentApplicationBuilder AddAgentWorker(this IHostApplicationBuilde
             });
         });
         builder.Services.TryAddSingleton(DistributedContextPropagator.Current);
+        builder.Services.AddSingleton<AgentClient>();
         builder.Services.AddSingleton<AgentWorkerRuntime>();
         builder.Services.AddSingleton<IHostedService>(sp => sp.GetRequiredService<AgentWorkerRuntime>());
         builder.Services.AddKeyedSingleton("EventTypes", (sp, key) =>