Merge pull request #524 from iceljc/features/add-agent-utility

change to agent utility

Author: Oceania2018

src/Infrastructure/BotSharp.Abstraction/Agents/Enums/AgentField.cs

@@ -17,7 +17,7 @@ public enum AgentField
     Response,
     Sample,
     LlmConfig,
-    Tool
+    Utility
 }
 
 public enum AgentTaskField

src/Infrastructure/BotSharp.Abstraction/Agents/Enums/AgentTool.cs renamed to src/Infrastructure/BotSharp.Abstraction/Agents/Enums/AgentUtility.cs

@@ -1,6 +1,6 @@
 namespace BotSharp.Abstraction.Agents.Enums;
 
-public class AgentTool
+public class AgentUtility
 {
     public const string FileAnalyzer = "file-analyzer";
     public const string ImageGenerator = "image-generator";

src/Infrastructure/BotSharp.Abstraction/Agents/IAgentService.cs

@@ -52,5 +52,5 @@ public interface IAgentService
 
     PluginDef GetPlugin(string agentId);
 
-    IEnumerable<string> GetAgentTools();
+    IEnumerable<string> GetAgentUtilities();
 }

src/Infrastructure/BotSharp.Abstraction/Agents/IAgentToolHook.cs

@@ -0,0 +1,6 @@
+namespace BotSharp.Abstraction.Agents;
+
+public interface IAgentUtilityHook
+{
+    void AddUtilities(List<string> utilities);
+}

src/Infrastructure/BotSharp.Abstraction/Agents/IAgentUtilityHook.cs

@@ -91,9 +91,9 @@ public class Agent
         = new List<string>();
 
     /// <summary>
-    /// Useful tools
+    /// Agent utilities
     /// </summary>
-    public List<string> Tools { get; set; }
+    public List<string> Utilities { get; set; }
         = new List<string>();
 
     /// <summary>
@@ -127,7 +127,7 @@ public static Agent Clone(Agent agent)
             Functions = agent.Functions,
             Responses = agent.Responses,
             Samples = agent.Samples,
-            Tools = agent.Tools,
+            Utilities = agent.Utilities,
             Knowledges = agent.Knowledges,
             IsPublic = agent.IsPublic,
             Disabled = agent.Disabled,
@@ -169,9 +169,9 @@ public Agent SetSamples(List<string> samples)
         return this;
     }
 
-    public Agent SetTools(List<string> tools)
+    public Agent SetUtilities(List<string> utilities)
     {
-        Tools = tools ?? new List<string>();
+        Utilities = utilities ?? new List<string>();
         return this;
     }
 

src/Infrastructure/BotSharp.Abstraction/Agents/Models/Agent.cs

@@ -6,7 +6,6 @@ public class AgentSettings
     public string TemplateFormat { get; set; } = "liquid";
     public string HostAgentId { get; set; } = string.Empty;
     public bool EnableTranslator { get; set; } = false;
-    public bool EnableHttpHandler { get; set; } = false;
 
     /// <summary>
     /// This is the default LLM config for agent

src/Infrastructure/BotSharp.Abstraction/Agents/Settings/AgentSettings.cs

@@ -32,7 +32,7 @@ public async Task UpdateAgent(Agent agent, AgentField updateField)
         record.Templates = agent.Templates ?? new List<AgentTemplate>();
         record.Responses = agent.Responses ?? new List<AgentResponse>();
         record.Samples = agent.Samples ?? new List<string>();
-        record.Tools = agent.Tools ?? new List<string>();
+        record.Utilities = agent.Utilities ?? new List<string>();
         if (agent.LlmConfig != null && !agent.LlmConfig.IsInherit)
         {
             record.LlmConfig = agent.LlmConfig;
@@ -94,7 +94,7 @@ public async Task<string> UpdateAgentFromFile(string id)
                        .SetFunctions(foundAgent.Functions)
                        .SetResponses(foundAgent.Responses)
                        .SetSamples(foundAgent.Samples)
-                       .SetTools(foundAgent.Tools)
+                       .SetUtilities(foundAgent.Utilities)
                        .SetLlmConfig(foundAgent.LlmConfig);
 
             _db.UpdateAgent(clonedAgent, AgentField.All);

src/Infrastructure/BotSharp.Core/Agents/Services/AgentService.UpdateAgent.cs

@@ -55,15 +55,14 @@ public List<Agent> GetAgentsByUser(string userId)
         return agents;
     }
 
