kiapi 0.1.0__py3-none-any.whl

kiapi/cli/activate/_operations/preflight_hf_access.py

@@ -0,0 +1,129 @@
+from dataclasses import dataclass
+from typing import Protocol
+from urllib.parse import quote
+
+import click
+from huggingface_hub import HfApi, get_token
+from huggingface_hub.utils import (
+    GatedRepoError,  # pyright: ignore
+    HfHubHTTPError,  # pyright: ignore
+    LocalTokenNotFoundError,  # pyright: ignore
+)
+
+from kiapi.core.setup import HfSnapshotResource, SetupResource
+
+
+class ResourceStatusManager(Protocol):
+    def status(self, resource: SetupResource) -> object: ...
+
+
+@dataclass(frozen=True)
+class HfAccessIssue:
+    repo: str
+    url: str
+    reason: str
+
+
+def preflight_hf_access(
+    resources: list[SetupResource],
+    manager: ResourceStatusManager,
+    *,
+    api: HfApi | None = None,
+) -> None:
+    """Fail early when selected gated Hugging Face snapshots are not accessible."""
+
+    hf_resources = [
+        resource for resource in resources if isinstance(resource, HfSnapshotResource)
+    ]
+    if not hf_resources:
+        return
+
+    api = api or HfApi()
+    issues: list[HfAccessIssue] = []
+    for resource in hf_resources:
+        state = manager.status(resource)
+        if bool(getattr(state, "ready", False)):
+            continue
+        if not _is_gated_repo(api, resource):
+            continue
+
+        issue = _check_gated_repo_access(api, resource)
+        if issue is not None:
+            issues.append(issue)
+
+    if issues:
+        raise click.ClickException(_format_hf_access_issues(issues))
+
+
+def _is_gated_repo(api: HfApi, resource: HfSnapshotResource) -> bool:
+    info = api.model_info(
+        resource.repo,
+        revision=resource.revision,
+        token=False,
+        timeout=15,
+    )
+    return bool(getattr(info, "gated", False))
+
+
+def _check_gated_repo_access(
+    api: HfApi, resource: HfSnapshotResource
+) -> HfAccessIssue | None:
+    url = _repo_url(resource.repo)
+    if not get_token():
+        return HfAccessIssue(
+            repo=resource.repo,
+            url=url,
+            reason="not authenticated",
+        )
+
+    try:
+        api.list_repo_files(
+            resource.repo,
+            revision=resource.revision,
+            repo_type="model",
+            token=True,
+        )
+    except LocalTokenNotFoundError:
+        return HfAccessIssue(
+            repo=resource.repo,
+            url=url,
+            reason="not authenticated",
+        )
+    except GatedRepoError:
+        return HfAccessIssue(
+            repo=resource.repo,
+            url=url,
+            reason="access has not been approved or terms have not been accepted",
+        )
+    except HfHubHTTPError as exc:
+        status_code = getattr(getattr(exc, "response", None), "status_code", None)
+        if status_code in {401, 403}:
+            return HfAccessIssue(
+                repo=resource.repo,
+                url=url,
+                reason="access has not been approved or terms have not been accepted",
+            )
+        raise
+
+    return None
+
+
+def _repo_url(repo: str) -> str:
+    return f"https://huggingface.co/{quote(repo, safe='/')}"
+
+
+def _format_hf_access_issues(issues: list[HfAccessIssue]) -> str:
+    repos = "\n".join(
+        f"- {issue.repo} ({issue.reason})\n  {issue.url}" for issue in issues
+    )
+    return (
+        "Hugging Face access is required before activation.\n\n"
+        "Open the following repository page(s), accept the terms or request access, "
+        "then authenticate locally:\n\n"
+        f"{repos}\n\n"
+        "Authenticate with one of:\n\n"
+        "  hf auth login\n\n"
+        "or:\n\n"
+        "  export HF_TOKEN=hf_...\n\n"
+        "Then run `kiapi activate ...` again."
+    )

