fix: add Plugins.* to PluginDeclaration.Category()

langgenius · Jan 24, 2025 · fc3cba5 · fc3cba5
1 parent 2f8d22f
commit fc3cba5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/entities/plugin_entities/plugin_declaration.go b/pkg/entities/plugin_entities/plugin_declaration.go
@@ -183,13 +183,13 @@ type PluginDeclaration struct {
 }
 
 func (p *PluginDeclaration) Category() PluginCategory {
-	if p.Tool != nil {
+	if p.Tool != nil || len(p.Plugins.Tools) != 0 {
 		return PLUGIN_CATEGORY_TOOL
 	}
-	if p.Model != nil {
+	if p.Model != nil || len(p.Plugins.Models) != 0 {
 		return PLUGIN_CATEGORY_MODEL
 	}
-	if p.AgentStrategy != nil {
+	if p.AgentStrategy != nil || len(p.Plugins.AgentStrategies) != 0 {
 		return PLUGIN_CATEGORY_AGENT_STRATEGY
 	}
 	return PLUGIN_CATEGORY_EXTENSION