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

@ -76,7 +76,7 @@ class PrintingBlock(Block):
class ObjectLookupBlock(Block):
class Input(BlockSchema):
input: Any = Field(description="Dictionary to lookup from")
key: str | int = Field(description="Key to lookup in the dictionary")
@ -84,7 +84,7 @@ class ObjectLookupBlock(Block):
class Output(BlockSchema):
output: Any = Field(description="Value found for the given key")
missing: Any = Field(description="Value of the input that missing the key")
def __init__(self):
super().__init__(
id="b2g2c3d4-5e6f-7g8h-9i0j-k1l2m3n4o5p6",
@ -107,7 +107,7 @@ class ObjectLookupBlock(Block):
("output", "key"),
("output", ["v1", "v3"]),
],
categories={BlockCategory.BASIC}
categories={BlockCategory.BASIC},
)
def run(self, input_data: Input) -> BlockOutput:

View File

@ -441,7 +441,7 @@ class ExecutionManager(AppService):
name = node.input_default.get("name")
if name and name in data:
input_data = {"value": data[name]}
input_data, error = validate_exec(node, input_data)
if input_data is None:
raise Exception(error)

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
)
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
)
assert len(executions) == 3
@ -230,7 +230,7 @@ async def test_static_input_link_on_graph(server):
graph_exec_id = await execute_graph(
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
)
assert len(executions) == 8