make consistency constants part of public interface again

Signed-off-by: Filinto Duran <[email protected]>

Author: filintod

docs/sessions/dapr_session.md

@@ -244,9 +244,9 @@ Consistency levels control how Dapr handles read/write operations across distrib
 Use the provided constants to avoid typos:
 
 ```python
-from agents.extensions.memory.dapr_session import (
-    CONSISTENCY_EVENTUAL,
-    CONSISTENCY_STRONG,
+from agents.extensions.memory import (
+    DAPR_CONSISTENCY_EVENTUAL,
+    DAPR_CONSISTENCY_STRONG,
     DaprSession,
 )
 
@@ -255,19 +255,19 @@ session = DaprSession.from_address(
     session_id="user-123",
     state_store_name="statestore",
     dapr_address="localhost:50001",
-    consistency=CONSISTENCY_EVENTUAL,  # or "eventual"
+    consistency=DAPR_CONSISTENCY_EVENTUAL,  # or "eventual"
 )
 
 # Strong consistency (guarantees read-after-write consistency)
 session = DaprSession.from_address(
     session_id="user-123",
     state_store_name="statestore",
     dapr_address="localhost:50001",
-    consistency=CONSISTENCY_STRONG,  # or "strong"
+    consistency=DAPR_CONSISTENCY_STRONG,  # or "strong"
 )
 ```
 
