Skip to content

Commit

Permalink
Merge pull request #696 from valory-xyz/feat/install-timeout
Browse files Browse the repository at this point in the history
`--timeout` flag on `aea install` command
  • Loading branch information
angrybayblade authored Nov 29, 2023
2 parents b20ef46 + 6c470fc commit e3d2da4
Showing 1 changed file with 16 additions and 3 deletions.
19 changes: 16 additions & 3 deletions aea/cli/install.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,29 +49,38 @@
help="Provide extra dependency.",
multiple=True,
)
@click.option(
"--timeout",
type=float,
default=300.0,
help="Specify timeout.",
)
@click.pass_context
@check_aea_project
def install(
click_context: click.Context,
requirement: Optional[str],
extra_dependencies: Tuple[Dependency],
timeout: float,
) -> None:
"""Install the dependencies of the agent."""
ctx = cast(Context, click_context.obj)
do_install(ctx, requirement, extra_dependencies)
do_install(ctx, requirement, extra_dependencies, timeout=timeout)


def do_install(
ctx: Context,
requirement: Optional[str] = None,
extra_dependencies: Optional[Tuple[Dependency]] = None,
timeout: float = 300.0,
) -> None:
"""
Install necessary dependencies.
:param ctx: context object.
:param requirement: optional str requirement.
:param extra_dependencies: List of the extra dependencies to use
:param timeout: timeout to wait pip to install
:raises ClickException: if AEAException occurs.
"""
Expand All @@ -82,11 +91,15 @@ def do_install(
"Extra dependencies will be ignored while installing from requirements file"
)
logger.debug("Installing the dependencies in '{}'...".format(requirement))
_install_from_requirement(requirement)
_install_from_requirement(requirement, install_timeout=timeout)
else:
logger.debug("Installing all the dependencies...")
dependencies = ctx.get_dependencies(extra_dependencies=extra_dependencies)
install_dependencies(list(dependencies.values()), logger=logger)
install_dependencies(
list(dependencies.values()),
logger=logger,
install_timeout=timeout,
)
except AEAException as e:
raise click.ClickException(str(e))

Expand Down

0 comments on commit e3d2da4

Please sign in to comment.