| | | 1 | | using Elsa.AI.Abstractions.Models; |
| | | 2 | | using Elsa.AI.Host.Services; |
| | | 3 | | using Elsa.AI.Host.Tools; |
| | | 4 | | using Elsa.Common.Models; |
| | | 5 | | using Elsa.Workflows.Management; |
| | | 6 | | using Elsa.Workflows.Management.Entities; |
| | | 7 | | using Elsa.Workflows.Management.Filters; |
| | | 8 | | |
| | | 9 | | namespace Elsa.AI.Host.Tools.Workflows; |
| | | 10 | | |
| | 226 | 11 | | public abstract class WorkflowToolBase(IServiceProvider serviceProvider, AIGroundingResultFormatter formatter) : Groundi |
| | | 12 | | { |
| | 228 | 13 | | protected IServiceProvider ServiceProvider { get; } = serviceProvider; |
| | 228 | 14 | | protected AIGroundingResultFormatter Formatter { get; } = formatter; |
| | 2 | 15 | | protected IWorkflowDefinitionStore? WorkflowDefinitionStore => ServiceProvider.GetService(typeof(IWorkflowDefinition |
| | | 16 | | |
| | | 17 | | protected AIToolResult WorkflowStoreUnavailable() => |
| | 0 | 18 | | Formatter.Unavailable("Workflow definition store"); |
| | | 19 | | |
| | | 20 | | protected static bool IsTenantAllowed(WorkflowDefinition definition, string? tenantId) => |
| | 2 | 21 | | string.Equals(NormalizeTenant(definition.TenantId), NormalizeTenant(tenantId), StringComparison.Ordinal); |
| | | 22 | | |
| | | 23 | | protected static WorkflowDefinitionFilter CreateDefinitionFilter(JsonObject arguments) |
| | | 24 | | { |
| | 2 | 25 | | var filter = new WorkflowDefinitionFilter |
| | 2 | 26 | | { |
| | 2 | 27 | | Id = GetString(arguments, "versionId") ?? GetString(arguments, "id"), |
| | 2 | 28 | | DefinitionId = GetString(arguments, "definitionId"), |
| | 2 | 29 | | SearchTerm = GetString(arguments, "query") ?? GetString(arguments, "searchTerm"), |
| | 2 | 30 | | Name = GetString(arguments, "name") |
| | 2 | 31 | | }; |
| | | 32 | | |
| | 2 | 33 | | var version = GetInt(arguments, "version"); |
| | 2 | 34 | | if (version != null) |
| | 0 | 35 | | filter.VersionOptions = VersionOptions.SpecificVersion(version.Value); |
| | 2 | 36 | | else if (GetBool(arguments, "published") == true) |
| | 0 | 37 | | filter.VersionOptions = VersionOptions.Published; |
| | | 38 | | else |
| | 2 | 39 | | filter.VersionOptions = VersionOptions.Latest; |
| | | 40 | | |
| | 2 | 41 | | return filter; |
| | | 42 | | } |
| | | 43 | | |
| | | 44 | | protected static string NormalizeTenant(string? tenantId) => |
| | 4 | 45 | | string.IsNullOrWhiteSpace(tenantId) ? "" : tenantId; |
| | | 46 | | } |