65 lines
1.6 KiB
Nix
65 lines
1.6 KiB
Nix
{
|
|
lib,
|
|
stdenv,
|
|
cmake,
|
|
fetchFromGitHub,
|
|
openssl,
|
|
postgresql,
|
|
postgresqlTestExtension,
|
|
buildPostgresqlExtension,
|
|
}:
|
|
|
|
buildPostgresqlExtension (finalAttrs: {
|
|
pname = "postgresql-lantern";
|
|
version = "0.5.0";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "lanterndata";
|
|
repo = "lantern";
|
|
rev = "v${finalAttrs.version}";
|
|
hash = "sha256-IsDD/um5pVvbzin8onf45DQVszl+Id/pJSQ2iijgHmg=";
|
|
fetchSubmodules = true;
|
|
};
|
|
|
|
postPatch = ''
|
|
patchShebangs --build lantern_hnsw/scripts/link_llvm_objects.sh
|
|
'';
|
|
|
|
nativeBuildInputs = [
|
|
cmake
|
|
];
|
|
|
|
buildInputs = [
|
|
openssl
|
|
];
|
|
|
|
cmakeFlags = [
|
|
"-DBUILD_FOR_DISTRIBUTING=ON"
|
|
"-S ../lantern_hnsw"
|
|
];
|
|
|
|
passthru.tests.extension = postgresqlTestExtension {
|
|
inherit (finalAttrs) finalPackage;
|
|
sql = ''
|
|
CREATE EXTENSION lantern;
|
|
|
|
CREATE TABLE small_world (id integer, vector real[3]);
|
|
INSERT INTO small_world (id, vector) VALUES (0, '{0,0,0}'), (1, '{0,0,1}');
|
|
|
|
CREATE INDEX ON small_world USING lantern_hnsw (vector dist_l2sq_ops)
|
|
WITH (M=2, ef_construction=10, ef=4, dim=3);
|
|
'';
|
|
};
|
|
|
|
meta = with lib; {
|
|
description = "PostgreSQL vector database extension for building AI applications";
|
|
homepage = "https://lantern.dev/";
|
|
changelog = "https://github.com/lanterndata/lantern/blob/${finalAttrs.src.rev}/CHANGELOG.md";
|
|
license = licenses.agpl3Only;
|
|
maintainers = [ ];
|
|
platforms = postgresql.meta.platforms;
|
|
# error: use of undeclared identifier 'aligned_alloc'
|
|
broken =
|
|
stdenv.hostPlatform.isDarwin && lib.versionOlder stdenv.hostPlatform.darwinMinVersion "10.13";
|
|
};
|
|
})
|