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
··· 5 azure-identity, 6 azure-keyvault-secrets, 7 backoff, 8 buildPythonPackage, 9 click, 10 cryptography, ··· 25 pydantic, 26 pyjwt, 27 pynacl, 28 python-dotenv, 29 python-multipart, 30 pythonOlder, 31 pyyaml, 32 requests, 33 resend, 34 rq, 35 tiktoken, 36 tokenizers, 37 uvloop, 38 uvicorn, 39 nixosTests, 40 nix-update-script, 41 }: ··· 63 importlib-metadata 64 jinja2 65 jsonschema 66 - mcp 67 openai 68 pydantic 69 python-dotenv ··· 76 proxy = [ 77 apscheduler 78 backoff 79 cryptography 80 fastapi 81 fastapi-sso 82 gunicorn 83 orjson 84 pyjwt 85 python-multipart 86 pyyaml 87 rq 88 uvloop 89 uvicorn 90 ]; 91 extra_proxy = [ 92 azure-identity 93 azure-keyvault-secrets 94 google-cloud-kms 95 prisma 96 - pynacl 97 resend 98 ]; 99 }; 100 101 - pythonImportsCheck = [ "litellm" ]; 102 103 # Relax dependency check on openai, may not be needed in the future 104 pythonRelaxDeps = [ "openai" ]; 105 106 # access network 107 doCheck = false; 108 109 passthru = { 110 tests = { inherit (nixosTests) litellm; };
··· 5 azure-identity, 6 azure-keyvault-secrets, 7 backoff, 8 + boto3, 9 buildPythonPackage, 10 click, 11 cryptography, ··· 26 pydantic, 27 pyjwt, 28 pynacl, 29 + python, 30 python-dotenv, 31 python-multipart, 32 pythonOlder, 33 pyyaml, 34 requests, 35 resend, 36 + rich, 37 rq, 38 tiktoken, 39 tokenizers, 40 uvloop, 41 uvicorn, 42 + websockets, 43 nixosTests, 44 nix-update-script, 45 }: ··· 67 importlib-metadata 68 jinja2 69 jsonschema 70 openai 71 pydantic 72 python-dotenv ··· 79 proxy = [ 80 apscheduler 81 backoff 82 + boto3 83 cryptography 84 fastapi 85 fastapi-sso 86 gunicorn 87 + mcp 88 orjson 89 pyjwt 90 + pynacl 91 python-multipart 92 pyyaml 93 + rich 94 rq 95 uvloop 96 uvicorn 97 + websockets 98 ]; 99 + 100 extra_proxy = [ 101 azure-identity 102 azure-keyvault-secrets 103 google-cloud-kms 104 prisma 105 resend 106 ]; 107 }; 108 109 + pythonImportsCheck = [ 110 + "litellm" 111 + "litellm_enterprise" 112 + ]; 113 114 # Relax dependency check on openai, may not be needed in the future 115 pythonRelaxDeps = [ "openai" ]; 116 117 # access network 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 + ''; 126 127 passthru = { 128 tests = { inherit (nixosTests) litellm; };