Fix VibeVoice

Author: mythz

MyApp.ServiceInterface/ComfyWorkflowConverter.cs

@@ -30,7 +30,7 @@ public Task<ApiPromptResult> CreateApiPromptAsync(WorkflowVersion workflowVersio
 
     public ApiPrompt CreateApiPrompt(Dictionary<string, object?> workflow, Dictionary<string,object?> args, ComfyAgent? agent=null, string? clientId=null) 
     {
-        var requiredNodes = ComfyWorkflowParser.ExtractNodeTypes(workflow, log);
+        var requiredNodes = ComfyWorkflowParser.ExtractRequiredNodeTypes(workflow, log);
         var requiredAssets = ComfyWorkflowParser.ExtractAssetPaths(workflow, log);
         var nodeDefs = appData.GetSupportedNodeDefinitions(requiredNodes, requiredAssets, agent);
         var apiPrompt = ComfyConverters.ConvertWorkflowToApiPrompt(workflow, nodeDefs, clientId, log:log);
@@ -83,7 +83,7 @@ public class NodeComfyWorkflowConverter(ILogger<NodeComfyWorkflowConverter> log,
     public async Task<ApiPromptResult> CreateApiPromptAsync(WorkflowVersion workflowVersion, Dictionary<string, object?> args, ComfyAgent? agent=null, string? clientId = null)
     {
         var workflow = workflowVersion.Workflow;
-        var requiredNodes = ComfyWorkflowParser.ExtractNodeTypes(workflow, log);
+        var requiredNodes = ComfyWorkflowParser.ExtractRequiredNodeTypes(workflow, log);
         var requiredAssets = ComfyWorkflowParser.ExtractAssetPaths(workflow, log);
         agent ??= appData.GetSupportedAgent(requiredNodes, requiredAssets);
         var nodeDefinitionPath = agent != null

MyApp.ServiceInterface/ComfyWorkflowParser.cs

@@ -20,15 +20,19 @@ public class ComfyWorkflowParser
         return nodesObj.Map(x => (Dictionary<string, object?>)x);
     }
 
-    public static List<string> IgnoreClientNodes { get; } =
+    public static HashSet<string> IgnoreClientNodes { get; } =
     [
         "Note",
         "MarkdownNote",
         "Reroute",
         "PrimitiveNode",
-        "PrimitiveNode",
     ];
 
+    public static HashSet<string> ExtractRequiredNodeTypes(Dictionary<string, object?> workflow, ILogger? log = null)
+    {
+        return ExtractRequiredNodeTypes(workflow, IgnoreClientNodes, log);
+    }
+    
     public static HashSet<string> ExtractRequiredNodeTypes(Dictionary<string, object?> workflow, HashSet<string> ignoreNodes, ILogger? log = null)
     {
         var workflowNodes = ExtractNodeTypes(workflow, log);
@@ -40,7 +44,7 @@ public static HashSet<string> ExtractRequiredNodeTypes(Dictionary<string, object
         return workflowNodes;
     }
 
-    public static HashSet<string> ExtractNodeTypes(Dictionary<string, object?> workflow, ILogger? log = null)
+    static HashSet<string> ExtractNodeTypes(Dictionary<string, object?> workflow, ILogger? log = null)
     {
         log ??= NullLogger.Instance;
         var ret = new HashSet<string>();