diff --git a/pkgs/development/python-modules/smolagents/default.nix b/pkgs/development/python-modules/smolagents/default.nix index 15c74d3454de..782a446b7ebe 100644 --- a/pkgs/development/python-modules/smolagents/default.nix +++ b/pkgs/development/python-modules/smolagents/default.nix @@ -47,13 +47,14 @@ # tests ipython, pytest-datadir, + pytest-timeout, pytestCheckHook, wikipedia-api, }: buildPythonPackage (finalAttrs: { pname = "smolagents"; - version = "1.21.3"; + version = "1.23.0"; pyproject = true; src = fetchFromGitHub { @@ -77,6 +78,10 @@ buildPythonPackage (finalAttrs: { optional-dependencies = lib.fix (self: { audio = [ soundfile ] ++ self.torch; bedrock = [ boto3 ]; + # blaxel = [ + # blaxel + # websocket-client + # ]; docker = [ docker websocket-client @@ -91,6 +96,10 @@ buildPythonPackage (finalAttrs: { mcp mcpadapt ]; + # modal = [ + # modal + # websocket-client + # ]; # mlx-lm = [ mlx-lm ]; openai = [ openai ]; # telemetry = [ @@ -129,6 +138,7 @@ buildPythonPackage (finalAttrs: { pytestCheckHook wikipedia-api ] + ++ lib.optionals (stdenv.hostPlatform.isDarwin) [ pytest-timeout ] ++ lib.concatAttrValues finalAttrs.passthru.optional-dependencies; pythonImportsCheck = [ "smolagents" ]; @@ -173,6 +183,16 @@ buildPythonPackage (finalAttrs: { "test_init_agent_with_different_toolsets" "test_multiagents_save" "test_new_instance" + + # Requires optional "blaxel" dependencies + "test_blaxel_executor_instantiation_with_blaxel_sdk" + "test_blaxel_executor_custom_parameters" + "test_blaxel_executor_cleanup" + # Requires optional "modal" dependencies + "test_sandbox_lifecycle" + # TypeError: 'function' object is not subscriptable + "test_stream_to_gradio_memory_step" + ]; __darwinAllowLocalNetworking = true;