patch.py

Go to the documentation of this file.

# Copyright (c) 2025-present Aquin Labs Private Limited. All Rights Reserved.
# This file is part of the Aquin Engine. Unauthorized copying, modification,
# distribution, or use of this file, via any medium, is strictly prohibited.
# Proprietary and confidential. See LICENSE for terms.
 
"""Patch — steer / sweep / multi-steer / vector save."""
 
from __future__ import annotations
 
from typing import Any
 
from aquin.sdk._runtime import invoke
 
 
def steer(**kwargs: Any) -> dict[str, Any]:
    if kwargs.get("save") or kwargs.get("save_path"):
        return invoke("extract_steer_vector", kwargs, command="steer --save")
    return invoke("run_steer_and_show", kwargs, command="steer")
 
 
def multi_steer(**kwargs: Any) -> dict[str, Any]:
    return invoke("run_multi_steer", kwargs, command="multi-steer")
 
 
def save_vector(**kwargs: Any) -> dict[str, Any]:
    return invoke("extract_steer_vector", kwargs, command="steer --save")
 
 
def sweep(**kwargs: Any) -> dict[str, Any]:
    """Strength grid over a steer intervention (uses CLI sweep engine)."""
    from aquin.sweep_cli import cmd_sweep
 
    # Prefer structured kwargs → temporary argv for existing sweep implementation
    argv: list[str] = []
    mapping = {
        "feature_idx": "--feature_idx",
        "feature": "--feature_idx",
        "strengths": "--strengths",
        "prompt": "--prompt",
        "eval": "--eval",
        "layer": "--layer",
    }
    for key, flag in mapping.items():
        if key in kwargs and kwargs[key] is not None:
            val = kwargs[key]
            if key == "eval" and val is True:
                argv.append(flag)
            else:
                argv.extend([flag, str(val)])
    cmd_sweep(argv)
    return {"ok": True, "argv": argv}