agsekit 1.7.3__tar.gz → 1.7.4__tar.gz

{agsekit-1.7.3/agsekit.egg-info → agsekit-1.7.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agsekit
-Version: 1.7.3
+Version: 1.7.4
 Summary: Agent Safety Kit command-line utilities
 Author-email: Mihanentalpo <mihanentalpo@yandex.ru>
 License: MIT License

{agsekit-1.7.3 → agsekit-1.7.4/agsekit.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agsekit
-Version: 1.7.3
+Version: 1.7.4
 Summary: Agent Safety Kit command-line utilities
 Author-email: Mihanentalpo <mihanentalpo@yandex.ru>
 License: MIT License

{agsekit-1.7.3 → agsekit-1.7.4}/agsekit.egg-info/SOURCES.txt

@@ -10,6 +10,7 @@ agsekit.egg-info/entry_points.txt
 agsekit.egg-info/requires.txt
 agsekit.egg-info/top_level.txt
 agsekit_cli/__init__.py
+agsekit_cli/agent_version_sources.py
 agsekit_cli/agents.py
 agsekit_cli/ansible_runners.py
 agsekit_cli/ansible_utils.py
@@ -65,6 +66,7 @@ agsekit_cli/ansible/agents/cline.yml
 agsekit_cli/ansible/agents/codex-glibc-prebuilt.yml
 agsekit_cli/ansible/agents/codex-glibc.yml
 agsekit_cli/ansible/agents/codex.yml
+agsekit_cli/ansible/agents/codex_code_mode_host.yml
 agsekit_cli/ansible/agents/codex_logrotate.yml
 agsekit_cli/ansible/agents/forgecode.yml
 agsekit_cli/ansible/agents/opencode.yml
@@ -125,6 +127,7 @@ prebuilt-agents/codex-glibc/Dockerfile
 prebuilt-agents/codex-glibc/build-codex-glibc.sh
 prebuilt-agents/codex-glibc/build.sh
 tests/test_addmount_command.py
+tests/test_agent_latest_versions.py
 tests/test_agents.py
 tests/test_agents_modules.py
 tests/test_ansible_agent_proxychains.py

agsekit-1.7.4/agsekit_cli/agent_version_sources.py

@@ -0,0 +1,90 @@
+from __future__ import annotations
+
+import json
+import os
+from typing import Any, Dict, Iterable, Optional
+from urllib.error import HTTPError, URLError
+from urllib.parse import quote
+from urllib.request import Request, urlopen
+
+from packaging.version import InvalidVersion, Version
+
+
+class AgentVersionCheckError(RuntimeError):
+    """Raised when an upstream agent version cannot be resolved."""
+
+
+def _headers() -> Dict[str, str]:
+    headers = {"Accept": "application/json", "User-Agent": "agsekit-agent-version-check"}
+    token = os.getenv("AGSEKIT_GITHUB_TOKEN") or os.getenv("GITHUB_TOKEN")
+    if token:
+        headers["Authorization"] = f"Bearer {token}"
+    return headers
+
+
+def fetch_json(url: str, *, timeout: float = 30.0) -> Any:
+    request = Request(url, headers=_headers())
+    try:
+        with urlopen(request, timeout=timeout) as response:
+            return json.load(response)
+    except HTTPError as exc:
+        raise AgentVersionCheckError(f"Version request failed for {url}: HTTP {exc.code}") from exc
+    except URLError as exc:
+        raise AgentVersionCheckError(f"Version request failed for {url}: {exc.reason}") from exc
+    except (OSError, ValueError) as exc:
+        raise AgentVersionCheckError(f"Invalid version response from {url}: {exc}") from exc
+
+
+def latest_npm_version(package_name: str, *, timeout: float = 30.0) -> str:
+    payload = fetch_json(
+        f"https://registry.npmjs.org/{quote(package_name, safe='')}/latest",
+        timeout=timeout,
+    )
+    version = payload.get("version") if isinstance(payload, dict) else None
+    if not isinstance(version, str) or not version.strip():
+        raise AgentVersionCheckError(f"npm did not report a latest version for {package_name}")
+    return version.strip()
+
+
+def latest_pypi_version(package_name: str, *, timeout: float = 30.0) -> str:
+    payload = fetch_json(f"https://pypi.org/pypi/{quote(package_name, safe='')}/json", timeout=timeout)
+    info = payload.get("info") if isinstance(payload, dict) else None
+    version = info.get("version") if isinstance(info, dict) else None
+    if not isinstance(version, str) or not version.strip():
+        raise AgentVersionCheckError(f"PyPI did not report a latest version for {package_name}")
+    return version.strip()
+
+
+def _stable_versions(values: Iterable[str]) -> Iterable[Version]:
+    for value in values:
+        try:
+            parsed = Version(value)
+        except InvalidVersion:
+            continue
+        if not parsed.is_prerelease and not parsed.is_devrelease:
+            yield parsed
+
+
+def latest_github_release_version(
+    repo: str,
+    *,
+    tag_prefix: str,
+    timeout: float = 30.0,
+) -> str:
+    payload = fetch_json(f"https://api.github.com/repos/{repo}/releases?per_page=100", timeout=timeout)
+    if not isinstance(payload, list):
+        raise AgentVersionCheckError(f"GitHub did not report releases for {repo}")
+
+    candidates = []
+    for release in payload:
+        if not isinstance(release, dict) or release.get("draft") or release.get("prerelease"):
+            continue
+        tag = release.get("tag_name")
+        if not isinstance(tag, str) or not tag.startswith(tag_prefix):
+            continue
+        candidates.append(tag[len(tag_prefix):])
+
+    versions = list(_stable_versions(candidates))
+    if not versions:
+        raise AgentVersionCheckError(f"GitHub did not report a stable {tag_prefix} release for {repo}")
+    return str(max(versions))

agsekit-1.7.4/agsekit_cli/agents_modules/aider.py

@@ -0,0 +1,14 @@
+from .base import BaseAgent
+from ..agent_version_sources import latest_pypi_version
+
+
+class AiderAgent(BaseAgent):
+    type_name = "aider"
+    runtime_binary = "aider"
+    default_version = "0.86.2"
+    default_env = {"AIDER_CHECK_UPDATE": "false"}
+
+    @classmethod
+    def check_latest_version(cls, *, architecture=None, timeout=30.0):
+        del architecture
+        return cls.normalize_version(latest_pypi_version("aider-chat", timeout=timeout))

{agsekit-1.7.3 → agsekit-1.7.4}/agsekit_cli/agents_modules/base.py

@@ -33,6 +33,12 @@ class BaseAgent:
     def needs_nvm(cls) -> bool:
         return bool(cls._needs_nvm)
 
+    @classmethod
+    def check_latest_version(cls, *, architecture: Optional[str] = None, timeout: float = 30.0) -> str:
+        del architecture
+        del timeout
+        raise NotImplementedError(f"Latest-version lookup is not implemented for {cls.type_name}")
+
     @classmethod
     def playbook_name(cls) -> str:
         if cls.installer_playbook:

agsekit-1.7.4/agsekit_cli/agents_modules/claude.py

@@ -0,0 +1,15 @@
+from .base import BaseAgent
+from ..agent_version_sources import latest_npm_version
+
+
+class ClaudeAgent(BaseAgent):
+    type_name = "claude"
+    runtime_binary = "claude"
+    default_version = "2.1.141"
+    default_env = {"DISABLE_AUTOUPDATER": "1"}
+    _needs_nvm = True
+
+    @classmethod
+    def check_latest_version(cls, *, architecture=None, timeout=30.0):
+        del architecture
+        return cls.normalize_version(latest_npm_version("@anthropic-ai/claude-code", timeout=timeout))

agsekit-1.7.4/agsekit_cli/agents_modules/cline.py

@@ -0,0 +1,15 @@
+from .base import BaseAgent
+from ..agent_version_sources import latest_npm_version
+
+
+class ClineAgent(BaseAgent):
+    type_name = "cline"
+    runtime_binary = "cline"
+    default_version = "2.17.0"
+    default_env = {"CLINE_NO_AUTO_UPDATE": "1"}
+    _needs_nvm = True
+
+    @classmethod
+    def check_latest_version(cls, *, architecture=None, timeout=30.0):
+        del architecture
+        return cls.normalize_version(latest_npm_version("cline", timeout=timeout))

agsekit-1.7.4/agsekit_cli/agents_modules/codex.py

@@ -0,0 +1,14 @@
+from .base import BaseAgent
+from ..agent_version_sources import latest_npm_version
+
+
+class CodexAgent(BaseAgent):
+    type_name = "codex"
+    runtime_binary = "codex"
+    default_version = "0.130.0"
+    _needs_nvm = True
+
+    @classmethod
+    def check_latest_version(cls, *, architecture=None, timeout=30.0):
+        del architecture
+        return cls.normalize_version(latest_npm_version("@openai/codex", timeout=timeout))

agsekit-1.7.4/agsekit_cli/agents_modules/codex_glibc.py

@@ -0,0 +1,15 @@
+from .base import BaseAgent
+from ..agent_version_sources import latest_github_release_version
+
+
+class CodexGlibcAgent(BaseAgent):
+    type_name = "codex-glibc"
+    runtime_binary = "codex-glibc"
+    default_version = "0.130.0"
+
+    @classmethod
+    def check_latest_version(cls, *, architecture=None, timeout=30.0):
+        del architecture
+        return cls.normalize_version(
+            latest_github_release_version("openai/codex", tag_prefix="rust-v", timeout=timeout)
+        )

agsekit-1.7.4/agsekit_cli/agents_modules/codex_glibc_prebuilt.py

@@ -0,0 +1,23 @@
+from .base import BaseAgent
+from ..agent_version_sources import AgentVersionCheckError
+from ..prebuilt import PrebuiltReleaseError, resolve_codex_glibc_prebuilt_release
+
+
+class CodexGlibcPrebuiltAgent(BaseAgent):
+    type_name = "codex-glibc-prebuilt"
+    runtime_binary = "codex-glibc-prebuilt"
+    default_version = "0.130.0"
+
+    @classmethod
+    def check_latest_version(cls, *, architecture=None, timeout=30.0):
+        del timeout
+        if not architecture:
+            raise AgentVersionCheckError("Current architecture is required for codex-glibc-prebuilt")
+        try:
+            release = resolve_codex_glibc_prebuilt_release(arch=architecture)
+        except PrebuiltReleaseError as exc:
+            raise AgentVersionCheckError(str(exc)) from exc
+        prefix = "codex-glibc-rust-v"
+        if not release.tag.startswith(prefix):
+            raise AgentVersionCheckError(f"Unexpected codex-glibc prebuilt tag: {release.tag}")
+        return cls.normalize_version(release.tag[len(prefix):])

agsekit-1.7.4/agsekit_cli/agents_modules/forgecode.py

@@ -0,0 +1,16 @@
+from .base import BaseAgent
+from ..agent_version_sources import latest_github_release_version
+
+
+class ForgecodeAgent(BaseAgent):
+    type_name = "forgecode"
+    runtime_binary = "forge"
+    default_version = "2.12.14"
+    default_env = {"FORGE_TRACKER": "false"}
+
+    @classmethod
+    def check_latest_version(cls, *, architecture=None, timeout=30.0):
+        del architecture
+        return cls.normalize_version(
+            latest_github_release_version("tailcallhq/forgecode", tag_prefix="v", timeout=timeout)
+        )

agsekit-1.7.4/agsekit_cli/agents_modules/opencode.py

@@ -0,0 +1,15 @@
+from .base import BaseAgent
+from ..agent_version_sources import latest_npm_version
+
+
+class OpencodeAgent(BaseAgent):
+    type_name = "opencode"
+    runtime_binary = "opencode"
+    default_version = "1.14.50"
+    default_env = {"OPENCODE_DISABLE_AUTOUPDATE": "true"}
+    _needs_nvm = True
+
+    @classmethod
+    def check_latest_version(cls, *, architecture=None, timeout=30.0):
+        del architecture
+        return cls.normalize_version(latest_npm_version("opencode-ai", timeout=timeout))

agsekit-1.7.4/agsekit_cli/agents_modules/qwen.py

@@ -0,0 +1,14 @@
+from .base import BaseAgent
+from ..agent_version_sources import latest_npm_version
+
+
+class QwenAgent(BaseAgent):
+    type_name = "qwen"
+    runtime_binary = "qwen"
+    default_version = "0.15.11"
+    _needs_nvm = True
+
+    @classmethod
+    def check_latest_version(cls, *, architecture=None, timeout=30.0):
+        del architecture
+        return cls.normalize_version(latest_npm_version("@qwen-code/qwen-code", timeout=timeout))

{agsekit-1.7.3 → agsekit-1.7.4}/agsekit_cli/ansible/agents/codex-glibc-prebuilt.yml

@@ -52,8 +52,7 @@
               (ansible_playbook_python | quote)
               ~ ' -m agsekit_cli.prebuilt resolve-codex-glibc-prebuilt --arch '
               ~ (codex_prebuilt_arch | quote)
-              ~ ' --tag '
-              ~ (('--tag ' ~ ('codex-glibc-rust-v' ~ requested_agent_version)) if requested_agent_version else '')
+              ~ ((' --tag ' ~ ('codex-glibc-rust-v' ~ requested_agent_version)) if requested_agent_version else '')
             ) | from_json
           }}
 