-    public IEnumerable<string> GetAgentTools()
+    public IEnumerable<string> GetAgentUtilities()
     {
-        var tools = new List<string>();
-
-        var hooks = _services.GetServices<IAgentToolHook>();
+        var utilities = new List<string>();
+        var hooks = _services.GetServices<IAgentUtilityHook>();
         foreach (var hook in hooks)
         {
-            hook.AddTools(tools);
+            hook.AddUtilities(utilities);
         }
-        return tools.Where(x => !string.IsNullOrWhiteSpace(x)).Distinct().OrderBy(x => x).ToList();
+        return utilities.Where(x => !string.IsNullOrWhiteSpace(x)).Distinct().OrderBy(x => x).ToList();
     }
 }

src/Infrastructure/BotSharp.Core/Agents/Services/AgentService.cs

@@ -18,6 +18,6 @@ public void RegisterDI(IServiceCollection services, IConfiguration config)
         services.AddScoped<IBotSharpFileService, BotSharpFileService>();
 
         services.AddScoped<IAgentHook, FileAnalyzerHook>();
-        services.AddScoped<IAgentToolHook, FileAnalyzerToolHook>();
+        services.AddScoped<IAgentUtilityHook, FileAnalyzerUtilityHook>();
     }
 }

src/Infrastructure/BotSharp.Core/Files/FilePlugin.cs

@@ -12,7 +12,7 @@ public class LoadAttachmentFn : IFunctionCallback
     private readonly ILogger<LoadAttachmentFn> _logger;
     private readonly IEnumerable<string> _imageTypes = new List<string> { "image", "images", "png", "jpg", "jpeg" };
     private readonly IEnumerable<string> _pdfTypes = new List<string> { "pdf" };
