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

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchFromGitHub,
   5
   6   # build-system
   7   poetry-core,
   8
   9   # dependencies
  10   langchain-core,
  11   numpy,
  12   pymongo,
  13
  14   freezegun,
  15   httpx,
  16   langchain,
  17   pytest-asyncio,
  18   pytestCheckHook,
  19   pytest-mock,
  20   syrupy,
  21 }:
  22
  23 buildPythonPackage rec {
  24   pname = "langchain-mongodb";
  25   version = "0.2.0";
  26   pyproject = true;
  27
  28   src = fetchFromGitHub {
  29     owner = "langchain-ai";
  30     repo = "langchain";
  31     tag = "langchain-mongodb==${version}";
  32     hash = "sha256-Jd9toXkS9dGtSIrJQ/5W+swV1z2BJOJKBtkyGzj3oSc=";
  33   };
  34
  35   sourceRoot = "${src.name}/libs/partners/mongodb";
  36
  37   build-system = [ poetry-core ];
  38
  39   pythonRelaxDeps = [
  40     "numpy"
  41   ];
  42
  43   dependencies = [
  44     langchain-core
  45     numpy
  46     pymongo
  47   ];
  48
  49   nativeCheckInputs = [
  50     freezegun
  51     httpx
  52     langchain
  53     pytest-asyncio
  54     pytestCheckHook
  55     pytest-mock
  56     syrupy
  57   ];
  58
  59   pytestFlagsArray = [ "tests/unit_tests" ];
  60
  61   pythonImportsCheck = [ "langchain_mongodb" ];
  62
  63   passthru = {
  64     inherit (langchain-core) updateScript;
  65   };
  66
  67   meta = {
  68     changelog = "https://github.com/langchain-ai/langchain/releases/tag/langchain-mongodb==${version}";
  69     description = "Integration package connecting MongoDB and LangChain";
  70     homepage = "https://github.com/langchain-ai/langchain/tree/master/libs/partners/mongodb";
  71     license = lib.licenses.mit;
  72     maintainers = with lib.maintainers; [
  73       natsukium
  74       sarahec
  75     ];
  76   };
  77 }