diff --git a/pkgs/development/python-modules/llm-tools-sqlite/default.nix b/pkgs/development/python-modules/llm-tools-sqlite/default.nix index 494c4c5994e5..94d623124cc8 100644 --- a/pkgs/development/python-modules/llm-tools-sqlite/default.nix +++ b/pkgs/development/python-modules/llm-tools-sqlite/default.nix @@ -22,6 +22,10 @@ buildPythonPackage rec { hash = "sha256-VAmK4cXzZWTWCU92TwMdhNJPvYPZ88t5BZe8vo60SZY="; }; + patches = [ + ./fix-test.patch + ]; + build-system = [ setuptools ]; dependencies = [ llm ]; diff --git a/pkgs/development/python-modules/llm-tools-sqlite/fix-test.patch b/pkgs/development/python-modules/llm-tools-sqlite/fix-test.patch new file mode 100644 index 000000000000..802b73d8583e --- /dev/null +++ b/pkgs/development/python-modules/llm-tools-sqlite/fix-test.patch @@ -0,0 +1,13 @@ +diff --git a/tests/test_llm_tools_sqlite.py b/tests/test_llm_tools_sqlite.py +index a676007..4d653b6 100644 +--- a/tests/test_llm_tools_sqlite.py ++++ b/tests/test_llm_tools_sqlite.py +@@ -47,7 +47,7 @@ def test_tool(monkeypatch, tmpdir): + {"name": "SQLite_query", "output": '[{"n": 3}]', "tool_call_id": None}, + { + "name": "SQLite_schema", +- "output": "CREATE TABLE [pelicans] (\n [name] TEXT\n);", ++ "output": "CREATE TABLE \"pelicans\" (\n \"name\" TEXT\n);", + "tool_call_id": None, + }, + {