load("@rules_shell//shell:sh_binary.bzl", "sh_binary")
load("@rules_shell//shell:sh_test.bzl", "sh_test")
load(":runtime.bzl", "cuda_runtime")

cuda_runtime(
    name = "llama_cpp_cuda_runtime",
    cuda_runtime = "@llama_cpp_win_cuda_12_4_cudart_x64//:runtime",
    llama_runtime = "@llama_cpp_win_cuda_12_4_x64//:runtime",
)

sh_binary(
    name = "model",
    srcs = ["qwen3_vl.sh"],
    args = [
        "$(rootpath :llama_cpp_cuda_runtime)",
        "config",
    ],
    data = [":llama_cpp_cuda_runtime"],
)

sh_binary(
    name = "download_model",
    srcs = ["qwen3_vl.sh"],
    args = [
        "$(rootpath :llama_cpp_cuda_runtime)",
        "download",
    ],
    data = [":llama_cpp_cuda_runtime"],
)

sh_binary(
    name = "preflight",
    srcs = ["qwen3_vl.sh"],
    args = [
        "$(rootpath :llama_cpp_cuda_runtime)",
        "preflight",
    ],
    data = [":llama_cpp_cuda_runtime"],
)

sh_binary(
    name = "chat",
    srcs = ["qwen3_vl.sh"],
    args = [
        "$(rootpath :llama_cpp_cuda_runtime)",
        "chat",
    ],
    data = [":llama_cpp_cuda_runtime"],
)

sh_binary(
    name = "serve",
    srcs = ["qwen3_vl.sh"],
    args = [
        "$(rootpath :llama_cpp_cuda_runtime)",
        "serve",
    ],
    data = [":llama_cpp_cuda_runtime"],
)

sh_binary(
    name = "serve_canvas",
    srcs = ["qwen3_vl.sh"],
    args = [
        "$(rootpath :llama_cpp_cuda_runtime)",
        "serve_canvas",
    ],
    data = [":llama_cpp_cuda_runtime"],
    visibility = ["//infinite_canvas:__pkg__"],
)

sh_test(
    name = "qwen3_vl_test",
    srcs = ["qwen3_vl_test.sh"],
    data = ["qwen3_vl.sh"],
    deps = ["@bazel_tools//tools/bash/runfiles"],
)