Python: PR1 — New session and context provider types (side-by-side)

python/packages/a2a/agent_framework_a2a/_agent.py

@@ -7,7 +7,7 @@
 import re
 import uuid
 from collections.abc import AsyncIterable, Awaitable, Sequence
-from typing import Any, Final, Literal, cast, overload
+from typing import Any, Final, Literal, overload
 
 import httpx
 from a2a.client import Client, ClientConfig, ClientFactory, minimal_agent_card
@@ -401,11 +401,15 @@ def _prepare_message_for_a2a(self, message: ChatMessage) -> A2AMessage:
                 case _:
                     raise ValueError(f"Unknown content type: {content.type}")
 
+        # Exclude framework-internal keys (e.g. attribution) from wire metadata
+        internal_keys = {"_attribution"}
+        metadata = {k: v for k, v in message.additional_properties.items() if k not in internal_keys} or None
+
         return A2AMessage(
             role=A2ARole("user"),
             parts=parts,
             message_id=message.message_id or uuid.uuid4().hex,
-            metadata=cast(dict[str, Any], message.additional_properties),
+            metadata=metadata,
         )
 
     def _parse_contents_from_a2a(self, parts: Sequence[A2APart]) -> list[Content]:

python/packages/azure-ai-search/agent_framework_azure_ai_search/__init__.py

@@ -2,6 +2,7 @@
 
 import importlib.metadata
 
+from ._context_provider import _AzureAISearchContextProvider
 from ._search_provider import AzureAISearchContextProvider, AzureAISearchSettings
 
 try:
@@ -12,5 +13,6 @@
 __all__ = [
     "AzureAISearchContextProvider",
     "AzureAISearchSettings",
+    "_AzureAISearchContextProvider",
     "__version__",
 ]