Skip to content

Commit 32a6b4a

Browse files
committed
debug: Specifying LOG_LEVEL explicitly for model download tests.
1 parent 80e3ba4 commit 32a6b4a

File tree

3 files changed

+4
-1
lines changed

3 files changed

+4
-1
lines changed

.github/workflows/uv-pytest-self-hosted.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,4 +38,4 @@ jobs:
3838
run: uv sync --all-groups
3939

4040
- name: Run tests using pytest
41-
run: LOG_LEVEL=INFO uv run --group test pytest tests/
41+
run: uv run --group test pytest tests/

src/ollama_downloader/common.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ def get_httpx_client(verify: bool, timeout: float) -> httpx.Client:
5555
trust_env=True,
5656
http2=True,
5757
timeout=timeout,
58+
# Set a custom User-Agent header
5859
headers={"User-Agent": user_agent},
5960
)
6061
return client

tests/test_typer.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ def test_model_download():
4747
[sys.executable, "-m", "ollama_downloader.cli", "model-download", model_tag],
4848
capture_output=True,
4949
text=True,
50+
env={"LOG_LEVEL": "INFO"},
5051
)
5152
assert result.returncode == 0
5253
assert f"{model_tag} successfully downloaded and saved" in result.stdout
@@ -67,6 +68,7 @@ def test_hf_model_download():
6768
],
6869
capture_output=True,
6970
text=True,
71+
env={"LOG_LEVEL": "INFO"},
7072
)
7173
assert result.returncode == 0
7274
assert f"{org_repo_model} successfully downloaded and saved" in result.stdout

0 commit comments

Comments
 (0)