[LI (ktz03)]

Motivation & Context

The Cosmos memory package guide places a stable user ID in flat session state, but the provider receives state scoped by its source ID. Following that entry therefore falls back to the session ID instead of the intended stable user ID.

Description & Review Guide

• What are the major changes? Use session.state.setdefault(provider.source_id, {})["user_id"], matching the existing README and three samples. Remove the missing-user warning promise, which the resolver does not implement.
• What is the impact of these changes? The package guide describes the existing state contract. Runtime code and APIs are unchanged.
• What do you want reviewers to focus on? Consistency with the provider-scoped state passed to before_run and after_run.

Validation: eight controlled Core Agent.run calls check two provider IDs, flat versus scoped state, and two independent sessions. Four flat-state runs omit user_id from the delivered provider state; four scoped-state runs deliver the stable ID. The second session in each case uses the public session snapshot round trip. The Cosmos resolver was checked in source; no Cosmos/Foundry backend, live model, full package suite or build was run.

Related Issue

N/A - trivial change. This corrects one guide entry to match the unchanged public contract already shown in the README, samples and resolver; the desired documentation result does not require a new design or behavior decision.

AI Assistance

• No material AI assistance was used.
• This is an AI-assisted contribution. I reviewed, understood, and verified all submitted content and accept responsibility for it.

AI assistance details: Documentation, validation, and PR description.

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 links to an agreed issue with no competing open PR, or the Related Issue section documents a trivial-change or repository-automation exception.
• 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]

🟢 Approval recommended

The documentation-only corrections match existing behavior, with no unresolved findings.

0 open findings

What changed in this PR

Corrects the Cosmos memory package guide to match the existing provider-scoped state contract. Runtime code and APIs are unchanged.

Changes:

• Shows how to set a stable user ID under provider.source_id.
• Removes the unsupported promise of a missing-user warning.

File	Description
python/​packages/​azure-cosmos-memory/​AGENTS.md	Corrects user ID scoping and fallback documentation.

🧠 Review effort: Balanced

---

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