diff --git a/pkgs/development/python-modules/llm-ollama/default.nix b/pkgs/development/python-modules/llm-ollama/default.nix index aebebd7543af..6c255aa76c55 100644 --- a/pkgs/development/python-modules/llm-ollama/default.nix +++ b/pkgs/development/python-modules/llm-ollama/default.nix @@ -15,18 +15,20 @@ # tests pytestCheckHook, + pytest-asyncio, + writableTmpDirAsHomeHook, }: buildPythonPackage rec { pname = "llm-ollama"; - version = "0.8.2"; + version = "0.9.1"; pyproject = true; src = fetchFromGitHub { owner = "taketwo"; repo = "llm-ollama"; tag = version; - hash = "sha256-/WAugfkI4izIQ7PoKM9epd/4vFxYPvsiwDbEqqTdMq4="; + hash = "sha256-NAJ0tfGRQOxYVAi2X0AI2d9+wyUS3ro1bfMSViZjaR0="; }; build-system = [ @@ -43,14 +45,8 @@ buildPythonPackage rec { nativeCheckInputs = [ pytestCheckHook - ]; - - # These tests try to access the filesystem and fail - disabledTests = [ - "test_registered_model" - "test_registered_chat_models" - "test_registered_embedding_models" - "test_registered_models_when_ollama_is_down" + pytest-asyncio + writableTmpDirAsHomeHook ]; pythonImportsCheck = [