github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-08 03:00:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(block): Avoid infinite loop execution on AddToListBlock self-loop

Browse Source
This commit is contained in:
Zamil Majdy
2025-03-13 15:27:28 +07:00
parent f4d4bb83b0
commit 266354e618
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
autogpt_platform/backend/backend/blocks/basic.py
View File

@@ -468,11 +468,27 @@ class AddToListBlock(Block):
@classmethod
def get_missing_links(cls, data: BlockInput, links: List["Link"]) -> set[str]:
# If `list` self-loop present, the `entry` pin can't be static.
list_self_loop_present = any(
link.sink_name == "list" and link.sink_id == link.source_id
for link in links
)
entry_static_output_present = any(
link.sink_name == "entry" and link.is_static for link in links
)
if list_self_loop_present and entry_static_output_present:
raise ValueError(
"The `entry` pin can't have static output from `AgentInput` or "
"`StoreValue` block as long as the `list` pin has a self-loop."
"This will cause an infinite execution loop."
)
return super().get_missing_links(
data,
[
link
for link in links
# Allow execution with `list` pin self-loop
if link.sink_name != "list" or link.sink_id != link.source_id
],
)