diff --git a/src/paperless_ai/tests/test_client.py b/src/paperless_ai/tests/test_client.py index 94ed13d93..96cc1b1f7 100644 --- a/src/paperless_ai/tests/test_client.py +++ b/src/paperless_ai/tests/test_client.py @@ -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,