@@ -149,3 +148,6 @@
       ansible.builtin.command: "{{ codex_install_path }} --version"
       register: codex_version_check
       changed_when: false
+
+    - name: Install version-matched Codex Code Mode host
+      ansible.builtin.include_tasks: "{{ playbook_dir }}/codex_code_mode_host.yml"

{agsekit-1.7.3 → agsekit-1.7.4}/agsekit_cli/ansible/agents/codex-glibc.yml

@@ -263,6 +263,9 @@
       register: codex_version_check
       changed_when: false
 
+    - name: Install version-matched Codex Code Mode host
+      ansible.builtin.include_tasks: "{{ playbook_dir }}/codex_code_mode_host.yml"
+
     - name: Cleanup rust and cargo toolchains used for build
       ansible.builtin.shell: |
         set -euo pipefail

agsekit-1.7.4/agsekit_cli/ansible/agents/codex_code_mode_host.yml

@@ -0,0 +1,52 @@
+---
+- name: Extract installed Codex version for Code Mode host
+  ansible.builtin.set_fact:
+    codex_code_mode_host_version: >-
+      {{ codex_version_check.stdout | regex_search('[0-9]+\.[0-9]+\.[0-9]+(?:[-+][0-9A-Za-z.-]+)?') }}
+
+- name: Fail when installed Codex version cannot be determined
+  ansible.builtin.fail:
+    msg: >-
+      Could not determine the installed Codex version from: {{ codex_version_check.stdout }}
+  when: not codex_code_mode_host_version
+
+- name: Determine Codex Code Mode host release target
+  ansible.builtin.set_fact:
+    codex_code_mode_host_target: >-
+      {{ 'aarch64-unknown-linux-musl' if ansible_architecture in ['aarch64', 'arm64'] else 'x86_64-unknown-linux-musl' }}
+    codex_code_mode_host_path: "{{ codex_install_path | dirname }}/codex-code-mode-host"
+
+- name: Remove previous Codex Code Mode host download
+  ansible.builtin.file:
+    path: /tmp/codex-code-mode-host.tar.gz
+    state: absent
+
+- name: Download version-matched Codex Code Mode host
+  ansible.builtin.command: >-
+    {{ proxychains_prefix }}curl -fsSL --write-out %{http_code}
+    "https://github.com/openai/codex/releases/download/rust-v{{ codex_code_mode_host_version }}/codex-code-mode-host-{{ codex_code_mode_host_target }}.tar.gz"
+    -o /tmp/codex-code-mode-host.tar.gz
+  register: codex_code_mode_host_download
+  changed_when: codex_code_mode_host_download.rc == 0
+  failed_when:
+    - codex_code_mode_host_download.rc != 0
+    - codex_code_mode_host_download.stdout | trim != "404"
+
+- name: Install Codex Code Mode host beside Codex binary
+  ansible.builtin.shell: |
+    set -euo pipefail
+    rm -rf /tmp/codex-code-mode-host-extract
+    mkdir -p /tmp/codex-code-mode-host-extract
+    tar -xzf /tmp/codex-code-mode-host.tar.gz -C /tmp/codex-code-mode-host-extract
+    helper="/tmp/codex-code-mode-host-extract/codex-code-mode-host-{{ codex_code_mode_host_target }}"
+    test -f "$helper"
+    {{ 'sudo ' if codex_install_mode == 'system' else '' }}install -m 0755 "$helper" "{{ codex_code_mode_host_path }}"
+    rm -rf /tmp/codex-code-mode-host-extract /tmp/codex-code-mode-host.tar.gz
+  args:
+    executable: /bin/bash
+  when: codex_code_mode_host_download.rc == 0
+
+- name: Verify Codex Code Mode host works
+  ansible.builtin.command: "{{ codex_code_mode_host_path }} --help"
+  changed_when: false
+  when: codex_code_mode_host_download.rc == 0

