Guard protocol generation against context truncation
This commit is contained in:
@@ -232,6 +232,7 @@ class DirectProtocolCliTests(unittest.TestCase):
|
||||
return_value=type("Result", (), {
|
||||
"protocol_text": protocol_text,
|
||||
"exact_prompt": "exact prompt\n",
|
||||
"transcript_input": "selected transcript\n",
|
||||
"raw_response": {"response": protocol_text},
|
||||
"runtime_metadata": {"request_count": 1},
|
||||
})(),
|
||||
@@ -245,6 +246,10 @@ class DirectProtocolCliTests(unittest.TestCase):
|
||||
(run_dir / "protocol/exact_prompt.txt").read_text(encoding="utf-8"),
|
||||
"exact prompt\n",
|
||||
)
|
||||
self.assertEqual(
|
||||
(run_dir / "protocol/transcript_input.txt").read_text(encoding="utf-8"),
|
||||
"selected transcript\n",
|
||||
)
|
||||
self.assertEqual(
|
||||
json.loads((run_dir / "protocol/raw_response.json").read_text())["response"],
|
||||
protocol_text,
|
||||
|
||||
Reference in New Issue
Block a user