| | | 1 | | using Elsa.AI.Abstractions.Contracts; |
| | | 2 | | using Elsa.AI.Abstractions.Models; |
| | | 3 | | using Elsa.AI.Host.Services; |
| | | 4 | | |
| | | 5 | | namespace Elsa.AI.Host.Tools.Workflows; |
| | | 6 | | |
| | 45 | 7 | | public class WorkflowProposeUpdateTool(IServiceProvider serviceProvider, WorkflowDraftValidationService validationServic |
| | | 8 | | { |
| | 90 | 9 | | public override AIToolDefinition Definition { get; } = ProposalDefinition( |
| | 45 | 10 | | "workflows.proposeUpdate", |
| | 45 | 11 | | "Propose workflow update", |
| | 45 | 12 | | "Create a reviewable AI workflow update proposal against a baseline workflow version. This never persists a work |
| | 45 | 13 | | GroundingToolSchemas.WithProperties( |
| | 45 | 14 | | ("definitionId", GroundingToolSchemas.String("Logical workflow definition ID.")), |
| | 45 | 15 | | ("baselineVersionId", GroundingToolSchemas.String("Baseline workflow version ID.")), |
| | 45 | 16 | | ("draft", GroundingToolSchemas.Object("Workflow draft JSON payload.")), |
| | 45 | 17 | | ("rationale", GroundingToolSchemas.String("Why the draft should update the workflow.")))); |
| | | 18 | | |
| | | 19 | | public override async ValueTask<AIToolResult> ExecuteAsync(AIToolExecutionContext context, CancellationToken cancell |
| | | 20 | | { |
| | 0 | 21 | | var proposalStore = ServiceProvider.GetService(typeof(IAIProposalStore)) as IAIProposalStore; |
| | 0 | 22 | | if (proposalStore == null) |
| | 0 | 23 | | return Formatter.Unavailable("AI proposal store"); |
| | | 24 | | |
| | 0 | 25 | | var workflowStore = WorkflowDefinitionStore; |
| | 0 | 26 | | if (workflowStore == null) |
| | 0 | 27 | | return WorkflowStoreUnavailable(); |
| | | 28 | | |
| | 0 | 29 | | var baselineVersionId = GetString(context.Arguments, "baselineVersionId"); |
| | 0 | 30 | | var baseline = string.IsNullOrWhiteSpace(baselineVersionId) |
| | 0 | 31 | | ? null |
| | 0 | 32 | | : await workflowStore.FindAsync(new() { Id = baselineVersionId }, cancellationToken); |
| | 0 | 33 | | if (baseline == null || !IsTenantAllowed(baseline, context.TenantId)) |
| | 0 | 34 | | return new AIToolResult { Status = AIToolInvocationStatus.Failed, Error = "Baseline workflow definition was |
| | | 35 | | |
| | 0 | 36 | | var latest = await workflowStore.FindAsync(new() { DefinitionId = baseline.DefinitionId, VersionOptions = Elsa.C |
| | 0 | 37 | | var draft = GetObject(context.Arguments, "draft") ?? []; |
| | 0 | 38 | | var diagnostics = validationService.Validate(draft, baselineVersionId, latest?.Id); |
| | 0 | 39 | | var proposal = new AIProposal |
| | 0 | 40 | | { |
| | 0 | 41 | | TenantId = context.TenantId, |
| | 0 | 42 | | ConversationId = context.ConversationId, |
| | 0 | 43 | | Kind = AIProposalKind.WorkflowUpdate, |
| | 0 | 44 | | Status = diagnostics.Any(x => x.Severity == AIValidationSeverity.Error) ? AIProposalStatus.Blocked : AIPropo |
| | 0 | 45 | | BaselineWorkflowDefinitionId = baseline.DefinitionId, |
| | 0 | 46 | | BaselineVersionId = baseline.Id, |
| | 0 | 47 | | WorkflowPayload = (JsonObject)draft.DeepClone(), |
| | 0 | 48 | | Rationale = GetString(context.Arguments, "rationale") ?? "", |
| | 0 | 49 | | ValidationDiagnostics = diagnostics.ToList(), |
| | 0 | 50 | | GraphDiff = diffService.CreateDiff(draft, workflowMapper.MapGraph(baseline)), |
| | 0 | 51 | | CreatedBy = context.ActorId |
| | 0 | 52 | | }; |
| | 0 | 53 | | await proposalStore.SaveAsync(proposal, cancellationToken); |
| | | 54 | | |
| | 0 | 55 | | return Formatter.CreateResult($"Created workflow update proposal {proposal.Id}.", [AIGroundingJson.ToJsonObject( |
| | 0 | 56 | | } |
| | | 57 | | } |