fix(demo): inline ternary to fix mypy assignment type error

Also:
- chore(pyproject): ignore E501 in tests
- test(client): 16 with_raw_response tests (sync + async)
- test(opencode): 7 ask_stream unit tests

Author: skislyakow

demo.py

@@ -51,8 +51,7 @@
 # 8. Code search
 print("\n[8] Searching text...")
 found = client.find_text("create_opencode")
-n = len(found) if isinstance(found, list) else "?"
-print(f"    Found: {n}")
+print(f"    Found: {len(found) if isinstance(found, list) else '?'}")
 
 # 9. Send prompt (no LLM — queue only)
 print("\n[9] Sending prompt (no API key)...")
@@ -69,15 +68,23 @@
 # 10. Session context messages
 print("\n[10] Session messages...")
 ctx = client.v2_session_context(sid)
-items = ctx.get("data", []) if isinstance(ctx, dict) else ctx if isinstance(ctx, list) else []
+items = (
+    ctx.get("data", [])
+    if isinstance(ctx, dict)
+    else ctx
+    if isinstance(ctx, list)
+    else []
+)
 print(f"     Messages: {len(items)}")
 
 # 11. Extra capabilities
 print("\n[11] Additional:")
 path = client.path_get()
 print(f"     Working directory: {path.worktree}")
 cmds = client.command_list()
-print(f"     Opencode commands: {len(cmds) if isinstance(cmds, list) else '?'}")
+print(
+    f"     Opencode commands: {len(cmds) if isinstance(cmds, list) else '?'}"
+)
 agents = client.app_agents()
 print(f"     Agents: {len(agents) if isinstance(agents, list) else '?'}")
 

pyproject.toml

@@ -64,6 +64,7 @@ select = ["E", "F", "I", "N", "W", "UP"]
 
 [tool.ruff.lint.per-file-ignores]
 "src/opencode/_response_models.py" = ["N815"]
+"tests/*" = ["E501"]
 
 [tool.mypy]
 python_version = "3.10"

tests/test_async_client.py

@@ -7,6 +7,7 @@
 from opencode._response_models import (
     FileContentResponse,
     HealthResponse,
+    RawResponse,
     SessionResponse,
     V1SessionResponse,
 )
@@ -150,3 +151,114 @@ async def test_file_read() -> None:
     result = await client.file_read("/path/to/file.py")
     assert isinstance(result, FileContentResponse)
     assert result.content == "print('hello')"
+
+
+# ---------------------------------------------------------------------------
+# with_raw_response
+# ---------------------------------------------------------------------------
+
+
+@pytest.mark.asyncio
+async def test_with_raw_response_parsed_model() -> None:
+    client = AsyncOpendcodeClient(
+        base_url="http://localhost:9999",
+        httpx_client=httpx.AsyncClient(
+            transport=httpx.MockTransport(lambda _: httpx.Response(200, json={"ok": True}))
+        ),
+    )
+    with client.with_raw_response:
+        raw = await client.health()
+    assert isinstance(raw, RawResponse)
+    assert isinstance(raw.parsed, HealthResponse)
+    assert raw.parsed.ok is True
+
+
+@pytest.mark.asyncio
+async def test_with_raw_response_status_code() -> None:
+    client = AsyncOpendcodeClient(
+        base_url="http://localhost:9999",
+        httpx_client=httpx.AsyncClient(
+            transport=httpx.MockTransport(lambda _: httpx.Response(201, json={"ok": True}))
+        ),
+    )
+    with client.with_raw_response:
+        raw = await client.health()
+    assert raw.status_code == 201
+
+
+@pytest.mark.asyncio
+async def test_with_raw_response_headers() -> None:
+    client = AsyncOpendcodeClient(
+        base_url="http://localhost:9999",
+        httpx_client=httpx.AsyncClient(
+            transport=httpx.MockTransport(
+                lambda _: httpx.Response(
+                    200, json={"ok": True}, headers={"x-custom": "test-val"}
+                )
+            )
+        ),
+    )
+    with client.with_raw_response:
+        raw = await client.health()
+    assert raw.headers["x-custom"] == "test-val"
+
+
+@pytest.mark.asyncio
+async def test_with_raw_response_204() -> None:
+    client = AsyncOpendcodeClient(
+        base_url="http://localhost:9999",
+        httpx_client=httpx.AsyncClient(
+            transport=httpx.MockTransport(lambda _: httpx.Response(204))
+        ),
+    )
+    with client.with_raw_response:
+        raw = await client.v2_session_wait("ses_1")
+    assert isinstance(raw, RawResponse)
+    assert raw.status_code == 204
+    assert raw.parsed is None
+
+
+@pytest.mark.asyncio
+async def test_with_raw_response_mode_resets() -> None:
+    client = AsyncOpendcodeClient(
+        base_url="http://localhost:9999",
+        httpx_client=httpx.AsyncClient(
+            transport=httpx.MockTransport(lambda _: httpx.Response(200, json={"ok": True}))
+        ),
+    )
+    with client.with_raw_response:
+        raw = await client.health()
+    assert isinstance(raw, RawResponse)
+    normal = await client.health()
+    assert isinstance(normal, HealthResponse)
+    assert not isinstance(normal, RawResponse)
+
+
+@pytest.mark.asyncio
+async def test_with_raw_response_still_raises_error() -> None:
+    client = AsyncOpendcodeClient(
+        base_url="http://localhost:9999",
+        httpx_client=httpx.AsyncClient(
+            transport=httpx.MockTransport(
+                lambda _: httpx.Response(500, json={"message": "boom"})
+            )
+        ),
+    )
+    with client.with_raw_response:
+        with pytest.raises(APIError) as exc:
+            await client.health()
+    assert exc.value.status == 500
+
+
+@pytest.mark.asyncio
+async def test_with_raw_response_cast_to_none() -> None:
+    client = AsyncOpendcodeClient(
+        base_url="http://localhost:9999",
+        httpx_client=httpx.AsyncClient(
+            transport=httpx.MockTransport(lambda _: httpx.Response(200, json={"foo": "bar"}))
+        ),
+    )
+    with client.with_raw_response:
+        raw = await client.global_dispose()
+    assert isinstance(raw, RawResponse)
+    assert raw.parsed == {"foo": "bar"}

