| | | 1 | | using Elsa.AI.Abstractions.Models; |
| | | 2 | | using Elsa.AI.Host.Services; |
| | | 3 | | |
| | | 4 | | namespace Elsa.AI.Host.Tools.Runtime; |
| | | 5 | | |
| | 45 | 6 | | public class WorkflowInstanceActivityStateTool(IServiceProvider serviceProvider, AIGroundingResultFormatter formatter) : |
| | | 7 | | { |
| | 90 | 8 | | public override AIToolDefinition Definition { get; } = ReadOnlyDefinition( |
| | 45 | 9 | | "instances.getActivityState", |
| | 45 | 10 | | "Get instance activity state", |
| | 45 | 11 | | "Get bounded state for a specific activity in a workflow instance.", |
| | 45 | 12 | | GroundingToolSchemas.WithProperties( |
| | 45 | 13 | | ("instanceId", GroundingToolSchemas.String("Workflow instance ID.")), |
| | 45 | 14 | | ("activityId", GroundingToolSchemas.String("Activity ID.")), |
| | 45 | 15 | | ("activityNodeId", GroundingToolSchemas.String("Activity node ID.")))); |
| | | 16 | | |
| | | 17 | | public override async ValueTask<AIToolResult> ExecuteAsync(AIToolExecutionContext context, CancellationToken cancell |
| | | 18 | | { |
| | 0 | 19 | | if (WorkflowInstanceStore == null) |
| | 0 | 20 | | return InstanceStoreUnavailable(); |
| | | 21 | | |
| | 0 | 22 | | var instance = await FindAuthorizedInstanceAsync(context.Arguments, context.TenantId, cancellationToken); |
| | 0 | 23 | | if (instance == null) |
| | 0 | 24 | | return new AIToolResult { Status = AIToolInvocationStatus.Failed, Error = "Workflow instance was not found." |
| | | 25 | | |
| | 0 | 26 | | var activityId = GetString(context.Arguments, "activityId"); |
| | 0 | 27 | | var activityNodeId = GetString(context.Arguments, "activityNodeId"); |
| | 0 | 28 | | var item = new JsonObject |
| | 0 | 29 | | { |
| | 0 | 30 | | ["instanceId"] = instance.Id, |
| | 0 | 31 | | ["activityId"] = activityId, |
| | 0 | 32 | | ["activityNodeId"] = activityNodeId, |
| | 0 | 33 | | ["incidents"] = AIGroundingJson.ToJsonArray(instance.WorkflowState.Incidents |
| | 0 | 34 | | .Where(x => Matches(x.ActivityId, activityId) || Matches(x.ActivityNodeId, activityNodeId)) |
| | 0 | 35 | | .Select(x => new { x.ActivityId, x.ActivityNodeId, x.ActivityType, x.Message, x.Timestamp })), |
| | 0 | 36 | | ["scheduledActivities"] = AIGroundingJson.ToJsonArray(instance.WorkflowState.ScheduledActivities |
| | 0 | 37 | | .Where(x => Matches(x.ActivityNodeId, activityNodeId) || Matches(x.ActivityNodeId, activityId)) |
| | 0 | 38 | | .Select(x => new { x.ActivityNodeId, x.OwnerContextId, x.ExistingActivityExecutionContextId })), |
| | 0 | 39 | | ["bookmarks"] = AIGroundingJson.ToJsonArray(instance.WorkflowState.Bookmarks |
| | 0 | 40 | | .Where(x => Matches(x.ActivityId, activityId)) |
| | 0 | 41 | | .Select(x => new { x.Id, x.Name, x.ActivityId })) |
| | 0 | 42 | | }; |
| | | 43 | | |
| | 0 | 44 | | return Formatter.CreateResult($"Resolved activity state for workflow instance {instance.Id}.", [Formatter.Redact |
| | 0 | 45 | | } |
| | | 46 | | |
| | | 47 | | private static bool Matches(string? value, string? expected) => |
| | 0 | 48 | | !string.IsNullOrWhiteSpace(value) && !string.IsNullOrWhiteSpace(expected) && string.Equals(value, expected, Stri |
| | | 49 | | } |