Merge pull request #240344 from natsukium/clarifai/init

python310Packages.clarifai: init at 9.5.2; python310Packages.langchain: add clarifai as an optional dependency

pkgs/development/python-modules/clarifai-grpc/default.nix

@@ -0,0 +1,42 @@
+{ lib
+, buildPythonPackage
+, fetchPypi
+, pythonOlder
+, googleapis-common-protos
+, grpcio
+, protobuf
+, requests
+}:
+
+buildPythonPackage rec {
+  pname = "clarifai-grpc";
+  version = "9.5.0";
+  format = "setuptools";
+
+  disabled = pythonOlder "3.8";
+
+  src = fetchPypi {
+    inherit pname version;
+    hash = "sha256-amMUgRXtvcrALfWwat7W0W60FJQnEeMvHhyxxBCMqUg=";
+  };
+
+  propagatedBuildInputs = [
+    googleapis-common-protos
+    grpcio
+    protobuf
+    requests
+  ];
+
+  # almost all tests require network access
+  doCheck = false;
+
+  pythonImportsCheck = [ "clarifai_grpc" ];
+
+  meta = with lib; {
+    description = "Clarifai gRPC API Client";
+    homepage = "https://github.com/Clarifai/clarifai-python-grpc";
+    changelog = "https://github.com/Clarifai/clarifai-python-grpc/releases/tag/${version}";
+    license = licenses.asl20;
+    maintainers = with maintainers; [ natsukium ];
+  };
+}

pkgs/development/python-modules/clarifai/default.nix

@@ -0,0 +1,40 @@
+{ lib
+, buildPythonPackage
+, fetchFromGitHub
+, pythonOlder
+, clarifai-grpc
+, pytestCheckHook
+}:
+
+buildPythonPackage rec {
+  pname = "clarifai";
+  version = "9.5.2";
+  format = "setuptools";
+
+  disabled = pythonOlder "3.8";
+
+  src = fetchFromGitHub {
+    owner = "Clarifai";
+    repo = "clarifai-python-utils";
+    rev = "refs/tags/${version}";
+    hash = "sha256-29by0YAQ7qc0gL/3lAFOk4FLDB5Qv4X9QDyK49gfyAo=";
+  };
+
+  propagatedBuildInputs = [
+    clarifai-grpc
+  ];
+
+  nativeCheckInputs = [
+    pytestCheckHook
+  ];
+
+  pythonImportsCheck = [ "clarifai" ];
+
+  meta = with lib; {
+    description = "Clarifai Python Utilities";
+    homepage = "https://github.com/Clarifai/clarifai-python-utils";
+    changelog = "https://github.com/Clarifai/clarifai-python-utils/releases/tag/${src.rev}";
+    license = licenses.asl20;
+    maintainers = with maintainers; [ natsukium ];
+  };
+}

pkgs/development/python-modules/langchain/default.nix

@@ -17,6 +17,7 @@
 , bash
   # optional dependencies
 , anthropic
+, clarifai
 , cohere
 , openai
 , nlpcloud
@@ -123,6 +124,7 @@ buildPythonPackage rec {
   passthru.optional-dependencies = {
     llms = [
       anthropic
+      clarifai
       cohere
       openai
       nlpcloud
@@ -140,6 +142,9 @@ buildPythonPackage rec {
     text_helpers = [
       chardet
     ];
+    clarifai = [
+      clarifai
+    ];
     cohere = [
       cohere
     ];
@@ -157,6 +162,7 @@ buildPythonPackage rec {
     ];
     all = [
       anthropic
+      clarifai
       cohere
       openai
       nlpcloud

pkgs/top-level/python-packages.nix

@@ -1861,6 +1861,10 @@ self: super: with self; {
 
   ckcc-protocol = callPackage ../development/python-modules/ckcc-protocol { };
 
+  clarifai = callPackage ../development/python-modules/clarifai { };
+
+  clarifai-grpc = callPackage ../development/python-modules/clarifai-grpc { };
+
   claripy = callPackage ../development/python-modules/claripy { };
 
   classify-imports = callPackage ../development/python-modules/classify-imports { };