Fix missing node causing connection error

9 months ago · 3653fb475e
1 changed files with 3 additions and 2 deletions
--- a/StabilityMatrix.Core/Inference/ComfyClient.cs
+++ b/StabilityMatrix.Core/Inference/ComfyClient.cs
@ -471,8 +471,9 @@ public class ComfyClient : InferenceClientBase
    {
        var response = await comfyApi.GetObjectInfo(nodeName, cancellationToken).ConfigureAwait(false);

-        var info = response[nodeName];
-        return info.Input.GetOptionalValueAsNestedList(optionName);
+        var info = response.GetValueOrDefault(nodeName);
+
+        return info?.Input.GetOptionalValueAsNestedList(optionName);
    }

    protected override void Dispose(bool disposing)