diff options
author | Alyssa Ross <hi@alyssa.is> | 2023-09-15 07:58:52 +0000 |
---|---|---|
committer | Alyssa Ross <hi@alyssa.is> | 2023-09-15 07:58:52 +0000 |
commit | 9d1daa60832979d5d361dfdac136fb9e5a1af2c5 (patch) | |
tree | 5192dd85903cb092cf7dff0e3403387b3b683d84 /nixpkgs/pkgs/development/python-modules/litellm/default.nix | |
parent | 67cdfc7d42f721bf85814af5a0095fb9f9ea455d (diff) | |
parent | f2ea252d23ebc9a5336bf6a61e0644921f64e67c (diff) | |
download | nixlib-9d1daa60832979d5d361dfdac136fb9e5a1af2c5.tar nixlib-9d1daa60832979d5d361dfdac136fb9e5a1af2c5.tar.gz nixlib-9d1daa60832979d5d361dfdac136fb9e5a1af2c5.tar.bz2 nixlib-9d1daa60832979d5d361dfdac136fb9e5a1af2c5.tar.lz nixlib-9d1daa60832979d5d361dfdac136fb9e5a1af2c5.tar.xz nixlib-9d1daa60832979d5d361dfdac136fb9e5a1af2c5.tar.zst nixlib-9d1daa60832979d5d361dfdac136fb9e5a1af2c5.zip |
Merge branch 'nixos-unstable' of https://github.com/NixOS/nixpkgs
Diffstat (limited to 'nixpkgs/pkgs/development/python-modules/litellm/default.nix')
-rw-r--r-- | nixpkgs/pkgs/development/python-modules/litellm/default.nix | 53 |
1 files changed, 53 insertions, 0 deletions
diff --git a/nixpkgs/pkgs/development/python-modules/litellm/default.nix b/nixpkgs/pkgs/development/python-modules/litellm/default.nix new file mode 100644 index 000000000000..1e75776f0c32 --- /dev/null +++ b/nixpkgs/pkgs/development/python-modules/litellm/default.nix @@ -0,0 +1,53 @@ +{ lib +, buildPythonPackage +, fetchFromGitHub +, poetry-core +, importlib-metadata +, openai +, python-dotenv +, tiktoken +}: +let + version = "0.1.590"; +in +buildPythonPackage rec { + pname = "litellm"; + format = "pyproject"; + inherit version; + + src = fetchFromGitHub { + owner = "BerriAI"; + repo = "litellm"; + rev = "7cb96e86b4753008cbf8d116aca514750e98d360"; + hash = "sha256-ITMcwGjelNfNGnfBmmdu0Xwph4u0mxiFSfHnysUxWCQ="; + }; + + postPatch = '' + rm -rf dist + ''; + + nativeBuildInputs = [ + poetry-core + ]; + + propagatedBuildInputs = [ + importlib-metadata + openai + python-dotenv + tiktoken + ]; + + # the import check phase fails trying to do a network request to openai + # pythonImportsCheck = [ "litellm" ]; + + # no tests + doCheck = false; + + meta = with lib; { + description = "Use any LLM as a drop in replacement for gpt-3.5-turbo. Use Azure, OpenAI, Cohere, Anthropic, Ollama, VLLM, Sagemaker, HuggingFace, Replicate (100+ LLMs)"; + homepage = "https://github.com/BerriAI/litellm"; + license = licenses.mit; + changelog = "https://github.com/BerriAI/litellm/releases/tag/v${version}"; + maintainers = with maintainers; [ happysalada ]; + }; +} |