Support Claude Code JSON schema execution

llm_connect/claude_code.py

@@ -3,7 +3,10 @@ Claude Code CLI adapter — runs the ``claude`` CLI as a subprocess.
 """
 
 import asyncio
+import json
+import os
 import subprocess
+from pathlib import Path
 from typing import Optional
 
 from llm_connect.adapter import LLMAdapter
@@ -25,21 +28,19 @@ class ClaudeCodeAdapter(LLMAdapter):
 
     def __init__(
         self,
-        cli_path: str = "claude",
+        cli_path: Optional[str] = None,
         model: Optional[str] = None,
         config: Optional[LLMConfig] = None,
     ):
         self._config = config or LLMConfig(provider="claude-code")
-        self._cli_path = cli_path or self._config.claude_cli_path
+        self._cli_path = cli_path or self._resolve_cli_path()
         self._model = model
 
     # ── LLMAdapter interface ────────────────────────────────────────
 
     def execute_prompt(self, prompt: str, config: RunConfig) -> LLMResponse:
         self._preflight_budget(config)
-        cmd = [self._cli_path, "--print"]
+        cmd = self._build_command(config)
-        if self._model:
-            cmd.extend(["--model", self._model])
 
         timeout = config.timeout_seconds or self._config.timeout_seconds
 
@@ -88,9 +89,7 @@ class ClaudeCodeAdapter(LLMAdapter):
     async def async_execute_prompt(self, prompt: str, config: RunConfig) -> LLMResponse:
         """Native async implementation using asyncio.create_subprocess_exec."""
         self._preflight_budget(config)
-        cmd = [self._cli_path, "--print"]
+        cmd = self._build_command(config)
-        if self._model:
-            cmd.extend(["--model", self._model])
 
         timeout = config.timeout_seconds or self._config.timeout_seconds
 
@@ -151,3 +150,38 @@ class ClaudeCodeAdapter(LLMAdapter):
             return result.returncode == 0
         except (subprocess.TimeoutExpired, FileNotFoundError, OSError):
             return False
+
+    def _build_command(self, config: RunConfig) -> list[str]:
+        cmd = [self._cli_path, "--print"]
+        if self._model:
+            cmd.extend(["--model", self._model])
+
+        json_schema = _json_schema_arg(config)
+        if json_schema:
+            cmd.extend(["--json-schema", json_schema])
+        return cmd
+
+    def _resolve_cli_path(self) -> str:
+        configured = (
+            os.environ.get("LLM_CONNECT_CLAUDE_CLI_PATH")
+            or os.environ.get("CLAUDE_CLI_PATH")
+            or self._config.claude_cli_path
+        )
+        if configured and configured != "claude":
+            return configured
+
+        local_cli = Path.home() / ".local" / "bin" / "claude"
+        if local_cli.exists():
+            return str(local_cli)
+        return configured or "claude"
+
+
+def _json_schema_arg(config: RunConfig) -> str | None:
+    schema = (config.model_params or {}).get("json_schema")
+    if not schema:
+        return None
+    if isinstance(schema, str):
+        return schema
+    if isinstance(schema, dict):
+        return json.dumps(schema, separators=(",", ":"))
+    return None

tests/test_claude_code.py

@@ -0,0 +1,58 @@
+from __future__ import annotations
+
+from types import SimpleNamespace
+
+from llm_connect.claude_code import ClaudeCodeAdapter
+from llm_connect.config import LLMConfig
+from llm_connect.models import RunConfig
+
+
+def test_execute_prompt_passes_json_schema_to_claude_cli(monkeypatch):
+    calls: dict[str, object] = {}
+
+    def fake_run(cmd, input, capture_output, text, timeout):  # noqa: ANN001
+        calls["cmd"] = cmd
+        calls["input"] = input
+        calls["capture_output"] = capture_output
+        calls["text"] = text
+        calls["timeout"] = timeout
+        return SimpleNamespace(
+            returncode=0,
+            stdout='{"summary":"ok","recommendations":[]}',
+            stderr="",
+        )
+
+    monkeypatch.setattr("llm_connect.claude_code.subprocess.run", fake_run)
+    adapter = ClaudeCodeAdapter(cli_path="/custom/claude")
+
+    response = adapter.execute_prompt(
+        "Produce a report.",
+        RunConfig(
+            timeout_seconds=42,
+            model_params={"json_schema": {"type": "object"}},
+        ),
+    )
+
+    assert calls == {
+        "cmd": [
+            "/custom/claude",
+            "--print",
+            "--json-schema",
+            '{"type":"object"}',
+        ],
+        "input": "Produce a report.",
+        "capture_output": True,
+        "text": True,
+        "timeout": 42,
+    }
+    assert response.content == '{"summary":"ok","recommendations":[]}'
+
+
+def test_claude_code_adapter_prefers_env_cli_path(monkeypatch):
+    monkeypatch.setenv("LLM_CONNECT_CLAUDE_CLI_PATH", "/home/me/bin/claude")
+
+    adapter = ClaudeCodeAdapter(
+        config=LLMConfig(provider="claude-code", claude_cli_path="claude")
+    )
+
+    assert adapter._cli_path == "/home/me/bin/claude"