diff --git a/pkgs/development/python-modules/google-generativeai/default.nix b/pkgs/development/python-modules/google-generativeai/default.nix new file mode 100644 index 000000000000..ffab9af49802 --- /dev/null +++ b/pkgs/development/python-modules/google-generativeai/default.nix @@ -0,0 +1,49 @@ +{ lib +, buildPythonPackage +, fetchFromGitHub +, google-ai-generativelanguage +, pythonOlder +, pythonRelaxDepsHook +}: + +buildPythonPackage rec { + pname = "google-generativeai"; + version = "0.1.0"; + format = "setuptools"; + + disabled = pythonOlder "3.9"; + + src = fetchFromGitHub { + owner = "google"; + repo = "generative-ai-python"; + rev = "refs/tags/v${version}"; + hash = "sha256-caFq7BZKRuNU7fKJWeWSbbDFNpav7OrIET4OQSHkPz4="; + }; + + pythonRelaxDeps = [ + "google-ai-generativelanguage" + ]; + + nativeBuildInputs = [ + pythonRelaxDepsHook + ]; + + propagatedBuildInputs = [ + google-ai-generativelanguage + ]; + + # Issue with the google.ai module. Check with the next release + doCheck = false; + + pythonImportsCheck = [ + "google.generativeai" + ]; + + meta = with lib; { + description = "Python client library for Google's large language model PaLM API"; + homepage = "https://github.com/google/generative-ai-python"; + changelog = "https://github.com/google/generative-ai-python/releases/tag/v${version}"; + license = licenses.asl20; + maintainers = with maintainers; [ fab ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 8468e89b4bb9..ffa4fc488df4 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -4295,6 +4295,8 @@ self: super: with self; { inherit (pkgs) crc32c; }; + google-generativeai = callPackage ../development/python-modules/google-generativeai { }; + google-i18n-address = callPackage ../development/python-modules/google-i18n-address { }; google-nest-sdm = callPackage ../development/python-modules/google-nest-sdm { };