ed/manual_slop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(ai_client): isolation of current_tier using threading.local() for parallel agent safety

Browse Source
This commit is contained in:
Edward R. Gonzalez
2026-03-06 12:59:10 -05:00
parent 1fb6ebc4d0
commit 684a6d1d3b
11 changed files with 75 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
tests/test_mma_agent_focus_phase1.py
View File

@@ -7,15 +7,14 @@ state when logging comms and tools.
from src import ai_client
def reset_tier():
ai_client.current_tier = None
ai_client.set_current_tier(None)
yield
ai_client.current_tier = None
ai_client.set_current_tier(None)
def test_current_tier_variable_exists() -> None:
"""ai_client must expose a module-level current_tier variable."""
assert hasattr(ai_client, "current_tier")
# current_tier might be None or a default
pass
def test_get_current_tier_exists() -> None:
"""ai_client must expose a get_current_tier function."""
assert hasattr(ai_client, "get_current_tier")
assert callable(ai_client.get_current_tier)
def test_append_comms_has_source_tier_key() -> None:
"""Dict entries in comms log must have a 'source_tier' key."""
@@ -30,7 +29,7 @@ def test_append_comms_has_source_tier_key() -> None:
def test_append_comms_source_tier_none_when_unset() -> None:
"""When current_tier is None, source_tier in log must be None."""
ai_client.reset_session()
ai_client.current_tier = None
ai_client.set_current_tier(None)
ai_client._append_comms("OUT", "request", {"msg": "hello"})
log = ai_client.get_comms_log()
@@ -39,22 +38,22 @@ def test_append_comms_source_tier_none_when_unset() -> None:
def test_append_comms_source_tier_set_when_current_tier_set() -> None:
"""When current_tier is 'Tier 1', source_tier in log must be 'Tier 1'."""
ai_client.reset_session()
ai_client.current_tier = "Tier 1"
ai_client.set_current_tier("Tier 1")
ai_client._append_comms("OUT", "request", {"msg": "hello"})
log = ai_client.get_comms_log()
assert log[-1]["source_tier"] == "Tier 1"
ai_client.current_tier = None
ai_client.set_current_tier(None)
def test_append_comms_source_tier_tier2() -> None:
"""When current_tier is 'Tier 2', source_tier in log must be 'Tier 2'."""
ai_client.reset_session()
ai_client.current_tier = "Tier 2"
ai_client.set_current_tier("Tier 2")
ai_client._append_comms("OUT", "request", {"msg": "hello"})
log = ai_client.get_comms_log()
assert log[-1]["source_tier"] == "Tier 2"
ai_client.current_tier = None
ai_client.set_current_tier(None)
def test_append_tool_log_stores_dict(app_instance) -> None:
"""App._append_tool_log must store a dict in self._tool_log."""