From 11094f504e8393bc3e4f6292c9e6f56b87cf675c Mon Sep 17 00:00:00 2001 From: Dev Khant Date: Tue, 18 Jun 2024 21:40:43 +0530 Subject: [PATCH] Fix Ollama test (#1428) --- tests/llm/test_ollama.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/llm/test_ollama.py b/tests/llm/test_ollama.py index 36a62b63..e45cb6bb 100644 --- a/tests/llm/test_ollama.py +++ b/tests/llm/test_ollama.py @@ -37,6 +37,7 @@ def test_get_answer_mocked_ollama(ollama_llm_config, mocker): def test_get_llm_model_answer_with_streaming(ollama_llm_config, mocker): ollama_llm_config.stream = True ollama_llm_config.callbacks = [StreamingStdOutCallbackHandler()] + mocker.patch("embedchain.llm.ollama.Client.list", return_value={"models": [{"name": "llama2"}]}) mocked_ollama_chat = mocker.patch("embedchain.llm.ollama.OllamaLlm._get_answer", return_value="Test answer") llm = OllamaLlm(ollama_llm_config)