Skip to content

Commit 383105b

Browse files
committed
fix prompt control panel rendering (wip)
1 parent b372b10 commit 383105b

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

server/app.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -153,11 +153,11 @@ async def on_message(message):
153153
"nodes": nodes_info
154154
}
155155
channel.send(json.dumps(response))
156-
elif params.get("type") == "update_prompts":
157-
if "prompts" not in params:
156+
elif params.get("type") == "update_prompt":
157+
if "prompt" not in params:
158158
logger.warning("[Control] Missing prompt in update_prompt message")
159159
return
160-
await pipeline.update_prompts(params["prompts"])
160+
await pipeline.update_prompts(params["prompt"])
161161
response = {
162162
"type": "prompts_updated",
163163
"success": True

src/comfystream/client.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ async def get_available_nodes(self):
125125

126126
# Now process any nodes in our prompt that use this class_type
127127
for node_id in list(remaining_nodes):
128-
node = self.prompt[node_id]
128+
node = prompt[node_id]
129129
if node.get('class_type') != class_type:
130130
continue
131131

@@ -149,7 +149,7 @@ async def get_available_nodes(self):
149149

150150
all_prompts_nodes_info[prompt_index] = nodes_info
151151

152-
return all_prompts_nodes_info
152+
return all_prompts_nodes_info[0] # TODO: make it for for multiple prompts
153153

154154
except Exception as e:
155155
logger.error(f"Error getting node info: {str(e)}")

0 commit comments

Comments
 (0)