Help Running pyinfra with cli api

[ascorbic-acid]

hello, thanks for this nice too, now would like to make cli tool. also if would appreciate if you help solve my issue & also give me tip on my simple structure if im also doing something wrong. My goal is to make simple reusable "run_cmd" which will work like "run_shell_command" in the normal mode, so far my code work fine but i get error when i run my "run_cmd" again

#!.venv/bin/python
import inspect
import click

from pyinfra import logger
from pyinfra.api.operation import OperationMeta

import logging
from infra_books.inv import inventory
from pyinfra.api import Config, Inventory, State
from pyinfra.api.connect import connect_all
from pyinfra.api.operation import add_op
from pyinfra.api.operations import run_ops
from pyinfra.facts.server import Os
from pyinfra.operations import server
from pyinfra.api.exceptions import PyinfraError

logging.getLogger('pyinfra').setLevel(logging.CRITICAL)


def run_cmd(state: State, cmd: str, cwd: str = "~/",):
    try:
        result = add_op(
            state,
            server.shell,
            name="Run some shell commands",
            commands=cmd,
            _chdir=cwd
        )
        run_ops(state)
    except PyinfraError as e:
        logger.critical(e)

    # op_meta: OperationMeta = list(result.values())[0]
    host = state.inventory.hosts['erp']
    print(result[host].stdout)
    print(result[host].stderr)


def main():
    state = State(inventory=inventory, config=Config(SUDO=False))
    connect_all(state)

    run_cmd(state, "whoami; hostname", cwd="~/")

    # Running Same function Twice causing the mentioned error
    # run_cmd(state, "whoami; hostname", cwd="~/")


if __name__ == "__main__":
    main()

error:

  File "/home/pop/ans-play/.venv/lib/python3.12/site-packages/pyinfra/api/operations.py", line 169, in _run_host_op
    op_data.operation_meta.set_complete(
      ^^^^^^^^^^^^^^^^^
  File "/home/pop/ans-play/.venv/lib/python3.12/site-packages/pyinfra/api/operation.py", line 78, in set_complete
    raise RuntimeError("Cannot complete an already complete operation")
^^^^^^^^^^^^^^^
RuntimeError: Cannot complete an already complete operation