amorfo77/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2026-09-21 08:58:31 +00:00
Code Issues Projects Releases Wiki Activity

Sure, cover ollama too

Browse Source
This commit is contained in:
shamoon
2026-09-20 21:09:20 -07:00
parent 285e3fec39
commit ac59404389
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/paperless_ai/tests/test_client.py
+13 -1
View File
@@ -199,7 +199,7 @@ def test_run_llm_query_openai_uses_tools(mock_ai_config, mock_openai_llm):
)
def test_get_llm_passes_extra_params(mock_ai_config, mock_openai_llm):
def test_get_llm_passes_extra_params(mock_ai_config, mock_openai_llm, mock_ollama_llm):
"""
GIVEN:
- Extra LLM params configured, e.g. for a provider that needs a
@@ -220,6 +220,18 @@ def test_get_llm_passes_extra_params(mock_ai_config, mock_openai_llm):
"reasoning_effort": "none",
}
# ollama
mock_ai_config.llm_backend = "ollama"
mock_ai_config.llm_model = "test_model"
mock_ai_config.llm_endpoint = "http://test-url"
mock_ai_config.llm_extra_params = {"reasoning_effort": "none"}
AIClient()
assert mock_ollama_llm.call_args.kwargs["additional_kwargs"] == {
"reasoning_effort": "none",
}
def test_run_llm_query_openai_timeout_raises_local_error(
mock_ai_config,