tests/test_client.py

@@ -7,6 +7,7 @@
 from opencode._response_models import (
     FileContentResponse,
     HealthResponse,
+    RawResponse,
     SessionResponse,
     V1SessionResponse,
 )
@@ -123,3 +124,127 @@ def test_file_read() -> None:
     result = client.file_read("/path/to/file.py")
     assert isinstance(result, FileContentResponse)
     assert result.content == "print('hello')"
+
+
+# ---------------------------------------------------------------------------
+# with_raw_response
+# ---------------------------------------------------------------------------
+
+
+def test_with_raw_response_parsed_model() -> None:
+    client = OpencodeClient(
+        base_url="http://localhost:9999",
+        httpx_client=httpx.Client(
+            transport=httpx.MockTransport(lambda _: httpx.Response(200, json={"ok": True}))
+        ),
+    )
+    with client.with_raw_response:
+        raw = client.health()
+    assert isinstance(raw, RawResponse)
+    assert isinstance(raw.parsed, HealthResponse)
+    assert raw.parsed.ok is True
+
+
+def test_with_raw_response_status_code() -> None:
+    client = OpencodeClient(
+        base_url="http://localhost:9999",
+        httpx_client=httpx.Client(
+            transport=httpx.MockTransport(lambda _: httpx.Response(201, json={"ok": True}))
+        ),
+    )
+    with client.with_raw_response:
+        raw = client.health()
+    assert raw.status_code == 201
+
+
+def test_with_raw_response_headers() -> None:
+    client = OpencodeClient(
+        base_url="http://localhost:9999",
+        httpx_client=httpx.Client(
+            transport=httpx.MockTransport(
+                lambda _: httpx.Response(
+                    200,
+                    json={"ok": True},
+                    headers={"x-custom": "test-val"},
+                )
+            )
+        ),
+    )
+    with client.with_raw_response:
+        raw = client.health()
+    assert raw.headers["x-custom"] == "test-val"
+
+
+def test_with_raw_response_content() -> None:
+    client = OpencodeClient(
+        base_url="http://localhost:9999",
+        httpx_client=httpx.Client(
+            transport=httpx.MockTransport(
+                lambda _: httpx.Response(200, json={"ok": True})
+            )
+        ),
+    )
+    with client.with_raw_response:
+        raw = client.health()
+    assert raw.content is not None
+    assert b"ok" in raw.content
+
+
+def test_with_raw_response_204() -> None:
+    client = OpencodeClient(
+        base_url="http://localhost:9999",
+        httpx_client=httpx.Client(
+            transport=httpx.MockTransport(lambda _: httpx.Response(204))
+        ),
+    )
+    with client.with_raw_response:
+        raw = client.v2_session_wait("ses_1")
+    assert isinstance(raw, RawResponse)
+    assert raw.status_code == 204
+    assert raw.parsed is None
+
+
+def test_with_raw_response_mode_resets() -> None:
+    client = OpencodeClient(
+        base_url="http://localhost:9999",
+        httpx_client=httpx.Client(
+            transport=httpx.MockTransport(lambda _: httpx.Response(200, json={"ok": True}))
+        ),
+    )
+    # Inside context — returns RawResponse
+    with client.with_raw_response:
+        raw = client.health()
+    assert isinstance(raw, RawResponse)
+    # After context — returns normal model
+    normal = client.health()
+    assert isinstance(normal, HealthResponse)
+    assert not isinstance(normal, RawResponse)
+
+
+def test_with_raw_response_still_raises_error() -> None:
+    client = OpencodeClient(
+        base_url="http://localhost:9999",
+        httpx_client=httpx.Client(
+            transport=httpx.MockTransport(
+                lambda _: httpx.Response(500, json={"message": "boom"})
+            )
+        ),
+    )
+    with client.with_raw_response:
+        with pytest.raises(APIError) as exc:
+            client.health()
+    assert exc.value.status == 500
+
+
+def test_with_raw_response_cast_to_none() -> None:
+    """Methods without cast_to should still return RawResponse with raw parsed data."""
+    client = OpencodeClient(
+        base_url="http://localhost:9999",
+        httpx_client=httpx.Client(
+            transport=httpx.MockTransport(lambda _: httpx.Response(200, json={"foo": "bar"}))
+        ),
+    )
+    with client.with_raw_response:
+        raw = client.global_dispose()
+    assert isinstance(raw, RawResponse)
+    assert raw.parsed == {"foo": "bar"}