-    private static string TOOL_ASSISTANT = Guid.Empty.ToString();
+    private static string UTILITY_ASSISTANT = Guid.Empty.ToString();
 
     public LoadAttachmentFn(
         IServiceProvider services,
@@ -31,7 +31,7 @@ public async Task<bool> Execute(RoleDialogModel message)
         var wholeDialogs = conv.GetDialogHistory();
         var fileTypes = args?.FileTypes?.Split(",", StringSplitOptions.RemoveEmptyEntries)?.ToList() ?? new List<string>();
         var dialogs = await AssembleFiles(conv.ConversationId, wholeDialogs, fileTypes);
-        var agent = await agentService.LoadAgent(TOOL_ASSISTANT);
+        var agent = await agentService.LoadAgent(UTILITY_ASSISTANT);
         var fileAgent = new Agent
         {
             Id = agent?.Id ?? Guid.Empty.ToString(),

src/Infrastructure/BotSharp.Core/Files/Functions/LoadAttachmentFn.cs

@@ -2,7 +2,7 @@ namespace BotSharp.Core.Files.Hooks;
 
 public class FileAnalyzerHook : AgentHookBase
 {
-    private static string TOOL_ASSISTANT = Guid.Empty.ToString();
+    private static string UTILITY_ASSISTANT = Guid.Empty.ToString();
 
     public override string SelfId => string.Empty;
 
@@ -15,7 +15,7 @@ public override void OnAgentLoaded(Agent agent)
     {
         var conv = _services.GetRequiredService<IConversationService>();
         var isConvMode = conv.IsConversationMode();
-        var isEnabled = !agent.Tools.IsNullOrEmpty() && agent.Tools.Contains(AgentTool.FileAnalyzer);
+        var isEnabled = !agent.Utilities.IsNullOrEmpty() && agent.Utilities.Contains(AgentUtility.FileAnalyzer);
 
         if (isConvMode && isEnabled)
         {
@@ -45,7 +45,7 @@ public override void OnAgentLoaded(Agent agent)
     {
         var fn = "load_attachment";
         var db = _services.GetRequiredService<IBotSharpRepository>();
-        var agent = db.GetAgent(TOOL_ASSISTANT);
+        var agent = db.GetAgent(UTILITY_ASSISTANT);
         var prompt = agent?.Templates?.FirstOrDefault(x => x.Name.IsEqualTo($"{fn}.fn"))?.Content ?? string.Empty;
         var loadAttachmentFn = agent?.Functions?.FirstOrDefault(x => x.Name.IsEqualTo(fn));
         return (prompt, loadAttachmentFn);

src/Infrastructure/BotSharp.Core/Files/Hooks/FileAnalyzerHook.cs

@@ -0,0 +1,9 @@
+namespace BotSharp.Core.Files.Hooks;
+
+public class FileAnalyzerUtilityHook : IAgentUtilityHook
+{
+    public void AddUtilities(List<string> utilities)
+    {
+        utilities.Add(AgentUtility.FileAnalyzer);
+    }
+}

src/Infrastructure/BotSharp.Core/Files/Hooks/FileAnalyzerToolHook.cs

@@ -54,8 +54,8 @@ public void UpdateAgent(Agent agent, AgentField field)
                 case AgentField.LlmConfig:
                     UpdateAgentLlmConfig(agent.Id, agent.LlmConfig);
                     break;
-                case AgentField.Tool:
-                    UpdateAgentTools(agent.Id, agent.Tools);
+                case AgentField.Utility:
+                    UpdateAgentUtilities(agent.Id, agent.Utilities);
                     break;
                 case AgentField.All:
                     UpdateAgentAllFields(agent);
@@ -149,14 +149,14 @@ private void UpdateAgentProfiles(string agentId, List<string> profiles)
             File.WriteAllText(agentFile, json);
         }
 
-        private void UpdateAgentTools(string agentId, List<string> tools)
+        private void UpdateAgentUtilities(string agentId, List<string> utilities)
         {
-            if (tools == null) return;
+            if (utilities == null) return;
 
             var (agent, agentFile) = GetAgentFromFile(agentId);
             if (agent == null) return;
 
-            agent.Tools = tools;
+            agent.Utilities = utilities;
             agent.UpdatedDateTime = DateTime.UtcNow;
             var json = JsonSerializer.Serialize(agent, _options);
             File.WriteAllText(agentFile, json);
@@ -301,7 +301,7 @@ private void UpdateAgentAllFields(Agent inputAgent)
             agent.Disabled = inputAgent.Disabled;
             agent.Type = inputAgent.Type;
             agent.Profiles = inputAgent.Profiles;
-            agent.Tools = inputAgent.Tools;
+            agent.Utilities = inputAgent.Utilities;
             agent.RoutingRules = inputAgent.RoutingRules;
             agent.LlmConfig = inputAgent.LlmConfig;
             agent.UpdatedDateTime = DateTime.UtcNow;

src/Infrastructure/BotSharp.Core/Files/Hooks/FileAnalyzerUtilityHook.cs

@@ -1,7 +1,7 @@
 {
   "id": "00000000-0000-0000-0000-000000000000",
-  "name": "Tool Assistant",
-  "description": "Tool assistant that can be used to complete many different tasks",
+  "name": "Utility Assistant",
+  "description": "Utility assistant that can be used to complete many different tasks",
   "type": "static",
   "createdDateTime": "2023-06-24T10:39:32.2349685Z",
   "updatedDateTime": "2023-06-24T14:39:32.2349686Z",

src/Infrastructure/BotSharp.Core/Repository/FileRepository/FileRepository.Agent.cs

@@ -1 +1 @@
-You are a tool agent.
+You are a utility agent.

src/Infrastructure/BotSharp.Core/data/agents/00000000-0000-0000-0000-000000000000/agent.json

@@ -141,9 +141,9 @@ public async Task<bool> DeleteAgent([FromRoute] string agentId)
         return await _agentService.DeleteAgent(agentId);
     }
 
-    [HttpGet("/agent/tools")]
-    public IEnumerable<string> GetAgentTools()
+    [HttpGet("/agent/utilities")]
+    public IEnumerable<string> GetAgentUtilities()
     {
-        return _agentService.GetAgentTools();
+        return _agentService.GetAgentUtilities();
     }
 }

src/Infrastructure/BotSharp.Core/data/agents/00000000-0000-0000-0000-000000000000/instruction.liquid

@@ -1,6 +1,7 @@
 using BotSharp.Abstraction.Agents.Models;
 using BotSharp.Abstraction.Functions.Models;
 using BotSharp.Abstraction.Routing.Models;
+using BotSharp.Core.Infrastructures;
 
 namespace BotSharp.OpenAPI.ViewModels.Agents;
 
@@ -43,7 +44,7 @@ public class AgentCreationModel
     /// Combine different Agents together to form a Profile.
     /// </summary>
     public List<string> Profiles { get; set; } = new List<string>();
-    public List<string> Tools { get; set; } = new List<string>();
+    public List<string> Utilities { get; set; } = new List<string>();
     public List<RoutingRuleUpdateModel> RoutingRules { get; set; } = new List<RoutingRuleUpdateModel>();
     public AgentLlmConfig? LlmConfig { get; set; }
 
@@ -58,7 +59,7 @@ public Agent ToAgent()
             Functions = Functions,
             Responses = Responses,
             Samples = Samples,
-            Tools = Tools,
+            Utilities = Utilities,
             IsPublic = IsPublic,
             Type = Type,
             Disabled = Disabled,

src/Infrastructure/BotSharp.OpenAPI/Controllers/AgentController.cs

@@ -26,9 +26,9 @@ public class AgentUpdateModel
     public List<string>? Samples { get; set; }
 
     /// <summary>
-    /// Tools
+    /// Utilities
     /// </summary>
-    public List<string>? Tools { get; set; }
+    public List<string>? Utilities { get; set; }
 
     /// <summary>
     /// Functions
@@ -76,7 +76,7 @@ public Agent ToAgent()
             Templates = Templates ?? new List<AgentTemplate>(),
             Functions = Functions ?? new List<FunctionDef>(),
             Responses = Responses ?? new List<AgentResponse>(),
-            Tools = Tools ?? new List<string>(),
+            Utilities = Utilities ?? new List<string>(),
             LlmConfig = LlmConfig
         };
 

src/Infrastructure/BotSharp.OpenAPI/ViewModels/Agents/AgentCreationModel.cs

@@ -17,7 +17,7 @@ public class AgentViewModel
     public List<FunctionDef> Functions { get; set; }
     public List<AgentResponse> Responses { get; set; }
     public List<string> Samples { get; set; }
-    public List<string> Tools { get; set; }
+    public List<string> Utilities { get; set; }
 
     [JsonPropertyName("is_public")]
     public bool IsPublic { get; set; }
@@ -64,7 +64,7 @@ public static AgentViewModel FromAgent(Agent agent)
             Functions = agent.Functions,
             Responses = agent.Responses,
             Samples = agent.Samples,
-            Tools = agent.Tools,
+            Utilities = agent.Utilities,
             IsPublic= agent.IsPublic,
             Disabled = agent.Disabled,
             IconUrl = agent.IconUrl,

src/Infrastructure/BotSharp.OpenAPI/ViewModels/Agents/AgentUpdateModel.cs

@@ -1,6 +1,6 @@
 namespace BotSharp.Plugin.HttpHandler.Enums;
 
-public class Tool
+public class Utility
 {
     public const string HttpHandler = "http-handler";
 }

src/Infrastructure/BotSharp.OpenAPI/ViewModels/Agents/AgentViewModel.cs

@@ -8,7 +8,7 @@ namespace BotSharp.Plugin.HttpHandler.Hooks;
 
 public class HttpHandlerHook : AgentHookBase
 {
-    private static string TOOL_ASSISTANT = Guid.Empty.ToString();
+    private static string UTILITY_ASSISTANT = Guid.Empty.ToString();
 
     public override string SelfId => string.Empty;
 
@@ -21,7 +21,7 @@ public override void OnAgentLoaded(Agent agent)
     {
         var conv = _services.GetRequiredService<IConversationService>();
         var isConvMode = conv.IsConversationMode();
-        var isEnabled = !agent.Tools.IsNullOrEmpty() && agent.Tools.Contains(Tool.HttpHandler);
+        var isEnabled = !agent.Utilities.IsNullOrEmpty() && agent.Utilities.Contains(Utility.HttpHandler);
 
         if (isConvMode && isEnabled)
         {
@@ -51,7 +51,7 @@ public override void OnAgentLoaded(Agent agent)
     {
         var fn = "handle_http_request";
         var db = _services.GetRequiredService<IBotSharpRepository>();
-        var agent = db.GetAgent(TOOL_ASSISTANT);
+        var agent = db.GetAgent(UTILITY_ASSISTANT);
         var prompt = agent?.Templates?.FirstOrDefault(x => x.Name.IsEqualTo($"{fn}.fn"))?.Content ?? string.Empty;
         var loadAttachmentFn = agent?.Functions?.FirstOrDefault(x => x.Name.IsEqualTo(fn));
         return (prompt, loadAttachmentFn);

src/Plugins/BotSharp.Plugin.HttpHandler/Enums/Tool.cs renamed to src/Plugins/BotSharp.Plugin.HttpHandler/Enums/Utility.cs

@@ -0,0 +1,12 @@
+using BotSharp.Abstraction.Agents;
+using BotSharp.Plugin.HttpHandler.Enums;
+
+namespace BotSharp.Plugin.HttpHandler.Hooks;
+
+public class HttpHandlerUtilityHook : IAgentUtilityHook
+{
+    public void AddUtilities(List<string> utilities)
+    {
+        utilities.Add(Utility.HttpHandler);
+    }
+}

src/Plugins/BotSharp.Plugin.HttpHandler/Hooks/HttpHandlerHook.cs

@@ -22,6 +22,6 @@ public void RegisterDI(IServiceCollection services, IConfiguration config)
         });
 
         services.AddScoped<IAgentHook, HttpHandlerHook>();
-        services.AddScoped<IAgentToolHook, HttpHandlerToolHook>();
+        services.AddScoped<IAgentUtilityHook, HttpHandlerUtilityHook>();
     }
 }