new model support
This commit is contained in:
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user