kiapi/cli/activate/cli.py

@@ -0,0 +1,37 @@
+import click
+
+from kiapi.cli import (
+    dedupe_resources,
+    register_all_capabilities,
+    run_resources,
+    select_specs,
+)
+from kiapi.core.setup import SetupManager
+
+from ._operations.preflight_hf_access import preflight_hf_access
+
+
+@click.command()
+@click.option("--all", "activate_all", is_flag=True, help="Activate all models")
+@click.option("--domain", "domains", multiple=True, help="Activate domain")
+@click.option("--family", "families", multiple=True, help="Activate family")
+@click.option("--repo", "repos", multiple=True, help="Activate repo/image")
+def activate(
+    activate_all: bool,
+    domains: tuple[str, ...],
+    families: tuple[str, ...],
+    repos: tuple[str, ...],
+) -> None:
+    """Download Hugging Face snapshots and pull Docker images."""
+    register_all_capabilities()
+    specs = select_specs(
+        all_selected=activate_all,
+        domains=domains,
+        families=families,
+        repos=repos,
+        interactive=True,
+    )
+    resources = dedupe_resources(specs)
+    manager = SetupManager()
+    preflight_hf_access(resources, manager)
+    run_resources("activate", resources, manager)

kiapi/cli/check/_helpers/check_audio_acestep_turbo.py

