[LI (ktz03)]

Motivation & Context

A pending Content Understanding vector-store upload retains a second reference to its document entry. Restoring the session reconstructs separate mappings, so a later upload success or failure leaves the canonical document and list_documents stuck at uploading after the queue is cleared.

Description & Review Guide

• What are the major changes? Resolve each pending upload to the current document entry before retrying. Add regressions for restored success/failure, timeout followed by another restore, and the existing in-memory and orphan-entry controls.
• What is the impact of these changes? The document status, upload duration, and failure details reflect the completed retry after restoration. Existing upload/timeout/cleanup behavior and the fallback for a missing document key are preserved; no public API or state format changes.
• What do you want reviewers to focus on? Preservation of the canonical document reference and the existing fallback. Six focused cases reproduce three failures before the fix and all pass afterward. On Windows/Python 3.12.14, the package non-integration suite passed 87 tests with five integration tests excluded (88.70% coverage); source Pyright, all five test type checkers, Ruff, changed-file hooks, and wheel/sdist builds passed. Offline SDK lifecycle checks also covered JSON/MessagePack file-store restoration with an explicitly registered DocumentStatus codec. The existing unconfigured-codec rejection and compatible to_dict() warnings remain unchanged. No live Azure upload, authentication, indexing, or model run was performed.

Related Issue

Fixes #9084

Contribution Checklist

• The code builds clean without any errors or warnings
• All unit tests pass, and I have added new tests where possible
• The PR follows the Contribution Guidelines
• This PR is linked to an issue and there is no other open PR for this issue (see Related Issue above).
• This is not a breaking change. If it is a breaking change, add the breaking change label (or add "[BREAKING]" to the title prefix, before or after any language prefix) — a workflow keeps the label and title prefix in sync automatically.

[Copilot]

Copilot review overview

🟢 Approval recommended

The targeted fix preserves fallback behavior and has comprehensive regression coverage.

Review effort: Balanced
Findings: None

What changed in this PR

Fixes restored upload retries so they update the canonical document state.

Changes:

• Resolves queued uploads against the current document mapping.
• Adds six regression cases covering success, failure, timeout, restoration, and fallback behavior.

File	Description
_context_provider.py	Uses the canonical document entry during retries.
test_context_provider.py	Adds focused retry and restoration tests.

---

💡 Add a code-review agent skill for context-aware, tailored reviews. Learn more in the docs.