Python: [BREAKING] Scope provider state by source_id and standardize source IDs (#3995)

* Initial plan

* Add FoundryMemoryProvider and tests

Co-authored-by: eavanvalkenburg <13749212+eavanvalkenburg@users.noreply.github.com>

* Add sample and documentation for FoundryMemoryProvider

Co-authored-by: eavanvalkenburg <13749212+eavanvalkenburg@users.noreply.github.com>

* Address code review feedback for FoundryMemoryProvider

Co-authored-by: eavanvalkenburg <13749212+eavanvalkenburg@users.noreply.github.com>

* Address PR review comments: Add DEFAULT_SOURCE_ID, use logging.getLogger, move state to session.state

Co-authored-by: eavanvalkenburg <13749212+eavanvalkenburg@users.noreply.github.com>

* Fix Foundry memory ItemParam usage and exports

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

* Refactor provider hook state and standardize source IDs

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

* Support endpoint-based Foundry memory init

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

* Fix core README workflows link

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

* updated implementation and sample

* Split out Foundry memory provider changes

Remove FoundryMemoryProvider implementation/tests/sample plus export and docs mentions from this branch so only non-Foundry changes remain.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

* Trigger CI rerun for PR #3995

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

---------

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: eavanvalkenburg <13749212+eavanvalkenburg@users.noreply.github.com>
Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

python/packages/lab/tau2/agent_framework_lab_tau2/_sliding_window.py

@@ -18,7 +18,7 @@ class SlidingWindowHistoryProvider(InMemoryHistoryProvider):
 
     def __init__(
         self,
-        source_id: str = "memory",
+        source_id: str = InMemoryHistoryProvider.DEFAULT_SOURCE_ID,
         *,
         max_tokens: int = 3800,
         system_message: str | None = None,

python/packages/lab/tau2/agent_framework_lab_tau2/runner.py

@@ -12,6 +12,7 @@ from agent_framework import (
     AgentExecutorResponse,
     AgentResponse,
     FunctionExecutor,
+    InMemoryHistoryProvider,
     Message,
     SupportsChatGetResponse,
     Workflow,
@@ -359,7 +360,9 @@ class TaskRunner:
         # 2. The assistant's session state (full history, not just the truncated window)
         # 3. The final user message (if any)
         session_state: dict[str, Any] = self._assistant_executor._session.state  # type: ignore
-        all_messages: list[Message] = list(session_state.get("memory", {}).get("messages", []))  # type: ignore
+        all_messages: list[Message] = list(
+            session_state.get(InMemoryHistoryProvider.DEFAULT_SOURCE_ID, {}).get("messages", [])
+        )  # type: ignore
         full_conversation = [first_message, *all_messages]
         if self._final_user_message is not None:
             full_conversation.extend(self._final_user_message)

python/packages/lab/tau2/tests/test_sliding_window.py

@@ -4,6 +4,7 @@
 
 from unittest.mock import patch
 
+from agent_framework import InMemoryHistoryProvider
 from agent_framework._types import Content, Message
 from agent_framework_lab_tau2._sliding_window import SlidingWindowHistoryProvider
 
@@ -12,7 +13,7 @@ def _make_state(provider: SlidingWindowHistoryProvider, messages: list[Message]
     """Helper to create a session state dict with messages pre-loaded."""
     state: dict = {}
     if messages:
-        state[provider.source_id] = {"messages": list(messages)}
+        state["messages"] = list(messages)
     return state
 
 
@@ -27,7 +28,7 @@ def test_initialization():
     assert provider.max_tokens == 2000
     assert provider.system_message == "You are a helpful assistant"
     assert provider.tool_definitions == [{"name": "test_tool"}]
-    assert provider.source_id == "memory"
+    assert provider.source_id == InMemoryHistoryProvider.DEFAULT_SOURCE_ID
 
 
 async def test_get_messages_empty():
@@ -66,7 +67,7 @@ async def test_save_and_get_messages():
     # get_messages returns truncated
     truncated = await provider.get_messages(None, state=state)
     # Full history is in session state
-    all_msgs = state[provider.source_id]["messages"]
+    all_msgs = state["messages"]
 
     assert len(all_msgs) == 10
     assert len(truncated) < len(all_msgs)
@@ -196,7 +197,7 @@ async def test_real_world_scenario():
     await provider.save_messages(None, conversation, state=state)
 
     truncated = await provider.get_messages(None, state=state)
-    all_msgs = state[provider.source_id]["messages"]
+    all_msgs = state["messages"]
 
     assert len(all_msgs) == 6
     assert len(truncated) <= 6