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 }:
  11
  12 buildPythonPackage rec {
  13   pname = "llama-index-vector-stores-postgres";
  14   version = "0.4.1";
  15   pyproject = true;
  16
  17   src = fetchPypi {
  18     pname = "llama_index_vector_stores_postgres";
  19     inherit version;
  20     hash = "sha256-4bbjXMH4T/BRJ/l+vDOYqbv2nn0HE9953h4qFRXFlXQ=";
  21   };
  22
  23   pythonRemoveDeps = [ "psycopg2-binary" ];
  24
  25   pythonRelaxDeps = [
  26     "pgvector"
  27   ];
  28
  29   build-system = [
  30     poetry-core
  31   ];
  32
  33   dependencies = [
  34     asyncpg
  35     llama-index-core
  36     pgvector
  37     psycopg2
  38   ];
  39
  40   pythonImportsCheck = [ "llama_index.vector_stores.postgres" ];
  41
  42   meta = {
  43     description = "LlamaIndex Vector Store Integration for Postgres";
  44     homepage = "https://github.com/run-llama/llama_index/tree/main/llama-index-integrations/vector_stores/llama-index-vector-stores-postgres";
  45     license = lib.licenses.mit;
  46     maintainers = with lib.maintainers; [ fab ];
  47   };
  48 }