diff --git a/pkgs/development/python-modules/llm-echo/default.nix b/pkgs/development/python-modules/llm-echo/default.nix new file mode 100644 index 000000000000..3347f5280145 --- /dev/null +++ b/pkgs/development/python-modules/llm-echo/default.nix @@ -0,0 +1,45 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + setuptools, + llm, + llm-echo, + pytestCheckHook, + writableTmpDirAsHomeHook, +}: + +buildPythonPackage rec { + pname = "llm-echo"; + version = "0.3"; + pyproject = true; + + src = fetchFromGitHub { + owner = "simonw"; + repo = "llm-echo"; + tag = version; + hash = "sha256-KtdsvXJQRJ3EEJ25i6ccbVxMsWlDsxa+HJ7bFzyYGNI="; + }; + + build-system = [ + setuptools + llm + ]; + + nativeCheckInputs = [ + pytestCheckHook + writableTmpDirAsHomeHook + ]; + + pythonImportsCheck = [ "llm_echo" ]; + + passthru.tests = llm.mkPluginTest llm-echo; + + meta = { + description = "Debug plugin for LLM"; + homepage = "https://github.com/simonw/llm-echo"; + changelog = "https://github.com/simonw/llm-echo/releases/tag/${version}/CHANGELOG.md"; + license = lib.licenses.asl20; + maintainers = with lib.maintainers; [ philiptaron ]; + }; +} diff --git a/pkgs/development/python-modules/llm/default.nix b/pkgs/development/python-modules/llm/default.nix index c1f71fa10e16..80bed4ff6c2e 100644 --- a/pkgs/development/python-modules/llm/default.nix +++ b/pkgs/development/python-modules/llm/default.nix @@ -63,6 +63,7 @@ let llm-cmd ? false, llm-command-r ? false, llm-deepseek ? false, + llm-echo ? false, llm-fragments-github ? false, llm-fragments-pypi ? false, llm-gemini ? false, diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 8bded0dc781f..ea594d439326 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -8302,6 +8302,8 @@ self: super: with self; { llm-deepseek = callPackage ../development/python-modules/llm-deepseek { }; + llm-echo = callPackage ../development/python-modules/llm-echo { }; + llm-fragments-github = callPackage ../development/python-modules/llm-fragments-github { }; llm-fragments-pypi = callPackage ../development/python-modules/llm-fragments-pypi { };