SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Support rendering submenu.

Browse source
This commit is contained in:
Haiping Chen 2024-01-15 15:35:01 -06:00
parent 6286393242
commit 9d698fc79d
9 changed files with 70 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/Infrastructure/BotSharp.Abstraction/Plugins/IBotSharpPlugin.cs
View file

@ -25,5 +25,5 @@ public interface IBotSharpPlugin
void RegisterDI(IServiceCollection services, IConfiguration config);
PluginMenuDef[] GetMenus() => new PluginMenuDef[0];
bool AttachMenu(List<PluginMenuDef> menu) => true;
}

5
src/Infrastructure/BotSharp.Abstraction/Plugins/Models/PluginDef.cs
View file

@ -6,6 +6,9 @@ public class PluginDef
public string Name { get; set; }
public string Description { get; set; }
public string Assembly { get; set; }
[JsonPropertyName("is_core")]
public bool IsCore => Assembly == "BotSharp.Core";
[JsonPropertyName("icon_url")]
public string? IconUrl { get; set; }
@ -17,6 +20,4 @@ public class PluginDef
public IBotSharpPlugin Module { get; set; }
public bool Enabled { get; set; }
public PluginMenuDef[] Menus { get; set; }
}

24
src/Infrastructure/BotSharp.Abstraction/Plugins/Models/PluginMenuDef.cs
View file

@ -2,18 +2,36 @@ namespace BotSharp.Abstraction.Plugins.Models;
public class PluginMenuDef
{
[JsonIgnore]
public string Id { get; set; }
public string Label { get; set; }
[JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
public string? Icon { get; set; }
[JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
public string? Link { get; set; }
public bool IsHeader { get; set; }
[JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
public bool? IsHeader { get; set; }
[JsonIgnore]
public int Weight { get; set; }
public PluginMenuDef(string lable, string? link = null, string? icon = null, int weight = 0, bool isHeader = false)
[JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
public List<PluginMenuDef>? SubMenu { get; set; }
public PluginMenuDef(string lable, string? link = null, string? icon = null, int weight = 0)
{
Label = lable;
Link = link;
Icon = icon;
Weight = weight;
IsHeader = isHeader;
}
public override string ToString()
{
return $"{Label} {Link} {Weight}";
}
}

17
src/Infrastructure/BotSharp.Core/Agents/AgentPlugin.cs
View file

@ -1,4 +1,5 @@
using BotSharp.Abstraction.MLTasks;
using BotSharp.Abstraction.Plugins.Models;
using BotSharp.Abstraction.Settings;
using Microsoft.Extensions.Configuration;
@ -26,4 +27,20 @@ public class AgentPlugin : IBotSharpPlugin
return settingService.Bind<AgentSettings>("Agent");
});
}
public bool AttachMenu(List<PluginMenuDef> menu)
{
var section = menu.First(x => x.Label == "Apps");
menu.Add(new PluginMenuDef("Agent", icon: "bx bx-bot", weight: section.Weight + 1)
{
SubMenu = new List<PluginMenuDef>
{
new PluginMenuDef("Router", link: "/page/agent/router"), // icon: "bx bx-map-pin"
new PluginMenuDef("Evaluator", link: "/page/agent/evaluator"), // icon: "bx bx-task"
new PluginMenuDef("Agents", link: "/page/agent"), // icon: "bx bx-bot"
}
});
return true;
}
}

8
src/Infrastructure/BotSharp.Core/Conversations/ConversationPlugin.cs
View file

@ -1,6 +1,7 @@
using BotSharp.Abstraction.Instructs;
using BotSharp.Abstraction.Messaging;
using BotSharp.Abstraction.Planning;
using BotSharp.Abstraction.Plugins.Models;
using BotSharp.Abstraction.Settings;
using BotSharp.Abstraction.Templating;
using BotSharp.Core.Instructs;
@ -45,4 +46,11 @@ public class ConversationPlugin : IBotSharpPlugin
services.AddScoped<IInstructService, InstructService>();
services.AddScoped<ITokenStatistics, TokenStatistics>();
}
public bool AttachMenu(List<PluginMenuDef> menu)
{
var section = menu.First(x => x.Label == "Apps");
menu.Add(new PluginMenuDef("Conversation", link: "/page/conversation", icon: "bx bx-conversation", weight: section.Weight + 5));
return true;
}
}

