fix cross partition query

eavanvalkenburg · eavanvalkenburg · commit f8525519e917 · 2026-03-03T11:55:50.000+01:00
diff --git a/python/packages/azure-cosmos/agent_framework_azure_cosmos/_history_provider.py b/python/packages/azure-cosmos/agent_framework_azure_cosmos/_history_provider.py
@@ -211,9 +211,8 @@ async def list_sessions(self) -> list[str]:
         parameters: list[dict[str, object]] = [
             {"name": "@source_id", "value": self.source_id}
         ]
-        items = self._container_proxy.query_items(  # type: ignore[union-attr]
-            query=query, parameters=parameters, enable_cross_partition_query=True
-        )
+        # without a partition key, it is automatically a cross-partition query
+        items = self._container_proxy.query_items(query=query, parameters=parameters)  # type: ignore[union-attr]
 
         session_ids: set[str] = set()
         async for item in items:
diff --git a/python/packages/azure-cosmos/tests/test_cosmos_history_provider.py b/python/packages/azure-cosmos/tests/test_cosmos_history_provider.py
@@ -9,17 +9,15 @@
 from typing import Any
 from unittest.mock import AsyncMock, MagicMock, patch
 
+import agent_framework_azure_cosmos._history_provider as history_provider_module
 import pytest
 from agent_framework import AgentResponse, Message
 from agent_framework._sessions import AgentSession, SessionContext
 from agent_framework.exceptions import SettingNotFoundError
+from agent_framework_azure_cosmos._history_provider import CosmosHistoryProvider
 from azure.cosmos.aio import CosmosClient
 from azure.cosmos.exceptions import CosmosResourceNotFoundError
 
-import agent_framework_azure_cosmos._history_provider as history_provider_module
-from agent_framework_azure_cosmos._history_provider import CosmosHistoryProvider
-
-
 skip_if_cosmos_integration_tests_disabled = pytest.mark.skipif(
     any(
         os.getenv(name, "") == ""
@@ -209,7 +207,6 @@ async def test_list_sessions_returns_unique_sorted_ids(self, mock_container: Mag
         kwargs = mock_container.query_items.call_args.kwargs
         assert kwargs["query"] == "SELECT DISTINCT VALUE c.session_id FROM c WHERE c.source_id = @source_id"
         assert kwargs["parameters"] == [{"name": "@source_id", "value": "mem"}]
-        assert kwargs["enable_cross_partition_query"] is True
 
 
 class TestCosmosHistoryProviderSaveMessages:
diff --git a/python/uv.lock b/python/uv.lock
