Merge branch 'feature/azure-ai-inference-client' of github.com:rohant…

…hacker/autogen into feature/azure-ai-inference-client

README.md

@@ -56,7 +56,7 @@ Currently, there are three main APIs your application can target:
 
 - [Core](https://microsoft.github.io/autogen/dev/user-guide/core-user-guide/index.html)
 - [AgentChat](https://microsoft.github.io/autogen/dev/user-guide/agentchat-user-guide/index.html)
-- [Extensions](https://microsoft.github.io/autogen/dev/reference/python/autogen_ext/autogen_ext.html)
+- [Extensions](https://microsoft.github.io/autogen/dev/user-guide/extensions-user-guide/index.html)
 
 ## Core
 
@@ -186,9 +186,9 @@ await app.WaitForShutdownAsync();
 
 [TopicSubscription("agents")]
 public class HelloAgent(
-    IAgentContext context,
+    IAgentContext worker,
     [FromKeyedServices("EventTypes")] EventTypes typeRegistry) : ConsoleAgent(
-        context,
+        worker,
         typeRegistry),
         ISayHello,
         IHandle<NewMessageReceived>,

dotnet/samples/Hello/HelloAIAgents/HelloAIAgent.cs

@@ -8,11 +8,11 @@
 namespace Hello;
 [TopicSubscription("agents")]
 public class HelloAIAgent(
-    IAgentRuntime context,
+    IAgentWorker worker,
     [FromKeyedServices("EventTypes")] EventTypes typeRegistry,
     IHostApplicationLifetime hostApplicationLifetime,
     IChatClient client) : HelloAgent(
-        context,
+        worker,
         typeRegistry,
         hostApplicationLifetime),
         IHandle<NewMessageReceived>

dotnet/samples/Hello/HelloAIAgents/Program.cs

@@ -33,10 +33,10 @@ namespace Hello
 {
     [TopicSubscription("agents")]
     public class HelloAgent(
-        IAgentRuntime context,
+        IAgentWorker worker,
         [FromKeyedServices("EventTypes")] EventTypes typeRegistry,
         IHostApplicationLifetime hostApplicationLifetime) : ConsoleAgent(
-            context,
+            worker,
             typeRegistry),
             ISayHello,
             IHandle<NewMessageReceived>,

dotnet/samples/Hello/HelloAgent/Program.cs

@@ -18,9 +18,9 @@ namespace Hello
 {
     [TopicSubscription("agents")]
     public class HelloAgent(
-        IAgentRuntime context, IHostApplicationLifetime hostApplicationLifetime,
+        IAgentWorker worker, IHostApplicationLifetime hostApplicationLifetime,
         [FromKeyedServices("EventTypes")] EventTypes typeRegistry) : Agent(
-            context,
+            worker,
             typeRegistry),
             ISayHello,
             IHandleConsole,
@@ -35,7 +35,7 @@ public async Task Handle(NewMessageReceived item)
             await PublishMessageAsync(evt).ConfigureAwait(false);
             var goodbye = new ConversationClosed
             {
-                UserId = this.AgentId.Key,
+                UserId = this.AgentId.Type,
                 UserMessage = "Goodbye"
             };
             await PublishMessageAsync(goodbye).ConfigureAwait(false);

dotnet/samples/Hello/HelloAgent/README.md

@@ -43,9 +43,9 @@ Within that event handler you may optionally *emit* new events, which are then s
 ```csharp
 TopicSubscription("HelloAgents")]
 public class HelloAgent(
-    IAgentContext context,
+    iAgentWorker worker,
     [FromKeyedServices("EventTypes")] EventTypes typeRegistry) : ConsoleAgent(
-        context,
+        worker,
         typeRegistry),
         ISayHello,
         IHandle<NewMessageReceived>,

dotnet/samples/Hello/HelloAgentState/Program.cs

@@ -18,10 +18,10 @@ namespace Hello
 {
     [TopicSubscription("agents")]
     public class HelloAgent(
-        IAgentRuntime context,
+        IAgentWorker worker,
         IHostApplicationLifetime hostApplicationLifetime,
         [FromKeyedServices("EventTypes")] EventTypes typeRegistry) : Agent(
-            context,
+            worker,
             typeRegistry),
             IHandleConsole,
             IHandle<NewMessageReceived>,

dotnet/samples/Hello/HelloAgentState/README.md

@@ -43,9 +43,9 @@ Within that event handler you may optionally *emit* new events, which are then s
 ```csharp
 TopicSubscription("HelloAgents")]
 public class HelloAgent(
-    IAgentContext context,
+    iAgentWorker worker,
     [FromKeyedServices("EventTypes")] EventTypes typeRegistry) : ConsoleAgent(
-        context,
+        worker,
         typeRegistry),
         ISayHello,
         IHandle<NewMessageReceived>,

dotnet/samples/dev-team/DevTeam.Agents/Developer/Developer.cs

@@ -11,8 +11,8 @@
 namespace DevTeam.Agents;
 
 [TopicSubscription("devteam")]
-public class Dev(IAgentRuntime context, Kernel kernel, ISemanticTextMemory memory, [FromKeyedServices("EventTypes")] EventTypes typeRegistry, ILogger<Dev> logger)
-    : SKAiAgent<DeveloperState>(context, memory, kernel, typeRegistry), IDevelopApps,
+public class Dev(IAgentWorker worker, Kernel kernel, ISemanticTextMemory memory, [FromKeyedServices("EventTypes")] EventTypes typeRegistry, ILogger<Dev> logger)
+    : SKAiAgent<DeveloperState>(worker, memory, kernel, typeRegistry), IDevelopApps,
     IHandle<CodeGenerationRequested>,
     IHandle<CodeChainClosed>
 {

dotnet/samples/dev-team/DevTeam.Agents/DeveloperLead/DeveloperLead.cs

@@ -12,8 +12,8 @@
 namespace DevTeam.Agents;
 
 [TopicSubscription("devteam")]
-public class DeveloperLead(IAgentRuntime context, Kernel kernel, ISemanticTextMemory memory, [FromKeyedServices("EventTypes")] EventTypes typeRegistry, ILogger<DeveloperLead> logger)
-    : SKAiAgent<DeveloperLeadState>(context, memory, kernel, typeRegistry), ILeadDevelopers,
+public class DeveloperLead(IAgentWorker worker, Kernel kernel, ISemanticTextMemory memory, [FromKeyedServices("EventTypes")] EventTypes typeRegistry, ILogger<DeveloperLead> logger)
+    : SKAiAgent<DeveloperLeadState>(worker, memory, kernel, typeRegistry), ILeadDevelopers,
     IHandle<DevPlanRequested>,
     IHandle<DevPlanChainClosed>
 {

dotnet/samples/dev-team/DevTeam.Agents/ProductManager/ProductManager.cs

@@ -11,8 +11,8 @@
 namespace DevTeam.Agents;
 
 [TopicSubscription("devteam")]
-public class ProductManager(IAgentRuntime context, Kernel kernel, ISemanticTextMemory memory, [FromKeyedServices("EventTypes")] EventTypes typeRegistry, ILogger<ProductManager> logger)
-    : SKAiAgent<ProductManagerState>(context, memory, kernel, typeRegistry), IManageProducts,
+public class ProductManager(IAgentWorker worker, Kernel kernel, ISemanticTextMemory memory, [FromKeyedServices("EventTypes")] EventTypes typeRegistry, ILogger<ProductManager> logger)
+    : SKAiAgent<ProductManagerState>(worker, memory, kernel, typeRegistry), IManageProducts,
     IHandle<ReadmeChainClosed>,
     IHandle<ReadmeRequested>
 {

dotnet/samples/dev-team/DevTeam.Backend/Agents/AzureGenie.cs

@@ -9,8 +9,8 @@
 using Microsoft.SemanticKernel.Memory;
 namespace Microsoft.AI.DevTeam;
 
-public class AzureGenie(IAgentRuntime context, Kernel kernel, ISemanticTextMemory memory, [FromKeyedServices("EventTypes")] EventTypes typeRegistry, IManageAzure azureService)
-    : SKAiAgent<object>(context, memory, kernel, typeRegistry),
+public class AzureGenie(IAgentWorker worker, Kernel kernel, ISemanticTextMemory memory, [FromKeyedServices("EventTypes")] EventTypes typeRegistry, IManageAzure azureService)
+    : SKAiAgent<object>(worker, memory, kernel, typeRegistry),
     IHandle<ReadmeCreated>,
     IHandle<CodeCreated>
 

dotnet/samples/dev-team/DevTeam.Backend/Agents/Hubber.cs

@@ -12,8 +12,8 @@
 
 namespace Microsoft.AI.DevTeam;
 
-public class Hubber(IAgentRuntime context, Kernel kernel, ISemanticTextMemory memory, [FromKeyedServices("EventTypes")] EventTypes typeRegistry, IManageGithub ghService)
-    : SKAiAgent<object>(context, memory, kernel, typeRegistry),
+public class Hubber(IAgentWorker worker, Kernel kernel, ISemanticTextMemory memory, [FromKeyedServices("EventTypes")] EventTypes typeRegistry, IManageGithub ghService)
+    : SKAiAgent<object>(worker, memory, kernel, typeRegistry),
     IHandle<NewAsk>,
     IHandle<ReadmeGenerated>,
     IHandle<DevPlanGenerated>,

dotnet/src/Microsoft.AutoGen/Agents/AIAgent/InferenceAgent.cs

@@ -5,10 +5,10 @@
 using Microsoft.Extensions.AI;
 namespace Microsoft.AutoGen.Agents;
 public abstract class InferenceAgent<T>(
-    IAgentRuntime context,
+    IAgentWorker worker,
     EventTypes typeRegistry,
     IChatClient client)
-    : Agent(context, typeRegistry)
+    : Agent(worker, typeRegistry)
     where T : IMessage, new()
 {
     protected IChatClient ChatClient { get; } = client;

dotnet/src/Microsoft.AutoGen/Agents/AIAgent/SKAiAgent.cs

@@ -8,18 +8,17 @@
 using Microsoft.SemanticKernel.Memory;
 
 namespace Microsoft.AutoGen.Agents;
-public abstract class SKAiAgent<T> : Agent where T : class, new()
+public abstract class SKAiAgent<T>(
+    IAgentWorker worker,
+    ISemanticTextMemory memory,
+    Kernel kernel,
+    EventTypes typeRegistry) : Agent(
+        worker,
+        typeRegistry) where T : class, new()
 {
-    protected AgentState<T> _state;
-    protected Kernel _kernel;
-    private readonly ISemanticTextMemory _memory;
-
-    public SKAiAgent(IAgentRuntime context, ISemanticTextMemory memory, Kernel kernel, EventTypes typeRegistry) : base(context, typeRegistry)
-    {
-        _state = new();
-        _memory = memory;
-        _kernel = kernel;
-    }
+    protected AgentState<T> _state = new();
+    protected Kernel _kernel = kernel;
+    private readonly ISemanticTextMemory _memory = memory;
 
     public void AddToHistory(string message, ChatUserType userType) => _state.History.Add(new ChatHistoryItem
     {

dotnet/src/Microsoft.AutoGen/Agents/IOAgent/ConsoleAgent/ConsoleAgent.cs

@@ -13,7 +13,7 @@ public abstract class ConsoleAgent : IOAgent,
 {
 
     // instead of the primary constructor above, make a constructr here that still calls the base constructor
-    public ConsoleAgent(IAgentRuntime context, [FromKeyedServices("EventTypes")] EventTypes typeRegistry) : base(context, typeRegistry)
+    public ConsoleAgent(IAgentWorker worker, [FromKeyedServices("EventTypes")] EventTypes typeRegistry) : base(worker, typeRegistry)
     {
         _route = "console";
     }

dotnet/src/Microsoft.AutoGen/Agents/IOAgent/FileAgent/FileAgent.cs

@@ -9,11 +9,11 @@ namespace Microsoft.AutoGen.Agents;
 
 [TopicSubscription("FileIO")]
 public abstract class FileAgent(
-    IAgentRuntime context,
+    IAgentWorker worker,
     [FromKeyedServices("EventTypes")] EventTypes typeRegistry,
     string inputPath = "input.txt",
     string outputPath = "output.txt"
-    ) : IOAgent(context, typeRegistry),
+    ) : IOAgent(worker, typeRegistry),
         IUseFiles,
         IHandle<Input>,
         IHandle<Output>

dotnet/src/Microsoft.AutoGen/Agents/IOAgent/IOAgent.cs

@@ -5,12 +5,10 @@
 using Microsoft.AutoGen.Core;
 namespace Microsoft.AutoGen.Agents;
 
-public abstract class IOAgent : Agent
+public abstract class IOAgent(IAgentWorker worker, EventTypes eventTypes) : Agent(worker, eventTypes)
 {
     public string _route = "base";
-    protected IOAgent(IAgentRuntime context, EventTypes eventTypes) : base(context, eventTypes)
-    {
-    }
+
     public virtual async Task Handle(Input item)
     {
 

dotnet/src/Microsoft.AutoGen/Agents/IOAgent/WebAPIAgent/WebAPIAgent.cs

@@ -17,11 +17,11 @@ public abstract class WebAPIAgent : IOAgent,
     private readonly string _url = "/agents/webio";
 
     public WebAPIAgent(
-    IAgentRuntime context,
+    IAgentWorker worker,
     [FromKeyedServices("EventTypes")] EventTypes typeRegistry,
     ILogger<WebAPIAgent> logger,
     string url = "/agents/webio") : base(
-        context,
+        worker,
         typeRegistry)
     {
         _url = url;

dotnet/src/Microsoft.AutoGen/Core.Grpc/GrpcAgentWorker.cs

@@ -2,7 +2,6 @@
 // GrpcAgentWorker.cs
 
 using System.Collections.Concurrent;
-using System.Diagnostics;
 using System.Reflection;
 using System.Threading.Channels;
 using Grpc.Core;
@@ -18,8 +17,7 @@ public sealed class GrpcAgentWorker(
     IHostApplicationLifetime hostApplicationLifetime,
     IServiceProvider serviceProvider,
     [FromKeyedServices("AgentTypes")] IEnumerable<Tuple<string, Type>> configuredAgentTypes,
-    ILogger<GrpcAgentWorker> logger,
-    DistributedContextPropagator distributedContextPropagator) :
+    ILogger<GrpcAgentWorker> logger) :
     IHostedService, IDisposable, IAgentWorker
 {
     private readonly object _channelLock = new();
@@ -37,7 +35,6 @@ public sealed class GrpcAgentWorker(
     private readonly IServiceProvider _serviceProvider = serviceProvider;
     private readonly IEnumerable<Tuple<string, Type>> _configuredAgentTypes = configuredAgentTypes;
     private readonly ILogger<GrpcAgentWorker> _logger = logger;
-    private readonly DistributedContextPropagator _distributedContextPropagator = distributedContextPropagator;
     private readonly CancellationTokenSource _shutdownCts = CancellationTokenSource.CreateLinkedTokenSource(hostApplicationLifetime.ApplicationStopping);
     private AsyncDuplexStreamingCall<Message, Message>? _channel;
     private Task? _readTask;
@@ -178,8 +175,7 @@ private Agent GetOrActivateAgent(AgentId agentId)
         {
             if (_agentTypes.TryGetValue(agentId.Type, out var agentType))
             {
-                var context = new AgentRuntime(agentId, this, _serviceProvider.GetRequiredService<ILogger<Agent>>(), _distributedContextPropagator);
-                agent = (Agent)ActivatorUtilities.CreateInstance(_serviceProvider, agentType, context);
+                agent = (Agent)ActivatorUtilities.CreateInstance(_serviceProvider, agentType, this);
                 _agents.TryAdd((agentId.Type, agentId.Key), agent);
             }
             else