Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license
This commit is contained in:
@@ -0,0 +1,37 @@
|
||||
"""Regression coverage for crashed TUI gateway turns."""
|
||||
|
||||
import threading
|
||||
import types
|
||||
|
||||
from tui_gateway import server
|
||||
|
||||
|
||||
def test_turn_error_restores_agent_transcript_to_gateway_history():
|
||||
agent = types.SimpleNamespace(
|
||||
_session_messages=[
|
||||
{"role": "user", "content": "first"},
|
||||
{"role": "assistant", "content": "completed work"},
|
||||
]
|
||||
)
|
||||
session = {
|
||||
"history": [{"role": "user", "content": "first"}],
|
||||
"history_lock": threading.Lock(),
|
||||
"history_version": 0,
|
||||
}
|
||||
|
||||
assert server._restore_agent_history_after_turn_error(session, agent) is True
|
||||
assert session["history"] == agent._session_messages
|
||||
assert session["history_version"] == 1
|
||||
|
||||
|
||||
def test_turn_error_with_no_agent_transcript_does_not_overwrite_history():
|
||||
agent = types.SimpleNamespace(_session_messages=None)
|
||||
session = {
|
||||
"history": [{"role": "user", "content": "first"}],
|
||||
"history_lock": threading.Lock(),
|
||||
"history_version": 4,
|
||||
}
|
||||
|
||||
assert server._restore_agent_history_after_turn_error(session, agent) is False
|
||||
assert session["history"] == [{"role": "user", "content": "first"}]
|
||||
assert session["history_version"] == 4
|
||||
Reference in New Issue
Block a user