5
src/Infrastructure/BotSharp.Core/Plugins/PluginLoader.cs
View file

@ -57,8 +57,7 @@ public class PluginLoader
Description = module.Description,
Assembly = assemblyName,
IconUrl = module.IconUrl,
AgentIds = module.AgentIds,
Menus = module.GetMenus(),
AgentIds = module.AgentIds
});
Console.Write($"Loaded plugin ");
Console.Write(name, Color.Green);
@ -85,7 +84,7 @@ public class PluginLoader
var config = db.GetPluginConfig();
foreach (var plugin in _plugins)
{
plugin.Enabled = config.EnabledPlugins.Contains(plugin.Id);
plugin.Enabled = plugin.IsCore || config.EnabledPlugins.Contains(plugin.Id);
}
return _plugins;
}

24
src/Infrastructure/BotSharp.OpenAPI/Controllers/PluginController.cs
View file

@ -26,19 +26,21 @@ public class PluginController : ControllerBase
[HttpGet("/plugin/menu")]
public List<PluginMenuDef> GetPluginMenu()
{
var menus = new List<PluginMenuDef>
var menu = new List<PluginMenuDef>
{
new PluginMenuDef("Dashboard", link: "/page/dashboard", icon: "bx bx-home-circle", weight: 1),
new PluginMenuDef("Agent", isHeader: true, weight: 5),
new PluginMenuDef("Router", link: "/page/agent/router", icon: "bx bx-map-pin", weight: 6),
// new PluginMenuDef("Evaluator", link: "/page/agent/evaluator", icon: "bx bx-task", weight: 7),
new PluginMenuDef("Agents", link: "/page/agent", icon: "bx bx-bot", weight: 8),
new PluginMenuDef("Conversation", isHeader: true, weight: 15),
new PluginMenuDef("Conversations", link: "/page/conversation", icon: "bx bx-conversation", weight: 16),
new PluginMenuDef("System", isHeader: true, weight: 30),
new PluginMenuDef("Apps", weight: 5)
{
IsHeader = true,
},
new PluginMenuDef("System", weight: 30)
{
IsHeader = true
},
new PluginMenuDef("Plugins", link: "/page/plugin", icon: "bx bx-plug", weight: 31),
new PluginMenuDef("Settings", link: "/page/setting", icon: "bx bx-cog", weight: 32),
};
var loader = _services.GetRequiredService<PluginLoader>();
foreach (var plugin in loader.GetPlugins(_services))
{
@ -46,10 +48,10 @@ public class PluginController : ControllerBase
{
continue;
}
menus.AddRange(plugin.Menus);
plugin.Module.AttachMenu(menu);
}
menus = menus.OrderBy(x => x.Weight).ToList();
return menus;
menu = menu.OrderBy(x => x.Weight).ToList();
return menu;
}
[HttpPost("/plugin/{id}/enable")]

10
src/Plugins/BotSharp.Plugin.KnowledgeBase/KnowledgeBasePlugin.cs
View file

@ -27,12 +27,10 @@ public class KnowledgeBasePlugin : IBotSharpPlugin
services.AddSingleton<IPdf2TextConverter, PigPdf2TextConverter>();
}
public PluginMenuDef[] GetMenus()
public bool AttachMenu(List<PluginMenuDef> menu)
{
return new PluginMenuDef[]
{
new PluginMenuDef("RAG", isHeader: true, weight: 20),
new PluginMenuDef("Knowledge Base", link: "/page/knowledge-base", icon: "bx bx-book-open", weight: 21),
};
var section = menu.First(x => x.Label == "Apps");
menu.Add(new PluginMenuDef("Knowledge Base", link: "/page/knowledge-base", icon: "bx bx-book-open", weight: section.Weight + 1));
return true;
}
}