pyrox.dev
lol
python3Packages.litellm: fix litellm_enterprise module not found (#406995)
authored by
Pol Dellaiera
and committed by
GitHub
dac052eb
a6dfa11f
+21
-3
+21
-3
pkgs/development/python-modules/litellm/default.nix
+21
-3
pkgs/development/python-modules/litellm/default.nix
···
5
5
azure-identity,
6
6
azure-keyvault-secrets,
7
7
backoff,
8
+
boto3,
8
9
buildPythonPackage,
9
10
click,
10
11
cryptography,
···
25
26
pydantic,
26
27
pyjwt,
27
28
pynacl,
29
+
python,
28
30
python-dotenv,
29
31
python-multipart,
30
32
pythonOlder,
31
33
pyyaml,
32
34
requests,
33
35
resend,
36
+
rich,
34
37
rq,
35
38
tiktoken,
36
39
tokenizers,
37
40
uvloop,
38
41
uvicorn,
42
+
websockets,
39
43
nixosTests,
40
44
nix-update-script,
41
45
}:
···
63
67
importlib-metadata
64
68
jinja2
65
69
jsonschema
66
-
mcp
67
70
openai
68
71
pydantic
69
72
python-dotenv
···
76
79
proxy = [
77
80
apscheduler
78
81
backoff
82
+
boto3
79
83
cryptography
80
84
fastapi
81
85
fastapi-sso
82
86
gunicorn
87
+
mcp
83
88
orjson
84
89
pyjwt
90
+
pynacl
85
91
python-multipart
86
92
pyyaml
93
+
rich
87
94
rq
88
95
uvloop
89
96
uvicorn
97
+
websockets
90
98
];
99
+
91
100
extra_proxy = [
92
101
azure-identity
93
102
azure-keyvault-secrets
94
103
google-cloud-kms
95
104
prisma
96
-
pynacl
97
105
resend
98
106
];
99
107
};
100
108
101
-
pythonImportsCheck = [ "litellm" ];
109
+
pythonImportsCheck = [
110
+
"litellm"
111
+
"litellm_enterprise"
112
+
];
102
113
103
114
# Relax dependency check on openai, may not be needed in the future
104
115
pythonRelaxDeps = [ "openai" ];
105
116
106
117
# access network
107
118
doCheck = false;
119
+
120
+
postFixup = ''
121
+
# Symlink litellm_enterprise to make it discoverable
122
+
pushd $out/lib/python${python.pythonVersion}/site-packages
123
+
ln -s enterprise/litellm_enterprise litellm_enterprise
124
+
popd
125
+
'';
108
126
109
127
passthru = {
110
128
tests = { inherit (nixosTests) litellm; };