| | | 1 | | using Elsa.AI.Abstractions.Models; |
| | | 2 | | using Elsa.AI.Host.Services; |
| | | 3 | | using Elsa.AI.Host.Tools; |
| | | 4 | | using Elsa.Workflows; |
| | | 5 | | |
| | | 6 | | namespace Elsa.AI.Host.Tools.Activities; |
| | | 7 | | |
| | 45 | 8 | | public class ActivitiesSearchTool(IServiceProvider serviceProvider, ActivityGroundingSearchService searchService, AIGrou |
| | | 9 | | { |
| | 91 | 10 | | public override AIToolDefinition Definition { get; } = ReadOnlyDefinition( |
| | 45 | 11 | | "activities.search", |
| | 45 | 12 | | "Search activities", |
| | 45 | 13 | | "Search installed Activity Registry descriptors by name, category, type, version, ports, and trigger behavior.", |
| | 45 | 14 | | GroundingToolSchemas.WithProperties( |
| | 45 | 15 | | ("query", GroundingToolSchemas.String("Free-text search term.")), |
| | 45 | 16 | | ("category", GroundingToolSchemas.String("Activity category.")), |
| | 45 | 17 | | ("type", GroundingToolSchemas.String("Activity type name or short name.")), |
| | 45 | 18 | | ("version", GroundingToolSchemas.Integer("Activity version.")), |
| | 45 | 19 | | ("input", GroundingToolSchemas.String("Input name or display name.")), |
| | 45 | 20 | | ("output", GroundingToolSchemas.String("Output name or display name.")), |
| | 45 | 21 | | ("trigger", GroundingToolSchemas.Boolean("Whether the activity can start a workflow.")))); |
| | | 22 | | |
| | | 23 | | public override ValueTask<AIToolResult> ExecuteAsync(AIToolExecutionContext context, CancellationToken cancellationT |
| | | 24 | | { |
| | 1 | 25 | | var registry = serviceProvider.GetService(typeof(IActivityRegistry)) as IActivityRegistry; |
| | 1 | 26 | | if (registry == null) |
| | 0 | 27 | | return ValueTask.FromResult(formatter.Unavailable("Activity Registry")); |
| | | 28 | | |
| | 1 | 29 | | var descriptors = searchService.Search( |
| | 1 | 30 | | registry, |
| | 1 | 31 | | GetString(context.Arguments, "query"), |
| | 1 | 32 | | GetString(context.Arguments, "category"), |
| | 1 | 33 | | GetString(context.Arguments, "type"), |
| | 1 | 34 | | GetInt(context.Arguments, "version"), |
| | 1 | 35 | | GetString(context.Arguments, "input"), |
| | 1 | 36 | | GetString(context.Arguments, "output"), |
| | 1 | 37 | | GetBool(context.Arguments, "trigger")); |
| | 1 | 38 | | var items = descriptors.Select(searchService.Map); |
| | | 39 | | |
| | 1 | 40 | | return ValueTask.FromResult(formatter.CreateResult($"Found {descriptors.Count} installed activities.", items, de |
| | | 41 | | } |
| | | 42 | | } |