python311Packages.google-generativeai: init at 0.1.0

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 ];
+  };
+}

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 { };