diff --git a/rnd/autogpt_server/autogpt_server/blocks/basic.py b/rnd/autogpt_server/autogpt_server/blocks/basic.py index 87cda8f65..17ab4762d 100644 --- a/rnd/autogpt_server/autogpt_server/blocks/basic.py +++ b/rnd/autogpt_server/autogpt_server/blocks/basic.py @@ -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: diff --git a/rnd/autogpt_server/autogpt_server/executor/manager.py b/rnd/autogpt_server/autogpt_server/executor/manager.py index 7057d9af4..16c7df311 100644 --- a/rnd/autogpt_server/autogpt_server/executor/manager.py +++ b/rnd/autogpt_server/autogpt_server/executor/manager.py @@ -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) diff --git a/rnd/autogpt_server/test/executor/test_manager.py b/rnd/autogpt_server/test/executor/test_manager.py index 26665fa2e..3cafa84f7 100644 --- a/rnd/autogpt_server/test/executor/test_manager.py +++ b/rnd/autogpt_server/test/executor/test_manager.py @@ -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