pull/7868/head^2
Zamil Majdy 2024-08-28 15:56:02 -05:00
parent 98c909f99f
commit b5db7f575e
3 changed files with 6 additions and 6 deletions

View File

@ -107,7 +107,7 @@ class ObjectLookupBlock(Block):
("output", "key"), ("output", "key"),
("output", ["v1", "v3"]), ("output", ["v1", "v3"]),
], ],
categories={BlockCategory.BASIC} categories={BlockCategory.BASIC},
) )
def run(self, input_data: Input) -> BlockOutput: def run(self, input_data: Input) -> BlockOutput:

View File

@ -150,7 +150,7 @@ async def test_input_pin_always_waited(server):
server.agent_server, server.exec_manager, test_graph, test_user, {}, 3 server.agent_server, server.exec_manager, test_graph, test_user, {}, 3
) )
executions = await server.agent_server.get_graph_run_node_execution_results( executions = await server.agent_server.get_run_execution_results(
test_graph.id, graph_exec_id, test_user.id test_graph.id, graph_exec_id, test_user.id
) )
assert len(executions) == 3 assert len(executions) == 3
@ -230,7 +230,7 @@ async def test_static_input_link_on_graph(server):
graph_exec_id = await execute_graph( graph_exec_id = await execute_graph(
server.agent_server, server.exec_manager, test_graph, test_user, {}, 8 server.agent_server, server.exec_manager, test_graph, test_user, {}, 8
) )
executions = await server.agent_server.get_graph_run_node_execution_results( executions = await server.agent_server.get_run_execution_results(
test_graph.id, graph_exec_id, test_user.id test_graph.id, graph_exec_id, test_user.id
) )
assert len(executions) == 8 assert len(executions) == 8