neuro_llc/ComfyBox
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

Prompt serializer and test fixes

Browse Source
This commit is contained in:
space-nuko
2023-05-20 21:42:38 -05:00
parent 8d031120f7
commit 09e806bd3e
17 changed files with 196 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/lib/nodes/ComfyConfigureQueuePromptButton.ts
View File

@@ -29,13 +29,10 @@ export default class ComfyConfigureQueuePromptButton extends ComfyGraphNode {
throw new Error(`Could not find layout attached to this node! ${this.id}`)
}
this.layoutState.update(state => {
if (typeof param === "string")
state.attrs.queuePromptButtonName = param || ""
else if (typeof param === "object" && "buttonName" in param)
state.attrs.queuePromptButtonName = param.buttonName || ""
return state
})
if (typeof param === "string")
this.workflow.attrs.queuePromptButtonName = param || ""
else if (typeof param === "object" && "buttonName" in param)
this.workflow.attrs.queuePromptButtonName = param.buttonName || ""
}
}
}

5
src/lib/nodes/ComfyGraphNode.ts
View File

@@ -10,6 +10,7 @@ import { get } from "svelte/store";
import configState from "$lib/stores/configState";
import type { WritableLayoutStateStore } from "$lib/stores/layoutStates";
import layoutStates from "$lib/stores/layoutStates";
import workflowStateStore, { ComfyWorkflow } from "$lib/stores/workflowState";
export type DefaultWidgetSpec = {
defaultWidgetNode: new (name?: string) => ComfyWidgetNode,
@@ -106,6 +107,10 @@ export default class ComfyGraphNode extends LGraphNode {
return layoutStates.getLayoutByNode(this);
}
get workflow(): ComfyWorkflow | null {
return workflowStateStore.getWorkflowByNode(this);
}
constructor(title?: string) {
super(title)
this.addProperty("tags", [], "array")