diff --git a/src/Infrastructure/BotSharp.Abstraction/Conversations/IConversationService.cs b/src/Infrastructure/BotSharp.Abstraction/Conversations/IConversationService.cs
index 4eb44ae4..656b397d 100644
--- a/src/Infrastructure/BotSharp.Abstraction/Conversations/IConversationService.cs
+++ b/src/Infrastructure/BotSharp.Abstraction/Conversations/IConversationService.cs
@@ -1,4 +1,3 @@
-using BotSharp.Abstraction.Loggers.Models;
 using BotSharp.Abstraction.Repositories.Filters;
 
 namespace BotSharp.Abstraction.Conversations;
@@ -53,7 +52,7 @@ public interface IConversationService
     /// <returns></returns>
     Task UpdateBreakpoint(bool resetStates = false, string? reason = null, params string[] excludedStates);
 
-    Task<string> GetConversationSummary(IEnumerable<string> conversationId);
+    Task<string> GetConversationSummary(ConversationSummaryModel model);
 
     Task<Conversation> GetConversationRecordOrCreateNew(string agentId);
 
diff --git a/src/Infrastructure/BotSharp.Abstraction/Models/ConversationSummaryModel.cs b/src/Infrastructure/BotSharp.Abstraction/Models/ConversationSummaryModel.cs
new file mode 100644
index 00000000..00860e2e
--- /dev/null
+++ b/src/Infrastructure/BotSharp.Abstraction/Models/ConversationSummaryModel.cs
@@ -0,0 +1,27 @@
+using System.Text.Json.Serialization;
+
+namespace BotSharp.Abstraction.Models;
+
+public class ConversationSummaryModel
+{
+    [JsonPropertyName("conversation_ids")]
+    public IEnumerable<string> ConversationIds { get; set; } = new List<string>();
+
+    private string _agentId;
+
+    [JsonPropertyName("agent_id")]
+    public string AgentId
+    {
+        get => _agentId ?? BuiltInAgentId.AIAssistant;
+        set => _agentId = value;
+    }
+
+    private string _templateName;
+
+    [JsonPropertyName("template_name")]
+    public string TemplateName
+    { 
+        get => _templateName ?? "conversation.summary";
+        set => _templateName = value;
+    }
+}
diff --git a/src/Infrastructure/BotSharp.Core/Conversations/Services/ConversationService.Summary.cs b/src/Infrastructure/BotSharp.Core/Conversations/Services/ConversationService.Summary.cs
index 88d3a52d..ba141e08 100644
--- a/src/Infrastructure/BotSharp.Core/Conversations/Services/ConversationService.Summary.cs
+++ b/src/Infrastructure/BotSharp.Core/Conversations/Services/ConversationService.Summary.cs
@@ -1,20 +1,21 @@
 using BotSharp.Abstraction.Conversations.Enums;
 using BotSharp.Abstraction.MLTasks;
+using BotSharp.Abstraction.Models;
 using BotSharp.Abstraction.Templating;
 
 namespace BotSharp.Core.Conversations.Services;
 
 public partial class ConversationService
 {
-    public async Task<string> GetConversationSummary(IEnumerable<string> conversationIds)
+    public async Task<string> GetConversationSummary(ConversationSummaryModel model)
     {
-        if (conversationIds.IsNullOrEmpty()) return string.Empty;
+        if (model.ConversationIds.IsNullOrEmpty()) return string.Empty;
 
         var routing = _services.GetRequiredService<IRoutingService>();
         var agentService = _services.GetRequiredService<IAgentService>();
 
         var contents = new List<string>();
-        foreach ( var conversationId in conversationIds)
+        foreach (var conversationId in model.ConversationIds)
         {
             if (string.IsNullOrEmpty(conversationId)) continue;
 
@@ -31,16 +32,16 @@ public partial class ConversationService
 
         if (contents.IsNullOrEmpty()) return string.Empty;
 
-        var router = await agentService.LoadAgent(AIAssistant);
-        var prompt = GetPrompt(router, contents);
-        var summary = await Summarize(router, prompt);
+        var agent = await agentService.LoadAgent(model.AgentId);
+        var prompt = GetPrompt(agent, model.TemplateName, contents);
+        var summary = await Summarize(agent, prompt);
 
         return summary;
     }
 
-    private string GetPrompt(Agent agent, List<string> contents)
+    private string GetPrompt(Agent agent, string templateName, List<string> contents)
     {
-        var template = agent.Templates.First(x => x.Name == "conversation.summary").Content;
+        var template = agent.Templates.First(x => x.Name == templateName).Content;
         var render = _services.GetRequiredService<ITemplateRender>();
 
         var texts = new List<string>();
diff --git a/src/Infrastructure/BotSharp.OpenAPI/Controllers/ConversationController.cs b/src/Infrastructure/BotSharp.OpenAPI/Controllers/ConversationController.cs
index bcb92f74..d3da550f 100644
--- a/src/Infrastructure/BotSharp.OpenAPI/Controllers/ConversationController.cs
+++ b/src/Infrastructure/BotSharp.OpenAPI/Controllers/ConversationController.cs
@@ -185,7 +185,7 @@ public class ConversationController : ControllerBase
     public async Task<string> GetConversationSummary([FromBody] ConversationSummaryModel input)
     {
         var service = _services.GetRequiredService<IConversationService>();
-        return await service.GetConversationSummary(input.ConversationIds);
+        return await service.GetConversationSummary(input);
     }
 
     [HttpPut("/conversation/{conversationId}/update-title")]
diff --git a/src/Infrastructure/BotSharp.OpenAPI/ViewModels/Conversations/Request/ConversationSummaryModel.cs b/src/Infrastructure/BotSharp.OpenAPI/ViewModels/Conversations/Request/ConversationSummaryModel.cs
deleted file mode 100644
index 0854ab2a..00000000
--- a/src/Infrastructure/BotSharp.OpenAPI/ViewModels/Conversations/Request/ConversationSummaryModel.cs
+++ /dev/null
@@ -1,9 +0,0 @@
-using System.Text.Json.Serialization;
-
-namespace BotSharp.OpenAPI.ViewModels.Conversations;
-
-public class ConversationSummaryModel
-{
-    [JsonPropertyName("conversation_ids")]
-    public List<string> ConversationIds { get; set; } = new List<string>();
-}