except Exception as e:
        ok = bool(
            await anyio.to_thread.run_sync(
                cm.__exit__, type(e), e, e.__traceback__, limiter=exit_limiter
            )
        )
        if not ok:
            raise e
    else:
        await anyio.to_thread.run_sync(
            cm.__exit__, None, None, None, limiter=exit_limiter

    MAX_ATTEMPTS = 3
    for attempt_no in range(1, MAX_ATTEMPTS + 1):
        print(f"Running agent for {out_path} (attempt {attempt_no}/{MAX_ATTEMPTS})")
        result = agent.run_sync(prompt)
        out_content = f"{result.output.strip()}\n"
        try:
            check_translation(
                doc_lines=out_content.splitlines(),
                en_doc_lines=original_content.splitlines(),

     * @return a future that completes when unregistration is done
     */
    public CompletableFuture<Void> unregister(WitnessRegistration registration) {
        return CompletableFuture.runAsync(() -> {
            try {
                registration.setState(WitnessRegistrationState.UNREGISTERING);

                WitnessUnregisterRequest request = new WitnessUnregisterRequest();

                throw new RuntimeException(e);
            }
        });
    }
    
    public CompletableFuture<Void> unregister(WitnessRegistration registration) {
        return CompletableFuture.runAsync(() -> {
            try {
                registration.setState(WitnessRegistrationState.UNREGISTERING);
                
                WitnessUnregisterRequest request = new WitnessUnregisterRequest();

        if (timeoutSeconds <= 0) {
            timeoutSeconds = DEFAULT_LEASE_BREAK_TIMEOUT;
        }

        CompletableFuture<Void> future = CompletableFuture.runAsync(() -> {
            handleLeaseBreak(key, newState);
        });

        try {
            future.get(timeoutSeconds, TimeUnit.SECONDS);
        } catch (TimeoutException e) {

            // Cancel any pending notifications
            cancelNotification(handle);
        }
    }
    
    private void startAsyncNotification(ChangeNotificationHandle handle) {
        CompletableFuture.runAsync(() -> {
            while (handle.active) {
                try {
                    // Send SMB2 Change Notify request
                    Smb2ChangeNotifyRequest request = new Smb2ChangeNotifyRequest();