1 files changed, 13 insertions, 12 deletions
diff --git a/pkgs/development/python-modules/litellm/default.nix b/pkgs/development/python-modules/litellm/default.nix
index 950563c88aecf..72a770848285d 100644
--- a/pkgs/development/python-modules/litellm/default.nix
+++ b/pkgs/development/python-modules/litellm/default.nix
@@ -7,6 +7,7 @@
   backoff,
   buildPythonPackage,
   click,
+  cryptography,
   fastapi,
   fastapi-sso,
   fetchFromGitHub,
@@ -14,11 +15,14 @@
   gunicorn,
   importlib-metadata,
   jinja2,
+  jsonschema,
   openai,
   orjson,
   poetry-core,
   prisma,
+  pydantic,
   pyjwt,
+  pynacl,
   python-dotenv,
   python-multipart,
   pythonOlder,
@@ -26,7 +30,6 @@
   requests,
   resend,
   rq,
-  streamlit,
   tiktoken,
   tokenizers,
   uvicorn,
@@ -34,7 +37,7 @@
 
 buildPythonPackage rec {
   pname = "litellm";
-  version = "1.40.6";
+  version = "1.41.28";
   pyproject = true;
 
   disabled = pythonOlder "3.8";
@@ -43,13 +46,9 @@ buildPythonPackage rec {
     owner = "BerriAI";
     repo = "litellm";
     rev = "refs/tags/v${version}";
-    hash = "sha256-mXrLhgOETrkwHYSJSmEPvFwZiO0kx9FmXlTJN2WZ8WI=";
+    hash = "sha256-DNFzBl2K4liphEMVPRbLWMzzCxtIcvUgQxvppAnv/10=";
   };
 
-  postPatch = ''
-    rm -rf dist
-  '';
-
   build-system = [ poetry-core ];
 
   dependencies = [
@@ -57,9 +56,11 @@ buildPythonPackage rec {
     click
     importlib-metadata
     jinja2
+    jsonschema
     openai
-    requests
+    pydantic
     python-dotenv
+    requests
     tiktoken
     tokenizers
   ];
@@ -68,6 +69,7 @@ buildPythonPackage rec {
     proxy = [
       apscheduler
       backoff
+      cryptography
       fastapi
       fastapi-sso
       gunicorn
@@ -83,15 +85,14 @@ buildPythonPackage rec {
       azure-keyvault-secrets
       google-cloud-kms
       prisma
+      pynacl
       resend
-      streamlit
     ];
   };
 
-  # the import check phase fails trying to do a network request to openai
-  # pythonImportsCheck = [ "litellm" ];
+  pythonImportsCheck = [ "litellm" ];
 
-  # no tests
+  # access network
   doCheck = false;
 
   meta = with lib; {