Leonid PershinandCursor
ec8bd28ce0
- Introduced `assistent-personas.yaml` to the project and updated the `.gitignore` accordingly. - Implemented functions to seed and merge assistent personas from local files to the VM. - Enhanced the model capture process to include merging of wanted models from the VM into `models.yaml`. - Updated documentation to reflect changes in the assistent personas and model management processes. Co-authored-by: Cursor <cursoragent@cursor.com>
119 lines
3.7 KiB
Python
119 lines
3.7 KiB
Python
"""Interactive setup wizard: files + LLM_RUNTIME + optional watchdog."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import shutil
|
|
from collections.abc import Callable
|
|
from pathlib import Path
|
|
|
|
from gpu_rent.llm_runtime import (
|
|
append_vars_llm_runtime,
|
|
ensure_ollama_manifest_from_example,
|
|
llm_runtime_menu,
|
|
normalize_runtime,
|
|
ollama_preset_menu,
|
|
write_ollama_models_preset,
|
|
)
|
|
from gpu_rent.paths import (
|
|
app_root,
|
|
assistent_personas_example_path,
|
|
assistent_personas_manifest_path,
|
|
env_path,
|
|
extensions_manifest_path,
|
|
models_manifest_path,
|
|
ollama_models_example_path,
|
|
ollama_models_manifest_path,
|
|
vars_example_path,
|
|
vars_path,
|
|
)
|
|
from gpu_rent.prompts import prompt_menu
|
|
|
|
Log = Callable[[str], None]
|
|
|
|
|
|
def _copy_if_missing(src: Path, dst: Path, label: str, log: Log) -> None:
|
|
if dst.is_file():
|
|
log(f"есть {label}")
|
|
return
|
|
if src.is_file():
|
|
shutil.copy2(src, dst)
|
|
log(f"создал {label} из example")
|
|
else:
|
|
log(f"нет example для {label}: {src}")
|
|
|
|
|
|
def run_setup(
|
|
*,
|
|
llm: str | None = None,
|
|
ollama_preset: str | None = None,
|
|
install_watchdog: bool | None = None,
|
|
confirm: Callable[[str], bool] | None = None,
|
|
ask: Callable[[str, str], str] | None = None,
|
|
log: Log = print,
|
|
) -> None:
|
|
root = app_root()
|
|
log(f"setup в {root}")
|
|
|
|
_copy_if_missing(root / "env.example", env_path(), ".env", log)
|
|
_copy_if_missing(root / "models.example.yaml", models_manifest_path(), "models.yaml", log)
|
|
_copy_if_missing(
|
|
assistent_personas_example_path(),
|
|
assistent_personas_manifest_path(),
|
|
"assistent-personas.yaml",
|
|
log,
|
|
)
|
|
_copy_if_missing(
|
|
root / "extensions.example.yaml", extensions_manifest_path(), "extensions.yaml", log
|
|
)
|
|
_copy_if_missing(vars_example_path(), vars_path(), "gpu-rent.vars", log)
|
|
_copy_if_missing(
|
|
ollama_models_example_path(), ollama_models_manifest_path(), "ollama-models.yaml", log
|
|
)
|
|
|
|
runtime = llm
|
|
if runtime is None:
|
|
if ask:
|
|
runtime = prompt_menu(
|
|
"LLM runtime",
|
|
llm_runtime_menu(),
|
|
default="none",
|
|
ask=ask,
|
|
show=log,
|
|
)
|
|
else:
|
|
runtime = "none"
|
|
runtime = normalize_runtime(runtime)
|
|
append_vars_llm_runtime(vars_path(), runtime)
|
|
log(f"LLM_RUNTIME={runtime} → gpu-rent.vars")
|
|
|
|
if runtime == "ollama":
|
|
preset = ollama_preset
|
|
if preset is None and ask:
|
|
preset = prompt_menu(
|
|
"Ollama preset",
|
|
ollama_preset_menu(include_keep=False),
|
|
default="recommended",
|
|
ask=ask,
|
|
show=log,
|
|
)
|
|
if preset is None:
|
|
preset = "recommended"
|
|
if preset.strip().lower() in {"keep", "example", ""}:
|
|
ensure_ollama_manifest_from_example()
|
|
log("ollama-models.yaml из example")
|
|
else:
|
|
write_ollama_models_preset(ollama_models_manifest_path(), preset)
|
|
log(f"ollama-models.yaml пресет={preset}")
|
|
|
|
do_wd = install_watchdog
|
|
if do_wd is None and confirm:
|
|
do_wd = confirm("Установить local-watchdog (аварийный stop, если процесс умер)?")
|
|
if do_wd:
|
|
from gpu_rent.local_watchdog import install_watchdog as _install
|
|
|
|
_install(log=log)
|
|
elif do_wd is False:
|
|
log("local-watchdog: skip")
|
|
|
|
log("готово. Заполни .env (OS_*), потом: gpu-rent doctor && gpu-rent up")
|