diff --git a/pkgs/development/python-modules/llm-fragments-reader/default.nix b/pkgs/development/python-modules/llm-fragments-reader/default.nix new file mode 100644 index 000000000000..72aa5aaea539 --- /dev/null +++ b/pkgs/development/python-modules/llm-fragments-reader/default.nix @@ -0,0 +1,50 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + setuptools, + httpx, + httpx-sse, + llm, + llm-fragments-reader, + pytestCheckHook, + pytest-asyncio, + pytest-httpx, + writableTmpDirAsHomeHook, +}: + +buildPythonPackage rec { + pname = "llm-fragments-reader"; + version = "0.1"; + pyproject = true; + + src = fetchFromGitHub { + owner = "simonw"; + repo = "llm-fragments-reader"; + tag = version; + hash = "sha256-2xdvOpMGsTtnerrlGiVSHoJrM+GQ7Zgv+zn2SAwYAL4="; + }; + + build-system = [ + setuptools + llm + ]; + + nativeCheckInputs = [ + pytestCheckHook + pytest-httpx + writableTmpDirAsHomeHook + ]; + + pythonImportsCheck = [ "llm_fragments_reader" ]; + + passthru.tests = llm.mkPluginTest llm-fragments-reader; + + meta = { + description = "Run URLs through the Jina Reader API"; + homepage = "https://github.com/simonw/llm-fragments-reader"; + changelog = "https://github.com/simonw/llm-fragments-reader/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 4cfc7e21ef7a..3089cdb53746 100644 --- a/pkgs/development/python-modules/llm/default.nix +++ b/pkgs/development/python-modules/llm/default.nix @@ -69,6 +69,7 @@ let llm-echo ? false, llm-fragments-github ? false, llm-fragments-pypi ? false, + llm-fragments-reader ? false, llm-gemini ? false, llm-gguf ? false, llm-grok ? false, diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 91a2b06a722e..eb077daabfdc 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -8308,6 +8308,8 @@ self: super: with self; { llm-fragments-pypi = callPackage ../development/python-modules/llm-fragments-pypi { }; + llm-fragments-reader = callPackage ../development/python-modules/llm-fragments-reader { }; + llm-gemini = callPackage ../development/python-modules/llm-gemini { }; llm-gguf = callPackage ../development/python-modules/llm-gguf { };