| | | 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 WorkflowProposeCreateTool(IServiceProvider serviceProvider, WorkflowDraftValidationService validationServic |
| | | 8 | | { |
| | 91 | 9 | | public override AIToolDefinition Definition { get; } = ProposalDefinition( |
| | 45 | 10 | | "workflows.proposeCreate", |
| | 45 | 11 | | "Propose workflow creation", |
| | 45 | 12 | | "Create a reviewable AI workflow creation proposal. This never persists a workflow definition.", |
| | 45 | 13 | | GroundingToolSchemas.WithProperties( |
| | 45 | 14 | | ("draft", GroundingToolSchemas.Object("Workflow draft JSON payload.")), |
| | 45 | 15 | | ("rationale", GroundingToolSchemas.String("Why the draft should be created.")))); |
| | | 16 | | |
| | | 17 | | public override async ValueTask<AIToolResult> ExecuteAsync(AIToolExecutionContext context, CancellationToken cancell |
| | | 18 | | { |
| | 1 | 19 | | var store = serviceProvider.GetService(typeof(IAIProposalStore)) as IAIProposalStore; |
| | 1 | 20 | | if (store == null) |
| | 0 | 21 | | return formatter.Unavailable("AI proposal store"); |
| | | 22 | | |
| | 1 | 23 | | var draft = GetObject(context.Arguments, "draft") ?? []; |
| | 1 | 24 | | var diagnostics = validationService.Validate(draft); |
| | 1 | 25 | | var proposal = new AIProposal |
| | 1 | 26 | | { |
| | 1 | 27 | | TenantId = context.TenantId, |
| | 1 | 28 | | ConversationId = context.ConversationId, |
| | 1 | 29 | | Kind = AIProposalKind.WorkflowCreate, |
| | 1 | 30 | | Status = diagnostics.Any(x => x.Severity == AIValidationSeverity.Error) ? AIProposalStatus.Blocked : AIPropo |
| | 1 | 31 | | WorkflowPayload = (JsonObject)draft.DeepClone(), |
| | 1 | 32 | | Rationale = GetString(context.Arguments, "rationale") ?? "", |
| | 1 | 33 | | ValidationDiagnostics = diagnostics.ToList(), |
| | 1 | 34 | | GraphDiff = diffService.CreateDiff(draft), |
| | 1 | 35 | | CreatedBy = context.ActorId |
| | 1 | 36 | | }; |
| | 1 | 37 | | await store.SaveAsync(proposal, cancellationToken); |
| | | 38 | | |
| | 1 | 39 | | return formatter.CreateResult($"Created workflow creation proposal {proposal.Id}.", [AIGroundingJson.ToJsonObjec |
| | 1 | 40 | | } |
| | | 41 | | } |