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

fix(simulation): Resolve simulation timeouts and stabilize history checks

Browse Source
This commit is contained in:
Edward R. Gonzalez
2026-03-03 00:56:35 -05:00
parent aed1f9a97e
commit dbd955a45b
4 changed files with 145 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
simulation/sim_context.py
View File

@@ -45,11 +45,15 @@ class ContextSimulation(BaseSimulation):
msg = "What is the current date and time? Answer in one sentence."
print(f"[Sim] Sending message: {msg}")
self.sim.run_discussion_turn(msg)
time.sleep(10)
# 4. Verify History
print("[Sim] Verifying history...")
session = self.client.get_session()
entries = session.get('session', {}).get('entries', [])
if not entries:
print("[Sim] !!! WARNING: entries list is EMPTY. Waiting another 2 seconds for eventual consistency...")
time.sleep(2)
session = self.client.get_session()
entries = session.get('session', {}).get('entries', [])
# We expect at least 2 entries (User and AI)
assert len(entries) >= 2, f"Expected at least 2 entries, found {len(entries)}"
assert entries[-2]['role'] == 'User', "Expected second to last entry to be User"
@@ -61,9 +65,9 @@ class ContextSimulation(BaseSimulation):
time.sleep(1)
session = self.client.get_session()
entries = session.get('session', {}).get('entries', [])
# Truncating to 1 pair means 2 entries max (if it's already at 2, it might not change,
# but if we had more, it would).
assert len(entries) <= 2, f"Expected <= 2 entries after truncation, found {len(entries)}"
print(f"[DEBUG] Entries after truncation: {entries}")
chat_entries = [e for e in entries if e.get('role') in ('User', 'AI')]
assert len(chat_entries) == 2, f"Expected exactly 2 chat entries after truncation, found {len(chat_entries)}"
if __name__ == "__main__":
run_sim(ContextSimulation)