Add Unsloth desktop deep links (#7560)
* Add Unsloth desktop deep links * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Address deep-link review feedback --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Wasim Yousef Said
GitHub
parent
71f7e1087b
commit
65b4d9d9e7
21 changed files with 707 additions and 22 deletions
188
tests/studio/test_tauri_deep_link_contract.py
Normal file
188
tests/studio/test_tauri_deep_link_contract.py
Normal file
|
|
@ -0,0 +1,188 @@
|
|||
# SPDX-License-Identifier: AGPL-3.0-only
|
||||
# Copyright 2026-present the Unsloth AI Inc. team. All rights reserved. See /studio/LICENSE.AGPL-3.0
|
||||
|
||||
"""Desktop deep-link configuration, routing, and validation contracts."""
|
||||
|
||||
import json
|
||||
from pathlib import Path
|
||||
import shutil
|
||||
import subprocess
|
||||
import textwrap
|
||||
|
||||
import pytest
|
||||
|
||||
try:
|
||||
import tomllib
|
||||
except ModuleNotFoundError:
|
||||
tomllib = pytest.importorskip("tomli")
|
||||
|
||||
|
||||
REPO = Path(__file__).resolve().parents[2]
|
||||
FRONTEND = REPO / "studio/frontend"
|
||||
TAURI = REPO / "studio/src-tauri"
|
||||
PARSER = FRONTEND / "src/features/deep-links/parse-deep-link.ts"
|
||||
INTENT_GATE = FRONTEND / "src/features/deep-links/deep-link-intent.ts"
|
||||
GGUF_FILENAME = FRONTEND / "src/features/hub/lib/gguf-filename.ts"
|
||||
|
||||
|
||||
def test_unsloth_deep_link_parser_guardrails(tmp_path: Path) -> None:
|
||||
if shutil.which("node") is None:
|
||||
pytest.skip("node not available")
|
||||
probe = subprocess.run(
|
||||
["node", "--experimental-strip-types", "--version"],
|
||||
capture_output = True,
|
||||
text = True,
|
||||
timeout = 5,
|
||||
)
|
||||
if probe.returncode != 0:
|
||||
pytest.skip("node --experimental-strip-types not available")
|
||||
|
||||
(tmp_path / "parse-deep-link.ts").write_text(
|
||||
PARSER.read_text(encoding = "utf-8"), encoding = "utf-8"
|
||||
)
|
||||
|
||||
(tmp_path / "gguf-filename.ts").write_text(
|
||||
GGUF_FILENAME.read_text(encoding = "utf-8"), encoding = "utf-8"
|
||||
)
|
||||
|
||||
(tmp_path / "deep-link-intent.ts").write_text(
|
||||
INTENT_GATE.read_text(encoding = "utf-8"), encoding = "utf-8"
|
||||
)
|
||||
script = textwrap.dedent("""
|
||||
import assert from "node:assert/strict";
|
||||
import { parseUnslothDeepLink } from "./parse-deep-link.ts";
|
||||
|
||||
import { createDeepLinkIntentGate } from "./deep-link-intent.ts";
|
||||
import {
|
||||
ggufFilenamesMatch,
|
||||
ggufSelectionOverrideMatchesIntent,
|
||||
} from "./gguf-filename.ts";
|
||||
|
||||
const valid = new Map([
|
||||
[
|
||||
"unsloth://open_from_hf?model=unsloth/Laguna-S-2.1-GGUF",
|
||||
{ model: "unsloth/Laguna-S-2.1-GGUF" },
|
||||
],
|
||||
[
|
||||
"unsloth://open_from_hf/?model=org/repo_name",
|
||||
{ model: "org/repo_name" },
|
||||
],
|
||||
[
|
||||
"unsloth://open_from_hf?model=org%2Frepo",
|
||||
{ model: "org/repo" },
|
||||
],
|
||||
[
|
||||
"unsloth://open_from_hf?model=unsloth/Laguna-S-2.1-GGUF&file=Laguna-S-2.1-UD-IQ3_XXS.gguf",
|
||||
{
|
||||
model: "unsloth/Laguna-S-2.1-GGUF",
|
||||
file: "Laguna-S-2.1-UD-IQ3_XXS.gguf",
|
||||
},
|
||||
],
|
||||
[
|
||||
"unsloth://open_from_hf?file=weights%2Fmodel-Q4_K_M.gguf&model=org/repo",
|
||||
{ model: "org/repo", file: "weights/model-Q4_K_M.gguf" },
|
||||
],
|
||||
[
|
||||
`unsloth://open_from_hf?model=${"a".repeat(96)}/${"b".repeat(96)}`,
|
||||
{ model: `${"a".repeat(96)}/${"b".repeat(96)}` },
|
||||
],
|
||||
]);
|
||||
for (const [url, intent] of valid) {
|
||||
assert.deepEqual(parseUnslothDeepLink(url), intent, url);
|
||||
}
|
||||
|
||||
assert.equal(
|
||||
ggufFilenamesMatch(
|
||||
"weights/model-Q4_K_M-00002-of-00002.gguf",
|
||||
"weights/model-Q4_K_M-00001-of-00002.gguf",
|
||||
),
|
||||
true,
|
||||
);
|
||||
assert.equal(
|
||||
ggufFilenamesMatch("model-Q4_K_M.GGUF", "model-q4_k_m.gguf"),
|
||||
true,
|
||||
);
|
||||
assert.equal(ggufFilenamesMatch("mmproj-F16.gguf", "model-F16.gguf"), false);
|
||||
|
||||
assert.equal(ggufSelectionOverrideMatchesIntent("a.gguf", 2, "a.gguf", 2), true);
|
||||
assert.equal(ggufSelectionOverrideMatchesIntent("a.gguf", 2, "a.gguf", 1), false);
|
||||
|
||||
let now = 1_000;
|
||||
const acceptIntent = createDeepLinkIntentGate(2_000, () => now);
|
||||
assert.equal(acceptIntent("org/repo", "a.gguf"), 1);
|
||||
assert.equal(acceptIntent("org/repo", "a.gguf"), null);
|
||||
assert.equal(acceptIntent("org/repo", "b.gguf"), 2);
|
||||
now = 3_000;
|
||||
assert.equal(acceptIntent("org/repo", "b.gguf"), 3);
|
||||
|
||||
|
||||
const invalid = [
|
||||
"",
|
||||
"https://open_from_hf?model=org/repo",
|
||||
"UNSLOTH://open_from_hf?model=org/repo",
|
||||
"unsloth://OPEN_FROM_HF?model=org/repo",
|
||||
"unsloth://open_from_hf/path?model=org/repo",
|
||||
"unsloth://open_from_hf/%2e%2e?model=org/repo",
|
||||
"unsloth://user@open_from_hf?model=org/repo",
|
||||
"unsloth://open_from_hf:42?model=org/repo",
|
||||
"unsloth://open_from_hf?model=org/repo#fragment",
|
||||
"unsloth://open_from_hf?model=org/repo&download=true",
|
||||
|
||||
"unsloth://open_from_hf?model=org/repo&file=model.gguf&file=other.gguf",
|
||||
"unsloth://open_from_hf?model=org/repo&file=",
|
||||
"unsloth://open_from_hf?model=org/repo&file=../model.gguf",
|
||||
"unsloth://open_from_hf?model=org/repo&file=%2Fmodel.gguf",
|
||||
"unsloth://open_from_hf?model=org/repo&file=model.safetensors",
|
||||
"unsloth://open_from_hf?model=org/repo&model=other/repo",
|
||||
"unsloth://open_from_hf?model=repo",
|
||||
"unsloth://open_from_hf?model=org/repo/extra",
|
||||
"unsloth://open_from_hf?model=-org/repo",
|
||||
"unsloth://open_from_hf?model=org/repo.",
|
||||
|
||||
"unsloth://open_from_hf?model=org/repo.git",
|
||||
"unsloth://open_from_hf?model=org/repo--name",
|
||||
"unsloth://open_from_hf?model=org/repo..name",
|
||||
];
|
||||
for (const url of invalid) {
|
||||
assert.equal(parseUnslothDeepLink(url), null, url);
|
||||
}
|
||||
""")
|
||||
result = subprocess.run(
|
||||
["node", "--experimental-strip-types", "--no-warnings", "--input-type=module"],
|
||||
input = script,
|
||||
cwd = tmp_path,
|
||||
capture_output = True,
|
||||
text = True,
|
||||
timeout = 30,
|
||||
)
|
||||
assert result.returncode == 0, f"stderr: {result.stderr}\nstdout: {result.stdout}"
|
||||
|
||||
|
||||
def test_tauri_registers_only_the_unsloth_scheme() -> None:
|
||||
cargo = tomllib.loads((TAURI / "Cargo.toml").read_text(encoding = "utf-8"))
|
||||
dependencies = cargo["dependencies"]
|
||||
assert "tauri-plugin-deep-link" in dependencies
|
||||
single_instance = dependencies["tauri-plugin-single-instance"]
|
||||
assert isinstance(single_instance, dict)
|
||||
assert "deep-link" in single_instance.get("features", [])
|
||||
|
||||
config = json.loads((TAURI / "tauri.conf.json").read_text(encoding = "utf-8"))
|
||||
assert config["plugins"]["deep-link"]["desktop"]["schemes"] == ["unsloth"]
|
||||
|
||||
capabilities = json.loads((TAURI / "capabilities/default.json").read_text(encoding = "utf-8"))
|
||||
assert "deep-link:default" in capabilities["permissions"]
|
||||
assert "core:window:allow-unminimize" in capabilities["permissions"]
|
||||
|
||||
main = (TAURI / "src/main.rs").read_text(encoding = "utf-8")
|
||||
assert main.index("tauri_plugin_single_instance::init") < main.index(
|
||||
"tauri_plugin_deep_link::init()"
|
||||
)
|
||||
assert "DeepLinkExt" in main
|
||||
assert "if let Err(error) = app.deep_link().register_all()" in main
|
||||
assert 'warn!("Failed to register deep-link handlers: {error}")' in main
|
||||
assert 'target_os = "linux"' in main
|
||||
desktop_template = TAURI / "linux/unsloth.desktop"
|
||||
assert config["bundle"]["linux"]["deb"]["desktopTemplate"] == "./linux/unsloth.desktop"
|
||||
desktop = desktop_template.read_text(encoding = "utf-8")
|
||||
assert "Exec={{exec}} %u" in desktop
|
||||
assert "MimeType=x-scheme-handler/unsloth;" in desktop
|
||||
Loading…
Add table
Add a link
Reference in a new issue