From b9129ea1a6dedc1b5d3c92d3a4085f75fdd7cbef Mon Sep 17 00:00:00 2001
From: Ionite <dev@ionite.io>
Date: Wed, 25 Oct 2023 14:50:05 -0400
Subject: [PATCH] Move IComfyStep to models

---
 StabilityMatrix.Avalonia/Models/Inference/IComfyStep.cs  | 6 ++++++
 .../Models/Inference/ModuleApplyStepEventArgs.cs         | 3 +++
 StabilityMatrix.Avalonia/ViewModels/Base/IComfyStep.cs   | 9 ---------
 3 files changed, 9 insertions(+), 9 deletions(-)
 create mode 100644 StabilityMatrix.Avalonia/Models/Inference/IComfyStep.cs
 delete mode 100644 StabilityMatrix.Avalonia/ViewModels/Base/IComfyStep.cs

diff --git a/StabilityMatrix.Avalonia/Models/Inference/IComfyStep.cs b/StabilityMatrix.Avalonia/Models/Inference/IComfyStep.cs
new file mode 100644
index 00000000..60b3e466
--- /dev/null
+++ b/StabilityMatrix.Avalonia/Models/Inference/IComfyStep.cs
@@ -0,0 +1,6 @@
+namespace StabilityMatrix.Avalonia.Models.Inference;
+
+public interface IComfyStep
+{
+    void ApplyStep(ModuleApplyStepEventArgs e);
+}
diff --git a/StabilityMatrix.Avalonia/Models/Inference/ModuleApplyStepEventArgs.cs b/StabilityMatrix.Avalonia/Models/Inference/ModuleApplyStepEventArgs.cs
index 5fa43a3c..2949787a 100644
--- a/StabilityMatrix.Avalonia/Models/Inference/ModuleApplyStepEventArgs.cs
+++ b/StabilityMatrix.Avalonia/Models/Inference/ModuleApplyStepEventArgs.cs
@@ -5,6 +5,9 @@ using StabilityMatrix.Core.Models.Api.Comfy.NodeTypes;
 
 namespace StabilityMatrix.Avalonia.Models.Inference;
 
+/// <summary>
+/// Event args for applying a <see cref="IComfyStep"/>.
+/// </summary>
 public class ModuleApplyStepEventArgs : EventArgs
 {
     public required ComfyNodeBuilder Builder { get; init; }
diff --git a/StabilityMatrix.Avalonia/ViewModels/Base/IComfyStep.cs b/StabilityMatrix.Avalonia/ViewModels/Base/IComfyStep.cs
deleted file mode 100644
index 30f4d716..00000000
--- a/StabilityMatrix.Avalonia/ViewModels/Base/IComfyStep.cs
+++ /dev/null
@@ -1,9 +0,0 @@
-using StabilityMatrix.Avalonia.Models.Inference;
-using StabilityMatrix.Core.Models.Api.Comfy.Nodes;
-
-namespace StabilityMatrix.Avalonia.ViewModels.Base;
-
-public interface IComfyStep
-{
-    void ApplyStep(ModuleApplyStepEventArgs e);
-}