diff --git a/pkgs/development/python-modules/clarifai-grpc/default.nix b/pkgs/development/python-modules/clarifai-grpc/default.nix new file mode 100644 index 000000000000..cec151526074 --- /dev/null +++ b/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 ]; + }; +} diff --git a/pkgs/development/python-modules/clarifai/default.nix b/pkgs/development/python-modules/clarifai/default.nix new file mode 100644 index 000000000000..ffc02aa60480 --- /dev/null +++ b/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 ]; + }; +} diff --git a/pkgs/development/python-modules/langchain/default.nix b/pkgs/development/python-modules/langchain/default.nix index ee16723f3afd..50c08bb67557 100644 --- a/pkgs/development/python-modules/langchain/default.nix +++ b/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 diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index fea53f104cdd..3b1fd05cb344 100644 --- a/pkgs/top-level/python-packages.nix +++ b/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 { };