use consistency constants in more places

Signed-off-by: Filinto Duran <1373693+filintod@users.noreply.github.com>

docs/sessions/index.md

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

tests/extensions/memory/test_dapr_redis_integration.py

@@ -25,7 +25,11 @@
 from testcontainers.redis import RedisContainer # type: ignore[import-untyped]
 
 from agents import Agent, Runner, TResponseInputItem
-from agents.extensions.memory.dapr_session import DaprSession
+from agents.extensions.memory.dapr_session import (
+ CONSISTENCY_EVENTUAL,
+ CONSISTENCY_STRONG,
+ DaprSession,
+)
 from tests.fake_model import FakeModel
 from tests.test_responses import get_text_message
 
@@ -380,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="eventual",
+ consistency=CONSISTENCY_EVENTUAL,
  )
 
  # Test strong consistency
  session_strong = DaprSession.from_address(
  session_id="strong_consistency_test",
  state_store_name="statestore",
  dapr_address=dapr_address,
- consistency="strong",
+ consistency=CONSISTENCY_STRONG,
  )
 
  try:

tests/extensions/memory/test_dapr_session.py

@@ -8,7 +8,11 @@
 pytest.importorskip("dapr") # Skip tests if Dapr is not installed
 
 from agents import Agent, Runner, TResponseInputItem
-from agents.extensions.memory.dapr_session import DaprSession
+from agents.extensions.memory.dapr_session import (
+ CONSISTENCY_EVENTUAL,
+ CONSISTENCY_STRONG,
+ DaprSession,
+)
 from tests.fake_model import FakeModel
 from tests.test_responses import get_text_message
 
@@ -448,15 +452,15 @@ async def test_consistency_levels(fake_dapr_client: FakeDaprClient):
  session_id="eventual_test",
  state_store_name="statestore",
  dapr_client=fake_dapr_client, # type: ignore[arg-type]
- consistency="eventual",
+ consistency=CONSISTENCY_EVENTUAL,
  )
 
  # Test strong consistency
  session_strong = DaprSession(
  session_id="strong_test",
  state_store_name="statestore",
  dapr_client=fake_dapr_client, # type: ignore[arg-type]
- consistency="strong",
+ consistency=CONSISTENCY_STRONG,
  )
 
  try: