898d3dd0b5* Studio: offer the in-app llama.cpp update for source-build (markerless) installs Source-build installs have no UNSLOTH_PREBUILT_INFO.json marker, so freshness reported supported=False and the Update button never showed (notably on macOS, where the fork shipped no prebuilt before b9585 and setup fell back to a source build). When an install has no marker but an official prebuilt now exists for the host, surface the update and let one click swap it in place. - install_llama_prebuilt.py: published_repo_for_host() (the setup.sh host->repo rule in Python) and a --resolve-prebuilt mode that reports whether a prebuilt exists for this host without downloading. - llama_cpp_update.py: markerless branch in get_update_status/start_update, version-suppressed so source builds already newer than latest are not nagged; fail-open throughout. * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Studio: run llama update detection off the event loop, expose source_build The markerless source-build check probes the host and reads GitHub, so run get_update_status and start_update in a worker thread to keep the API responsive. Expose source_build in the status response so the banner can label the source-build switch. * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Keep the llama-route auth stub out of sys.modules for the rest of the suite test_llama_route.py replaced sys.modules['auth.authentication'] with a bare stub at collection time and never restored it, so every later test importing create_access_token got the stub: 17 failures across test_desktop_auth, test_middleware, test_openai_tool_passthrough and test_rag_preview on all four Backend CI Python versions. Import the real module when its deps are available and only stub in minimal envs, popping the stubs after the standalone route load either way. * Studio: address review on the source-build update path - published_repo_for_host: route CPU-only Windows to ggml-org too (mirrors setup.ps1; the fork ships no win-cpu bundle), macOS always the fork. - markerless detection compares/display the upstream llama_tag, not a possible fork wrapper release_tag, so a source build is not wrongly judged newer. - do not offer when there is no resolvable install root (a pinned LLAMA_SERVER_PATH outside a managed dir): an apply would not take effect. * Ignore version probes in the update tests' subprocess capture The status polls in these tests trigger the new source-build detection, which shells out to llama-server --version through the same patched subprocess.run. On slow runners that probe lands after the installer call and clobbers the single captured argv, failing the flag assertions (seen on the 3.10/3.11 Backend CI jobs). Skip probe calls in all three fakes so only the installer invocation is captured. * Skip markerless re-detection while the update job is swapping the tree On a source-build install the frontend polls update-status every 3s during an apply, and each poll ran _source_build_status, which execs the very llama-server binary the job is concurrently replacing. On Windows that exec can hold the exe long enough to fail the installer's os.replace; everywhere it is a per-poll subprocess spawn for a status the poller does not read (it only consumes job progress). Gate the markerless branch on the job not running; the marked path is probe-free and still returns the live job state. * Studio: tighten source-build update root, repo routing, and downgrade guard Only manage a markerless install when the active binary lives under a resolvable llama.cpp root (marker dir, UNSLOTH_LLAMA_CPP_PATH it sits in, or a llama.cpp ancestor); a pinned LLAMA_SERVER_PATH or a PATH/system binary is left alone so an apply cannot install where it would not take effect. Gate start_update on the same suppression as detection so a direct POST cannot downgrade a source build newer than the latest prebuilt. Route Linux hosts with AMD tooling (rocminfo/amd-smi/hipconfig/ hipinfo) to the fork in --resolve-prebuilt, matching setup.sh, so a HIP source build is not offered an upstream CPU prebuilt. * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Studio: cover inactive env root and pinned llama.cpp checkout in update root tests --------- Co-authored-by: danielhanchen <michaelhan2050@gmail.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
82 lines
2.9 KiB
Python
82 lines
2.9 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-only
|
|
# Copyright 2026-present the Unsloth AI Inc. team. All rights reserved. See /studio/LICENSE.AGPL-3.0
|
|
|
|
"""llama.cpp prebuilt update endpoints.
|
|
|
|
GET /api/llama/update-status -> is a newer prebuilt available + job state
|
|
POST /api/llama/update -> download + atomically swap to the latest
|
|
|
|
Detection reuses utils.llama_cpp_freshness; the swap reuses
|
|
install_llama_prebuilt.py via utils.llama_cpp_update. Both fail open so the UI
|
|
never blocks on a missing marker / offline GitHub.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from typing import Optional
|
|
|
|
from fastapi import APIRouter, Depends, Query
|
|
from pydantic import BaseModel, Field
|
|
|
|
from auth.authentication import get_current_subject
|
|
from utils.llama_cpp_update import get_update_status, start_update
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
class LlamaUpdateJob(BaseModel):
|
|
state: str = Field("idle", description = "idle | running | success | error")
|
|
message: str = ""
|
|
from_tag: Optional[str] = None
|
|
to_tag: Optional[str] = None
|
|
error: Optional[str] = None
|
|
started_at: Optional[str] = None
|
|
finished_at: Optional[str] = None
|
|
|
|
|
|
class LlamaUpdateStatusResponse(BaseModel):
|
|
supported: bool = Field(
|
|
False,
|
|
description = "True when the install came from an Unsloth prebuilt (has a marker).",
|
|
)
|
|
update_available: bool = Field(False, description = "True when installed_tag != latest_tag.")
|
|
stale: bool = Field(
|
|
False, description = "Update available AND install older than the staleness threshold."
|
|
)
|
|
installed_tag: Optional[str] = None
|
|
latest_tag: Optional[str] = None
|
|
published_repo: Optional[str] = None
|
|
installed_at_utc: Optional[str] = None
|
|
age_days: Optional[int] = None
|
|
source_build: bool = Field(
|
|
False, description = "True when there is no marker (source build) but a prebuilt is offered."
|
|
)
|
|
job: LlamaUpdateJob = Field(default_factory = LlamaUpdateJob)
|
|
|
|
|
|
class LlamaUpdateActionResponse(BaseModel):
|
|
started: bool
|
|
reason: Optional[str] = None
|
|
message: Optional[str] = None
|
|
job: LlamaUpdateJob = Field(default_factory = LlamaUpdateJob)
|
|
|
|
|
|
@router.get("/update-status", response_model = LlamaUpdateStatusResponse)
|
|
async def llama_update_status(
|
|
force_refresh: bool = Query(
|
|
False, description = "Bypass the 24h release cache for an explicit check."
|
|
),
|
|
current_subject: str = Depends(get_current_subject),
|
|
) -> LlamaUpdateStatusResponse:
|
|
# Off the event loop: detection may probe the host and read GitHub.
|
|
status = await asyncio.to_thread(get_update_status, force_refresh = force_refresh)
|
|
return LlamaUpdateStatusResponse(**status)
|
|
|
|
|
|
@router.post("/update", response_model = LlamaUpdateActionResponse)
|
|
async def llama_update(
|
|
current_subject: str = Depends(get_current_subject),
|
|
) -> LlamaUpdateActionResponse:
|
|
action = await asyncio.to_thread(start_update)
|
|
return LlamaUpdateActionResponse(**action)
|