feat: opencode runner (#22)

Co-authored-by: banteg <4562643+banteg@users.noreply.github.com>
2026-01-02 09:57:19 -03:00
parent 1204524bef
commit 7c30674e53
9 changed files with 1189 additions and 0 deletions
@@ -0,0 +1,2 @@
+{"type":"step_start","timestamp":1767037000000,"sessionID":"ses_error123","part":{"id":"prt_error1","sessionID":"ses_error123","messageID":"msg_error1","type":"step-start"}}
+{"type":"error","timestamp":1767037001000,"sessionID":"ses_error123","error":{"name":"APIError","data":{"message":"Rate limit exceeded","statusCode":429,"isRetryable":true}}}
@@ -0,0 +1,6 @@
+{"type":"step_start","timestamp":1767036059338,"sessionID":"ses_494719016ffe85dkDMj0FPRbHK","part":{"id":"prt_b6b8e7ec7001qAZUB7eTENxPpI","sessionID":"ses_494719016ffe85dkDMj0FPRbHK","messageID":"msg_b6b8e702b0012XuEC4bGe0XhKa","type":"step-start","snapshot":"71db24a798b347669c0ebadb2dfad238f991753d"}}
+{"type":"tool_use","timestamp":1767036061199,"sessionID":"ses_494719016ffe85dkDMj0FPRbHK","part":{"id":"prt_b6b8e85bb001CzBoN2dDlEZJnP","sessionID":"ses_494719016ffe85dkDMj0FPRbHK","messageID":"msg_b6b8e702b0012XuEC4bGe0XhKa","type":"tool","callID":"r9bQWsNLvOrJGIOz","tool":"bash","state":{"status":"completed","input":{"command":"echo hello","description":"Print hello to stdout"},"output":"hello\n","title":"Print hello to stdout","metadata":{"output":"hello\n","exit":0,"description":"Print hello to stdout"},"time":{"start":1767036061123,"end":1767036061173}}}}
+{"type":"step_finish","timestamp":1767036061205,"sessionID":"ses_494719016ffe85dkDMj0FPRbHK","part":{"id":"prt_b6b8e85fb001L4I3WHMqH6EQNI","sessionID":"ses_494719016ffe85dkDMj0FPRbHK","messageID":"msg_b6b8e702b0012XuEC4bGe0XhKa","type":"step-finish","reason":"tool-calls","snapshot":"ee3406d50c7d9048674bbb1a3e325d82513b74ed","cost":0,"tokens":{"input":21772,"output":110,"reasoning":0,"cache":{"read":0,"write":0}}}}
+{"type":"step_start","timestamp":1767036063732,"sessionID":"ses_494719016ffe85dkDMj0FPRbHK","part":{"id":"prt_b6b8e8ff2001hIElz1HRSMdHJY","sessionID":"ses_494719016ffe85dkDMj0FPRbHK","messageID":"msg_b6b8e8627001yM4qKJCXdC7W1L","type":"step-start","snapshot":"9017313c64af88e12921b4c81d57fd4806192416"}}
+{"type":"text","timestamp":1767036064268,"sessionID":"ses_494719016ffe85dkDMj0FPRbHK","part":{"id":"prt_b6b8e8ff2002mxSx9LtvAlf8Ng","sessionID":"ses_494719016ffe85dkDMj0FPRbHK","messageID":"msg_b6b8e8627001yM4qKJCXdC7W1L","type":"text","text":"```\nhello\n```","time":{"start":1767036064265,"end":1767036064265}}}
+{"type":"step_finish","timestamp":1767036064273,"sessionID":"ses_494719016ffe85dkDMj0FPRbHK","part":{"id":"prt_b6b8e9209001ojZ4ECN1geZISm","sessionID":"ses_494719016ffe85dkDMj0FPRbHK","messageID":"msg_b6b8e8627001yM4qKJCXdC7W1L","type":"step-finish","reason":"stop","snapshot":"09dd05d11a4ac013136c1df10932efc0ad9116e8","cost":0.001,"tokens":{"input":671,"output":8,"reasoning":0,"cache":{"read":21415,"write":0}}}}
@@ -0,0 +1,3 @@
+{"type":"step_start","timestamp":1767038000000,"sessionID":"ses_no_reason","part":{"id":"prt_nr_start","sessionID":"ses_no_reason","messageID":"msg_nr_1","type":"step-start"}}
+{"type":"text","timestamp":1767038000500,"sessionID":"ses_no_reason","part":{"id":"prt_nr_text","sessionID":"ses_no_reason","messageID":"msg_nr_1","type":"text","text":"All done.","time":{"start":1767038000500,"end":1767038000500}}}
+{"type":"step_finish","timestamp":1767038001000,"sessionID":"ses_no_reason","part":{"id":"prt_nr_finish","sessionID":"ses_no_reason","messageID":"msg_nr_1","type":"step-finish","cost":0.002,"tokens":{"input":12,"output":3,"reasoning":0,"cache":{"read":0,"write":0}}}}
@@ -0,0 +1,341 @@
+import json
+from pathlib import Path
+
+import anyio
+import pytest
+
+from takopi.model import ActionEvent, CompletedEvent, ResumeToken, StartedEvent
+from takopi.runners.opencode import (
+    OpenCodeRunner,
+    OpenCodeStreamState,
+    ENGINE,
+    translate_opencode_event,
+)
+
+
+def _load_fixture(name: str) -> list[dict]:
+    path = Path(__file__).parent / "fixtures" / name
+    return [json.loads(line) for line in path.read_text().splitlines() if line.strip()]
+
+
+def test_opencode_resume_format_and_extract() -> None:
+    runner = OpenCodeRunner(opencode_cmd="opencode")
+    token = ResumeToken(engine=ENGINE, value="ses_abc123")
+
+    assert runner.format_resume(token) == "`opencode --session ses_abc123`"
+    assert runner.extract_resume("`opencode --session ses_abc123`") == token
+    assert runner.extract_resume("opencode run -s ses_other") == ResumeToken(
+        engine=ENGINE, value="ses_other"
+    )
+    assert runner.extract_resume("opencode -s ses_other") == ResumeToken(
+        engine=ENGINE, value="ses_other"
+    )
+    assert runner.extract_resume("`claude --resume sid`") is None
+    assert runner.extract_resume("`codex resume sid`") is None
+
+
+def test_translate_success_fixture() -> None:
+    state = OpenCodeStreamState()
+    events: list = []
+    for event in _load_fixture("opencode_stream_success.jsonl"):
+        events.extend(translate_opencode_event(event, title="opencode", state=state))
+
+    assert isinstance(events[0], StartedEvent)
+    started = next(evt for evt in events if isinstance(evt, StartedEvent))
+    assert started.resume.value == "ses_494719016ffe85dkDMj0FPRbHK"
+    assert started.resume.engine == ENGINE
+
+    action_events = [evt for evt in events if isinstance(evt, ActionEvent)]
+    assert len(action_events) == 1
+
+    completed_actions = [evt for evt in action_events if evt.phase == "completed"]
+    assert len(completed_actions) == 1
+    assert completed_actions[0].action.kind == "command"
+    assert completed_actions[0].ok is True
+
+    completed = next(evt for evt in events if isinstance(evt, CompletedEvent))
+    assert events[-1] == completed
+    assert completed.ok is True
+    assert completed.resume == started.resume
+    assert completed.answer == "```\nhello\n```"
+
+    assert completed.usage is not None
+    assert "tokens" in completed.usage
+
+
+def test_translate_missing_reason_success() -> None:
+    state = OpenCodeStreamState()
+    events: list = []
+    for event in _load_fixture("opencode_stream_success_no_reason.jsonl"):
+        events.extend(translate_opencode_event(event, title="opencode", state=state))
+
+    started = next(evt for evt in events if isinstance(evt, StartedEvent))
+    runner = OpenCodeRunner(opencode_cmd="opencode")
+    fallback = runner.stream_end_events(
+        resume=None,
+        found_session=started.resume,
+        stderr_tail="",
+        state=state,
+    )
+
+    completed = next(evt for evt in fallback if isinstance(evt, CompletedEvent))
+    assert completed.ok is True
+    assert completed.resume == started.resume
+    assert completed.answer == "All done."
+    assert completed.usage is not None
+
+
+def test_translate_accumulates_text() -> None:
+    state = OpenCodeStreamState()
+
+    events = translate_opencode_event(
+        {"type": "step_start", "sessionID": "ses_test123", "part": {}},
+        title="opencode",
+        state=state,
+    )
+    assert len(events) == 1
+    assert isinstance(events[0], StartedEvent)
+
+    translate_opencode_event(
+        {
+            "type": "text",
+            "sessionID": "ses_test123",
+            "part": {"type": "text", "text": "Hello "},
+        },
+        title="opencode",
+        state=state,
+    )
+    translate_opencode_event(
+        {
+            "type": "text",
+            "sessionID": "ses_test123",
+            "part": {"type": "text", "text": "World"},
+        },
+        title="opencode",
+        state=state,
+    )
+
+    assert state.last_text == "Hello World"
+
+    events = translate_opencode_event(
+        {
+            "type": "step_finish",
+            "sessionID": "ses_test123",
+            "part": {"reason": "stop", "tokens": {"input": 100, "output": 10}},
+        },
+        title="opencode",
+        state=state,
+    )
+
+    assert len(events) == 1
+    completed = events[0]
+    assert isinstance(completed, CompletedEvent)
+    assert completed.answer == "Hello World"
+    assert completed.ok is True
+
+
+def test_translate_tool_use_completed() -> None:
+    state = OpenCodeStreamState()
+    state.session_id = "ses_test123"
+    state.emitted_started = True
+
+    events = translate_opencode_event(
+        {
+            "type": "tool_use",
+            "sessionID": "ses_test123",
+            "part": {
+                "id": "prt_123",
+                "callID": "call_abc",
+                "tool": "bash",
+                "state": {
+                    "status": "completed",
+                    "input": {"command": "ls -la"},
+                    "output": "file1.txt\nfile2.txt",
+                    "title": "List files",
+                    "metadata": {"exit": 0},
+                },
+            },
+        },
+        title="opencode",
+        state=state,
+    )
+
+    assert len(events) == 1
+    action_event = events[0]
+    assert isinstance(action_event, ActionEvent)
+    assert action_event.phase == "completed"
+    assert action_event.action.kind == "command"
+    assert action_event.action.title == "List files"
+    assert action_event.ok is True
+
+
+def test_translate_tool_use_with_error() -> None:
+    state = OpenCodeStreamState()
+    state.session_id = "ses_test123"
+    state.emitted_started = True
+
+    events = translate_opencode_event(
+        {
+            "type": "tool_use",
+            "sessionID": "ses_test123",
+            "part": {
+                "id": "prt_123",
+                "callID": "call_abc",
+                "tool": "bash",
+                "state": {
+                    "status": "completed",
+                    "input": {"command": "exit 1"},
+                    "output": "error",
+                    "title": "Run failing command",
+                    "metadata": {"exit": 1},
+                },
+            },
+        },
+        title="opencode",
+        state=state,
+    )
+
+    assert len(events) == 1
+    action_event = events[0]
+    assert isinstance(action_event, ActionEvent)
+    assert action_event.phase == "completed"
+    assert action_event.ok is False
+
+
+def test_translate_tool_use_read_title_wraps_path() -> None:
+    state = OpenCodeStreamState()
+    state.session_id = "ses_test123"
+    state.emitted_started = True
+    path = Path.cwd() / "src" / "takopi" / "runners" / "opencode.py"
+
+    events = translate_opencode_event(
+        {
+            "type": "tool_use",
+            "sessionID": "ses_test123",
+            "part": {
+                "id": "prt_123",
+                "callID": "call_abc",
+                "tool": "read",
+                "state": {
+                    "status": "completed",
+                    "input": {"filePath": str(path)},
+                    "output": "file contents",
+                    "title": "src/takopi/runners/opencode.py",
+                },
+            },
+        },
+        title="opencode",
+        state=state,
+    )
+
+    assert len(events) == 1
+    action_event = events[0]
+    assert isinstance(action_event, ActionEvent)
+    assert action_event.action.kind == "tool"
+    assert action_event.action.title == "`src/takopi/runners/opencode.py`"
+
+
+def test_translate_error_fixture() -> None:
+    state = OpenCodeStreamState()
+    events: list = []
+    for event in _load_fixture("opencode_stream_error.jsonl"):
+        events.extend(translate_opencode_event(event, title="opencode", state=state))
+
+    started = next(evt for evt in events if isinstance(evt, StartedEvent))
+    completed = next(evt for evt in events if isinstance(evt, CompletedEvent))
+
+    assert completed.ok is False
+    assert completed.error == "Rate limit exceeded"
+    assert completed.resume == started.resume
+
+
+def test_step_finish_tool_calls_does_not_complete() -> None:
+    state = OpenCodeStreamState()
+    state.session_id = "ses_test123"
+    state.emitted_started = True
+
+    events = translate_opencode_event(
+        {
+            "type": "step_finish",
+            "sessionID": "ses_test123",
+            "part": {"reason": "tool-calls", "tokens": {"input": 100, "output": 10}},
+        },
+        title="opencode",
+        state=state,
+    )
+
+    assert len(events) == 0
+
+
+def test_build_args_new_session() -> None:
+    runner = OpenCodeRunner(opencode_cmd="opencode", model="claude-sonnet")
+    args = runner.build_args("hello world", None, state=OpenCodeStreamState())
+
+    assert args == [
+        "run",
+        "--format",
+        "json",
+        "--model",
+        "claude-sonnet",
+        "--",
+        "hello world",
+    ]
+
+
+def test_build_args_with_resume() -> None:
+    runner = OpenCodeRunner(opencode_cmd="opencode")
+    token = ResumeToken(engine=ENGINE, value="ses_abc123")
+    args = runner.build_args("continue", token, state=OpenCodeStreamState())
+
+    assert args == [
+        "run",
+        "--format",
+        "json",
+        "--session",
+        "ses_abc123",
+        "--",
+        "continue",
+    ]
+
+
+def test_stdin_payload_returns_none() -> None:
+    runner = OpenCodeRunner(opencode_cmd="opencode")
+    payload = runner.stdin_payload("prompt", None, state=OpenCodeStreamState())
+    assert payload is None
+
+
+@pytest.mark.anyio
+async def test_run_serializes_same_session() -> None:
+    runner = OpenCodeRunner(opencode_cmd="opencode")
+    gate = anyio.Event()
+    in_flight = 0
+    max_in_flight = 0
+
+    async def run_stub(*_args, **_kwargs):
+        nonlocal in_flight, max_in_flight
+        in_flight += 1
+        max_in_flight = max(max_in_flight, in_flight)
+        try:
+            await gate.wait()
+            yield CompletedEvent(
+                engine=ENGINE,
+                resume=ResumeToken(engine=ENGINE, value="ses_test"),
+                ok=True,
+                answer="ok",
+            )
+        finally:
+            in_flight -= 1
+
+    runner.run_impl = run_stub  # type: ignore[assignment]
+
+    async def drain(prompt: str, resume: ResumeToken | None) -> None:
+        async for _event in runner.run(prompt, resume):
+            pass
+
+    token = ResumeToken(engine=ENGINE, value="ses_test")
+    async with anyio.create_task_group() as tg:
+        tg.start_soon(drain, "a", token)
+        tg.start_soon(drain, "b", token)
+        await anyio.sleep(0)
+        gate.set()
+    assert max_in_flight == 1