new model support

This commit is contained in:
Dobromir Popov
2026-07-06 15:51:51 +03:00
parent cdc9f11128
commit ee2711a38a
3 changed files with 64 additions and 1 deletions

View File

@@ -2,6 +2,7 @@
import json
import io
import os
import sys
import types
import urllib.request
@@ -1416,14 +1417,35 @@ def test_layers_from_config_get_text_config_and_variants():
assert layers_from_config(types.SimpleNamespace()) is None
def test_download_dir_env_override(monkeypatch):
def test_download_dir_env_override(tmp_path, monkeypatch):
import importlib
from meshnet_node import config as config_mod
monkeypatch.chdir(tmp_path)
monkeypatch.setenv("MESHNET_DOWNLOAD_DIR", "/tmp/llm-store")
importlib.reload(config_mod)
assert config_mod.DEFAULTS["download_dir"] == "/tmp/llm-store"
monkeypatch.delenv("MESHNET_DOWNLOAD_DIR")
importlib.reload(config_mod)
assert config_mod.DEFAULTS["download_dir"].endswith("models")
def test_cli_loads_local_env_before_config_defaults(tmp_path, monkeypatch):
import importlib
from meshnet_node import cli as cli_mod
from meshnet_node import config as config_mod
monkeypatch.delenv("MESHNET_DOWNLOAD_DIR", raising=False)
monkeypatch.chdir(tmp_path)
(tmp_path / ".env").write_text(
"MESHNET_DOWNLOAD_DIR=/run/media/popov/DATA/llm/safetensor/models\n"
"HF_TOKEN=hf_test_token\n"
)
cli_mod._load_env_defaults()
importlib.reload(config_mod)
assert config_mod.DEFAULTS["download_dir"] == "/run/media/popov/DATA/llm/safetensor/models"
assert os.environ["HF_TOKEN"] == "hf_test_token"