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

   1 {
   2   lib,
   3   asyncpg,
   4   buildPythonPackage,
   5   fetchPypi,
   6   llama-index-core,
   7   pgvector,
   8   poetry-core,
   9   psycopg2,
  10   pythonRelaxDepsHook,
  11   pythonOlder,
  12 }:
  13
  14 buildPythonPackage rec {
  15   pname = "llama-index-vector-stores-postgres";
  16   version = "0.1.5";
  17   pyproject = true;
  18
  19   disabled = pythonOlder "3.8";
  20
  21   src = fetchPypi {
  22     pname = "llama_index_vector_stores_postgres";
  23     inherit version;
  24     hash = "sha256-9jE+1Gbx2y/CSqkpSfuYqgyX49yZwhwmJbiG/EHwTLw=";
  25   };
  26
  27   pythonRemoveDeps = [ "psycopg2-binary" ];
  28
  29   build-system = [
  30     poetry-core
  31     pythonRelaxDepsHook
  32   ];
  33
  34   dependencies = [
  35     asyncpg
  36     llama-index-core
  37     pgvector
  38     psycopg2
  39   ];
  40
  41   pythonImportsCheck = [ "llama_index.vector_stores.postgres" ];
  42
  43   meta = with lib; {
  44     description = "LlamaIndex Vector Store Integration for Postgres";
  45     homepage = "https://github.com/run-llama/llama_index/tree/main/llama-index-integrations/vector_stores/llama-index-vector-stores-postgres";
  46     license = licenses.mit;
  47     maintainers = with maintainers; [ fab ];
  48   };
  49 }