Method _run_module_agent_sync

codewiki/src/be/caw_backend.py:206–324

(
        self,
        module_name: str,
        components: Dict[str, Node],
        core_component_ids: List[str],
        module_path: List[str],
        working_dir: str,
        start_depth: int = 1,
        module_tree: Dict[str, Any] | None = None,
    )

_patch_claude_allowed_tools()
# --- end stopgap --------------------------------------------------------------


class CawBackend(LLMBackend):
    """Routes LLM operations through the claude / codex CLI subscription."""

    def __init__(self, config: Config) -> None:
        self._config = config
        self._caw_provider = _resolve_caw_provider(config.provider)
        # main_model is passed straight through; empty string → caw default.
        self._model: str | None = config.main_model or None
        # Resolve once, before any agent-run chdir can move the cwd —
        # os.path.abspath is cwd-relative and _run_module_agent_sync recurses
        # (via generate_sub_module_documentation) while the cwd is pinned.
        self._repo_root = str(os.path.abspath(config.repo_path))

        # Fail loudly here rather than producing a confusing caw error mid-run.
        cli = _CLI_BINARY[config.provider]
        if shutil.which(cli) is None:
            raise RuntimeError(
                f"Subscription mode requires the '{cli}' CLI on PATH. "
                f"Install it and run '{cli} login', then try again."
            )

        if self._caw_provider == "claude_code":
            # Prevent claude-code CLI from cancelling long sub-module recursion;
            # setdefault preserves a user-supplied value (e.g. shell override).
            os.environ.setdefault("MCP_TOOL_TIMEOUT", "86400000")
            os.environ.setdefault("MCP_TIMEOUT", "60000")
            logger.info(
                "claude-code MCP timeouts: MCP_TOOL_TIMEOUT=%s MCP_TIMEOUT=%s",
                os.environ["MCP_TOOL_TIMEOUT"],
                os.environ["MCP_TIMEOUT"],
            )

    # ------------------------------------------------------------------
    # Single-shot completion (clustering, parent / repo overviews)
    # ------------------------------------------------------------------

    def complete(
        self,
        prompt: str,
        *,
        model: str | None = None,
    ) -> str:
        # Blocks the calling thread for the lifetime of the claude/codex
        # subprocess.  Callers running this from an async context (e.g. the
        # documentation_generator) accept this — there is no concurrent work
        # to do while clustering is in flight anyway.
        effective_model = model or self._model
        agent = CawAgent(
            provider=self._caw_provider,
            model=effective_model,
            tools=ToolGroup.READER,
        )
        traj = agent.completion(prompt)
        self.last_usage = usage_to_dict(getattr(traj, "total_usage", None))
        return traj.result