Spaces:
Running on Zero
Running on Zero
File size: 2,770 Bytes
849ee7b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | from __future__ import annotations
import json
import tempfile
import unittest
from pathlib import Path
from parser import parse_trace
class ParserTests(unittest.TestCase):
def write_jsonl(self, records: list[dict]) -> Path:
handle = tempfile.NamedTemporaryFile("w", encoding="utf-8", suffix=".jsonl", delete=False)
with handle:
for record in records:
handle.write(json.dumps(record) + "\n")
return Path(handle.name)
def test_parses_codex_response_items_and_ignores_tools(self) -> None:
path = self.write_jsonl(
[
{"type": "session_meta", "payload": {"originator": "codex_cli"}},
{
"timestamp": "2026-06-06T00:00:00Z",
"type": "response_item",
"payload": {
"type": "message",
"role": "assistant",
"content": [
{"type": "output_text", "text": "I will inspect the failing path."},
{"type": "tool_call", "name": "exec", "arguments": "cat secret.py"},
],
},
},
{
"timestamp": "2026-06-06T00:00:02Z",
"type": "response_item",
"payload": {
"type": "message",
"role": "user",
"content": [{"type": "input_text", "text": "Please fix it."}],
},
},
]
)
messages, agent_type = parse_trace(path)
self.assertEqual(agent_type, "codex")
self.assertEqual([message.role for message in messages], ["assistant", "user"])
self.assertIn("inspect", messages[0].text)
self.assertNotIn("secret.py", messages[0].text)
def test_parses_claude_message_shape(self) -> None:
path = self.write_jsonl(
[
{
"parentUuid": None,
"sessionId": "session",
"type": "assistant",
"timestamp": "2026-06-06T00:00:00Z",
"message": {
"role": "assistant",
"content": [{"type": "text", "text": "The test failed, so I will verify another path."}],
},
}
]
)
messages, agent_type = parse_trace(path)
self.assertEqual(agent_type, "claude_code")
self.assertEqual(len(messages), 1)
self.assertEqual(messages[0].role, "assistant")
self.assertIn("test failed", messages[0].text)
if __name__ == "__main__":
unittest.main()
|