{agsekit-1.7.3 → agsekit-1.7.4}/agsekit_cli/commands/check_new_version.py

@@ -5,7 +5,7 @@ from typing import Optional
 
 import click
 
-from ..config import resolve_config_path
+from ..config import ConfigError, resolve_config_path
 from ..i18n import tr
 from ..state import CHANGELOG_URL, get_state_manager, initialize_state, run_check_new_version
 from ..versioning import is_newer_version
@@ -24,7 +24,7 @@ from . import debug_option, non_interactive_option
     help=tr("config.option_path"),
 )
 def check_new_version_command(config_path: Optional[str], non_interactive: bool, debug: bool) -> None:
-    """Check PyPI through pip, update state.yaml, and report whether a newer version exists."""
+    """Refresh cached agsekit and configured-agent versions in state.yaml."""
     del non_interactive
     del debug
 
@@ -32,8 +32,8 @@ def check_new_version_command(config_path: Optional[str], non_interactive: bool,
     initialize_state(resolved_path)
     manager = get_state_manager()
     try:
-        latest = run_check_new_version()
-    except RuntimeError as exc:
+        latest = run_check_new_version(resolved_path)
+    except (ConfigError, RuntimeError) as exc:
         raise click.ClickException(tr("check_new_version.failed", error=str(exc)))
 
     if is_newer_version(latest, manager.current_version):

{agsekit-1.7.3 → agsekit-1.7.4}/agsekit_cli/commands/daemon.py

@@ -77,7 +77,7 @@ def run_daemon_status() -> None:
 
 @click.group(name="daemon", help=tr("daemon.group_help"))
 def daemon_group() -> None:
-    """Commands for managing the portforward daemon."""
+    """Commands for managing the agsekit background daemon."""
 
 
 @daemon_group.command(name="install", help=tr("daemon.install_help"))

{agsekit-1.7.3 → agsekit-1.7.4}/agsekit_cli/commands/install_agents.py

@@ -34,6 +34,7 @@ from ..interactive import is_interactive_terminal
 from ..i18n import tr
 from ..progress import ProgressManager
 from ..provision_handlers import PreparedVmSsh, choose_provision_handler
+from ..state import initialize_state, refresh_agent_versions
 from ..vm import MultipassError, ensure_multipass_available, resolve_proxychains
 from ..vm_prepare import _ensure_vm_ssh_access, _fetch_vm_ips, ensure_host_ssh_keypair, vm_ssh_ansible_vars
 from . import debug_option, non_interactive_option
@@ -246,6 +247,8 @@ def run_install_agents(
     debug: bool,
     interactive: bool,
     progress: Optional[ProgressManager] = None,
+    upgrade: bool = False,
+    force_check_versions: bool = False,
 ) -> None:
     with debug_scope(debug):
         if not progress:
@@ -253,6 +256,8 @@ def run_install_agents(
 
         if all_agents and agent_name:
             raise click.ClickException(tr("install_agents.agent_conflict"))
+        if force_check_versions and not upgrade:
+            raise click.ClickException(tr("install_agents.force_check_requires_upgrade"))
 
         resolved_path = resolve_config_path(Path(config_path) if config_path else None)
         try:
@@ -307,6 +312,31 @@ def run_install_agents(
                             raise click.ClickException(tr("install_agents.vm_missing", vm_name=target_vm_name))
                         targets.append((agent.name, vms_config[target_vm_name]))
 
+        upgrade_versions: Dict[str, str] = {}
+        if upgrade:
+            state_manager = initialize_state(resolved_path)
+            selected_agent_types = {
+                find_agent(agents_config, target_agent_name).type
+                for target_agent_name, _target_vm in targets
+            }
+            if force_check_versions:
+                errors = refresh_agent_versions(selected_agent_types, force=True)
+                if errors:
+                    details = "; ".join(
+                        f"{agent_type}: {message}" for agent_type, message in sorted(errors.items())
+                    )
+                    raise click.ClickException(
+                        tr("install_agents.force_check_failed", errors=details)
+                    )
+            for target_agent_name in dict.fromkeys(name for name, _vm in targets):
+                target_agent = find_agent(agents_config, target_agent_name)
+                cached = state_manager.get_agent_version(target_agent.type)
+                if cached is None:
+                    raise click.ClickException(
+                        tr("install_agents.upgrade_version_missing", agent_type=target_agent.type)
+                    )
+                upgrade_versions[target_agent.type] = cached.latest_version
+
         if progress is None:
             with ProgressManager(debug=debug) as owned_progress:
                 _run_install_targets(
@@ -317,6 +347,7 @@ def run_install_agents(
                     debug=debug,
                     progress=owned_progress,
                     show_overall_task=True,
+                    upgrade_versions=upgrade_versions,
                 )
             _emit_install_success(targets)
             return
@@ -329,6 +360,7 @@ def run_install_agents(
             debug=debug,
             progress=progress,
             show_overall_task=False,
+            upgrade_versions=upgrade_versions,
         )
 
 
@@ -341,6 +373,7 @@ def _run_install_targets(
     debug: bool,
     progress: ProgressManager,
     show_overall_task: bool,
+    upgrade_versions: Optional[Dict[str, str]] = None,
 ) -> None:
     overall_task = progress.add_task(tr("progress.install_agents_title"), total=len(targets)) if show_overall_task else None
     node_ready_vms: set[str] = set()
@@ -349,8 +382,9 @@ def _run_install_targets(
         agent = find_agent(agents_config, target_agent_name)
         agent_cls = get_agent_class(agent.type)
         playbook_path = _playbook_for(agent)
+        requested_version = upgrade_versions.get(agent.type) if upgrade_versions else agent.version
         installed_version = _installed_agent_version(agent, target_vm, debug=debug)
-        if agent.version is None and installed_version is not None:
+        if requested_version is None and installed_version is not None:
             click.echo(
                 tr(
                     "install_agents.version_unpinned",
@@ -363,14 +397,14 @@ def _run_install_targets(
             if overall_task is not None:
                 progress.advance(overall_task)
             continue
-        if agent.version is not None and installed_version == agent.version:
+        if requested_version is not None and installed_version == requested_version:
             click.echo(
                 tr(
                     "install_agents.version_ok",
                     agent_name=agent.name,
                     agent_type=agent.type,
                     vm_name=target_vm.name,
-                    version=agent.version or "latest",
+                    version=requested_version or "latest",
                 )
             )
             if overall_task is not None:
@@ -384,7 +418,7 @@ def _run_install_targets(
                     agent_name=agent.name,
                     agent_type=agent.type,
                     vm_name=target_vm.name,
-                    version=agent.version or "latest",
+                    version=requested_version or "latest",
                 )
             )
         else:
@@ -395,7 +429,7 @@ def _run_install_targets(
                     agent_type=agent.type,
                     vm_name=target_vm.name,
                     current_version=installed_version,
-                    required_version=agent.version or "latest",
+                    required_version=requested_version or "latest",
                 )
             )
         if debug:
@@ -405,7 +439,7 @@ def _run_install_targets(
                     agent_name=agent.name,
                     agent_type=agent.type,
                     vm_name=target_vm.name,
-                    version=agent.version or "latest",
+                    version=requested_version or "latest",
                     script=playbook_path.name,
                 )
             )
@@ -422,7 +456,7 @@ def _run_install_targets(
             if overall_task is not None:
                 progress.update(overall_task, description=install_label)
             extra_vars_overrides: Dict[str, object] = {
-                "agent_version": agent.version,
+                "agent_version": requested_version,
                 "force_reinstall": force_reinstall,
             }
             if agent_cls.needs_nvm() and target_vm.name in node_ready_vms:
@@ -467,6 +501,12 @@ def _run_install_targets(
 @click.argument("vm", required=False)
 @click.option("--all-vms", is_flag=True, help=tr("install_agents.option_all_vms"))
 @click.option("--all-agents", is_flag=True, help=tr("install_agents.option_all_agents"))
+@click.option("--upgrade", is_flag=True, help=tr("install_agents.option_upgrade"))
+@click.option(
+    "--force-check-versions",
+    is_flag=True,
+    help=tr("install_agents.option_force_check_versions"),
+)
 @click.option(
     "config_path",
     "--config",
@@ -487,6 +527,8 @@ def install_agents_command(
     vm: Optional[str],
     all_vms: bool,
     all_agents: bool,
+    upgrade: bool,
+    force_check_versions: bool,
     config_path: Optional[str],
     proxychains: Optional[str],
     debug: bool,
@@ -503,4 +545,6 @@ def install_agents_command(
         proxychains=proxychains,
         debug=debug,
         interactive=interactive,
+        upgrade=upgrade,
+        force_check_versions=force_check_versions,
     )