pkgs/development/python-modules/llama-index-vector-stores-chroma/default.nix

   1 {
   2   lib,
   3   buildPythonPackage,
   4   chromadb,
   5   fetchPypi,
   6   llama-index-core,
   7   pythonOlder,
   8   poetry-core,
   9 }:
  10
  11 buildPythonPackage rec {
  12   pname = "llama-index-vector-stores-chroma";
  13   version = "0.3.0";
  14   pyproject = true;
  15
  16   disabled = pythonOlder "3.8";
  17
  18   src = fetchPypi {
  19     pname = "llama_index_vector_stores_chroma";
  20     inherit version;
  21     hash = "sha256-VAWSFqI71r0dw69MWYbbOLKTNYawQujuEmOF16XeziM=";
  22   };
  23
  24   build-system = [ poetry-core ];
  25
  26   dependencies = [
  27     chromadb
  28     llama-index-core
  29   ];
  30
  31   pythonImportsCheck = [ "llama_index.vector_stores.chroma" ];
  32
  33   meta = with lib; {
  34     description = "LlamaIndex Vector Store Integration for Chroma";
  35     homepage = "https://github.com/run-llama/llama_index/tree/main/llama-index-integrations/vector_stores/llama-index-vector-stores-chroma";
  36     license = licenses.mit;
  37     maintainers = with maintainers; [ fab ];
  38   };
  39 }