blob: 318f673533089304402efda3c1db4740e3704532 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
{
buildPythonApplication,
buildPythonPackage,
fetchFromGitHub,
lib,
makeWrapper,
pytestCheckHook,
python3,
pythonOlder,
ruff,
setuptools,
}: let
llm = buildPythonPackage rec {
pname = "llm";
version = "0.12";
pyproject = true;
disabled = pythonOlder "3.8";
src = fetchFromGitHub {
owner = "simonw";
repo = pname;
rev = "refs/tags/${version}";
hash = "sha256-aCqdw2co/cXrBwVY/k/aSLl3C22nlH5LvU2yir1/NnQ=";
};
patches = [
./001-disable-install-uninstall-commands.patch
];
nativeBuildInputs = [
setuptools
];
propagatedBuildInputs = with python3.pkgs; [
click-default-group
numpy
openai
pluggy
pydantic
python-ulid
pyyaml
setuptools # for pkg_resources
sqlite-migrate
sqlite-utils
];
nativeCheckInputs = with python3.pkgs; [
cogapp
numpy
pytestCheckHook
requests-mock
];
doCheck = true;
pytestFlagsArray = [
"-svv"
"tests/"
];
pythonImportsCheck = [
"llm"
];
passthru = {inherit withPlugins;};
meta = with lib; {
homepage = "https://github.com/simonw/llm";
description = "Access large language models from the command-line";
changelog = "https://github.com/simonw/llm/releases/tag/${version}";
license = licenses.asl20;
mainProgram = "llm";
maintainers = with maintainers; [aldoborrero];
};
};
withPlugins = plugins: buildPythonApplication {
inherit (llm) pname version;
format = "other";
disabled = pythonOlder "3.8";
dontUnpack = true;
dontBuild = true;
doCheck = false;
nativeBuildInputs = [
makeWrapper
];
installPhase = ''
makeWrapper ${llm}/bin/llm $out/bin/llm \
--prefix PYTHONPATH : "${llm}/${python3.sitePackages}:$PYTHONPATH"
ln -sfv ${llm}/lib $out/lib
'';
propagatedBuildInputs = llm.propagatedBuildInputs ++ plugins;
passthru = llm.passthru // {
withPlugins = morePlugins: withPlugins (morePlugins ++ plugins);
};
inherit (llm) meta;
};
in
llm
|