@@ -0,0 +1,22 @@
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    from kiapi.capabilities.acestep import GenerateRequest, handle_generate
+
+    result, artifacts = handle_generate(
+        ctx,
+        GenerateRequest(
+            model=spec.name,
+            prompt="short simple electronic tone",
+            lyrics="[Instrumental]",
+            duration=5,
+            inference_steps=1,
+            seed=1,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_audio_acestep_xl_base.py

@@ -0,0 +1,22 @@
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    from kiapi.capabilities.acestep import GenerateRequest, handle_generate
+
+    result, artifacts = handle_generate(
+        ctx,
+        GenerateRequest(
+            model=spec.name,
+            prompt="short simple electronic tone",
+            lyrics="[Instrumental]",
+            duration=5,
+            inference_steps=1,
+            seed=1,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_audio_audiogen_medium.py

@@ -0,0 +1,24 @@
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    from kiapi.capabilities.audiogen import GenerateRequest, handle_generate
+
+    result, artifacts = handle_generate(
+        ctx,
+        GenerateRequest(
+            model=spec.name,
+            prompt="single soft beep",
+            duration=1.0,
+            seed=1,
+            top_k=1,
+            top_p=0.0,
+            temperature=1.0,
+            cfg_coef=3.0,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_chat_chat_qwen3_6_27b.py

@@ -0,0 +1,19 @@
+from kiapi.capabilities.chat import ChatRequest, handle_chat
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    result, artifacts = handle_chat(
+        ctx,
+        ChatRequest(
+            model=spec.name,
+            messages=[{"role": "user", "content": "Reply with only: ok"}],
+            max_completion_tokens=4,
+            temperature=0.0,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_chat_chat_qwen3_omni.py

@@ -0,0 +1,19 @@
+from kiapi.capabilities.chat import ChatRequest, handle_chat
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    result, artifacts = handle_chat(
+        ctx,
+        ChatRequest(
+            model=spec.name,
+            messages=[{"role": "user", "content": "Reply with only: ok"}],
+            max_completion_tokens=4,
+            temperature=0.0,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_embedding_embedding_qwen3_embedding_8b.py

@@ -0,0 +1,14 @@
+from kiapi.capabilities.embedding import EmbedRequest, handle_embed
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    result, artifacts = handle_embed(
+        ctx,
+        EmbedRequest(model=spec.name, text="kiapi check"),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_embedding_embedding_qwen3_vl_embedding_2b.py

@@ -0,0 +1,19 @@
+from kiapi.capabilities.embedding import EmbedRequest, handle_embed
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._operations.create_sample_image import create_sample_image
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    image_file_id = create_sample_image(ctx)
+    rec = ctx.file_store.get(image_file_id)
+    if rec is None:
+        raise RuntimeError("failed to create check input image")
+    result, artifacts = handle_embed(
+        ctx,
+        EmbedRequest(model=spec.name, image=rec.path),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_image_depthpro_base.py

@@ -0,0 +1,21 @@
+from kiapi.capabilities.depthpro import EstimateRequest, handle_estimate
+from kiapi.core.app import AppContext
+from kiapi.core.file import FileIDRef
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._operations.create_sample_image import create_sample_image
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    image_file_id = create_sample_image(ctx)
+    result, artifacts = handle_estimate(
+        ctx,
+        EstimateRequest(
+            model=spec.name,
+            image=FileIDRef(file_id=image_file_id),
+            include_depth_data=False,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_image_ernie_base.py

@@ -0,0 +1,23 @@
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    from kiapi.capabilities.ernie import GenerateRequest, handle_generate
+
+    result, artifacts = handle_generate(
+        ctx,
+        GenerateRequest(
+            model=spec.name,
+            prompt="a tiny blue cube on a white table",
+            width=256,
+            height=256,
+            steps=1,
+            seed=1,
+            quality=90,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_image_ernie_turbo.py

@@ -0,0 +1,23 @@
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    from kiapi.capabilities.ernie import GenerateRequest, handle_generate
+
+    result, artifacts = handle_generate(
+        ctx,
+        GenerateRequest(
+            model=spec.name,
+            prompt="a tiny blue cube on a white table",
+            width=256,
+            height=256,
+            steps=1,
+            seed=1,
+            quality=90,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_image_flux2_klein_9b.py

@@ -0,0 +1,23 @@
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    from kiapi.capabilities.flux2 import GenerateRequest, handle_generate
+
+    result, artifacts = handle_generate(
+        ctx,
+        GenerateRequest(
+            model=spec.name,
+            prompt="a tiny blue cube on a white table",
+            width=256,
+            height=256,
+            steps=2,
+            seed=1,
+            quality=90,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_image_flux2_klein_base_4b.py

@@ -0,0 +1,23 @@
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    from kiapi.capabilities.flux2 import GenerateRequest, handle_generate
+
+    result, artifacts = handle_generate(
+        ctx,
+        GenerateRequest(
+            model=spec.name,
+            prompt="a tiny blue cube on a white table",
+            width=256,
+            height=256,
+            steps=2,
+            seed=1,
+            quality=90,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_image_flux2_klein_base_9b.py

@@ -0,0 +1,23 @@
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    from kiapi.capabilities.flux2 import GenerateRequest, handle_generate
+
+    result, artifacts = handle_generate(
+        ctx,
+        GenerateRequest(
+            model=spec.name,
+            prompt="a tiny blue cube on a white table",
+            width=256,
+            height=256,
+            steps=2,
+            seed=1,
+            quality=90,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_image_ideogram4_fp8.py

@@ -0,0 +1,23 @@
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    from kiapi.capabilities.ideogram4 import GenerateRequest, handle_generate
+
+    result, artifacts = handle_generate(
+        ctx,
+        GenerateRequest(
+            model=spec.name,
+            prompt="simple word OK on a plain white card",
+            preset="V4_TURBO_12",
+            width=256,
+            height=256,
+            seed=1,
+            quality=90,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_image_qwen_edit_2509.py

@@ -0,0 +1,27 @@
+from kiapi.core.app import AppContext
+from kiapi.core.file import FileIDRef
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._operations.create_sample_image import create_sample_image
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    from kiapi.capabilities.qwen import EditRequest, handle_edit
+
+    image_file_id = create_sample_image(ctx)
+    result, artifacts = handle_edit(
+        ctx,
+        EditRequest(
+            model=spec.name,
+            prompt="make the object blue, keep the simple composition",
+            images=[FileIDRef(file_id=image_file_id)],
+            width=256,
+            height=256,
+            steps=1,
+            seed=1,
+            quality=90,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_image_qwen_image.py

@@ -0,0 +1,23 @@
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    from kiapi.capabilities.qwen import GenerateRequest, handle_generate
+
+    result, artifacts = handle_generate(
+        ctx,
+        GenerateRequest(
+            model=spec.name,
+            prompt="a tiny blue cube on a white table",
+            width=256,
+            height=256,
+            steps=1,
+            seed=1,
+            quality=90,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_image_seedvr2_3b.py

@@ -0,0 +1,23 @@
+from kiapi.capabilities.seedvr2 import UpscaleRequest, handle_upscale
+from kiapi.core.app import AppContext
+from kiapi.core.file import FileIDRef
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._operations.create_sample_image import create_sample_image
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    image_file_id = create_sample_image(ctx, size=128)
+    result, artifacts = handle_upscale(
+        ctx,
+        UpscaleRequest(
+            model=spec.name,
+            image=FileIDRef(file_id=image_file_id),
+            resolution="2x",
+            seed=1,
+            quality=90,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_image_seedvr2_7b.py

@@ -0,0 +1,23 @@
+from kiapi.capabilities.seedvr2 import UpscaleRequest, handle_upscale
+from kiapi.core.app import AppContext
+from kiapi.core.file import FileIDRef
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._operations.create_sample_image import create_sample_image
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    image_file_id = create_sample_image(ctx, size=128)
+    result, artifacts = handle_upscale(
+        ctx,
+        UpscaleRequest(
+            model=spec.name,
+            image=FileIDRef(file_id=image_file_id),
+            resolution="2x",
+            seed=1,
+            quality=90,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_image_zimage_base.py

@@ -0,0 +1,23 @@
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    from kiapi.capabilities.zimage import GenerateRequest, handle_generate
+
+    result, artifacts = handle_generate(
+        ctx,
+        GenerateRequest(
+            model=spec.name,
+            prompt="a tiny blue cube on a white table",
+            width=256,
+            height=256,
+            steps=2,
+            seed=1,
+            quality=90,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_image_zimage_turbo.py

@@ -0,0 +1,23 @@
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    from kiapi.capabilities.zimage import GenerateRequest, handle_generate
+
+    result, artifacts = handle_generate(
+        ctx,
+        GenerateRequest(
+            model=spec.name,
+            prompt="a tiny blue cube on a white table",
+            width=256,
+            height=256,
+            steps=2,
+            seed=1,
+            quality=90,
+        ),
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_video_ltx2_distilled.py

@@ -0,0 +1,26 @@
+from kiapi.capabilities.ltx2 import GenerateRequest, handle_generate
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    result, artifacts = handle_generate(
+        ctx,
+        GenerateRequest(
+            model=spec.name,
+            prompt="a small blue cube rotating slowly",
+            width=256,
+            height=256,
+            num_frames=9,
+            fps=8,
+            seed=1,
+            image_strength=1.0,
+            end_image_strength=None,
+            generate_audio=False,
+        ),
+        mode="T2V",
+    )
+    return build_check_result(spec, result, artifacts)

kiapi/cli/check/_helpers/check_web_web_fetch.py

@@ -0,0 +1,14 @@
+from kiapi.capabilities.web import FetchRequest, handle_fetch
+from kiapi.core.app import AppContext
+from kiapi.core.model import ModelSpec
+
+from .._operations.build_check_result import build_check_result
+from .._schemas.check_result import CheckResult
+
+
+def check(ctx: AppContext, spec: ModelSpec) -> CheckResult:
+    result, artifacts = handle_fetch(
+        ctx,
+        FetchRequest(url="https://example.com"),
+    )
+    return build_check_result(spec, result, artifacts)