Commit 383105b 1 parent b372b10 commit 383105b Copy full SHA for 383105b
File tree 2 files changed +5
-5
lines changed
2 files changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -153,11 +153,11 @@ async def on_message(message):
153
153
"nodes" : nodes_info
154
154
}
155
155
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 :
158
158
logger .warning ("[Control] Missing prompt in update_prompt message" )
159
159
return
160
- await pipeline .update_prompts (params ["prompts " ])
160
+ await pipeline .update_prompts (params ["prompt " ])
161
161
response = {
162
162
"type" : "prompts_updated" ,
163
163
"success" : True
Original file line number Diff line number Diff line change @@ -125,7 +125,7 @@ async def get_available_nodes(self):
125
125
126
126
# Now process any nodes in our prompt that use this class_type
127
127
for node_id in list (remaining_nodes ):
128
- node = self . prompt [node_id ]
128
+ node = prompt [node_id ]
129
129
if node .get ('class_type' ) != class_type :
130
130
continue
131
131
@@ -149,7 +149,7 @@ async def get_available_nodes(self):
149
149
150
150
all_prompts_nodes_info [prompt_index ] = nodes_info
151
151
152
- return all_prompts_nodes_info
152
+ return all_prompts_nodes_info [ 0 ] # TODO: make it for for multiple prompts
153
153
154
154
except Exception as e :
155
155
logger .error (f"Error getting node info: { str (e )} " )
You can’t perform that action at this time.
0 commit comments