Unverified Commit bce8e5a9 authored by Pavol Rusnak's avatar Pavol Rusnak
Browse files

python3Packages.mistral-common: init at 1.5.2

parent 9abe16fa
Loading
Loading
Loading
Loading
+57 −0
Original line number Diff line number Diff line
{
  lib,
  buildPythonPackage,
  fetchPypi,
  poetry-core,
  numpy,
  pydantic,
  jsonschema,
  sentencepiece,
  typing-extensions,
  tiktoken,
  pillow,
  requests,
}:

buildPythonPackage rec {
  pname = "mistral-common";
  version = "1.5.2";
  pyproject = true;

  src = fetchPypi {
    pname = "mistral_common";
    inherit version;
    hash = "sha256-nRFXsTdsSdNav8dD2+AITyyjezpavQPnQSdqG8ZshS8=";
  };

  # relax dependencies
  postPatch = ''
    substituteInPlace pyproject.toml \
      --replace-fail 'pillow = "^10.3.0"' 'pillow = ">=10.3.0"' \
      --replace-fail 'tiktoken = "^0.7.0"' 'tiktoken = ">=0.7.0"' \
  '';

  build-system = [ poetry-core ];

  dependencies = [
    numpy
    pydantic
    jsonschema
    sentencepiece
    typing-extensions
    tiktoken
    pillow
    requests
  ];

  doCheck = true;

  pythonImportsCheck = [ "mistral_common" ];

  meta = with lib; {
    description = "mistral-common is a set of tools to help you work with Mistral models.";
    homepage = "https://github.com/mistralai/mistral-common";
    license = licenses.asl20;
    maintainers = with maintainers; [ bgamari ];
  };
}
+2 −0
Original line number Diff line number Diff line
@@ -8278,6 +8278,8 @@ self: super: with self; {
  mistletoe = callPackage ../development/python-modules/mistletoe { };
  mistral-common = callPackage ../development/python-modules/mistral-common { };
  mistune = callPackage ../development/python-modules/mistune { };
  mitmproxy = callPackage ../development/python-modules/mitmproxy { };