Agent: Usa agent config object instead of dict should_propagate

This commit is contained in:
vakaris_zilius 2022-06-23 15:08:20 +00:00 committed by Mike Salvatore
parent ab67853192
commit 0f848eb284
1 changed files with 5 additions and 2 deletions

View File

@ -1,2 +1,5 @@
def should_propagate(config: dict, current_depth: int) -> bool: from common.configuration import AgentConfiguration
return config["config"]["depth"] > current_depth
def should_propagate(config: AgentConfiguration, current_depth: int) -> bool:
return config.propagation.maximum_depth > current_depth