-**Important**: Consistency levels apply to both read and write operations. When using `CONSISTENCY_STRONG`, the session ensures that reads always reflect the most recent writes, preventing stale data after updates.
+**Important**: Consistency levels apply to both read and write operations. When using `DAPR_CONSISTENCY_STRONG`, the session ensures that reads always reflect the most recent writes, preventing stale data after updates.
 
 Support varies by state store. See [Dapr consistency documentation](https://docs.dapr.io/developing-applications/building-blocks/state-management/state-management-overview/#consistency).
 

docs/sessions/index.md

@@ -211,14 +211,14 @@ session = DaprSession.from_address(
 )
 
 # Optional: configure TTL and consistency
-from agents.extensions.memory import CONSISTENCY_STRONG
+from agents.extensions.memory import DAPR_CONSISTENCY_STRONG
 
 session = DaprSession.from_address(
     "user_123",
     state_store_name="statestore",
     dapr_address="localhost:50001",  # The default Dapr gRPC port is 50001 and can be omitted.
     ttl=3600,  # 1 hour
-    consistency=CONSISTENCY_STRONG,  # or CONSISTENCY_EVENTUAL
+    consistency=DAPR_CONSISTENCY_STRONG,  # or DAPR_CONSISTENCY_EVENTUAL
 )
 ```
 

examples/memory/dapr_session_example.py

@@ -19,8 +19,8 @@
 
 from agents import Agent, Runner
 from agents.extensions.memory import (
-    CONSISTENCY_EVENTUAL,
-    CONSISTENCY_STRONG,
+    DAPR_CONSISTENCY_EVENTUAL,
+    DAPR_CONSISTENCY_STRONG,
     DaprSession,
 )
 
@@ -183,15 +183,15 @@ async def demonstrate_advanced_features():
             "eventual_session",
             state_store_name="statestore",
             dapr_address="localhost:50001",
-            consistency=CONSISTENCY_EVENTUAL,
+            consistency=DAPR_CONSISTENCY_EVENTUAL,
         )
 
         # Strong consistency (guaranteed read-after-write)
         strong_session = DaprSession.from_address(
             "strong_session",
             state_store_name="statestore",
             dapr_address="localhost:50001",
-            consistency=CONSISTENCY_STRONG,
+            consistency=DAPR_CONSISTENCY_STRONG,
         )
 
         if await eventual_session.ping():

src/agents/extensions/memory/__init__.py

@@ -12,6 +12,8 @@
 
 __all__: list[str] = [
     "AdvancedSQLiteSession",
+    "DAPR_CONSISTENCY_EVENTUAL",
+    "DAPR_CONSISTENCY_STRONG",
     "DaprSession",
     "EncryptedSession",
     "RedisSession",
@@ -72,4 +74,26 @@ def __getattr__(name: str) -> Any:
                 "Install it with: pip install openai-agents[dapr]"
             ) from e
 
+    if name == "DAPR_CONSISTENCY_EVENTUAL":
+        try:
+            from .dapr_session import DAPR_CONSISTENCY_EVENTUAL  # noqa: F401
+
+            return DAPR_CONSISTENCY_EVENTUAL
+        except ModuleNotFoundError as e:
+            raise ImportError(
+                "DAPR_CONSISTENCY_EVENTUAL requires the 'dapr' extra. "
+                "Install it with: pip install openai-agents[dapr]"
+            ) from e
+
+    if name == "DAPR_CONSISTENCY_STRONG":
+        try:
+            from .dapr_session import DAPR_CONSISTENCY_STRONG  # noqa: F401
+
+            return DAPR_CONSISTENCY_STRONG
+        except ModuleNotFoundError as e:
+            raise ImportError(
+                "DAPR_CONSISTENCY_STRONG requires the 'dapr' extra. "
+                "Install it with: pip install openai-agents[dapr]"
+            ) from e
+
     raise AttributeError(f"module {__name__} has no attribute {name}")

src/agents/extensions/memory/dapr_session.py

@@ -43,8 +43,8 @@
 ConsistencyLevel = Literal["eventual", "strong"]
 
 # Consistency level constants
-CONSISTENCY_EVENTUAL: ConsistencyLevel = "eventual"
-CONSISTENCY_STRONG: ConsistencyLevel = "strong"
+DAPR_CONSISTENCY_EVENTUAL: ConsistencyLevel = "eventual"
+DAPR_CONSISTENCY_STRONG: ConsistencyLevel = "strong"
 
 
 class DaprSession(SessionABC):
@@ -57,7 +57,7 @@ def __init__(
         state_store_name: str,
         dapr_client: DaprClient,
         ttl: int | None = None,
-        consistency: ConsistencyLevel = CONSISTENCY_EVENTUAL,
+        consistency: ConsistencyLevel = DAPR_CONSISTENCY_EVENTUAL,
     ):
         """Initializes a new DaprSession.
 
@@ -125,9 +125,9 @@ def _get_read_metadata(self) -> dict[str, str]:
 
     def _get_state_options(self) -> StateOptions | None:
         """Get StateOptions for write/delete consistency level."""
-        if self._consistency == CONSISTENCY_STRONG:
+        if self._consistency == DAPR_CONSISTENCY_STRONG:
             return StateOptions(consistency=Consistency.strong)
-        elif self._consistency == CONSISTENCY_EVENTUAL:
+        elif self._consistency == DAPR_CONSISTENCY_EVENTUAL:
             return StateOptions(consistency=Consistency.eventual)
         return None
 

tests/extensions/memory/test_dapr_redis_integration.py

@@ -26,12 +26,10 @@
 
 from agents import Agent, Runner, TResponseInputItem
 from agents.extensions.memory import (
+    DAPR_CONSISTENCY_EVENTUAL,
+    DAPR_CONSISTENCY_STRONG,
     DaprSession,
 )
-from agents.extensions.memory.dapr_session import (
-    CONSISTENCY_EVENTUAL,
-    CONSISTENCY_STRONG,
-)
 from tests.fake_model import FakeModel
 from tests.test_responses import get_text_message
 
@@ -386,15 +384,15 @@ async def test_dapr_consistency_levels(dapr_container, monkeypatch):
         session_id="eventual_consistency_test",
         state_store_name="statestore",
         dapr_address=dapr_address,
-        consistency=CONSISTENCY_EVENTUAL,
+        consistency=DAPR_CONSISTENCY_EVENTUAL,
     )
 
     # Test strong consistency
     session_strong = DaprSession.from_address(
         session_id="strong_consistency_test",
         state_store_name="statestore",
         dapr_address=dapr_address,
-        consistency=CONSISTENCY_STRONG,
+        consistency=DAPR_CONSISTENCY_STRONG,
     )
 
     try:

tests/extensions/memory/test_dapr_session.py

@@ -10,12 +10,10 @@
 
 from agents import Agent, Runner, TResponseInputItem
 from agents.extensions.memory import (
+    DAPR_CONSISTENCY_EVENTUAL,
+    DAPR_CONSISTENCY_STRONG,
     DaprSession,
 )
-from agents.extensions.memory.dapr_session import (
-    CONSISTENCY_EVENTUAL,
-    CONSISTENCY_STRONG,
-)
 from tests.fake_model import FakeModel
 from tests.test_responses import get_text_message
 
@@ -455,15 +453,15 @@ async def test_consistency_levels(fake_dapr_client: FakeDaprClient):
         session_id="eventual_test",
         state_store_name="statestore",
         dapr_client=fake_dapr_client,
-        consistency=CONSISTENCY_EVENTUAL,
+        consistency=DAPR_CONSISTENCY_EVENTUAL,
     )
 
     # Test strong consistency
     session_strong = DaprSession(
         session_id="strong_test",
         state_store_name="statestore",
         dapr_client=fake_dapr_client,
-        consistency=CONSISTENCY_STRONG,
+        consistency=DAPR_CONSISTENCY_STRONG,
     )
 
     try: