mirror of
https://github.com/lancedb/lancedb.git
synced 2025-12-23 13:29:57 +00:00
Compare commits
843 Commits
python-v0.
...
python-v0.
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
bb809abd4b | ||
|
|
c87530f7a3 | ||
|
|
1eb1beecd6 | ||
|
|
ce550e6c45 | ||
|
|
d3bae1f3a3 | ||
|
|
dcf53c4506 | ||
|
|
941eada703 | ||
|
|
ed640a76d9 | ||
|
|
296205ef96 | ||
|
|
16beaaa656 | ||
|
|
4ff87b1f4a | ||
|
|
0532ef2358 | ||
|
|
dcf7334c1f | ||
|
|
8ffe992a6f | ||
|
|
9d683e4f0b | ||
|
|
0a1ea1858d | ||
|
|
7d0127b376 | ||
|
|
02595dc475 | ||
|
|
f23327af79 | ||
|
|
c7afa724dd | ||
|
|
c359cec504 | ||
|
|
fe76496a59 | ||
|
|
67ec1fe75c | ||
|
|
70d9b04ba5 | ||
|
|
b0d4a79c35 | ||
|
|
f79295c697 | ||
|
|
381fad9b65 | ||
|
|
055bf91d3e | ||
|
|
050f0086b8 | ||
|
|
10fa23e0d6 | ||
|
|
43d9fc28b0 | ||
|
|
f45f0d0431 | ||
|
|
b9e3c36d82 | ||
|
|
3cd7dd3375 | ||
|
|
12d4ce4cfe | ||
|
|
3d1f102087 | ||
|
|
81afd8a42f | ||
|
|
c2aa03615a | ||
|
|
d2c6759e7f | ||
|
|
94fb9f364a | ||
|
|
fbff244ed8 | ||
|
|
7e7466d224 | ||
|
|
cceaf27d79 | ||
|
|
7a15337e03 | ||
|
|
96c66fd087 | ||
|
|
0579303602 | ||
|
|
75edb8756c | ||
|
|
88283110f4 | ||
|
|
b3a637fdeb | ||
|
|
ce24457531 | ||
|
|
087fe6343d | ||
|
|
ab8cbe62dd | ||
|
|
f076bb41f4 | ||
|
|
902fb83d54 | ||
|
|
779118339f | ||
|
|
03b62599d7 | ||
|
|
4c999fb651 | ||
|
|
6d23d32ab5 | ||
|
|
704cec34e1 | ||
|
|
a300a238db | ||
|
|
a41ff1df0a | ||
|
|
77b005d849 | ||
|
|
167fccc427 | ||
|
|
2bffbcefa5 | ||
|
|
905552f993 | ||
|
|
e4898c9313 | ||
|
|
cab36d94b2 | ||
|
|
b64252d4fd | ||
|
|
6fc006072c | ||
|
|
d4bb59b542 | ||
|
|
6b2dd6de51 | ||
|
|
dbccd9e4f1 | ||
|
|
b12ebfed4c | ||
|
|
1dadb2aefa | ||
|
|
eb9784d7f2 | ||
|
|
ba755626cc | ||
|
|
7760799cb8 | ||
|
|
4beb2d2877 | ||
|
|
a00b8595d1 | ||
|
|
9c8314b4fd | ||
|
|
c625b6f2b2 | ||
|
|
bec8fe6547 | ||
|
|
dc1150c011 | ||
|
|
afaefc6264 | ||
|
|
cb70ff8cee | ||
|
|
cbb5a841b1 | ||
|
|
c72f6770fd | ||
|
|
e5a80a5e86 | ||
|
|
8d0a7fad1f | ||
|
|
b80d4d0134 | ||
|
|
9645fe52c2 | ||
|
|
b77314168d | ||
|
|
e08d45e090 | ||
|
|
2e3ddb8382 | ||
|
|
627ca4c810 | ||
|
|
f8dae4ffe9 | ||
|
|
9eb6119468 | ||
|
|
59b57e30ed | ||
|
|
fec8d58f06 | ||
|
|
84ded9d678 | ||
|
|
65696d9713 | ||
|
|
e2f2ea32e4 | ||
|
|
d5f2eca754 | ||
|
|
7fa455a8a5 | ||
|
|
8f42b5874e | ||
|
|
274f19f560 | ||
|
|
fbcbc75b5b | ||
|
|
008f389bd0 | ||
|
|
91af6518d9 | ||
|
|
af6819762c | ||
|
|
7acece493d | ||
|
|
20e017fedc | ||
|
|
74e578b3c8 | ||
|
|
d92d9eb3d2 | ||
|
|
b6cdce7bc9 | ||
|
|
316b406265 | ||
|
|
8825c7c1dd | ||
|
|
81c85ff702 | ||
|
|
570f2154d5 | ||
|
|
0525c055fc | ||
|
|
38d11291da | ||
|
|
258e682574 | ||
|
|
d7afa600b8 | ||
|
|
5c7303ab2e | ||
|
|
5895ef4039 | ||
|
|
0528cd858a | ||
|
|
6582f43422 | ||
|
|
5c7f63388d | ||
|
|
d0bc671cac | ||
|
|
d37e17593d | ||
|
|
cb726d370e | ||
|
|
23ee132546 | ||
|
|
7fa090d330 | ||
|
|
07bc1c5397 | ||
|
|
d7a9dbb9fc | ||
|
|
00487afc7d | ||
|
|
1902d65aad | ||
|
|
c4fbb65b8e | ||
|
|
875ed7ae6f | ||
|
|
95a46a57ba | ||
|
|
51561e31a0 | ||
|
|
7b19120578 | ||
|
|
745c34a6a9 | ||
|
|
db8fa2454d | ||
|
|
a67a7b4b42 | ||
|
|
496846e532 | ||
|
|
dadcfebf8e | ||
|
|
67033dbd7f | ||
|
|
05a85cfc2a | ||
|
|
40c5d3d72b | ||
|
|
198f0f80c6 | ||
|
|
e3f2fd3892 | ||
|
|
f401ccc599 | ||
|
|
81b59139f8 | ||
|
|
1026781ab6 | ||
|
|
9c699b8cd9 | ||
|
|
34bec59bc3 | ||
|
|
a5fbbf0d66 | ||
|
|
b42721167b | ||
|
|
543dec9ff0 | ||
|
|
04f962f6b0 | ||
|
|
19e896ff69 | ||
|
|
272e4103b2 | ||
|
|
75c257ebb6 | ||
|
|
9ee152eb42 | ||
|
|
c9ae1b1737 | ||
|
|
89dc80c42a | ||
|
|
7b020ac799 | ||
|
|
529e774bbb | ||
|
|
7c12239305 | ||
|
|
d83424d6b4 | ||
|
|
8bf89f887c | ||
|
|
b2160b2304 | ||
|
|
1bb82597be | ||
|
|
e4eee38b3c | ||
|
|
64fc2be503 | ||
|
|
dc8054e90d | ||
|
|
1684940946 | ||
|
|
695813463c | ||
|
|
ed594b0f76 | ||
|
|
cee2b5ea42 | ||
|
|
f315f9665a | ||
|
|
5deb26bc8b | ||
|
|
3cc670ac38 | ||
|
|
4ade3e31e2 | ||
|
|
a222d2cd91 | ||
|
|
508e621f3d | ||
|
|
a1a0472f3f | ||
|
|
3425a6d339 | ||
|
|
af54e0ce06 | ||
|
|
089905fe8f | ||
|
|
554939e5d2 | ||
|
|
7a13814922 | ||
|
|
e9f25f6a12 | ||
|
|
419a433244 | ||
|
|
a9311c4dc0 | ||
|
|
178bcf9c90 | ||
|
|
b9be092cb1 | ||
|
|
e8c0c52315 | ||
|
|
a60fa0d3b7 | ||
|
|
726d629b9b | ||
|
|
b493f56dee | ||
|
|
a8b5ad7e74 | ||
|
|
f8f6264883 | ||
|
|
d8517117f1 | ||
|
|
ab66dd5ed2 | ||
|
|
cbb9a7877c | ||
|
|
b7fc223535 | ||
|
|
1fdaf7a1a4 | ||
|
|
d11819c90c | ||
|
|
9b902272f1 | ||
|
|
8c0622fa2c | ||
|
|
2191f948c3 | ||
|
|
acc3b03004 | ||
|
|
7f091b8c8e | ||
|
|
c19bdd9a24 | ||
|
|
dad0ff5cd2 | ||
|
|
a705621067 | ||
|
|
39614fdb7d | ||
|
|
96d534d4bc | ||
|
|
5051d30d09 | ||
|
|
db853c4041 | ||
|
|
76d1d22bdc | ||
|
|
d8746c61c6 | ||
|
|
1a66df2627 | ||
|
|
44670076c1 | ||
|
|
92f0b16e46 | ||
|
|
1620ba3508 | ||
|
|
3ae90dde80 | ||
|
|
4f07fea6df | ||
|
|
3d7d82cf86 | ||
|
|
edc4e40a7b | ||
|
|
ca3806a02f | ||
|
|
35cff12e31 | ||
|
|
c6c20cb2bd | ||
|
|
26080ee4c1 | ||
|
|
ef3a2b5357 | ||
|
|
c42a201389 | ||
|
|
24e42ccd4d | ||
|
|
8a50944061 | ||
|
|
40e066bc7c | ||
|
|
b3ad105fa0 | ||
|
|
6e701d3e1b | ||
|
|
2248aa9508 | ||
|
|
a6fa69ab89 | ||
|
|
b3a4efd587 | ||
|
|
4708b60bb1 | ||
|
|
080ea2f9a4 | ||
|
|
32fdde23f8 | ||
|
|
c44e5c046c | ||
|
|
f23aa0a793 | ||
|
|
83fc2b1851 | ||
|
|
56aa133ee6 | ||
|
|
27d9e5c596 | ||
|
|
ec8271931f | ||
|
|
6c6966600c | ||
|
|
2e170c3c7b | ||
|
|
fd92e651d1 | ||
|
|
c298482ee1 | ||
|
|
d59f64b5a3 | ||
|
|
30ed8c4c43 | ||
|
|
4a2cdbf299 | ||
|
|
657843d9e9 | ||
|
|
1cd76b8498 | ||
|
|
a38f784081 | ||
|
|
647dee4e94 | ||
|
|
0844c2dd64 | ||
|
|
fd2692295c | ||
|
|
d4ea50fba1 | ||
|
|
0d42297cf8 | ||
|
|
a6d4125cbf | ||
|
|
5c32a99e61 | ||
|
|
cefaa75b24 | ||
|
|
bd62c2384f | ||
|
|
f0bc08c0d7 | ||
|
|
e52ac79c69 | ||
|
|
f091f57594 | ||
|
|
a997fd4108 | ||
|
|
1486514ccc | ||
|
|
a505bc3965 | ||
|
|
c1738250a3 | ||
|
|
1ee63984f5 | ||
|
|
2eb2c8862a | ||
|
|
4ea8e178d3 | ||
|
|
e4485a630e | ||
|
|
fb95f9b3bd | ||
|
|
625bab3f21 | ||
|
|
e59f9382a0 | ||
|
|
fdee7ba477 | ||
|
|
c44fa3abc4 | ||
|
|
fc43aac0ed | ||
|
|
e67cd0baf9 | ||
|
|
26dab93f2a | ||
|
|
b9bdb8d937 | ||
|
|
a1d1833a40 | ||
|
|
a547c523c2 | ||
|
|
dc8b75feab | ||
|
|
c1600cdc06 | ||
|
|
f5dee46970 | ||
|
|
346cbf8bf7 | ||
|
|
3c7dfe9f28 | ||
|
|
f52d05d3fa | ||
|
|
c321cccc12 | ||
|
|
cba14a5743 | ||
|
|
72057b743d | ||
|
|
698f329598 | ||
|
|
79fa745130 | ||
|
|
2ad71bdeca | ||
|
|
7c13615096 | ||
|
|
f882f5b69a | ||
|
|
a68311a893 | ||
|
|
846a5cea33 | ||
|
|
e3dec647b5 | ||
|
|
c58104cecc | ||
|
|
b3b5362632 | ||
|
|
abe06fee3d | ||
|
|
93a82fd371 | ||
|
|
0d379e6ffa | ||
|
|
e1388bdfdd | ||
|
|
315a24c2bc | ||
|
|
6dd4cf6038 | ||
|
|
f97e751b3c | ||
|
|
e803a626a1 | ||
|
|
9403254442 | ||
|
|
b2a38ac366 | ||
|
|
bdb6c09c3b | ||
|
|
2bfdef2624 | ||
|
|
7982d5c082 | ||
|
|
7ff6ec7fe3 | ||
|
|
ba1ded933a | ||
|
|
b595d8a579 | ||
|
|
2a1d6d8abf | ||
|
|
440a466a13 | ||
|
|
b9afd9c860 | ||
|
|
a6b6f6a806 | ||
|
|
ae1548b507 | ||
|
|
4e03ee82bc | ||
|
|
46a6846d07 | ||
|
|
a207213358 | ||
|
|
6c321c694a | ||
|
|
5c00b2904c | ||
|
|
14677d7c18 | ||
|
|
dd22a379b2 | ||
|
|
7747c9bcbf | ||
|
|
c9d6fc43a6 | ||
|
|
581bcfbb88 | ||
|
|
3750639b5f | ||
|
|
e744d54460 | ||
|
|
9d1ce4b5a5 | ||
|
|
729ce5e542 | ||
|
|
de6739e7ec | ||
|
|
495216efdb | ||
|
|
a3b45a4d00 | ||
|
|
c316c2f532 | ||
|
|
3966b16b63 | ||
|
|
5661cc15ac | ||
|
|
4e7220400f | ||
|
|
ae4928fe77 | ||
|
|
e80a405dee | ||
|
|
a53e19e386 | ||
|
|
c0097c5f0a | ||
|
|
c199708e64 | ||
|
|
4a47150ae7 | ||
|
|
f86b20a564 | ||
|
|
cc81f3e1a5 | ||
|
|
bc49c4db82 | ||
|
|
d2eec46f17 | ||
|
|
51437bc228 | ||
|
|
fa53cfcfd2 | ||
|
|
374fe0ad95 | ||
|
|
35e5b84ba9 | ||
|
|
7c12d497b0 | ||
|
|
dfe4ba8dad | ||
|
|
fa1b9ad5bd | ||
|
|
8877eb020d | ||
|
|
01e4291d21 | ||
|
|
ab3ea76ad1 | ||
|
|
728ef8657d | ||
|
|
0b13901a16 | ||
|
|
84b110e0ef | ||
|
|
e1836e54e3 | ||
|
|
4ba5326880 | ||
|
|
b036a69300 | ||
|
|
5b12a47119 | ||
|
|
769d483e50 | ||
|
|
9ecb11fe5a | ||
|
|
22bd8329f3 | ||
|
|
a736fad149 | ||
|
|
072adc41aa | ||
|
|
c6f25ef1f0 | ||
|
|
2f0c5baea2 | ||
|
|
a63dd66d41 | ||
|
|
d6b3ccb37b | ||
|
|
c4f99e82e5 | ||
|
|
979a2d3d9d | ||
|
|
7ac5f74c80 | ||
|
|
ecdee4d2b1 | ||
|
|
f391ed828a | ||
|
|
a99a450f2b | ||
|
|
6fa1f37506 | ||
|
|
544382df5e | ||
|
|
784f00ef6d | ||
|
|
96d7446f70 | ||
|
|
99ea78fb55 | ||
|
|
8eef4cdc28 | ||
|
|
0f102f02c3 | ||
|
|
a33a0670f6 | ||
|
|
14c9ff46d1 | ||
|
|
1865f7decf | ||
|
|
a608621476 | ||
|
|
00514999ff | ||
|
|
b3b597fef6 | ||
|
|
bf17144591 | ||
|
|
09e110525f | ||
|
|
40f0dbb64d | ||
|
|
3b19e96ae7 | ||
|
|
78a17ad54c | ||
|
|
a8e6b491e2 | ||
|
|
cea541ca46 | ||
|
|
873ffc1042 | ||
|
|
83273ad997 | ||
|
|
d18d63c69d | ||
|
|
c3e865e8d0 | ||
|
|
a7755cb313 | ||
|
|
3490f3456f | ||
|
|
0a1d0693e1 | ||
|
|
fd330b4b4b | ||
|
|
d4e9fc08e0 | ||
|
|
3626f2f5e1 | ||
|
|
e64712cfa5 | ||
|
|
3e3118f85c | ||
|
|
592598a333 | ||
|
|
5ad21341c9 | ||
|
|
6e08caa091 | ||
|
|
7e259d8b0f | ||
|
|
e84f747464 | ||
|
|
998cd43fe6 | ||
|
|
4bc7eebe61 | ||
|
|
2e3b34e79b | ||
|
|
e7574698eb | ||
|
|
801a9e5f6f | ||
|
|
4e5fbe6c99 | ||
|
|
1a449fa49e | ||
|
|
6bf742c759 | ||
|
|
ef3093bc23 | ||
|
|
16851389ea | ||
|
|
c269524b2f | ||
|
|
f6eef14313 | ||
|
|
32716adaa3 | ||
|
|
5e98b7f4c0 | ||
|
|
3f2589c11f | ||
|
|
e3b99694d6 | ||
|
|
9d42dc349c | ||
|
|
482f1ee1d3 | ||
|
|
2f39274a66 | ||
|
|
2fc174f532 | ||
|
|
dba85f4d6f | ||
|
|
555fa26147 | ||
|
|
e05c0cd87e | ||
|
|
25c17ebf4e | ||
|
|
87b12b57dc | ||
|
|
3dc9b71914 | ||
|
|
2622f34d1a | ||
|
|
a677a4b651 | ||
|
|
e6b4f14c1f | ||
|
|
15f8f4d627 | ||
|
|
6526d6c3b1 | ||
|
|
da4d7e3ca7 | ||
|
|
8fbadca9aa | ||
|
|
29120219cf | ||
|
|
a9897d9d85 | ||
|
|
acda7a4589 | ||
|
|
dac0857745 | ||
|
|
0a9e1eab75 | ||
|
|
d999d72c8d | ||
|
|
de4720993e | ||
|
|
6c14a307e2 | ||
|
|
43747278c8 | ||
|
|
e5f42a850e | ||
|
|
7920ecf66e | ||
|
|
28e1b70e4b | ||
|
|
52b79d2b1e | ||
|
|
c05d45150d | ||
|
|
48ed3bb544 | ||
|
|
bcfc93cc88 | ||
|
|
214d0debf5 | ||
|
|
f059372137 | ||
|
|
3dc1803c07 | ||
|
|
d0501f65f1 | ||
|
|
4703cc6894 | ||
|
|
493f9ce467 | ||
|
|
5c759505b8 | ||
|
|
bb6a39727e | ||
|
|
d57bed90e5 | ||
|
|
648327e90c | ||
|
|
6c7e81ee57 | ||
|
|
905e9d4738 | ||
|
|
38642e349c | ||
|
|
6879861ea8 | ||
|
|
88325e488e | ||
|
|
995bd9bf37 | ||
|
|
36cc06697f | ||
|
|
35da464591 | ||
|
|
31f9c30ffb | ||
|
|
92dcf24b0c | ||
|
|
6b0adba2d9 | ||
|
|
66cbf6b6c5 | ||
|
|
ce9506db71 | ||
|
|
b66cd943a7 | ||
|
|
d8d11f48e7 | ||
|
|
7ec5df3022 | ||
|
|
b17304172c | ||
|
|
fbe5408434 | ||
|
|
3f3f845c5a | ||
|
|
fbffe532a8 | ||
|
|
55ffc96e56 | ||
|
|
998c5f3f74 | ||
|
|
6eacae18c4 | ||
|
|
d3ea75cc2b | ||
|
|
f4afe456e8 | ||
|
|
ea5c2266b8 | ||
|
|
c557e77f09 | ||
|
|
3c0a64be8f | ||
|
|
0e496ed3b5 | ||
|
|
17c9e9afea | ||
|
|
0b45ef93c0 | ||
|
|
b474f98049 | ||
|
|
2c05ffed52 | ||
|
|
8b31540b21 | ||
|
|
ba844318f8 | ||
|
|
f007b76153 | ||
|
|
5d8d258f59 | ||
|
|
4172140f74 | ||
|
|
a27c5cf12b | ||
|
|
f4dea72cc5 | ||
|
|
f76c4a5ce1 | ||
|
|
164ce397c2 | ||
|
|
445a312667 | ||
|
|
92d845fa72 | ||
|
|
397813f6a4 | ||
|
|
50c30c5d34 | ||
|
|
c9f248b058 | ||
|
|
0cb6da6b7e | ||
|
|
aec8332eb5 | ||
|
|
46061070e6 | ||
|
|
dae8334d0b | ||
|
|
8c81968b59 | ||
|
|
16cf2990f3 | ||
|
|
0a0f667bbd | ||
|
|
03753fd84b | ||
|
|
55cceaa309 | ||
|
|
c3797eb834 | ||
|
|
c0d0f38494 | ||
|
|
6a8ab78d0a | ||
|
|
27404c8623 | ||
|
|
f181c7e77f | ||
|
|
e70fd4fecc | ||
|
|
ac0068b80e | ||
|
|
ebac960571 | ||
|
|
59b57055e7 | ||
|
|
591c8de8fc | ||
|
|
f835ff310f | ||
|
|
cf8c2edaf4 | ||
|
|
61a714a459 | ||
|
|
5ddd84cec0 | ||
|
|
27ef0bb0a2 | ||
|
|
25402ba6ec | ||
|
|
37c359ed40 | ||
|
|
06cdf00987 | ||
|
|
144b7f5d54 | ||
|
|
edc9b9adec | ||
|
|
d11b2a6975 | ||
|
|
980aa70e2d | ||
|
|
d83e5a0208 | ||
|
|
16a6b9ce8f | ||
|
|
e3c6213333 | ||
|
|
00552439d9 | ||
|
|
c0ee370f83 | ||
|
|
17e4022045 | ||
|
|
c3ebac1a92 | ||
|
|
10f919a0a9 | ||
|
|
8af5476395 | ||
|
|
bcbbeb7a00 | ||
|
|
d6c0f75078 | ||
|
|
e820e356a0 | ||
|
|
509286492f | ||
|
|
f9789ec962 | ||
|
|
347515aa51 | ||
|
|
3324e7d525 | ||
|
|
ab5316b4fa | ||
|
|
db125013fc | ||
|
|
a43193c99b | ||
|
|
b70513ca72 | ||
|
|
78165801c6 | ||
|
|
6e5927ce6d | ||
|
|
6c1f32ac11 | ||
|
|
4fdf084777 | ||
|
|
1fad24fcd8 | ||
|
|
6ef20b85ca | ||
|
|
35bacdd57e | ||
|
|
a5ebe5a6c4 | ||
|
|
bf03ad1b4a | ||
|
|
2a9e3e2084 | ||
|
|
f298f15360 | ||
|
|
679b031b99 | ||
|
|
f50b5d532b | ||
|
|
fe655a15f0 | ||
|
|
9d0af794d0 | ||
|
|
048a2d10f8 | ||
|
|
c78a9849b4 | ||
|
|
c663085203 | ||
|
|
8b628854d5 | ||
|
|
a8d8c17b2a | ||
|
|
3c487e5fc7 | ||
|
|
d6219d687c | ||
|
|
239f725b32 | ||
|
|
5f261cf2d8 | ||
|
|
79eaa52184 | ||
|
|
bd82e1f66d | ||
|
|
ba34c3bee1 | ||
|
|
d4d0873e2b | ||
|
|
12c7bd18a5 | ||
|
|
c6bf6a25d6 | ||
|
|
c998a47e17 | ||
|
|
d8c758513c | ||
|
|
3795e02ee3 | ||
|
|
c7d424b2f3 | ||
|
|
1efb9914ee | ||
|
|
83e26a231e | ||
|
|
72a17b2de4 | ||
|
|
4231925476 | ||
|
|
84a6693294 | ||
|
|
6c2d4c10a4 | ||
|
|
d914722f79 | ||
|
|
a6e4034dba | ||
|
|
2616a50502 | ||
|
|
7b5e9d824a | ||
|
|
3b173e7cb9 | ||
|
|
d496ab13a0 | ||
|
|
69d9beebc7 | ||
|
|
d32360b99d | ||
|
|
9fa08bfa93 | ||
|
|
d6d9cb7415 | ||
|
|
990d93f553 | ||
|
|
0832cba3c6 | ||
|
|
38b0d91848 | ||
|
|
6826039575 | ||
|
|
3e9321fc40 | ||
|
|
2ded17452b | ||
|
|
dfd9d2ac99 | ||
|
|
162880140e | ||
|
|
99d9ced6d5 | ||
|
|
96933d7df8 | ||
|
|
d369233b3d | ||
|
|
43a670ed4b | ||
|
|
cb9a00a28d | ||
|
|
72af977a73 | ||
|
|
7cecb71df0 | ||
|
|
285071e5c8 | ||
|
|
114866fbcf | ||
|
|
5387c0e243 | ||
|
|
53d1535de1 | ||
|
|
b2f88f0b29 | ||
|
|
f2e3989831 | ||
|
|
83ae52938a | ||
|
|
267aa83bf8 | ||
|
|
cc72050206 | ||
|
|
72543c8b9d | ||
|
|
97d6210c33 | ||
|
|
a3d0c27b0a | ||
|
|
b23d8abcdd | ||
|
|
e3ea5cf9b9 | ||
|
|
4f8b086175 | ||
|
|
72330fb759 | ||
|
|
e3b2c5f438 | ||
|
|
66a881b33a | ||
|
|
a7515d6ee2 | ||
|
|
587c0824af | ||
|
|
b38a4269d0 | ||
|
|
119d88b9db | ||
|
|
74f660d223 | ||
|
|
b2b0979b90 | ||
|
|
ee2a40b182 | ||
|
|
4ca0b15354 | ||
|
|
d8c217b47d | ||
|
|
b724b1a01f | ||
|
|
abd75e0ead | ||
|
|
0fd8a50bd7 | ||
|
|
9f228feb0e | ||
|
|
90e9c52d0a | ||
|
|
68974a4e06 | ||
|
|
4c9bab0d92 | ||
|
|
5117aecc38 | ||
|
|
729718cb09 | ||
|
|
b1c84e0bda | ||
|
|
cbbc07d0f5 | ||
|
|
21021f94ca | ||
|
|
0ed77fa990 | ||
|
|
4372c231cd | ||
|
|
fa9ca8f7a6 | ||
|
|
2a35d24ee6 | ||
|
|
dd9ce337e2 | ||
|
|
b9921d56cc | ||
|
|
0cfd9ed18e | ||
|
|
975398c3a8 | ||
|
|
08d5f93f34 | ||
|
|
91cab3b556 | ||
|
|
c61bfc3af8 | ||
|
|
4e8c7b0adf | ||
|
|
26f4a80e10 | ||
|
|
3604d20ad3 | ||
|
|
9708d829a9 | ||
|
|
059c9794b5 | ||
|
|
15ed7f75a0 | ||
|
|
96181ab421 | ||
|
|
f3fc339ef6 | ||
|
|
113cd6995b | ||
|
|
02535bdc88 | ||
|
|
facc7d61c0 | ||
|
|
f947259f16 | ||
|
|
e291212ecf | ||
|
|
edc6445f6f | ||
|
|
a324f4ad7a | ||
|
|
55104c5bae | ||
|
|
d71df4572e | ||
|
|
aa269199ad | ||
|
|
32fdcf97db | ||
|
|
b9802a0d23 | ||
|
|
2ea5939f85 | ||
|
|
04e1f1ee4c | ||
|
|
bbc588e27d | ||
|
|
5517e102c3 | ||
|
|
82197c54e4 | ||
|
|
48f46d4751 | ||
|
|
437316cbbc | ||
|
|
d406eab2c8 | ||
|
|
1f41101897 | ||
|
|
99e4db0d6a | ||
|
|
46486d4d22 | ||
|
|
f43cb8bba1 | ||
|
|
38eb05f297 | ||
|
|
679a70231e | ||
|
|
e7b56b7b2a | ||
|
|
5ccd0edec2 | ||
|
|
9c74c435e0 | ||
|
|
6de53ce393 | ||
|
|
9f42fbba96 | ||
|
|
d892f7a622 | ||
|
|
515ab5f417 | ||
|
|
8d0055fe6b | ||
|
|
5f9d8509b3 | ||
|
|
f3b6a1f55b | ||
|
|
aff25e3bf9 | ||
|
|
8509f73221 | ||
|
|
607476788e | ||
|
|
4d458d5829 | ||
|
|
e61ba7f4e2 | ||
|
|
408bc96a44 | ||
|
|
6ceaf8b06e | ||
|
|
e2ca8daee1 | ||
|
|
f305f34d9b | ||
|
|
a416925ca1 | ||
|
|
2c4b07eb17 | ||
|
|
33b402c861 | ||
|
|
7b2cdd2269 | ||
|
|
d6b5054778 | ||
|
|
f0e7f5f665 | ||
|
|
f958f4d2e8 | ||
|
|
c1d9d6f70b | ||
|
|
1778219ea9 | ||
|
|
ee6c18f207 | ||
|
|
e606a455df | ||
|
|
8f0eb34109 | ||
|
|
2f2721e242 | ||
|
|
f00b21c98c | ||
|
|
962b3afd17 | ||
|
|
b72ac073ab | ||
|
|
3152ccd13c | ||
|
|
d5021356b4 | ||
|
|
e82f63b40a | ||
|
|
f81ce68e41 | ||
|
|
f5c25b6fff | ||
|
|
86978e7588 | ||
|
|
7c314d61cc | ||
|
|
7a8d2f37c4 | ||
|
|
11072b9edc | ||
|
|
915d828cee | ||
|
|
d9a72adc58 | ||
|
|
d6cf2dafc6 | ||
|
|
38f0031d0b | ||
|
|
e118c37228 | ||
|
|
abeaae3d80 | ||
|
|
b3c0227065 | ||
|
|
521e665f57 | ||
|
|
ffb28dd4fc | ||
|
|
32af962c0c | ||
|
|
18484d0b6c | ||
|
|
c02ee3c80c | ||
|
|
dcd5f51036 | ||
|
|
9b8472850e | ||
|
|
36d05ea641 | ||
|
|
7ed86cadfb | ||
|
|
1c123b58d8 | ||
|
|
bf7d2d6fb0 | ||
|
|
c7732585bf | ||
|
|
b3bf6386c3 | ||
|
|
4b79db72bf | ||
|
|
622a2922e2 | ||
|
|
c91221d710 | ||
|
|
56da5ebd13 | ||
|
|
64eb43229d | ||
|
|
c31c92122f | ||
|
|
205fc530cf | ||
|
|
2bde5401eb | ||
|
|
a405847f9b | ||
|
|
bcc19665ce | ||
|
|
2a6586d6fb | ||
|
|
029b01bbbf | ||
|
|
cd32944e54 | ||
|
|
7eb3b52297 | ||
|
|
8dcd328dce | ||
|
|
1d61717d0e | ||
|
|
4ee7225e91 | ||
|
|
2bc7dca3ca | ||
|
|
b24810a011 | ||
|
|
2b8e872be0 | ||
|
|
03ef1dc081 | ||
|
|
fde636ca2e | ||
|
|
51966a84f5 | ||
|
|
38015ffa7c | ||
|
|
dc72ece847 | ||
|
|
1521435193 | ||
|
|
bfe8fccfab | ||
|
|
6f6eb170a9 | ||
|
|
dd1c16bbaf | ||
|
|
a76186ee83 | ||
|
|
ae85008714 | ||
|
|
a85f039352 | ||
|
|
9c25998110 | ||
|
|
549ca51a8a | ||
|
|
632007d0e2 | ||
|
|
02d85a4ea4 | ||
|
|
a9d0625e2b |
@@ -1,5 +1,5 @@
|
|||||||
[tool.bumpversion]
|
[tool.bumpversion]
|
||||||
current_version = "0.10.0-beta.0"
|
current_version = "0.21.2"
|
||||||
parse = """(?x)
|
parse = """(?x)
|
||||||
(?P<major>0|[1-9]\\d*)\\.
|
(?P<major>0|[1-9]\\d*)\\.
|
||||||
(?P<minor>0|[1-9]\\d*)\\.
|
(?P<minor>0|[1-9]\\d*)\\.
|
||||||
@@ -24,34 +24,51 @@ commit = true
|
|||||||
message = "Bump version: {current_version} → {new_version}"
|
message = "Bump version: {current_version} → {new_version}"
|
||||||
commit_args = ""
|
commit_args = ""
|
||||||
|
|
||||||
[tool.bumpversion.parts.pre_l]
|
# Java maven files
|
||||||
values = ["beta", "final"]
|
pre_commit_hooks = [
|
||||||
optional_value = "final"
|
"""
|
||||||
|
NEW_VERSION="${BVHOOK_NEW_MAJOR}.${BVHOOK_NEW_MINOR}.${BVHOOK_NEW_PATCH}"
|
||||||
|
if [ ! -z "$BVHOOK_NEW_PRE_L" ] && [ ! -z "$BVHOOK_NEW_PRE_N" ]; then
|
||||||
|
NEW_VERSION="${NEW_VERSION}-${BVHOOK_NEW_PRE_L}.${BVHOOK_NEW_PRE_N}"
|
||||||
|
fi
|
||||||
|
echo "Constructed new version: $NEW_VERSION"
|
||||||
|
cd java && mvn versions:set -DnewVersion=$NEW_VERSION && mvn versions:commit
|
||||||
|
|
||||||
[[tool.bumpversion.files]]
|
# Check for any modified but unstaged pom.xml files
|
||||||
filename = "node/package.json"
|
MODIFIED_POMS=$(git ls-files -m | grep pom.xml)
|
||||||
search = "\"version\": \"{current_version}\","
|
if [ ! -z "$MODIFIED_POMS" ]; then
|
||||||
replace = "\"version\": \"{new_version}\","
|
echo "The following pom.xml files were modified but not staged. Adding them now:"
|
||||||
|
echo "$MODIFIED_POMS" | while read -r file; do
|
||||||
|
git add "$file"
|
||||||
|
echo "Added: $file"
|
||||||
|
done
|
||||||
|
fi
|
||||||
|
""",
|
||||||
|
]
|
||||||
|
|
||||||
|
[tool.bumpversion.parts.pre_l]
|
||||||
|
optional_value = "final"
|
||||||
|
values = ["beta", "final"]
|
||||||
|
|
||||||
[[tool.bumpversion.files]]
|
[[tool.bumpversion.files]]
|
||||||
filename = "nodejs/package.json"
|
filename = "nodejs/package.json"
|
||||||
search = "\"version\": \"{current_version}\","
|
|
||||||
replace = "\"version\": \"{new_version}\","
|
replace = "\"version\": \"{new_version}\","
|
||||||
|
search = "\"version\": \"{current_version}\","
|
||||||
|
|
||||||
# nodejs binary packages
|
# nodejs binary packages
|
||||||
[[tool.bumpversion.files]]
|
[[tool.bumpversion.files]]
|
||||||
glob = "nodejs/npm/*/package.json"
|
glob = "nodejs/npm/*/package.json"
|
||||||
search = "\"version\": \"{current_version}\","
|
|
||||||
replace = "\"version\": \"{new_version}\","
|
replace = "\"version\": \"{new_version}\","
|
||||||
|
search = "\"version\": \"{current_version}\","
|
||||||
|
|
||||||
# Cargo files
|
# Cargo files
|
||||||
# ------------
|
# ------------
|
||||||
[[tool.bumpversion.files]]
|
[[tool.bumpversion.files]]
|
||||||
filename = "rust/ffi/node/Cargo.toml"
|
filename = "rust/lancedb/Cargo.toml"
|
||||||
search = "\nversion = \"{current_version}\""
|
|
||||||
replace = "\nversion = \"{new_version}\""
|
replace = "\nversion = \"{new_version}\""
|
||||||
|
search = "\nversion = \"{current_version}\""
|
||||||
|
|
||||||
[[tool.bumpversion.files]]
|
[[tool.bumpversion.files]]
|
||||||
filename = "rust/lancedb/Cargo.toml"
|
filename = "nodejs/Cargo.toml"
|
||||||
search = "\nversion = \"{current_version}\""
|
|
||||||
replace = "\nversion = \"{new_version}\""
|
replace = "\nversion = \"{new_version}\""
|
||||||
|
search = "\nversion = \"{current_version}\""
|
||||||
|
|||||||
@@ -31,6 +31,13 @@ rustflags = [
|
|||||||
[target.x86_64-unknown-linux-gnu]
|
[target.x86_64-unknown-linux-gnu]
|
||||||
rustflags = ["-C", "target-cpu=haswell", "-C", "target-feature=+avx2,+fma,+f16c"]
|
rustflags = ["-C", "target-cpu=haswell", "-C", "target-feature=+avx2,+fma,+f16c"]
|
||||||
|
|
||||||
|
[target.x86_64-unknown-linux-musl]
|
||||||
|
rustflags = ["-C", "target-cpu=haswell", "-C", "target-feature=-crt-static,+avx2,+fma,+f16c"]
|
||||||
|
|
||||||
|
[target.aarch64-unknown-linux-musl]
|
||||||
|
linker = "aarch64-linux-musl-gcc"
|
||||||
|
rustflags = ["-C", "target-feature=-crt-static"]
|
||||||
|
|
||||||
[target.aarch64-apple-darwin]
|
[target.aarch64-apple-darwin]
|
||||||
rustflags = ["-C", "target-cpu=apple-m1", "-C", "target-feature=+neon,+fp16,+fhm,+dotprod"]
|
rustflags = ["-C", "target-cpu=apple-m1", "-C", "target-feature=+neon,+fp16,+fhm,+dotprod"]
|
||||||
|
|
||||||
@@ -38,3 +45,7 @@ rustflags = ["-C", "target-cpu=apple-m1", "-C", "target-feature=+neon,+fp16,+fhm
|
|||||||
# not found errors on systems that are missing it.
|
# not found errors on systems that are missing it.
|
||||||
[target.x86_64-pc-windows-msvc]
|
[target.x86_64-pc-windows-msvc]
|
||||||
rustflags = ["-Ctarget-feature=+crt-static"]
|
rustflags = ["-Ctarget-feature=+crt-static"]
|
||||||
|
|
||||||
|
# Experimental target for Arm64 Windows
|
||||||
|
[target.aarch64-pc-windows-msvc]
|
||||||
|
rustflags = ["-Ctarget-feature=+crt-static"]
|
||||||
|
|||||||
12
.github/workflows/build_linux_wheel/action.yml
vendored
12
.github/workflows/build_linux_wheel/action.yml
vendored
@@ -36,8 +36,7 @@ runs:
|
|||||||
args: ${{ inputs.args }}
|
args: ${{ inputs.args }}
|
||||||
before-script-linux: |
|
before-script-linux: |
|
||||||
set -e
|
set -e
|
||||||
yum install -y openssl-devel \
|
curl -L https://github.com/protocolbuffers/protobuf/releases/download/v24.4/protoc-24.4-linux-$(uname -m).zip > /tmp/protoc.zip \
|
||||||
&& curl -L https://github.com/protocolbuffers/protobuf/releases/download/v24.4/protoc-24.4-linux-$(uname -m).zip > /tmp/protoc.zip \
|
|
||||||
&& unzip /tmp/protoc.zip -d /usr/local \
|
&& unzip /tmp/protoc.zip -d /usr/local \
|
||||||
&& rm /tmp/protoc.zip
|
&& rm /tmp/protoc.zip
|
||||||
- name: Build Arm Manylinux Wheel
|
- name: Build Arm Manylinux Wheel
|
||||||
@@ -52,12 +51,7 @@ runs:
|
|||||||
args: ${{ inputs.args }}
|
args: ${{ inputs.args }}
|
||||||
before-script-linux: |
|
before-script-linux: |
|
||||||
set -e
|
set -e
|
||||||
apt install -y unzip
|
yum install -y clang \
|
||||||
if [ $(uname -m) = "x86_64" ]; then
|
&& curl -L https://github.com/protocolbuffers/protobuf/releases/download/v24.4/protoc-24.4-linux-aarch_64.zip > /tmp/protoc.zip \
|
||||||
PROTOC_ARCH="x86_64"
|
|
||||||
else
|
|
||||||
PROTOC_ARCH="aarch_64"
|
|
||||||
fi
|
|
||||||
curl -L https://github.com/protocolbuffers/protobuf/releases/download/v24.4/protoc-24.4-linux-$PROTOC_ARCH.zip > /tmp/protoc.zip \
|
|
||||||
&& unzip /tmp/protoc.zip -d /usr/local \
|
&& unzip /tmp/protoc.zip -d /usr/local \
|
||||||
&& rm /tmp/protoc.zip
|
&& rm /tmp/protoc.zip
|
||||||
|
|||||||
2
.github/workflows/build_mac_wheel/action.yml
vendored
2
.github/workflows/build_mac_wheel/action.yml
vendored
@@ -20,7 +20,7 @@ runs:
|
|||||||
uses: PyO3/maturin-action@v1
|
uses: PyO3/maturin-action@v1
|
||||||
with:
|
with:
|
||||||
command: build
|
command: build
|
||||||
|
# TODO: pass through interpreter
|
||||||
args: ${{ inputs.args }}
|
args: ${{ inputs.args }}
|
||||||
docker-options: "-e PIP_EXTRA_INDEX_URL=https://pypi.fury.io/lancedb/"
|
docker-options: "-e PIP_EXTRA_INDEX_URL=https://pypi.fury.io/lancedb/"
|
||||||
working-directory: python
|
working-directory: python
|
||||||
interpreter: 3.${{ inputs.python-minor-version }}
|
|
||||||
|
|||||||
@@ -28,7 +28,7 @@ runs:
|
|||||||
args: ${{ inputs.args }}
|
args: ${{ inputs.args }}
|
||||||
docker-options: "-e PIP_EXTRA_INDEX_URL=https://pypi.fury.io/lancedb/"
|
docker-options: "-e PIP_EXTRA_INDEX_URL=https://pypi.fury.io/lancedb/"
|
||||||
working-directory: python
|
working-directory: python
|
||||||
- uses: actions/upload-artifact@v3
|
- uses: actions/upload-artifact@v4
|
||||||
with:
|
with:
|
||||||
name: windows-wheels
|
name: windows-wheels
|
||||||
path: python\target\wheels
|
path: python\target\wheels
|
||||||
|
|||||||
10
.github/workflows/cargo-publish.yml
vendored
10
.github/workflows/cargo-publish.yml
vendored
@@ -5,8 +5,8 @@ on:
|
|||||||
tags-ignore:
|
tags-ignore:
|
||||||
# We don't publish pre-releases for Rust. Crates.io is just a source
|
# We don't publish pre-releases for Rust. Crates.io is just a source
|
||||||
# distribution, so we don't need to publish pre-releases.
|
# distribution, so we don't need to publish pre-releases.
|
||||||
- 'v*-beta*'
|
- "v*-beta*"
|
||||||
- '*-v*' # for example, python-vX.Y.Z
|
- "*-v*" # for example, python-vX.Y.Z
|
||||||
|
|
||||||
env:
|
env:
|
||||||
# This env var is used by Swatinem/rust-cache@v2 for the cache
|
# This env var is used by Swatinem/rust-cache@v2 for the cache
|
||||||
@@ -19,6 +19,8 @@ env:
|
|||||||
jobs:
|
jobs:
|
||||||
build:
|
build:
|
||||||
runs-on: ubuntu-22.04
|
runs-on: ubuntu-22.04
|
||||||
|
permissions:
|
||||||
|
id-token: write
|
||||||
timeout-minutes: 30
|
timeout-minutes: 30
|
||||||
# Only runs on tags that matches the make-release action
|
# Only runs on tags that matches the make-release action
|
||||||
if: startsWith(github.ref, 'refs/tags/v')
|
if: startsWith(github.ref, 'refs/tags/v')
|
||||||
@@ -31,6 +33,8 @@ jobs:
|
|||||||
run: |
|
run: |
|
||||||
sudo apt update
|
sudo apt update
|
||||||
sudo apt install -y protobuf-compiler libssl-dev
|
sudo apt install -y protobuf-compiler libssl-dev
|
||||||
|
- uses: rust-lang/crates-io-auth-action@v1
|
||||||
|
id: auth
|
||||||
- name: Publish the package
|
- name: Publish the package
|
||||||
run: |
|
run: |
|
||||||
cargo publish -p lancedb --all-features --token ${{ secrets.CARGO_REGISTRY_TOKEN }}
|
cargo publish -p lancedb --all-features --token ${{ steps.auth.outputs.token }}
|
||||||
|
|||||||
32
.github/workflows/docs.yml
vendored
32
.github/workflows/docs.yml
vendored
@@ -18,17 +18,24 @@ concurrency:
|
|||||||
group: "pages"
|
group: "pages"
|
||||||
cancel-in-progress: true
|
cancel-in-progress: true
|
||||||
|
|
||||||
|
env:
|
||||||
|
# This reduces the disk space needed for the build
|
||||||
|
RUSTFLAGS: "-C debuginfo=0"
|
||||||
|
# according to: https://matklad.github.io/2021/09/04/fast-rust-builds.html
|
||||||
|
# CI builds are faster with incremental disabled.
|
||||||
|
CARGO_INCREMENTAL: "0"
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
# Single deploy job since we're just deploying
|
# Single deploy job since we're just deploying
|
||||||
build:
|
build:
|
||||||
environment:
|
environment:
|
||||||
name: github-pages
|
name: github-pages
|
||||||
url: ${{ steps.deployment.outputs.page_url }}
|
url: ${{ steps.deployment.outputs.page_url }}
|
||||||
runs-on: buildjet-8vcpu-ubuntu-2204
|
runs-on: ubuntu-24.04
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
- name: Install dependecies needed for ubuntu
|
- name: Install dependencies needed for ubuntu
|
||||||
run: |
|
run: |
|
||||||
sudo apt install -y protobuf-compiler libssl-dev
|
sudo apt install -y protobuf-compiler libssl-dev
|
||||||
rustup update && rustup default
|
rustup update && rustup default
|
||||||
@@ -38,33 +45,22 @@ jobs:
|
|||||||
python-version: "3.10"
|
python-version: "3.10"
|
||||||
cache: "pip"
|
cache: "pip"
|
||||||
cache-dependency-path: "docs/requirements.txt"
|
cache-dependency-path: "docs/requirements.txt"
|
||||||
|
- uses: Swatinem/rust-cache@v2
|
||||||
- name: Build Python
|
- name: Build Python
|
||||||
working-directory: python
|
working-directory: python
|
||||||
run: |
|
run: |
|
||||||
python -m pip install -e .
|
python -m pip install --extra-index-url https://pypi.fury.io/lancedb/ -e .
|
||||||
python -m pip install -r ../docs/requirements.txt
|
python -m pip install --extra-index-url https://pypi.fury.io/lancedb/ -r ../docs/requirements.txt
|
||||||
- name: Set up node
|
- name: Set up node
|
||||||
uses: actions/setup-node@v3
|
uses: actions/setup-node@v3
|
||||||
with:
|
with:
|
||||||
node-version: 20
|
node-version: 20
|
||||||
cache: 'npm'
|
cache: 'npm'
|
||||||
cache-dependency-path: node/package-lock.json
|
|
||||||
- uses: Swatinem/rust-cache@v2
|
|
||||||
- name: Install node dependencies
|
- name: Install node dependencies
|
||||||
working-directory: node
|
working-directory: node
|
||||||
run: |
|
run: |
|
||||||
sudo apt update
|
sudo apt update
|
||||||
sudo apt install -y protobuf-compiler libssl-dev
|
sudo apt install -y protobuf-compiler libssl-dev
|
||||||
- name: Build node
|
|
||||||
working-directory: node
|
|
||||||
run: |
|
|
||||||
npm ci
|
|
||||||
npm run build
|
|
||||||
npm run tsc
|
|
||||||
- name: Create markdown files
|
|
||||||
working-directory: node
|
|
||||||
run: |
|
|
||||||
npx typedoc --plugin typedoc-plugin-markdown --out ../docs/src/javascript src/index.ts
|
|
||||||
- name: Build docs
|
- name: Build docs
|
||||||
working-directory: docs
|
working-directory: docs
|
||||||
run: |
|
run: |
|
||||||
@@ -72,9 +68,9 @@ jobs:
|
|||||||
- name: Setup Pages
|
- name: Setup Pages
|
||||||
uses: actions/configure-pages@v2
|
uses: actions/configure-pages@v2
|
||||||
- name: Upload artifact
|
- name: Upload artifact
|
||||||
uses: actions/upload-pages-artifact@v1
|
uses: actions/upload-pages-artifact@v3
|
||||||
with:
|
with:
|
||||||
path: "docs/site"
|
path: "docs/site"
|
||||||
- name: Deploy to GitHub Pages
|
- name: Deploy to GitHub Pages
|
||||||
id: deployment
|
id: deployment
|
||||||
uses: actions/deploy-pages@v1
|
uses: actions/deploy-pages@v4
|
||||||
|
|||||||
54
.github/workflows/docs_test.yml
vendored
54
.github/workflows/docs_test.yml
vendored
@@ -24,15 +24,19 @@ env:
|
|||||||
jobs:
|
jobs:
|
||||||
test-python:
|
test-python:
|
||||||
name: Test doc python code
|
name: Test doc python code
|
||||||
runs-on: "warp-ubuntu-latest-x64-4x"
|
runs-on: ubuntu-24.04
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
- name: Print CPU capabilities
|
- name: Print CPU capabilities
|
||||||
run: cat /proc/cpuinfo
|
run: cat /proc/cpuinfo
|
||||||
|
- name: Install protobuf
|
||||||
|
run: |
|
||||||
|
sudo apt update
|
||||||
|
sudo apt install -y protobuf-compiler
|
||||||
- name: Install dependecies needed for ubuntu
|
- name: Install dependecies needed for ubuntu
|
||||||
run: |
|
run: |
|
||||||
sudo apt install -y protobuf-compiler libssl-dev
|
sudo apt install -y libssl-dev
|
||||||
rustup update && rustup default
|
rustup update && rustup default
|
||||||
- name: Set up Python
|
- name: Set up Python
|
||||||
uses: actions/setup-python@v5
|
uses: actions/setup-python@v5
|
||||||
@@ -45,7 +49,7 @@ jobs:
|
|||||||
- name: Build Python
|
- name: Build Python
|
||||||
working-directory: docs/test
|
working-directory: docs/test
|
||||||
run:
|
run:
|
||||||
python -m pip install -r requirements.txt
|
python -m pip install --extra-index-url https://pypi.fury.io/lancedb/ -r requirements.txt
|
||||||
- name: Create test files
|
- name: Create test files
|
||||||
run: |
|
run: |
|
||||||
cd docs/test
|
cd docs/test
|
||||||
@@ -54,47 +58,3 @@ jobs:
|
|||||||
run: |
|
run: |
|
||||||
cd docs/test/python
|
cd docs/test/python
|
||||||
for d in *; do cd "$d"; echo "$d".py; python "$d".py; cd ..; done
|
for d in *; do cd "$d"; echo "$d".py; python "$d".py; cd ..; done
|
||||||
test-node:
|
|
||||||
name: Test doc nodejs code
|
|
||||||
runs-on: "warp-ubuntu-latest-x64-4x"
|
|
||||||
timeout-minutes: 60
|
|
||||||
strategy:
|
|
||||||
fail-fast: false
|
|
||||||
steps:
|
|
||||||
- name: Checkout
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
with:
|
|
||||||
fetch-depth: 0
|
|
||||||
lfs: true
|
|
||||||
- name: Print CPU capabilities
|
|
||||||
run: cat /proc/cpuinfo
|
|
||||||
- name: Set up Node
|
|
||||||
uses: actions/setup-node@v4
|
|
||||||
with:
|
|
||||||
node-version: 20
|
|
||||||
- name: Install dependecies needed for ubuntu
|
|
||||||
run: |
|
|
||||||
sudo apt install -y protobuf-compiler libssl-dev
|
|
||||||
rustup update && rustup default
|
|
||||||
- name: Rust cache
|
|
||||||
uses: swatinem/rust-cache@v2
|
|
||||||
- name: Install node dependencies
|
|
||||||
run: |
|
|
||||||
sudo swapoff -a
|
|
||||||
sudo fallocate -l 8G /swapfile
|
|
||||||
sudo chmod 600 /swapfile
|
|
||||||
sudo mkswap /swapfile
|
|
||||||
sudo swapon /swapfile
|
|
||||||
sudo swapon --show
|
|
||||||
cd node
|
|
||||||
npm ci
|
|
||||||
npm run build-release
|
|
||||||
cd ../docs
|
|
||||||
npm install
|
|
||||||
- name: Test
|
|
||||||
env:
|
|
||||||
LANCEDB_URI: ${{ secrets.LANCEDB_URI }}
|
|
||||||
LANCEDB_DEV_API_KEY: ${{ secrets.LANCEDB_DEV_API_KEY }}
|
|
||||||
run: |
|
|
||||||
cd docs
|
|
||||||
npm t
|
|
||||||
|
|||||||
114
.github/workflows/java-publish.yml
vendored
Normal file
114
.github/workflows/java-publish.yml
vendored
Normal file
@@ -0,0 +1,114 @@
|
|||||||
|
name: Build and publish Java packages
|
||||||
|
on:
|
||||||
|
release:
|
||||||
|
types: [released]
|
||||||
|
pull_request:
|
||||||
|
paths:
|
||||||
|
- .github/workflows/java-publish.yml
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
macos-arm64:
|
||||||
|
name: Build on MacOS Arm64
|
||||||
|
runs-on: macos-14
|
||||||
|
timeout-minutes: 45
|
||||||
|
defaults:
|
||||||
|
run:
|
||||||
|
working-directory: ./java/core/lancedb-jni
|
||||||
|
steps:
|
||||||
|
- name: Checkout repository
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
- uses: Swatinem/rust-cache@v2
|
||||||
|
- name: Install dependencies
|
||||||
|
run: |
|
||||||
|
brew install protobuf
|
||||||
|
- name: Build release
|
||||||
|
run: |
|
||||||
|
cargo build --release
|
||||||
|
- uses: actions/upload-artifact@v4
|
||||||
|
with:
|
||||||
|
name: liblancedb_jni_darwin_aarch64.zip
|
||||||
|
path: target/release/liblancedb_jni.dylib
|
||||||
|
retention-days: 1
|
||||||
|
if-no-files-found: error
|
||||||
|
linux-arm64:
|
||||||
|
name: Build on Linux Arm64
|
||||||
|
runs-on: warp-ubuntu-2204-arm64-8x
|
||||||
|
timeout-minutes: 45
|
||||||
|
defaults:
|
||||||
|
run:
|
||||||
|
working-directory: ./java/core/lancedb-jni
|
||||||
|
steps:
|
||||||
|
- name: Checkout repository
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
- uses: Swatinem/rust-cache@v2
|
||||||
|
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
||||||
|
with:
|
||||||
|
toolchain: "1.81.0"
|
||||||
|
cache-workspaces: "./java/core/lancedb-jni"
|
||||||
|
# Disable full debug symbol generation to speed up CI build and keep memory down
|
||||||
|
# "1" means line tables only, which is useful for panic tracebacks.
|
||||||
|
rustflags: "-C debuginfo=1"
|
||||||
|
- name: Install dependencies
|
||||||
|
run: |
|
||||||
|
sudo apt -y -qq update
|
||||||
|
sudo apt install -y protobuf-compiler libssl-dev pkg-config
|
||||||
|
- name: Build release
|
||||||
|
run: |
|
||||||
|
cargo build --release
|
||||||
|
- uses: actions/upload-artifact@v4
|
||||||
|
with:
|
||||||
|
name: liblancedb_jni_linux_aarch64.zip
|
||||||
|
path: target/release/liblancedb_jni.so
|
||||||
|
retention-days: 1
|
||||||
|
if-no-files-found: error
|
||||||
|
linux-x86:
|
||||||
|
runs-on: warp-ubuntu-2204-x64-8x
|
||||||
|
timeout-minutes: 30
|
||||||
|
needs: [macos-arm64, linux-arm64]
|
||||||
|
defaults:
|
||||||
|
run:
|
||||||
|
working-directory: ./java
|
||||||
|
steps:
|
||||||
|
- name: Checkout repository
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
- uses: Swatinem/rust-cache@v2
|
||||||
|
- name: Set up Java 8
|
||||||
|
uses: actions/setup-java@v4
|
||||||
|
with:
|
||||||
|
distribution: temurin
|
||||||
|
java-version: 8
|
||||||
|
cache: "maven"
|
||||||
|
server-id: ossrh
|
||||||
|
server-username: SONATYPE_USER
|
||||||
|
server-password: SONATYPE_TOKEN
|
||||||
|
gpg-private-key: ${{ secrets.GPG_PRIVATE_KEY }}
|
||||||
|
gpg-passphrase: ${{ secrets.GPG_PASSPHRASE }}
|
||||||
|
- name: Install dependencies
|
||||||
|
run: |
|
||||||
|
sudo apt -y -qq update
|
||||||
|
sudo apt install -y protobuf-compiler libssl-dev pkg-config
|
||||||
|
- name: Download artifact
|
||||||
|
uses: actions/download-artifact@v4
|
||||||
|
- name: Copy native libs
|
||||||
|
run: |
|
||||||
|
mkdir -p ./core/target/classes/nativelib/darwin-aarch64 ./core/target/classes/nativelib/linux-aarch64
|
||||||
|
cp ../liblancedb_jni_darwin_aarch64.zip/liblancedb_jni.dylib ./core/target/classes/nativelib/darwin-aarch64/liblancedb_jni.dylib
|
||||||
|
cp ../liblancedb_jni_linux_aarch64.zip/liblancedb_jni.so ./core/target/classes/nativelib/linux-aarch64/liblancedb_jni.so
|
||||||
|
- name: Dry run
|
||||||
|
if: github.event_name == 'pull_request'
|
||||||
|
run: |
|
||||||
|
mvn --batch-mode -DskipTests -Drust.release.build=true package
|
||||||
|
- name: Set github
|
||||||
|
run: |
|
||||||
|
git config --global user.email "LanceDB Github Runner"
|
||||||
|
git config --global user.name "dev+gha@lancedb.com"
|
||||||
|
- name: Publish with Java 8
|
||||||
|
if: github.event_name == 'release'
|
||||||
|
run: |
|
||||||
|
echo "use-agent" >> ~/.gnupg/gpg.conf
|
||||||
|
echo "pinentry-mode loopback" >> ~/.gnupg/gpg.conf
|
||||||
|
export GPG_TTY=$(tty)
|
||||||
|
mvn --batch-mode -DskipTests -Drust.release.build=true -DpushChanges=false -Dgpg.passphrase=${{ secrets.GPG_PASSPHRASE }} deploy -P deploy-to-ossrh
|
||||||
|
env:
|
||||||
|
SONATYPE_USER: ${{ secrets.SONATYPE_USER }}
|
||||||
|
SONATYPE_TOKEN: ${{ secrets.SONATYPE_TOKEN }}
|
||||||
7
.github/workflows/java.yml
vendored
7
.github/workflows/java.yml
vendored
@@ -35,6 +35,9 @@ jobs:
|
|||||||
- uses: Swatinem/rust-cache@v2
|
- uses: Swatinem/rust-cache@v2
|
||||||
with:
|
with:
|
||||||
workspaces: java/core/lancedb-jni
|
workspaces: java/core/lancedb-jni
|
||||||
|
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
||||||
|
with:
|
||||||
|
components: rustfmt
|
||||||
- name: Run cargo fmt
|
- name: Run cargo fmt
|
||||||
run: cargo fmt --check
|
run: cargo fmt --check
|
||||||
working-directory: ./java/core/lancedb-jni
|
working-directory: ./java/core/lancedb-jni
|
||||||
@@ -68,6 +71,9 @@ jobs:
|
|||||||
- uses: Swatinem/rust-cache@v2
|
- uses: Swatinem/rust-cache@v2
|
||||||
with:
|
with:
|
||||||
workspaces: java/core/lancedb-jni
|
workspaces: java/core/lancedb-jni
|
||||||
|
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
||||||
|
with:
|
||||||
|
components: rustfmt
|
||||||
- name: Run cargo fmt
|
- name: Run cargo fmt
|
||||||
run: cargo fmt --check
|
run: cargo fmt --check
|
||||||
working-directory: ./java/core/lancedb-jni
|
working-directory: ./java/core/lancedb-jni
|
||||||
@@ -110,4 +116,3 @@ jobs:
|
|||||||
-Djdk.reflect.useDirectMethodHandle=false \
|
-Djdk.reflect.useDirectMethodHandle=false \
|
||||||
-Dio.netty.tryReflectionSetAccessible=true"
|
-Dio.netty.tryReflectionSetAccessible=true"
|
||||||
JAVA_HOME=$JAVA_17 mvn clean test
|
JAVA_HOME=$JAVA_17 mvn clean test
|
||||||
|
|
||||||
|
|||||||
31
.github/workflows/license-header-check.yml
vendored
Normal file
31
.github/workflows/license-header-check.yml
vendored
Normal file
@@ -0,0 +1,31 @@
|
|||||||
|
name: Check license headers
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
branches:
|
||||||
|
- main
|
||||||
|
pull_request:
|
||||||
|
paths:
|
||||||
|
- rust/**
|
||||||
|
- python/**
|
||||||
|
- nodejs/**
|
||||||
|
- java/**
|
||||||
|
- .github/workflows/license-header-check.yml
|
||||||
|
jobs:
|
||||||
|
check-licenses:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: Check out code
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
- name: Install license-header-checker
|
||||||
|
working-directory: /tmp
|
||||||
|
run: |
|
||||||
|
curl -s https://raw.githubusercontent.com/lluissm/license-header-checker/master/install.sh | bash
|
||||||
|
mv /tmp/bin/license-header-checker /usr/local/bin/
|
||||||
|
- name: Check license headers (rust)
|
||||||
|
run: license-header-checker -a -v ./rust/license_header.txt ./ rs && [[ -z `git status -s` ]]
|
||||||
|
- name: Check license headers (python)
|
||||||
|
run: license-header-checker -a -v ./python/license_header.txt python py && [[ -z `git status -s` ]]
|
||||||
|
- name: Check license headers (typescript)
|
||||||
|
run: license-header-checker -a -v ./nodejs/license_header.txt nodejs ts && [[ -z `git status -s` ]]
|
||||||
|
- name: Check license headers (java)
|
||||||
|
run: license-header-checker -a -v ./nodejs/license_header.txt java java && [[ -z `git status -s` ]]
|
||||||
16
.github/workflows/make-release-commit.yml
vendored
16
.github/workflows/make-release-commit.yml
vendored
@@ -30,7 +30,7 @@ on:
|
|||||||
default: true
|
default: true
|
||||||
type: boolean
|
type: boolean
|
||||||
other:
|
other:
|
||||||
description: 'Make a Node/Rust release'
|
description: 'Make a Node/Rust/Java release'
|
||||||
required: true
|
required: true
|
||||||
default: true
|
default: true
|
||||||
type: boolean
|
type: boolean
|
||||||
@@ -43,7 +43,7 @@ on:
|
|||||||
jobs:
|
jobs:
|
||||||
make-release:
|
make-release:
|
||||||
# Creates tag and GH release. The GH release will trigger the build and release jobs.
|
# Creates tag and GH release. The GH release will trigger the build and release jobs.
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-24.04
|
||||||
permissions:
|
permissions:
|
||||||
contents: write
|
contents: write
|
||||||
steps:
|
steps:
|
||||||
@@ -57,15 +57,14 @@ jobs:
|
|||||||
# trigger any workflows watching for new tags. See:
|
# trigger any workflows watching for new tags. See:
|
||||||
# https://docs.github.com/en/actions/using-workflows/triggering-a-workflow#triggering-a-workflow-from-a-workflow
|
# https://docs.github.com/en/actions/using-workflows/triggering-a-workflow#triggering-a-workflow-from-a-workflow
|
||||||
token: ${{ secrets.LANCEDB_RELEASE_TOKEN }}
|
token: ${{ secrets.LANCEDB_RELEASE_TOKEN }}
|
||||||
|
- name: Validate Lance dependency is at stable version
|
||||||
|
if: ${{ inputs.type == 'stable' }}
|
||||||
|
run: python ci/validate_stable_lance.py
|
||||||
- name: Set git configs for bumpversion
|
- name: Set git configs for bumpversion
|
||||||
shell: bash
|
shell: bash
|
||||||
run: |
|
run: |
|
||||||
git config user.name 'Lance Release'
|
git config user.name 'Lance Release'
|
||||||
git config user.email 'lance-dev@lancedb.com'
|
git config user.email 'lance-dev@lancedb.com'
|
||||||
- name: Set up Python 3.11
|
|
||||||
uses: actions/setup-python@v5
|
|
||||||
with:
|
|
||||||
python-version: "3.11"
|
|
||||||
- name: Bump Python version
|
- name: Bump Python version
|
||||||
if: ${{ inputs.python }}
|
if: ${{ inputs.python }}
|
||||||
working-directory: python
|
working-directory: python
|
||||||
@@ -85,6 +84,7 @@ jobs:
|
|||||||
run: |
|
run: |
|
||||||
pip install bump-my-version PyGithub packaging
|
pip install bump-my-version PyGithub packaging
|
||||||
bash ci/bump_version.sh ${{ inputs.type }} ${{ inputs.bump-minor }} v $COMMIT_BEFORE_BUMP
|
bash ci/bump_version.sh ${{ inputs.type }} ${{ inputs.bump-minor }} v $COMMIT_BEFORE_BUMP
|
||||||
|
bash ci/update_lockfiles.sh --amend
|
||||||
- name: Push new version tag
|
- name: Push new version tag
|
||||||
if: ${{ !inputs.dry_run }}
|
if: ${{ !inputs.dry_run }}
|
||||||
uses: ad-m/github-push-action@master
|
uses: ad-m/github-push-action@master
|
||||||
@@ -93,7 +93,3 @@ jobs:
|
|||||||
github_token: ${{ secrets.LANCEDB_RELEASE_TOKEN }}
|
github_token: ${{ secrets.LANCEDB_RELEASE_TOKEN }}
|
||||||
branch: ${{ github.ref }}
|
branch: ${{ github.ref }}
|
||||||
tags: true
|
tags: true
|
||||||
- uses: ./.github/workflows/update_package_lock
|
|
||||||
if: ${{ !inputs.dry_run && inputs.other }}
|
|
||||||
with:
|
|
||||||
github_token: ${{ secrets.GITHUB_TOKEN }}
|
|
||||||
|
|||||||
147
.github/workflows/node.yml
vendored
147
.github/workflows/node.yml
vendored
@@ -1,147 +0,0 @@
|
|||||||
name: Node
|
|
||||||
|
|
||||||
on:
|
|
||||||
push:
|
|
||||||
branches:
|
|
||||||
- main
|
|
||||||
pull_request:
|
|
||||||
paths:
|
|
||||||
- node/**
|
|
||||||
- rust/ffi/node/**
|
|
||||||
- .github/workflows/node.yml
|
|
||||||
- docker-compose.yml
|
|
||||||
|
|
||||||
concurrency:
|
|
||||||
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
|
|
||||||
cancel-in-progress: true
|
|
||||||
|
|
||||||
env:
|
|
||||||
# Disable full debug symbol generation to speed up CI build and keep memory down
|
|
||||||
# "1" means line tables only, which is useful for panic tracebacks.
|
|
||||||
#
|
|
||||||
# Use native CPU to accelerate tests if possible, especially for f16
|
|
||||||
# target-cpu=haswell fixes failing ci build
|
|
||||||
RUSTFLAGS: "-C debuginfo=1 -C target-cpu=haswell -C target-feature=+f16c,+avx2,+fma"
|
|
||||||
RUST_BACKTRACE: "1"
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
linux:
|
|
||||||
name: Linux (Node ${{ matrix.node-version }})
|
|
||||||
timeout-minutes: 30
|
|
||||||
strategy:
|
|
||||||
matrix:
|
|
||||||
node-version: [ "18", "20" ]
|
|
||||||
runs-on: "ubuntu-22.04"
|
|
||||||
defaults:
|
|
||||||
run:
|
|
||||||
shell: bash
|
|
||||||
working-directory: node
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@v4
|
|
||||||
with:
|
|
||||||
fetch-depth: 0
|
|
||||||
lfs: true
|
|
||||||
- uses: actions/setup-node@v3
|
|
||||||
with:
|
|
||||||
node-version: ${{ matrix.node-version }}
|
|
||||||
cache: 'npm'
|
|
||||||
cache-dependency-path: node/package-lock.json
|
|
||||||
- uses: Swatinem/rust-cache@v2
|
|
||||||
- name: Install dependencies
|
|
||||||
run: |
|
|
||||||
sudo apt update
|
|
||||||
sudo apt install -y protobuf-compiler libssl-dev
|
|
||||||
- name: Build
|
|
||||||
run: |
|
|
||||||
npm ci
|
|
||||||
npm run build
|
|
||||||
npm run pack-build
|
|
||||||
npm install --no-save ./dist/lancedb-vectordb-*.tgz
|
|
||||||
# Remove index.node to test with dependency installed
|
|
||||||
rm index.node
|
|
||||||
- name: Test
|
|
||||||
run: npm run test
|
|
||||||
macos:
|
|
||||||
timeout-minutes: 30
|
|
||||||
runs-on: "macos-13"
|
|
||||||
defaults:
|
|
||||||
run:
|
|
||||||
shell: bash
|
|
||||||
working-directory: node
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@v4
|
|
||||||
with:
|
|
||||||
fetch-depth: 0
|
|
||||||
lfs: true
|
|
||||||
- uses: actions/setup-node@v3
|
|
||||||
with:
|
|
||||||
node-version: 20
|
|
||||||
cache: 'npm'
|
|
||||||
cache-dependency-path: node/package-lock.json
|
|
||||||
- uses: Swatinem/rust-cache@v2
|
|
||||||
- name: Install dependencies
|
|
||||||
run: brew install protobuf
|
|
||||||
- name: Build
|
|
||||||
run: |
|
|
||||||
npm ci
|
|
||||||
npm run build
|
|
||||||
npm run pack-build
|
|
||||||
npm install --no-save ./dist/lancedb-vectordb-*.tgz
|
|
||||||
# Remove index.node to test with dependency installed
|
|
||||||
rm index.node
|
|
||||||
- name: Test
|
|
||||||
run: |
|
|
||||||
npm run test
|
|
||||||
aws-integtest:
|
|
||||||
timeout-minutes: 45
|
|
||||||
runs-on: "ubuntu-22.04"
|
|
||||||
defaults:
|
|
||||||
run:
|
|
||||||
shell: bash
|
|
||||||
working-directory: node
|
|
||||||
env:
|
|
||||||
AWS_ACCESS_KEY_ID: ACCESSKEY
|
|
||||||
AWS_SECRET_ACCESS_KEY: SECRETKEY
|
|
||||||
AWS_DEFAULT_REGION: us-west-2
|
|
||||||
# this one is for s3
|
|
||||||
AWS_ENDPOINT: http://localhost:4566
|
|
||||||
# this one is for dynamodb
|
|
||||||
DYNAMODB_ENDPOINT: http://localhost:4566
|
|
||||||
ALLOW_HTTP: true
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@v4
|
|
||||||
with:
|
|
||||||
fetch-depth: 0
|
|
||||||
lfs: true
|
|
||||||
- uses: actions/setup-node@v3
|
|
||||||
with:
|
|
||||||
node-version: 20
|
|
||||||
cache: 'npm'
|
|
||||||
cache-dependency-path: node/package-lock.json
|
|
||||||
- name: start local stack
|
|
||||||
run: docker compose -f ../docker-compose.yml up -d --wait
|
|
||||||
- name: create s3
|
|
||||||
run: aws s3 mb s3://lancedb-integtest --endpoint $AWS_ENDPOINT
|
|
||||||
- name: create ddb
|
|
||||||
run: |
|
|
||||||
aws dynamodb create-table \
|
|
||||||
--table-name lancedb-integtest \
|
|
||||||
--attribute-definitions '[{"AttributeName": "base_uri", "AttributeType": "S"}, {"AttributeName": "version", "AttributeType": "N"}]' \
|
|
||||||
--key-schema '[{"AttributeName": "base_uri", "KeyType": "HASH"}, {"AttributeName": "version", "KeyType": "RANGE"}]' \
|
|
||||||
--provisioned-throughput '{"ReadCapacityUnits": 10, "WriteCapacityUnits": 10}' \
|
|
||||||
--endpoint-url $DYNAMODB_ENDPOINT
|
|
||||||
- uses: Swatinem/rust-cache@v2
|
|
||||||
- name: Install dependencies
|
|
||||||
run: |
|
|
||||||
sudo apt update
|
|
||||||
sudo apt install -y protobuf-compiler libssl-dev
|
|
||||||
- name: Build
|
|
||||||
run: |
|
|
||||||
npm ci
|
|
||||||
npm run build
|
|
||||||
npm run pack-build
|
|
||||||
npm install --no-save ./dist/lancedb-vectordb-*.tgz
|
|
||||||
# Remove index.node to test with dependency installed
|
|
||||||
rm index.node
|
|
||||||
- name: Test
|
|
||||||
run: npm run integration-test
|
|
||||||
34
.github/workflows/nodejs.yml
vendored
34
.github/workflows/nodejs.yml
vendored
@@ -47,12 +47,18 @@ jobs:
|
|||||||
run: |
|
run: |
|
||||||
sudo apt update
|
sudo apt update
|
||||||
sudo apt install -y protobuf-compiler libssl-dev
|
sudo apt install -y protobuf-compiler libssl-dev
|
||||||
|
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
||||||
|
with:
|
||||||
|
components: rustfmt, clippy
|
||||||
- name: Lint
|
- name: Lint
|
||||||
run: |
|
run: |
|
||||||
cargo fmt --all -- --check
|
cargo fmt --all -- --check
|
||||||
cargo clippy --all --all-features -- -D warnings
|
cargo clippy --all --all-features -- -D warnings
|
||||||
npm ci
|
npm ci
|
||||||
npm run lint-ci
|
npm run lint-ci
|
||||||
|
- name: Lint examples
|
||||||
|
working-directory: nodejs/examples
|
||||||
|
run: npm ci && npm run lint-ci
|
||||||
linux:
|
linux:
|
||||||
name: Linux (NodeJS ${{ matrix.node-version }})
|
name: Linux (NodeJS ${{ matrix.node-version }})
|
||||||
timeout-minutes: 30
|
timeout-minutes: 30
|
||||||
@@ -73,7 +79,7 @@ jobs:
|
|||||||
with:
|
with:
|
||||||
node-version: ${{ matrix.node-version }}
|
node-version: ${{ matrix.node-version }}
|
||||||
cache: 'npm'
|
cache: 'npm'
|
||||||
cache-dependency-path: node/package-lock.json
|
cache-dependency-path: nodejs/package-lock.json
|
||||||
- uses: Swatinem/rust-cache@v2
|
- uses: Swatinem/rust-cache@v2
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: |
|
run: |
|
||||||
@@ -91,6 +97,30 @@ jobs:
|
|||||||
env:
|
env:
|
||||||
S3_TEST: "1"
|
S3_TEST: "1"
|
||||||
run: npm run test
|
run: npm run test
|
||||||
|
- name: Setup examples
|
||||||
|
working-directory: nodejs/examples
|
||||||
|
run: npm ci
|
||||||
|
- name: Test examples
|
||||||
|
working-directory: ./
|
||||||
|
env:
|
||||||
|
OPENAI_API_KEY: test
|
||||||
|
OPENAI_BASE_URL: http://0.0.0.0:8000
|
||||||
|
run: |
|
||||||
|
python ci/mock_openai.py &
|
||||||
|
cd nodejs/examples
|
||||||
|
npm test
|
||||||
|
- name: Check docs
|
||||||
|
run: |
|
||||||
|
# We run this as part of the job because the binary needs to be built
|
||||||
|
# first to export the types of the native code.
|
||||||
|
set -e
|
||||||
|
npm ci
|
||||||
|
npm run docs
|
||||||
|
if ! git diff --exit-code -- . ':(exclude)Cargo.lock'; then
|
||||||
|
echo "Docs need to be updated"
|
||||||
|
echo "Run 'npm run docs', fix any warnings, and commit the changes."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
macos:
|
macos:
|
||||||
timeout-minutes: 30
|
timeout-minutes: 30
|
||||||
runs-on: "macos-14"
|
runs-on: "macos-14"
|
||||||
@@ -107,7 +137,7 @@ jobs:
|
|||||||
with:
|
with:
|
||||||
node-version: 20
|
node-version: 20
|
||||||
cache: 'npm'
|
cache: 'npm'
|
||||||
cache-dependency-path: node/package-lock.json
|
cache-dependency-path: nodejs/package-lock.json
|
||||||
- uses: Swatinem/rust-cache@v2
|
- uses: Swatinem/rust-cache@v2
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: |
|
run: |
|
||||||
|
|||||||
681
.github/workflows/npm-publish.yml
vendored
681
.github/workflows/npm-publish.yml
vendored
@@ -1,399 +1,32 @@
|
|||||||
name: NPM Publish
|
name: NPM Publish
|
||||||
|
|
||||||
|
env:
|
||||||
|
MACOSX_DEPLOYMENT_TARGET: '10.13'
|
||||||
|
CARGO_INCREMENTAL: '0'
|
||||||
|
|
||||||
|
permissions:
|
||||||
|
contents: write
|
||||||
|
id-token: write
|
||||||
|
|
||||||
on:
|
on:
|
||||||
push:
|
push:
|
||||||
|
branches:
|
||||||
|
- main
|
||||||
tags:
|
tags:
|
||||||
- "v*"
|
- "v*"
|
||||||
|
pull_request:
|
||||||
|
# This should trigger a dry run (we skip the final publish step)
|
||||||
|
paths:
|
||||||
|
- .github/workflows/npm-publish.yml
|
||||||
|
- Cargo.toml # Change in dependency frequently breaks builds
|
||||||
|
|
||||||
|
concurrency:
|
||||||
|
group: ${{ github.workflow }}-${{ github.ref }}
|
||||||
|
cancel-in-progress: true
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
node:
|
|
||||||
name: vectordb Typescript
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
# Only runs on tags that matches the make-release action
|
|
||||||
if: startsWith(github.ref, 'refs/tags/v')
|
|
||||||
defaults:
|
|
||||||
run:
|
|
||||||
shell: bash
|
|
||||||
working-directory: node
|
|
||||||
steps:
|
|
||||||
- name: Checkout
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
- uses: actions/setup-node@v3
|
|
||||||
with:
|
|
||||||
node-version: 20
|
|
||||||
cache: "npm"
|
|
||||||
cache-dependency-path: node/package-lock.json
|
|
||||||
- name: Install dependencies
|
|
||||||
run: |
|
|
||||||
sudo apt update
|
|
||||||
sudo apt install -y protobuf-compiler libssl-dev
|
|
||||||
- name: Build
|
|
||||||
run: |
|
|
||||||
npm ci
|
|
||||||
npm run tsc
|
|
||||||
npm pack
|
|
||||||
- name: Upload Linux Artifacts
|
|
||||||
uses: actions/upload-artifact@v4
|
|
||||||
with:
|
|
||||||
name: node-package
|
|
||||||
path: |
|
|
||||||
node/vectordb-*.tgz
|
|
||||||
|
|
||||||
node-macos:
|
|
||||||
name: vectordb ${{ matrix.config.arch }}
|
|
||||||
strategy:
|
|
||||||
matrix:
|
|
||||||
config:
|
|
||||||
- arch: x86_64-apple-darwin
|
|
||||||
runner: macos-13
|
|
||||||
- arch: aarch64-apple-darwin
|
|
||||||
# xlarge is implicitly arm64.
|
|
||||||
runner: macos-14
|
|
||||||
runs-on: ${{ matrix.config.runner }}
|
|
||||||
# Only runs on tags that matches the make-release action
|
|
||||||
if: startsWith(github.ref, 'refs/tags/v')
|
|
||||||
steps:
|
|
||||||
- name: Checkout
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
- name: Install system dependencies
|
|
||||||
run: brew install protobuf
|
|
||||||
- name: Install npm dependencies
|
|
||||||
run: |
|
|
||||||
cd node
|
|
||||||
npm ci
|
|
||||||
- name: Build MacOS native node modules
|
|
||||||
run: bash ci/build_macos_artifacts.sh ${{ matrix.config.arch }}
|
|
||||||
- name: Upload Darwin Artifacts
|
|
||||||
uses: actions/upload-artifact@v4
|
|
||||||
with:
|
|
||||||
name: node-native-darwin-${{ matrix.config.arch }}
|
|
||||||
path: |
|
|
||||||
node/dist/lancedb-vectordb-darwin*.tgz
|
|
||||||
|
|
||||||
nodejs-macos:
|
|
||||||
name: lancedb ${{ matrix.config.arch }}
|
|
||||||
strategy:
|
|
||||||
matrix:
|
|
||||||
config:
|
|
||||||
- arch: x86_64-apple-darwin
|
|
||||||
runner: macos-13
|
|
||||||
- arch: aarch64-apple-darwin
|
|
||||||
# xlarge is implicitly arm64.
|
|
||||||
runner: macos-14
|
|
||||||
runs-on: ${{ matrix.config.runner }}
|
|
||||||
# Only runs on tags that matches the make-release action
|
|
||||||
if: startsWith(github.ref, 'refs/tags/v')
|
|
||||||
steps:
|
|
||||||
- name: Checkout
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
- name: Install system dependencies
|
|
||||||
run: brew install protobuf
|
|
||||||
- name: Install npm dependencies
|
|
||||||
run: |
|
|
||||||
cd nodejs
|
|
||||||
npm ci
|
|
||||||
- name: Build MacOS native nodejs modules
|
|
||||||
run: bash ci/build_macos_artifacts_nodejs.sh ${{ matrix.config.arch }}
|
|
||||||
- name: Upload Darwin Artifacts
|
|
||||||
uses: actions/upload-artifact@v4
|
|
||||||
with:
|
|
||||||
name: nodejs-native-darwin-${{ matrix.config.arch }}
|
|
||||||
path: |
|
|
||||||
nodejs/dist/*.node
|
|
||||||
|
|
||||||
node-linux:
|
|
||||||
name: vectordb (${{ matrix.config.arch}}-unknown-linux-gnu)
|
|
||||||
runs-on: ${{ matrix.config.runner }}
|
|
||||||
# Only runs on tags that matches the make-release action
|
|
||||||
if: startsWith(github.ref, 'refs/tags/v')
|
|
||||||
strategy:
|
|
||||||
fail-fast: false
|
|
||||||
matrix:
|
|
||||||
config:
|
|
||||||
- arch: x86_64
|
|
||||||
runner: ubuntu-latest
|
|
||||||
- arch: aarch64
|
|
||||||
# For successful fat LTO builds, we need a large runner to avoid OOM errors.
|
|
||||||
runner: warp-ubuntu-latest-arm64-4x
|
|
||||||
steps:
|
|
||||||
- name: Checkout
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
# To avoid OOM errors on ARM, we create a swap file.
|
|
||||||
- name: Configure aarch64 build
|
|
||||||
if: ${{ matrix.config.arch == 'aarch64' }}
|
|
||||||
run: |
|
|
||||||
free -h
|
|
||||||
sudo fallocate -l 16G /swapfile
|
|
||||||
sudo chmod 600 /swapfile
|
|
||||||
sudo mkswap /swapfile
|
|
||||||
sudo swapon /swapfile
|
|
||||||
echo "/swapfile swap swap defaults 0 0" >> sudo /etc/fstab
|
|
||||||
# print info
|
|
||||||
swapon --show
|
|
||||||
free -h
|
|
||||||
- name: Build Linux Artifacts
|
|
||||||
run: |
|
|
||||||
bash ci/build_linux_artifacts.sh ${{ matrix.config.arch }}
|
|
||||||
- name: Upload Linux Artifacts
|
|
||||||
uses: actions/upload-artifact@v4
|
|
||||||
with:
|
|
||||||
name: node-native-linux-${{ matrix.config.arch }}
|
|
||||||
path: |
|
|
||||||
node/dist/lancedb-vectordb-linux*.tgz
|
|
||||||
|
|
||||||
nodejs-linux:
|
|
||||||
name: lancedb (${{ matrix.config.arch}}-unknown-linux-gnu
|
|
||||||
runs-on: ${{ matrix.config.runner }}
|
|
||||||
# Only runs on tags that matches the make-release action
|
|
||||||
if: startsWith(github.ref, 'refs/tags/v')
|
|
||||||
strategy:
|
|
||||||
fail-fast: false
|
|
||||||
matrix:
|
|
||||||
config:
|
|
||||||
- arch: x86_64
|
|
||||||
runner: ubuntu-latest
|
|
||||||
- arch: aarch64
|
|
||||||
# For successful fat LTO builds, we need a large runner to avoid OOM errors.
|
|
||||||
runner: buildjet-16vcpu-ubuntu-2204-arm
|
|
||||||
steps:
|
|
||||||
- name: Checkout
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
# Buildjet aarch64 runners have only 1.5 GB RAM per core, vs 3.5 GB per core for
|
|
||||||
# x86_64 runners. To avoid OOM errors on ARM, we create a swap file.
|
|
||||||
- name: Configure aarch64 build
|
|
||||||
if: ${{ matrix.config.arch == 'aarch64' }}
|
|
||||||
run: |
|
|
||||||
free -h
|
|
||||||
sudo fallocate -l 16G /swapfile
|
|
||||||
sudo chmod 600 /swapfile
|
|
||||||
sudo mkswap /swapfile
|
|
||||||
sudo swapon /swapfile
|
|
||||||
echo "/swapfile swap swap defaults 0 0" >> sudo /etc/fstab
|
|
||||||
# print info
|
|
||||||
swapon --show
|
|
||||||
free -h
|
|
||||||
- name: Build Linux Artifacts
|
|
||||||
run: |
|
|
||||||
bash ci/build_linux_artifacts_nodejs.sh ${{ matrix.config.arch }}
|
|
||||||
- name: Upload Linux Artifacts
|
|
||||||
uses: actions/upload-artifact@v4
|
|
||||||
with:
|
|
||||||
name: nodejs-native-linux-${{ matrix.config.arch }}
|
|
||||||
path: |
|
|
||||||
nodejs/dist/*.node
|
|
||||||
# The generic files are the same in all distros so we just pick
|
|
||||||
# one to do the upload.
|
|
||||||
- name: Upload Generic Artifacts
|
|
||||||
if: ${{ matrix.config.arch == 'x86_64' }}
|
|
||||||
uses: actions/upload-artifact@v4
|
|
||||||
with:
|
|
||||||
name: nodejs-dist
|
|
||||||
path: |
|
|
||||||
nodejs/dist/*
|
|
||||||
!nodejs/dist/*.node
|
|
||||||
|
|
||||||
node-windows:
|
|
||||||
name: vectordb ${{ matrix.target }}
|
|
||||||
runs-on: windows-2022
|
|
||||||
# Only runs on tags that matches the make-release action
|
|
||||||
if: startsWith(github.ref, 'refs/tags/v')
|
|
||||||
strategy:
|
|
||||||
fail-fast: false
|
|
||||||
matrix:
|
|
||||||
target: [x86_64-pc-windows-msvc]
|
|
||||||
steps:
|
|
||||||
- name: Checkout
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
- name: Install Protoc v21.12
|
|
||||||
working-directory: C:\
|
|
||||||
run: |
|
|
||||||
New-Item -Path 'C:\protoc' -ItemType Directory
|
|
||||||
Set-Location C:\protoc
|
|
||||||
Invoke-WebRequest https://github.com/protocolbuffers/protobuf/releases/download/v21.12/protoc-21.12-win64.zip -OutFile C:\protoc\protoc.zip
|
|
||||||
7z x protoc.zip
|
|
||||||
Add-Content $env:GITHUB_PATH "C:\protoc\bin"
|
|
||||||
shell: powershell
|
|
||||||
- name: Install npm dependencies
|
|
||||||
run: |
|
|
||||||
cd node
|
|
||||||
npm ci
|
|
||||||
- name: Build Windows native node modules
|
|
||||||
run: .\ci\build_windows_artifacts.ps1 ${{ matrix.target }}
|
|
||||||
- name: Upload Windows Artifacts
|
|
||||||
uses: actions/upload-artifact@v4
|
|
||||||
with:
|
|
||||||
name: node-native-windows
|
|
||||||
path: |
|
|
||||||
node/dist/lancedb-vectordb-win32*.tgz
|
|
||||||
|
|
||||||
nodejs-windows:
|
|
||||||
name: lancedb ${{ matrix.target }}
|
|
||||||
runs-on: windows-2022
|
|
||||||
# Only runs on tags that matches the make-release action
|
|
||||||
if: startsWith(github.ref, 'refs/tags/v')
|
|
||||||
strategy:
|
|
||||||
fail-fast: false
|
|
||||||
matrix:
|
|
||||||
target: [x86_64-pc-windows-msvc]
|
|
||||||
steps:
|
|
||||||
- name: Checkout
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
- name: Install Protoc v21.12
|
|
||||||
working-directory: C:\
|
|
||||||
run: |
|
|
||||||
New-Item -Path 'C:\protoc' -ItemType Directory
|
|
||||||
Set-Location C:\protoc
|
|
||||||
Invoke-WebRequest https://github.com/protocolbuffers/protobuf/releases/download/v21.12/protoc-21.12-win64.zip -OutFile C:\protoc\protoc.zip
|
|
||||||
7z x protoc.zip
|
|
||||||
Add-Content $env:GITHUB_PATH "C:\protoc\bin"
|
|
||||||
shell: powershell
|
|
||||||
- name: Install npm dependencies
|
|
||||||
run: |
|
|
||||||
cd nodejs
|
|
||||||
npm ci
|
|
||||||
- name: Build Windows native node modules
|
|
||||||
run: .\ci\build_windows_artifacts_nodejs.ps1 ${{ matrix.target }}
|
|
||||||
- name: Upload Windows Artifacts
|
|
||||||
uses: actions/upload-artifact@v4
|
|
||||||
with:
|
|
||||||
name: nodejs-native-windows
|
|
||||||
path: |
|
|
||||||
nodejs/dist/*.node
|
|
||||||
|
|
||||||
release:
|
|
||||||
name: vectordb NPM Publish
|
|
||||||
needs: [node, node-macos, node-linux, node-windows]
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
# Only runs on tags that matches the make-release action
|
|
||||||
if: startsWith(github.ref, 'refs/tags/v')
|
|
||||||
steps:
|
|
||||||
- uses: actions/download-artifact@v4
|
|
||||||
with:
|
|
||||||
pattern: node-*
|
|
||||||
- name: Display structure of downloaded files
|
|
||||||
run: ls -R
|
|
||||||
- uses: actions/setup-node@v3
|
|
||||||
with:
|
|
||||||
node-version: 20
|
|
||||||
registry-url: "https://registry.npmjs.org"
|
|
||||||
- name: Publish to NPM
|
|
||||||
env:
|
|
||||||
NODE_AUTH_TOKEN: ${{ secrets.LANCEDB_NPM_REGISTRY_TOKEN }}
|
|
||||||
run: |
|
|
||||||
# Tag beta as "preview" instead of default "latest". See lancedb
|
|
||||||
# npm publish step for more info.
|
|
||||||
if [[ $GITHUB_REF =~ refs/tags/v(.*)-beta.* ]]; then
|
|
||||||
PUBLISH_ARGS="--tag preview"
|
|
||||||
fi
|
|
||||||
|
|
||||||
mv */*.tgz .
|
|
||||||
for filename in *.tgz; do
|
|
||||||
npm publish $PUBLISH_ARGS $filename
|
|
||||||
done
|
|
||||||
- name: Notify Slack Action
|
|
||||||
uses: ravsamhq/notify-slack-action@2.3.0
|
|
||||||
if: ${{ always() }}
|
|
||||||
with:
|
|
||||||
status: ${{ job.status }}
|
|
||||||
notify_when: "failure"
|
|
||||||
notification_title: "{workflow} is failing"
|
|
||||||
env:
|
|
||||||
SLACK_WEBHOOK_URL: ${{ secrets.ACTION_MONITORING_SLACK }}
|
|
||||||
|
|
||||||
release-nodejs:
|
|
||||||
name: lancedb NPM Publish
|
|
||||||
needs: [nodejs-macos, nodejs-linux, nodejs-windows]
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
# Only runs on tags that matches the make-release action
|
|
||||||
if: startsWith(github.ref, 'refs/tags/v')
|
|
||||||
defaults:
|
|
||||||
run:
|
|
||||||
shell: bash
|
|
||||||
working-directory: nodejs
|
|
||||||
steps:
|
|
||||||
- name: Checkout
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
- uses: actions/download-artifact@v4
|
|
||||||
with:
|
|
||||||
name: nodejs-dist
|
|
||||||
path: nodejs/dist
|
|
||||||
- uses: actions/download-artifact@v4
|
|
||||||
name: Download arch-specific binaries
|
|
||||||
with:
|
|
||||||
pattern: nodejs-*
|
|
||||||
path: nodejs/nodejs-artifacts
|
|
||||||
merge-multiple: true
|
|
||||||
- name: Display structure of downloaded files
|
|
||||||
run: find .
|
|
||||||
- uses: actions/setup-node@v3
|
|
||||||
with:
|
|
||||||
node-version: 20
|
|
||||||
registry-url: "https://registry.npmjs.org"
|
|
||||||
- name: Install napi-rs
|
|
||||||
run: npm install -g @napi-rs/cli
|
|
||||||
- name: Prepare artifacts
|
|
||||||
run: npx napi artifacts -d nodejs-artifacts
|
|
||||||
- name: Display structure of staged files
|
|
||||||
run: find npm
|
|
||||||
- name: Publish to NPM
|
|
||||||
env:
|
|
||||||
NODE_AUTH_TOKEN: ${{ secrets.LANCEDB_NPM_REGISTRY_TOKEN }}
|
|
||||||
# By default, things are published to the latest tag. This is what is
|
|
||||||
# installed by default if the user does not specify a version. This is
|
|
||||||
# good for stable releases, but for pre-releases, we want to publish to
|
|
||||||
# the "preview" tag so they can install with `npm install lancedb@preview`.
|
|
||||||
# See: https://medium.com/@mbostock/prereleases-and-npm-e778fc5e2420
|
|
||||||
run: |
|
|
||||||
if [[ $GITHUB_REF =~ refs/tags/v(.*)-beta.* ]]; then
|
|
||||||
npm publish --access public --tag preview
|
|
||||||
else
|
|
||||||
npm publish --access public
|
|
||||||
fi
|
|
||||||
- name: Notify Slack Action
|
|
||||||
uses: ravsamhq/notify-slack-action@2.3.0
|
|
||||||
if: ${{ always() }}
|
|
||||||
with:
|
|
||||||
status: ${{ job.status }}
|
|
||||||
notify_when: "failure"
|
|
||||||
notification_title: "{workflow} is failing"
|
|
||||||
env:
|
|
||||||
SLACK_WEBHOOK_URL: ${{ secrets.ACTION_MONITORING_SLACK }}
|
|
||||||
|
|
||||||
update-package-lock:
|
|
||||||
needs: [release]
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
permissions:
|
|
||||||
contents: write
|
|
||||||
steps:
|
|
||||||
- name: Checkout
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
with:
|
|
||||||
ref: main
|
|
||||||
persist-credentials: false
|
|
||||||
fetch-depth: 0
|
|
||||||
lfs: true
|
|
||||||
- uses: ./.github/workflows/update_package_lock
|
|
||||||
with:
|
|
||||||
github_token: ${{ secrets.GITHUB_TOKEN }}
|
|
||||||
|
|
||||||
update-package-lock-nodejs:
|
|
||||||
needs: [release-nodejs]
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
permissions:
|
|
||||||
contents: write
|
|
||||||
steps:
|
|
||||||
- name: Checkout
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
with:
|
|
||||||
ref: main
|
|
||||||
persist-credentials: false
|
|
||||||
fetch-depth: 0
|
|
||||||
lfs: true
|
|
||||||
- uses: ./.github/workflows/update_package_lock_nodejs
|
|
||||||
with:
|
|
||||||
github_token: ${{ secrets.GITHUB_TOKEN }}
|
|
||||||
|
|
||||||
gh-release:
|
gh-release:
|
||||||
|
if: startsWith(github.ref, 'refs/tags/v')
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
permissions:
|
permissions:
|
||||||
contents: write
|
contents: write
|
||||||
@@ -458,3 +91,277 @@ jobs:
|
|||||||
generate_release_notes: false
|
generate_release_notes: false
|
||||||
name: Node/Rust LanceDB v${{ steps.extract_version.outputs.version }}
|
name: Node/Rust LanceDB v${{ steps.extract_version.outputs.version }}
|
||||||
body: ${{ steps.release_notes.outputs.changelog }}
|
body: ${{ steps.release_notes.outputs.changelog }}
|
||||||
|
|
||||||
|
build-lancedb:
|
||||||
|
strategy:
|
||||||
|
fail-fast: false
|
||||||
|
matrix:
|
||||||
|
settings:
|
||||||
|
- target: x86_64-apple-darwin
|
||||||
|
host: macos-latest
|
||||||
|
features: ","
|
||||||
|
pre_build: |-
|
||||||
|
brew install protobuf
|
||||||
|
rustup target add x86_64-apple-darwin
|
||||||
|
- target: aarch64-apple-darwin
|
||||||
|
host: macos-latest
|
||||||
|
features: fp16kernels
|
||||||
|
pre_build: brew install protobuf
|
||||||
|
- target: x86_64-pc-windows-msvc
|
||||||
|
host: windows-latest
|
||||||
|
features: ","
|
||||||
|
pre_build: |-
|
||||||
|
choco install --no-progress protoc ninja nasm
|
||||||
|
tail -n 1000 /c/ProgramData/chocolatey/logs/chocolatey.log
|
||||||
|
# There is an issue where choco doesn't add nasm to the path
|
||||||
|
export PATH="$PATH:/c/Program Files/NASM"
|
||||||
|
nasm -v
|
||||||
|
- target: aarch64-pc-windows-msvc
|
||||||
|
host: windows-latest
|
||||||
|
features: ","
|
||||||
|
pre_build: |-
|
||||||
|
choco install --no-progress protoc
|
||||||
|
rustup target add aarch64-pc-windows-msvc
|
||||||
|
- target: x86_64-unknown-linux-gnu
|
||||||
|
host: ubuntu-latest
|
||||||
|
features: fp16kernels
|
||||||
|
# https://github.com/napi-rs/napi-rs/blob/main/debian.Dockerfile
|
||||||
|
docker: ghcr.io/napi-rs/napi-rs/nodejs-rust:lts-debian
|
||||||
|
pre_build: |-
|
||||||
|
set -e &&
|
||||||
|
apt-get update &&
|
||||||
|
apt-get install -y protobuf-compiler pkg-config
|
||||||
|
- target: x86_64-unknown-linux-musl
|
||||||
|
# This one seems to need some extra memory
|
||||||
|
host: ubuntu-2404-8x-x64
|
||||||
|
# https://github.com/napi-rs/napi-rs/blob/main/alpine.Dockerfile
|
||||||
|
docker: ghcr.io/napi-rs/napi-rs/nodejs-rust:lts-alpine
|
||||||
|
features: fp16kernels
|
||||||
|
pre_build: |-
|
||||||
|
set -e &&
|
||||||
|
apk add protobuf-dev curl &&
|
||||||
|
ln -s /usr/lib/gcc/x86_64-alpine-linux-musl/14.2.0/crtbeginS.o /usr/lib/crtbeginS.o &&
|
||||||
|
ln -s /usr/lib/libgcc_s.so /usr/lib/libgcc.so &&
|
||||||
|
CC=gcc &&
|
||||||
|
CXX=g++
|
||||||
|
- target: aarch64-unknown-linux-gnu
|
||||||
|
host: ubuntu-2404-8x-x64
|
||||||
|
# https://github.com/napi-rs/napi-rs/blob/main/debian-aarch64.Dockerfile
|
||||||
|
docker: ghcr.io/napi-rs/napi-rs/nodejs-rust:lts-debian-aarch64
|
||||||
|
features: "fp16kernels"
|
||||||
|
pre_build: |-
|
||||||
|
set -e &&
|
||||||
|
apt-get update &&
|
||||||
|
apt-get install -y protobuf-compiler pkg-config &&
|
||||||
|
# https://github.com/aws/aws-lc-rs/issues/737#issuecomment-2725918627
|
||||||
|
ln -s /usr/aarch64-unknown-linux-gnu/lib/gcc/aarch64-unknown-linux-gnu/4.8.5/crtbeginS.o /usr/aarch64-unknown-linux-gnu/aarch64-unknown-linux-gnu/sysroot/usr/lib/crtbeginS.o &&
|
||||||
|
ln -s /usr/aarch64-unknown-linux-gnu/lib/gcc /usr/aarch64-unknown-linux-gnu/aarch64-unknown-linux-gnu/sysroot/usr/lib/gcc &&
|
||||||
|
rustup target add aarch64-unknown-linux-gnu
|
||||||
|
- target: aarch64-unknown-linux-musl
|
||||||
|
host: ubuntu-2404-8x-x64
|
||||||
|
# https://github.com/napi-rs/napi-rs/blob/main/alpine.Dockerfile
|
||||||
|
docker: ghcr.io/napi-rs/napi-rs/nodejs-rust:lts-alpine
|
||||||
|
features: ","
|
||||||
|
pre_build: |-
|
||||||
|
set -e &&
|
||||||
|
apk add protobuf-dev &&
|
||||||
|
rustup target add aarch64-unknown-linux-musl &&
|
||||||
|
export CC_aarch64_unknown_linux_musl=aarch64-linux-musl-gcc &&
|
||||||
|
export CXX_aarch64_unknown_linux_musl=aarch64-linux-musl-g++
|
||||||
|
name: build - ${{ matrix.settings.target }}
|
||||||
|
runs-on: ${{ matrix.settings.host }}
|
||||||
|
defaults:
|
||||||
|
run:
|
||||||
|
working-directory: nodejs
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v4
|
||||||
|
- name: Setup node
|
||||||
|
uses: actions/setup-node@v4
|
||||||
|
if: ${{ !matrix.settings.docker }}
|
||||||
|
with:
|
||||||
|
node-version: 20
|
||||||
|
cache: npm
|
||||||
|
cache-dependency-path: nodejs/package-lock.json
|
||||||
|
- name: Install
|
||||||
|
uses: dtolnay/rust-toolchain@stable
|
||||||
|
if: ${{ !matrix.settings.docker }}
|
||||||
|
with:
|
||||||
|
toolchain: stable
|
||||||
|
targets: ${{ matrix.settings.target }}
|
||||||
|
- name: Cache cargo
|
||||||
|
uses: actions/cache@v4
|
||||||
|
with:
|
||||||
|
path: |
|
||||||
|
~/.cargo/registry/index/
|
||||||
|
~/.cargo/registry/cache/
|
||||||
|
~/.cargo/git/db/
|
||||||
|
.cargo-cache
|
||||||
|
target/
|
||||||
|
key: nodejs-${{ matrix.settings.target }}-cargo-${{ matrix.settings.host }}
|
||||||
|
- name: Setup toolchain
|
||||||
|
run: ${{ matrix.settings.setup }}
|
||||||
|
if: ${{ matrix.settings.setup }}
|
||||||
|
shell: bash
|
||||||
|
- name: Install dependencies
|
||||||
|
run: npm ci
|
||||||
|
- name: Build in docker
|
||||||
|
uses: addnab/docker-run-action@v3
|
||||||
|
if: ${{ matrix.settings.docker }}
|
||||||
|
with:
|
||||||
|
image: ${{ matrix.settings.docker }}
|
||||||
|
options: "--user 0:0 -v ${{ github.workspace }}/.cargo-cache/git/db:/usr/local/cargo/git/db \
|
||||||
|
-v ${{ github.workspace }}/.cargo/registry/cache:/usr/local/cargo/registry/cache \
|
||||||
|
-v ${{ github.workspace }}/.cargo/registry/index:/usr/local/cargo/registry/index \
|
||||||
|
-v ${{ github.workspace }}:/build -w /build/nodejs"
|
||||||
|
run: |
|
||||||
|
set -e
|
||||||
|
${{ matrix.settings.pre_build }}
|
||||||
|
npx napi build --platform --release --no-const-enum \
|
||||||
|
--features ${{ matrix.settings.features }} \
|
||||||
|
--target ${{ matrix.settings.target }} \
|
||||||
|
--dts ../lancedb/native.d.ts \
|
||||||
|
--js ../lancedb/native.js \
|
||||||
|
--strip \
|
||||||
|
dist/
|
||||||
|
- name: Build
|
||||||
|
run: |
|
||||||
|
${{ matrix.settings.pre_build }}
|
||||||
|
npx napi build --platform --release --no-const-enum \
|
||||||
|
--features ${{ matrix.settings.features }} \
|
||||||
|
--target ${{ matrix.settings.target }} \
|
||||||
|
--dts ../lancedb/native.d.ts \
|
||||||
|
--js ../lancedb/native.js \
|
||||||
|
--strip \
|
||||||
|
$EXTRA_ARGS \
|
||||||
|
dist/
|
||||||
|
if: ${{ !matrix.settings.docker }}
|
||||||
|
shell: bash
|
||||||
|
- name: Upload artifact
|
||||||
|
uses: actions/upload-artifact@v4
|
||||||
|
with:
|
||||||
|
name: lancedb-${{ matrix.settings.target }}
|
||||||
|
path: nodejs/dist/*.node
|
||||||
|
if-no-files-found: error
|
||||||
|
# The generic files are the same in all distros so we just pick
|
||||||
|
# one to do the upload.
|
||||||
|
- name: Make generic artifacts
|
||||||
|
if: ${{ matrix.settings.target == 'aarch64-apple-darwin' }}
|
||||||
|
run: npm run tsc
|
||||||
|
- name: Upload Generic Artifacts
|
||||||
|
if: ${{ matrix.settings.target == 'aarch64-apple-darwin' }}
|
||||||
|
uses: actions/upload-artifact@v4
|
||||||
|
with:
|
||||||
|
name: nodejs-dist
|
||||||
|
path: |
|
||||||
|
nodejs/dist/*
|
||||||
|
!nodejs/dist/*.node
|
||||||
|
test-lancedb:
|
||||||
|
name: "Test: ${{ matrix.settings.target }} - node@${{ matrix.node }}"
|
||||||
|
needs:
|
||||||
|
- build-lancedb
|
||||||
|
strategy:
|
||||||
|
fail-fast: false
|
||||||
|
matrix:
|
||||||
|
settings:
|
||||||
|
# TODO: Get tests passing on Windows (failing from test tmpdir issue)
|
||||||
|
# - host: windows-latest
|
||||||
|
# target: x86_64-pc-windows-msvc
|
||||||
|
- host: macos-latest
|
||||||
|
target: aarch64-apple-darwin
|
||||||
|
- target: x86_64-unknown-linux-gnu
|
||||||
|
host: ubuntu-latest
|
||||||
|
- target: aarch64-unknown-linux-gnu
|
||||||
|
host: buildjet-16vcpu-ubuntu-2204-arm
|
||||||
|
node:
|
||||||
|
- '20'
|
||||||
|
runs-on: ${{ matrix.settings.host }}
|
||||||
|
defaults:
|
||||||
|
run:
|
||||||
|
shell: bash
|
||||||
|
working-directory: nodejs
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v4
|
||||||
|
- name: Setup node
|
||||||
|
uses: actions/setup-node@v4
|
||||||
|
with:
|
||||||
|
node-version: ${{ matrix.node }}
|
||||||
|
cache: npm
|
||||||
|
cache-dependency-path: nodejs/package-lock.json
|
||||||
|
- name: Install dependencies
|
||||||
|
run: npm ci
|
||||||
|
- name: Download artifacts
|
||||||
|
uses: actions/download-artifact@v4
|
||||||
|
with:
|
||||||
|
name: lancedb-${{ matrix.settings.target }}
|
||||||
|
path: nodejs/dist/
|
||||||
|
# For testing purposes:
|
||||||
|
# run-id: 13982782871
|
||||||
|
# github-token: ${{ secrets.GITHUB_TOKEN }} # token with actions:read permissions on target repo
|
||||||
|
- uses: actions/download-artifact@v4
|
||||||
|
with:
|
||||||
|
name: nodejs-dist
|
||||||
|
path: nodejs/dist
|
||||||
|
# For testing purposes:
|
||||||
|
# github-token: ${{ secrets.GITHUB_TOKEN }} # token with actions:read permissions on target repo
|
||||||
|
# run-id: 13982782871
|
||||||
|
- name: List packages
|
||||||
|
run: ls -R dist
|
||||||
|
- name: Move built files
|
||||||
|
run: cp dist/native.d.ts dist/native.js dist/*.node lancedb/
|
||||||
|
- name: Test bindings
|
||||||
|
run: npm test
|
||||||
|
publish:
|
||||||
|
name: Publish
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
defaults:
|
||||||
|
run:
|
||||||
|
shell: bash
|
||||||
|
working-directory: nodejs
|
||||||
|
needs:
|
||||||
|
- test-lancedb
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v4
|
||||||
|
- name: Setup node
|
||||||
|
uses: actions/setup-node@v4
|
||||||
|
with:
|
||||||
|
node-version: 20
|
||||||
|
cache: npm
|
||||||
|
cache-dependency-path: nodejs/package-lock.json
|
||||||
|
registry-url: "https://registry.npmjs.org"
|
||||||
|
- name: Install dependencies
|
||||||
|
run: npm ci
|
||||||
|
- uses: actions/download-artifact@v4
|
||||||
|
with:
|
||||||
|
name: nodejs-dist
|
||||||
|
path: nodejs/dist
|
||||||
|
# For testing purposes:
|
||||||
|
# run-id: 13982782871
|
||||||
|
# github-token: ${{ secrets.GITHUB_TOKEN }} # token with actions:read permissions on target repo
|
||||||
|
- uses: actions/download-artifact@v4
|
||||||
|
name: Download arch-specific binaries
|
||||||
|
with:
|
||||||
|
pattern: lancedb-*
|
||||||
|
path: nodejs/nodejs-artifacts
|
||||||
|
merge-multiple: true
|
||||||
|
# For testing purposes:
|
||||||
|
# run-id: 13982782871
|
||||||
|
# github-token: ${{ secrets.GITHUB_TOKEN }} # token with actions:read permissions on target repo
|
||||||
|
- name: Display structure of downloaded files
|
||||||
|
run: find dist && find nodejs-artifacts
|
||||||
|
- name: Move artifacts
|
||||||
|
run: npx napi artifacts -d nodejs-artifacts
|
||||||
|
- name: List packages
|
||||||
|
run: find npm
|
||||||
|
- name: Publish
|
||||||
|
env:
|
||||||
|
NODE_AUTH_TOKEN: ${{ secrets.LANCEDB_NPM_REGISTRY_TOKEN }}
|
||||||
|
DRY_RUN: ${{ !startsWith(github.ref, 'refs/tags/v') }}
|
||||||
|
run: |
|
||||||
|
ARGS="--access public"
|
||||||
|
if [[ $DRY_RUN == "true" ]]; then
|
||||||
|
ARGS="$ARGS --dry-run"
|
||||||
|
fi
|
||||||
|
if [[ $GITHUB_REF =~ refs/tags/v(.*)-beta.* ]]; then
|
||||||
|
ARGS="$ARGS --tag preview"
|
||||||
|
fi
|
||||||
|
npm publish $ARGS
|
||||||
|
|||||||
25
.github/workflows/pypi-publish.yml
vendored
25
.github/workflows/pypi-publish.yml
vendored
@@ -4,6 +4,11 @@ on:
|
|||||||
push:
|
push:
|
||||||
tags:
|
tags:
|
||||||
- 'python-v*'
|
- 'python-v*'
|
||||||
|
pull_request:
|
||||||
|
# This should trigger a dry run (we skip the final publish step)
|
||||||
|
paths:
|
||||||
|
- .github/workflows/pypi-publish.yml
|
||||||
|
- Cargo.toml # Change in dependency frequently breaks builds
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
linux:
|
linux:
|
||||||
@@ -15,15 +20,21 @@ jobs:
|
|||||||
- platform: x86_64
|
- platform: x86_64
|
||||||
manylinux: "2_17"
|
manylinux: "2_17"
|
||||||
extra_args: ""
|
extra_args: ""
|
||||||
|
runner: ubuntu-22.04
|
||||||
- platform: x86_64
|
- platform: x86_64
|
||||||
manylinux: "2_28"
|
manylinux: "2_28"
|
||||||
extra_args: "--features fp16kernels"
|
extra_args: "--features fp16kernels"
|
||||||
|
runner: ubuntu-22.04
|
||||||
- platform: aarch64
|
- platform: aarch64
|
||||||
manylinux: "2_24"
|
manylinux: "2_17"
|
||||||
extra_args: ""
|
extra_args: ""
|
||||||
# We don't build fp16 kernels for aarch64, because it uses
|
# For successful fat LTO builds, we need a large runner to avoid OOM errors.
|
||||||
# cross compilation image, which doesn't have a new enough compiler.
|
runner: ubuntu-2404-8x-arm64
|
||||||
runs-on: "ubuntu-22.04"
|
- platform: aarch64
|
||||||
|
manylinux: "2_28"
|
||||||
|
extra_args: "--features fp16kernels"
|
||||||
|
runner: ubuntu-2404-8x-arm64
|
||||||
|
runs-on: ${{ matrix.config.runner }}
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
@@ -40,6 +51,7 @@ jobs:
|
|||||||
arm-build: ${{ matrix.config.platform == 'aarch64' }}
|
arm-build: ${{ matrix.config.platform == 'aarch64' }}
|
||||||
manylinux: ${{ matrix.config.manylinux }}
|
manylinux: ${{ matrix.config.manylinux }}
|
||||||
- uses: ./.github/workflows/upload_wheel
|
- uses: ./.github/workflows/upload_wheel
|
||||||
|
if: startsWith(github.ref, 'refs/tags/python-v')
|
||||||
with:
|
with:
|
||||||
pypi_token: ${{ secrets.LANCEDB_PYPI_API_TOKEN }}
|
pypi_token: ${{ secrets.LANCEDB_PYPI_API_TOKEN }}
|
||||||
fury_token: ${{ secrets.FURY_TOKEN }}
|
fury_token: ${{ secrets.FURY_TOKEN }}
|
||||||
@@ -69,6 +81,7 @@ jobs:
|
|||||||
python-minor-version: 8
|
python-minor-version: 8
|
||||||
args: "--release --strip --target ${{ matrix.config.target }} --features fp16kernels"
|
args: "--release --strip --target ${{ matrix.config.target }} --features fp16kernels"
|
||||||
- uses: ./.github/workflows/upload_wheel
|
- uses: ./.github/workflows/upload_wheel
|
||||||
|
if: startsWith(github.ref, 'refs/tags/python-v')
|
||||||
with:
|
with:
|
||||||
pypi_token: ${{ secrets.LANCEDB_PYPI_API_TOKEN }}
|
pypi_token: ${{ secrets.LANCEDB_PYPI_API_TOKEN }}
|
||||||
fury_token: ${{ secrets.FURY_TOKEN }}
|
fury_token: ${{ secrets.FURY_TOKEN }}
|
||||||
@@ -83,17 +96,19 @@ jobs:
|
|||||||
- name: Set up Python
|
- name: Set up Python
|
||||||
uses: actions/setup-python@v4
|
uses: actions/setup-python@v4
|
||||||
with:
|
with:
|
||||||
python-version: 3.8
|
python-version: 3.12
|
||||||
- uses: ./.github/workflows/build_windows_wheel
|
- uses: ./.github/workflows/build_windows_wheel
|
||||||
with:
|
with:
|
||||||
python-minor-version: 8
|
python-minor-version: 8
|
||||||
args: "--release --strip"
|
args: "--release --strip"
|
||||||
vcpkg_token: ${{ secrets.VCPKG_GITHUB_PACKAGES }}
|
vcpkg_token: ${{ secrets.VCPKG_GITHUB_PACKAGES }}
|
||||||
- uses: ./.github/workflows/upload_wheel
|
- uses: ./.github/workflows/upload_wheel
|
||||||
|
if: startsWith(github.ref, 'refs/tags/python-v')
|
||||||
with:
|
with:
|
||||||
pypi_token: ${{ secrets.LANCEDB_PYPI_API_TOKEN }}
|
pypi_token: ${{ secrets.LANCEDB_PYPI_API_TOKEN }}
|
||||||
fury_token: ${{ secrets.FURY_TOKEN }}
|
fury_token: ${{ secrets.FURY_TOKEN }}
|
||||||
gh-release:
|
gh-release:
|
||||||
|
if: startsWith(github.ref, 'refs/tags/python-v')
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
permissions:
|
permissions:
|
||||||
contents: write
|
contents: write
|
||||||
|
|||||||
62
.github/workflows/python.yml
vendored
62
.github/workflows/python.yml
vendored
@@ -13,6 +13,11 @@ concurrency:
|
|||||||
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
|
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
|
||||||
cancel-in-progress: true
|
cancel-in-progress: true
|
||||||
|
|
||||||
|
env:
|
||||||
|
# Color output for pytest is off by default.
|
||||||
|
PYTEST_ADDOPTS: "--color=yes"
|
||||||
|
FORCE_COLOR: "1"
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
lint:
|
lint:
|
||||||
name: "Lint"
|
name: "Lint"
|
||||||
@@ -30,16 +35,17 @@ jobs:
|
|||||||
- name: Set up Python
|
- name: Set up Python
|
||||||
uses: actions/setup-python@v5
|
uses: actions/setup-python@v5
|
||||||
with:
|
with:
|
||||||
python-version: "3.11"
|
python-version: "3.12"
|
||||||
- name: Install ruff
|
- name: Install ruff
|
||||||
run: |
|
run: |
|
||||||
pip install ruff==0.5.4
|
pip install ruff==0.9.9
|
||||||
- name: Format check
|
- name: Format check
|
||||||
run: ruff format --check .
|
run: ruff format --check .
|
||||||
- name: Lint
|
- name: Lint
|
||||||
run: ruff check .
|
run: ruff check .
|
||||||
doctest:
|
|
||||||
name: "Doctest"
|
type-check:
|
||||||
|
name: "Type Check"
|
||||||
timeout-minutes: 30
|
timeout-minutes: 30
|
||||||
runs-on: "ubuntu-22.04"
|
runs-on: "ubuntu-22.04"
|
||||||
defaults:
|
defaults:
|
||||||
@@ -54,7 +60,36 @@ jobs:
|
|||||||
- name: Set up Python
|
- name: Set up Python
|
||||||
uses: actions/setup-python@v5
|
uses: actions/setup-python@v5
|
||||||
with:
|
with:
|
||||||
python-version: "3.11"
|
python-version: "3.12"
|
||||||
|
- name: Install protobuf compiler
|
||||||
|
run: |
|
||||||
|
sudo apt update
|
||||||
|
sudo apt install -y protobuf-compiler
|
||||||
|
pip install toml
|
||||||
|
- name: Install dependencies
|
||||||
|
run: |
|
||||||
|
python ../ci/parse_requirements.py pyproject.toml --extras dev,tests,embeddings > requirements.txt
|
||||||
|
pip install -r requirements.txt
|
||||||
|
- name: Run pyright
|
||||||
|
run: pyright
|
||||||
|
|
||||||
|
doctest:
|
||||||
|
name: "Doctest"
|
||||||
|
timeout-minutes: 30
|
||||||
|
runs-on: "ubuntu-24.04"
|
||||||
|
defaults:
|
||||||
|
run:
|
||||||
|
shell: bash
|
||||||
|
working-directory: python
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
fetch-depth: 0
|
||||||
|
lfs: true
|
||||||
|
- name: Set up Python
|
||||||
|
uses: actions/setup-python@v5
|
||||||
|
with:
|
||||||
|
python-version: "3.12"
|
||||||
cache: "pip"
|
cache: "pip"
|
||||||
- name: Install protobuf
|
- name: Install protobuf
|
||||||
run: |
|
run: |
|
||||||
@@ -75,8 +110,8 @@ jobs:
|
|||||||
timeout-minutes: 30
|
timeout-minutes: 30
|
||||||
strategy:
|
strategy:
|
||||||
matrix:
|
matrix:
|
||||||
python-minor-version: ["9", "11"]
|
python-minor-version: ["9", "12"]
|
||||||
runs-on: "ubuntu-22.04"
|
runs-on: "ubuntu-24.04"
|
||||||
defaults:
|
defaults:
|
||||||
run:
|
run:
|
||||||
shell: bash
|
shell: bash
|
||||||
@@ -101,6 +136,10 @@ jobs:
|
|||||||
- uses: ./.github/workflows/run_tests
|
- uses: ./.github/workflows/run_tests
|
||||||
with:
|
with:
|
||||||
integration: true
|
integration: true
|
||||||
|
- name: Test without pylance or pandas
|
||||||
|
run: |
|
||||||
|
pip uninstall -y pylance pandas
|
||||||
|
pytest -vv python/tests/test_table.py
|
||||||
# Make sure wheels are not included in the Rust cache
|
# Make sure wheels are not included in the Rust cache
|
||||||
- name: Delete wheels
|
- name: Delete wheels
|
||||||
run: rm -rf target/wheels
|
run: rm -rf target/wheels
|
||||||
@@ -127,7 +166,7 @@ jobs:
|
|||||||
- name: Set up Python
|
- name: Set up Python
|
||||||
uses: actions/setup-python@v5
|
uses: actions/setup-python@v5
|
||||||
with:
|
with:
|
||||||
python-version: "3.11"
|
python-version: "3.12"
|
||||||
- uses: Swatinem/rust-cache@v2
|
- uses: Swatinem/rust-cache@v2
|
||||||
with:
|
with:
|
||||||
workspaces: python
|
workspaces: python
|
||||||
@@ -138,7 +177,7 @@ jobs:
|
|||||||
run: rm -rf target/wheels
|
run: rm -rf target/wheels
|
||||||
windows:
|
windows:
|
||||||
name: "Windows: ${{ matrix.config.name }}"
|
name: "Windows: ${{ matrix.config.name }}"
|
||||||
timeout-minutes: 30
|
timeout-minutes: 60
|
||||||
strategy:
|
strategy:
|
||||||
matrix:
|
matrix:
|
||||||
config:
|
config:
|
||||||
@@ -157,7 +196,7 @@ jobs:
|
|||||||
- name: Set up Python
|
- name: Set up Python
|
||||||
uses: actions/setup-python@v5
|
uses: actions/setup-python@v5
|
||||||
with:
|
with:
|
||||||
python-version: "3.11"
|
python-version: "3.12"
|
||||||
- uses: Swatinem/rust-cache@v2
|
- uses: Swatinem/rust-cache@v2
|
||||||
with:
|
with:
|
||||||
workspaces: python
|
workspaces: python
|
||||||
@@ -168,7 +207,7 @@ jobs:
|
|||||||
run: rm -rf target/wheels
|
run: rm -rf target/wheels
|
||||||
pydantic1x:
|
pydantic1x:
|
||||||
timeout-minutes: 30
|
timeout-minutes: 30
|
||||||
runs-on: "ubuntu-22.04"
|
runs-on: "ubuntu-24.04"
|
||||||
defaults:
|
defaults:
|
||||||
run:
|
run:
|
||||||
shell: bash
|
shell: bash
|
||||||
@@ -189,6 +228,7 @@ jobs:
|
|||||||
- name: Install lancedb
|
- name: Install lancedb
|
||||||
run: |
|
run: |
|
||||||
pip install "pydantic<2"
|
pip install "pydantic<2"
|
||||||
|
pip install pyarrow==16
|
||||||
pip install --extra-index-url https://pypi.fury.io/lancedb/ -e .[tests]
|
pip install --extra-index-url https://pypi.fury.io/lancedb/ -e .[tests]
|
||||||
pip install tantivy
|
pip install tantivy
|
||||||
- name: Run tests
|
- name: Run tests
|
||||||
|
|||||||
4
.github/workflows/run_tests/action.yml
vendored
4
.github/workflows/run_tests/action.yml
vendored
@@ -24,8 +24,8 @@ runs:
|
|||||||
- name: pytest (with integration)
|
- name: pytest (with integration)
|
||||||
shell: bash
|
shell: bash
|
||||||
if: ${{ inputs.integration == 'true' }}
|
if: ${{ inputs.integration == 'true' }}
|
||||||
run: pytest -m "not slow" -x -v --durations=30 python/python/tests
|
run: pytest -m "not slow" -vv --durations=30 python/python/tests
|
||||||
- name: pytest (no integration tests)
|
- name: pytest (no integration tests)
|
||||||
shell: bash
|
shell: bash
|
||||||
if: ${{ inputs.integration != 'true' }}
|
if: ${{ inputs.integration != 'true' }}
|
||||||
run: pytest -m "not slow and not s3_test" -x -v --durations=30 python/python/tests
|
run: pytest -m "not slow and not s3_test" -vv --durations=30 python/python/tests
|
||||||
|
|||||||
144
.github/workflows/rust.yml
vendored
144
.github/workflows/rust.yml
vendored
@@ -22,24 +22,27 @@ env:
|
|||||||
# "1" means line tables only, which is useful for panic tracebacks.
|
# "1" means line tables only, which is useful for panic tracebacks.
|
||||||
RUSTFLAGS: "-C debuginfo=1"
|
RUSTFLAGS: "-C debuginfo=1"
|
||||||
RUST_BACKTRACE: "1"
|
RUST_BACKTRACE: "1"
|
||||||
|
CARGO_INCREMENTAL: 0
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
lint:
|
lint:
|
||||||
timeout-minutes: 30
|
timeout-minutes: 30
|
||||||
runs-on: ubuntu-22.04
|
runs-on: ubuntu-24.04
|
||||||
defaults:
|
defaults:
|
||||||
run:
|
run:
|
||||||
shell: bash
|
shell: bash
|
||||||
working-directory: rust
|
|
||||||
env:
|
env:
|
||||||
# Need up-to-date compilers for kernels
|
# Need up-to-date compilers for kernels
|
||||||
CC: gcc-12
|
CC: clang-18
|
||||||
CXX: g++-12
|
CXX: clang++-18
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
fetch-depth: 0
|
fetch-depth: 0
|
||||||
lfs: true
|
lfs: true
|
||||||
|
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
||||||
|
with:
|
||||||
|
components: rustfmt, clippy
|
||||||
- uses: Swatinem/rust-cache@v2
|
- uses: Swatinem/rust-cache@v2
|
||||||
with:
|
with:
|
||||||
workspaces: rust
|
workspaces: rust
|
||||||
@@ -50,21 +53,49 @@ jobs:
|
|||||||
- name: Run format
|
- name: Run format
|
||||||
run: cargo fmt --all -- --check
|
run: cargo fmt --all -- --check
|
||||||
- name: Run clippy
|
- name: Run clippy
|
||||||
run: cargo clippy --all --all-features -- -D warnings
|
run: cargo clippy --workspace --tests --all-features -- -D warnings
|
||||||
|
|
||||||
|
build-no-lock:
|
||||||
|
runs-on: ubuntu-24.04
|
||||||
|
timeout-minutes: 30
|
||||||
|
env:
|
||||||
|
# Need up-to-date compilers for kernels
|
||||||
|
CC: clang
|
||||||
|
CXX: clang++
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v4
|
||||||
|
# Building without a lock file often requires the latest Rust version since downstream
|
||||||
|
# dependencies may have updated their minimum Rust version.
|
||||||
|
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
||||||
|
with:
|
||||||
|
toolchain: "stable"
|
||||||
|
# Remove cargo.lock to force a fresh build
|
||||||
|
- name: Remove Cargo.lock
|
||||||
|
run: rm -f Cargo.lock
|
||||||
|
- uses: rui314/setup-mold@v1
|
||||||
|
- uses: Swatinem/rust-cache@v2
|
||||||
|
- name: Install dependencies
|
||||||
|
run: |
|
||||||
|
sudo apt update
|
||||||
|
sudo apt install -y protobuf-compiler libssl-dev
|
||||||
|
- name: Build all
|
||||||
|
run: |
|
||||||
|
cargo build --benches --all-features --tests
|
||||||
|
|
||||||
linux:
|
linux:
|
||||||
timeout-minutes: 30
|
timeout-minutes: 30
|
||||||
# To build all features, we need more disk space than is available
|
# To build all features, we need more disk space than is available
|
||||||
# on the GitHub-provided runner. This is mostly due to the the
|
# on the free OSS github runner. This is mostly due to the the
|
||||||
# sentence-transformers feature.
|
# sentence-transformers feature.
|
||||||
runs-on: warp-ubuntu-latest-x64-4x
|
runs-on: ubuntu-2404-4x-x64
|
||||||
defaults:
|
defaults:
|
||||||
run:
|
run:
|
||||||
shell: bash
|
shell: bash
|
||||||
working-directory: rust
|
working-directory: rust
|
||||||
env:
|
env:
|
||||||
# Need up-to-date compilers for kernels
|
# Need up-to-date compilers for kernels
|
||||||
CC: gcc-12
|
CC: clang-18
|
||||||
CXX: g++-12
|
CXX: clang++-18
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
@@ -75,22 +106,32 @@ jobs:
|
|||||||
workspaces: rust
|
workspaces: rust
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: |
|
run: |
|
||||||
sudo apt update
|
# This shaves 2 minutes off this step in CI. This doesn't seem to be
|
||||||
|
# necessary in standard runners, but it is in the 4x runners.
|
||||||
|
sudo rm /var/lib/man-db/auto-update
|
||||||
sudo apt install -y protobuf-compiler libssl-dev
|
sudo apt install -y protobuf-compiler libssl-dev
|
||||||
|
- uses: rui314/setup-mold@v1
|
||||||
|
- name: Make Swap
|
||||||
|
run: |
|
||||||
|
sudo fallocate -l 16G /swapfile
|
||||||
|
sudo chmod 600 /swapfile
|
||||||
|
sudo mkswap /swapfile
|
||||||
|
sudo swapon /swapfile
|
||||||
- name: Start S3 integration test environment
|
- name: Start S3 integration test environment
|
||||||
working-directory: .
|
working-directory: .
|
||||||
run: docker compose up --detach --wait
|
run: docker compose up --detach --wait
|
||||||
- name: Build
|
- name: Build
|
||||||
run: cargo build --all-features
|
run: cargo build --all-features --tests --locked --examples
|
||||||
- name: Run tests
|
- name: Run tests
|
||||||
run: cargo test --all-features
|
run: cargo test --all-features --locked
|
||||||
- name: Run examples
|
- name: Run examples
|
||||||
run: cargo run --example simple
|
run: cargo run --example simple --locked
|
||||||
|
|
||||||
macos:
|
macos:
|
||||||
timeout-minutes: 30
|
timeout-minutes: 30
|
||||||
strategy:
|
strategy:
|
||||||
matrix:
|
matrix:
|
||||||
mac-runner: [ "macos-13", "macos-14" ]
|
mac-runner: ["macos-13", "macos-14"]
|
||||||
runs-on: "${{ matrix.mac-runner }}"
|
runs-on: "${{ matrix.mac-runner }}"
|
||||||
defaults:
|
defaults:
|
||||||
run:
|
run:
|
||||||
@@ -108,29 +149,78 @@ jobs:
|
|||||||
workspaces: rust
|
workspaces: rust
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: brew install protobuf
|
run: brew install protobuf
|
||||||
- name: Build
|
|
||||||
run: cargo build --all-features
|
|
||||||
- name: Run tests
|
- name: Run tests
|
||||||
# Run with everything except the integration tests.
|
run: |
|
||||||
run: cargo test --features remote,fp16kernels
|
# Don't run the s3 integration tests since docker isn't available
|
||||||
|
# on this image.
|
||||||
|
ALL_FEATURES=`cargo metadata --format-version=1 --no-deps \
|
||||||
|
| jq -r '.packages[] | .features | keys | .[]' \
|
||||||
|
| grep -v s3-test | sort | uniq | paste -s -d "," -`
|
||||||
|
cargo test --features $ALL_FEATURES --locked
|
||||||
|
|
||||||
windows:
|
windows:
|
||||||
runs-on: windows-2022
|
runs-on: windows-2022
|
||||||
|
strategy:
|
||||||
|
matrix:
|
||||||
|
target:
|
||||||
|
- x86_64-pc-windows-msvc
|
||||||
|
- aarch64-pc-windows-msvc
|
||||||
|
defaults:
|
||||||
|
run:
|
||||||
|
working-directory: rust/lancedb
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
- uses: Swatinem/rust-cache@v2
|
- uses: Swatinem/rust-cache@v2
|
||||||
with:
|
with:
|
||||||
workspaces: rust
|
workspaces: rust
|
||||||
- name: Install Protoc v21.12
|
- name: Install Protoc v21.12
|
||||||
working-directory: C:\
|
run: choco install --no-progress protoc
|
||||||
|
- name: Build
|
||||||
run: |
|
run: |
|
||||||
New-Item -Path 'C:\protoc' -ItemType Directory
|
rustup target add ${{ matrix.target }}
|
||||||
Set-Location C:\protoc
|
$env:VCPKG_ROOT = $env:VCPKG_INSTALLATION_ROOT
|
||||||
Invoke-WebRequest https://github.com/protocolbuffers/protobuf/releases/download/v21.12/protoc-21.12-win64.zip -OutFile C:\protoc\protoc.zip
|
cargo build --features remote --tests --locked --target ${{ matrix.target }}
|
||||||
7z x protoc.zip
|
|
||||||
Add-Content $env:GITHUB_PATH "C:\protoc\bin"
|
|
||||||
shell: powershell
|
|
||||||
- name: Run tests
|
- name: Run tests
|
||||||
|
# Can only run tests when target matches host
|
||||||
|
if: ${{ matrix.target == 'x86_64-pc-windows-msvc' }}
|
||||||
run: |
|
run: |
|
||||||
$env:VCPKG_ROOT = $env:VCPKG_INSTALLATION_ROOT
|
$env:VCPKG_ROOT = $env:VCPKG_INSTALLATION_ROOT
|
||||||
cargo build
|
cargo test --features remote --locked
|
||||||
cargo test
|
|
||||||
|
msrv:
|
||||||
|
# Check the minimum supported Rust version
|
||||||
|
name: MSRV Check - Rust v${{ matrix.msrv }}
|
||||||
|
runs-on: ubuntu-24.04
|
||||||
|
strategy:
|
||||||
|
matrix:
|
||||||
|
msrv: ["1.78.0"] # This should match up with rust-version in Cargo.toml
|
||||||
|
env:
|
||||||
|
# Need up-to-date compilers for kernels
|
||||||
|
CC: clang-18
|
||||||
|
CXX: clang++-18
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
submodules: true
|
||||||
|
- name: Install dependencies
|
||||||
|
run: |
|
||||||
|
sudo apt update
|
||||||
|
sudo apt install -y protobuf-compiler libssl-dev
|
||||||
|
- name: Install ${{ matrix.msrv }}
|
||||||
|
uses: dtolnay/rust-toolchain@master
|
||||||
|
with:
|
||||||
|
toolchain: ${{ matrix.msrv }}
|
||||||
|
- name: Downgrade dependencies
|
||||||
|
# These packages have newer requirements for MSRV
|
||||||
|
run: |
|
||||||
|
cargo update -p aws-sdk-bedrockruntime --precise 1.64.0
|
||||||
|
cargo update -p aws-sdk-dynamodb --precise 1.55.0
|
||||||
|
cargo update -p aws-config --precise 1.5.10
|
||||||
|
cargo update -p aws-sdk-kms --precise 1.51.0
|
||||||
|
cargo update -p aws-sdk-s3 --precise 1.65.0
|
||||||
|
cargo update -p aws-sdk-sso --precise 1.50.0
|
||||||
|
cargo update -p aws-sdk-ssooidc --precise 1.51.0
|
||||||
|
cargo update -p aws-sdk-sts --precise 1.51.0
|
||||||
|
cargo update -p home --precise 0.5.9
|
||||||
|
- name: cargo +${{ matrix.msrv }} check
|
||||||
|
run: cargo check --workspace --tests --benches --all-features
|
||||||
|
|||||||
33
.github/workflows/update_package_lock/action.yml
vendored
33
.github/workflows/update_package_lock/action.yml
vendored
@@ -1,33 +0,0 @@
|
|||||||
name: update_package_lock
|
|
||||||
description: "Update node's package.lock"
|
|
||||||
|
|
||||||
inputs:
|
|
||||||
github_token:
|
|
||||||
required: true
|
|
||||||
description: "github token for the repo"
|
|
||||||
|
|
||||||
runs:
|
|
||||||
using: "composite"
|
|
||||||
steps:
|
|
||||||
- uses: actions/setup-node@v3
|
|
||||||
with:
|
|
||||||
node-version: 20
|
|
||||||
- name: Set git configs
|
|
||||||
shell: bash
|
|
||||||
run: |
|
|
||||||
git config user.name 'Lance Release'
|
|
||||||
git config user.email 'lance-dev@lancedb.com'
|
|
||||||
- name: Update package-lock.json file
|
|
||||||
working-directory: ./node
|
|
||||||
run: |
|
|
||||||
npm install
|
|
||||||
git add package-lock.json
|
|
||||||
git commit -m "Updating package-lock.json"
|
|
||||||
shell: bash
|
|
||||||
- name: Push changes
|
|
||||||
if: ${{ inputs.dry_run }} == "false"
|
|
||||||
uses: ad-m/github-push-action@master
|
|
||||||
with:
|
|
||||||
github_token: ${{ inputs.github_token }}
|
|
||||||
branch: main
|
|
||||||
tags: true
|
|
||||||
@@ -1,33 +0,0 @@
|
|||||||
name: update_package_lock_nodejs
|
|
||||||
description: "Update nodejs's package.lock"
|
|
||||||
|
|
||||||
inputs:
|
|
||||||
github_token:
|
|
||||||
required: true
|
|
||||||
description: "github token for the repo"
|
|
||||||
|
|
||||||
runs:
|
|
||||||
using: "composite"
|
|
||||||
steps:
|
|
||||||
- uses: actions/setup-node@v3
|
|
||||||
with:
|
|
||||||
node-version: 20
|
|
||||||
- name: Set git configs
|
|
||||||
shell: bash
|
|
||||||
run: |
|
|
||||||
git config user.name 'Lance Release'
|
|
||||||
git config user.email 'lance-dev@lancedb.com'
|
|
||||||
- name: Update package-lock.json file
|
|
||||||
working-directory: ./nodejs
|
|
||||||
run: |
|
|
||||||
npm install
|
|
||||||
git add package-lock.json
|
|
||||||
git commit -m "Updating package-lock.json"
|
|
||||||
shell: bash
|
|
||||||
- name: Push changes
|
|
||||||
if: ${{ inputs.dry_run }} == "false"
|
|
||||||
uses: ad-m/github-push-action@master
|
|
||||||
with:
|
|
||||||
github_token: ${{ inputs.github_token }}
|
|
||||||
branch: main
|
|
||||||
tags: true
|
|
||||||
5
.github/workflows/upload_wheel/action.yml
vendored
5
.github/workflows/upload_wheel/action.yml
vendored
@@ -17,11 +17,12 @@ runs:
|
|||||||
run: |
|
run: |
|
||||||
python -m pip install --upgrade pip
|
python -m pip install --upgrade pip
|
||||||
pip install twine
|
pip install twine
|
||||||
|
python3 -m pip install --upgrade pkginfo
|
||||||
- name: Choose repo
|
- name: Choose repo
|
||||||
shell: bash
|
shell: bash
|
||||||
id: choose_repo
|
id: choose_repo
|
||||||
run: |
|
run: |
|
||||||
if [ ${{ github.ref }} == "*beta*" ]; then
|
if [[ ${{ github.ref }} == *beta* ]]; then
|
||||||
echo "repo=fury" >> $GITHUB_OUTPUT
|
echo "repo=fury" >> $GITHUB_OUTPUT
|
||||||
else
|
else
|
||||||
echo "repo=pypi" >> $GITHUB_OUTPUT
|
echo "repo=pypi" >> $GITHUB_OUTPUT
|
||||||
@@ -32,7 +33,7 @@ runs:
|
|||||||
FURY_TOKEN: ${{ inputs.fury_token }}
|
FURY_TOKEN: ${{ inputs.fury_token }}
|
||||||
PYPI_TOKEN: ${{ inputs.pypi_token }}
|
PYPI_TOKEN: ${{ inputs.pypi_token }}
|
||||||
run: |
|
run: |
|
||||||
if [ ${{ steps.choose_repo.outputs.repo }} == "fury" ]; then
|
if [[ ${{ steps.choose_repo.outputs.repo }} == fury ]]; then
|
||||||
WHEEL=$(ls target/wheels/lancedb-*.whl 2> /dev/null | head -n 1)
|
WHEEL=$(ls target/wheels/lancedb-*.whl 2> /dev/null | head -n 1)
|
||||||
echo "Uploading $WHEEL to Fury"
|
echo "Uploading $WHEEL to Fury"
|
||||||
curl -f -F package=@$WHEEL https://$FURY_TOKEN@push.fury.io/lancedb/
|
curl -f -F package=@$WHEEL https://$FURY_TOKEN@push.fury.io/lancedb/
|
||||||
|
|||||||
6
.gitignore
vendored
6
.gitignore
vendored
@@ -9,7 +9,6 @@ venv
|
|||||||
.vscode
|
.vscode
|
||||||
.zed
|
.zed
|
||||||
rust/target
|
rust/target
|
||||||
rust/Cargo.lock
|
|
||||||
|
|
||||||
site
|
site
|
||||||
|
|
||||||
@@ -32,9 +31,6 @@ python/dist
|
|||||||
*.node
|
*.node
|
||||||
**/node_modules
|
**/node_modules
|
||||||
**/.DS_Store
|
**/.DS_Store
|
||||||
node/dist
|
|
||||||
node/examples/**/package-lock.json
|
|
||||||
node/examples/**/dist
|
|
||||||
nodejs/lancedb/native*
|
nodejs/lancedb/native*
|
||||||
dist
|
dist
|
||||||
|
|
||||||
@@ -42,5 +38,3 @@ dist
|
|||||||
target
|
target
|
||||||
|
|
||||||
**/sccache.log
|
**/sccache.log
|
||||||
|
|
||||||
Cargo.lock
|
|
||||||
|
|||||||
@@ -1,16 +1,22 @@
|
|||||||
repos:
|
repos:
|
||||||
- repo: https://github.com/pre-commit/pre-commit-hooks
|
- repo: https://github.com/pre-commit/pre-commit-hooks
|
||||||
rev: v3.2.0
|
rev: v3.2.0
|
||||||
hooks:
|
hooks:
|
||||||
- id: check-yaml
|
- id: check-yaml
|
||||||
- id: end-of-file-fixer
|
- id: end-of-file-fixer
|
||||||
- id: trailing-whitespace
|
- id: trailing-whitespace
|
||||||
- repo: https://github.com/astral-sh/ruff-pre-commit
|
- repo: https://github.com/astral-sh/ruff-pre-commit
|
||||||
# Ruff version.
|
# Ruff version.
|
||||||
rev: v0.2.2
|
rev: v0.9.9
|
||||||
hooks:
|
hooks:
|
||||||
- id: ruff
|
- id: ruff
|
||||||
- repo: local
|
# - repo: https://github.com/RobertCraigie/pyright-python
|
||||||
|
# rev: v1.1.395
|
||||||
|
# hooks:
|
||||||
|
# - id: pyright
|
||||||
|
# args: ["--project", "python"]
|
||||||
|
# additional_dependencies: [pyarrow-stubs]
|
||||||
|
- repo: local
|
||||||
hooks:
|
hooks:
|
||||||
- id: local-biome-check
|
- id: local-biome-check
|
||||||
name: biome check
|
name: biome check
|
||||||
|
|||||||
22
CLAUDE.md
Normal file
22
CLAUDE.md
Normal file
@@ -0,0 +1,22 @@
|
|||||||
|
LanceDB is a database designed for retrieval, including vector, full-text, and hybrid search.
|
||||||
|
It is a wrapper around Lance. There are two backends: local (in-process like SQLite) and
|
||||||
|
remote (against LanceDB Cloud).
|
||||||
|
|
||||||
|
The core of LanceDB is written in Rust. There are bindings in Python, Typescript, and Java.
|
||||||
|
|
||||||
|
Project layout:
|
||||||
|
|
||||||
|
* `rust/lancedb`: The LanceDB core Rust implementation.
|
||||||
|
* `python`: The Python bindings, using PyO3.
|
||||||
|
* `nodejs`: The Typescript bindings, using napi-rs
|
||||||
|
* `java`: The Java bindings
|
||||||
|
|
||||||
|
Common commands:
|
||||||
|
|
||||||
|
* Check for compiler errors: `cargo check --features remote --tests --examples`
|
||||||
|
* Run tests: `cargo test --features remote --tests`
|
||||||
|
* Run specific test: `cargo test --features remote -p <package_name> --test <test_name>`
|
||||||
|
* Lint: `cargo clippy --features remote --tests --examples`
|
||||||
|
* Format: `cargo fmt --all`
|
||||||
|
|
||||||
|
Before committing changes, run formatting.
|
||||||
78
CONTRIBUTING.md
Normal file
78
CONTRIBUTING.md
Normal file
@@ -0,0 +1,78 @@
|
|||||||
|
# Contributing to LanceDB
|
||||||
|
|
||||||
|
LanceDB is an open-source project and we welcome contributions from the community.
|
||||||
|
This document outlines the process for contributing to LanceDB.
|
||||||
|
|
||||||
|
## Reporting Issues
|
||||||
|
|
||||||
|
If you encounter a bug or have a feature request, please open an issue on the
|
||||||
|
[GitHub issue tracker](https://github.com/lancedb/lancedb).
|
||||||
|
|
||||||
|
## Picking an issue
|
||||||
|
|
||||||
|
We track issues on the GitHub issue tracker. If you are looking for something to
|
||||||
|
work on, check the [good first issue](https://github.com/lancedb/lancedb/contribute) label. These issues are typically the best described and have the smallest scope.
|
||||||
|
|
||||||
|
If there's an issue you are interested in working on, please leave a comment on the issue. This will help us avoid duplicate work. Additionally, if you have questions about the issue, please ask them in the issue comments. We are happy to provide guidance on how to approach the issue.
|
||||||
|
|
||||||
|
## Configuring Git
|
||||||
|
|
||||||
|
First, fork the repository on GitHub, then clone your fork:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
git clone https://github.com/<username>/lancedb.git
|
||||||
|
cd lancedb
|
||||||
|
```
|
||||||
|
|
||||||
|
Then add the main repository as a remote:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
git remote add upstream https://github.com/lancedb/lancedb.git
|
||||||
|
git fetch upstream
|
||||||
|
```
|
||||||
|
|
||||||
|
## Setting up your development environment
|
||||||
|
|
||||||
|
We have development environments for Python, Typescript, and Java. Each environment has its own setup instructions.
|
||||||
|
|
||||||
|
* [Python](python/CONTRIBUTING.md)
|
||||||
|
* [Typescript](nodejs/CONTRIBUTING.md)
|
||||||
|
<!-- TODO: add Java contributing guide -->
|
||||||
|
* [Documentation](docs/README.md)
|
||||||
|
|
||||||
|
|
||||||
|
## Best practices for pull requests
|
||||||
|
|
||||||
|
For the best chance of having your pull request accepted, please follow these guidelines:
|
||||||
|
|
||||||
|
1. Unit test all bug fixes and new features. Your code will not be merged if it
|
||||||
|
doesn't have tests.
|
||||||
|
1. If you change the public API, update the documentation in the `docs` directory.
|
||||||
|
1. Aim to minimize the number of changes in each pull request. Keep to solving
|
||||||
|
one problem at a time, when possible.
|
||||||
|
1. Before marking a pull request ready-for-review, do a self review of your code.
|
||||||
|
Is it clear why you are making the changes? Are the changes easy to understand?
|
||||||
|
1. Use [conventional commit messages](https://www.conventionalcommits.org/en/) as pull request titles. Examples:
|
||||||
|
* New feature: `feat: adding foo API`
|
||||||
|
* Bug fix: `fix: issue with foo API`
|
||||||
|
* Documentation change: `docs: adding foo API documentation`
|
||||||
|
1. If your pull request is a work in progress, leave the pull request as a draft.
|
||||||
|
We will assume the pull request is ready for review when it is opened.
|
||||||
|
1. When writing tests, test the error cases. Make sure they have understandable
|
||||||
|
error messages.
|
||||||
|
|
||||||
|
## Project structure
|
||||||
|
|
||||||
|
The core library is written in Rust. The Python, Typescript, and Java libraries
|
||||||
|
are wrappers around the Rust library.
|
||||||
|
|
||||||
|
* `src/lancedb`: Rust library source code
|
||||||
|
* `python`: Python package source code
|
||||||
|
* `nodejs`: Typescript package source code
|
||||||
|
* `node`: **Deprecated** Typescript package source code
|
||||||
|
* `java`: Java package source code
|
||||||
|
* `docs`: Documentation source code
|
||||||
|
|
||||||
|
## Release process
|
||||||
|
|
||||||
|
For information on the release process, see: [release_process.md](release_process.md)
|
||||||
8896
Cargo.lock
generated
Normal file
8896
Cargo.lock
generated
Normal file
File diff suppressed because it is too large
Load Diff
65
Cargo.toml
65
Cargo.toml
@@ -1,11 +1,5 @@
|
|||||||
[workspace]
|
[workspace]
|
||||||
members = [
|
members = ["rust/lancedb", "nodejs", "python", "java/core/lancedb-jni"]
|
||||||
"rust/ffi/node",
|
|
||||||
"rust/lancedb",
|
|
||||||
"nodejs",
|
|
||||||
"python",
|
|
||||||
"java/core/lancedb-jni",
|
|
||||||
]
|
|
||||||
# Python package needs to be built by maturin.
|
# Python package needs to be built by maturin.
|
||||||
exclude = ["python"]
|
exclude = ["python"]
|
||||||
resolver = "2"
|
resolver = "2"
|
||||||
@@ -18,35 +12,54 @@ repository = "https://github.com/lancedb/lancedb"
|
|||||||
description = "Serverless, low-latency vector database for AI applications"
|
description = "Serverless, low-latency vector database for AI applications"
|
||||||
keywords = ["lancedb", "lance", "database", "vector", "search"]
|
keywords = ["lancedb", "lance", "database", "vector", "search"]
|
||||||
categories = ["database-implementations"]
|
categories = ["database-implementations"]
|
||||||
|
rust-version = "1.78.0"
|
||||||
|
|
||||||
[workspace.dependencies]
|
[workspace.dependencies]
|
||||||
lance = { "version" = "=0.16.1", "features" = ["dynamodb"] }
|
lance = { "version" = "=0.33.0", "features" = ["dynamodb"] }
|
||||||
lance-index = { "version" = "=0.16.1" }
|
lance-io = "=0.33.0"
|
||||||
lance-linalg = { "version" = "=0.16.1" }
|
lance-index = "=0.33.0"
|
||||||
lance-testing = { "version" = "=0.16.1" }
|
lance-linalg = "=0.33.0"
|
||||||
lance-datafusion = { "version" = "=0.16.1" }
|
lance-table = "=0.33.0"
|
||||||
lance-encoding = { "version" = "=0.16.1" }
|
lance-testing = "=0.33.0"
|
||||||
|
lance-datafusion = "=0.33.0"
|
||||||
|
lance-encoding = "=0.33.0"
|
||||||
# Note that this one does not include pyarrow
|
# Note that this one does not include pyarrow
|
||||||
arrow = { version = "52.2", optional = false }
|
arrow = { version = "55.1", optional = false }
|
||||||
arrow-array = "52.2"
|
arrow-array = "55.1"
|
||||||
arrow-data = "52.2"
|
arrow-data = "55.1"
|
||||||
arrow-ipc = "52.2"
|
arrow-ipc = "55.1"
|
||||||
arrow-ord = "52.2"
|
arrow-ord = "55.1"
|
||||||
arrow-schema = "52.2"
|
arrow-schema = "55.1"
|
||||||
arrow-arith = "52.2"
|
arrow-arith = "55.1"
|
||||||
arrow-cast = "52.2"
|
arrow-cast = "55.1"
|
||||||
async-trait = "0"
|
async-trait = "0"
|
||||||
chrono = "0.4.35"
|
datafusion = { version = "48.0", default-features = false }
|
||||||
datafusion-physical-plan = "40.0"
|
datafusion-catalog = "48.0"
|
||||||
half = { "version" = "=2.4.1", default-features = false, features = [
|
datafusion-common = { version = "48.0", default-features = false }
|
||||||
|
datafusion-execution = "48.0"
|
||||||
|
datafusion-expr = "48.0"
|
||||||
|
datafusion-physical-plan = "48.0"
|
||||||
|
env_logger = "0.11"
|
||||||
|
half = { "version" = "2.6.0", default-features = false, features = [
|
||||||
"num-traits",
|
"num-traits",
|
||||||
] }
|
] }
|
||||||
futures = "0"
|
futures = "0"
|
||||||
log = "0.4"
|
log = "0.4"
|
||||||
object_store = "0.10.2"
|
moka = { version = "0.12", features = ["future"] }
|
||||||
|
object_store = "0.12.0"
|
||||||
pin-project = "1.0.7"
|
pin-project = "1.0.7"
|
||||||
snafu = "0.7.4"
|
snafu = "0.8"
|
||||||
url = "2"
|
url = "2"
|
||||||
num-traits = "0.2"
|
num-traits = "0.2"
|
||||||
|
rand = "0.9"
|
||||||
regex = "1.10"
|
regex = "1.10"
|
||||||
lazy_static = "1"
|
lazy_static = "1"
|
||||||
|
semver = "1.0.25"
|
||||||
|
crunchy = "0.2.4"
|
||||||
|
# Temporary pins to work around downstream issues
|
||||||
|
# https://github.com/apache/arrow-rs/commit/2fddf85afcd20110ce783ed5b4cdeb82293da30b
|
||||||
|
chrono = "=0.4.41"
|
||||||
|
# https://github.com/RustCrypto/formats/issues/1684
|
||||||
|
base64ct = "=1.6.0"
|
||||||
|
# Workaround for: https://github.com/Lokathor/bytemuck/issues/306
|
||||||
|
bytemuck_derive = ">=1.8.1, <1.9.0"
|
||||||
|
|||||||
126
README.md
126
README.md
@@ -1,85 +1,97 @@
|
|||||||
|
<a href="https://cloud.lancedb.com" target="_blank">
|
||||||
|
<img src="https://github.com/user-attachments/assets/92dad0a2-2a37-4ce1-b783-0d1b4f30a00c" alt="LanceDB Cloud Public Beta" width="100%" style="max-width: 100%;">
|
||||||
|
</a>
|
||||||
<div align="center">
|
<div align="center">
|
||||||
<p align="center">
|
|
||||||
|
|
||||||
<img width="275" alt="LanceDB Logo" src="https://github.com/lancedb/lancedb/assets/5846846/37d7c7ad-c2fd-4f56-9f16-fffb0d17c73a">
|
[](https://lancedb.com)
|
||||||
|
[](https://lancedb.com/)
|
||||||
|
[](https://blog.lancedb.com/)
|
||||||
|
[](https://discord.gg/zMM32dvNtd)
|
||||||
|
[](https://twitter.com/lancedb)
|
||||||
|
[](https://www.linkedin.com/company/lancedb/)
|
||||||
|
|
||||||
**Developer-friendly, database for multimodal AI**
|
|
||||||
|
|
||||||
<a href='https://github.com/lancedb/vectordb-recipes/tree/main' target="_blank"><img alt='LanceDB' src='https://img.shields.io/badge/VectorDB_Recipes-100000?style=for-the-badge&logo=LanceDB&logoColor=white&labelColor=645cfb&color=645cfb'/></a>
|
<img src="docs/src/assets/lancedb.png" alt="LanceDB" width="50%">
|
||||||
<a href='https://lancedb.github.io/lancedb/' target="_blank"><img alt='lancdb' src='https://img.shields.io/badge/DOCS-100000?style=for-the-badge&logo=lancdb&logoColor=white&labelColor=645cfb&color=645cfb'/></a>
|
|
||||||
[](https://blog.lancedb.com/)
|
|
||||||
[](https://discord.gg/zMM32dvNtd)
|
|
||||||
[](https://twitter.com/lancedb)
|
|
||||||
|
|
||||||
</p>
|
# **The Multimodal AI Lakehouse**
|
||||||
|
|
||||||
<img max-width="750px" alt="LanceDB Multimodal Search" src="https://github.com/lancedb/lancedb/assets/917119/09c5afc5-7816-4687-bae4-f2ca194426ec">
|
[**How to Install** ](#how-to-install) ✦ [**Detailed Documentation**](https://lancedb.github.io/lancedb/) ✦ [**Tutorials and Recipes**](https://github.com/lancedb/vectordb-recipes/tree/main) ✦ [**Contributors**](#contributors)
|
||||||
|
|
||||||
|
**The ultimate multimodal data platform for AI/ML applications.**
|
||||||
|
|
||||||
|
LanceDB is designed for fast, scalable, and production-ready vector search. It is built on top of the Lance columnar format. You can store, index, and search over petabytes of multimodal data and vectors with ease.
|
||||||
|
LanceDB is a central location where developers can build, train and analyze their AI workloads.
|
||||||
|
|
||||||
</p>
|
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<hr />
|
<br>
|
||||||
|
|
||||||
LanceDB is an open-source database for vector-search built with persistent storage, which greatly simplifies retrieval, filtering and management of embeddings.
|
## **Demo: Multimodal Search by Keyword, Vector or with SQL**
|
||||||
|
<img max-width="750px" alt="LanceDB Multimodal Search" src="https://github.com/lancedb/lancedb/assets/917119/09c5afc5-7816-4687-bae4-f2ca194426ec">
|
||||||
|
|
||||||
The key features of LanceDB include:
|
## **Star LanceDB to get updates!**
|
||||||
|
|
||||||
* Production-scale vector search with no servers to manage.
|
<details>
|
||||||
|
<summary>⭐ Click here ⭐ to see how fast we're growing!</summary>
|
||||||
|
<picture>
|
||||||
|
<source media="(prefers-color-scheme: dark)" srcset="https://api.star-history.com/svg?repos=lancedb/lancedb&theme=dark&type=Date">
|
||||||
|
<img width="100%" src="https://api.star-history.com/svg?repos=lancedb/lancedb&theme=dark&type=Date">
|
||||||
|
</picture>
|
||||||
|
</details>
|
||||||
|
|
||||||
* Store, query and filter vectors, metadata and multi-modal data (text, images, videos, point clouds, and more).
|
## **Key Features**:
|
||||||
|
|
||||||
* Support for vector similarity search, full-text search and SQL.
|
- **Fast Vector Search**: Search billions of vectors in milliseconds with state-of-the-art indexing.
|
||||||
|
- **Comprehensive Search**: Support for vector similarity search, full-text search and SQL.
|
||||||
|
- **Multimodal Support**: Store, query and filter vectors, metadata and multimodal data (text, images, videos, point clouds, and more).
|
||||||
|
- **Advanced Features**: Zero-copy, automatic versioning, manage versions of your data without needing extra infrastructure. GPU support in building vector index.
|
||||||
|
|
||||||
* Native Python and Javascript/Typescript support.
|
### **Products**:
|
||||||
|
- **Open Source & Local**: 100% open source, runs locally or in your cloud. No vendor lock-in.
|
||||||
|
- **Cloud and Enterprise**: Production-scale vector search with no servers to manage. Complete data sovereignty and security.
|
||||||
|
|
||||||
* Zero-copy, automatic versioning, manage versions of your data without needing extra infrastructure.
|
### **Ecosystem**:
|
||||||
|
- **Columnar Storage**: Built on the Lance columnar format for efficient storage and analytics.
|
||||||
|
- **Seamless Integration**: Python, Node.js, Rust, and REST APIs for easy integration. Native Python and Javascript/Typescript support.
|
||||||
|
- **Rich Ecosystem**: Integrations with [**LangChain** 🦜️🔗](https://python.langchain.com/docs/integrations/vectorstores/lancedb/), [**LlamaIndex** 🦙](https://gpt-index.readthedocs.io/en/latest/examples/vector_stores/LanceDBIndexDemo.html), Apache-Arrow, Pandas, Polars, DuckDB and more on the way.
|
||||||
|
|
||||||
* GPU support in building vector index(*).
|
## **How to Install**:
|
||||||
|
|
||||||
* Ecosystem integrations with [LangChain 🦜️🔗](https://python.langchain.com/docs/integrations/vectorstores/lancedb/), [LlamaIndex 🦙](https://gpt-index.readthedocs.io/en/latest/examples/vector_stores/LanceDBIndexDemo.html), Apache-Arrow, Pandas, Polars, DuckDB and more on the way.
|
Follow the [Quickstart](https://lancedb.github.io/lancedb/basic/) doc to set up LanceDB locally.
|
||||||
|
|
||||||
LanceDB's core is written in Rust 🦀 and is built using <a href="https://github.com/lancedb/lance">Lance</a>, an open-source columnar format designed for performant ML workloads.
|
**API & SDK:** We also support Python, Typescript and Rust SDKs
|
||||||
|
|
||||||
## Quick Start
|
| Interface | Documentation |
|
||||||
|
|-----------|---------------|
|
||||||
|
| Python SDK | https://lancedb.github.io/lancedb/python/python/ |
|
||||||
|
| Typescript SDK | https://lancedb.github.io/lancedb/js/globals/ |
|
||||||
|
| Rust SDK | https://docs.rs/lancedb/latest/lancedb/index.html |
|
||||||
|
| REST API | https://docs.lancedb.com/api-reference/introduction |
|
||||||
|
|
||||||
**Javascript**
|
## **Join Us and Contribute**
|
||||||
```shell
|
|
||||||
npm install @lancedb/lancedb
|
|
||||||
```
|
|
||||||
|
|
||||||
```javascript
|
We welcome contributions from everyone! Whether you're a developer, researcher, or just someone who wants to help out.
|
||||||
import * as lancedb from "@lancedb/lancedb";
|
|
||||||
|
|
||||||
const db = await lancedb.connect("data/sample-lancedb");
|
If you have any suggestions or feature requests, please feel free to open an issue on GitHub or discuss it on our [**Discord**](https://discord.gg/G5DcmnZWKB) server.
|
||||||
const table = await db.createTable("vectors", [
|
|
||||||
{ id: 1, vector: [0.1, 0.2], item: "foo", price: 10 },
|
[**Check out the GitHub Issues**](https://github.com/lancedb/lancedb/issues) if you would like to work on the features that are planned for the future. If you have any suggestions or feature requests, please feel free to open an issue on GitHub.
|
||||||
{ id: 2, vector: [1.1, 1.2], item: "bar", price: 50 },
|
|
||||||
], {mode: 'overwrite'});
|
## **Contributors**
|
||||||
|
|
||||||
|
<a href="https://github.com/lancedb/lancedb/graphs/contributors">
|
||||||
|
<img src="https://contrib.rocks/image?repo=lancedb/lancedb" />
|
||||||
|
</a>
|
||||||
|
|
||||||
|
|
||||||
const query = table.vectorSearch([0.1, 0.3]).limit(2);
|
## **Stay in Touch With Us**
|
||||||
const results = await query.toArray();
|
<div align="center">
|
||||||
|
|
||||||
// You can also search for rows by specific criteria without involving a vector search.
|
</br>
|
||||||
const rowsByCriteria = await table.query().where("price >= 10").toArray();
|
|
||||||
```
|
|
||||||
|
|
||||||
**Python**
|
[](https://lancedb.com/)
|
||||||
```shell
|
[](https://blog.lancedb.com/)
|
||||||
pip install lancedb
|
[](https://discord.gg/zMM32dvNtd)
|
||||||
```
|
[](https://twitter.com/lancedb)
|
||||||
|
[](https://www.linkedin.com/company/lancedb/)
|
||||||
|
|
||||||
```python
|
</div>
|
||||||
import lancedb
|
|
||||||
|
|
||||||
uri = "data/sample-lancedb"
|
|
||||||
db = lancedb.connect(uri)
|
|
||||||
table = db.create_table("my_table",
|
|
||||||
data=[{"vector": [3.1, 4.1], "item": "foo", "price": 10.0},
|
|
||||||
{"vector": [5.9, 26.5], "item": "bar", "price": 20.0}])
|
|
||||||
result = table.search([100, 100]).limit(2).to_pandas()
|
|
||||||
```
|
|
||||||
|
|
||||||
## Blogs, Tutorials & Videos
|
|
||||||
* 📈 <a href="https://blog.lancedb.com/benchmarking-random-access-in-lance/">2000x better performance with Lance over Parquet</a>
|
|
||||||
* 🤖 <a href="https://github.com/lancedb/lancedb/blob/main/docs/src/notebooks/youtube_transcript_search.ipynb">Build a question and answer bot with LanceDB</a>
|
|
||||||
|
|||||||
@@ -1,21 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
set -e
|
|
||||||
ARCH=${1:-x86_64}
|
|
||||||
|
|
||||||
# We pass down the current user so that when we later mount the local files
|
|
||||||
# into the container, the files are accessible by the current user.
|
|
||||||
pushd ci/manylinux_node
|
|
||||||
docker build \
|
|
||||||
-t lancedb-node-manylinux \
|
|
||||||
--build-arg="ARCH=$ARCH" \
|
|
||||||
--build-arg="DOCKER_USER=$(id -u)" \
|
|
||||||
--progress=plain \
|
|
||||||
.
|
|
||||||
popd
|
|
||||||
|
|
||||||
# We turn on memory swap to avoid OOM killer
|
|
||||||
docker run \
|
|
||||||
-v $(pwd):/io -w /io \
|
|
||||||
--memory-swap=-1 \
|
|
||||||
lancedb-node-manylinux \
|
|
||||||
bash ci/manylinux_node/build_vectordb.sh $ARCH
|
|
||||||
@@ -1,21 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
set -e
|
|
||||||
ARCH=${1:-x86_64}
|
|
||||||
|
|
||||||
# We pass down the current user so that when we later mount the local files
|
|
||||||
# into the container, the files are accessible by the current user.
|
|
||||||
pushd ci/manylinux_node
|
|
||||||
docker build \
|
|
||||||
-t lancedb-node-manylinux-$ARCH \
|
|
||||||
--build-arg="ARCH=$ARCH" \
|
|
||||||
--build-arg="DOCKER_USER=$(id -u)" \
|
|
||||||
--progress=plain \
|
|
||||||
.
|
|
||||||
popd
|
|
||||||
|
|
||||||
# We turn on memory swap to avoid OOM killer
|
|
||||||
docker run \
|
|
||||||
-v $(pwd):/io -w /io \
|
|
||||||
--memory-swap=-1 \
|
|
||||||
lancedb-node-manylinux-$ARCH \
|
|
||||||
bash ci/manylinux_node/build_lancedb.sh $ARCH
|
|
||||||
@@ -1,34 +0,0 @@
|
|||||||
# Builds the macOS artifacts (node binaries).
|
|
||||||
# Usage: ./ci/build_macos_artifacts.sh [target]
|
|
||||||
# Targets supported: x86_64-apple-darwin aarch64-apple-darwin
|
|
||||||
set -e
|
|
||||||
|
|
||||||
prebuild_rust() {
|
|
||||||
# Building here for the sake of easier debugging.
|
|
||||||
pushd rust/ffi/node
|
|
||||||
echo "Building rust library for $1"
|
|
||||||
export RUST_BACKTRACE=1
|
|
||||||
cargo build --release --target $1
|
|
||||||
popd
|
|
||||||
}
|
|
||||||
|
|
||||||
build_node_binaries() {
|
|
||||||
pushd node
|
|
||||||
echo "Building node library for $1"
|
|
||||||
npm run build-release -- --target $1
|
|
||||||
npm run pack-build -- --target $1
|
|
||||||
popd
|
|
||||||
}
|
|
||||||
|
|
||||||
if [ -n "$1" ]; then
|
|
||||||
targets=$1
|
|
||||||
else
|
|
||||||
targets="x86_64-apple-darwin aarch64-apple-darwin"
|
|
||||||
fi
|
|
||||||
|
|
||||||
echo "Building artifacts for targets: $targets"
|
|
||||||
for target in $targets
|
|
||||||
do
|
|
||||||
prebuild_rust $target
|
|
||||||
build_node_binaries $target
|
|
||||||
done
|
|
||||||
@@ -1,34 +0,0 @@
|
|||||||
# Builds the macOS artifacts (nodejs binaries).
|
|
||||||
# Usage: ./ci/build_macos_artifacts_nodejs.sh [target]
|
|
||||||
# Targets supported: x86_64-apple-darwin aarch64-apple-darwin
|
|
||||||
set -e
|
|
||||||
|
|
||||||
prebuild_rust() {
|
|
||||||
# Building here for the sake of easier debugging.
|
|
||||||
pushd rust/lancedb
|
|
||||||
echo "Building rust library for $1"
|
|
||||||
export RUST_BACKTRACE=1
|
|
||||||
cargo build --release --target $1
|
|
||||||
popd
|
|
||||||
}
|
|
||||||
|
|
||||||
build_node_binaries() {
|
|
||||||
pushd nodejs
|
|
||||||
echo "Building nodejs library for $1"
|
|
||||||
export RUST_TARGET=$1
|
|
||||||
npm run build-release
|
|
||||||
popd
|
|
||||||
}
|
|
||||||
|
|
||||||
if [ -n "$1" ]; then
|
|
||||||
targets=$1
|
|
||||||
else
|
|
||||||
targets="x86_64-apple-darwin aarch64-apple-darwin"
|
|
||||||
fi
|
|
||||||
|
|
||||||
echo "Building artifacts for targets: $targets"
|
|
||||||
for target in $targets
|
|
||||||
do
|
|
||||||
prebuild_rust $target
|
|
||||||
build_node_binaries $target
|
|
||||||
done
|
|
||||||
@@ -1,41 +0,0 @@
|
|||||||
# Builds the Windows artifacts (node binaries).
|
|
||||||
# Usage: .\ci\build_windows_artifacts.ps1 [target]
|
|
||||||
# Targets supported:
|
|
||||||
# - x86_64-pc-windows-msvc
|
|
||||||
# - i686-pc-windows-msvc
|
|
||||||
|
|
||||||
function Prebuild-Rust {
|
|
||||||
param (
|
|
||||||
[string]$target
|
|
||||||
)
|
|
||||||
|
|
||||||
# Building here for the sake of easier debugging.
|
|
||||||
Push-Location -Path "rust/ffi/node"
|
|
||||||
Write-Host "Building rust library for $target"
|
|
||||||
$env:RUST_BACKTRACE=1
|
|
||||||
cargo build --release --target $target
|
|
||||||
Pop-Location
|
|
||||||
}
|
|
||||||
|
|
||||||
function Build-NodeBinaries {
|
|
||||||
param (
|
|
||||||
[string]$target
|
|
||||||
)
|
|
||||||
|
|
||||||
Push-Location -Path "node"
|
|
||||||
Write-Host "Building node library for $target"
|
|
||||||
npm run build-release -- --target $target
|
|
||||||
npm run pack-build -- --target $target
|
|
||||||
Pop-Location
|
|
||||||
}
|
|
||||||
|
|
||||||
$targets = $args[0]
|
|
||||||
if (-not $targets) {
|
|
||||||
$targets = "x86_64-pc-windows-msvc"
|
|
||||||
}
|
|
||||||
|
|
||||||
Write-Host "Building artifacts for targets: $targets"
|
|
||||||
foreach ($target in $targets) {
|
|
||||||
Prebuild-Rust $target
|
|
||||||
Build-NodeBinaries $target
|
|
||||||
}
|
|
||||||
@@ -1,41 +0,0 @@
|
|||||||
# Builds the Windows artifacts (nodejs binaries).
|
|
||||||
# Usage: .\ci\build_windows_artifacts_nodejs.ps1 [target]
|
|
||||||
# Targets supported:
|
|
||||||
# - x86_64-pc-windows-msvc
|
|
||||||
# - i686-pc-windows-msvc
|
|
||||||
|
|
||||||
function Prebuild-Rust {
|
|
||||||
param (
|
|
||||||
[string]$target
|
|
||||||
)
|
|
||||||
|
|
||||||
# Building here for the sake of easier debugging.
|
|
||||||
Push-Location -Path "rust/lancedb"
|
|
||||||
Write-Host "Building rust library for $target"
|
|
||||||
$env:RUST_BACKTRACE=1
|
|
||||||
cargo build --release --target $target
|
|
||||||
Pop-Location
|
|
||||||
}
|
|
||||||
|
|
||||||
function Build-NodeBinaries {
|
|
||||||
param (
|
|
||||||
[string]$target
|
|
||||||
)
|
|
||||||
|
|
||||||
Push-Location -Path "nodejs"
|
|
||||||
Write-Host "Building nodejs library for $target"
|
|
||||||
$env:RUST_TARGET=$target
|
|
||||||
npm run build-release
|
|
||||||
Pop-Location
|
|
||||||
}
|
|
||||||
|
|
||||||
$targets = $args[0]
|
|
||||||
if (-not $targets) {
|
|
||||||
$targets = "x86_64-pc-windows-msvc"
|
|
||||||
}
|
|
||||||
|
|
||||||
Write-Host "Building artifacts for targets: $targets"
|
|
||||||
foreach ($target in $targets) {
|
|
||||||
Prebuild-Rust $target
|
|
||||||
Build-NodeBinaries $target
|
|
||||||
}
|
|
||||||
@@ -1,31 +0,0 @@
|
|||||||
# Many linux dockerfile with Rust, Node, and Lance dependencies installed.
|
|
||||||
# This container allows building the node modules native libraries in an
|
|
||||||
# environment with a very old glibc, so that we are compatible with a wide
|
|
||||||
# range of linux distributions.
|
|
||||||
ARG ARCH=x86_64
|
|
||||||
|
|
||||||
FROM quay.io/pypa/manylinux_2_28_${ARCH}
|
|
||||||
|
|
||||||
ARG ARCH=x86_64
|
|
||||||
ARG DOCKER_USER=default_user
|
|
||||||
|
|
||||||
# Install static openssl
|
|
||||||
COPY install_openssl.sh install_openssl.sh
|
|
||||||
RUN ./install_openssl.sh ${ARCH} > /dev/null
|
|
||||||
|
|
||||||
# Protobuf is also installed as root.
|
|
||||||
COPY install_protobuf.sh install_protobuf.sh
|
|
||||||
RUN ./install_protobuf.sh ${ARCH}
|
|
||||||
|
|
||||||
ENV DOCKER_USER=${DOCKER_USER}
|
|
||||||
# Create a group and user, but only if it doesn't exist
|
|
||||||
RUN echo ${ARCH} && id -u ${DOCKER_USER} >/dev/null 2>&1 || adduser --user-group --create-home --uid ${DOCKER_USER} build_user
|
|
||||||
|
|
||||||
# We switch to the user to install Rust and Node, since those like to be
|
|
||||||
# installed at the user level.
|
|
||||||
USER ${DOCKER_USER}
|
|
||||||
|
|
||||||
COPY prepare_manylinux_node.sh prepare_manylinux_node.sh
|
|
||||||
RUN cp /prepare_manylinux_node.sh $HOME/ && \
|
|
||||||
cd $HOME && \
|
|
||||||
./prepare_manylinux_node.sh ${ARCH}
|
|
||||||
@@ -1,18 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
# Builds the nodejs module for manylinux. Invoked by ci/build_linux_artifacts_nodejs.sh.
|
|
||||||
set -e
|
|
||||||
ARCH=${1:-x86_64}
|
|
||||||
|
|
||||||
if [ "$ARCH" = "x86_64" ]; then
|
|
||||||
export OPENSSL_LIB_DIR=/usr/local/lib64/
|
|
||||||
else
|
|
||||||
export OPENSSL_LIB_DIR=/usr/local/lib/
|
|
||||||
fi
|
|
||||||
export OPENSSL_STATIC=1
|
|
||||||
export OPENSSL_INCLUDE_DIR=/usr/local/include/openssl
|
|
||||||
|
|
||||||
source $HOME/.bashrc
|
|
||||||
|
|
||||||
cd nodejs
|
|
||||||
npm ci
|
|
||||||
npm run build-release
|
|
||||||
@@ -1,19 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
# Builds the node module for manylinux. Invoked by ci/build_linux_artifacts.sh.
|
|
||||||
set -e
|
|
||||||
ARCH=${1:-x86_64}
|
|
||||||
|
|
||||||
if [ "$ARCH" = "x86_64" ]; then
|
|
||||||
export OPENSSL_LIB_DIR=/usr/local/lib64/
|
|
||||||
else
|
|
||||||
export OPENSSL_LIB_DIR=/usr/local/lib/
|
|
||||||
fi
|
|
||||||
export OPENSSL_STATIC=1
|
|
||||||
export OPENSSL_INCLUDE_DIR=/usr/local/include/openssl
|
|
||||||
|
|
||||||
source $HOME/.bashrc
|
|
||||||
|
|
||||||
cd node
|
|
||||||
npm ci
|
|
||||||
npm run build-release
|
|
||||||
npm run pack-build
|
|
||||||
@@ -1,26 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
# Builds openssl from source so we can statically link to it
|
|
||||||
|
|
||||||
# this is to avoid the error we get with the system installation:
|
|
||||||
# /usr/bin/ld: <library>: version node not found for symbol SSLeay@@OPENSSL_1.0.1
|
|
||||||
# /usr/bin/ld: failed to set dynamic section sizes: Bad value
|
|
||||||
set -e
|
|
||||||
|
|
||||||
git clone -b OpenSSL_1_1_1v \
|
|
||||||
--single-branch \
|
|
||||||
https://github.com/openssl/openssl.git
|
|
||||||
|
|
||||||
pushd openssl
|
|
||||||
|
|
||||||
if [[ $1 == x86_64* ]]; then
|
|
||||||
ARCH=linux-x86_64
|
|
||||||
else
|
|
||||||
# gnu target
|
|
||||||
ARCH=linux-aarch64
|
|
||||||
fi
|
|
||||||
|
|
||||||
./Configure no-shared $ARCH
|
|
||||||
|
|
||||||
make
|
|
||||||
|
|
||||||
make install
|
|
||||||
@@ -1,15 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
# Installs protobuf compiler. Should be run as root.
|
|
||||||
set -e
|
|
||||||
|
|
||||||
if [[ $1 == x86_64* ]]; then
|
|
||||||
ARCH=x86_64
|
|
||||||
else
|
|
||||||
# gnu target
|
|
||||||
ARCH=aarch_64
|
|
||||||
fi
|
|
||||||
|
|
||||||
PB_REL=https://github.com/protocolbuffers/protobuf/releases
|
|
||||||
PB_VERSION=23.1
|
|
||||||
curl -LO $PB_REL/download/v$PB_VERSION/protoc-$PB_VERSION-linux-$ARCH.zip
|
|
||||||
unzip protoc-$PB_VERSION-linux-$ARCH.zip -d /usr/local
|
|
||||||
@@ -1,21 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
set -e
|
|
||||||
|
|
||||||
install_node() {
|
|
||||||
echo "Installing node..."
|
|
||||||
|
|
||||||
curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.34.0/install.sh | bash
|
|
||||||
|
|
||||||
source "$HOME"/.bashrc
|
|
||||||
|
|
||||||
nvm install --no-progress 18
|
|
||||||
}
|
|
||||||
|
|
||||||
install_rust() {
|
|
||||||
echo "Installing rust..."
|
|
||||||
curl https://sh.rustup.rs -sSf | bash -s -- -y
|
|
||||||
export PATH="$PATH:/root/.cargo/bin"
|
|
||||||
}
|
|
||||||
|
|
||||||
install_node
|
|
||||||
install_rust
|
|
||||||
57
ci/mock_openai.py
Normal file
57
ci/mock_openai.py
Normal file
@@ -0,0 +1,57 @@
|
|||||||
|
# SPDX-License-Identifier: Apache-2.0
|
||||||
|
# SPDX-FileCopyrightText: Copyright The LanceDB Authors
|
||||||
|
"""A zero-dependency mock OpenAI embeddings API endpoint for testing purposes."""
|
||||||
|
import argparse
|
||||||
|
import json
|
||||||
|
import http.server
|
||||||
|
|
||||||
|
|
||||||
|
class MockOpenAIRequestHandler(http.server.BaseHTTPRequestHandler):
|
||||||
|
def do_POST(self):
|
||||||
|
content_length = int(self.headers["Content-Length"])
|
||||||
|
post_data = self.rfile.read(content_length)
|
||||||
|
post_data = json.loads(post_data.decode("utf-8"))
|
||||||
|
# See: https://platform.openai.com/docs/api-reference/embeddings/create
|
||||||
|
|
||||||
|
if isinstance(post_data["input"], str):
|
||||||
|
num_inputs = 1
|
||||||
|
else:
|
||||||
|
num_inputs = len(post_data["input"])
|
||||||
|
|
||||||
|
model = post_data.get("model", "text-embedding-ada-002")
|
||||||
|
|
||||||
|
data = []
|
||||||
|
for i in range(num_inputs):
|
||||||
|
data.append({
|
||||||
|
"object": "embedding",
|
||||||
|
"embedding": [0.1] * 1536,
|
||||||
|
"index": i,
|
||||||
|
})
|
||||||
|
|
||||||
|
response = {
|
||||||
|
"object": "list",
|
||||||
|
"data": data,
|
||||||
|
"model": model,
|
||||||
|
"usage": {
|
||||||
|
"prompt_tokens": 0,
|
||||||
|
"total_tokens": 0,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
self.send_response(200)
|
||||||
|
self.send_header("Content-type", "application/json")
|
||||||
|
self.end_headers()
|
||||||
|
self.wfile.write(json.dumps(response).encode("utf-8"))
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
parser = argparse.ArgumentParser(description="Mock OpenAI embeddings API endpoint")
|
||||||
|
parser.add_argument("--port", type=int, default=8000, help="Port to listen on")
|
||||||
|
args = parser.parse_args()
|
||||||
|
port = args.port
|
||||||
|
|
||||||
|
print(f"server started on port {port}. Press Ctrl-C to stop.")
|
||||||
|
print(f"To use, set OPENAI_BASE_URL=http://localhost:{port} in your environment.")
|
||||||
|
|
||||||
|
with http.server.HTTPServer(("0.0.0.0", port), MockOpenAIRequestHandler) as server:
|
||||||
|
server.serve_forever()
|
||||||
41
ci/parse_requirements.py
Normal file
41
ci/parse_requirements.py
Normal file
@@ -0,0 +1,41 @@
|
|||||||
|
import argparse
|
||||||
|
import toml
|
||||||
|
|
||||||
|
|
||||||
|
def parse_dependencies(pyproject_path, extras=None):
|
||||||
|
with open(pyproject_path, "r") as file:
|
||||||
|
pyproject = toml.load(file)
|
||||||
|
|
||||||
|
dependencies = pyproject.get("project", {}).get("dependencies", [])
|
||||||
|
for dependency in dependencies:
|
||||||
|
print(dependency)
|
||||||
|
|
||||||
|
optional_dependencies = pyproject.get("project", {}).get(
|
||||||
|
"optional-dependencies", {}
|
||||||
|
)
|
||||||
|
|
||||||
|
if extras:
|
||||||
|
for extra in extras.split(","):
|
||||||
|
for dep in optional_dependencies.get(extra, []):
|
||||||
|
print(dep)
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
parser = argparse.ArgumentParser(
|
||||||
|
description="Generate requirements.txt from pyproject.toml"
|
||||||
|
)
|
||||||
|
parser.add_argument("path", type=str, help="Path to pyproject.toml")
|
||||||
|
parser.add_argument(
|
||||||
|
"--extras",
|
||||||
|
type=str,
|
||||||
|
help="Comma-separated list of extras to include",
|
||||||
|
default="",
|
||||||
|
)
|
||||||
|
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
parse_dependencies(args.path, args.extras)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
188
ci/set_lance_version.py
Normal file
188
ci/set_lance_version.py
Normal file
@@ -0,0 +1,188 @@
|
|||||||
|
import argparse
|
||||||
|
import sys
|
||||||
|
import json
|
||||||
|
|
||||||
|
|
||||||
|
def run_command(command: str) -> str:
|
||||||
|
"""
|
||||||
|
Run a shell command and return stdout as a string.
|
||||||
|
If exit code is not 0, raise an exception with the stderr output.
|
||||||
|
"""
|
||||||
|
import subprocess
|
||||||
|
|
||||||
|
result = subprocess.run(command, shell=True, capture_output=True, text=True)
|
||||||
|
if result.returncode != 0:
|
||||||
|
raise Exception(f"Command failed with error: {result.stderr.strip()}")
|
||||||
|
return result.stdout.strip()
|
||||||
|
|
||||||
|
|
||||||
|
def get_latest_stable_version() -> str:
|
||||||
|
version_line = run_command("cargo info lance | grep '^version:'")
|
||||||
|
version = version_line.split(" ")[1].strip()
|
||||||
|
return version
|
||||||
|
|
||||||
|
|
||||||
|
def get_latest_preview_version() -> str:
|
||||||
|
lance_tags = run_command(
|
||||||
|
"git ls-remote --tags https://github.com/lancedb/lance.git | grep 'refs/tags/v[0-9beta.-]\\+$'"
|
||||||
|
).splitlines()
|
||||||
|
lance_tags = (
|
||||||
|
tag.split("refs/tags/")[1]
|
||||||
|
for tag in lance_tags
|
||||||
|
if "refs/tags/" in tag and "beta" in tag
|
||||||
|
)
|
||||||
|
from packaging.version import Version
|
||||||
|
|
||||||
|
latest = max(
|
||||||
|
(tag[1:] for tag in lance_tags if tag.startswith("v")), key=lambda t: Version(t)
|
||||||
|
)
|
||||||
|
return str(latest)
|
||||||
|
|
||||||
|
|
||||||
|
def extract_features(line: str) -> list:
|
||||||
|
"""
|
||||||
|
Extracts the features from a line in Cargo.toml.
|
||||||
|
Example: 'lance = { "version" = "=0.29.0", "features" = ["dynamodb"] }'
|
||||||
|
Returns: ['dynamodb']
|
||||||
|
"""
|
||||||
|
import re
|
||||||
|
|
||||||
|
match = re.search(r'"features"\s*=\s*\[\s*(.*?)\s*\]', line, re.DOTALL)
|
||||||
|
if match:
|
||||||
|
features_str = match.group(1)
|
||||||
|
return [f.strip('"') for f in features_str.split(",") if len(f) > 0]
|
||||||
|
return []
|
||||||
|
|
||||||
|
|
||||||
|
def update_cargo_toml(line_updater):
|
||||||
|
"""
|
||||||
|
Updates the Cargo.toml file by applying the line_updater function to each line.
|
||||||
|
The line_updater function should take a line as input and return the updated line.
|
||||||
|
"""
|
||||||
|
with open("Cargo.toml", "r") as f:
|
||||||
|
lines = f.readlines()
|
||||||
|
|
||||||
|
new_lines = []
|
||||||
|
lance_line = ""
|
||||||
|
is_parsing_lance_line = False
|
||||||
|
for line in lines:
|
||||||
|
if line.startswith("lance"):
|
||||||
|
# Update the line using the provided function
|
||||||
|
if line.strip().endswith("}"):
|
||||||
|
new_lines.append(line_updater(line))
|
||||||
|
else:
|
||||||
|
lance_line = line
|
||||||
|
is_parsing_lance_line = True
|
||||||
|
elif is_parsing_lance_line:
|
||||||
|
lance_line += line
|
||||||
|
if line.strip().endswith("}"):
|
||||||
|
new_lines.append(line_updater(lance_line))
|
||||||
|
lance_line = ""
|
||||||
|
is_parsing_lance_line = False
|
||||||
|
else:
|
||||||
|
print("doesn't end with }:", line)
|
||||||
|
else:
|
||||||
|
# Keep the line unchanged
|
||||||
|
new_lines.append(line)
|
||||||
|
|
||||||
|
with open("Cargo.toml", "w") as f:
|
||||||
|
f.writelines(new_lines)
|
||||||
|
|
||||||
|
|
||||||
|
def set_stable_version(version: str):
|
||||||
|
"""
|
||||||
|
Sets lines to
|
||||||
|
lance = { "version" = "=0.29.0", "features" = ["dynamodb"] }
|
||||||
|
lance-io = "=0.29.0"
|
||||||
|
...
|
||||||
|
"""
|
||||||
|
|
||||||
|
def line_updater(line: str) -> str:
|
||||||
|
package_name = line.split("=", maxsplit=1)[0].strip()
|
||||||
|
features = extract_features(line)
|
||||||
|
if features:
|
||||||
|
return f'{package_name} = {{ "version" = "={version}", "features" = {json.dumps(features)} }}\n'
|
||||||
|
else:
|
||||||
|
return f'{package_name} = "={version}"\n'
|
||||||
|
|
||||||
|
update_cargo_toml(line_updater)
|
||||||
|
|
||||||
|
|
||||||
|
def set_preview_version(version: str):
|
||||||
|
"""
|
||||||
|
Sets lines to
|
||||||
|
lance = { "version" = "=0.29.0", "features" = ["dynamodb"], tag = "v0.29.0-beta.2", git="https://github.com/lancedb/lance.git" }
|
||||||
|
lance-io = { version = "=0.29.0", tag = "v0.29.0-beta.2", git="https://github.com/lancedb/lance.git" }
|
||||||
|
...
|
||||||
|
"""
|
||||||
|
|
||||||
|
def line_updater(line: str) -> str:
|
||||||
|
package_name = line.split("=", maxsplit=1)[0].strip()
|
||||||
|
features = extract_features(line)
|
||||||
|
base_version = version.split("-")[0] # Get the base version without beta suffix
|
||||||
|
if features:
|
||||||
|
return f'{package_name} = {{ "version" = "={base_version}", "features" = {json.dumps(features)}, "tag" = "v{version}", "git" = "https://github.com/lancedb/lance.git" }}\n'
|
||||||
|
else:
|
||||||
|
return f'{package_name} = {{ "version" = "={base_version}", "tag" = "v{version}", "git" = "https://github.com/lancedb/lance.git" }}\n'
|
||||||
|
|
||||||
|
update_cargo_toml(line_updater)
|
||||||
|
|
||||||
|
|
||||||
|
def set_local_version():
|
||||||
|
"""
|
||||||
|
Sets lines to
|
||||||
|
lance = { path = "../lance/rust/lance", features = ["dynamodb"] }
|
||||||
|
lance-io = { path = "../lance/rust/lance-io" }
|
||||||
|
...
|
||||||
|
"""
|
||||||
|
|
||||||
|
def line_updater(line: str) -> str:
|
||||||
|
package_name = line.split("=", maxsplit=1)[0].strip()
|
||||||
|
features = extract_features(line)
|
||||||
|
if features:
|
||||||
|
return f'{package_name} = {{ "path" = "../lance/rust/{package_name}", "features" = {json.dumps(features)} }}\n'
|
||||||
|
else:
|
||||||
|
return f'{package_name} = {{ "path" = "../lance/rust/{package_name}" }}\n'
|
||||||
|
|
||||||
|
update_cargo_toml(line_updater)
|
||||||
|
|
||||||
|
|
||||||
|
parser = argparse.ArgumentParser(description="Set the version of the Lance package.")
|
||||||
|
parser.add_argument(
|
||||||
|
"version",
|
||||||
|
type=str,
|
||||||
|
help="The version to set for the Lance package. Use 'stable' for the latest stable version, 'preview' for latest preview version, or a specific version number (e.g., '0.1.0'). You can also specify 'local' to use a local path.",
|
||||||
|
)
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
if args.version == "stable":
|
||||||
|
latest_stable_version = get_latest_stable_version()
|
||||||
|
print(
|
||||||
|
f"Found latest stable version: \033[1mv{latest_stable_version}\033[0m",
|
||||||
|
file=sys.stderr,
|
||||||
|
)
|
||||||
|
set_stable_version(latest_stable_version)
|
||||||
|
elif args.version == "preview":
|
||||||
|
latest_preview_version = get_latest_preview_version()
|
||||||
|
print(
|
||||||
|
f"Found latest preview version: \033[1mv{latest_preview_version}\033[0m",
|
||||||
|
file=sys.stderr,
|
||||||
|
)
|
||||||
|
set_preview_version(latest_preview_version)
|
||||||
|
elif args.version == "local":
|
||||||
|
set_local_version()
|
||||||
|
else:
|
||||||
|
# Parse the version number.
|
||||||
|
version = args.version
|
||||||
|
# Ignore initial v if present.
|
||||||
|
if version.startswith("v"):
|
||||||
|
version = version[1:]
|
||||||
|
|
||||||
|
if "beta" in version:
|
||||||
|
set_preview_version(version)
|
||||||
|
else:
|
||||||
|
set_stable_version(version)
|
||||||
|
|
||||||
|
print("Updating lockfiles...", file=sys.stderr, end="")
|
||||||
|
run_command("cargo metadata > /dev/null")
|
||||||
|
print(" done.", file=sys.stderr)
|
||||||
105
ci/sysroot-aarch64-pc-windows-msvc.sh
Normal file
105
ci/sysroot-aarch64-pc-windows-msvc.sh
Normal file
@@ -0,0 +1,105 @@
|
|||||||
|
#!/bin/sh
|
||||||
|
|
||||||
|
# https://github.com/mstorsjo/msvc-wine/blob/master/vsdownload.py
|
||||||
|
# https://github.com/mozilla/gecko-dev/blob/6027d1d91f2d3204a3992633b3ef730ff005fc64/build/vs/vs2022-car.yaml
|
||||||
|
|
||||||
|
# function dl() {
|
||||||
|
# curl -O https://download.visualstudio.microsoft.com/download/pr/$1
|
||||||
|
# }
|
||||||
|
|
||||||
|
# [[.h]]
|
||||||
|
|
||||||
|
# "id": "Win11SDK_10.0.26100"
|
||||||
|
# "version": "10.0.26100.7"
|
||||||
|
|
||||||
|
# libucrt.lib
|
||||||
|
|
||||||
|
# example: <assert.h>
|
||||||
|
# dir: ucrt/
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/2ee3a5fc6e9fc832af7295b138e93839/universal%20crt%20headers%20libraries%20and%20sources-x86_en-us.msi
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/b1aa09b90fe314aceb090f6ec7626624/16ab2ea2187acffa6435e334796c8c89.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/400609bb0ff5804e36dbe6dcd42a7f01/6ee7bbee8435130a869cf971694fd9e2.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/2ac327317abb865a0e3f56b2faefa918/78fa3c824c2c48bd4a49ab5969adaaf7.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/f034bc0b2680f67dccd4bfeea3d0f932/7afc7b670accd8e3cc94cfffd516f5cb.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/7ed5e12f9d50f80825a8b27838cf4c7f/96076045170fe5db6d5dcf14b6f6688e.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/764edc185a696bda9e07df8891dddbbb/a1e2a83aa8a71c48c742eeaff6e71928.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/66854bedc6dbd5ccb5dd82c8e2412231/b2f03f34ff83ec013b9e45c7cd8e8a73.cab
|
||||||
|
|
||||||
|
# example: <windows.h>
|
||||||
|
# dir: um/
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/b286efac4d83a54fc49190bddef1edc9/windows%20sdk%20for%20windows%20store%20apps%20headers-x86_en-us.msi
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/e0dc3811d92ab96fcb72bf63d6c08d71/766c0ffd568bbb31bf7fb6793383e24a.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/613503da4b5628768497822826aed39f/8125ee239710f33ea485965f76fae646.cab
|
||||||
|
|
||||||
|
# example: <winapifamily.h>
|
||||||
|
# dir: /shared
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/122979f0348d3a2a36b6aa1a111d5d0c/windows%20sdk%20for%20windows%20store%20apps%20headers%20onecoreuap-x86_en-us.msi
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/766e04beecdfccff39e91dd9eb32834a/e89e3dcbb016928c7e426238337d69eb.cab
|
||||||
|
|
||||||
|
|
||||||
|
# "id": "Microsoft.VisualC.14.16.CRT.Headers"
|
||||||
|
# "version": "14.16.27045"
|
||||||
|
|
||||||
|
# example: <vcruntime.h>
|
||||||
|
# dir: MSVC/
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/bac0afd7-cc9e-4182-8a83-9898fa20e092/87bbe41e09a2f83711e72696f49681429327eb7a4b90618c35667a6ba2e2880e/Microsoft.VisualC.14.16.CRT.Headers.vsix
|
||||||
|
|
||||||
|
# [[.lib]]
|
||||||
|
|
||||||
|
# advapi32.lib bcrypt.lib kernel32.lib ntdll.lib user32.lib uuid.lib ws2_32.lib userenv.lib cfgmgr32.lib runtimeobject.lib
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/944c4153b849a1f7d0c0404a4f1c05ea/windows%20sdk%20for%20windows%20store%20apps%20libs-x86_en-us.msi
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/5306aed3e1a38d1e8bef5934edeb2a9b/05047a45609f311645eebcac2739fc4c.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/13c8a73a0f5a6474040b26d016a26fab/13d68b8a7b6678a368e2d13ff4027521.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/149578fb3b621cdb61ee1813b9b3e791/463ad1b0783ebda908fd6c16a4abfe93.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/5c986c4f393c6b09d5aec3b539e9fb4a/5a22e5cde814b041749fb271547f4dd5.cab
|
||||||
|
|
||||||
|
# dbghelp.lib fwpuclnt.lib arm64rt.lib
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/7a332420d812f7c1d41da865ae5a7c52/windows%20sdk%20desktop%20libs%20arm64-x86_en-us.msi
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/19de98ed4a79938d0045d19c047936b3/3e2f7be479e3679d700ce0782e4cc318.cab
|
||||||
|
|
||||||
|
# libcmt.lib libvcruntime.lib
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/bac0afd7-cc9e-4182-8a83-9898fa20e092/227f40682a88dc5fa0ccb9cadc9ad30af99ad1f1a75db63407587d079f60d035/Microsoft.VisualC.14.16.CRT.ARM64.Desktop.vsix
|
||||||
|
|
||||||
|
|
||||||
|
msiextract universal%20crt%20headers%20libraries%20and%20sources-x86_en-us.msi
|
||||||
|
msiextract windows%20sdk%20for%20windows%20store%20apps%20headers-x86_en-us.msi
|
||||||
|
msiextract windows%20sdk%20for%20windows%20store%20apps%20headers%20onecoreuap-x86_en-us.msi
|
||||||
|
msiextract windows%20sdk%20for%20windows%20store%20apps%20libs-x86_en-us.msi
|
||||||
|
msiextract windows%20sdk%20desktop%20libs%20arm64-x86_en-us.msi
|
||||||
|
unzip -o Microsoft.VisualC.14.16.CRT.Headers.vsix
|
||||||
|
unzip -o Microsoft.VisualC.14.16.CRT.ARM64.Desktop.vsix
|
||||||
|
|
||||||
|
mkdir -p /usr/aarch64-pc-windows-msvc/usr/include
|
||||||
|
mkdir -p /usr/aarch64-pc-windows-msvc/usr/lib
|
||||||
|
|
||||||
|
# lowercase folder/file names
|
||||||
|
echo "$(find . -regex ".*/[^/]*[A-Z][^/]*")" | xargs -I{} sh -c 'mv "$(echo "{}" | sed -E '"'"'s/(.*\/)/\L\1/'"'"')" "$(echo "{}" | tr [A-Z] [a-z])"'
|
||||||
|
|
||||||
|
# .h
|
||||||
|
(cd 'program files/windows kits/10/include/10.0.26100.0' && cp -r ucrt/* um/* shared/* -t /usr/aarch64-pc-windows-msvc/usr/include)
|
||||||
|
|
||||||
|
cp -r contents/vc/tools/msvc/14.16.27023/include/* /usr/aarch64-pc-windows-msvc/usr/include
|
||||||
|
|
||||||
|
# lowercase #include "" and #include <>
|
||||||
|
find /usr/aarch64-pc-windows-msvc/usr/include -type f -exec sed -i -E 's/(#include <[^<>]*?[A-Z][^<>]*?>)|(#include "[^"]*?[A-Z][^"]*?")/\L\1\2/' "{}" ';'
|
||||||
|
|
||||||
|
# ARM intrinsics
|
||||||
|
# original dir: MSVC/
|
||||||
|
|
||||||
|
# '__n128x4' redefined in arm_neon.h
|
||||||
|
# "arm64_neon.h" included from intrin.h
|
||||||
|
|
||||||
|
(cd /usr/lib/llvm19/lib/clang/19/include && cp arm_neon.h intrin.h -t /usr/aarch64-pc-windows-msvc/usr/include)
|
||||||
|
|
||||||
|
# .lib
|
||||||
|
|
||||||
|
# _Interlocked intrinsics
|
||||||
|
# must always link with arm64rt.lib
|
||||||
|
# reason: https://developercommunity.visualstudio.com/t/libucrtlibstreamobj-error-lnk2001-unresolved-exter/1544787#T-ND1599818
|
||||||
|
# I don't understand the 'correct' fix for this, arm64rt.lib is supposed to be the workaround
|
||||||
|
|
||||||
|
(cd 'program files/windows kits/10/lib/10.0.26100.0/um/arm64' && cp advapi32.lib bcrypt.lib kernel32.lib ntdll.lib user32.lib uuid.lib ws2_32.lib userenv.lib cfgmgr32.lib runtimeobject.lib dbghelp.lib fwpuclnt.lib arm64rt.lib -t /usr/aarch64-pc-windows-msvc/usr/lib)
|
||||||
|
|
||||||
|
(cd 'contents/vc/tools/msvc/14.16.27023/lib/arm64' && cp libcmt.lib libvcruntime.lib -t /usr/aarch64-pc-windows-msvc/usr/lib)
|
||||||
|
|
||||||
|
cp 'program files/windows kits/10/lib/10.0.26100.0/ucrt/arm64/libucrt.lib' /usr/aarch64-pc-windows-msvc/usr/lib
|
||||||
105
ci/sysroot-x86_64-pc-windows-msvc.sh
Normal file
105
ci/sysroot-x86_64-pc-windows-msvc.sh
Normal file
@@ -0,0 +1,105 @@
|
|||||||
|
#!/bin/sh
|
||||||
|
|
||||||
|
# https://github.com/mstorsjo/msvc-wine/blob/master/vsdownload.py
|
||||||
|
# https://github.com/mozilla/gecko-dev/blob/6027d1d91f2d3204a3992633b3ef730ff005fc64/build/vs/vs2022-car.yaml
|
||||||
|
|
||||||
|
# function dl() {
|
||||||
|
# curl -O https://download.visualstudio.microsoft.com/download/pr/$1
|
||||||
|
# }
|
||||||
|
|
||||||
|
# [[.h]]
|
||||||
|
|
||||||
|
# "id": "Win11SDK_10.0.26100"
|
||||||
|
# "version": "10.0.26100.7"
|
||||||
|
|
||||||
|
# libucrt.lib
|
||||||
|
|
||||||
|
# example: <assert.h>
|
||||||
|
# dir: ucrt/
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/2ee3a5fc6e9fc832af7295b138e93839/universal%20crt%20headers%20libraries%20and%20sources-x86_en-us.msi
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/b1aa09b90fe314aceb090f6ec7626624/16ab2ea2187acffa6435e334796c8c89.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/400609bb0ff5804e36dbe6dcd42a7f01/6ee7bbee8435130a869cf971694fd9e2.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/2ac327317abb865a0e3f56b2faefa918/78fa3c824c2c48bd4a49ab5969adaaf7.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/f034bc0b2680f67dccd4bfeea3d0f932/7afc7b670accd8e3cc94cfffd516f5cb.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/7ed5e12f9d50f80825a8b27838cf4c7f/96076045170fe5db6d5dcf14b6f6688e.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/764edc185a696bda9e07df8891dddbbb/a1e2a83aa8a71c48c742eeaff6e71928.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/66854bedc6dbd5ccb5dd82c8e2412231/b2f03f34ff83ec013b9e45c7cd8e8a73.cab
|
||||||
|
|
||||||
|
# example: <windows.h>
|
||||||
|
# dir: um/
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/b286efac4d83a54fc49190bddef1edc9/windows%20sdk%20for%20windows%20store%20apps%20headers-x86_en-us.msi
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/e0dc3811d92ab96fcb72bf63d6c08d71/766c0ffd568bbb31bf7fb6793383e24a.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/613503da4b5628768497822826aed39f/8125ee239710f33ea485965f76fae646.cab
|
||||||
|
|
||||||
|
# example: <winapifamily.h>
|
||||||
|
# dir: /shared
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/122979f0348d3a2a36b6aa1a111d5d0c/windows%20sdk%20for%20windows%20store%20apps%20headers%20onecoreuap-x86_en-us.msi
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/766e04beecdfccff39e91dd9eb32834a/e89e3dcbb016928c7e426238337d69eb.cab
|
||||||
|
|
||||||
|
|
||||||
|
# "id": "Microsoft.VisualC.14.16.CRT.Headers"
|
||||||
|
# "version": "14.16.27045"
|
||||||
|
|
||||||
|
# example: <vcruntime.h>
|
||||||
|
# dir: MSVC/
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/bac0afd7-cc9e-4182-8a83-9898fa20e092/87bbe41e09a2f83711e72696f49681429327eb7a4b90618c35667a6ba2e2880e/Microsoft.VisualC.14.16.CRT.Headers.vsix
|
||||||
|
|
||||||
|
# [[.lib]]
|
||||||
|
|
||||||
|
# advapi32.lib bcrypt.lib kernel32.lib ntdll.lib user32.lib uuid.lib ws2_32.lib userenv.lib cfgmgr32.lib
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/944c4153b849a1f7d0c0404a4f1c05ea/windows%20sdk%20for%20windows%20store%20apps%20libs-x86_en-us.msi
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/5306aed3e1a38d1e8bef5934edeb2a9b/05047a45609f311645eebcac2739fc4c.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/13c8a73a0f5a6474040b26d016a26fab/13d68b8a7b6678a368e2d13ff4027521.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/149578fb3b621cdb61ee1813b9b3e791/463ad1b0783ebda908fd6c16a4abfe93.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/5c986c4f393c6b09d5aec3b539e9fb4a/5a22e5cde814b041749fb271547f4dd5.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/bfc3904a0195453419ae4dfea7abd6fb/e10768bb6e9d0ea730280336b697da66.cab
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/637f9f3be880c71f9e3ca07b4d67345c/f9b24c8280986c0683fbceca5326d806.cab
|
||||||
|
|
||||||
|
# dbghelp.lib fwpuclnt.lib
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/9f51690d5aa804b1340ce12d1ec80f89/windows%20sdk%20desktop%20libs%20x64-x86_en-us.msi
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/32863b8d-a46d-4231-8e84-0888519d20a9/d3a7df4ca3303a698640a29e558a5e5b/58314d0646d7e1a25e97c902166c3155.cab
|
||||||
|
|
||||||
|
# libcmt.lib libvcruntime.lib
|
||||||
|
curl -O https://download.visualstudio.microsoft.com/download/pr/bac0afd7-cc9e-4182-8a83-9898fa20e092/8728f21ae09940f1f4b4ee47b4a596be2509e2a47d2f0c83bbec0ea37d69644b/Microsoft.VisualC.14.16.CRT.x64.Desktop.vsix
|
||||||
|
|
||||||
|
|
||||||
|
msiextract universal%20crt%20headers%20libraries%20and%20sources-x86_en-us.msi
|
||||||
|
msiextract windows%20sdk%20for%20windows%20store%20apps%20headers-x86_en-us.msi
|
||||||
|
msiextract windows%20sdk%20for%20windows%20store%20apps%20headers%20onecoreuap-x86_en-us.msi
|
||||||
|
msiextract windows%20sdk%20for%20windows%20store%20apps%20libs-x86_en-us.msi
|
||||||
|
msiextract windows%20sdk%20desktop%20libs%20x64-x86_en-us.msi
|
||||||
|
unzip -o Microsoft.VisualC.14.16.CRT.Headers.vsix
|
||||||
|
unzip -o Microsoft.VisualC.14.16.CRT.x64.Desktop.vsix
|
||||||
|
|
||||||
|
mkdir -p /usr/x86_64-pc-windows-msvc/usr/include
|
||||||
|
mkdir -p /usr/x86_64-pc-windows-msvc/usr/lib
|
||||||
|
|
||||||
|
# lowercase folder/file names
|
||||||
|
echo "$(find . -regex ".*/[^/]*[A-Z][^/]*")" | xargs -I{} sh -c 'mv "$(echo "{}" | sed -E '"'"'s/(.*\/)/\L\1/'"'"')" "$(echo "{}" | tr [A-Z] [a-z])"'
|
||||||
|
|
||||||
|
# .h
|
||||||
|
(cd 'program files/windows kits/10/include/10.0.26100.0' && cp -r ucrt/* um/* shared/* -t /usr/x86_64-pc-windows-msvc/usr/include)
|
||||||
|
|
||||||
|
cp -r contents/vc/tools/msvc/14.16.27023/include/* /usr/x86_64-pc-windows-msvc/usr/include
|
||||||
|
|
||||||
|
# lowercase #include "" and #include <>
|
||||||
|
find /usr/x86_64-pc-windows-msvc/usr/include -type f -exec sed -i -E 's/(#include <[^<>]*?[A-Z][^<>]*?>)|(#include "[^"]*?[A-Z][^"]*?")/\L\1\2/' "{}" ';'
|
||||||
|
|
||||||
|
# x86 intrinsics
|
||||||
|
# original dir: MSVC/
|
||||||
|
|
||||||
|
# '_mm_movemask_epi8' defined in emmintrin.h
|
||||||
|
# '__v4sf' defined in xmmintrin.h
|
||||||
|
# '__v2si' defined in mmintrin.h
|
||||||
|
# '__m128d' redefined in immintrin.h
|
||||||
|
# '__m128i' redefined in intrin.h
|
||||||
|
# '_mm_comlt_epu8' defined in ammintrin.h
|
||||||
|
|
||||||
|
(cd /usr/lib/llvm19/lib/clang/19/include && cp emmintrin.h xmmintrin.h mmintrin.h immintrin.h intrin.h ammintrin.h -t /usr/x86_64-pc-windows-msvc/usr/include)
|
||||||
|
|
||||||
|
# .lib
|
||||||
|
(cd 'program files/windows kits/10/lib/10.0.26100.0/um/x64' && cp advapi32.lib bcrypt.lib kernel32.lib ntdll.lib user32.lib uuid.lib ws2_32.lib userenv.lib cfgmgr32.lib dbghelp.lib fwpuclnt.lib -t /usr/x86_64-pc-windows-msvc/usr/lib)
|
||||||
|
|
||||||
|
(cd 'contents/vc/tools/msvc/14.16.27023/lib/x64' && cp libcmt.lib libvcruntime.lib -t /usr/x86_64-pc-windows-msvc/usr/lib)
|
||||||
|
|
||||||
|
cp 'program files/windows kits/10/lib/10.0.26100.0/ucrt/x64/libucrt.lib' /usr/x86_64-pc-windows-msvc/usr/lib
|
||||||
27
ci/update_lockfiles.sh
Executable file
27
ci/update_lockfiles.sh
Executable file
@@ -0,0 +1,27 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
AMEND=false
|
||||||
|
|
||||||
|
for arg in "$@"; do
|
||||||
|
if [[ "$arg" == "--amend" ]]; then
|
||||||
|
AMEND=true
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
# This updates the lockfile without building
|
||||||
|
cargo metadata --quiet > /dev/null
|
||||||
|
|
||||||
|
pushd nodejs || exit 1
|
||||||
|
npm install --package-lock-only --silent
|
||||||
|
popd
|
||||||
|
|
||||||
|
if git diff --quiet --exit-code; then
|
||||||
|
echo "No lockfile changes to commit; skipping amend."
|
||||||
|
elif $AMEND; then
|
||||||
|
git add Cargo.lock nodejs/package-lock.json
|
||||||
|
git commit --amend --no-edit
|
||||||
|
else
|
||||||
|
git add Cargo.lock nodejs/package-lock.json
|
||||||
|
git commit -m "Update lockfiles"
|
||||||
|
fi
|
||||||
34
ci/validate_stable_lance.py
Normal file
34
ci/validate_stable_lance.py
Normal file
@@ -0,0 +1,34 @@
|
|||||||
|
import tomllib
|
||||||
|
|
||||||
|
found_preview_lance = False
|
||||||
|
|
||||||
|
with open("Cargo.toml", "rb") as f:
|
||||||
|
cargo_data = tomllib.load(f)
|
||||||
|
|
||||||
|
for name, dep in cargo_data["workspace"]["dependencies"].items():
|
||||||
|
if name == "lance" or name.startswith("lance-"):
|
||||||
|
if isinstance(dep, str):
|
||||||
|
version = dep
|
||||||
|
elif isinstance(dep, dict):
|
||||||
|
# Version doesn't have the beta tag in it, so we instead look
|
||||||
|
# at the git tag.
|
||||||
|
version = dep.get('tag', dep.get('version'))
|
||||||
|
else:
|
||||||
|
raise ValueError("Unexpected type for dependency: " + str(dep))
|
||||||
|
|
||||||
|
if "beta" in version:
|
||||||
|
found_preview_lance = True
|
||||||
|
print(f"Dependency '{name}' is a preview version: {version}")
|
||||||
|
|
||||||
|
with open("python/pyproject.toml", "rb") as f:
|
||||||
|
py_proj_data = tomllib.load(f)
|
||||||
|
|
||||||
|
for dep in py_proj_data["project"]["dependencies"]:
|
||||||
|
if dep.startswith("pylance"):
|
||||||
|
if "b" in dep:
|
||||||
|
found_preview_lance = True
|
||||||
|
print(f"Dependency '{dep}' is a preview version")
|
||||||
|
break # Only one pylance dependency
|
||||||
|
|
||||||
|
if found_preview_lance:
|
||||||
|
raise ValueError("Found preview version of Lance in dependencies")
|
||||||
@@ -2,43 +2,88 @@
|
|||||||
|
|
||||||
LanceDB docs are deployed to https://lancedb.github.io/lancedb/.
|
LanceDB docs are deployed to https://lancedb.github.io/lancedb/.
|
||||||
|
|
||||||
Docs is built and deployed automatically by [Github Actions](.github/workflows/docs.yml)
|
Docs is built and deployed automatically by [Github Actions](../.github/workflows/docs.yml)
|
||||||
whenever a commit is pushed to the `main` branch. So it is possible for the docs to show
|
whenever a commit is pushed to the `main` branch. So it is possible for the docs to show
|
||||||
unreleased features.
|
unreleased features.
|
||||||
|
|
||||||
## Building the docs
|
## Building the docs
|
||||||
|
|
||||||
### Setup
|
### Setup
|
||||||
1. Install LanceDB. From LanceDB repo root: `pip install -e python`
|
1. Install LanceDB Python. See setup in [Python contributing guide](../python/CONTRIBUTING.md).
|
||||||
2. Install dependencies. From LanceDB repo root: `pip install -r docs/requirements.txt`
|
Run `make develop` to install the Python package.
|
||||||
3. Make sure you have node and npm setup
|
2. Install documentation dependencies. From LanceDB repo root: `pip install -r docs/requirements.txt`
|
||||||
4. Make sure protobuf and libssl are installed
|
|
||||||
|
|
||||||
### Building node module and create markdown files
|
### Preview the docs
|
||||||
|
|
||||||
See [Javascript docs README](./src/javascript/README.md)
|
```shell
|
||||||
|
|
||||||
### Build docs
|
|
||||||
From LanceDB repo root:
|
|
||||||
|
|
||||||
Run: `PYTHONPATH=. mkdocs build -f docs/mkdocs.yml`
|
|
||||||
|
|
||||||
If successful, you should see a `docs/site` directory that you can verify locally.
|
|
||||||
|
|
||||||
### Run local server
|
|
||||||
|
|
||||||
You can run a local server to test the docs prior to deployment by navigating to the `docs` directory and running the following command:
|
|
||||||
|
|
||||||
```bash
|
|
||||||
cd docs
|
cd docs
|
||||||
mkdocs serve
|
mkdocs serve
|
||||||
```
|
```
|
||||||
|
|
||||||
### Run doctest for typescript example
|
If you want to just generate the HTML files:
|
||||||
|
|
||||||
```bash
|
```shell
|
||||||
cd lancedb/docs
|
PYTHONPATH=. mkdocs build -f docs/mkdocs.yml
|
||||||
npm i
|
```
|
||||||
npm run build
|
|
||||||
npm run all
|
If successful, you should see a `docs/site` directory that you can verify locally.
|
||||||
|
|
||||||
|
## Adding examples
|
||||||
|
|
||||||
|
To make sure examples are correct, we put examples in test files so they can be
|
||||||
|
run as part of our test suites.
|
||||||
|
|
||||||
|
You can see the tests are at:
|
||||||
|
|
||||||
|
* Python: `python/python/tests/docs`
|
||||||
|
* Typescript: `nodejs/examples/`
|
||||||
|
|
||||||
|
### Checking python examples
|
||||||
|
|
||||||
|
```shell
|
||||||
|
cd python
|
||||||
|
pytest -vv python/tests/docs
|
||||||
|
```
|
||||||
|
|
||||||
|
### Checking typescript examples
|
||||||
|
|
||||||
|
The `@lancedb/lancedb` package must be built before running the tests:
|
||||||
|
|
||||||
|
```shell
|
||||||
|
pushd nodejs
|
||||||
|
npm ci
|
||||||
|
npm run build
|
||||||
|
popd
|
||||||
|
```
|
||||||
|
|
||||||
|
Then you can run the examples by going to the `nodejs/examples` directory and
|
||||||
|
running the tests like a normal npm package:
|
||||||
|
|
||||||
|
```shell
|
||||||
|
pushd nodejs/examples
|
||||||
|
npm ci
|
||||||
|
npm test
|
||||||
|
popd
|
||||||
|
```
|
||||||
|
|
||||||
|
## API documentation
|
||||||
|
|
||||||
|
### Python
|
||||||
|
|
||||||
|
The Python API documentation is organized based on the file `docs/src/python/python.md`.
|
||||||
|
We manually add entries there so we can control the organization of the reference page.
|
||||||
|
**However, this means any new types must be manually added to the file.** No additional
|
||||||
|
steps are needed to generate the API documentation.
|
||||||
|
|
||||||
|
### Typescript
|
||||||
|
|
||||||
|
The typescript API documentation is generated from the typescript source code using [typedoc](https://typedoc.org/).
|
||||||
|
|
||||||
|
When new APIs are added, you must manually re-run the typedoc command to update the API documentation.
|
||||||
|
The new files should be checked into the repository.
|
||||||
|
|
||||||
|
```shell
|
||||||
|
pushd nodejs
|
||||||
|
npm run docs
|
||||||
|
popd
|
||||||
```
|
```
|
||||||
|
|||||||
158
docs/mkdocs.yml
158
docs/mkdocs.yml
@@ -4,6 +4,9 @@ repo_url: https://github.com/lancedb/lancedb
|
|||||||
edit_uri: https://github.com/lancedb/lancedb/tree/main/docs/src
|
edit_uri: https://github.com/lancedb/lancedb/tree/main/docs/src
|
||||||
repo_name: lancedb/lancedb
|
repo_name: lancedb/lancedb
|
||||||
docs_dir: src
|
docs_dir: src
|
||||||
|
watch:
|
||||||
|
- src
|
||||||
|
- ../python/python
|
||||||
|
|
||||||
theme:
|
theme:
|
||||||
name: "material"
|
name: "material"
|
||||||
@@ -26,6 +29,7 @@ theme:
|
|||||||
- content.code.copy
|
- content.code.copy
|
||||||
- content.tabs.link
|
- content.tabs.link
|
||||||
- content.action.edit
|
- content.action.edit
|
||||||
|
- content.tooltips
|
||||||
- toc.follow
|
- toc.follow
|
||||||
- navigation.top
|
- navigation.top
|
||||||
- navigation.tabs
|
- navigation.tabs
|
||||||
@@ -33,8 +37,10 @@ theme:
|
|||||||
- navigation.footer
|
- navigation.footer
|
||||||
- navigation.tracking
|
- navigation.tracking
|
||||||
- navigation.instant
|
- navigation.instant
|
||||||
|
- content.footnote.tooltips
|
||||||
icon:
|
icon:
|
||||||
repo: fontawesome/brands/github
|
repo: fontawesome/brands/github
|
||||||
|
annotation: material/arrow-right-circle
|
||||||
custom_dir: overrides
|
custom_dir: overrides
|
||||||
|
|
||||||
plugins:
|
plugins:
|
||||||
@@ -52,10 +58,15 @@ plugins:
|
|||||||
show_signature_annotations: true
|
show_signature_annotations: true
|
||||||
show_root_heading: true
|
show_root_heading: true
|
||||||
members_order: source
|
members_order: source
|
||||||
|
docstring_section_style: list
|
||||||
|
signature_crossrefs: true
|
||||||
|
separate_signature: true
|
||||||
import:
|
import:
|
||||||
# for cross references
|
# for cross references
|
||||||
- https://arrow.apache.org/docs/objects.inv
|
- https://arrow.apache.org/docs/objects.inv
|
||||||
- https://pandas.pydata.org/docs/objects.inv
|
- https://pandas.pydata.org/docs/objects.inv
|
||||||
|
- https://lancedb.github.io/lance/objects.inv
|
||||||
|
- https://docs.pydantic.dev/latest/objects.inv
|
||||||
- mkdocs-jupyter
|
- mkdocs-jupyter
|
||||||
- render_swagger:
|
- render_swagger:
|
||||||
allow_arbitrary_locations: true
|
allow_arbitrary_locations: true
|
||||||
@@ -63,6 +74,11 @@ plugins:
|
|||||||
markdown_extensions:
|
markdown_extensions:
|
||||||
- admonition
|
- admonition
|
||||||
- footnotes
|
- footnotes
|
||||||
|
- pymdownx.critic
|
||||||
|
- pymdownx.caret
|
||||||
|
- pymdownx.keys
|
||||||
|
- pymdownx.mark
|
||||||
|
- pymdownx.tilde
|
||||||
- pymdownx.details
|
- pymdownx.details
|
||||||
- pymdownx.highlight:
|
- pymdownx.highlight:
|
||||||
anchor_linenums: true
|
anchor_linenums: true
|
||||||
@@ -76,7 +92,15 @@ markdown_extensions:
|
|||||||
- pymdownx.tabbed:
|
- pymdownx.tabbed:
|
||||||
alternate_style: true
|
alternate_style: true
|
||||||
- md_in_html
|
- md_in_html
|
||||||
|
- abbr
|
||||||
- attr_list
|
- attr_list
|
||||||
|
- pymdownx.snippets
|
||||||
|
- pymdownx.emoji:
|
||||||
|
emoji_index: !!python/name:material.extensions.emoji.twemoji
|
||||||
|
emoji_generator: !!python/name:material.extensions.emoji.to_svg
|
||||||
|
- markdown.extensions.toc:
|
||||||
|
baselevel: 1
|
||||||
|
permalink: ""
|
||||||
|
|
||||||
nav:
|
nav:
|
||||||
- Home:
|
- Home:
|
||||||
@@ -84,19 +108,37 @@ nav:
|
|||||||
- 🏃🏼♂️ Quick start: basic.md
|
- 🏃🏼♂️ Quick start: basic.md
|
||||||
- 📚 Concepts:
|
- 📚 Concepts:
|
||||||
- Vector search: concepts/vector_search.md
|
- Vector search: concepts/vector_search.md
|
||||||
- Indexing: concepts/index_ivfpq.md
|
- Indexing:
|
||||||
|
- IVFPQ: concepts/index_ivfpq.md
|
||||||
|
- HNSW: concepts/index_hnsw.md
|
||||||
- Storage: concepts/storage.md
|
- Storage: concepts/storage.md
|
||||||
- Data management: concepts/data_management.md
|
- Data management: concepts/data_management.md
|
||||||
- 🔨 Guides:
|
- 🔨 Guides:
|
||||||
- Working with tables: guides/tables.md
|
- Working with tables: guides/tables.md
|
||||||
- Building a vector index: ann_indexes.md
|
- Building a vector index: ann_indexes.md
|
||||||
- Vector Search: search.md
|
- Vector Search: search.md
|
||||||
- Full-text search: fts.md
|
- Full-text search (native): fts.md
|
||||||
|
- Full-text search (tantivy-based): fts_tantivy.md
|
||||||
- Building a scalar index: guides/scalar_index.md
|
- Building a scalar index: guides/scalar_index.md
|
||||||
- Hybrid search:
|
- Hybrid search:
|
||||||
- Overview: hybrid_search/hybrid_search.md
|
- Overview: hybrid_search/hybrid_search.md
|
||||||
- Comparing Rerankers: hybrid_search/eval.md
|
- Comparing Rerankers: hybrid_search/eval.md
|
||||||
- Airbnb financial data example: notebooks/hybrid_search.ipynb
|
- Airbnb financial data example: notebooks/hybrid_search.ipynb
|
||||||
|
- Late interaction with MultiVector search:
|
||||||
|
- Overview: guides/multi-vector.md
|
||||||
|
- Example: notebooks/Multivector_on_LanceDB.ipynb
|
||||||
|
- RAG:
|
||||||
|
- Vanilla RAG: rag/vanilla_rag.md
|
||||||
|
- Multi-head RAG: rag/multi_head_rag.md
|
||||||
|
- Corrective RAG: rag/corrective_rag.md
|
||||||
|
- Agentic RAG: rag/agentic_rag.md
|
||||||
|
- Graph RAG: rag/graph_rag.md
|
||||||
|
- Self RAG: rag/self_rag.md
|
||||||
|
- Adaptive RAG: rag/adaptive_rag.md
|
||||||
|
- SFR RAG: rag/sfr_rag.md
|
||||||
|
- Advanced Techniques:
|
||||||
|
- HyDE: rag/advanced_techniques/hyde.md
|
||||||
|
- FLARE: rag/advanced_techniques/flare.md
|
||||||
- Reranking:
|
- Reranking:
|
||||||
- Quickstart: reranking/index.md
|
- Quickstart: reranking/index.md
|
||||||
- Cohere Reranker: reranking/cohere.md
|
- Cohere Reranker: reranking/cohere.md
|
||||||
@@ -106,10 +148,14 @@ nav:
|
|||||||
- ColBERT Reranker: reranking/colbert.md
|
- ColBERT Reranker: reranking/colbert.md
|
||||||
- Jina Reranker: reranking/jina.md
|
- Jina Reranker: reranking/jina.md
|
||||||
- OpenAI Reranker: reranking/openai.md
|
- OpenAI Reranker: reranking/openai.md
|
||||||
|
- AnswerDotAi Rerankers: reranking/answerdotai.md
|
||||||
|
- Voyage AI Rerankers: reranking/voyageai.md
|
||||||
- Building Custom Rerankers: reranking/custom_reranker.md
|
- Building Custom Rerankers: reranking/custom_reranker.md
|
||||||
- Example: notebooks/lancedb_reranking.ipynb
|
- Example: notebooks/lancedb_reranking.ipynb
|
||||||
- Filtering: sql.md
|
- Filtering: sql.md
|
||||||
- Versioning & Reproducibility: notebooks/reproducibility.ipynb
|
- Versioning & Reproducibility:
|
||||||
|
- sync API: notebooks/reproducibility.ipynb
|
||||||
|
- async API: notebooks/reproducibility_async.ipynb
|
||||||
- Configuring Storage: guides/storage.md
|
- Configuring Storage: guides/storage.md
|
||||||
- Migration Guide: migration.md
|
- Migration Guide: migration.md
|
||||||
- Tuning retrieval performance:
|
- Tuning retrieval performance:
|
||||||
@@ -117,10 +163,29 @@ nav:
|
|||||||
- Reranking: guides/tuning_retrievers/2_reranking.md
|
- Reranking: guides/tuning_retrievers/2_reranking.md
|
||||||
- Embedding fine-tuning: guides/tuning_retrievers/3_embed_tuning.md
|
- Embedding fine-tuning: guides/tuning_retrievers/3_embed_tuning.md
|
||||||
- 🧬 Managing embeddings:
|
- 🧬 Managing embeddings:
|
||||||
- Overview: embeddings/index.md
|
- Understand Embeddings: embeddings/understanding_embeddings.md
|
||||||
|
- Get Started: embeddings/index.md
|
||||||
- Embedding functions: embeddings/embedding_functions.md
|
- Embedding functions: embeddings/embedding_functions.md
|
||||||
- Available models: embeddings/default_embedding_functions.md
|
- Available models:
|
||||||
|
- Overview: embeddings/default_embedding_functions.md
|
||||||
|
- Text Embedding Functions:
|
||||||
|
- Sentence Transformers: embeddings/available_embedding_models/text_embedding_functions/sentence_transformers.md
|
||||||
|
- Huggingface Embedding Models: embeddings/available_embedding_models/text_embedding_functions/huggingface_embedding.md
|
||||||
|
- Ollama Embeddings: embeddings/available_embedding_models/text_embedding_functions/ollama_embedding.md
|
||||||
|
- OpenAI Embeddings: embeddings/available_embedding_models/text_embedding_functions/openai_embedding.md
|
||||||
|
- Instructor Embeddings: embeddings/available_embedding_models/text_embedding_functions/instructor_embedding.md
|
||||||
|
- Gemini Embeddings: embeddings/available_embedding_models/text_embedding_functions/gemini_embedding.md
|
||||||
|
- Cohere Embeddings: embeddings/available_embedding_models/text_embedding_functions/cohere_embedding.md
|
||||||
|
- Jina Embeddings: embeddings/available_embedding_models/text_embedding_functions/jina_embedding.md
|
||||||
|
- AWS Bedrock Text Embedding Functions: embeddings/available_embedding_models/text_embedding_functions/aws_bedrock_embedding.md
|
||||||
|
- IBM watsonx.ai Embeddings: embeddings/available_embedding_models/text_embedding_functions/ibm_watsonx_ai_embedding.md
|
||||||
|
- Voyage AI Embeddings: embeddings/available_embedding_models/text_embedding_functions/voyageai_embedding.md
|
||||||
|
- Multimodal Embedding Functions:
|
||||||
|
- OpenClip embeddings: embeddings/available_embedding_models/multimodal_embedding_functions/openclip_embedding.md
|
||||||
|
- Imagebind embeddings: embeddings/available_embedding_models/multimodal_embedding_functions/imagebind_embedding.md
|
||||||
|
- Jina Embeddings: embeddings/available_embedding_models/multimodal_embedding_functions/jina_multimodal_embedding.md
|
||||||
- User-defined embedding functions: embeddings/custom_embedding_function.md
|
- User-defined embedding functions: embeddings/custom_embedding_function.md
|
||||||
|
- Variables and secrets: embeddings/variables_and_secrets.md
|
||||||
- "Example: Multi-lingual semantic search": notebooks/multi_lingual_example.ipynb
|
- "Example: Multi-lingual semantic search": notebooks/multi_lingual_example.ipynb
|
||||||
- "Example: MultiModal CLIP Embeddings": notebooks/DisappearingEmbeddingFunction.ipynb
|
- "Example: MultiModal CLIP Embeddings": notebooks/DisappearingEmbeddingFunction.ipynb
|
||||||
- 🔌 Integrations:
|
- 🔌 Integrations:
|
||||||
@@ -128,6 +193,7 @@ nav:
|
|||||||
- Pandas and PyArrow: python/pandas_and_pyarrow.md
|
- Pandas and PyArrow: python/pandas_and_pyarrow.md
|
||||||
- Polars: python/polars_arrow.md
|
- Polars: python/polars_arrow.md
|
||||||
- DuckDB: python/duckdb.md
|
- DuckDB: python/duckdb.md
|
||||||
|
- Datafusion: python/datafusion.md
|
||||||
- LangChain:
|
- LangChain:
|
||||||
- LangChain 🔗: integrations/langchain.md
|
- LangChain 🔗: integrations/langchain.md
|
||||||
- LangChain demo: notebooks/langchain_demo.ipynb
|
- LangChain demo: notebooks/langchain_demo.ipynb
|
||||||
@@ -139,6 +205,8 @@ nav:
|
|||||||
- Voxel51: integrations/voxel51.md
|
- Voxel51: integrations/voxel51.md
|
||||||
- PromptTools: integrations/prompttools.md
|
- PromptTools: integrations/prompttools.md
|
||||||
- dlt: integrations/dlt.md
|
- dlt: integrations/dlt.md
|
||||||
|
- phidata: integrations/phidata.md
|
||||||
|
- Genkit: integrations/genkit.md
|
||||||
- 🎯 Examples:
|
- 🎯 Examples:
|
||||||
- Overview: examples/index.md
|
- Overview: examples/index.md
|
||||||
- 🐍 Python:
|
- 🐍 Python:
|
||||||
@@ -150,10 +218,8 @@ nav:
|
|||||||
- Chatbot: examples/python_examples/chatbot.md
|
- Chatbot: examples/python_examples/chatbot.md
|
||||||
- Evaluation: examples/python_examples/evaluations.md
|
- Evaluation: examples/python_examples/evaluations.md
|
||||||
- AI Agent: examples/python_examples/aiagent.md
|
- AI Agent: examples/python_examples/aiagent.md
|
||||||
|
- Recommender System: examples/python_examples/recommendersystem.md
|
||||||
- Miscellaneous:
|
- Miscellaneous:
|
||||||
- YouTube Transcript Search: notebooks/youtube_transcript_search.ipynb
|
|
||||||
- Documentation QA Bot using LangChain: notebooks/code_qa_bot.ipynb
|
|
||||||
- Multimodal search using CLIP: notebooks/multimodal_search.ipynb
|
|
||||||
- Serverless QA Bot with S3 and Lambda: examples/serverless_lancedb_with_s3_and_lambda.md
|
- Serverless QA Bot with S3 and Lambda: examples/serverless_lancedb_with_s3_and_lambda.md
|
||||||
- Serverless QA Bot with Modal: examples/serverless_qa_bot_with_modal_and_langchain.md
|
- Serverless QA Bot with Modal: examples/serverless_qa_bot_with_modal_and_langchain.md
|
||||||
- 👾 JavaScript:
|
- 👾 JavaScript:
|
||||||
@@ -163,35 +229,51 @@ nav:
|
|||||||
- TransformersJS Embedding Search: examples/transformerjs_embedding_search_nodejs.md
|
- TransformersJS Embedding Search: examples/transformerjs_embedding_search_nodejs.md
|
||||||
- 🦀 Rust:
|
- 🦀 Rust:
|
||||||
- Overview: examples/examples_rust.md
|
- Overview: examples/examples_rust.md
|
||||||
|
- 📓 Studies:
|
||||||
|
- ↗Improve retrievers with hybrid search and reranking: https://blog.lancedb.com/hybrid-search-and-reranking-report/
|
||||||
- 💭 FAQs: faq.md
|
- 💭 FAQs: faq.md
|
||||||
|
- 🔍 Troubleshooting: troubleshooting.md
|
||||||
- ⚙️ API reference:
|
- ⚙️ API reference:
|
||||||
- 🐍 Python: python/python.md
|
- 🐍 Python: python/python.md
|
||||||
- 👾 JavaScript (vectordb): javascript/modules.md
|
- 👾 JavaScript (vectordb): javascript/modules.md
|
||||||
- 👾 JavaScript (lancedb): js/globals.md
|
- 👾 JavaScript (lancedb): js/globals.md
|
||||||
- 🦀 Rust: https://docs.rs/lancedb/latest/lancedb/
|
- 🦀 Rust: https://docs.rs/lancedb/latest/lancedb/
|
||||||
- ☁️ LanceDB Cloud:
|
|
||||||
- Overview: cloud/index.md
|
|
||||||
- API reference:
|
|
||||||
- 🐍 Python: python/saas-python.md
|
|
||||||
- 👾 JavaScript: javascript/modules.md
|
|
||||||
- REST API: cloud/rest.md
|
|
||||||
|
|
||||||
- Quick start: basic.md
|
- Quick start: basic.md
|
||||||
- Concepts:
|
- Concepts:
|
||||||
- Vector search: concepts/vector_search.md
|
- Vector search: concepts/vector_search.md
|
||||||
- Indexing: concepts/index_ivfpq.md
|
- Indexing:
|
||||||
|
- IVFPQ: concepts/index_ivfpq.md
|
||||||
|
- HNSW: concepts/index_hnsw.md
|
||||||
- Storage: concepts/storage.md
|
- Storage: concepts/storage.md
|
||||||
- Data management: concepts/data_management.md
|
- Data management: concepts/data_management.md
|
||||||
- Guides:
|
- Guides:
|
||||||
- Working with tables: guides/tables.md
|
- Working with tables: guides/tables.md
|
||||||
|
- Working with SQL: guides/sql_querying.md
|
||||||
- Building an ANN index: ann_indexes.md
|
- Building an ANN index: ann_indexes.md
|
||||||
- Vector Search: search.md
|
- Vector Search: search.md
|
||||||
- Full-text search: fts.md
|
- Full-text search (native): fts.md
|
||||||
|
- Full-text search (tantivy-based): fts_tantivy.md
|
||||||
- Building a scalar index: guides/scalar_index.md
|
- Building a scalar index: guides/scalar_index.md
|
||||||
- Hybrid search:
|
- Hybrid search:
|
||||||
- Overview: hybrid_search/hybrid_search.md
|
- Overview: hybrid_search/hybrid_search.md
|
||||||
- Comparing Rerankers: hybrid_search/eval.md
|
- Comparing Rerankers: hybrid_search/eval.md
|
||||||
- Airbnb financial data example: notebooks/hybrid_search.ipynb
|
- Airbnb financial data example: notebooks/hybrid_search.ipynb
|
||||||
|
- Late interaction with MultiVector search:
|
||||||
|
- Overview: guides/multi-vector.md
|
||||||
|
- Document search Example: notebooks/Multivector_on_LanceDB.ipynb
|
||||||
|
- RAG:
|
||||||
|
- Vanilla RAG: rag/vanilla_rag.md
|
||||||
|
- Multi-head RAG: rag/multi_head_rag.md
|
||||||
|
- Corrective RAG: rag/corrective_rag.md
|
||||||
|
- Agentic RAG: rag/agentic_rag.md
|
||||||
|
- Graph RAG: rag/graph_rag.md
|
||||||
|
- Self RAG: rag/self_rag.md
|
||||||
|
- Adaptive RAG: rag/adaptive_rag.md
|
||||||
|
- SFR RAG: rag/sfr_rag.md
|
||||||
|
- Advanced Techniques:
|
||||||
|
- HyDE: rag/advanced_techniques/hyde.md
|
||||||
|
- FLARE: rag/advanced_techniques/flare.md
|
||||||
- Reranking:
|
- Reranking:
|
||||||
- Quickstart: reranking/index.md
|
- Quickstart: reranking/index.md
|
||||||
- Cohere Reranker: reranking/cohere.md
|
- Cohere Reranker: reranking/cohere.md
|
||||||
@@ -201,10 +283,13 @@ nav:
|
|||||||
- ColBERT Reranker: reranking/colbert.md
|
- ColBERT Reranker: reranking/colbert.md
|
||||||
- Jina Reranker: reranking/jina.md
|
- Jina Reranker: reranking/jina.md
|
||||||
- OpenAI Reranker: reranking/openai.md
|
- OpenAI Reranker: reranking/openai.md
|
||||||
|
- AnswerDotAi Rerankers: reranking/answerdotai.md
|
||||||
- Building Custom Rerankers: reranking/custom_reranker.md
|
- Building Custom Rerankers: reranking/custom_reranker.md
|
||||||
- Example: notebooks/lancedb_reranking.ipynb
|
- Example: notebooks/lancedb_reranking.ipynb
|
||||||
- Filtering: sql.md
|
- Filtering: sql.md
|
||||||
- Versioning & Reproducibility: notebooks/reproducibility.ipynb
|
- Versioning & Reproducibility:
|
||||||
|
- sync API: notebooks/reproducibility.ipynb
|
||||||
|
- async API: notebooks/reproducibility_async.ipynb
|
||||||
- Configuring Storage: guides/storage.md
|
- Configuring Storage: guides/storage.md
|
||||||
- Migration Guide: migration.md
|
- Migration Guide: migration.md
|
||||||
- Tuning retrieval performance:
|
- Tuning retrieval performance:
|
||||||
@@ -212,10 +297,28 @@ nav:
|
|||||||
- Reranking: guides/tuning_retrievers/2_reranking.md
|
- Reranking: guides/tuning_retrievers/2_reranking.md
|
||||||
- Embedding fine-tuning: guides/tuning_retrievers/3_embed_tuning.md
|
- Embedding fine-tuning: guides/tuning_retrievers/3_embed_tuning.md
|
||||||
- Managing Embeddings:
|
- Managing Embeddings:
|
||||||
- Overview: embeddings/index.md
|
- Understand Embeddings: embeddings/understanding_embeddings.md
|
||||||
|
- Get Started: embeddings/index.md
|
||||||
- Embedding functions: embeddings/embedding_functions.md
|
- Embedding functions: embeddings/embedding_functions.md
|
||||||
- Available models: embeddings/default_embedding_functions.md
|
- Available models:
|
||||||
|
- Overview: embeddings/default_embedding_functions.md
|
||||||
|
- Text Embedding Functions:
|
||||||
|
- Sentence Transformers: embeddings/available_embedding_models/text_embedding_functions/sentence_transformers.md
|
||||||
|
- Huggingface Embedding Models: embeddings/available_embedding_models/text_embedding_functions/huggingface_embedding.md
|
||||||
|
- Ollama Embeddings: embeddings/available_embedding_models/text_embedding_functions/ollama_embedding.md
|
||||||
|
- OpenAI Embeddings: embeddings/available_embedding_models/text_embedding_functions/openai_embedding.md
|
||||||
|
- Instructor Embeddings: embeddings/available_embedding_models/text_embedding_functions/instructor_embedding.md
|
||||||
|
- Gemini Embeddings: embeddings/available_embedding_models/text_embedding_functions/gemini_embedding.md
|
||||||
|
- Cohere Embeddings: embeddings/available_embedding_models/text_embedding_functions/cohere_embedding.md
|
||||||
|
- Jina Embeddings: embeddings/available_embedding_models/text_embedding_functions/jina_embedding.md
|
||||||
|
- AWS Bedrock Text Embedding Functions: embeddings/available_embedding_models/text_embedding_functions/aws_bedrock_embedding.md
|
||||||
|
- IBM watsonx.ai Embeddings: embeddings/available_embedding_models/text_embedding_functions/ibm_watsonx_ai_embedding.md
|
||||||
|
- Multimodal Embedding Functions:
|
||||||
|
- OpenClip embeddings: embeddings/available_embedding_models/multimodal_embedding_functions/openclip_embedding.md
|
||||||
|
- Imagebind embeddings: embeddings/available_embedding_models/multimodal_embedding_functions/imagebind_embedding.md
|
||||||
|
- Jina Embeddings: embeddings/available_embedding_models/multimodal_embedding_functions/jina_multimodal_embedding.md
|
||||||
- User-defined embedding functions: embeddings/custom_embedding_function.md
|
- User-defined embedding functions: embeddings/custom_embedding_function.md
|
||||||
|
- Variables and secrets: embeddings/variables_and_secrets.md
|
||||||
- "Example: Multi-lingual semantic search": notebooks/multi_lingual_example.ipynb
|
- "Example: Multi-lingual semantic search": notebooks/multi_lingual_example.ipynb
|
||||||
- "Example: MultiModal CLIP Embeddings": notebooks/DisappearingEmbeddingFunction.ipynb
|
- "Example: MultiModal CLIP Embeddings": notebooks/DisappearingEmbeddingFunction.ipynb
|
||||||
- Integrations:
|
- Integrations:
|
||||||
@@ -223,6 +326,7 @@ nav:
|
|||||||
- Pandas and PyArrow: python/pandas_and_pyarrow.md
|
- Pandas and PyArrow: python/pandas_and_pyarrow.md
|
||||||
- Polars: python/polars_arrow.md
|
- Polars: python/polars_arrow.md
|
||||||
- DuckDB: python/duckdb.md
|
- DuckDB: python/duckdb.md
|
||||||
|
- Datafusion: python/datafusion.md
|
||||||
- LangChain 🦜️🔗↗: integrations/langchain.md
|
- LangChain 🦜️🔗↗: integrations/langchain.md
|
||||||
- LangChain.js 🦜️🔗↗: https://js.langchain.com/docs/integrations/vectorstores/lancedb
|
- LangChain.js 🦜️🔗↗: https://js.langchain.com/docs/integrations/vectorstores/lancedb
|
||||||
- LlamaIndex 🦙↗: integrations/llamaIndex.md
|
- LlamaIndex 🦙↗: integrations/llamaIndex.md
|
||||||
@@ -230,6 +334,8 @@ nav:
|
|||||||
- Voxel51: integrations/voxel51.md
|
- Voxel51: integrations/voxel51.md
|
||||||
- PromptTools: integrations/prompttools.md
|
- PromptTools: integrations/prompttools.md
|
||||||
- dlt: integrations/dlt.md
|
- dlt: integrations/dlt.md
|
||||||
|
- phidata: integrations/phidata.md
|
||||||
|
- Genkit: integrations/genkit.md
|
||||||
- Examples:
|
- Examples:
|
||||||
- examples/index.md
|
- examples/index.md
|
||||||
- 🐍 Python:
|
- 🐍 Python:
|
||||||
@@ -241,10 +347,8 @@ nav:
|
|||||||
- Chatbot: examples/python_examples/chatbot.md
|
- Chatbot: examples/python_examples/chatbot.md
|
||||||
- Evaluation: examples/python_examples/evaluations.md
|
- Evaluation: examples/python_examples/evaluations.md
|
||||||
- AI Agent: examples/python_examples/aiagent.md
|
- AI Agent: examples/python_examples/aiagent.md
|
||||||
|
- Recommender System: examples/python_examples/recommendersystem.md
|
||||||
- Miscellaneous:
|
- Miscellaneous:
|
||||||
- YouTube Transcript Search: notebooks/youtube_transcript_search.ipynb
|
|
||||||
- Documentation QA Bot using LangChain: notebooks/code_qa_bot.ipynb
|
|
||||||
- Multimodal search using CLIP: notebooks/multimodal_search.ipynb
|
|
||||||
- Serverless QA Bot with S3 and Lambda: examples/serverless_lancedb_with_s3_and_lambda.md
|
- Serverless QA Bot with S3 and Lambda: examples/serverless_lancedb_with_s3_and_lambda.md
|
||||||
- Serverless QA Bot with Modal: examples/serverless_qa_bot_with_modal_and_langchain.md
|
- Serverless QA Bot with Modal: examples/serverless_qa_bot_with_modal_and_langchain.md
|
||||||
- 👾 JavaScript:
|
- 👾 JavaScript:
|
||||||
@@ -254,18 +358,15 @@ nav:
|
|||||||
- TransformersJS Embedding Search: examples/transformerjs_embedding_search_nodejs.md
|
- TransformersJS Embedding Search: examples/transformerjs_embedding_search_nodejs.md
|
||||||
- 🦀 Rust:
|
- 🦀 Rust:
|
||||||
- Overview: examples/examples_rust.md
|
- Overview: examples/examples_rust.md
|
||||||
|
- Studies:
|
||||||
|
- studies/overview.md
|
||||||
|
- ↗Improve retrievers with hybrid search and reranking: https://blog.lancedb.com/hybrid-search-and-reranking-report/
|
||||||
- API reference:
|
- API reference:
|
||||||
- Overview: api_reference.md
|
- Overview: api_reference.md
|
||||||
- Python: python/python.md
|
- Python: python/python.md
|
||||||
- Javascript (vectordb): javascript/modules.md
|
- Javascript (vectordb): javascript/modules.md
|
||||||
- Javascript (lancedb): js/globals.md
|
- Javascript (lancedb): js/globals.md
|
||||||
- Rust: https://docs.rs/lancedb/latest/lancedb/index.html
|
- Rust: https://docs.rs/lancedb/latest/lancedb/index.html
|
||||||
- LanceDB Cloud:
|
|
||||||
- Overview: cloud/index.md
|
|
||||||
- API reference:
|
|
||||||
- 🐍 Python: python/saas-python.md
|
|
||||||
- 👾 JavaScript: javascript/modules.md
|
|
||||||
- REST API: cloud/rest.md
|
|
||||||
|
|
||||||
extra_css:
|
extra_css:
|
||||||
- styles/global.css
|
- styles/global.css
|
||||||
@@ -273,6 +374,7 @@ extra_css:
|
|||||||
|
|
||||||
extra_javascript:
|
extra_javascript:
|
||||||
- "extra_js/init_ask_ai_widget.js"
|
- "extra_js/init_ask_ai_widget.js"
|
||||||
|
- "extra_js/reo.js"
|
||||||
|
|
||||||
extra:
|
extra:
|
||||||
analytics:
|
analytics:
|
||||||
|
|||||||
@@ -38,6 +38,13 @@ components:
|
|||||||
required: true
|
required: true
|
||||||
schema:
|
schema:
|
||||||
type: string
|
type: string
|
||||||
|
index_name:
|
||||||
|
name: index_name
|
||||||
|
in: path
|
||||||
|
description: name of the index
|
||||||
|
required: true
|
||||||
|
schema:
|
||||||
|
type: string
|
||||||
responses:
|
responses:
|
||||||
invalid_request:
|
invalid_request:
|
||||||
description: Invalid request
|
description: Invalid request
|
||||||
@@ -164,7 +171,7 @@ paths:
|
|||||||
distance_type:
|
distance_type:
|
||||||
type: string
|
type: string
|
||||||
description: |
|
description: |
|
||||||
The distance metric to use for search. L2, Cosine, Dot and Hamming are supported. Default is L2.
|
The distance metric to use for search. l2, Cosine, Dot and Hamming are supported. Default is l2.
|
||||||
bypass_vector_index:
|
bypass_vector_index:
|
||||||
type: boolean
|
type: boolean
|
||||||
description: |
|
description: |
|
||||||
@@ -443,7 +450,7 @@ paths:
|
|||||||
type: string
|
type: string
|
||||||
nullable: false
|
nullable: false
|
||||||
description: |
|
description: |
|
||||||
The metric type to use for the index. L2, Cosine, Dot are supported.
|
The metric type to use for the index. l2, Cosine, Dot are supported.
|
||||||
index_type:
|
index_type:
|
||||||
type: string
|
type: string
|
||||||
responses:
|
responses:
|
||||||
@@ -485,3 +492,22 @@ paths:
|
|||||||
$ref: "#/components/responses/unauthorized"
|
$ref: "#/components/responses/unauthorized"
|
||||||
"404":
|
"404":
|
||||||
$ref: "#/components/responses/not_found"
|
$ref: "#/components/responses/not_found"
|
||||||
|
/v1/table/{name}/index/{index_name}/drop/:
|
||||||
|
post:
|
||||||
|
description: Drop an index from the table
|
||||||
|
tags:
|
||||||
|
- Tables
|
||||||
|
summary: Drop an index from the table
|
||||||
|
operationId: dropIndex
|
||||||
|
parameters:
|
||||||
|
- $ref: "#/components/parameters/table_name"
|
||||||
|
- $ref: "#/components/parameters/index_name"
|
||||||
|
responses:
|
||||||
|
"200":
|
||||||
|
description: Index successfully dropped
|
||||||
|
"400":
|
||||||
|
$ref: "#/components/responses/invalid_request"
|
||||||
|
"401":
|
||||||
|
$ref: "#/components/responses/unauthorized"
|
||||||
|
"404":
|
||||||
|
$ref: "#/components/responses/not_found"
|
||||||
5
docs/overrides/partials/main.html
Normal file
5
docs/overrides/partials/main.html
Normal file
@@ -0,0 +1,5 @@
|
|||||||
|
{% extends "base.html" %}
|
||||||
|
|
||||||
|
{% block announce %}
|
||||||
|
📚 Starting June 1st, 2025, please use <a href="https://lancedb.github.io/documentation" target="_blank" rel="noopener noreferrer">lancedb.github.io/documentation</a> for the latest docs.
|
||||||
|
{% endblock %}
|
||||||
21
docs/package-lock.json
generated
21
docs/package-lock.json
generated
@@ -19,7 +19,7 @@
|
|||||||
},
|
},
|
||||||
"../node": {
|
"../node": {
|
||||||
"name": "vectordb",
|
"name": "vectordb",
|
||||||
"version": "0.4.6",
|
"version": "0.21.2-beta.0",
|
||||||
"cpu": [
|
"cpu": [
|
||||||
"x64",
|
"x64",
|
||||||
"arm64"
|
"arm64"
|
||||||
@@ -31,9 +31,7 @@
|
|||||||
"win32"
|
"win32"
|
||||||
],
|
],
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"@apache-arrow/ts": "^14.0.2",
|
|
||||||
"@neon-rs/load": "^0.0.74",
|
"@neon-rs/load": "^0.0.74",
|
||||||
"apache-arrow": "^14.0.2",
|
|
||||||
"axios": "^1.4.0"
|
"axios": "^1.4.0"
|
||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
@@ -46,6 +44,7 @@
|
|||||||
"@types/temp": "^0.9.1",
|
"@types/temp": "^0.9.1",
|
||||||
"@types/uuid": "^9.0.3",
|
"@types/uuid": "^9.0.3",
|
||||||
"@typescript-eslint/eslint-plugin": "^5.59.1",
|
"@typescript-eslint/eslint-plugin": "^5.59.1",
|
||||||
|
"apache-arrow-old": "npm:apache-arrow@13.0.0",
|
||||||
"cargo-cp-artifact": "^0.1",
|
"cargo-cp-artifact": "^0.1",
|
||||||
"chai": "^4.3.7",
|
"chai": "^4.3.7",
|
||||||
"chai-as-promised": "^7.1.1",
|
"chai-as-promised": "^7.1.1",
|
||||||
@@ -62,15 +61,19 @@
|
|||||||
"ts-node-dev": "^2.0.0",
|
"ts-node-dev": "^2.0.0",
|
||||||
"typedoc": "^0.24.7",
|
"typedoc": "^0.24.7",
|
||||||
"typedoc-plugin-markdown": "^3.15.3",
|
"typedoc-plugin-markdown": "^3.15.3",
|
||||||
"typescript": "*",
|
"typescript": "^5.1.0",
|
||||||
"uuid": "^9.0.0"
|
"uuid": "^9.0.0"
|
||||||
},
|
},
|
||||||
"optionalDependencies": {
|
"optionalDependencies": {
|
||||||
"@lancedb/vectordb-darwin-arm64": "0.4.6",
|
"@lancedb/vectordb-darwin-arm64": "0.21.2-beta.0",
|
||||||
"@lancedb/vectordb-darwin-x64": "0.4.6",
|
"@lancedb/vectordb-darwin-x64": "0.21.2-beta.0",
|
||||||
"@lancedb/vectordb-linux-arm64-gnu": "0.4.6",
|
"@lancedb/vectordb-linux-arm64-gnu": "0.21.2-beta.0",
|
||||||
"@lancedb/vectordb-linux-x64-gnu": "0.4.6",
|
"@lancedb/vectordb-linux-x64-gnu": "0.21.2-beta.0",
|
||||||
"@lancedb/vectordb-win32-x64-msvc": "0.4.6"
|
"@lancedb/vectordb-win32-x64-msvc": "0.21.2-beta.0"
|
||||||
|
},
|
||||||
|
"peerDependencies": {
|
||||||
|
"@apache-arrow/ts": "^14.0.2",
|
||||||
|
"apache-arrow": "^14.0.2"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"../node/node_modules/apache-arrow": {
|
"../node/node_modules/apache-arrow": {
|
||||||
|
|||||||
@@ -18,24 +18,23 @@ See the [indexing](concepts/index_ivfpq.md) concepts guide for more information
|
|||||||
Lance supports `IVF_PQ` index type by default.
|
Lance supports `IVF_PQ` index type by default.
|
||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
Creating indexes is done via the [create_index](https://lancedb.github.io/lancedb/python/#lancedb.table.LanceTable.create_index) method.
|
Creating indexes is done via the [create_index](https://lancedb.github.io/lancedb/python/#lancedb.table.LanceTable.create_index) method.
|
||||||
|
|
||||||
```python
|
```python
|
||||||
import lancedb
|
--8<-- "python/python/tests/docs/test_guide_index.py:import-lancedb"
|
||||||
import numpy as np
|
--8<-- "python/python/tests/docs/test_guide_index.py:import-numpy"
|
||||||
uri = "data/sample-lancedb"
|
--8<-- "python/python/tests/docs/test_guide_index.py:create_ann_index"
|
||||||
db = lancedb.connect(uri)
|
```
|
||||||
|
=== "Async API"
|
||||||
|
Creating indexes is done via the [create_index](https://lancedb.github.io/lancedb/python/#lancedb.table.LanceTable.create_index) method.
|
||||||
|
|
||||||
# Create 10,000 sample vectors
|
```python
|
||||||
data = [{"vector": row, "item": f"item {i}"}
|
--8<-- "python/python/tests/docs/test_guide_index.py:import-lancedb"
|
||||||
for i, row in enumerate(np.random.random((10_000, 1536)).astype('float32'))]
|
--8<-- "python/python/tests/docs/test_guide_index.py:import-numpy"
|
||||||
|
--8<-- "python/python/tests/docs/test_guide_index.py:import-lancedb-ivfpq"
|
||||||
# Add the vectors to a table
|
--8<-- "python/python/tests/docs/test_guide_index.py:create_ann_index_async"
|
||||||
tbl = db.create_table("my_vectors", data=data)
|
|
||||||
|
|
||||||
# Create and train the index - you need to have enough data in the table for an effective training step
|
|
||||||
tbl.create_index(num_partitions=256, num_sub_vectors=96)
|
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "TypeScript"
|
=== "TypeScript"
|
||||||
@@ -45,9 +44,9 @@ Lance supports `IVF_PQ` index type by default.
|
|||||||
Creating indexes is done via the [lancedb.Table.createIndex](../js/classes/Table.md/#createIndex) method.
|
Creating indexes is done via the [lancedb.Table.createIndex](../js/classes/Table.md/#createIndex) method.
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
--8<--- "nodejs/examples/ann_indexes.ts:import"
|
--8<--- "nodejs/examples/ann_indexes.test.ts:import"
|
||||||
|
|
||||||
--8<-- "nodejs/examples/ann_indexes.ts:ingest"
|
--8<-- "nodejs/examples/ann_indexes.test.ts:ingest"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)"
|
=== "vectordb (deprecated)"
|
||||||
@@ -70,7 +69,7 @@ Lance supports `IVF_PQ` index type by default.
|
|||||||
|
|
||||||
The following IVF_PQ paramters can be specified:
|
The following IVF_PQ paramters can be specified:
|
||||||
|
|
||||||
- **distance_type**: The distance metric to use. By default it uses euclidean distance "`L2`".
|
- **distance_type**: The distance metric to use. By default it uses euclidean distance "`l2`".
|
||||||
We also support "cosine" and "dot" distance as well.
|
We also support "cosine" and "dot" distance as well.
|
||||||
- **num_partitions**: The number of partitions in the index. The default is the square root
|
- **num_partitions**: The number of partitions in the index. The default is the square root
|
||||||
of the number of rows.
|
of the number of rows.
|
||||||
@@ -83,6 +82,7 @@ The following IVF_PQ paramters can be specified:
|
|||||||
- **num_sub_vectors**: The number of sub-vectors (M) that will be created during Product Quantization (PQ).
|
- **num_sub_vectors**: The number of sub-vectors (M) that will be created during Product Quantization (PQ).
|
||||||
For D dimensional vector, it will be divided into `M` subvectors with dimension `D/M`, each of which is replaced by
|
For D dimensional vector, it will be divided into `M` subvectors with dimension `D/M`, each of which is replaced by
|
||||||
a single PQ code. The default is the dimension of the vector divided by 16.
|
a single PQ code. The default is the dimension of the vector divided by 16.
|
||||||
|
- **num_bits**: The number of bits used to encode each sub-vector. Only 4 and 8 are supported. The higher the number of bits, the higher the accuracy of the index, also the slower search. The default is 8.
|
||||||
|
|
||||||
!!! note
|
!!! note
|
||||||
|
|
||||||
@@ -126,6 +126,8 @@ You can specify the GPU device to train IVF partitions via
|
|||||||
accelerator="mps"
|
accelerator="mps"
|
||||||
)
|
)
|
||||||
```
|
```
|
||||||
|
!!! note
|
||||||
|
GPU based indexing is not yet supported with our asynchronous client.
|
||||||
|
|
||||||
Troubleshooting:
|
Troubleshooting:
|
||||||
|
|
||||||
@@ -140,22 +142,26 @@ There are a couple of parameters that can be used to fine-tune the search:
|
|||||||
|
|
||||||
- **limit** (default: 10): The amount of results that will be returned
|
- **limit** (default: 10): The amount of results that will be returned
|
||||||
- **nprobes** (default: 20): The number of probes used. A higher number makes search more accurate but also slower.<br/>
|
- **nprobes** (default: 20): The number of probes used. A higher number makes search more accurate but also slower.<br/>
|
||||||
Most of the time, setting nprobes to cover 5-10% of the dataset should achieve high recall with low latency.<br/>
|
Most of the time, setting nprobes to cover 5-15% of the dataset should achieve high recall with low latency.<br/>
|
||||||
e.g., for 1M vectors divided up into 256 partitions, nprobes should be set to ~20-40.<br/>
|
- _For example_, For a dataset of 1 million vectors divided into 256 partitions, `nprobes` should be set to ~20-40. This value can be adjusted to achieve the optimal balance between search latency and search quality. <br/>
|
||||||
Note: nprobes is only applicable if an ANN index is present. If specified on a table without an ANN index, it is ignored.
|
|
||||||
- **refine_factor** (default: None): Refine the results by reading extra elements and re-ranking them in memory.<br/>
|
- **refine_factor** (default: None): Refine the results by reading extra elements and re-ranking them in memory.<br/>
|
||||||
A higher number makes search more accurate but also slower. If you find the recall is less than ideal, try refine_factor=10 to start.<br/>
|
A higher number makes search more accurate but also slower. If you find the recall is less than ideal, try refine_factor=10 to start.<br/>
|
||||||
e.g., for 1M vectors divided into 256 partitions, if you're looking for top 20, then refine_factor=200 reranks the whole partition.<br/>
|
- _For example_, For a dataset of 1 million vectors divided into 256 partitions, setting the `refine_factor` to 200 will initially retrieve the top 4,000 candidates (top k * refine_factor) from all searched partitions. These candidates are then reranked to determine the final top 20 results.<br/>
|
||||||
Note: refine_factor is only applicable if an ANN index is present. If specified on a table without an ANN index, it is ignored.
|
!!! note
|
||||||
|
Both `nprobes` and `refine_factor` are only applicable if an ANN index is present. If specified on a table without an ANN index, those parameters are ignored.
|
||||||
|
|
||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
```python
|
```python
|
||||||
tbl.search(np.random.random((1536))) \
|
--8<-- "python/python/tests/docs/test_guide_index.py:vector_search"
|
||||||
.limit(2) \
|
```
|
||||||
.nprobes(20) \
|
=== "Async API"
|
||||||
.refine_factor(10) \
|
|
||||||
.to_pandas()
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_guide_index.py:vector_search_async"
|
||||||
```
|
```
|
||||||
|
|
||||||
```text
|
```text
|
||||||
@@ -169,7 +175,7 @@ There are a couple of parameters that can be used to fine-tune the search:
|
|||||||
=== "@lancedb/lancedb"
|
=== "@lancedb/lancedb"
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
--8<-- "nodejs/examples/ann_indexes.ts:search1"
|
--8<-- "nodejs/examples/ann_indexes.test.ts:search1"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)"
|
=== "vectordb (deprecated)"
|
||||||
@@ -193,9 +199,15 @@ The search will return the data requested in addition to the distance of each it
|
|||||||
You can further filter the elements returned by a search using a where clause.
|
You can further filter the elements returned by a search using a where clause.
|
||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
```python
|
```python
|
||||||
tbl.search(np.random.random((1536))).where("item != 'item 1141'").to_pandas()
|
--8<-- "python/python/tests/docs/test_guide_index.py:vector_search_with_filter"
|
||||||
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_guide_index.py:vector_search_async_with_filter"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "TypeScript"
|
=== "TypeScript"
|
||||||
@@ -203,7 +215,7 @@ You can further filter the elements returned by a search using a where clause.
|
|||||||
=== "@lancedb/lancedb"
|
=== "@lancedb/lancedb"
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
--8<-- "nodejs/examples/ann_indexes.ts:search2"
|
--8<-- "nodejs/examples/ann_indexes.test.ts:search2"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)"
|
=== "vectordb (deprecated)"
|
||||||
@@ -218,10 +230,16 @@ You can select the columns returned by the query using a select clause.
|
|||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
|
||||||
```python
|
=== "Sync API"
|
||||||
tbl.search(np.random.random((1536))).select(["vector"]).to_pandas()
|
|
||||||
```
|
|
||||||
|
|
||||||
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_guide_index.py:vector_search_with_select"
|
||||||
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_guide_index.py:vector_search_async_with_select"
|
||||||
|
```
|
||||||
|
|
||||||
```text
|
```text
|
||||||
vector _distance
|
vector _distance
|
||||||
@@ -235,7 +253,7 @@ You can select the columns returned by the query using a select clause.
|
|||||||
=== "@lancedb/lancedb"
|
=== "@lancedb/lancedb"
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
--8<-- "nodejs/examples/ann_indexes.ts:search3"
|
--8<-- "nodejs/examples/ann_indexes.test.ts:search3"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)"
|
=== "vectordb (deprecated)"
|
||||||
@@ -273,9 +291,17 @@ Product quantization can lead to approximately `16 * sizeof(float32) / 1 = 64` t
|
|||||||
|
|
||||||
`num_partitions` is used to decide how many partitions the first level `IVF` index uses.
|
`num_partitions` is used to decide how many partitions the first level `IVF` index uses.
|
||||||
Higher number of partitions could lead to more efficient I/O during queries and better accuracy, but it takes much more time to train.
|
Higher number of partitions could lead to more efficient I/O during queries and better accuracy, but it takes much more time to train.
|
||||||
On `SIFT-1M` dataset, our benchmark shows that keeping each partition 1K-4K rows lead to a good latency / recall.
|
On `SIFT-1M` dataset, our benchmark shows that keeping each partition 4K-8K rows lead to a good latency / recall.
|
||||||
|
|
||||||
`num_sub_vectors` specifies how many Product Quantization (PQ) short codes to generate on each vector. Because
|
`num_sub_vectors` specifies how many Product Quantization (PQ) short codes to generate on each vector. The number should be a factor of the vector dimension. Because
|
||||||
PQ is a lossy compression of the original vector, a higher `num_sub_vectors` usually results in
|
PQ is a lossy compression of the original vector, a higher `num_sub_vectors` usually results in
|
||||||
less space distortion, and thus yields better accuracy. However, a higher `num_sub_vectors` also causes heavier I/O and
|
less space distortion, and thus yields better accuracy. However, a higher `num_sub_vectors` also causes heavier I/O and more PQ computation, and thus, higher latency. `dimension / num_sub_vectors` should be a multiple of 8 for optimum SIMD efficiency.
|
||||||
more PQ computation, and thus, higher latency. `dimension / num_sub_vectors` should be a multiple of 8 for optimum SIMD efficiency.
|
|
||||||
|
!!! note
|
||||||
|
if `num_sub_vectors` is set to be greater than the vector dimension, you will see errors like `attempt to divide by zero`
|
||||||
|
|
||||||
|
### How to choose `m` and `ef_construction` for `IVF_HNSW_*` index?
|
||||||
|
|
||||||
|
`m` determines the number of connections a new node establishes with its closest neighbors upon entering the graph. Typically, `m` falls within the range of 5 to 48. Lower `m` values are suitable for low-dimensional data or scenarios where recall is less critical. Conversely, higher `m` values are beneficial for high-dimensional data or when high recall is required. In essence, a larger `m` results in a denser graph with increased connectivity, but at the expense of higher memory consumption.
|
||||||
|
|
||||||
|
`ef_construction` balances build speed and accuracy. Higher values increase accuracy but slow down the build process. A typical range is 150 to 300. For good search results, a minimum value of 100 is recommended. In most cases, setting this value above 500 offers no additional benefit. Ensure that `ef_construction` is always set to a value equal to or greater than `ef` in the search phase
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ import * as vectordb from "vectordb";
|
|||||||
// --8<-- [end:import]
|
// --8<-- [end:import]
|
||||||
|
|
||||||
(async () => {
|
(async () => {
|
||||||
|
console.log("ann_indexes.ts: start");
|
||||||
// --8<-- [start:ingest]
|
// --8<-- [start:ingest]
|
||||||
const db = await vectordb.connect("data/sample-lancedb");
|
const db = await vectordb.connect("data/sample-lancedb");
|
||||||
|
|
||||||
@@ -49,5 +50,5 @@ import * as vectordb from "vectordb";
|
|||||||
.execute();
|
.execute();
|
||||||
// --8<-- [end:search3]
|
// --8<-- [end:search3]
|
||||||
|
|
||||||
console.log("Ann indexes: done");
|
console.log("ann_indexes.ts: done");
|
||||||
})();
|
})();
|
||||||
|
|||||||
BIN
docs/src/assets/hero-header.png
Normal file
BIN
docs/src/assets/hero-header.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 1.7 MiB |
BIN
docs/src/assets/lancedb.png
Normal file
BIN
docs/src/assets/lancedb.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 40 KiB |
BIN
docs/src/assets/maxsim.png
Normal file
BIN
docs/src/assets/maxsim.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 10 KiB |
@@ -133,22 +133,23 @@ recommend switching to stable releases.
|
|||||||
## Connect to a database
|
## Connect to a database
|
||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
```python
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:imports"
|
--8<-- "python/python/tests/docs/test_basic.py:imports"
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:connect"
|
|
||||||
|
|
||||||
|
--8<-- "python/python/tests/docs/test_basic.py:set_uri"
|
||||||
|
--8<-- "python/python/tests/docs/test_basic.py:connect"
|
||||||
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_basic.py:imports"
|
||||||
|
|
||||||
|
--8<-- "python/python/tests/docs/test_basic.py:set_uri"
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:connect_async"
|
--8<-- "python/python/tests/docs/test_basic.py:connect_async"
|
||||||
```
|
```
|
||||||
|
|
||||||
!!! note "Asynchronous Python API"
|
|
||||||
|
|
||||||
The asynchronous Python API is new and has some slight differences compared
|
|
||||||
to the synchronous API. Feel free to start using the asynchronous version.
|
|
||||||
Once all features have migrated we will start to move the synchronous API to
|
|
||||||
use the same syntax as the asynchronous API. To help with this migration we
|
|
||||||
have created a [migration guide](migration.md) detailing the differences.
|
|
||||||
|
|
||||||
=== "Typescript[^1]"
|
=== "Typescript[^1]"
|
||||||
|
|
||||||
=== "@lancedb/lancedb"
|
=== "@lancedb/lancedb"
|
||||||
@@ -157,7 +158,7 @@ recommend switching to stable releases.
|
|||||||
import * as lancedb from "@lancedb/lancedb";
|
import * as lancedb from "@lancedb/lancedb";
|
||||||
import * as arrow from "apache-arrow";
|
import * as arrow from "apache-arrow";
|
||||||
|
|
||||||
--8<-- "nodejs/examples/basic.ts:connect"
|
--8<-- "nodejs/examples/basic.test.ts:connect"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)"
|
=== "vectordb (deprecated)"
|
||||||
@@ -191,19 +192,31 @@ table.
|
|||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
|
||||||
```python
|
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:create_table"
|
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:create_table_async"
|
|
||||||
```
|
|
||||||
|
|
||||||
If the table already exists, LanceDB will raise an error by default.
|
If the table already exists, LanceDB will raise an error by default.
|
||||||
If you want to overwrite the table, you can pass in `mode="overwrite"`
|
If you want to overwrite the table, you can pass in `mode="overwrite"`
|
||||||
to the `create_table` method.
|
to the `create_table` method.
|
||||||
|
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_basic.py:create_table"
|
||||||
|
```
|
||||||
|
|
||||||
You can also pass in a pandas DataFrame directly:
|
You can also pass in a pandas DataFrame directly:
|
||||||
|
|
||||||
```python
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:create_table_pandas"
|
--8<-- "python/python/tests/docs/test_basic.py:create_table_pandas"
|
||||||
|
```
|
||||||
|
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_basic.py:create_table_async"
|
||||||
|
```
|
||||||
|
|
||||||
|
You can also pass in a pandas DataFrame directly:
|
||||||
|
|
||||||
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:create_table_async_pandas"
|
--8<-- "python/python/tests/docs/test_basic.py:create_table_async_pandas"
|
||||||
```
|
```
|
||||||
|
|
||||||
@@ -212,7 +225,7 @@ table.
|
|||||||
=== "@lancedb/lancedb"
|
=== "@lancedb/lancedb"
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
--8<-- "nodejs/examples/basic.ts:create_table"
|
--8<-- "nodejs/examples/basic.test.ts:create_table"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)"
|
=== "vectordb (deprecated)"
|
||||||
@@ -255,8 +268,14 @@ similar to a `CREATE TABLE` statement in SQL.
|
|||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
```python
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:create_empty_table"
|
--8<-- "python/python/tests/docs/test_basic.py:create_empty_table"
|
||||||
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:create_empty_table_async"
|
--8<-- "python/python/tests/docs/test_basic.py:create_empty_table_async"
|
||||||
```
|
```
|
||||||
|
|
||||||
@@ -268,7 +287,7 @@ similar to a `CREATE TABLE` statement in SQL.
|
|||||||
=== "@lancedb/lancedb"
|
=== "@lancedb/lancedb"
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
--8<-- "nodejs/examples/basic.ts:create_empty_table"
|
--8<-- "nodejs/examples/basic.test.ts:create_empty_table"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)"
|
=== "vectordb (deprecated)"
|
||||||
@@ -289,8 +308,14 @@ Once created, you can open a table as follows:
|
|||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
```python
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:open_table"
|
--8<-- "python/python/tests/docs/test_basic.py:open_table"
|
||||||
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:open_table_async"
|
--8<-- "python/python/tests/docs/test_basic.py:open_table_async"
|
||||||
```
|
```
|
||||||
|
|
||||||
@@ -298,7 +323,7 @@ Once created, you can open a table as follows:
|
|||||||
=== "@lancedb/lancedb"
|
=== "@lancedb/lancedb"
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
--8<-- "nodejs/examples/basic.ts:open_table"
|
--8<-- "nodejs/examples/basic.test.ts:open_table"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)"
|
=== "vectordb (deprecated)"
|
||||||
@@ -318,8 +343,14 @@ If you forget the name of your table, you can always get a listing of all table
|
|||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
```python
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:table_names"
|
--8<-- "python/python/tests/docs/test_basic.py:table_names"
|
||||||
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:table_names_async"
|
--8<-- "python/python/tests/docs/test_basic.py:table_names_async"
|
||||||
```
|
```
|
||||||
|
|
||||||
@@ -327,7 +358,7 @@ If you forget the name of your table, you can always get a listing of all table
|
|||||||
=== "@lancedb/lancedb"
|
=== "@lancedb/lancedb"
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
--8<-- "nodejs/examples/basic.ts:table_names"
|
--8<-- "nodejs/examples/basic.test.ts:table_names"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)"
|
=== "vectordb (deprecated)"
|
||||||
@@ -348,8 +379,14 @@ After a table has been created, you can always add more data to it as follows:
|
|||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
```python
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:add_data"
|
--8<-- "python/python/tests/docs/test_basic.py:add_data"
|
||||||
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:add_data_async"
|
--8<-- "python/python/tests/docs/test_basic.py:add_data_async"
|
||||||
```
|
```
|
||||||
|
|
||||||
@@ -357,7 +394,7 @@ After a table has been created, you can always add more data to it as follows:
|
|||||||
=== "@lancedb/lancedb"
|
=== "@lancedb/lancedb"
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
--8<-- "nodejs/examples/basic.ts:add_data"
|
--8<-- "nodejs/examples/basic.test.ts:add_data"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)"
|
=== "vectordb (deprecated)"
|
||||||
@@ -378,8 +415,14 @@ Once you've embedded the query, you can find its nearest neighbors as follows:
|
|||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
```python
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:vector_search"
|
--8<-- "python/python/tests/docs/test_basic.py:vector_search"
|
||||||
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:vector_search_async"
|
--8<-- "python/python/tests/docs/test_basic.py:vector_search_async"
|
||||||
```
|
```
|
||||||
|
|
||||||
@@ -389,7 +432,7 @@ Once you've embedded the query, you can find its nearest neighbors as follows:
|
|||||||
=== "@lancedb/lancedb"
|
=== "@lancedb/lancedb"
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
--8<-- "nodejs/examples/basic.ts:vector_search"
|
--8<-- "nodejs/examples/basic.test.ts:vector_search"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)"
|
=== "vectordb (deprecated)"
|
||||||
@@ -420,8 +463,14 @@ LanceDB allows you to create an ANN index on a table as follows:
|
|||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
|
||||||
```py
|
=== "Sync API"
|
||||||
|
|
||||||
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:create_index"
|
--8<-- "python/python/tests/docs/test_basic.py:create_index"
|
||||||
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:create_index_async"
|
--8<-- "python/python/tests/docs/test_basic.py:create_index_async"
|
||||||
```
|
```
|
||||||
|
|
||||||
@@ -429,7 +478,7 @@ LanceDB allows you to create an ANN index on a table as follows:
|
|||||||
=== "@lancedb/lancedb"
|
=== "@lancedb/lancedb"
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
--8<-- "nodejs/examples/basic.ts:create_index"
|
--8<-- "nodejs/examples/basic.test.ts:create_index"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)"
|
=== "vectordb (deprecated)"
|
||||||
@@ -459,8 +508,14 @@ This can delete any number of rows that match the filter.
|
|||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
```python
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:delete_rows"
|
--8<-- "python/python/tests/docs/test_basic.py:delete_rows"
|
||||||
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:delete_rows_async"
|
--8<-- "python/python/tests/docs/test_basic.py:delete_rows_async"
|
||||||
```
|
```
|
||||||
|
|
||||||
@@ -469,7 +524,7 @@ This can delete any number of rows that match the filter.
|
|||||||
=== "@lancedb/lancedb"
|
=== "@lancedb/lancedb"
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
--8<-- "nodejs/examples/basic.ts:delete_rows"
|
--8<-- "nodejs/examples/basic.test.ts:delete_rows"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)"
|
=== "vectordb (deprecated)"
|
||||||
@@ -491,7 +546,10 @@ simple or complex as needed. To see what expressions are supported, see the
|
|||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
|
||||||
|
=== "Sync API"
|
||||||
Read more: [lancedb.table.Table.delete][]
|
Read more: [lancedb.table.Table.delete][]
|
||||||
|
=== "Async API"
|
||||||
|
Read more: [lancedb.table.AsyncTable.delete][]
|
||||||
|
|
||||||
=== "Typescript[^1]"
|
=== "Typescript[^1]"
|
||||||
|
|
||||||
@@ -513,8 +571,14 @@ Use the `drop_table()` method on the database to remove a table.
|
|||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
```python
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:drop_table"
|
--8<-- "python/python/tests/docs/test_basic.py:drop_table"
|
||||||
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_basic.py:drop_table_async"
|
--8<-- "python/python/tests/docs/test_basic.py:drop_table_async"
|
||||||
```
|
```
|
||||||
|
|
||||||
@@ -527,7 +591,7 @@ Use the `drop_table()` method on the database to remove a table.
|
|||||||
=== "@lancedb/lancedb"
|
=== "@lancedb/lancedb"
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
--8<-- "nodejs/examples/basic.ts:drop_table"
|
--8<-- "nodejs/examples/basic.test.ts:drop_table"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)"
|
=== "vectordb (deprecated)"
|
||||||
@@ -551,18 +615,25 @@ You can use the embedding API when working with embedding models. It automatical
|
|||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
```python
|
```python
|
||||||
--8<-- "python/python/tests/docs/test_embeddings_optional.py:imports"
|
--8<-- "python/python/tests/docs/test_embeddings_optional.py:imports"
|
||||||
|
|
||||||
--8<-- "python/python/tests/docs/test_embeddings_optional.py:openai_embeddings"
|
--8<-- "python/python/tests/docs/test_embeddings_optional.py:openai_embeddings"
|
||||||
```
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
Coming soon to the async API.
|
||||||
|
https://github.com/lancedb/lancedb/issues/1938
|
||||||
|
|
||||||
=== "Typescript[^1]"
|
=== "Typescript[^1]"
|
||||||
|
|
||||||
=== "@lancedb/lancedb"
|
=== "@lancedb/lancedb"
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
--8<-- "nodejs/examples/embedding.ts:imports"
|
--8<-- "nodejs/examples/embedding.test.ts:imports"
|
||||||
--8<-- "nodejs/examples/embedding.ts:openai_embeddings"
|
--8<-- "nodejs/examples/embedding.test.ts:openai_embeddings"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "Rust"
|
=== "Rust"
|
||||||
@@ -572,7 +643,7 @@ You can use the embedding API when working with embedding models. It automatical
|
|||||||
--8<-- "rust/lancedb/examples/openai.rs:openai_embeddings"
|
--8<-- "rust/lancedb/examples/openai.rs:openai_embeddings"
|
||||||
```
|
```
|
||||||
|
|
||||||
Learn about using the existing integrations and creating custom embedding functions in the [embedding API guide](./embeddings/).
|
Learn about using the existing integrations and creating custom embedding functions in the [embedding API guide](./embeddings/index.md).
|
||||||
|
|
||||||
|
|
||||||
## What's next
|
## What's next
|
||||||
|
|||||||
@@ -107,7 +107,6 @@ const example = async () => {
|
|||||||
// --8<-- [start:search]
|
// --8<-- [start:search]
|
||||||
const query = await tbl.search([100, 100]).limit(2).execute();
|
const query = await tbl.search([100, 100]).limit(2).execute();
|
||||||
// --8<-- [end:search]
|
// --8<-- [end:search]
|
||||||
console.log(query);
|
|
||||||
|
|
||||||
// --8<-- [start:delete]
|
// --8<-- [start:delete]
|
||||||
await tbl.delete('item = "fizz"');
|
await tbl.delete('item = "fizz"');
|
||||||
@@ -119,8 +118,9 @@ const example = async () => {
|
|||||||
};
|
};
|
||||||
|
|
||||||
async function main() {
|
async function main() {
|
||||||
|
console.log("basic_legacy.ts: start");
|
||||||
await example();
|
await example();
|
||||||
console.log("Basic example: done");
|
console.log("basic_legacy.ts: done");
|
||||||
}
|
}
|
||||||
|
|
||||||
main();
|
main();
|
||||||
|
|||||||
34
docs/src/cloud/cloud_faq.md
Normal file
34
docs/src/cloud/cloud_faq.md
Normal file
@@ -0,0 +1,34 @@
|
|||||||
|
This section provides answers to the most common questions asked about LanceDB Cloud. By following these guidelines, you can ensure a smooth, performant experience with LanceDB Cloud.
|
||||||
|
|
||||||
|
### Should I reuse the database connection?
|
||||||
|
Yes! It is recommended to establish a single database connection and maintain it throughout your interaction with the tables within.
|
||||||
|
|
||||||
|
LanceDB uses HTTP connections to communicate with the servers. By re-using the Connection object, you avoid the overhead of repeatedly establishing HTTP connections, significantly improving efficiency.
|
||||||
|
|
||||||
|
### Should I re-use the `Table` object?
|
||||||
|
`table = db.open_table()` should be called once and used for all subsequent table operations. If there are changes to the opened table, `table` always reflect the **latest version** of the data.
|
||||||
|
|
||||||
|
### What should I do if I need to search for rows by `id`?
|
||||||
|
LanceDB Cloud currently does not support an ID or primary key column. You are recommended to add a
|
||||||
|
user-defined ID column. To significantly improve the query performance with SQL causes, a scalar BITMAP/BTREE index should be created on this column.
|
||||||
|
|
||||||
|
### What are the vector indexing types supported by LanceDB Cloud?
|
||||||
|
We support `IVF_PQ` and `IVF_HNSW_SQ` as the `index_type` which is passed to `create_index`. LanceDB Cloud tunes the indexing parameters automatically to achieve the best tradeoff between query latency and query quality.
|
||||||
|
|
||||||
|
### When I add new rows to a table, do I need to manually update the index?
|
||||||
|
No! LanceDB Cloud triggers an asynchronous background job to index the new vectors.
|
||||||
|
|
||||||
|
Even though indexing is asynchronous, your vectors will still be immediately searchable. LanceDB uses brute-force search to search over unindexed rows. This makes you new data is immediately available, but does increase latency temporarily. To disable the brute-force part of search, set the `fast_search` flag in your query to `true`.
|
||||||
|
|
||||||
|
### Do I need to reindex the whole dataset if only a small portion of the data is deleted or updated?
|
||||||
|
No! Similar to adding data to the table, LanceDB Cloud triggers an asynchronous background job to update the existing indices. Therefore, no action is needed from users and there is absolutely no
|
||||||
|
downtime expected.
|
||||||
|
|
||||||
|
### How do I know whether an index has been created?
|
||||||
|
While index creation in LanceDB Cloud is generally fast, querying immediately after a `create_index` call may result in errors. It's recommended to use `list_indices` to verify index creation before querying.
|
||||||
|
|
||||||
|
### Why is my query latency higher than expected?
|
||||||
|
Multiple factors can impact query latency. To reduce query latency, consider the following:
|
||||||
|
- Send pre-warm queries: send a few queries to warm up the cache before an actual user query.
|
||||||
|
- Check network latency: LanceDB Cloud is hosted in AWS `us-east-1` region. It is recommended to run queries from an EC2 instance that is in the same region.
|
||||||
|
- Create scalar indices: If you are filtering on metadata, it is recommended to create scalar indices on those columns. This will speedup searches with metadata filtering. See [here](../guides/scalar_index.md) for more details on creating a scalar index.
|
||||||
@@ -2,7 +2,7 @@
|
|||||||
|
|
||||||
LanceDB Cloud is a SaaS (software-as-a-service) solution that runs serverless in the cloud, clearly separating storage from compute. It's designed to be highly scalable without breaking the bank. LanceDB Cloud is currently in private beta with general availability coming soon, but you can apply for early access with the private beta release by signing up below.
|
LanceDB Cloud is a SaaS (software-as-a-service) solution that runs serverless in the cloud, clearly separating storage from compute. It's designed to be highly scalable without breaking the bank. LanceDB Cloud is currently in private beta with general availability coming soon, but you can apply for early access with the private beta release by signing up below.
|
||||||
|
|
||||||
[Try out LanceDB Cloud](https://noteforms.com/forms/lancedb-mailing-list-cloud-kty1o5?notionforms=1&utm_source=notionforms){ .md-button .md-button--primary }
|
[Try out LanceDB Cloud (Public Beta)](https://cloud.lancedb.com){ .md-button .md-button--primary }
|
||||||
|
|
||||||
## Architecture
|
## Architecture
|
||||||
|
|
||||||
|
|||||||
@@ -37,6 +37,10 @@ Depending on the use case and dataset, optimal compaction will have different re
|
|||||||
- It’s always better to use *batch* inserts rather than adding 1 row at a time (to avoid too small fragments). If single-row inserts are unavoidable, run compaction on a regular basis to merge them into larger fragments.
|
- It’s always better to use *batch* inserts rather than adding 1 row at a time (to avoid too small fragments). If single-row inserts are unavoidable, run compaction on a regular basis to merge them into larger fragments.
|
||||||
- Keep the number of fragments under 100, which is suitable for most use cases (for *really* large datasets of >500M rows, more fragments might be needed)
|
- Keep the number of fragments under 100, which is suitable for most use cases (for *really* large datasets of >500M rows, more fragments might be needed)
|
||||||
|
|
||||||
|
!!! note
|
||||||
|
|
||||||
|
LanceDB Cloud/Enterprise supports [auto-compaction](https://docs.lancedb.com/enterprise/architecture/architecture#write-path) which automatically optimizes fragments in the background as data changes.
|
||||||
|
|
||||||
## Deletion
|
## Deletion
|
||||||
|
|
||||||
Although Lance allows you to delete rows from a dataset, it does not actually delete the data immediately. It simply marks the row as deleted in the `DataFile` that represents a fragment. For a given version of the dataset, each fragment can have up to one deletion file (if no rows were ever deleted from that fragment, it will not have a deletion file). This is important to keep in mind because it means that the data is still there, and can be recovered if needed, as long as that version still exists based on your backup policy.
|
Although Lance allows you to delete rows from a dataset, it does not actually delete the data immediately. It simply marks the row as deleted in the `DataFile` that represents a fragment. For a given version of the dataset, each fragment can have up to one deletion file (if no rows were ever deleted from that fragment, it will not have a deletion file). This is important to keep in mind because it means that the data is still there, and can be recovered if needed, as long as that version still exists based on your backup policy.
|
||||||
@@ -50,13 +54,9 @@ Reindexing is the process of updating the index to account for new data, keeping
|
|||||||
|
|
||||||
Both LanceDB OSS and Cloud support reindexing, but the process (at least for now) is different for each, depending on the type of index.
|
Both LanceDB OSS and Cloud support reindexing, but the process (at least for now) is different for each, depending on the type of index.
|
||||||
|
|
||||||
When a reindex job is triggered in the background, the entire data is reindexed, but in the interim as new queries come in, LanceDB will combine results from the existing index with exhaustive kNN search on the new data. This is done to ensure that you're still searching on all your data, but it does come at a performance cost. The more data that you add without reindexing, the impact on latency (due to exhaustive search) can be noticeable.
|
In LanceDB OSS, re-indexing happens synchronously when you call either `create_index` or `optimize` on a table. In LanceDB Cloud, re-indexing happens asynchronously as you add and update data in your table.
|
||||||
|
|
||||||
### Vector reindex
|
By default, queries will search new data even if it has yet to be indexed. This is done using brute-force methods, such as kNN for vector search, and combined with the fast index search results. This is done to ensure that you're always searching over all your data, but it does come at a performance cost. Without reindexing, adding more data to a table will make queries slower and more expensive. This behavior can be disabled by setting the [fast_search](https://lancedb.github.io/lancedb/python/python/#lancedb.query.AsyncQuery.fast_search) parameter which will instruct the query to ignore un-indexed data.
|
||||||
|
|
||||||
* LanceDB Cloud supports incremental reindexing, where a background process will trigger a new index build for you automatically when new data is added to a dataset
|
* LanceDB Cloud/Enterprise supports [automatic incremental reindexing](https://docs.lancedb.com/core#vector-index) for vector, scalar, and FTS indices, where a background process will trigger a new index build for you automatically when new data is added or modified in a dataset
|
||||||
* LanceDB OSS requires you to manually trigger a reindex operation -- we are working on adding incremental reindexing to LanceDB OSS as well
|
* LanceDB OSS requires you to manually trigger a reindex operation -- we are working on adding incremental reindexing to LanceDB OSS as well
|
||||||
|
|
||||||
### FTS reindex
|
|
||||||
|
|
||||||
FTS reindexing is supported in both LanceDB OSS and Cloud, but requires that it's manually rebuilt once you have a significant enough amount of new data added that needs to be reindexed. We [updated](https://github.com/lancedb/lancedb/pull/762) Tantivy's default heap size from 128MB to 1GB in LanceDB to make it much faster to reindex, by up to 10x from the default settings.
|
|
||||||
|
|||||||
99
docs/src/concepts/index_hnsw.md
Normal file
99
docs/src/concepts/index_hnsw.md
Normal file
@@ -0,0 +1,99 @@
|
|||||||
|
|
||||||
|
# Understanding HNSW index
|
||||||
|
|
||||||
|
Approximate Nearest Neighbor (ANN) search is a method for finding data points near a given point in a dataset, though not always the exact nearest one. HNSW is one of the most accurate and fastest Approximate Nearest Neighbour search algorithms, It’s beneficial in high-dimensional spaces where finding the same nearest neighbor would be too slow and costly
|
||||||
|
|
||||||
|
[Jump to usage](#usage)
|
||||||
|
There are three main types of ANN search algorithms:
|
||||||
|
|
||||||
|
* **Tree-based search algorithms**: Use a tree structure to organize and store data points.
|
||||||
|
* **Hash-based search algorithms**: Use a specialized geometric hash table to store and manage data points. These algorithms typically focus on theoretical guarantees, and don't usually perform as well as the other approaches in practice.
|
||||||
|
* **Graph-based search algorithms**: Use a graph structure to store data points, which can be a bit complex.
|
||||||
|
|
||||||
|
HNSW is a graph-based algorithm. All graph-based search algorithms rely on the idea of a k-nearest neighbor (or k-approximate nearest neighbor) graph, which we outline below.
|
||||||
|
HNSW also combines this with the ideas behind a classic 1-dimensional search data structure: the skip list.
|
||||||
|
|
||||||
|
## k-Nearest Neighbor Graphs and k-approximate Nearest neighbor Graphs
|
||||||
|
The k-nearest neighbor graph actually predates its use for ANN search. Its construction is quite simple:
|
||||||
|
|
||||||
|
* Each vector in the dataset is given an associated vertex.
|
||||||
|
* Each vertex has outgoing edges to its k nearest neighbors. That is, the k closest other vertices by Euclidean distance between the two corresponding vectors. This can be thought of as a "friend list" for the vertex.
|
||||||
|
* For some applications (including nearest-neighbor search), the incoming edges are also added.
|
||||||
|
|
||||||
|
Eventually, it was realized that the following greedy search method over such a graph typically results in good approximate nearest neighbors:
|
||||||
|
|
||||||
|
* Given a query vector, start at some fixed "entry point" vertex (e.g. the approximate center node).
|
||||||
|
* Look at that vertex's neighbors. If any of them are closer to the query vector than the current vertex, then move to that vertex.
|
||||||
|
* Repeat until a local optimum is found.
|
||||||
|
|
||||||
|
The above algorithm also generalizes to e.g. top 10 approximate nearest neighbors.
|
||||||
|
|
||||||
|
Computing a k-nearest neighbor graph is actually quite slow, taking quadratic time in the dataset size. It was quickly realized that near-identical performance can be achieved using a k-approximate nearest neighbor graph. That is, instead of obtaining the k-nearest neighbors for each vertex, an approximate nearest neighbor search data structure is used to build much faster.
|
||||||
|
In fact, another data structure is not needed: This can be done "incrementally".
|
||||||
|
That is, if you start with a k-ANN graph for n-1 vertices, you can extend it to a k-ANN graph for n vertices as well by using the graph to obtain the k-ANN for the new vertex.
|
||||||
|
|
||||||
|
One downside of k-NN and k-ANN graphs alone is that one must typically build them with a large value of k to get decent results, resulting in a large index.
|
||||||
|
|
||||||
|
|
||||||
|
## HNSW: Hierarchical Navigable Small Worlds
|
||||||
|
|
||||||
|
HNSW builds on k-ANN in two main ways:
|
||||||
|
|
||||||
|
* Instead of getting the k-approximate nearest neighbors for a large value of k, it sparsifies the k-ANN graph using a carefully chosen "edge pruning" heuristic, allowing for the number of edges per vertex to be limited to a relatively small constant.
|
||||||
|
* The "entry point" vertex is chosen dynamically using a recursively constructed data structure on a subset of the data, similarly to a skip list.
|
||||||
|
|
||||||
|
This recursive structure can be thought of as separating into layers:
|
||||||
|
|
||||||
|
* At the bottom-most layer, an k-ANN graph on the whole dataset is present.
|
||||||
|
* At the second layer, a k-ANN graph on a fraction of the dataset (e.g. 10%) is present.
|
||||||
|
* At the Lth layer, a k-ANN graph is present. It is over a (constant) fraction (e.g. 10%) of the vectors/vertices present in the L-1th layer.
|
||||||
|
|
||||||
|
Then the greedy search routine operates as follows:
|
||||||
|
|
||||||
|
* At the top layer (using an arbitrary vertex as an entry point), use the greedy local search routine on the k-ANN graph to get an approximate nearest neighbor at that layer.
|
||||||
|
* Using the approximate nearest neighbor found in the previous layer as an entry point, find an approximate nearest neighbor in the next layer with the same method.
|
||||||
|
* Repeat until the bottom-most layer is reached. Then use the entry point to find multiple nearest neighbors (e.g. top 10).
|
||||||
|
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
|
||||||
|
There are three key parameters to set when constructing an HNSW index:
|
||||||
|
|
||||||
|
* `metric`: Use an `l2` euclidean distance metric. We also support `dot` and `cosine` distance.
|
||||||
|
* `m`: The number of neighbors to select for each vector in the HNSW graph.
|
||||||
|
* `ef_construction`: The number of candidates to evaluate during the construction of the HNSW graph.
|
||||||
|
|
||||||
|
|
||||||
|
We can combine the above concepts to understand how to build and query an HNSW index in LanceDB.
|
||||||
|
|
||||||
|
### Construct index
|
||||||
|
|
||||||
|
```python
|
||||||
|
import lancedb
|
||||||
|
import numpy as np
|
||||||
|
uri = "/tmp/lancedb"
|
||||||
|
db = lancedb.connect(uri)
|
||||||
|
|
||||||
|
# Create 10,000 sample vectors
|
||||||
|
data = [
|
||||||
|
{"vector": row, "item": f"item {i}"}
|
||||||
|
for i, row in enumerate(np.random.random((10_000, 1536)).astype('float32'))
|
||||||
|
]
|
||||||
|
|
||||||
|
# Add the vectors to a table
|
||||||
|
tbl = db.create_table("my_vectors", data=data)
|
||||||
|
|
||||||
|
# Create and train the HNSW index for a 1536-dimensional vector
|
||||||
|
# Make sure you have enough data in the table for an effective training step
|
||||||
|
tbl.create_index(index_type=IVF_HNSW_SQ)
|
||||||
|
|
||||||
|
```
|
||||||
|
|
||||||
|
### Query the index
|
||||||
|
|
||||||
|
```python
|
||||||
|
# Search using a random 1536-dimensional embedding
|
||||||
|
tbl.search(np.random.random((1536))) \
|
||||||
|
.limit(2) \
|
||||||
|
.to_pandas()
|
||||||
|
```
|
||||||
@@ -47,7 +47,7 @@ We can combine the above concepts to understand how to build and query an IVF-PQ
|
|||||||
|
|
||||||
There are three key parameters to set when constructing an IVF-PQ index:
|
There are three key parameters to set when constructing an IVF-PQ index:
|
||||||
|
|
||||||
* `metric`: Use an `L2` euclidean distance metric. We also support `dot` and `cosine` distance.
|
* `metric`: Use an `l2` euclidean distance metric. We also support `dot` and `cosine` distance.
|
||||||
* `num_partitions`: The number of partitions in the IVF portion of the index.
|
* `num_partitions`: The number of partitions in the IVF portion of the index.
|
||||||
* `num_sub_vectors`: The number of sub-vectors that will be created during Product Quantization (PQ).
|
* `num_sub_vectors`: The number of sub-vectors that will be created during Product Quantization (PQ).
|
||||||
|
|
||||||
@@ -56,10 +56,12 @@ In Python, the index can be created as follows:
|
|||||||
```python
|
```python
|
||||||
# Create and train the index for a 1536-dimensional vector
|
# Create and train the index for a 1536-dimensional vector
|
||||||
# Make sure you have enough data in the table for an effective training step
|
# Make sure you have enough data in the table for an effective training step
|
||||||
tbl.create_index(metric="L2", num_partitions=256, num_sub_vectors=96)
|
tbl.create_index(metric="l2", num_partitions=256, num_sub_vectors=96)
|
||||||
```
|
```
|
||||||
|
!!! note
|
||||||
|
`num_partitions`=256 and `num_sub_vectors`=96 does not work for every dataset. Those values needs to be adjusted for your particular dataset.
|
||||||
|
|
||||||
The `num_partitions` is usually chosen to target a particular number of vectors per partition. `num_sub_vectors` is typically chosen based on the desired recall and the dimensionality of the vector. See the [FAQs](#faq) below for best practices on choosing these parameters.
|
The `num_partitions` is usually chosen to target a particular number of vectors per partition. `num_sub_vectors` is typically chosen based on the desired recall and the dimensionality of the vector. See [here](../ann_indexes.md/#how-to-choose-num_partitions-and-num_sub_vectors-for-ivf_pq-index) for best practices on choosing these parameters.
|
||||||
|
|
||||||
|
|
||||||
### Query the index
|
### Query the index
|
||||||
|
|||||||
@@ -0,0 +1,67 @@
|
|||||||
|
# Imagebind embeddings
|
||||||
|
We have support for [imagebind](https://github.com/facebookresearch/ImageBind) model embeddings. You can download our version of the packaged model via - `pip install imagebind-packaged==0.1.2`.
|
||||||
|
|
||||||
|
This function is registered as `imagebind` and supports Audio, Video and Text modalities(extending to Thermal,Depth,IMU data):
|
||||||
|
|
||||||
|
| Parameter | Type | Default Value | Description |
|
||||||
|
|---|---|---|---|
|
||||||
|
| `name` | `str` | `"imagebind_huge"` | Name of the model. |
|
||||||
|
| `device` | `str` | `"cpu"` | The device to run the model on. Can be `"cpu"` or `"gpu"`. |
|
||||||
|
| `normalize` | `bool` | `False` | set to `True` to normalize your inputs before model ingestion. |
|
||||||
|
|
||||||
|
Below is an example demonstrating how the API works:
|
||||||
|
|
||||||
|
```python
|
||||||
|
import lancedb
|
||||||
|
from lancedb.pydantic import LanceModel, Vector
|
||||||
|
from lancedb.embeddings import get_registry
|
||||||
|
|
||||||
|
db = lancedb.connect(tmp_path)
|
||||||
|
func = get_registry().get("imagebind").create()
|
||||||
|
|
||||||
|
class ImageBindModel(LanceModel):
|
||||||
|
text: str
|
||||||
|
image_uri: str = func.SourceField()
|
||||||
|
audio_path: str
|
||||||
|
vector: Vector(func.ndims()) = func.VectorField()
|
||||||
|
|
||||||
|
# add locally accessible image paths
|
||||||
|
text_list=["A dog.", "A car", "A bird"]
|
||||||
|
image_paths=[".assets/dog_image.jpg", ".assets/car_image.jpg", ".assets/bird_image.jpg"]
|
||||||
|
audio_paths=[".assets/dog_audio.wav", ".assets/car_audio.wav", ".assets/bird_audio.wav"]
|
||||||
|
|
||||||
|
# Load data
|
||||||
|
inputs = [
|
||||||
|
{"text": a, "audio_path": b, "image_uri": c}
|
||||||
|
for a, b, c in zip(text_list, audio_paths, image_paths)
|
||||||
|
]
|
||||||
|
|
||||||
|
#create table and add data
|
||||||
|
table = db.create_table("img_bind", schema=ImageBindModel)
|
||||||
|
table.add(inputs)
|
||||||
|
```
|
||||||
|
|
||||||
|
Now, we can search using any modality:
|
||||||
|
|
||||||
|
#### image search
|
||||||
|
```python
|
||||||
|
query_image = "./assets/dog_image2.jpg" #download an image and enter that path here
|
||||||
|
actual = table.search(query_image).limit(1).to_pydantic(ImageBindModel)[0]
|
||||||
|
print(actual.text == "dog")
|
||||||
|
```
|
||||||
|
#### audio search
|
||||||
|
|
||||||
|
```python
|
||||||
|
query_audio = "./assets/car_audio2.wav" #download an audio clip and enter path here
|
||||||
|
actual = table.search(query_audio).limit(1).to_pydantic(ImageBindModel)[0]
|
||||||
|
print(actual.text == "car")
|
||||||
|
```
|
||||||
|
#### Text search
|
||||||
|
You can add any input query and fetch the result as follows:
|
||||||
|
```python
|
||||||
|
query = "an animal which flies and tweets"
|
||||||
|
actual = table.search(query).limit(1).to_pydantic(ImageBindModel)[0]
|
||||||
|
print(actual.text == "bird")
|
||||||
|
```
|
||||||
|
|
||||||
|
If you have any questions about the embeddings API, supported models, or see a relevant model missing, please raise an issue [on GitHub](https://github.com/lancedb/lancedb/issues).
|
||||||
@@ -0,0 +1,51 @@
|
|||||||
|
# Jina Embeddings : Multimodal
|
||||||
|
|
||||||
|
Jina embeddings can also be used to embed both text and image data, only some of the models support image data and you can check the list
|
||||||
|
under [https://jina.ai/embeddings/](https://jina.ai/embeddings/)
|
||||||
|
|
||||||
|
Supported parameters (to be passed in `create` method) are:
|
||||||
|
|
||||||
|
| Parameter | Type | Default Value | Description |
|
||||||
|
|---|---|---|---|
|
||||||
|
| `name` | `str` | `"jina-clip-v1"` | The model ID of the jina model to use |
|
||||||
|
|
||||||
|
Usage Example:
|
||||||
|
|
||||||
|
```python
|
||||||
|
import os
|
||||||
|
import requests
|
||||||
|
import lancedb
|
||||||
|
from lancedb.pydantic import LanceModel, Vector
|
||||||
|
from lancedb.embeddings import get_registry
|
||||||
|
import pandas as pd
|
||||||
|
|
||||||
|
os.environ['JINA_API_KEY'] = 'jina_*'
|
||||||
|
|
||||||
|
db = lancedb.connect("~/.lancedb")
|
||||||
|
func = get_registry().get("jina").create()
|
||||||
|
|
||||||
|
|
||||||
|
class Images(LanceModel):
|
||||||
|
label: str
|
||||||
|
image_uri: str = func.SourceField() # image uri as the source
|
||||||
|
image_bytes: bytes = func.SourceField() # image bytes as the source
|
||||||
|
vector: Vector(func.ndims()) = func.VectorField() # vector column
|
||||||
|
vec_from_bytes: Vector(func.ndims()) = func.VectorField() # Another vector column
|
||||||
|
|
||||||
|
|
||||||
|
table = db.create_table("images", schema=Images)
|
||||||
|
labels = ["cat", "cat", "dog", "dog", "horse", "horse"]
|
||||||
|
uris = [
|
||||||
|
"http://farm1.staticflickr.com/53/167798175_7c7845bbbd_z.jpg",
|
||||||
|
"http://farm1.staticflickr.com/134/332220238_da527d8140_z.jpg",
|
||||||
|
"http://farm9.staticflickr.com/8387/8602747737_2e5c2a45d4_z.jpg",
|
||||||
|
"http://farm5.staticflickr.com/4092/5017326486_1f46057f5f_z.jpg",
|
||||||
|
"http://farm9.staticflickr.com/8216/8434969557_d37882c42d_z.jpg",
|
||||||
|
"http://farm6.staticflickr.com/5142/5835678453_4f3a4edb45_z.jpg",
|
||||||
|
]
|
||||||
|
# get each uri as bytes
|
||||||
|
image_bytes = [requests.get(uri).content for uri in uris]
|
||||||
|
table.add(
|
||||||
|
pd.DataFrame({"label": labels, "image_uri": uris, "image_bytes": image_bytes})
|
||||||
|
)
|
||||||
|
```
|
||||||
@@ -0,0 +1,82 @@
|
|||||||
|
# OpenClip embeddings
|
||||||
|
We support CLIP model embeddings using the open source alternative, [open-clip](https://github.com/mlfoundations/open_clip) which supports various customizations. It is registered as `open-clip` and supports the following customizations:
|
||||||
|
|
||||||
|
| Parameter | Type | Default Value | Description |
|
||||||
|
|---|---|---|---|
|
||||||
|
| `name` | `str` | `"ViT-B-32"` | The name of the model. |
|
||||||
|
| `pretrained` | `str` | `"laion2b_s34b_b79k"` | The name of the pretrained model to load. |
|
||||||
|
| `device` | `str` | `"cpu"` | The device to run the model on. Can be `"cpu"` or `"gpu"`. |
|
||||||
|
| `batch_size` | `int` | `64` | The number of images to process in a batch. |
|
||||||
|
| `normalize` | `bool` | `True` | Whether to normalize the input images before feeding them to the model. |
|
||||||
|
|
||||||
|
This embedding function supports ingesting images as both bytes and urls. You can query them using both test and other images.
|
||||||
|
|
||||||
|
!!! info
|
||||||
|
LanceDB supports ingesting images directly from accessible links.
|
||||||
|
|
||||||
|
```python
|
||||||
|
import lancedb
|
||||||
|
from lancedb.pydantic import LanceModel, Vector
|
||||||
|
from lancedb.embeddings import get_registry
|
||||||
|
|
||||||
|
db = lancedb.connect(tmp_path)
|
||||||
|
func = get_registry().get("open-clip").create()
|
||||||
|
|
||||||
|
class Images(LanceModel):
|
||||||
|
label: str
|
||||||
|
image_uri: str = func.SourceField() # image uri as the source
|
||||||
|
image_bytes: bytes = func.SourceField() # image bytes as the source
|
||||||
|
vector: Vector(func.ndims()) = func.VectorField() # vector column
|
||||||
|
vec_from_bytes: Vector(func.ndims()) = func.VectorField() # Another vector column
|
||||||
|
|
||||||
|
table = db.create_table("images", schema=Images)
|
||||||
|
labels = ["cat", "cat", "dog", "dog", "horse", "horse"]
|
||||||
|
uris = [
|
||||||
|
"http://farm1.staticflickr.com/53/167798175_7c7845bbbd_z.jpg",
|
||||||
|
"http://farm1.staticflickr.com/134/332220238_da527d8140_z.jpg",
|
||||||
|
"http://farm9.staticflickr.com/8387/8602747737_2e5c2a45d4_z.jpg",
|
||||||
|
"http://farm5.staticflickr.com/4092/5017326486_1f46057f5f_z.jpg",
|
||||||
|
"http://farm9.staticflickr.com/8216/8434969557_d37882c42d_z.jpg",
|
||||||
|
"http://farm6.staticflickr.com/5142/5835678453_4f3a4edb45_z.jpg",
|
||||||
|
]
|
||||||
|
# get each uri as bytes
|
||||||
|
image_bytes = [requests.get(uri).content for uri in uris]
|
||||||
|
table.add(
|
||||||
|
pd.DataFrame({"label": labels, "image_uri": uris, "image_bytes": image_bytes})
|
||||||
|
)
|
||||||
|
```
|
||||||
|
Now we can search using text from both the default vector column and the custom vector column
|
||||||
|
```python
|
||||||
|
|
||||||
|
# text search
|
||||||
|
actual = table.search("man's best friend").limit(1).to_pydantic(Images)[0]
|
||||||
|
print(actual.label) # prints "dog"
|
||||||
|
|
||||||
|
frombytes = (
|
||||||
|
table.search("man's best friend", vector_column_name="vec_from_bytes")
|
||||||
|
.limit(1)
|
||||||
|
.to_pydantic(Images)[0]
|
||||||
|
)
|
||||||
|
print(frombytes.label)
|
||||||
|
|
||||||
|
```
|
||||||
|
|
||||||
|
Because we're using a multi-modal embedding function, we can also search using images
|
||||||
|
|
||||||
|
```python
|
||||||
|
# image search
|
||||||
|
query_image_uri = "http://farm1.staticflickr.com/200/467715466_ed4a31801f_z.jpg"
|
||||||
|
image_bytes = requests.get(query_image_uri).content
|
||||||
|
query_image = Image.open(io.BytesIO(image_bytes))
|
||||||
|
actual = table.search(query_image).limit(1).to_pydantic(Images)[0]
|
||||||
|
print(actual.label == "dog")
|
||||||
|
|
||||||
|
# image search using a custom vector column
|
||||||
|
other = (
|
||||||
|
table.search(query_image, vector_column_name="vec_from_bytes")
|
||||||
|
.limit(1)
|
||||||
|
.to_pydantic(Images)[0]
|
||||||
|
)
|
||||||
|
print(actual.label)
|
||||||
|
|
||||||
|
```
|
||||||
@@ -0,0 +1,51 @@
|
|||||||
|
# AWS Bedrock Text Embedding Functions
|
||||||
|
|
||||||
|
AWS Bedrock supports multiple base models for generating text embeddings. You need to setup the AWS credentials to use this embedding function.
|
||||||
|
You can do so by using `awscli` and also add your session_token:
|
||||||
|
```shell
|
||||||
|
aws configure
|
||||||
|
aws configure set aws_session_token "<your_session_token>"
|
||||||
|
```
|
||||||
|
to ensure that the credentials are set up correctly, you can run the following command:
|
||||||
|
```shell
|
||||||
|
aws sts get-caller-identity
|
||||||
|
```
|
||||||
|
|
||||||
|
Supported Embedding modelIDs are:
|
||||||
|
* `amazon.titan-embed-text-v1`
|
||||||
|
* `cohere.embed-english-v3`
|
||||||
|
* `cohere.embed-multilingual-v3`
|
||||||
|
|
||||||
|
Supported parameters (to be passed in `create` method) are:
|
||||||
|
|
||||||
|
| Parameter | Type | Default Value | Description |
|
||||||
|
|---|---|---|---|
|
||||||
|
| **name** | str | "amazon.titan-embed-text-v1" | The model ID of the bedrock model to use. Supported base models for Text Embeddings: amazon.titan-embed-text-v1, cohere.embed-english-v3, cohere.embed-multilingual-v3 |
|
||||||
|
| **region** | str | "us-east-1" | Optional name of the AWS Region in which the service should be called (e.g., "us-east-1"). |
|
||||||
|
| **profile_name** | str | None | Optional name of the AWS profile to use for calling the Bedrock service. If not specified, the default profile will be used. |
|
||||||
|
| **assumed_role** | str | None | Optional ARN of an AWS IAM role to assume for calling the Bedrock service. If not specified, the current active credentials will be used. |
|
||||||
|
| **role_session_name** | str | "lancedb-embeddings" | Optional name of the AWS IAM role session to use for calling the Bedrock service. If not specified, a "lancedb-embeddings" name will be used. |
|
||||||
|
| **runtime** | bool | True | Optional choice of getting different client to perform operations with the Amazon Bedrock service. |
|
||||||
|
| **max_retries** | int | 7 | Optional number of retries to perform when a request fails. |
|
||||||
|
|
||||||
|
Usage Example:
|
||||||
|
|
||||||
|
```python
|
||||||
|
import lancedb
|
||||||
|
from lancedb.pydantic import LanceModel, Vector
|
||||||
|
from lancedb.embeddings import get_registry
|
||||||
|
import pandas as pd
|
||||||
|
|
||||||
|
model = get_registry().get("bedrock-text").create()
|
||||||
|
|
||||||
|
class TextModel(LanceModel):
|
||||||
|
text: str = model.SourceField()
|
||||||
|
vector: Vector(model.ndims()) = model.VectorField()
|
||||||
|
|
||||||
|
df = pd.DataFrame({"text": ["hello world", "goodbye world"]})
|
||||||
|
db = lancedb.connect("tmp_path")
|
||||||
|
tbl = db.create_table("test", schema=TextModel, mode="overwrite")
|
||||||
|
|
||||||
|
tbl.add(df)
|
||||||
|
rs = tbl.search("hello").limit(1).to_pandas()
|
||||||
|
```
|
||||||
@@ -0,0 +1,63 @@
|
|||||||
|
# Cohere Embeddings
|
||||||
|
|
||||||
|
Using cohere API requires cohere package, which can be installed using `pip install cohere`. Cohere embeddings are used to generate embeddings for text data. The embeddings can be used for various tasks like semantic search, clustering, and classification.
|
||||||
|
You also need to set the `COHERE_API_KEY` environment variable to use the Cohere API.
|
||||||
|
|
||||||
|
Supported models are:
|
||||||
|
|
||||||
|
- embed-english-v3.0
|
||||||
|
- embed-multilingual-v3.0
|
||||||
|
- embed-english-light-v3.0
|
||||||
|
- embed-multilingual-light-v3.0
|
||||||
|
- embed-english-v2.0
|
||||||
|
- embed-english-light-v2.0
|
||||||
|
- embed-multilingual-v2.0
|
||||||
|
|
||||||
|
|
||||||
|
Supported parameters (to be passed in `create` method) are:
|
||||||
|
|
||||||
|
| Parameter | Type | Default Value | Description |
|
||||||
|
|---|---|--------|---------|
|
||||||
|
| `name` | `str` | `"embed-english-v2.0"` | The model ID of the cohere model to use. Supported base models for Text Embeddings: embed-english-v3.0, embed-multilingual-v3.0, embed-english-light-v3.0, embed-multilingual-light-v3.0, embed-english-v2.0, embed-english-light-v2.0, embed-multilingual-v2.0 |
|
||||||
|
| `source_input_type` | `str` | `"search_document"` | The type of input data to be used for the source column. |
|
||||||
|
| `query_input_type` | `str` | `"search_query"` | The type of input data to be used for the query. |
|
||||||
|
|
||||||
|
Cohere supports following input types:
|
||||||
|
|
||||||
|
| Input Type | Description |
|
||||||
|
|-------------------------|---------------------------------------|
|
||||||
|
| "`search_document`" | Used for embeddings stored in a vector|
|
||||||
|
| | database for search use-cases. |
|
||||||
|
| "`search_query`" | Used for embeddings of search queries |
|
||||||
|
| | run against a vector DB |
|
||||||
|
| "`semantic_similarity`" | Specifies the given text will be used |
|
||||||
|
| | for Semantic Textual Similarity (STS) |
|
||||||
|
| "`classification`" | Used for embeddings passed through a |
|
||||||
|
| | text classifier. |
|
||||||
|
| "`clustering`" | Used for the embeddings run through a |
|
||||||
|
| | clustering algorithm |
|
||||||
|
|
||||||
|
Usage Example:
|
||||||
|
|
||||||
|
```python
|
||||||
|
import lancedb
|
||||||
|
from lancedb.pydantic import LanceModel, Vector
|
||||||
|
from lancedb.embeddings import EmbeddingFunctionRegistry
|
||||||
|
|
||||||
|
cohere = EmbeddingFunctionRegistry
|
||||||
|
.get_instance()
|
||||||
|
.get("cohere")
|
||||||
|
.create(name="embed-multilingual-v2.0")
|
||||||
|
|
||||||
|
class TextModel(LanceModel):
|
||||||
|
text: str = cohere.SourceField()
|
||||||
|
vector: Vector(cohere.ndims()) = cohere.VectorField()
|
||||||
|
|
||||||
|
data = [ { "text": "hello world" },
|
||||||
|
{ "text": "goodbye world" }]
|
||||||
|
|
||||||
|
db = lancedb.connect("~/.lancedb")
|
||||||
|
tbl = db.create_table("test", schema=TextModel, mode="overwrite")
|
||||||
|
|
||||||
|
tbl.add(data)
|
||||||
|
```
|
||||||
@@ -0,0 +1,35 @@
|
|||||||
|
# Gemini Embeddings
|
||||||
|
With Google's Gemini, you can represent text (words, sentences, and blocks of text) in a vectorized form, making it easier to compare and contrast embeddings. For example, two texts that share a similar subject matter or sentiment should have similar embeddings, which can be identified through mathematical comparison techniques such as cosine similarity. For more on how and why you should use embeddings, refer to the Embeddings guide.
|
||||||
|
The Gemini Embedding Model API supports various task types:
|
||||||
|
|
||||||
|
| Task Type | Description |
|
||||||
|
|-------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
||||||
|
| "`retrieval_query`" | Specifies the given text is a query in a search/retrieval setting. |
|
||||||
|
| "`retrieval_document`" | Specifies the given text is a document in a search/retrieval setting. Using this task type requires a title but is automatically proided by Embeddings API |
|
||||||
|
| "`semantic_similarity`" | Specifies the given text will be used for Semantic Textual Similarity (STS). |
|
||||||
|
| "`classification`" | Specifies that the embeddings will be used for classification. |
|
||||||
|
| "`clusering`" | Specifies that the embeddings will be used for clustering. |
|
||||||
|
|
||||||
|
|
||||||
|
Usage Example:
|
||||||
|
|
||||||
|
```python
|
||||||
|
import lancedb
|
||||||
|
import pandas as pd
|
||||||
|
from lancedb.pydantic import LanceModel, Vector
|
||||||
|
from lancedb.embeddings import get_registry
|
||||||
|
|
||||||
|
|
||||||
|
model = get_registry().get("gemini-text").create()
|
||||||
|
|
||||||
|
class TextModel(LanceModel):
|
||||||
|
text: str = model.SourceField()
|
||||||
|
vector: Vector(model.ndims()) = model.VectorField()
|
||||||
|
|
||||||
|
df = pd.DataFrame({"text": ["hello world", "goodbye world"]})
|
||||||
|
db = lancedb.connect("~/.lancedb")
|
||||||
|
tbl = db.create_table("test", schema=TextModel, mode="overwrite")
|
||||||
|
|
||||||
|
tbl.add(df)
|
||||||
|
rs = tbl.search("hello").limit(1).to_pandas()
|
||||||
|
```
|
||||||
@@ -0,0 +1,24 @@
|
|||||||
|
# Huggingface embedding models
|
||||||
|
We offer support for all Hugging Face models (which can be loaded via [transformers](https://huggingface.co/docs/transformers/en/index) library). The default model is `colbert-ir/colbertv2.0` which also has its own special callout - `registry.get("colbert")`. Some Hugging Face models might require custom models defined on the HuggingFace Hub in their own modeling files. You may enable this by setting `trust_remote_code=True`. This option should only be set to True for repositories you trust and in which you have read the code, as it will execute code present on the Hub on your local machine.
|
||||||
|
|
||||||
|
Example usage -
|
||||||
|
```python
|
||||||
|
import lancedb
|
||||||
|
import pandas as pd
|
||||||
|
|
||||||
|
from lancedb.embeddings import get_registry
|
||||||
|
from lancedb.pydantic import LanceModel, Vector
|
||||||
|
|
||||||
|
model = get_registry().get("huggingface").create(name='facebook/bart-base')
|
||||||
|
|
||||||
|
class Words(LanceModel):
|
||||||
|
text: str = model.SourceField()
|
||||||
|
vector: Vector(model.ndims()) = model.VectorField()
|
||||||
|
|
||||||
|
df = pd.DataFrame({"text": ["hi hello sayonara", "goodbye world"]})
|
||||||
|
table = db.create_table("greets", schema=Words)
|
||||||
|
table.add(df)
|
||||||
|
query = "old greeting"
|
||||||
|
actual = table.search(query).limit(1).to_pydantic(Words)[0]
|
||||||
|
print(actual.text)
|
||||||
|
```
|
||||||
@@ -0,0 +1,75 @@
|
|||||||
|
# IBM watsonx.ai Embeddings
|
||||||
|
|
||||||
|
Generate text embeddings using IBM's watsonx.ai platform.
|
||||||
|
|
||||||
|
## Supported Models
|
||||||
|
|
||||||
|
You can find a list of supported models at [IBM watsonx.ai Documentation](https://dataplatform.cloud.ibm.com/docs/content/wsj/analyze-data/fm-models-embed.html?context=wx). The currently supported model names are:
|
||||||
|
|
||||||
|
- `ibm/slate-125m-english-rtrvr`
|
||||||
|
- `ibm/slate-30m-english-rtrvr`
|
||||||
|
- `sentence-transformers/all-minilm-l12-v2`
|
||||||
|
- `intfloat/multilingual-e5-large`
|
||||||
|
|
||||||
|
## Parameters
|
||||||
|
|
||||||
|
The following parameters can be passed to the `create` method:
|
||||||
|
|
||||||
|
| Parameter | Type | Default Value | Description |
|
||||||
|
|------------|----------|----------------------------------|-----------------------------------------------------------|
|
||||||
|
| name | str | "ibm/slate-125m-english-rtrvr" | The model ID of the watsonx.ai model to use |
|
||||||
|
| api_key | str | None | Optional IBM Cloud API key (or set `WATSONX_API_KEY`) |
|
||||||
|
| project_id | str | None | Optional watsonx project ID (or set `WATSONX_PROJECT_ID`) |
|
||||||
|
| url | str | None | Optional custom URL for the watsonx.ai instance |
|
||||||
|
| params | dict | None | Optional additional parameters for the embedding model |
|
||||||
|
|
||||||
|
## Usage Example
|
||||||
|
|
||||||
|
First, the watsonx.ai library is an optional dependency, so must be installed seperately:
|
||||||
|
|
||||||
|
```
|
||||||
|
pip install ibm-watsonx-ai
|
||||||
|
```
|
||||||
|
|
||||||
|
Optionally set environment variables (if not passing credentials to `create` directly):
|
||||||
|
|
||||||
|
```sh
|
||||||
|
export WATSONX_API_KEY="YOUR_WATSONX_API_KEY"
|
||||||
|
export WATSONX_PROJECT_ID="YOUR_WATSONX_PROJECT_ID"
|
||||||
|
```
|
||||||
|
|
||||||
|
```python
|
||||||
|
import os
|
||||||
|
import lancedb
|
||||||
|
from lancedb.pydantic import LanceModel, Vector
|
||||||
|
from lancedb.embeddings import EmbeddingFunctionRegistry
|
||||||
|
|
||||||
|
watsonx_embed = EmbeddingFunctionRegistry
|
||||||
|
.get_instance()
|
||||||
|
.get("watsonx")
|
||||||
|
.create(
|
||||||
|
name="ibm/slate-125m-english-rtrvr",
|
||||||
|
# Uncomment and set these if not using environment variables
|
||||||
|
# api_key="your_api_key_here",
|
||||||
|
# project_id="your_project_id_here",
|
||||||
|
# url="your_watsonx_url_here",
|
||||||
|
# params={...},
|
||||||
|
)
|
||||||
|
|
||||||
|
class TextModel(LanceModel):
|
||||||
|
text: str = watsonx_embed.SourceField()
|
||||||
|
vector: Vector(watsonx_embed.ndims()) = watsonx_embed.VectorField()
|
||||||
|
|
||||||
|
data = [
|
||||||
|
{"text": "hello world"},
|
||||||
|
{"text": "goodbye world"},
|
||||||
|
]
|
||||||
|
|
||||||
|
db = lancedb.connect("~/.lancedb")
|
||||||
|
tbl = db.create_table("watsonx_test", schema=TextModel, mode="overwrite")
|
||||||
|
|
||||||
|
tbl.add(data)
|
||||||
|
|
||||||
|
rs = tbl.search("hello").limit(1).to_pandas()
|
||||||
|
print(rs)
|
||||||
|
```
|
||||||
@@ -0,0 +1,50 @@
|
|||||||
|
# Instructor Embeddings
|
||||||
|
[Instructor](https://instructor-embedding.github.io/) is an instruction-finetuned text embedding model that can generate text embeddings tailored to any task (e.g. classification, retrieval, clustering, text evaluation, etc.) and domains (e.g. science, finance, etc.) by simply providing the task instruction, without any finetuning.
|
||||||
|
|
||||||
|
If you want to calculate customized embeddings for specific sentences, you can follow the unified template to write instructions.
|
||||||
|
|
||||||
|
!!! info
|
||||||
|
Represent the `domain` `text_type` for `task_objective`:
|
||||||
|
|
||||||
|
* `domain` is optional, and it specifies the domain of the text, e.g. science, finance, medicine, etc.
|
||||||
|
* `text_type` is required, and it specifies the encoding unit, e.g. sentence, document, paragraph, etc.
|
||||||
|
* `task_objective` is optional, and it specifies the objective of embedding, e.g. retrieve a document, classify the sentence, etc.
|
||||||
|
|
||||||
|
More information about the model can be found at the [source URL](https://github.com/xlang-ai/instructor-embedding).
|
||||||
|
|
||||||
|
| Argument | Type | Default | Description |
|
||||||
|
|---|---|---|---|
|
||||||
|
| `name` | `str` | "hkunlp/instructor-base" | The name of the model to use |
|
||||||
|
| `batch_size` | `int` | `32` | The batch size to use when generating embeddings |
|
||||||
|
| `device` | `str` | `"cpu"` | The device to use when generating embeddings |
|
||||||
|
| `show_progress_bar` | `bool` | `True` | Whether to show a progress bar when generating embeddings |
|
||||||
|
| `normalize_embeddings` | `bool` | `True` | Whether to normalize the embeddings |
|
||||||
|
| `quantize` | `bool` | `False` | Whether to quantize the model |
|
||||||
|
| `source_instruction` | `str` | `"represent the docuement for retreival"` | The instruction for the source column |
|
||||||
|
| `query_instruction` | `str` | `"represent the document for retreiving the most similar documents"` | The instruction for the query |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
```python
|
||||||
|
import lancedb
|
||||||
|
from lancedb.pydantic import LanceModel, Vector
|
||||||
|
from lancedb.embeddings import get_registry, InstuctorEmbeddingFunction
|
||||||
|
|
||||||
|
instructor = get_registry().get("instructor").create(
|
||||||
|
source_instruction="represent the docuement for retreival",
|
||||||
|
query_instruction="represent the document for retreiving the most similar documents"
|
||||||
|
)
|
||||||
|
|
||||||
|
class Schema(LanceModel):
|
||||||
|
vector: Vector(instructor.ndims()) = instructor.VectorField()
|
||||||
|
text: str = instructor.SourceField()
|
||||||
|
|
||||||
|
db = lancedb.connect("~/.lancedb")
|
||||||
|
tbl = db.create_table("test", schema=Schema, mode="overwrite")
|
||||||
|
|
||||||
|
texts = [{"text": "Capitalism has been dominant in the Western world since the end of feudalism, but most feel[who?] that..."},
|
||||||
|
{"text": "The disparate impact theory is especially controversial under the Fair Housing Act because the Act..."},
|
||||||
|
{"text": "Disparate impact in United States labor law refers to practices in employment, housing, and other areas that.."}]
|
||||||
|
|
||||||
|
tbl.add(texts)
|
||||||
|
```
|
||||||
@@ -0,0 +1,39 @@
|
|||||||
|
# Jina Embeddings
|
||||||
|
|
||||||
|
Jina embeddings are used to generate embeddings for text and image data.
|
||||||
|
You also need to set the `JINA_API_KEY` environment variable to use the Jina API.
|
||||||
|
|
||||||
|
You can find a list of supported models under [https://jina.ai/embeddings/](https://jina.ai/embeddings/)
|
||||||
|
|
||||||
|
Supported parameters (to be passed in `create` method) are:
|
||||||
|
|
||||||
|
| Parameter | Type | Default Value | Description |
|
||||||
|
|---|---|---|---|
|
||||||
|
| `name` | `str` | `"jina-clip-v1"` | The model ID of the jina model to use |
|
||||||
|
|
||||||
|
Usage Example:
|
||||||
|
|
||||||
|
```python
|
||||||
|
import os
|
||||||
|
import lancedb
|
||||||
|
from lancedb.pydantic import LanceModel, Vector
|
||||||
|
from lancedb.embeddings import EmbeddingFunctionRegistry
|
||||||
|
|
||||||
|
os.environ['JINA_API_KEY'] = 'jina_*'
|
||||||
|
|
||||||
|
jina_embed = EmbeddingFunctionRegistry.get_instance().get("jina").create(name="jina-embeddings-v2-base-en")
|
||||||
|
|
||||||
|
|
||||||
|
class TextModel(LanceModel):
|
||||||
|
text: str = jina_embed.SourceField()
|
||||||
|
vector: Vector(jina_embed.ndims()) = jina_embed.VectorField()
|
||||||
|
|
||||||
|
|
||||||
|
data = [{"text": "hello world"},
|
||||||
|
{"text": "goodbye world"}]
|
||||||
|
|
||||||
|
db = lancedb.connect("~/.lancedb-2")
|
||||||
|
tbl = db.create_table("test", schema=TextModel, mode="overwrite")
|
||||||
|
|
||||||
|
tbl.add(data)
|
||||||
|
```
|
||||||
@@ -0,0 +1,37 @@
|
|||||||
|
# Ollama embeddings
|
||||||
|
|
||||||
|
Generate embeddings via the [ollama](https://github.com/ollama/ollama-python) python library. More details:
|
||||||
|
|
||||||
|
- [Ollama docs on embeddings](https://github.com/ollama/ollama/blob/main/docs/api.md#generate-embeddings)
|
||||||
|
- [Ollama blog on embeddings](https://ollama.com/blog/embedding-models)
|
||||||
|
|
||||||
|
| Parameter | Type | Default Value | Description |
|
||||||
|
|------------------------|----------------------------|--------------------------|------------------------------------------------------------------------------------------------------------------------------------------------|
|
||||||
|
| `name` | `str` | `nomic-embed-text` | The name of the model. |
|
||||||
|
| `host` | `str` | `http://localhost:11434` | The Ollama host to connect to. |
|
||||||
|
| `options` | `ollama.Options` or `dict` | `None` | Additional model parameters listed in the documentation for the Modelfile such as `temperature`. |
|
||||||
|
| `keep_alive` | `float` or `str` | `"5m"` | Controls how long the model will stay loaded into memory following the request. |
|
||||||
|
| `ollama_client_kwargs` | `dict` | `{}` | kwargs that can be past to the `ollama.Client`. |
|
||||||
|
|
||||||
|
```python
|
||||||
|
import lancedb
|
||||||
|
from lancedb.pydantic import LanceModel, Vector
|
||||||
|
from lancedb.embeddings import get_registry
|
||||||
|
|
||||||
|
db = lancedb.connect("/tmp/db")
|
||||||
|
func = get_registry().get("ollama").create(name="nomic-embed-text")
|
||||||
|
|
||||||
|
class Words(LanceModel):
|
||||||
|
text: str = func.SourceField()
|
||||||
|
vector: Vector(func.ndims()) = func.VectorField()
|
||||||
|
|
||||||
|
table = db.create_table("words", schema=Words, mode="overwrite")
|
||||||
|
table.add([
|
||||||
|
{"text": "hello world"},
|
||||||
|
{"text": "goodbye world"}
|
||||||
|
])
|
||||||
|
|
||||||
|
query = "greetings"
|
||||||
|
actual = table.search(query).limit(1).to_pydantic(Words)[0]
|
||||||
|
print(actual.text)
|
||||||
|
```
|
||||||
@@ -0,0 +1,35 @@
|
|||||||
|
# OpenAI embeddings
|
||||||
|
|
||||||
|
LanceDB registers the OpenAI embeddings function in the registry by default, as `openai`. Below are the parameters that you can customize when creating the instances:
|
||||||
|
|
||||||
|
| Parameter | Type | Default Value | Description |
|
||||||
|
|---|---|---|---|
|
||||||
|
| `name` | `str` | `"text-embedding-ada-002"` | The name of the model. |
|
||||||
|
| `dim` | `int` | Model default | For OpenAI's newer text-embedding-3 model, we can specify a dimensionality that is smaller than the 1536 size. This feature supports it |
|
||||||
|
| `use_azure` | bool | `False` | Set true to use Azure OpenAPI SDK |
|
||||||
|
|
||||||
|
|
||||||
|
```python
|
||||||
|
import lancedb
|
||||||
|
from lancedb.pydantic import LanceModel, Vector
|
||||||
|
from lancedb.embeddings import get_registry
|
||||||
|
|
||||||
|
db = lancedb.connect("/tmp/db")
|
||||||
|
func = get_registry().get("openai").create(name="text-embedding-ada-002")
|
||||||
|
|
||||||
|
class Words(LanceModel):
|
||||||
|
text: str = func.SourceField()
|
||||||
|
vector: Vector(func.ndims()) = func.VectorField()
|
||||||
|
|
||||||
|
table = db.create_table("words", schema=Words, mode="overwrite")
|
||||||
|
table.add(
|
||||||
|
[
|
||||||
|
{"text": "hello world"},
|
||||||
|
{"text": "goodbye world"}
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|
||||||
|
query = "greetings"
|
||||||
|
actual = table.search(query).limit(1).to_pydantic(Words)[0]
|
||||||
|
print(actual.text)
|
||||||
|
```
|
||||||
@@ -0,0 +1,174 @@
|
|||||||
|
# Sentence transformers
|
||||||
|
Allows you to set parameters when registering a `sentence-transformers` object.
|
||||||
|
|
||||||
|
!!! info
|
||||||
|
Sentence transformer embeddings are normalized by default. It is recommended to use normalized embeddings for similarity search.
|
||||||
|
|
||||||
|
| Parameter | Type | Default Value | Description |
|
||||||
|
|---|---|---|---|
|
||||||
|
| `name` | `str` | `all-MiniLM-L6-v2` | The name of the model |
|
||||||
|
| `device` | `str` | `cpu` | The device to run the model on (can be `cpu` or `gpu`) |
|
||||||
|
| `normalize` | `bool` | `True` | Whether to normalize the input text before feeding it to the model |
|
||||||
|
| `trust_remote_code` | `bool` | `False` | Whether to trust and execute remote code from the model's Huggingface repository |
|
||||||
|
|
||||||
|
|
||||||
|
??? "Check out available sentence-transformer models here!"
|
||||||
|
```markdown
|
||||||
|
- sentence-transformers/all-MiniLM-L12-v2
|
||||||
|
- sentence-transformers/paraphrase-mpnet-base-v2
|
||||||
|
- sentence-transformers/gtr-t5-base
|
||||||
|
- sentence-transformers/LaBSE
|
||||||
|
- sentence-transformers/all-MiniLM-L6-v2
|
||||||
|
- sentence-transformers/bert-base-nli-max-tokens
|
||||||
|
- sentence-transformers/bert-base-nli-mean-tokens
|
||||||
|
- sentence-transformers/bert-base-nli-stsb-mean-tokens
|
||||||
|
- sentence-transformers/bert-base-wikipedia-sections-mean-tokens
|
||||||
|
- sentence-transformers/bert-large-nli-cls-token
|
||||||
|
- sentence-transformers/bert-large-nli-max-tokens
|
||||||
|
- sentence-transformers/bert-large-nli-mean-tokens
|
||||||
|
- sentence-transformers/bert-large-nli-stsb-mean-tokens
|
||||||
|
- sentence-transformers/distilbert-base-nli-max-tokens
|
||||||
|
- sentence-transformers/distilbert-base-nli-mean-tokens
|
||||||
|
- sentence-transformers/distilbert-base-nli-stsb-mean-tokens
|
||||||
|
- sentence-transformers/distilroberta-base-msmarco-v1
|
||||||
|
- sentence-transformers/distilroberta-base-msmarco-v2
|
||||||
|
- sentence-transformers/nli-bert-base-cls-pooling
|
||||||
|
- sentence-transformers/nli-bert-base-max-pooling
|
||||||
|
- sentence-transformers/nli-bert-base
|
||||||
|
- sentence-transformers/nli-bert-large-cls-pooling
|
||||||
|
- sentence-transformers/nli-bert-large-max-pooling
|
||||||
|
- sentence-transformers/nli-bert-large
|
||||||
|
- sentence-transformers/nli-distilbert-base-max-pooling
|
||||||
|
- sentence-transformers/nli-distilbert-base
|
||||||
|
- sentence-transformers/nli-roberta-base
|
||||||
|
- sentence-transformers/nli-roberta-large
|
||||||
|
- sentence-transformers/roberta-base-nli-mean-tokens
|
||||||
|
- sentence-transformers/roberta-base-nli-stsb-mean-tokens
|
||||||
|
- sentence-transformers/roberta-large-nli-mean-tokens
|
||||||
|
- sentence-transformers/roberta-large-nli-stsb-mean-tokens
|
||||||
|
- sentence-transformers/stsb-bert-base
|
||||||
|
- sentence-transformers/stsb-bert-large
|
||||||
|
- sentence-transformers/stsb-distilbert-base
|
||||||
|
- sentence-transformers/stsb-roberta-base
|
||||||
|
- sentence-transformers/stsb-roberta-large
|
||||||
|
- sentence-transformers/xlm-r-100langs-bert-base-nli-mean-tokens
|
||||||
|
- sentence-transformers/xlm-r-100langs-bert-base-nli-stsb-mean-tokens
|
||||||
|
- sentence-transformers/xlm-r-base-en-ko-nli-ststb
|
||||||
|
- sentence-transformers/xlm-r-bert-base-nli-mean-tokens
|
||||||
|
- sentence-transformers/xlm-r-bert-base-nli-stsb-mean-tokens
|
||||||
|
- sentence-transformers/xlm-r-large-en-ko-nli-ststb
|
||||||
|
- sentence-transformers/bert-base-nli-cls-token
|
||||||
|
- sentence-transformers/all-distilroberta-v1
|
||||||
|
- sentence-transformers/multi-qa-MiniLM-L6-dot-v1
|
||||||
|
- sentence-transformers/multi-qa-distilbert-cos-v1
|
||||||
|
- sentence-transformers/multi-qa-distilbert-dot-v1
|
||||||
|
- sentence-transformers/multi-qa-mpnet-base-cos-v1
|
||||||
|
- sentence-transformers/multi-qa-mpnet-base-dot-v1
|
||||||
|
- sentence-transformers/nli-distilroberta-base-v2
|
||||||
|
- sentence-transformers/all-MiniLM-L6-v1
|
||||||
|
- sentence-transformers/all-mpnet-base-v1
|
||||||
|
- sentence-transformers/all-mpnet-base-v2
|
||||||
|
- sentence-transformers/all-roberta-large-v1
|
||||||
|
- sentence-transformers/allenai-specter
|
||||||
|
- sentence-transformers/average_word_embeddings_glove.6B.300d
|
||||||
|
- sentence-transformers/average_word_embeddings_glove.840B.300d
|
||||||
|
- sentence-transformers/average_word_embeddings_komninos
|
||||||
|
- sentence-transformers/average_word_embeddings_levy_dependency
|
||||||
|
- sentence-transformers/clip-ViT-B-32-multilingual-v1
|
||||||
|
- sentence-transformers/clip-ViT-B-32
|
||||||
|
- sentence-transformers/distilbert-base-nli-stsb-quora-ranking
|
||||||
|
- sentence-transformers/distilbert-multilingual-nli-stsb-quora-ranking
|
||||||
|
- sentence-transformers/distilroberta-base-paraphrase-v1
|
||||||
|
- sentence-transformers/distiluse-base-multilingual-cased-v1
|
||||||
|
- sentence-transformers/distiluse-base-multilingual-cased-v2
|
||||||
|
- sentence-transformers/distiluse-base-multilingual-cased
|
||||||
|
- sentence-transformers/facebook-dpr-ctx_encoder-multiset-base
|
||||||
|
- sentence-transformers/facebook-dpr-ctx_encoder-single-nq-base
|
||||||
|
- sentence-transformers/facebook-dpr-question_encoder-multiset-base
|
||||||
|
- sentence-transformers/facebook-dpr-question_encoder-single-nq-base
|
||||||
|
- sentence-transformers/gtr-t5-large
|
||||||
|
- sentence-transformers/gtr-t5-xl
|
||||||
|
- sentence-transformers/gtr-t5-xxl
|
||||||
|
- sentence-transformers/msmarco-MiniLM-L-12-v3
|
||||||
|
- sentence-transformers/msmarco-MiniLM-L-6-v3
|
||||||
|
- sentence-transformers/msmarco-MiniLM-L12-cos-v5
|
||||||
|
- sentence-transformers/msmarco-MiniLM-L6-cos-v5
|
||||||
|
- sentence-transformers/msmarco-bert-base-dot-v5
|
||||||
|
- sentence-transformers/msmarco-bert-co-condensor
|
||||||
|
- sentence-transformers/msmarco-distilbert-base-dot-prod-v3
|
||||||
|
- sentence-transformers/msmarco-distilbert-base-tas-b
|
||||||
|
- sentence-transformers/msmarco-distilbert-base-v2
|
||||||
|
- sentence-transformers/msmarco-distilbert-base-v3
|
||||||
|
- sentence-transformers/msmarco-distilbert-base-v4
|
||||||
|
- sentence-transformers/msmarco-distilbert-cos-v5
|
||||||
|
- sentence-transformers/msmarco-distilbert-dot-v5
|
||||||
|
- sentence-transformers/msmarco-distilbert-multilingual-en-de-v2-tmp-lng-aligned
|
||||||
|
- sentence-transformers/msmarco-distilbert-multilingual-en-de-v2-tmp-trained-scratch
|
||||||
|
- sentence-transformers/msmarco-distilroberta-base-v2
|
||||||
|
- sentence-transformers/msmarco-roberta-base-ance-firstp
|
||||||
|
- sentence-transformers/msmarco-roberta-base-v2
|
||||||
|
- sentence-transformers/msmarco-roberta-base-v3
|
||||||
|
- sentence-transformers/multi-qa-MiniLM-L6-cos-v1
|
||||||
|
- sentence-transformers/nli-mpnet-base-v2
|
||||||
|
- sentence-transformers/nli-roberta-base-v2
|
||||||
|
- sentence-transformers/nq-distilbert-base-v1
|
||||||
|
- sentence-transformers/paraphrase-MiniLM-L12-v2
|
||||||
|
- sentence-transformers/paraphrase-MiniLM-L3-v2
|
||||||
|
- sentence-transformers/paraphrase-MiniLM-L6-v2
|
||||||
|
- sentence-transformers/paraphrase-TinyBERT-L6-v2
|
||||||
|
- sentence-transformers/paraphrase-albert-base-v2
|
||||||
|
- sentence-transformers/paraphrase-albert-small-v2
|
||||||
|
- sentence-transformers/paraphrase-distilroberta-base-v1
|
||||||
|
- sentence-transformers/paraphrase-distilroberta-base-v2
|
||||||
|
- sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2
|
||||||
|
- sentence-transformers/paraphrase-multilingual-mpnet-base-v2
|
||||||
|
- sentence-transformers/paraphrase-xlm-r-multilingual-v1
|
||||||
|
- sentence-transformers/quora-distilbert-base
|
||||||
|
- sentence-transformers/quora-distilbert-multilingual
|
||||||
|
- sentence-transformers/sentence-t5-base
|
||||||
|
- sentence-transformers/sentence-t5-large
|
||||||
|
- sentence-transformers/sentence-t5-xxl
|
||||||
|
- sentence-transformers/sentence-t5-xl
|
||||||
|
- sentence-transformers/stsb-distilroberta-base-v2
|
||||||
|
- sentence-transformers/stsb-mpnet-base-v2
|
||||||
|
- sentence-transformers/stsb-roberta-base-v2
|
||||||
|
- sentence-transformers/stsb-xlm-r-multilingual
|
||||||
|
- sentence-transformers/xlm-r-distilroberta-base-paraphrase-v1
|
||||||
|
- sentence-transformers/clip-ViT-L-14
|
||||||
|
- sentence-transformers/clip-ViT-B-16
|
||||||
|
- sentence-transformers/use-cmlm-multilingual
|
||||||
|
- sentence-transformers/all-MiniLM-L12-v1
|
||||||
|
```
|
||||||
|
|
||||||
|
!!! info
|
||||||
|
You can also load many other model architectures from the library. For example models from sources such as BAAI, nomic, salesforce research, etc.
|
||||||
|
See this HF hub page for all [supported models](https://huggingface.co/models?library=sentence-transformers).
|
||||||
|
|
||||||
|
!!! note "BAAI Embeddings example"
|
||||||
|
Here is an example that uses BAAI embedding model from the HuggingFace Hub [supported models](https://huggingface.co/models?library=sentence-transformers)
|
||||||
|
```python
|
||||||
|
import lancedb
|
||||||
|
from lancedb.pydantic import LanceModel, Vector
|
||||||
|
from lancedb.embeddings import get_registry
|
||||||
|
|
||||||
|
db = lancedb.connect("/tmp/db")
|
||||||
|
model = get_registry().get("sentence-transformers").create(name="BAAI/bge-small-en-v1.5", device="cpu")
|
||||||
|
|
||||||
|
class Words(LanceModel):
|
||||||
|
text: str = model.SourceField()
|
||||||
|
vector: Vector(model.ndims()) = model.VectorField()
|
||||||
|
|
||||||
|
table = db.create_table("words", schema=Words)
|
||||||
|
table.add(
|
||||||
|
[
|
||||||
|
{"text": "hello world"},
|
||||||
|
{"text": "goodbye world"}
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|
||||||
|
query = "greetings"
|
||||||
|
actual = table.search(query).limit(1).to_pydantic(Words)[0]
|
||||||
|
print(actual.text)
|
||||||
|
```
|
||||||
|
Visit sentence-transformers [HuggingFace HUB](https://huggingface.co/sentence-transformers) page for more information on the available models.
|
||||||
|
|
||||||
@@ -0,0 +1,51 @@
|
|||||||
|
# VoyageAI Embeddings
|
||||||
|
|
||||||
|
Voyage AI provides cutting-edge embedding and rerankers.
|
||||||
|
|
||||||
|
|
||||||
|
Using voyageai API requires voyageai package, which can be installed using `pip install voyageai`. Voyage AI embeddings are used to generate embeddings for text data. The embeddings can be used for various tasks like semantic search, clustering, and classification.
|
||||||
|
You also need to set the `VOYAGE_API_KEY` environment variable to use the VoyageAI API.
|
||||||
|
|
||||||
|
Supported models are:
|
||||||
|
|
||||||
|
- voyage-3
|
||||||
|
- voyage-3-lite
|
||||||
|
- voyage-finance-2
|
||||||
|
- voyage-multilingual-2
|
||||||
|
- voyage-law-2
|
||||||
|
- voyage-code-2
|
||||||
|
|
||||||
|
|
||||||
|
Supported parameters (to be passed in `create` method) are:
|
||||||
|
|
||||||
|
| Parameter | Type | Default Value | Description |
|
||||||
|
|---|---|--------|---------|
|
||||||
|
| `name` | `str` | `None` | The model ID of the model to use. Supported base models for Text Embeddings: voyage-3, voyage-3-lite, voyage-finance-2, voyage-multilingual-2, voyage-law-2, voyage-code-2 |
|
||||||
|
| `input_type` | `str` | `None` | Type of the input text. Default to None. Other options: query, document. |
|
||||||
|
| `truncation` | `bool` | `True` | Whether to truncate the input texts to fit within the context length. |
|
||||||
|
|
||||||
|
|
||||||
|
Usage Example:
|
||||||
|
|
||||||
|
```python
|
||||||
|
import lancedb
|
||||||
|
from lancedb.pydantic import LanceModel, Vector
|
||||||
|
from lancedb.embeddings import EmbeddingFunctionRegistry
|
||||||
|
|
||||||
|
voyageai = EmbeddingFunctionRegistry
|
||||||
|
.get_instance()
|
||||||
|
.get("voyageai")
|
||||||
|
.create(name="voyage-3")
|
||||||
|
|
||||||
|
class TextModel(LanceModel):
|
||||||
|
text: str = voyageai.SourceField()
|
||||||
|
vector: Vector(voyageai.ndims()) = voyageai.VectorField()
|
||||||
|
|
||||||
|
data = [ { "text": "hello world" },
|
||||||
|
{ "text": "goodbye world" }]
|
||||||
|
|
||||||
|
db = lancedb.connect("~/.lancedb")
|
||||||
|
tbl = db.create_table("test", schema=TextModel, mode="overwrite")
|
||||||
|
|
||||||
|
tbl.add(data)
|
||||||
|
```
|
||||||
@@ -47,14 +47,22 @@ Let's implement `SentenceTransformerEmbeddings` class. All you need to do is imp
|
|||||||
=== "TypeScript"
|
=== "TypeScript"
|
||||||
|
|
||||||
```ts
|
```ts
|
||||||
--8<--- "nodejs/examples/custom_embedding_function.ts:imports"
|
--8<--- "nodejs/examples/custom_embedding_function.test.ts:imports"
|
||||||
|
|
||||||
--8<--- "nodejs/examples/custom_embedding_function.ts:embedding_impl"
|
--8<--- "nodejs/examples/custom_embedding_function.test.ts:embedding_impl"
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|
||||||
This is a stripped down version of our implementation of `SentenceTransformerEmbeddings` that removes certain optimizations and default settings.
|
This is a stripped down version of our implementation of `SentenceTransformerEmbeddings` that removes certain optimizations and default settings.
|
||||||
|
|
||||||
|
!!! danger "Use sensitive keys to prevent leaking secrets"
|
||||||
|
To prevent leaking secrets, such as API keys, you should add any sensitive
|
||||||
|
parameters of an embedding function to the output of the
|
||||||
|
[sensitive_keys()][lancedb.embeddings.base.EmbeddingFunction.sensitive_keys] /
|
||||||
|
[getSensitiveKeys()](../../js/namespaces/embedding/classes/EmbeddingFunction/#getsensitivekeys)
|
||||||
|
method. This prevents users from accidentally instantiating the embedding
|
||||||
|
function with hard-coded secrets.
|
||||||
|
|
||||||
Now you can use this embedding function to create your table schema and that's it! you can then ingest data and run queries without manually vectorizing the inputs.
|
Now you can use this embedding function to create your table schema and that's it! you can then ingest data and run queries without manually vectorizing the inputs.
|
||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
@@ -78,7 +86,7 @@ Now you can use this embedding function to create your table schema and that's i
|
|||||||
=== "TypeScript"
|
=== "TypeScript"
|
||||||
|
|
||||||
```ts
|
```ts
|
||||||
--8<--- "nodejs/examples/custom_embedding_function.ts:call_custom_function"
|
--8<--- "nodejs/examples/custom_embedding_function.test.ts:call_custom_function"
|
||||||
```
|
```
|
||||||
|
|
||||||
!!! note
|
!!! note
|
||||||
|
|||||||
@@ -1,800 +1,86 @@
|
|||||||
There are various embedding functions available out of the box with LanceDB to manage your embeddings implicitly. We're actively working on adding other popular embedding APIs and models.
|
# 📚 Available Embedding Models
|
||||||
|
|
||||||
## Text embedding functions
|
There are various embedding functions available out of the box with LanceDB to manage your embeddings implicitly. We're actively working on adding other popular embedding APIs and models. 🚀
|
||||||
Contains the text embedding functions registered by default.
|
|
||||||
|
|
||||||
* Embedding functions have an inbuilt rate limit handler wrapper for source and query embedding function calls that retry with exponential backoff.
|
Before jumping on the list of available models, let's understand how to get an embedding model initialized and configured to use in our code:
|
||||||
* Each `EmbeddingFunction` implementation automatically takes `max_retries` as an argument which has the default value of 7.
|
|
||||||
|
|
||||||
### Sentence transformers
|
!!! example "Example usage"
|
||||||
Allows you to set parameters when registering a `sentence-transformers` object.
|
|
||||||
|
|
||||||
!!! info
|
|
||||||
Sentence transformer embeddings are normalized by default. It is recommended to use normalized embeddings for similarity search.
|
|
||||||
|
|
||||||
| Parameter | Type | Default Value | Description |
|
|
||||||
|---|---|---|---|
|
|
||||||
| `name` | `str` | `all-MiniLM-L6-v2` | The name of the model |
|
|
||||||
| `device` | `str` | `cpu` | The device to run the model on (can be `cpu` or `gpu`) |
|
|
||||||
| `normalize` | `bool` | `True` | Whether to normalize the input text before feeding it to the model |
|
|
||||||
| `trust_remote_code` | `bool` | `False` | Whether to trust and execute remote code from the model's Huggingface repository |
|
|
||||||
|
|
||||||
|
|
||||||
??? "Check out available sentence-transformer models here!"
|
|
||||||
```markdown
|
|
||||||
- sentence-transformers/all-MiniLM-L12-v2
|
|
||||||
- sentence-transformers/paraphrase-mpnet-base-v2
|
|
||||||
- sentence-transformers/gtr-t5-base
|
|
||||||
- sentence-transformers/LaBSE
|
|
||||||
- sentence-transformers/all-MiniLM-L6-v2
|
|
||||||
- sentence-transformers/bert-base-nli-max-tokens
|
|
||||||
- sentence-transformers/bert-base-nli-mean-tokens
|
|
||||||
- sentence-transformers/bert-base-nli-stsb-mean-tokens
|
|
||||||
- sentence-transformers/bert-base-wikipedia-sections-mean-tokens
|
|
||||||
- sentence-transformers/bert-large-nli-cls-token
|
|
||||||
- sentence-transformers/bert-large-nli-max-tokens
|
|
||||||
- sentence-transformers/bert-large-nli-mean-tokens
|
|
||||||
- sentence-transformers/bert-large-nli-stsb-mean-tokens
|
|
||||||
- sentence-transformers/distilbert-base-nli-max-tokens
|
|
||||||
- sentence-transformers/distilbert-base-nli-mean-tokens
|
|
||||||
- sentence-transformers/distilbert-base-nli-stsb-mean-tokens
|
|
||||||
- sentence-transformers/distilroberta-base-msmarco-v1
|
|
||||||
- sentence-transformers/distilroberta-base-msmarco-v2
|
|
||||||
- sentence-transformers/nli-bert-base-cls-pooling
|
|
||||||
- sentence-transformers/nli-bert-base-max-pooling
|
|
||||||
- sentence-transformers/nli-bert-base
|
|
||||||
- sentence-transformers/nli-bert-large-cls-pooling
|
|
||||||
- sentence-transformers/nli-bert-large-max-pooling
|
|
||||||
- sentence-transformers/nli-bert-large
|
|
||||||
- sentence-transformers/nli-distilbert-base-max-pooling
|
|
||||||
- sentence-transformers/nli-distilbert-base
|
|
||||||
- sentence-transformers/nli-roberta-base
|
|
||||||
- sentence-transformers/nli-roberta-large
|
|
||||||
- sentence-transformers/roberta-base-nli-mean-tokens
|
|
||||||
- sentence-transformers/roberta-base-nli-stsb-mean-tokens
|
|
||||||
- sentence-transformers/roberta-large-nli-mean-tokens
|
|
||||||
- sentence-transformers/roberta-large-nli-stsb-mean-tokens
|
|
||||||
- sentence-transformers/stsb-bert-base
|
|
||||||
- sentence-transformers/stsb-bert-large
|
|
||||||
- sentence-transformers/stsb-distilbert-base
|
|
||||||
- sentence-transformers/stsb-roberta-base
|
|
||||||
- sentence-transformers/stsb-roberta-large
|
|
||||||
- sentence-transformers/xlm-r-100langs-bert-base-nli-mean-tokens
|
|
||||||
- sentence-transformers/xlm-r-100langs-bert-base-nli-stsb-mean-tokens
|
|
||||||
- sentence-transformers/xlm-r-base-en-ko-nli-ststb
|
|
||||||
- sentence-transformers/xlm-r-bert-base-nli-mean-tokens
|
|
||||||
- sentence-transformers/xlm-r-bert-base-nli-stsb-mean-tokens
|
|
||||||
- sentence-transformers/xlm-r-large-en-ko-nli-ststb
|
|
||||||
- sentence-transformers/bert-base-nli-cls-token
|
|
||||||
- sentence-transformers/all-distilroberta-v1
|
|
||||||
- sentence-transformers/multi-qa-MiniLM-L6-dot-v1
|
|
||||||
- sentence-transformers/multi-qa-distilbert-cos-v1
|
|
||||||
- sentence-transformers/multi-qa-distilbert-dot-v1
|
|
||||||
- sentence-transformers/multi-qa-mpnet-base-cos-v1
|
|
||||||
- sentence-transformers/multi-qa-mpnet-base-dot-v1
|
|
||||||
- sentence-transformers/nli-distilroberta-base-v2
|
|
||||||
- sentence-transformers/all-MiniLM-L6-v1
|
|
||||||
- sentence-transformers/all-mpnet-base-v1
|
|
||||||
- sentence-transformers/all-mpnet-base-v2
|
|
||||||
- sentence-transformers/all-roberta-large-v1
|
|
||||||
- sentence-transformers/allenai-specter
|
|
||||||
- sentence-transformers/average_word_embeddings_glove.6B.300d
|
|
||||||
- sentence-transformers/average_word_embeddings_glove.840B.300d
|
|
||||||
- sentence-transformers/average_word_embeddings_komninos
|
|
||||||
- sentence-transformers/average_word_embeddings_levy_dependency
|
|
||||||
- sentence-transformers/clip-ViT-B-32-multilingual-v1
|
|
||||||
- sentence-transformers/clip-ViT-B-32
|
|
||||||
- sentence-transformers/distilbert-base-nli-stsb-quora-ranking
|
|
||||||
- sentence-transformers/distilbert-multilingual-nli-stsb-quora-ranking
|
|
||||||
- sentence-transformers/distilroberta-base-paraphrase-v1
|
|
||||||
- sentence-transformers/distiluse-base-multilingual-cased-v1
|
|
||||||
- sentence-transformers/distiluse-base-multilingual-cased-v2
|
|
||||||
- sentence-transformers/distiluse-base-multilingual-cased
|
|
||||||
- sentence-transformers/facebook-dpr-ctx_encoder-multiset-base
|
|
||||||
- sentence-transformers/facebook-dpr-ctx_encoder-single-nq-base
|
|
||||||
- sentence-transformers/facebook-dpr-question_encoder-multiset-base
|
|
||||||
- sentence-transformers/facebook-dpr-question_encoder-single-nq-base
|
|
||||||
- sentence-transformers/gtr-t5-large
|
|
||||||
- sentence-transformers/gtr-t5-xl
|
|
||||||
- sentence-transformers/gtr-t5-xxl
|
|
||||||
- sentence-transformers/msmarco-MiniLM-L-12-v3
|
|
||||||
- sentence-transformers/msmarco-MiniLM-L-6-v3
|
|
||||||
- sentence-transformers/msmarco-MiniLM-L12-cos-v5
|
|
||||||
- sentence-transformers/msmarco-MiniLM-L6-cos-v5
|
|
||||||
- sentence-transformers/msmarco-bert-base-dot-v5
|
|
||||||
- sentence-transformers/msmarco-bert-co-condensor
|
|
||||||
- sentence-transformers/msmarco-distilbert-base-dot-prod-v3
|
|
||||||
- sentence-transformers/msmarco-distilbert-base-tas-b
|
|
||||||
- sentence-transformers/msmarco-distilbert-base-v2
|
|
||||||
- sentence-transformers/msmarco-distilbert-base-v3
|
|
||||||
- sentence-transformers/msmarco-distilbert-base-v4
|
|
||||||
- sentence-transformers/msmarco-distilbert-cos-v5
|
|
||||||
- sentence-transformers/msmarco-distilbert-dot-v5
|
|
||||||
- sentence-transformers/msmarco-distilbert-multilingual-en-de-v2-tmp-lng-aligned
|
|
||||||
- sentence-transformers/msmarco-distilbert-multilingual-en-de-v2-tmp-trained-scratch
|
|
||||||
- sentence-transformers/msmarco-distilroberta-base-v2
|
|
||||||
- sentence-transformers/msmarco-roberta-base-ance-firstp
|
|
||||||
- sentence-transformers/msmarco-roberta-base-v2
|
|
||||||
- sentence-transformers/msmarco-roberta-base-v3
|
|
||||||
- sentence-transformers/multi-qa-MiniLM-L6-cos-v1
|
|
||||||
- sentence-transformers/nli-mpnet-base-v2
|
|
||||||
- sentence-transformers/nli-roberta-base-v2
|
|
||||||
- sentence-transformers/nq-distilbert-base-v1
|
|
||||||
- sentence-transformers/paraphrase-MiniLM-L12-v2
|
|
||||||
- sentence-transformers/paraphrase-MiniLM-L3-v2
|
|
||||||
- sentence-transformers/paraphrase-MiniLM-L6-v2
|
|
||||||
- sentence-transformers/paraphrase-TinyBERT-L6-v2
|
|
||||||
- sentence-transformers/paraphrase-albert-base-v2
|
|
||||||
- sentence-transformers/paraphrase-albert-small-v2
|
|
||||||
- sentence-transformers/paraphrase-distilroberta-base-v1
|
|
||||||
- sentence-transformers/paraphrase-distilroberta-base-v2
|
|
||||||
- sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2
|
|
||||||
- sentence-transformers/paraphrase-multilingual-mpnet-base-v2
|
|
||||||
- sentence-transformers/paraphrase-xlm-r-multilingual-v1
|
|
||||||
- sentence-transformers/quora-distilbert-base
|
|
||||||
- sentence-transformers/quora-distilbert-multilingual
|
|
||||||
- sentence-transformers/sentence-t5-base
|
|
||||||
- sentence-transformers/sentence-t5-large
|
|
||||||
- sentence-transformers/sentence-t5-xxl
|
|
||||||
- sentence-transformers/sentence-t5-xl
|
|
||||||
- sentence-transformers/stsb-distilroberta-base-v2
|
|
||||||
- sentence-transformers/stsb-mpnet-base-v2
|
|
||||||
- sentence-transformers/stsb-roberta-base-v2
|
|
||||||
- sentence-transformers/stsb-xlm-r-multilingual
|
|
||||||
- sentence-transformers/xlm-r-distilroberta-base-paraphrase-v1
|
|
||||||
- sentence-transformers/clip-ViT-L-14
|
|
||||||
- sentence-transformers/clip-ViT-B-16
|
|
||||||
- sentence-transformers/use-cmlm-multilingual
|
|
||||||
- sentence-transformers/all-MiniLM-L12-v1
|
|
||||||
```
|
|
||||||
|
|
||||||
!!! info
|
|
||||||
You can also load many other model architectures from the library. For example models from sources such as BAAI, nomic, salesforce research, etc.
|
|
||||||
See this HF hub page for all [supported models](https://huggingface.co/models?library=sentence-transformers).
|
|
||||||
|
|
||||||
!!! note "BAAI Embeddings example"
|
|
||||||
Here is an example that uses BAAI embedding model from the HuggingFace Hub [supported models](https://huggingface.co/models?library=sentence-transformers)
|
|
||||||
```python
|
```python
|
||||||
import lancedb
|
model = get_registry()
|
||||||
from lancedb.pydantic import LanceModel, Vector
|
.get("openai")
|
||||||
from lancedb.embeddings import get_registry
|
.create(name="text-embedding-ada-002")
|
||||||
|
|
||||||
db = lancedb.connect("/tmp/db")
|
|
||||||
model = get_registry().get("sentence-transformers").create(name="BAAI/bge-small-en-v1.5", device="cpu")
|
|
||||||
|
|
||||||
class Words(LanceModel):
|
|
||||||
text: str = model.SourceField()
|
|
||||||
vector: Vector(model.ndims()) = model.VectorField()
|
|
||||||
|
|
||||||
table = db.create_table("words", schema=Words)
|
|
||||||
table.add(
|
|
||||||
[
|
|
||||||
{"text": "hello world"},
|
|
||||||
{"text": "goodbye world"}
|
|
||||||
]
|
|
||||||
)
|
|
||||||
|
|
||||||
query = "greetings"
|
|
||||||
actual = table.search(query).limit(1).to_pydantic(Words)[0]
|
|
||||||
print(actual.text)
|
|
||||||
```
|
|
||||||
Visit sentence-transformers [HuggingFace HUB](https://huggingface.co/sentence-transformers) page for more information on the available models.
|
|
||||||
|
|
||||||
|
|
||||||
### Huggingface embedding models
|
|
||||||
We offer support for all huggingface models (which can be loaded via [transformers](https://huggingface.co/docs/transformers/en/index) library). The default model is `colbert-ir/colbertv2.0` which also has its own special callout - `registry.get("colbert")`
|
|
||||||
|
|
||||||
Example usage -
|
|
||||||
```python
|
|
||||||
import lancedb
|
|
||||||
import pandas as pd
|
|
||||||
|
|
||||||
from lancedb.embeddings import get_registry
|
|
||||||
from lancedb.pydantic import LanceModel, Vector
|
|
||||||
|
|
||||||
model = get_registry().get("huggingface").create(name='facebook/bart-base')
|
|
||||||
|
|
||||||
class Words(LanceModel):
|
|
||||||
text: str = model.SourceField()
|
|
||||||
vector: Vector(model.ndims()) = model.VectorField()
|
|
||||||
|
|
||||||
df = pd.DataFrame({"text": ["hi hello sayonara", "goodbye world"]})
|
|
||||||
table = db.create_table("greets", schema=Words)
|
|
||||||
table.add(df)
|
|
||||||
query = "old greeting"
|
|
||||||
actual = table.search(query).limit(1).to_pydantic(Words)[0]
|
|
||||||
print(actual.text)
|
|
||||||
```
|
|
||||||
|
|
||||||
|
|
||||||
### Ollama embeddings
|
|
||||||
Generate embeddings via the [ollama](https://github.com/ollama/ollama-python) python library. More details:
|
|
||||||
|
|
||||||
- [Ollama docs on embeddings](https://github.com/ollama/ollama/blob/main/docs/api.md#generate-embeddings)
|
|
||||||
- [Ollama blog on embeddings](https://ollama.com/blog/embedding-models)
|
|
||||||
|
|
||||||
| Parameter | Type | Default Value | Description |
|
|
||||||
|------------------------|----------------------------|--------------------------|------------------------------------------------------------------------------------------------------------------------------------------------|
|
|
||||||
| `name` | `str` | `nomic-embed-text` | The name of the model. |
|
|
||||||
| `host` | `str` | `http://localhost:11434` | The Ollama host to connect to. |
|
|
||||||
| `options` | `ollama.Options` or `dict` | `None` | Additional model parameters listed in the documentation for the Modelfile such as `temperature`. |
|
|
||||||
| `keep_alive` | `float` or `str` | `"5m"` | Controls how long the model will stay loaded into memory following the request. |
|
|
||||||
| `ollama_client_kwargs` | `dict` | `{}` | kwargs that can be past to the `ollama.Client`. |
|
|
||||||
|
|
||||||
```python
|
|
||||||
import lancedb
|
|
||||||
from lancedb.pydantic import LanceModel, Vector
|
|
||||||
from lancedb.embeddings import get_registry
|
|
||||||
|
|
||||||
db = lancedb.connect("/tmp/db")
|
|
||||||
func = get_registry().get("ollama").create(name="nomic-embed-text")
|
|
||||||
|
|
||||||
class Words(LanceModel):
|
|
||||||
text: str = func.SourceField()
|
|
||||||
vector: Vector(func.ndims()) = func.VectorField()
|
|
||||||
|
|
||||||
table = db.create_table("words", schema=Words, mode="overwrite")
|
|
||||||
table.add([
|
|
||||||
{"text": "hello world"},
|
|
||||||
{"text": "goodbye world"}
|
|
||||||
])
|
|
||||||
|
|
||||||
query = "greetings"
|
|
||||||
actual = table.search(query).limit(1).to_pydantic(Words)[0]
|
|
||||||
print(actual.text)
|
|
||||||
```
|
|
||||||
|
|
||||||
|
|
||||||
### OpenAI embeddings
|
|
||||||
LanceDB registers the OpenAI embeddings function in the registry by default, as `openai`. Below are the parameters that you can customize when creating the instances:
|
|
||||||
|
|
||||||
| Parameter | Type | Default Value | Description |
|
|
||||||
|---|---|---|---|
|
|
||||||
| `name` | `str` | `"text-embedding-ada-002"` | The name of the model. |
|
|
||||||
| `dim` | `int` | Model default | For OpenAI's newer text-embedding-3 model, we can specify a dimensionality that is smaller than the 1536 size. This feature supports it |
|
|
||||||
|
|
||||||
|
|
||||||
```python
|
|
||||||
import lancedb
|
|
||||||
from lancedb.pydantic import LanceModel, Vector
|
|
||||||
from lancedb.embeddings import get_registry
|
|
||||||
|
|
||||||
db = lancedb.connect("/tmp/db")
|
|
||||||
func = get_registry().get("openai").create(name="text-embedding-ada-002")
|
|
||||||
|
|
||||||
class Words(LanceModel):
|
|
||||||
text: str = func.SourceField()
|
|
||||||
vector: Vector(func.ndims()) = func.VectorField()
|
|
||||||
|
|
||||||
table = db.create_table("words", schema=Words, mode="overwrite")
|
|
||||||
table.add(
|
|
||||||
[
|
|
||||||
{"text": "hello world"},
|
|
||||||
{"text": "goodbye world"}
|
|
||||||
]
|
|
||||||
)
|
|
||||||
|
|
||||||
query = "greetings"
|
|
||||||
actual = table.search(query).limit(1).to_pydantic(Words)[0]
|
|
||||||
print(actual.text)
|
|
||||||
```
|
|
||||||
|
|
||||||
### Instructor Embeddings
|
|
||||||
[Instructor](https://instructor-embedding.github.io/) is an instruction-finetuned text embedding model that can generate text embeddings tailored to any task (e.g. classification, retrieval, clustering, text evaluation, etc.) and domains (e.g. science, finance, etc.) by simply providing the task instruction, without any finetuning.
|
|
||||||
|
|
||||||
If you want to calculate customized embeddings for specific sentences, you can follow the unified template to write instructions.
|
|
||||||
|
|
||||||
!!! info
|
|
||||||
Represent the `domain` `text_type` for `task_objective`:
|
|
||||||
|
|
||||||
* `domain` is optional, and it specifies the domain of the text, e.g. science, finance, medicine, etc.
|
|
||||||
* `text_type` is required, and it specifies the encoding unit, e.g. sentence, document, paragraph, etc.
|
|
||||||
* `task_objective` is optional, and it specifies the objective of embedding, e.g. retrieve a document, classify the sentence, etc.
|
|
||||||
|
|
||||||
More information about the model can be found at the [source URL](https://github.com/xlang-ai/instructor-embedding).
|
|
||||||
|
|
||||||
| Argument | Type | Default | Description |
|
|
||||||
|---|---|---|---|
|
|
||||||
| `name` | `str` | "hkunlp/instructor-base" | The name of the model to use |
|
|
||||||
| `batch_size` | `int` | `32` | The batch size to use when generating embeddings |
|
|
||||||
| `device` | `str` | `"cpu"` | The device to use when generating embeddings |
|
|
||||||
| `show_progress_bar` | `bool` | `True` | Whether to show a progress bar when generating embeddings |
|
|
||||||
| `normalize_embeddings` | `bool` | `True` | Whether to normalize the embeddings |
|
|
||||||
| `quantize` | `bool` | `False` | Whether to quantize the model |
|
|
||||||
| `source_instruction` | `str` | `"represent the docuement for retreival"` | The instruction for the source column |
|
|
||||||
| `query_instruction` | `str` | `"represent the document for retreiving the most similar documents"` | The instruction for the query |
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
```python
|
|
||||||
import lancedb
|
|
||||||
from lancedb.pydantic import LanceModel, Vector
|
|
||||||
from lancedb.embeddings import get_registry, InstuctorEmbeddingFunction
|
|
||||||
|
|
||||||
instructor = get_registry().get("instructor").create(
|
|
||||||
source_instruction="represent the docuement for retreival",
|
|
||||||
query_instruction="represent the document for retreiving the most similar documents"
|
|
||||||
)
|
|
||||||
|
|
||||||
class Schema(LanceModel):
|
|
||||||
vector: Vector(instructor.ndims()) = instructor.VectorField()
|
|
||||||
text: str = instructor.SourceField()
|
|
||||||
|
|
||||||
db = lancedb.connect("~/.lancedb")
|
|
||||||
tbl = db.create_table("test", schema=Schema, mode="overwrite")
|
|
||||||
|
|
||||||
texts = [{"text": "Capitalism has been dominant in the Western world since the end of feudalism, but most feel[who?] that..."},
|
|
||||||
{"text": "The disparate impact theory is especially controversial under the Fair Housing Act because the Act..."},
|
|
||||||
{"text": "Disparate impact in United States labor law refers to practices in employment, housing, and other areas that.."}]
|
|
||||||
|
|
||||||
tbl.add(texts)
|
|
||||||
```
|
|
||||||
|
|
||||||
### Gemini Embeddings
|
|
||||||
With Google's Gemini, you can represent text (words, sentences, and blocks of text) in a vectorized form, making it easier to compare and contrast embeddings. For example, two texts that share a similar subject matter or sentiment should have similar embeddings, which can be identified through mathematical comparison techniques such as cosine similarity. For more on how and why you should use embeddings, refer to the Embeddings guide.
|
|
||||||
The Gemini Embedding Model API supports various task types:
|
|
||||||
|
|
||||||
| Task Type | Description |
|
|
||||||
|-------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
|
||||||
| "`retrieval_query`" | Specifies the given text is a query in a search/retrieval setting. |
|
|
||||||
| "`retrieval_document`" | Specifies the given text is a document in a search/retrieval setting. Using this task type requires a title but is automatically proided by Embeddings API |
|
|
||||||
| "`semantic_similarity`" | Specifies the given text will be used for Semantic Textual Similarity (STS). |
|
|
||||||
| "`classification`" | Specifies that the embeddings will be used for classification. |
|
|
||||||
| "`clusering`" | Specifies that the embeddings will be used for clustering. |
|
|
||||||
|
|
||||||
|
|
||||||
Usage Example:
|
|
||||||
|
|
||||||
```python
|
|
||||||
import lancedb
|
|
||||||
import pandas as pd
|
|
||||||
from lancedb.pydantic import LanceModel, Vector
|
|
||||||
from lancedb.embeddings import get_registry
|
|
||||||
|
|
||||||
|
|
||||||
model = get_registry().get("gemini-text").create()
|
|
||||||
|
|
||||||
class TextModel(LanceModel):
|
|
||||||
text: str = model.SourceField()
|
|
||||||
vector: Vector(model.ndims()) = model.VectorField()
|
|
||||||
|
|
||||||
df = pd.DataFrame({"text": ["hello world", "goodbye world"]})
|
|
||||||
db = lancedb.connect("~/.lancedb")
|
|
||||||
tbl = db.create_table("test", schema=TextModel, mode="overwrite")
|
|
||||||
|
|
||||||
tbl.add(df)
|
|
||||||
rs = tbl.search("hello").limit(1).to_pandas()
|
|
||||||
```
|
|
||||||
|
|
||||||
### Cohere Embeddings
|
|
||||||
Using cohere API requires cohere package, which can be installed using `pip install cohere`. Cohere embeddings are used to generate embeddings for text data. The embeddings can be used for various tasks like semantic search, clustering, and classification.
|
|
||||||
You also need to set the `COHERE_API_KEY` environment variable to use the Cohere API.
|
|
||||||
|
|
||||||
Supported models are:
|
|
||||||
```
|
|
||||||
* embed-english-v3.0
|
|
||||||
* embed-multilingual-v3.0
|
|
||||||
* embed-english-light-v3.0
|
|
||||||
* embed-multilingual-light-v3.0
|
|
||||||
* embed-english-v2.0
|
|
||||||
* embed-english-light-v2.0
|
|
||||||
* embed-multilingual-v2.0
|
|
||||||
```
|
|
||||||
|
|
||||||
Supported parameters (to be passed in `create` method) are:
|
|
||||||
|
|
||||||
| Parameter | Type | Default Value | Description |
|
|
||||||
|---|---|---|---|
|
|
||||||
| `name` | `str` | `"embed-english-v2.0"` | The model ID of the cohere model to use. Supported base models for Text Embeddings: embed-english-v3.0, embed-multilingual-v3.0, embed-english-light-v3.0, embed-multilingual-light-v3.0, embed-english-v2.0, embed-english-light-v2.0, embed-multilingual-v2.0 |
|
|
||||||
| `source_input_type` | `str` | `"search_document"` | The type of input data to be used for the source column. |
|
|
||||||
| `query_input_type` | `str` | `"search_query"` | The type of input data to be used for the query. |
|
|
||||||
|
|
||||||
Cohere supports following input types:
|
|
||||||
|
|
||||||
| Input Type | Description |
|
|
||||||
|-------------------------|---------------------------------------|
|
|
||||||
| "`search_document`" | Used for embeddings stored in a vector|
|
|
||||||
| | database for search use-cases. |
|
|
||||||
| "`search_query`" | Used for embeddings of search queries |
|
|
||||||
| | run against a vector DB |
|
|
||||||
| "`semantic_similarity`" | Specifies the given text will be used |
|
|
||||||
| | for Semantic Textual Similarity (STS) |
|
|
||||||
| "`classification`" | Used for embeddings passed through a |
|
|
||||||
| | text classifier. |
|
|
||||||
| "`clustering`" | Used for the embeddings run through a |
|
|
||||||
| | clustering algorithm |
|
|
||||||
|
|
||||||
Usage Example:
|
|
||||||
|
|
||||||
```python
|
|
||||||
import lancedb
|
|
||||||
from lancedb.pydantic import LanceModel, Vector
|
|
||||||
from lancedb.embeddings import EmbeddingFunctionRegistry
|
|
||||||
|
|
||||||
cohere = EmbeddingFunctionRegistry
|
|
||||||
.get_instance()
|
|
||||||
.get("cohere")
|
|
||||||
.create(name="embed-multilingual-v2.0")
|
|
||||||
|
|
||||||
class TextModel(LanceModel):
|
|
||||||
text: str = cohere.SourceField()
|
|
||||||
vector: Vector(cohere.ndims()) = cohere.VectorField()
|
|
||||||
|
|
||||||
data = [ { "text": "hello world" },
|
|
||||||
{ "text": "goodbye world" }]
|
|
||||||
|
|
||||||
db = lancedb.connect("~/.lancedb")
|
|
||||||
tbl = db.create_table("test", schema=TextModel, mode="overwrite")
|
|
||||||
|
|
||||||
tbl.add(data)
|
|
||||||
```
|
```
|
||||||
|
|
||||||
### Jina Embeddings
|
Now let's understand the above syntax:
|
||||||
Jina embeddings are used to generate embeddings for text and image data.
|
|
||||||
You also need to set the `JINA_API_KEY` environment variable to use the Jina API.
|
|
||||||
|
|
||||||
You can find a list of supported models under [https://jina.ai/embeddings/](https://jina.ai/embeddings/)
|
|
||||||
|
|
||||||
Supported parameters (to be passed in `create` method) are:
|
|
||||||
|
|
||||||
| Parameter | Type | Default Value | Description |
|
|
||||||
|---|---|---|---|
|
|
||||||
| `name` | `str` | `"jina-clip-v1"` | The model ID of the jina model to use |
|
|
||||||
|
|
||||||
Usage Example:
|
|
||||||
|
|
||||||
```python
|
```python
|
||||||
import os
|
model = get_registry().get("model_id").create(...params)
|
||||||
import lancedb
|
|
||||||
from lancedb.pydantic import LanceModel, Vector
|
|
||||||
from lancedb.embeddings import EmbeddingFunctionRegistry
|
|
||||||
|
|
||||||
os.environ['JINA_API_KEY'] = 'jina_*'
|
|
||||||
|
|
||||||
jina_embed = EmbeddingFunctionRegistry.get_instance().get("jina").create(name="jina-embeddings-v2-base-en")
|
|
||||||
|
|
||||||
|
|
||||||
class TextModel(LanceModel):
|
|
||||||
text: str = jina_embed.SourceField()
|
|
||||||
vector: Vector(jina_embed.ndims()) = jina_embed.VectorField()
|
|
||||||
|
|
||||||
|
|
||||||
data = [{"text": "hello world"},
|
|
||||||
{"text": "goodbye world"}]
|
|
||||||
|
|
||||||
db = lancedb.connect("~/.lancedb-2")
|
|
||||||
tbl = db.create_table("test", schema=TextModel, mode="overwrite")
|
|
||||||
|
|
||||||
tbl.add(data)
|
|
||||||
```
|
```
|
||||||
|
**This👆 line effectively creates a configured instance of an `embedding function` with `model` of choice that is ready for use.**
|
||||||
|
|
||||||
### AWS Bedrock Text Embedding Functions
|
- `get_registry()` : This function call returns an instance of a `EmbeddingFunctionRegistry` object. This registry manages the registration and retrieval of embedding functions.
|
||||||
AWS Bedrock supports multiple base models for generating text embeddings. You need to setup the AWS credentials to use this embedding function.
|
|
||||||
You can do so by using `awscli` and also add your session_token:
|
|
||||||
```shell
|
|
||||||
aws configure
|
|
||||||
aws configure set aws_session_token "<your_session_token>"
|
|
||||||
```
|
|
||||||
to ensure that the credentials are set up correctly, you can run the following command:
|
|
||||||
```shell
|
|
||||||
aws sts get-caller-identity
|
|
||||||
```
|
|
||||||
|
|
||||||
Supported Embedding modelIDs are:
|
- `.get("model_id")` : This method call on the registry object and retrieves the **embedding models functions** associated with the `"model_id"` (1) .
|
||||||
* `amazon.titan-embed-text-v1`
|
{ .annotate }
|
||||||
* `cohere.embed-english-v3`
|
|
||||||
* `cohere.embed-multilingual-v3`
|
|
||||||
|
|
||||||
Supported parameters (to be passed in `create` method) are:
|
1. Hover over the names in table below to find out the `model_id` of different embedding functions.
|
||||||
|
|
||||||
| Parameter | Type | Default Value | Description |
|
- `.create(...params)` : This method call is on the object returned by the `get` method. It instantiates an embedding model function using the **specified parameters**.
|
||||||
|---|---|---|---|
|
|
||||||
| **name** | str | "amazon.titan-embed-text-v1" | The model ID of the bedrock model to use. Supported base models for Text Embeddings: amazon.titan-embed-text-v1, cohere.embed-english-v3, cohere.embed-multilingual-v3 |
|
|
||||||
| **region** | str | "us-east-1" | Optional name of the AWS Region in which the service should be called (e.g., "us-east-1"). |
|
|
||||||
| **profile_name** | str | None | Optional name of the AWS profile to use for calling the Bedrock service. If not specified, the default profile will be used. |
|
|
||||||
| **assumed_role** | str | None | Optional ARN of an AWS IAM role to assume for calling the Bedrock service. If not specified, the current active credentials will be used. |
|
|
||||||
| **role_session_name** | str | "lancedb-embeddings" | Optional name of the AWS IAM role session to use for calling the Bedrock service. If not specified, a "lancedb-embeddings" name will be used. |
|
|
||||||
| **runtime** | bool | True | Optional choice of getting different client to perform operations with the Amazon Bedrock service. |
|
|
||||||
| **max_retries** | int | 7 | Optional number of retries to perform when a request fails. |
|
|
||||||
|
|
||||||
Usage Example:
|
??? question "What parameters does the `.create(...params)` method accepts?"
|
||||||
|
**Checkout the documentation of specific embedding models (links in the table below👇) to know what parameters it takes**.
|
||||||
|
|
||||||
```python
|
!!! tip "Moving on"
|
||||||
import lancedb
|
Now that we know how to get the **desired embedding model** and use it in our code, let's explore the comprehensive **list** of embedding models **supported by LanceDB**, in the tables below.
|
||||||
from lancedb.pydantic import LanceModel, Vector
|
|
||||||
from lancedb.embeddings import get_registry
|
|
||||||
|
|
||||||
model = get_registry().get("bedrock-text").create()
|
## Text Embedding Functions 📝
|
||||||
|
These functions are registered by default to handle text embeddings.
|
||||||
|
|
||||||
class TextModel(LanceModel):
|
- 🔄 **Embedding functions** have an inbuilt rate limit handler wrapper for source and query embedding function calls that retry with **exponential backoff**.
|
||||||
text: str = model.SourceField()
|
|
||||||
vector: Vector(model.ndims()) = model.VectorField()
|
|
||||||
|
|
||||||
df = pd.DataFrame({"text": ["hello world", "goodbye world"]})
|
- 🌕 Each `EmbeddingFunction` implementation automatically takes `max_retries` as an argument which has the default value of 7.
|
||||||
db = lancedb.connect("tmp_path")
|
|
||||||
tbl = db.create_table("test", schema=TextModel, mode="overwrite")
|
|
||||||
|
|
||||||
tbl.add(df)
|
🌟 **Available Text Embeddings**
|
||||||
rs = tbl.search("hello").limit(1).to_pandas()
|
|
||||||
```
|
|
||||||
|
|
||||||
# IBM watsonx.ai Embeddings
|
| **Embedding** :material-information-outline:{ title="Hover over the name to find out the model_id" } | **Description** | **Documentation** |
|
||||||
|
|-----------|-------------|---------------|
|
||||||
|
| [**Sentence Transformers**](available_embedding_models/text_embedding_functions/sentence_transformers.md "sentence-transformers") | 🧠 **SentenceTransformers** is a Python framework for state-of-the-art sentence, text, and image embeddings. | [<img src="https://raw.githubusercontent.com/lancedb/assets/main/docs/assets/logos/sbert_2.png" alt="Sentence Transformers Icon" width="90" height="35">](available_embedding_models/text_embedding_functions/sentence_transformers.md)|
|
||||||
|
| [**Huggingface Models**](available_embedding_models/text_embedding_functions/huggingface_embedding.md "huggingface") |🤗 We offer support for all **Huggingface** models. The default model is `colbert-ir/colbertv2.0`. | [<img src="https://raw.githubusercontent.com/lancedb/assets/main/docs/assets/logos/hugging_face.png" alt="Huggingface Icon" width="130" height="35">](available_embedding_models/text_embedding_functions/huggingface_embedding.md) |
|
||||||
|
| [**Ollama Embeddings**](available_embedding_models/text_embedding_functions/ollama_embedding.md "ollama") | 🔍 Generate embeddings via the **Ollama** python library. Ollama supports embedding models, making it possible to build RAG apps. | [<img src="https://raw.githubusercontent.com/lancedb/assets/main/docs/assets/logos/Ollama.png" alt="Ollama Icon" width="110" height="35">](available_embedding_models/text_embedding_functions/ollama_embedding.md)|
|
||||||
|
| [**OpenAI Embeddings**](available_embedding_models/text_embedding_functions/openai_embedding.md "openai")| 🔑 **OpenAI’s** text embeddings measure the relatedness of text strings. **LanceDB** supports state-of-the-art embeddings from OpenAI. | [<img src="https://raw.githubusercontent.com/lancedb/assets/main/docs/assets/logos/openai.png" alt="OpenAI Icon" width="100" height="35">](available_embedding_models/text_embedding_functions/openai_embedding.md)|
|
||||||
|
| [**Instructor Embeddings**](available_embedding_models/text_embedding_functions/instructor_embedding.md "instructor") | 📚 **Instructor**: An instruction-finetuned text embedding model that can generate text embeddings tailored to any task and domains by simply providing the task instruction, without any finetuning. | [<img src="https://raw.githubusercontent.com/lancedb/assets/main/docs/assets/logos/instructor_embedding.png" alt="Instructor Embedding Icon" width="140" height="35">](available_embedding_models/text_embedding_functions/instructor_embedding.md) |
|
||||||
|
| [**Gemini Embeddings**](available_embedding_models/text_embedding_functions/gemini_embedding.md "gemini-text") | 🌌 Google’s Gemini API generates state-of-the-art embeddings for words, phrases, and sentences. | [<img src="https://raw.githubusercontent.com/lancedb/assets/main/docs/assets/logos/gemini.png" alt="Gemini Icon" width="95" height="35">](available_embedding_models/text_embedding_functions/gemini_embedding.md) |
|
||||||
|
| [**Cohere Embeddings**](available_embedding_models/text_embedding_functions/cohere_embedding.md "cohere") | 💬 This will help you get started with **Cohere** embedding models using LanceDB. Using cohere API requires cohere package. Install it via `pip`. | [<img src="https://raw.githubusercontent.com/lancedb/assets/main/docs/assets/logos/cohere.png" alt="Cohere Icon" width="140" height="35">](available_embedding_models/text_embedding_functions/cohere_embedding.md) |
|
||||||
|
| [**Jina Embeddings**](available_embedding_models/text_embedding_functions/jina_embedding.md "jina") | 🔗 World-class embedding models to improve your search and RAG systems. You will need **jina api key**. | [<img src="https://raw.githubusercontent.com/lancedb/assets/main/docs/assets/logos/jina.png" alt="Jina Icon" width="90" height="35">](available_embedding_models/text_embedding_functions/jina_embedding.md) |
|
||||||
|
| [ **AWS Bedrock Functions**](available_embedding_models/text_embedding_functions/aws_bedrock_embedding.md "bedrock-text") | ☁️ AWS Bedrock supports multiple base models for generating text embeddings. You need to setup the AWS credentials to use this embedding function. | [<img src="https://raw.githubusercontent.com/lancedb/assets/main/docs/assets/logos/aws_bedrock.png" alt="AWS Bedrock Icon" width="120" height="35">](available_embedding_models/text_embedding_functions/aws_bedrock_embedding.md) |
|
||||||
|
| [**IBM Watsonx.ai**](available_embedding_models/text_embedding_functions/ibm_watsonx_ai_embedding.md "watsonx") | 💡 Generate text embeddings using IBM's watsonx.ai platform. **Note**: watsonx.ai library is an optional dependency. | [<img src="https://raw.githubusercontent.com/lancedb/assets/main/docs/assets/logos/watsonx.png" alt="Watsonx Icon" width="140" height="35">](available_embedding_models/text_embedding_functions/ibm_watsonx_ai_embedding.md) |
|
||||||
|
| [**VoyageAI Embeddings**](available_embedding_models/text_embedding_functions/voyageai_embedding.md "voyageai") | 🌕 Voyage AI provides cutting-edge embedding and rerankers. This will help you get started with **VoyageAI** embedding models using LanceDB. Using voyageai API requires voyageai package. Install it via `pip`. | [<img src="https://www.voyageai.com/logo.svg" alt="VoyageAI Icon" width="140" height="35">](available_embedding_models/text_embedding_functions/voyageai_embedding.md) |
|
||||||
|
|
||||||
Generate text embeddings using IBM's watsonx.ai platform.
|
|
||||||
|
|
||||||
## Supported Models
|
|
||||||
|
|
||||||
You can find a list of supported models at [IBM watsonx.ai Documentation](https://dataplatform.cloud.ibm.com/docs/content/wsj/analyze-data/fm-models-embed.html?context=wx). The currently supported model names are:
|
[st-key]: "sentence-transformers"
|
||||||
|
[hf-key]: "huggingface"
|
||||||
|
[ollama-key]: "ollama"
|
||||||
|
[openai-key]: "openai"
|
||||||
|
[instructor-key]: "instructor"
|
||||||
|
[gemini-key]: "gemini-text"
|
||||||
|
[cohere-key]: "cohere"
|
||||||
|
[jina-key]: "jina"
|
||||||
|
[aws-key]: "bedrock-text"
|
||||||
|
[watsonx-key]: "watsonx"
|
||||||
|
[voyageai-key]: "voyageai"
|
||||||
|
|
||||||
- `ibm/slate-125m-english-rtrvr`
|
|
||||||
- `ibm/slate-30m-english-rtrvr`
|
|
||||||
- `sentence-transformers/all-minilm-l12-v2`
|
|
||||||
- `intfloat/multilingual-e5-large`
|
|
||||||
|
|
||||||
## Parameters
|
## Multi-modal Embedding Functions🖼️
|
||||||
|
|
||||||
The following parameters can be passed to the `create` method:
|
Multi-modal embedding functions allow you to query your table using both images and text. 💬🖼️
|
||||||
|
|
||||||
| Parameter | Type | Default Value | Description |
|
🌐 **Available Multi-modal Embeddings**
|
||||||
|------------|----------|----------------------------------|-----------------------------------------------------------|
|
|
||||||
| name | str | "ibm/slate-125m-english-rtrvr" | The model ID of the watsonx.ai model to use |
|
|
||||||
| api_key | str | None | Optional IBM Cloud API key (or set `WATSONX_API_KEY`) |
|
|
||||||
| project_id | str | None | Optional watsonx project ID (or set `WATSONX_PROJECT_ID`) |
|
|
||||||
| url | str | None | Optional custom URL for the watsonx.ai instance |
|
|
||||||
| params | dict | None | Optional additional parameters for the embedding model |
|
|
||||||
|
|
||||||
## Usage Example
|
| Embedding :material-information-outline:{ title="Hover over the name to find out the model_id" } | Description | Documentation |
|
||||||
|
|-----------|-------------|---------------|
|
||||||
|
| [**OpenClip Embeddings**](available_embedding_models/multimodal_embedding_functions/openclip_embedding.md "open-clip") | 🎨 We support CLIP model embeddings using the open source alternative, **open-clip** which supports various customizations. | [<img src="https://raw.githubusercontent.com/lancedb/assets/main/docs/assets/logos/openclip_github.png" alt="openclip Icon" width="150" height="35">](available_embedding_models/multimodal_embedding_functions/openclip_embedding.md) |
|
||||||
|
| [**Imagebind Embeddings**](available_embedding_models/multimodal_embedding_functions/imagebind_embedding.md "imageind") | 🌌 We have support for **imagebind model embeddings**. You can download our version of the packaged model via - `pip install imagebind-packaged==0.1.2`. | [<img src="https://raw.githubusercontent.com/lancedb/assets/main/docs/assets/logos/imagebind_meta.png" alt="imagebind Icon" width="150" height="35">](available_embedding_models/multimodal_embedding_functions/imagebind_embedding.md)|
|
||||||
|
| [**Jina Multi-modal Embeddings**](available_embedding_models/multimodal_embedding_functions/jina_multimodal_embedding.md "jina") | 🔗 **Jina embeddings** can also be used to embed both **text** and **image** data, only some of the models support image data and you can check the detailed documentation. 👉 | [<img src="https://raw.githubusercontent.com/lancedb/assets/main/docs/assets/logos/jina.png" alt="jina Icon" width="90" height="35">](available_embedding_models/multimodal_embedding_functions/jina_multimodal_embedding.md) |
|
||||||
|
|
||||||
First, the watsonx.ai library is an optional dependency, so must be installed seperately:
|
!!! note
|
||||||
|
If you'd like to request support for additional **embedding functions**, please feel free to open an issue on our LanceDB [GitHub issue page](https://github.com/lancedb/lancedb/issues).
|
||||||
```
|
|
||||||
pip install ibm-watsonx-ai
|
|
||||||
```
|
|
||||||
|
|
||||||
Optionally set environment variables (if not passing credentials to `create` directly):
|
|
||||||
|
|
||||||
```sh
|
|
||||||
export WATSONX_API_KEY="YOUR_WATSONX_API_KEY"
|
|
||||||
export WATSONX_PROJECT_ID="YOUR_WATSONX_PROJECT_ID"
|
|
||||||
```
|
|
||||||
|
|
||||||
```python
|
|
||||||
import os
|
|
||||||
import lancedb
|
|
||||||
from lancedb.pydantic import LanceModel, Vector
|
|
||||||
from lancedb.embeddings import EmbeddingFunctionRegistry
|
|
||||||
|
|
||||||
watsonx_embed = EmbeddingFunctionRegistry
|
|
||||||
.get_instance()
|
|
||||||
.get("watsonx")
|
|
||||||
.create(
|
|
||||||
name="ibm/slate-125m-english-rtrvr",
|
|
||||||
# Uncomment and set these if not using environment variables
|
|
||||||
# api_key="your_api_key_here",
|
|
||||||
# project_id="your_project_id_here",
|
|
||||||
# url="your_watsonx_url_here",
|
|
||||||
# params={...},
|
|
||||||
)
|
|
||||||
|
|
||||||
class TextModel(LanceModel):
|
|
||||||
text: str = watsonx_embed.SourceField()
|
|
||||||
vector: Vector(watsonx_embed.ndims()) = watsonx_embed.VectorField()
|
|
||||||
|
|
||||||
data = [
|
|
||||||
{"text": "hello world"},
|
|
||||||
{"text": "goodbye world"},
|
|
||||||
]
|
|
||||||
|
|
||||||
db = lancedb.connect("~/.lancedb")
|
|
||||||
tbl = db.create_table("watsonx_test", schema=TextModel, mode="overwrite")
|
|
||||||
|
|
||||||
tbl.add(data)
|
|
||||||
|
|
||||||
rs = tbl.search("hello").limit(1).to_pandas()
|
|
||||||
print(rs)
|
|
||||||
```
|
|
||||||
|
|
||||||
## Multi-modal embedding functions
|
|
||||||
Multi-modal embedding functions allow you to query your table using both images and text.
|
|
||||||
|
|
||||||
### OpenClip embeddings
|
|
||||||
We support CLIP model embeddings using the open source alternative, [open-clip](https://github.com/mlfoundations/open_clip) which supports various customizations. It is registered as `open-clip` and supports the following customizations:
|
|
||||||
|
|
||||||
| Parameter | Type | Default Value | Description |
|
|
||||||
|---|---|---|---|
|
|
||||||
| `name` | `str` | `"ViT-B-32"` | The name of the model. |
|
|
||||||
| `pretrained` | `str` | `"laion2b_s34b_b79k"` | The name of the pretrained model to load. |
|
|
||||||
| `device` | `str` | `"cpu"` | The device to run the model on. Can be `"cpu"` or `"gpu"`. |
|
|
||||||
| `batch_size` | `int` | `64` | The number of images to process in a batch. |
|
|
||||||
| `normalize` | `bool` | `True` | Whether to normalize the input images before feeding them to the model. |
|
|
||||||
|
|
||||||
This embedding function supports ingesting images as both bytes and urls. You can query them using both test and other images.
|
|
||||||
|
|
||||||
!!! info
|
|
||||||
LanceDB supports ingesting images directly from accessible links.
|
|
||||||
|
|
||||||
```python
|
|
||||||
import lancedb
|
|
||||||
from lancedb.pydantic import LanceModel, Vector
|
|
||||||
from lancedb.embeddings import get_registry
|
|
||||||
|
|
||||||
db = lancedb.connect(tmp_path)
|
|
||||||
func = get_registry.get("open-clip").create()
|
|
||||||
|
|
||||||
class Images(LanceModel):
|
|
||||||
label: str
|
|
||||||
image_uri: str = func.SourceField() # image uri as the source
|
|
||||||
image_bytes: bytes = func.SourceField() # image bytes as the source
|
|
||||||
vector: Vector(func.ndims()) = func.VectorField() # vector column
|
|
||||||
vec_from_bytes: Vector(func.ndims()) = func.VectorField() # Another vector column
|
|
||||||
|
|
||||||
table = db.create_table("images", schema=Images)
|
|
||||||
labels = ["cat", "cat", "dog", "dog", "horse", "horse"]
|
|
||||||
uris = [
|
|
||||||
"http://farm1.staticflickr.com/53/167798175_7c7845bbbd_z.jpg",
|
|
||||||
"http://farm1.staticflickr.com/134/332220238_da527d8140_z.jpg",
|
|
||||||
"http://farm9.staticflickr.com/8387/8602747737_2e5c2a45d4_z.jpg",
|
|
||||||
"http://farm5.staticflickr.com/4092/5017326486_1f46057f5f_z.jpg",
|
|
||||||
"http://farm9.staticflickr.com/8216/8434969557_d37882c42d_z.jpg",
|
|
||||||
"http://farm6.staticflickr.com/5142/5835678453_4f3a4edb45_z.jpg",
|
|
||||||
]
|
|
||||||
# get each uri as bytes
|
|
||||||
image_bytes = [requests.get(uri).content for uri in uris]
|
|
||||||
table.add(
|
|
||||||
pd.DataFrame({"label": labels, "image_uri": uris, "image_bytes": image_bytes})
|
|
||||||
)
|
|
||||||
```
|
|
||||||
Now we can search using text from both the default vector column and the custom vector column
|
|
||||||
```python
|
|
||||||
|
|
||||||
# text search
|
|
||||||
actual = table.search("man's best friend").limit(1).to_pydantic(Images)[0]
|
|
||||||
print(actual.label) # prints "dog"
|
|
||||||
|
|
||||||
frombytes = (
|
|
||||||
table.search("man's best friend", vector_column_name="vec_from_bytes")
|
|
||||||
.limit(1)
|
|
||||||
.to_pydantic(Images)[0]
|
|
||||||
)
|
|
||||||
print(frombytes.label)
|
|
||||||
|
|
||||||
```
|
|
||||||
|
|
||||||
Because we're using a multi-modal embedding function, we can also search using images
|
|
||||||
|
|
||||||
```python
|
|
||||||
# image search
|
|
||||||
query_image_uri = "http://farm1.staticflickr.com/200/467715466_ed4a31801f_z.jpg"
|
|
||||||
image_bytes = requests.get(query_image_uri).content
|
|
||||||
query_image = Image.open(io.BytesIO(image_bytes))
|
|
||||||
actual = table.search(query_image).limit(1).to_pydantic(Images)[0]
|
|
||||||
print(actual.label == "dog")
|
|
||||||
|
|
||||||
# image search using a custom vector column
|
|
||||||
other = (
|
|
||||||
table.search(query_image, vector_column_name="vec_from_bytes")
|
|
||||||
.limit(1)
|
|
||||||
.to_pydantic(Images)[0]
|
|
||||||
)
|
|
||||||
print(actual.label)
|
|
||||||
|
|
||||||
```
|
|
||||||
|
|
||||||
### Imagebind embeddings
|
|
||||||
We have support for [imagebind](https://github.com/facebookresearch/ImageBind) model embeddings. You can download our version of the packaged model via - `pip install imagebind-packaged==0.1.2`.
|
|
||||||
|
|
||||||
This function is registered as `imagebind` and supports Audio, Video and Text modalities(extending to Thermal,Depth,IMU data):
|
|
||||||
|
|
||||||
| Parameter | Type | Default Value | Description |
|
|
||||||
|---|---|---|---|
|
|
||||||
| `name` | `str` | `"imagebind_huge"` | Name of the model. |
|
|
||||||
| `device` | `str` | `"cpu"` | The device to run the model on. Can be `"cpu"` or `"gpu"`. |
|
|
||||||
| `normalize` | `bool` | `False` | set to `True` to normalize your inputs before model ingestion. |
|
|
||||||
|
|
||||||
Below is an example demonstrating how the API works:
|
|
||||||
|
|
||||||
```python
|
|
||||||
import lancedb
|
|
||||||
from lancedb.pydantic import LanceModel, Vector
|
|
||||||
from lancedb.embeddings import get_registry
|
|
||||||
|
|
||||||
db = lancedb.connect(tmp_path)
|
|
||||||
func = get_registry.get("imagebind").create()
|
|
||||||
|
|
||||||
class ImageBindModel(LanceModel):
|
|
||||||
text: str
|
|
||||||
image_uri: str = func.SourceField()
|
|
||||||
audio_path: str
|
|
||||||
vector: Vector(func.ndims()) = func.VectorField()
|
|
||||||
|
|
||||||
# add locally accessible image paths
|
|
||||||
text_list=["A dog.", "A car", "A bird"]
|
|
||||||
image_paths=[".assets/dog_image.jpg", ".assets/car_image.jpg", ".assets/bird_image.jpg"]
|
|
||||||
audio_paths=[".assets/dog_audio.wav", ".assets/car_audio.wav", ".assets/bird_audio.wav"]
|
|
||||||
|
|
||||||
# Load data
|
|
||||||
inputs = [
|
|
||||||
{"text": a, "audio_path": b, "image_uri": c}
|
|
||||||
for a, b, c in zip(text_list, audio_paths, image_paths)
|
|
||||||
]
|
|
||||||
|
|
||||||
#create table and add data
|
|
||||||
table = db.create_table("img_bind", schema=ImageBindModel)
|
|
||||||
table.add(inputs)
|
|
||||||
```
|
|
||||||
|
|
||||||
Now, we can search using any modality:
|
|
||||||
|
|
||||||
#### image search
|
|
||||||
```python
|
|
||||||
query_image = "./assets/dog_image2.jpg" #download an image and enter that path here
|
|
||||||
actual = table.search(query_image).limit(1).to_pydantic(ImageBindModel)[0]
|
|
||||||
print(actual.text == "dog")
|
|
||||||
```
|
|
||||||
#### audio search
|
|
||||||
|
|
||||||
```python
|
|
||||||
query_audio = "./assets/car_audio2.wav" #download an audio clip and enter path here
|
|
||||||
actual = table.search(query_audio).limit(1).to_pydantic(ImageBindModel)[0]
|
|
||||||
print(actual.text == "car")
|
|
||||||
```
|
|
||||||
#### Text search
|
|
||||||
You can add any input query and fetch the result as follows:
|
|
||||||
```python
|
|
||||||
query = "an animal which flies and tweets"
|
|
||||||
actual = table.search(query).limit(1).to_pydantic(ImageBindModel)[0]
|
|
||||||
print(actual.text == "bird")
|
|
||||||
```
|
|
||||||
|
|
||||||
If you have any questions about the embeddings API, supported models, or see a relevant model missing, please raise an issue [on GitHub](https://github.com/lancedb/lancedb/issues).
|
|
||||||
|
|
||||||
### Jina Embeddings
|
|
||||||
Jina embeddings can also be used to embed both text and image data, only some of the models support image data and you can check the list
|
|
||||||
under [https://jina.ai/embeddings/](https://jina.ai/embeddings/)
|
|
||||||
|
|
||||||
Supported parameters (to be passed in `create` method) are:
|
|
||||||
|
|
||||||
| Parameter | Type | Default Value | Description |
|
|
||||||
|---|---|---|---|
|
|
||||||
| `name` | `str` | `"jina-clip-v1"` | The model ID of the jina model to use |
|
|
||||||
|
|
||||||
Usage Example:
|
|
||||||
|
|
||||||
```python
|
|
||||||
import os
|
|
||||||
import requests
|
|
||||||
import lancedb
|
|
||||||
from lancedb.pydantic import LanceModel, Vector
|
|
||||||
from lancedb.embeddings import get_registry
|
|
||||||
import pandas as pd
|
|
||||||
|
|
||||||
os.environ['JINA_API_KEY'] = 'jina_*'
|
|
||||||
|
|
||||||
db = lancedb.connect("~/.lancedb")
|
|
||||||
func = get_registry().get("jina").create()
|
|
||||||
|
|
||||||
|
|
||||||
class Images(LanceModel):
|
|
||||||
label: str
|
|
||||||
image_uri: str = func.SourceField() # image uri as the source
|
|
||||||
image_bytes: bytes = func.SourceField() # image bytes as the source
|
|
||||||
vector: Vector(func.ndims()) = func.VectorField() # vector column
|
|
||||||
vec_from_bytes: Vector(func.ndims()) = func.VectorField() # Another vector column
|
|
||||||
|
|
||||||
|
|
||||||
table = db.create_table("images", schema=Images)
|
|
||||||
labels = ["cat", "cat", "dog", "dog", "horse", "horse"]
|
|
||||||
uris = [
|
|
||||||
"http://farm1.staticflickr.com/53/167798175_7c7845bbbd_z.jpg",
|
|
||||||
"http://farm1.staticflickr.com/134/332220238_da527d8140_z.jpg",
|
|
||||||
"http://farm9.staticflickr.com/8387/8602747737_2e5c2a45d4_z.jpg",
|
|
||||||
"http://farm5.staticflickr.com/4092/5017326486_1f46057f5f_z.jpg",
|
|
||||||
"http://farm9.staticflickr.com/8216/8434969557_d37882c42d_z.jpg",
|
|
||||||
"http://farm6.staticflickr.com/5142/5835678453_4f3a4edb45_z.jpg",
|
|
||||||
]
|
|
||||||
# get each uri as bytes
|
|
||||||
image_bytes = [requests.get(uri).content for uri in uris]
|
|
||||||
table.add(
|
|
||||||
pd.DataFrame({"label": labels, "image_uri": uris, "image_bytes": image_bytes})
|
|
||||||
)
|
|
||||||
```
|
|
||||||
@@ -94,8 +94,8 @@ the embeddings at all:
|
|||||||
=== "@lancedb/lancedb"
|
=== "@lancedb/lancedb"
|
||||||
|
|
||||||
```ts
|
```ts
|
||||||
--8<-- "nodejs/examples/embedding.ts:imports"
|
--8<-- "nodejs/examples/embedding.test.ts:imports"
|
||||||
--8<-- "nodejs/examples/embedding.ts:embedding_function"
|
--8<-- "nodejs/examples/embedding.test.ts:embedding_function"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)"
|
=== "vectordb (deprecated)"
|
||||||
@@ -150,7 +150,7 @@ need to worry about it when you query the table:
|
|||||||
.toArray()
|
.toArray()
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "vectordb (deprecated)
|
=== "vectordb (deprecated)"
|
||||||
|
|
||||||
```ts
|
```ts
|
||||||
const results = await table
|
const results = await table
|
||||||
|
|||||||
@@ -51,8 +51,8 @@ LanceDB registers the OpenAI embeddings function in the registry as `openai`. Yo
|
|||||||
=== "TypeScript"
|
=== "TypeScript"
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
--8<--- "nodejs/examples/embedding.ts:imports"
|
--8<--- "nodejs/examples/embedding.test.ts:imports"
|
||||||
--8<--- "nodejs/examples/embedding.ts:openai_embeddings"
|
--8<--- "nodejs/examples/embedding.test.ts:openai_embeddings"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "Rust"
|
=== "Rust"
|
||||||
@@ -121,12 +121,10 @@ class Words(LanceModel):
|
|||||||
vector: Vector(func.ndims()) = func.VectorField()
|
vector: Vector(func.ndims()) = func.VectorField()
|
||||||
|
|
||||||
table = db.create_table("words", schema=Words)
|
table = db.create_table("words", schema=Words)
|
||||||
table.add(
|
table.add([
|
||||||
[
|
|
||||||
{"text": "hello world"},
|
{"text": "hello world"},
|
||||||
{"text": "goodbye world"}
|
{"text": "goodbye world"}
|
||||||
]
|
])
|
||||||
)
|
|
||||||
|
|
||||||
query = "greetings"
|
query = "greetings"
|
||||||
actual = table.search(query).limit(1).to_pydantic(Words)[0]
|
actual = table.search(query).limit(1).to_pydantic(Words)[0]
|
||||||
|
|||||||
133
docs/src/embeddings/understanding_embeddings.md
Normal file
133
docs/src/embeddings/understanding_embeddings.md
Normal file
@@ -0,0 +1,133 @@
|
|||||||
|
# Understand Embeddings
|
||||||
|
|
||||||
|
The term **dimension** is a synonym for the number of elements in a feature vector. Each feature can be thought of as a different axis in a geometric space.
|
||||||
|
|
||||||
|
High-dimensional data means there are many features(or attributes) in the data.
|
||||||
|
|
||||||
|
!!! example
|
||||||
|
1. An image is a data point and it might have thousands of dimensions because each pixel could be considered as a feature.
|
||||||
|
|
||||||
|
2. Text data, when represented by each word or character, can also lead to high dimensions, especially when considering all possible words in a language.
|
||||||
|
|
||||||
|
Embedding captures **meaning and relationships** within data by mapping high-dimensional data into a lower-dimensional space. It captures it by placing inputs that are more **similar in meaning** closer together in the **embedding space**.
|
||||||
|
|
||||||
|
## What are Vector Embeddings?
|
||||||
|
|
||||||
|
Vector embeddings is a way to convert complex data, like text, images, or audio into numerical coordinates (called vectors) that can be plotted in an n-dimensional space(embedding space).
|
||||||
|
|
||||||
|
The closer these data points are related in the real world, the closer their corresponding numerical coordinates (vectors) will be to each other in the embedding space. This proximity in the embedding space reflects their semantic similarities, allowing machines to intuitively understand and process the data in a way that mirrors human perception of relationships and meaning.
|
||||||
|
|
||||||
|
In a way, it captures the most important aspects of the data while ignoring the less important ones. As a result, tasks like searching for related content or identifying patterns become more efficient and accurate, as the embeddings make it possible to quantify how **closely related** different **data points** are and **reduce** the **computational complexity**.
|
||||||
|
|
||||||
|
??? question "Are vectors and embeddings the same thing?"
|
||||||
|
|
||||||
|
When we say “vectors” we mean - **list of numbers** that **represents the data**.
|
||||||
|
When we say “embeddings” we mean - **list of numbers** that **capture important details and relationships**.
|
||||||
|
|
||||||
|
Although the terms are often used interchangeably, “embeddings” highlight how the data is represented with meaning and structure, while “vector” simply refers to the numerical form of that representation.
|
||||||
|
|
||||||
|
## Embedding vs Indexing
|
||||||
|
|
||||||
|
We already saw that creating **embeddings** on data is a method of creating **vectors** for a **n-dimensional embedding space** that captures the meaning and relationships inherent in the data.
|
||||||
|
|
||||||
|
Once we have these **vectors**, indexing comes into play. Indexing is a method of organizing these vector embeddings, that allows us to quickly and efficiently locate and retrieve them from the entire dataset of vector embeddings.
|
||||||
|
|
||||||
|
## What types of data/objects can be embedded?
|
||||||
|
|
||||||
|
The following are common types of data that can be embedded:
|
||||||
|
|
||||||
|
1. **Text**: Text data includes sentences, paragraphs, documents, or any written content.
|
||||||
|
2. **Images**: Image data encompasses photographs, illustrations, or any visual content.
|
||||||
|
3. **Audio**: Audio data includes sounds, music, speech, or any auditory content.
|
||||||
|
4. **Video**: Video data consists of moving images and sound, which can convey complex information.
|
||||||
|
|
||||||
|
Large datasets of multi-modal data (text, audio, images, etc.) can be converted into embeddings with the appropriate model.
|
||||||
|
|
||||||
|
!!! tip "LanceDB vs Other traditional Vector DBs"
|
||||||
|
While many vector databases primarily focus on the storage and retrieval of vector embeddings, **LanceDB** uses **Lance file format** (operates on a disk-based architecture), which allows for the storage and management of not just embeddings but also **raw file data (bytes)**. This capability means that users can integrate various types of data, including images and text, alongside their vector embeddings in a unified system.
|
||||||
|
|
||||||
|
With the ability to store both vectors and associated file data, LanceDB enhances the querying process. Users can perform semantic searches that not only retrieve similar embeddings but also access related files and metadata, thus streamlining the workflow.
|
||||||
|
|
||||||
|
## How does embedding works?
|
||||||
|
|
||||||
|
As mentioned, after creating embedding, each data point is represented as a vector in a n-dimensional space (embedding space). The dimensionality of this space can vary depending on the complexity of the data and the specific embedding technique used.
|
||||||
|
|
||||||
|
Points that are close to each other in vector space are considered similar (or appear in similar contexts), and points that are far away are considered dissimilar. To quantify this closeness, we use distance as a metric which can be measured in the following way -
|
||||||
|
|
||||||
|
1. **Euclidean Distance (l2)**: It calculates the straight-line distance between two points (vectors) in a multidimensional space.
|
||||||
|
2. **Cosine Similarity**: It measures the cosine of the angle between two vectors, providing a normalized measure of similarity based on their direction.
|
||||||
|
3. **Dot product**: It is calculated as the sum of the products of their corresponding components. To measure relatedness it considers both the magnitude and direction of the vectors.
|
||||||
|
|
||||||
|
## How do you create and store vector embeddings for your data?
|
||||||
|
|
||||||
|
1. **Creating embeddings**: Choose an embedding model, it can be a pre-trained model (open-source or commercial) or you can train a custom embedding model for your scenario. Then feed your preprocessed data into the chosen model to obtain embeddings.
|
||||||
|
|
||||||
|
??? question "Popular choices for embedding models"
|
||||||
|
For text data, popular choices are OpenAI’s text-embedding models, Google Gemini text-embedding models, Cohere’s Embed models, and SentenceTransformers, etc.
|
||||||
|
|
||||||
|
For image data, popular choices are CLIP (Contrastive Language–Image Pretraining), Imagebind embeddings by meta (supports audio, video, and image), and Jina multi-modal embeddings, etc.
|
||||||
|
|
||||||
|
2. **Storing vector embeddings**: This effectively requires **specialized databases** that can handle the complexity of vector data, as traditional databases often struggle with this task. Vector databases are designed specifically for storing and querying vector embeddings. They optimize for efficient nearest-neighbor searches and provide built-in indexing mechanisms.
|
||||||
|
|
||||||
|
!!! tip "Why LanceDB"
|
||||||
|
LanceDB **automates** the entire process of creating and storing embeddings for your data. LanceDB allows you to define and use **embedding functions**, which can be **pre-trained models** or **custom models**.
|
||||||
|
|
||||||
|
This enables you to **generate** embeddings tailored to the nature of your data (e.g., text, images) and **store** both the **original data** and **embeddings** in a **structured schema** thus providing efficient querying capabilities for similarity searches.
|
||||||
|
|
||||||
|
Let's quickly [get started](./index.md) and learn how to manage embeddings in LanceDB.
|
||||||
|
|
||||||
|
## Bonus: As a developer, what you can create using embeddings?
|
||||||
|
|
||||||
|
As a developer, you can create a variety of innovative applications using vector embeddings. Check out the following -
|
||||||
|
|
||||||
|
<div class="grid cards" markdown>
|
||||||
|
|
||||||
|
- __Chatbots__
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
Develop chatbots that utilize embeddings to retrieve relevant context and generate coherent, contextually aware responses to user queries.
|
||||||
|
|
||||||
|
[:octicons-arrow-right-24: Check out examples](../examples/python_examples/chatbot.md)
|
||||||
|
|
||||||
|
- __Recommendation Systems__
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
Develop systems that recommend content (such as articles, movies, or products) based on the similarity of keywords and descriptions, enhancing user experience.
|
||||||
|
|
||||||
|
[:octicons-arrow-right-24: Check out examples](../examples/python_examples/recommendersystem.md)
|
||||||
|
|
||||||
|
- __Vector Search__
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
Build powerful applications that harness the full potential of semantic search, enabling them to retrieve relevant data quickly and effectively.
|
||||||
|
|
||||||
|
[:octicons-arrow-right-24: Check out examples](../examples/python_examples/vector_search.md)
|
||||||
|
|
||||||
|
- __RAG Applications__
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
Combine the strengths of large language models (LLMs) with retrieval-based approaches to create more useful applications.
|
||||||
|
|
||||||
|
[:octicons-arrow-right-24: Check out examples](../examples/python_examples/rag.md)
|
||||||
|
|
||||||
|
- __Many more examples__
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
Explore applied examples available as Colab notebooks or Python scripts to integrate into your applications.
|
||||||
|
|
||||||
|
[:octicons-arrow-right-24: More](../examples/examples_python.md)
|
||||||
|
|
||||||
|
</div>
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
53
docs/src/embeddings/variables_and_secrets.md
Normal file
53
docs/src/embeddings/variables_and_secrets.md
Normal file
@@ -0,0 +1,53 @@
|
|||||||
|
# Variable and Secrets
|
||||||
|
|
||||||
|
Most embedding configuration options are saved in the table's metadata. However,
|
||||||
|
this isn't always appropriate. For example, API keys should never be stored in the
|
||||||
|
metadata. Additionally, other configuration options might be best set at runtime,
|
||||||
|
such as the `device` configuration that controls whether to use GPU or CPU for
|
||||||
|
inference. If you hardcoded this to GPU, you wouldn't be able to run the code on
|
||||||
|
a server without one.
|
||||||
|
|
||||||
|
To handle these cases, you can set variables on the embedding registry and
|
||||||
|
reference them in the embedding configuration. These variables will be available
|
||||||
|
during the runtime of your program, but not saved in the table's metadata. When
|
||||||
|
the table is loaded from a different process, the variables must be set again.
|
||||||
|
|
||||||
|
To set a variable, use the `set_var()` / `setVar()` method on the embedding registry.
|
||||||
|
To reference a variable, use the syntax `$env:VARIABLE_NAME`. If there is a default
|
||||||
|
value, you can use the syntax `$env:VARIABLE_NAME:DEFAULT_VALUE`.
|
||||||
|
|
||||||
|
## Using variables to set secrets
|
||||||
|
|
||||||
|
Sensitive configuration, such as API keys, must either be set as environment
|
||||||
|
variables or using variables on the embedding registry. If you pass in a hardcoded
|
||||||
|
value, LanceDB will raise an error. Instead, if you want to set an API key via
|
||||||
|
configuration, use a variable:
|
||||||
|
|
||||||
|
=== "Python"
|
||||||
|
|
||||||
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_embeddings_optional.py:register_secret"
|
||||||
|
```
|
||||||
|
|
||||||
|
=== "Typescript"
|
||||||
|
|
||||||
|
```typescript
|
||||||
|
--8<-- "nodejs/examples/embedding.test.ts:register_secret"
|
||||||
|
```
|
||||||
|
|
||||||
|
## Using variables to set the device parameter
|
||||||
|
|
||||||
|
Many embedding functions that run locally have a `device` parameter that controls
|
||||||
|
whether to use GPU or CPU for inference. Because not all computers have a GPU,
|
||||||
|
it's helpful to be able to set the `device` parameter at runtime, rather than
|
||||||
|
have it hard coded in the embedding configuration. To make it work even if the
|
||||||
|
variable isn't set, you could provide a default value of `cpu` in the embedding
|
||||||
|
configuration.
|
||||||
|
|
||||||
|
Some embedding libraries even have a method to detect which devices are available,
|
||||||
|
which could be used to dynamically set the device at runtime. For example, in Python
|
||||||
|
you can check if a CUDA GPU is available using `torch.cuda.is_available()`.
|
||||||
|
|
||||||
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_embeddings_optional.py:register_device"
|
||||||
|
```
|
||||||
@@ -1,17 +1,22 @@
|
|||||||
# Examples: Python
|
# Overview : Python Examples
|
||||||
|
|
||||||
To help you get started, we provide some examples, projects and applications that use the LanceDB Python API. You can always find the latest examples in our [VectorDB Recipes](https://github.com/lancedb/vectordb-recipes) repository.
|
To help you get started, we provide some examples, projects, and applications that use the LanceDB Python API. These examples are designed to get you right into the code with minimal introduction, enabling you to move from an idea to a proof of concept in minutes.
|
||||||
|
|
||||||
| Example | Interactive Envs | Scripts |
|
You can find the latest examples in our [VectorDB Recipes](https://github.com/lancedb/vectordb-recipes) repository.
|
||||||
|-------- | ---------------- | ------ |
|
|
||||||
| | | |
|
**Introduction**
|
||||||
| [Youtube transcript search bot](https://github.com/lancedb/vectordb-recipes/tree/main/examples/youtube_bot/) | <a href="https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/youtube_bot/main.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"></a>| [](https://github.com/lancedb/vectordb-recipes/tree/main/examples/youtube_bot/main.py)|
|
|
||||||
| [Langchain: Code Docs QA bot](https://github.com/lancedb/vectordb-recipes/tree/main/examples/Code-Documentation-QA-Bot/) | <a href="https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/Code-Documentation-QA-Bot/main.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"></a>| [](https://github.com/lancedb/vectordb-recipes/tree/main/examples/Code-Documentation-QA-Bot/main.py) |
|
Explore applied examples available as Colab notebooks or Python scripts to integrate into your applications. You can also checkout our blog posts related to the particular example for deeper understanding.
|
||||||
| [AI Agents: Reducing Hallucination](https://github.com/lancedb/vectordb-recipes/tree/main/examples/reducing_hallucinations_ai_agents/) | <a href="https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/reducing_hallucinations_ai_agents/main.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"></a>| [](https://github.com/lancedb/vectordb-recipes/tree/main/examples/reducing_hallucinations_ai_agents/main.py)|
|
|
||||||
| [Multimodal CLIP: DiffusionDB](https://github.com/lancedb/vectordb-recipes/tree/main/examples/multimodal_clip/) | <a href="https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/multimodal_clip/main.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"></a>| [](https://github.com/lancedb/vectordb-recipes/tree/main/examples/multimodal_clip/main.py) |
|
| Explore | Description |
|
||||||
| [Multimodal CLIP: Youtube videos](https://github.com/lancedb/vectordb-recipes/tree/main/examples/multimodal_video_search/) | <a href="https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/multimodal_video_search/main.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"></a>| [](https://github.com/lancedb/vectordb-recipes/tree/main/examples/multimodal_video_search/main.py) |
|
|----------------------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
||||||
| [Movie Recommender](https://github.com/lancedb/vectordb-recipes/tree/main/examples/movie-recommender/) | <a href="https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/movie-recommender/main.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"></a> | [](https://github.com/lancedb/vectordb-recipes/tree/main/examples/movie-recommender/main.py) |
|
| [**Build from Scratch with LanceDB** 🛠️🚀](python_examples/build_from_scratch.md) | Start building your **GenAI applications** from the **ground up** using **LanceDB's** efficient vector-based document retrieval capabilities! Get started quickly with a solid foundation. |
|
||||||
| [Audio Search](https://github.com/lancedb/vectordb-recipes/tree/main/examples/audio_search/) | <a href="https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/audio_search/main.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"></a> | [](https://github.com/lancedb/vectordb-recipes/tree/main/examples/audio_search/main.py) |
|
| [**Multimodal Search with LanceDB** 🤹♂️🔍](python_examples/multimodal.md) | Combine **text** and **image queries** to find the most relevant results using **LanceDB’s multimodal** capabilities. Leverage the efficient vector-based similarity search. |
|
||||||
| [Multimodal Image + Text Search](https://github.com/lancedb/vectordb-recipes/tree/main/examples/multimodal_search/) | <a href="https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/multimodal_search/main.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"></a> | [](https://github.com/lancedb/vectordb-recipes/tree/main/examples/multimodal_search/main.py) |
|
| [**RAG (Retrieval-Augmented Generation) with LanceDB** 🔓🧐](python_examples/rag.md) | Build RAG (Retrieval-Augmented Generation) with **LanceDB** for efficient **vector-based information retrieval** and more accurate responses from AI. |
|
||||||
| [Evaluating Prompts with Prompttools](https://github.com/lancedb/vectordb-recipes/tree/main/examples/prompttools-eval-prompts/) | <a href="https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/prompttools-eval-prompts/main.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"></a> | |
|
| [**Vector Search: Efficient Retrieval** 🔓👀](python_examples/vector_search.md) | Use **LanceDB's** vector search capabilities to perform efficient and accurate **similarity searches**, enabling rapid discovery and retrieval of relevant documents in Large datasets. |
|
||||||
|
| [**Chatbot applications with LanceDB** 🤖](python_examples/chatbot.md) | Create **chatbots** that retrieves relevant context for **coherent and context-aware replies**, enhancing user experience through advanced conversational AI. |
|
||||||
|
| [**Evaluation: Assessing Text Performance with Precision** 📊💡](python_examples/evaluations.md) | Develop **evaluation** applications that allows you to input reference and candidate texts to **measure** their performance across various metrics. |
|
||||||
|
| [**AI Agents: Intelligent Collaboration** 🤖](python_examples/aiagent.md) | Enable **AI agents** to communicate and collaborate efficiently through dense vector representations, achieving shared goals seamlessly. |
|
||||||
|
| [**Recommender Systems: Personalized Discovery** 🍿📺](python_examples/recommendersystem.md) | Deliver **personalized experiences** by efficiently storing and querying item embeddings with **LanceDB's** powerful vector database capabilities. |
|
||||||
|
| **Miscellaneous Examples🌟** | Find other **unique examples** and **creative solutions** using **LanceDB**, showcasing the flexibility and broad applicability of the platform. |
|
||||||
|
|
||||||
|
|||||||
@@ -8,9 +8,5 @@ LanceDB provides language APIs, allowing you to embed a database in your languag
|
|||||||
* 👾 [JavaScript](examples_js.md) examples
|
* 👾 [JavaScript](examples_js.md) examples
|
||||||
* 🦀 Rust examples (coming soon)
|
* 🦀 Rust examples (coming soon)
|
||||||
|
|
||||||
## Applications powered by LanceDB
|
!!! tip "Hosted LanceDB"
|
||||||
|
If you want S3 cost-efficiency and local performance via a simple serverless API, checkout **LanceDB Cloud**. For private deployments, high performance at extreme scale, or if you have strict security requirements, talk to us about **LanceDB Enterprise**. [Learn more](https://docs.lancedb.com/)
|
||||||
| Project Name | Description |
|
|
||||||
| --- | --- |
|
|
||||||
| **Ultralytics Explorer 🚀**<br>[](https://docs.ultralytics.com/datasets/explorer/)<br>[](https://colab.research.google.com/github/ultralytics/ultralytics/blob/main/docs/en/datasets/explorer/explorer.ipynb) | - 🔍 **Explore CV Datasets**: Semantic search, SQL queries, vector similarity, natural language.<br>- 🖥️ **GUI & Python API**: Seamless dataset interaction.<br>- ⚡ **Efficient & Scalable**: Leverages LanceDB for large datasets.<br>- 📊 **Detailed Analysis**: Easily analyze data patterns.<br>- 🌐 **Browser GUI Demo**: Create embeddings, search images, run queries. |
|
|
||||||
| **Website Chatbot🤖**<br>[](https://github.com/lancedb/lancedb-vercel-chatbot)<br>[](https://vercel.com/new/clone?repository-url=https%3A%2F%2Fgithub.com%2Flancedb%2Flancedb-vercel-chatbot&env=OPENAI_API_KEY&envDescription=OpenAI%20API%20Key%20for%20chat%20completion.&project-name=lancedb-vercel-chatbot&repository-name=lancedb-vercel-chatbot&demo-title=LanceDB%20Chatbot%20Demo&demo-description=Demo%20website%20chatbot%20with%20LanceDB.&demo-url=https%3A%2F%2Flancedb.vercel.app&demo-image=https%3A%2F%2Fi.imgur.com%2FazVJtvr.png) | - 🌐 **Chatbot from Sitemap/Docs**: Create a chatbot using site or document context.<br>- 🚀 **Embed LanceDB in Next.js**: Lightweight, on-prem storage.<br>- 🧠 **AI-Powered Context Retrieval**: Efficiently access relevant data.<br>- 🔧 **Serverless & Native JS**: Seamless integration with Next.js.<br>- ⚡ **One-Click Deploy on Vercel**: Quick and easy setup.. |
|
|
||||||
@@ -1,15 +1,15 @@
|
|||||||
# AI Agents: Intelligent Collaboration🤖
|
# AI Agents: Intelligent Collaboration🤖
|
||||||
|
|
||||||
Think of a platform💻 where AI Agents🤖 can seamlessly exchange information, coordinate over tasks, and achieve shared targets with great efficiency📈🚀.
|
Think of a platform where AI Agents can seamlessly exchange information, coordinate over tasks, and achieve shared targets with great efficiency💻📈.
|
||||||
|
|
||||||
## Vector-Based Coordination: The Technical Advantage
|
## Vector-Based Coordination: The Technical Advantage
|
||||||
Leveraging LanceDB's vector-based capabilities, our coordination application enables AI agents to communicate and collaborate through dense vector representations 🤖. AI agents can exchange information, coordinate on a task or work towards a common goal, just by giving queries📝.
|
Leveraging LanceDB's vector-based capabilities, we can enable **AI agents 🤖** to communicate and collaborate through dense vector representations. AI agents can exchange information, coordinate on a task or work towards a common goal, just by giving queries📝.
|
||||||
|
|
||||||
| **AI Agents** | **Description** | **Links** |
|
| **AI Agents** | **Description** | **Links** |
|
||||||
|:--------------|:----------------|:----------|
|
|:--------------|:----------------|:----------|
|
||||||
| **AI Agents: Reducing Hallucinationt📊** | 🤖💡 Reduce AI hallucinations using Critique-Based Contexting! Learn by Simplifying and Automating tedious workflows by going through fitness trainer agent example.💪 | [][hullucination_github] <br>[][hullucination_colab] <br>[][hullucination_python] <br>[][hullucination_ghost] |
|
| **AI Agents: Reducing Hallucinationt📊** | 🤖💡 **Reduce AI hallucinations** using Critique-Based Contexting! Learn by Simplifying and Automating tedious workflows by going through fitness trainer agent example.💪 | [][hullucination_github] <br>[][hullucination_colab] <br>[][hullucination_python] <br>[][hullucination_ghost] |
|
||||||
| **AI Trends Searcher: CrewAI🔍️** | 🔍️ Learn about CrewAI Agents ! Utilize the features of CrewAI - Role-based Agents, Task Management, and Inter-agent Delegation ! Make AI agents work together to do tricky stuff 😺| [][trend_github] <br>[][trend_colab] <br>[][trend_ghost] |
|
| **AI Trends Searcher: CrewAI🔍️** | 🔍️ Learn about **CrewAI Agents** ! Utilize the features of CrewAI - Role-based Agents, Task Management, and Inter-agent Delegation ! Make AI agents work together to do tricky stuff 😺| [][trend_github] <br>[][trend_colab] <br>[][trend_ghost] |
|
||||||
| **SuperAgent Autogen🤖** | 💻 AI interactions with the Super Agent! Integrating Autogen, LanceDB, LangChain, LiteLLM, and Ollama to create AI agent that excels in understanding and processing complex queries.🤖 | [][superagent_github] <br>[][superagent_colab] |
|
| **SuperAgent Autogen🤖** | 💻 AI interactions with the Super Agent! Integrating **Autogen**, **LanceDB**, **LangChain**, **LiteLLM**, and **Ollama** to create AI agent that excels in understanding and processing complex queries.🤖 | [][superagent_github] <br>[][superagent_colab] |
|
||||||
|
|
||||||
|
|
||||||
[hullucination_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/reducing_hallucinations_ai_agents
|
[hullucination_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/reducing_hallucinations_ai_agents
|
||||||
|
|||||||
@@ -1,10 +1,10 @@
|
|||||||
# **Build from Scratch with LanceDB 🛠️🚀**
|
# **Build from Scratch with LanceDB 🛠️🚀**
|
||||||
|
|
||||||
Start building your GenAI applications from the ground up using LanceDB's efficient vector-based document retrieval capabilities! 📑
|
Start building your GenAI applications from the ground up using **LanceDB's** efficient vector-based document retrieval capabilities! 📑
|
||||||
|
|
||||||
**Get Started in Minutes ⏱️**
|
**Get Started in Minutes ⏱️**
|
||||||
|
|
||||||
These examples provide a solid foundation for building your own GenAI applications using LanceDB. Jump from idea to proof of concept quickly with applied examples. Get started and see what you can create! 💻
|
These examples provide a solid foundation for building your own GenAI applications using LanceDB. Jump from idea to **proof of concept** quickly with applied examples. Get started and see what you can create! 💻
|
||||||
|
|
||||||
| **Build From Scratch** | **Description** | **Links** |
|
| **Build From Scratch** | **Description** | **Links** |
|
||||||
|:-------------------------------------------|:-------------------------------------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
|:-------------------------------------------|:-------------------------------------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
**Chatbot Application with LanceDB 🤖**
|
**Chatbot applications with LanceDB 🤖**
|
||||||
====================================================================
|
====================================================================
|
||||||
|
|
||||||
Create an innovative chatbot application that utilizes LanceDB for efficient vector-based response generation! 🌐✨
|
Create innovative chatbot applications that utilizes LanceDB for efficient vector-based response generation! 🌐✨
|
||||||
|
|
||||||
**Introduction 👋✨**
|
**Introduction 👋✨**
|
||||||
|
|
||||||
@@ -10,12 +10,12 @@
|
|||||||
|
|
||||||
| **Chatbot** | **Description** | **Links** |
|
| **Chatbot** | **Description** | **Links** |
|
||||||
|:----------------|:-----------------|:-----------|
|
|:----------------|:-----------------|:-----------|
|
||||||
| **Databricks DBRX Website Bot ⚡️** | Unlock magical conversations with the Hogwarts chatbot, powered by Open-source RAG, DBRX, LanceDB, LLama-index, and Hugging Face Embeddings, delivering enchanting user experiences and spellbinding interactions ✨ | [][databricks_github] <br>[][databricks_python] |
|
| **Databricks DBRX Website Bot ⚡️** | Engage with the **Hogwarts chatbot**, that uses Open-source RAG with **DBRX**, **LanceDB** and **LLama-index with Hugging Face Embeddings**, to provide interactive and engaging user experiences. ✨ | [][databricks_github] <br>[][databricks_python] |
|
||||||
| **CLI SDK Manual Chatbot Locally 💻** | CLI chatbot for SDK/hardware documents, powered by Local RAG, LLama3, Ollama, LanceDB, and Openhermes Embeddings, built with Phidata Assistant and Knowledge Base for instant technical support 🤖 | [][clisdk_github] <br>[][clisdk_python] |
|
| **CLI SDK Manual Chatbot Locally 💻** | CLI chatbot for SDK/hardware documents using **Local RAG** with **LLama3**, **Ollama**, **LanceDB**, and **Openhermes Embeddings**, built with **Phidata** Assistant and Knowledge Base 🤖 | [][clisdk_github] <br>[][clisdk_python] |
|
||||||
| **Youtube Transcript Search QA Bot 📹** | Unlock the power of YouTube transcripts with a Q&A bot, leveraging natural language search and LanceDB for effortless data management and instant answers 💬 | [][youtube_github] <br>[][youtube_colab] <br>[][youtube_python] |
|
| **Youtube Transcript Search QA Bot 📹** | Search through **youtube transcripts** using natural language with a Q&A bot, leveraging **LanceDB** for effortless data storage and management 💬 | [][youtube_github] <br>[][youtube_colab] <br>[][youtube_python] |
|
||||||
| **Code Documentation Q&A Bot with LangChain 🤖** | Revolutionize code documentation with a Q&A bot, powered by LangChain and LanceDB, allowing effortless querying of documentation using natural language, demonstrated with Numpy 1.26 docs 📚 | [][docs_github] <br>[][docs_colab] <br>[][docs_python] |
|
| **Code Documentation Q&A Bot with LangChain 🤖** | Query your own documentation easily using questions in natural language with a Q&A bot, powered by **LangChain** and **LanceDB**, demonstrated with **Numpy 1.26 docs** 📚 | [][docs_github] <br>[][docs_colab] <br>[][docs_python] |
|
||||||
| **Context-aware Chatbot using Llama 2 & LanceDB 🤖** | Experience the future of conversational AI with a context-aware chatbot, powered by Llama 2, LanceDB, and LangChain, enabling intuitive and meaningful conversations with your data 📚💬 | [][aware_github] <br>[][aware_colab] <br>[][aware_ghost] |
|
| **Context-aware Chatbot using Llama 2 & LanceDB 🤖** | Build **conversational AI** with a **context-aware chatbot**, powered by **Llama 2**, **LanceDB**, and **LangChain**, that enables intuitive and meaningful conversations with your data 📚💬 | [][aware_github] <br>[][aware_colab] <br>[][aware_ghost] |
|
||||||
| **Chat with csv using Hybrid Search 📊** | Revolutionize data interaction with a chat application that harnesses LanceDB's hybrid search capabilities to converse with CSV and Excel files, enabling efficient and scalable data exploration and analysis 🚀 | [][csv_github] <br>[][csv_colab] <br>[][csv_ghost] |
|
| **Chat with csv using Hybrid Search 📊** | **Chat** application that interacts with **CSV** and **Excel files** using **LanceDB’s** hybrid search capabilities, performing direct operations on large-scale columnar data efficiently 🚀 | [][csv_github] <br>[][csv_colab] <br>[][csv_ghost] |
|
||||||
|
|
||||||
|
|
||||||
[databricks_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/databricks_DBRX_website_bot
|
[databricks_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/databricks_DBRX_website_bot
|
||||||
@@ -36,6 +36,6 @@
|
|||||||
[aware_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/tutorials/chatbot_using_Llama2_&_lanceDB/main.ipynb
|
[aware_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/tutorials/chatbot_using_Llama2_&_lanceDB/main.ipynb
|
||||||
[aware_ghost]: https://blog.lancedb.com/context-aware-chatbot-using-llama-2-lancedb-as-vector-database-4d771d95c755
|
[aware_ghost]: https://blog.lancedb.com/context-aware-chatbot-using-llama-2-lancedb-as-vector-database-4d771d95c755
|
||||||
|
|
||||||
[csv_github]: https://github.com/lancedb/vectordb-recipes/blob/main/tutorials/Chat_with_csv_file
|
[csv_github]: https://github.com/lancedb/vectordb-recipes/tree/main/examples/archived_examples/Chat_with_csv_file
|
||||||
[csv_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/tutorials/Chat_with_csv_file/main.ipynb
|
[csv_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/archived_examples/Chat_with_csv_file/main.ipynb
|
||||||
[csv_ghost]: https://blog.lancedb.com/p/d8c71df4-e55f-479a-819e-cde13354a6a3/
|
[csv_ghost]: https://blog.lancedb.com/p/d8c71df4-e55f-479a-819e-cde13354a6a3/
|
||||||
|
|||||||
@@ -1,18 +1,16 @@
|
|||||||
**Evaluation: Assessing Text Performance with Precision 📊💡**
|
**Evaluation: Assessing Text Performance with Precision 📊💡**
|
||||||
====================================================================
|
====================================================================
|
||||||
|
|
||||||
**Evaluation Fundamentals 📊**
|
|
||||||
|
|
||||||
Evaluation is a comprehensive tool designed to measure the performance of text-based inputs, enabling data-driven optimization and improvement 📈.
|
Evaluation is a comprehensive tool designed to measure the performance of text-based inputs, enabling data-driven optimization and improvement 📈.
|
||||||
|
|
||||||
**Text Evaluation 101 📚**
|
**Text Evaluation 101 📚**
|
||||||
|
|
||||||
By leveraging cutting-edge technologies, this provides a robust framework for evaluating reference and candidate texts across various metrics 📊, ensuring high-quality text outputs that meet specific requirements and standards 📝.
|
Using robust framework for assessing reference and candidate texts across various metrics📊, ensure that the text outputs are high-quality and meet specific requirements and standards📝.
|
||||||
|
|
||||||
| **Evaluation** | **Description** | **Links** |
|
| **Evaluation** | **Description** | **Links** |
|
||||||
| -------------- | --------------- | --------- |
|
| -------------- | --------------- | --------- |
|
||||||
| **Evaluating Prompts with Prompttools 🤖** | Compare, visualize & evaluate embedding functions (incl. OpenAI) across metrics like latency & custom evaluation 📈📊 | [][prompttools_github] <br>[][prompttools_colab] |
|
| **Evaluating Prompts with Prompttools 🤖** | Compare, visualize & evaluate **embedding functions** (incl. OpenAI) across metrics like latency & custom evaluation 📈📊 | [][prompttools_github] <br>[][prompttools_colab] |
|
||||||
| **Evaluating RAG with RAGAs and GPT-4o 📊** | Evaluate RAG pipelines with cutting-edge metrics and tools, integrate with CI/CD for continuous performance checks, and generate responses with GPT-4o 🤖📈 | [][RAGAs_github] <br>[][RAGAs_colab] |
|
| **Evaluating RAG with RAGAs and GPT-4o 📊** | Evaluate **RAG pipelines** with cutting-edge metrics and tools, integrate with CI/CD for continuous performance checks, and generate responses with GPT-4o 🤖📈 | [][RAGAs_github] <br>[][RAGAs_colab] |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
# **Multimodal Search with LanceDB 🤹♂️🔍**
|
# **Multimodal Search with LanceDB 🤹♂️🔍**
|
||||||
|
|
||||||
Experience the future of search with LanceDB's multimodal capabilities. Combine text and image queries to find the most relevant results in your corpus ! 🔓💡
|
Using LanceDB's multimodal capabilities, combine text and image queries to find the most relevant results in your corpus ! 🔓💡
|
||||||
|
|
||||||
**Explore the Future of Search 🚀**
|
**Explore the Future of Search 🚀**
|
||||||
|
|
||||||
@@ -10,10 +10,10 @@ LanceDB supports multimodal search by indexing and querying vector representatio
|
|||||||
|
|
||||||
| **Multimodal** | **Description** | **Links** |
|
| **Multimodal** | **Description** | **Links** |
|
||||||
|:----------------|:-----------------|:-----------|
|
|:----------------|:-----------------|:-----------|
|
||||||
| **Multimodal CLIP: DiffusionDB 🌐💥** | Revolutionize search with Multimodal CLIP and DiffusionDB, combining text and image understanding for a new dimension of discovery! 🔓 | [][Clip_diffusionDB_github] <br>[][Clip_diffusionDB_colab] <br>[][Clip_diffusionDB_python] <br>[][Clip_diffusionDB_ghost] |
|
| **Multimodal CLIP: DiffusionDB 🌐💥** | Multi-Modal Search with **CLIP** and **LanceDB** Using **DiffusionDB** Data for Combined Text and Image Understanding ! 🔓 | [][Clip_diffusionDB_github] <br>[][Clip_diffusionDB_colab] <br>[][Clip_diffusionDB_python] <br>[][Clip_diffusionDB_ghost] |
|
||||||
| **Multimodal CLIP: Youtube Videos 📹👀** | Search Youtube videos using Multimodal CLIP, finding relevant content with ease and accuracy! 🎯 | [][Clip_youtube_github] <br>[][Clip_youtube_colab] <br> [][Clip_youtube_python] <br>[][Clip_youtube_python] |
|
| **Multimodal CLIP: Youtube Videos 📹👀** | Search **Youtube videos** using Multimodal CLIP, finding relevant content with ease and accuracy! 🎯 | [][Clip_youtube_github] <br>[][Clip_youtube_colab] <br> [][Clip_youtube_python] <br>[][Clip_youtube_python] |
|
||||||
| **Multimodal Image + Text Search 📸🔍** | Discover relevant documents and images with a single query, using LanceDB's multimodal search capabilities to bridge the gap between text and visuals! 🌉 | [](https://github.com/lancedb/vectordb-recipes/blob/main/examples/multimodal_search) <br>[](https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/multimodal_search/main.ipynb) <br> [](https://github.com/lancedb/vectordb-recipes/blob/main/examples/multimodal_search/main.py)<br> [](https://blog.lancedb.com/multi-modal-ai-made-easy-with-lancedb-clip-5aaf8801c939/) |
|
| **Multimodal Image + Text Search 📸🔍** | Find **relevant documents** and **images** with a single query using **LanceDB's** multimodal search capabilities, to seamlessly integrate text and visuals ! 🌉 | [](https://github.com/lancedb/vectordb-recipes/tree/main/examples/archived_examples/multimodal_search) <br>[](https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/archived_examples/multimodal_search/main.ipynb) <br> [](https://github.com/lancedb/vectordb-recipes/blob/main/examples/multimodal_search/main.py)<br> [](https://blog.lancedb.com/multi-modal-ai-made-easy-with-lancedb-clip-5aaf8801c939/) |
|
||||||
| **Cambrian-1: Vision-Centric Image Exploration 🔍👀** | Dive into vision-centric exploration of images with Cambrian-1, powered by LanceDB's multimodal search to uncover new insights! 🔎 | [](https://www.kaggle.com/code/prasantdixit/cambrian-1-vision-centric-exploration-of-images/)<br> [](https://blog.lancedb.com/cambrian-1-vision-centric-exploration/) |
|
| **Cambrian-1: Vision-Centric Image Exploration 🔍👀** | Learn how **Cambrian-1** works, using an example of **Vision-Centric** exploration on images found through vector search ! Work on **Flickr-8k** dataset 🔎 | [](https://www.kaggle.com/code/prasantdixit/cambrian-1-vision-centric-exploration-of-images/)<br> [](https://blog.lancedb.com/cambrian-1-vision-centric-exploration/) |
|
||||||
|
|
||||||
|
|
||||||
[Clip_diffusionDB_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/multimodal_clip_diffusiondb
|
[Clip_diffusionDB_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/multimodal_clip_diffusiondb
|
||||||
|
|||||||
@@ -1,12 +1,11 @@
|
|||||||
|
**RAG (Retrieval-Augmented Generation) with LanceDB 🔓🧐**
|
||||||
**RAG: Revolutionize Information Retrieval with LanceDB 🔓🧐**
|
|
||||||
====================================================================
|
====================================================================
|
||||||
|
|
||||||
Unlock the full potential of Retrieval-Augmented Generation (RAG) with LanceDB, a solution for efficient vector-based information retrieval 📊.
|
Build RAG (Retrieval-Augmented Generation) with LanceDB, a powerful solution for efficient vector-based information retrieval 📊.
|
||||||
|
|
||||||
**Experience the Future of Search 🔄**
|
**Experience the Future of Search 🔄**
|
||||||
|
|
||||||
RAG integrates large language models (LLMs) with scalable knowledge bases, enabling efficient information retrieval and answer generation 🤖. By applying RAG to industry-specific use cases, developers can optimize query processing 📊, reduce response latency ⏱️, and improve resource utilization 💻. LanceDB provides a robust framework for integrating LLMs with external knowledge sources, facilitating accurate and informative responses 📝.
|
🤖 RAG enables AI to **retrieve** relevant information from external sources and use it to **generate** more accurate and context-specific responses. 💻 LanceDB provides a robust framework for integrating LLMs with external knowledge sources 📝.
|
||||||
|
|
||||||
| **RAG** | **Description** | **Links** |
|
| **RAG** | **Description** | **Links** |
|
||||||
|----------------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------|----------------------------|
|
|----------------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------|----------------------------|
|
||||||
@@ -18,10 +17,10 @@ RAG integrates large language models (LLMs) with scalable knowledge bases, enabl
|
|||||||
| **Advanced RAG: Parent Document Retriever** 📑🔗 | Use **Parent Document & Bigger Chunk Retriever** to maintain context and relevance when generating related content. 🎵📄 | [][parent_doc_retriever_github] <br>[][parent_doc_retriever_colab] <br>[][parent_doc_retriever_ghost] |
|
| **Advanced RAG: Parent Document Retriever** 📑🔗 | Use **Parent Document & Bigger Chunk Retriever** to maintain context and relevance when generating related content. 🎵📄 | [][parent_doc_retriever_github] <br>[][parent_doc_retriever_colab] <br>[][parent_doc_retriever_ghost] |
|
||||||
| **Corrective RAG with Langgraph** 🔧📊 | Enhance RAG reliability with **Corrective RAG (CRAG)** by self-reflecting and fact-checking for accurate and trustworthy results. ✅🔍 |[][corrective_rag_github] <br>[][corrective_rag_colab] <br>[][corrective_rag_ghost] |
|
| **Corrective RAG with Langgraph** 🔧📊 | Enhance RAG reliability with **Corrective RAG (CRAG)** by self-reflecting and fact-checking for accurate and trustworthy results. ✅🔍 |[][corrective_rag_github] <br>[][corrective_rag_colab] <br>[][corrective_rag_ghost] |
|
||||||
| **Contextual Compression with RAG** 🗜️🧠 | Apply **contextual compression techniques** to condense large documents while retaining essential information. 📄🗜️ | [][compression_rag_github] <br>[][compression_rag_colab] <br>[][compression_rag_ghost] |
|
| **Contextual Compression with RAG** 🗜️🧠 | Apply **contextual compression techniques** to condense large documents while retaining essential information. 📄🗜️ | [][compression_rag_github] <br>[][compression_rag_colab] <br>[][compression_rag_ghost] |
|
||||||
| **Improve RAG with FLARE** 🔥| Enable users to ask questions directly to academic papers, focusing on ArXiv papers, with Forward-Looking Active REtrieval augmented generation.🚀🌟 | [][flare_github] <br>[][flare_colab] <br>[][flare_ghost] |
|
| **Improve RAG with FLARE** 🔥| Enable users to ask questions directly to **academic papers**, focusing on **ArXiv papers**, with **F**orward-**L**ooking **A**ctive **RE**trieval augmented generation.🚀🌟 | [][flare_github] <br>[][flare_colab] <br>[][flare_ghost] |
|
||||||
| **Query Expansion and Reranker** 🔍🔄 | Enhance RAG with query expansion using Large Language Models and advanced **reranking methods** like Cross Encoders, ColBERT v2, and FlashRank for improved document retrieval precision and recall 🔍📈 | [][query_github] <br>[][query_colab] |
|
| **Query Expansion and Reranker** 🔍🔄 | Enhance RAG with query expansion using Large Language Models and advanced **reranking methods** like **Cross Encoders**, **ColBERT v2**, and **FlashRank** for improved document retrieval precision and recall 🔍📈 | [][query_github] <br>[][query_colab] |
|
||||||
| **RAG Fusion** ⚡🌐 | Revolutionize search with RAG Fusion, utilizing the **RRF algorithm** to rerank documents based on user queries, and leveraging LanceDB and OPENAI Embeddings for efficient information retrieval ⚡🌐 | [][fusion_github] <br>[][fusion_colab] |
|
| **RAG Fusion** ⚡🌐 | Build RAG Fusion, utilize the **RRF algorithm** to rerank documents based on user queries ! Use **LanceDB** as vector database to store and retrieve documents related to queries via **OPENAI Embeddings**⚡🌐 | [][fusion_github] <br>[][fusion_colab] |
|
||||||
| **Agentic RAG** 🤖📚 | Unlock autonomous information retrieval with **Agentic RAG**, a framework of **intelligent agents** that collaborate to synthesize, summarize, and compare data across sources, enabling proactive and informed decision-making 🤖📚 | [][agentic_github] <br>[][agentic_colab] |
|
| **Agentic RAG** 🤖📚 | Build autonomous information retrieval with **Agentic RAG**, a framework of **intelligent agents** that collaborate to synthesize, summarize, and compare data across sources, that enables proactive and informed decision-making 🤖📚 | [][agentic_github] <br>[][agentic_colab] |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@@ -71,12 +70,12 @@ RAG integrates large language models (LLMs) with scalable knowledge bases, enabl
|
|||||||
[flare_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/better-rag-FLAIR/main.ipynb
|
[flare_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/better-rag-FLAIR/main.ipynb
|
||||||
[flare_ghost]: https://blog.lancedb.com/better-rag-with-active-retrieval-augmented-generation-flare-3b66646e2a9f/
|
[flare_ghost]: https://blog.lancedb.com/better-rag-with-active-retrieval-augmented-generation-flare-3b66646e2a9f/
|
||||||
|
|
||||||
[query_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/QueryExpansion&Reranker
|
[query_github]: https://github.com/lancedb/vectordb-recipes/tree/main/examples/archived_examples/QueryExpansion%26Reranker
|
||||||
[query_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/QueryExpansion&Reranker/main.ipynb
|
[query_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/archived_examples/QueryExpansion&Reranker/main.ipynb
|
||||||
|
|
||||||
|
|
||||||
[fusion_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/RAG_Fusion
|
[fusion_github]: https://github.com/lancedb/vectordb-recipes/tree/main/examples/archived_examples/RAG_Fusion
|
||||||
[fusion_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/RAG_Fusion/main.ipynb
|
[fusion_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/archived_examples/RAG_Fusion/main.ipynb
|
||||||
|
|
||||||
[agentic_github]: https://github.com/lancedb/vectordb-recipes/blob/main/tutorials/Agentic_RAG
|
[agentic_github]: https://github.com/lancedb/vectordb-recipes/blob/main/tutorials/Agentic_RAG
|
||||||
[agentic_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/tutorials/Agentic_RAG/main.ipynb
|
[agentic_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/tutorials/Agentic_RAG/main.ipynb
|
||||||
|
|||||||
37
docs/src/examples/python_examples/recommendersystem.md
Normal file
37
docs/src/examples/python_examples/recommendersystem.md
Normal file
@@ -0,0 +1,37 @@
|
|||||||
|
**Recommender Systems: Personalized Discovery🍿📺**
|
||||||
|
==============================================================
|
||||||
|
Deliver personalized experiences with Recommender Systems. 🎁
|
||||||
|
|
||||||
|
**Technical Overview📜**
|
||||||
|
|
||||||
|
🔍️ LanceDB's powerful vector database capabilities can efficiently store and query item embeddings. Recommender Systems can utilize it and provide personalized recommendations based on user preferences 🤝 and item features 📊 and therefore enhance the user experience.🗂️
|
||||||
|
|
||||||
|
| **Recommender System** | **Description** | **Links** |
|
||||||
|
| ---------------------- | --------------- | --------- |
|
||||||
|
| **Movie Recommender System🎬** | 🤝 Use **collaborative filtering** to predict user preferences, assuming similar users will like similar movies, and leverage **Singular Value Decomposition** (SVD) from Numpy for precise matrix factorization and accurate recommendations📊 | [][movie_github] <br>[][movie_colab] <br>[][movie_python] |
|
||||||
|
| **🎥 Movie Recommendation with Genres** | 🔍 Creates movie embeddings using **Doc2Vec**, capturing genre and characteristic nuances, and leverages VectorDB for efficient storage and querying, enabling accurate genre classification and personalized movie recommendations through **similarity searches**🎥 | [][genre_github] <br>[][genre_colab] <br>[][genre_ghost] |
|
||||||
|
| **🛍️ Product Recommender using Collaborative Filtering and LanceDB** | 📈 Using **Collaborative Filtering** and **LanceDB** to analyze your past purchases, recommends products based on user's past purchases. Demonstrated with the Instacart dataset in our example🛒 | [][product_github] <br>[][product_colab] <br>[][product_python] |
|
||||||
|
| **🔍 Arxiv Search with OpenCLIP and LanceDB** | 💡 Build a semantic search engine for **Arxiv papers** using **LanceDB**, and benchmarks its performance against traditional keyword-based search on **Nomic's Atlas**, to demonstrate the power of semantic search in finding relevant research papers📚 | [][arxiv_github] <br>[][arxiv_colab] <br>[][arxiv_python] |
|
||||||
|
| **Food Recommendation System🍴** | 🍔 Build a food recommendation system with **LanceDB**, featuring vector-based recommendations, full-text search, hybrid search, and reranking model integration for personalized and accurate food suggestions👌 | [][food_github] <br>[][food_colab] |
|
||||||
|
|
||||||
|
[movie_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/movie-recommender
|
||||||
|
[movie_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/movie-recommender/main.ipynb
|
||||||
|
[movie_python]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/movie-recommender/main.py
|
||||||
|
|
||||||
|
|
||||||
|
[genre_github]: https://github.com/lancedb/vectordb-recipes/tree/main/examples/archived_examples/movie-recommendation-with-genres
|
||||||
|
[genre_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/archived_examples/movie-recommendation-with-genres/movie_recommendation_with_doc2vec_and_lancedb.ipynb
|
||||||
|
[genre_ghost]: https://blog.lancedb.com/movie-recommendation-system-using-lancedb-and-doc2vec/
|
||||||
|
|
||||||
|
[product_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/product-recommender
|
||||||
|
[product_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/product-recommender/main.ipynb
|
||||||
|
[product_python]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/product-recommender/main.py
|
||||||
|
|
||||||
|
|
||||||
|
[arxiv_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/arxiv-recommender
|
||||||
|
[arxiv_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/arxiv-recommender/main.ipynb
|
||||||
|
[arxiv_python]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/arxiv-recommender/main.py
|
||||||
|
|
||||||
|
|
||||||
|
[food_github]: https://github.com/lancedb/vectordb-recipes/tree/main/examples/archived_examples/Food_recommendation
|
||||||
|
[food_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/archived_examples/Food_recommendation/main.ipynb
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
**Vector Search: Unlock Efficient Document Retrieval 🔓👀**
|
**Vector Search: Efficient Retrieval 🔓👀**
|
||||||
====================================================================
|
====================================================================
|
||||||
|
|
||||||
Unlock the power of vector search with LanceDB, a cutting-edge solution for efficient vector-based document retrieval 📊.
|
Vector search with LanceDB, is a solution for efficient and accurate similarity searches in large datasets 📊.
|
||||||
|
|
||||||
**Vector Search Capabilities in LanceDB🔝**
|
**Vector Search Capabilities in LanceDB🔝**
|
||||||
|
|
||||||
@@ -9,19 +9,19 @@ LanceDB implements vector search algorithms for efficient document retrieval and
|
|||||||
|
|
||||||
| **Vector Search** | **Description** | **Links** |
|
| **Vector Search** | **Description** | **Links** |
|
||||||
|:-----------------|:---------------|:---------|
|
|:-----------------|:---------------|:---------|
|
||||||
| **Inbuilt Hybrid Search 🔄** | Combine the power of traditional search algorithms with LanceDB's vector-based search for a robust and efficient search experience 📊 | [][inbuilt_hybrid_search_github] <br>[][inbuilt_hybrid_search_colab] |
|
| **Inbuilt Hybrid Search 🔄** | Perform hybrid search in **LanceDB** by combining the results of semantic and full-text search via a reranking algorithm of your choice 📊 | [][inbuilt_hybrid_search_github] <br>[][inbuilt_hybrid_search_colab] |
|
||||||
| **Hybrid Search with BM25 and LanceDB 💡** | Synergizes BM25's keyword-focused precision (term frequency, document length normalization, bias-free retrieval) with LanceDB's semantic understanding (contextual analysis, query intent alignment) for nuanced search results in complex datasets 📈 | [][BM25_github] <br>[][BM25_colab] <br>[][BM25_ghost] |
|
| **Hybrid Search with BM25 and LanceDB 💡** | Use **Synergizes BM25's** keyword-focused precision (term frequency, document length normalization, bias-free retrieval) with **LanceDB's** semantic understanding (contextual analysis, query intent alignment) for nuanced search results in complex datasets 📈 | [][BM25_github] <br>[][BM25_colab] <br>[][BM25_ghost] |
|
||||||
| **NER-powered Semantic Search 🔎** | Unlock contextual understanding with Named Entity Recognition (NER) methods: Dictionary-Based, Rule-Based, and Deep Learning-Based, to accurately identify and extract entities, enabling precise semantic search results 🗂️ | [][NER_github] <br>[][NER_colab] <br>[][NER_ghost]|
|
| **NER-powered Semantic Search 🔎** | Extract and identify essential information from text with Named Entity Recognition **(NER)** methods: Dictionary-Based, Rule-Based, and Deep Learning-Based, to accurately extract and categorize entities, enabling precise semantic search results 🗂️ | [][NER_github] <br>[][NER_colab] <br>[][NER_ghost]|
|
||||||
| **Audio Similarity Search using Vector Embeddings 🎵** | Create vector embeddings of audio files to find similar audio content, enabling efficient audio similarity search and retrieval in LanceDB's vector store 📻 |[][audio_search_github] <br>[][audio_search_colab] <br>[][audio_search_python]|
|
| **Audio Similarity Search using Vector Embeddings 🎵** | Create vector **embeddings of audio files** to find similar audio content, enabling efficient audio similarity search and retrieval in **LanceDB's** vector store 📻 |[][audio_search_github] <br>[][audio_search_colab] <br>[][audio_search_python]|
|
||||||
| **LanceDB Embeddings API: Multi-lingual Semantic Search 🌎** | Build a universal semantic search table with LanceDB's Embeddings API, supporting multiple languages (e.g., English, French) using cohere's multi-lingual model, for accurate cross-lingual search results 📄 | [][mls_github] <br>[][mls_colab] <br>[][mls_python] |
|
| **LanceDB Embeddings API: Multi-lingual Semantic Search 🌎** | Build a universal semantic search table with **LanceDB's Embeddings API**, supporting multiple languages (e.g., English, French) using **cohere's** multi-lingual model, for accurate cross-lingual search results 📄 | [][mls_github] <br>[][mls_colab] <br>[][mls_python] |
|
||||||
| **Facial Recognition: Face Embeddings 🤖** | Detect, crop, and embed faces using Facenet, then store and query face embeddings in LanceDB for efficient facial recognition and top-K matching results 👥 | [][fr_github] <br>[][fr_colab] |
|
| **Facial Recognition: Face Embeddings 🤖** | Detect, crop, and embed faces using Facenet, then store and query face embeddings in **LanceDB** for efficient facial recognition and top-K matching results 👥 | [][fr_github] <br>[][fr_colab] |
|
||||||
| **Sentiment Analysis: Hotel Reviews 🏨** | Analyze customer sentiments towards the hotel industry using BERT models, storing sentiment labels, scores, and embeddings in LanceDB, enabling queries on customer opinions and potential areas for improvement 💬 | [][sentiment_analysis_github] <br>[][sentiment_analysis_colab] <br>[][sentiment_analysis_ghost] |
|
| **Sentiment Analysis: Hotel Reviews 🏨** | Analyze customer sentiments towards the hotel industry using **BERT models**, storing sentiment labels, scores, and embeddings in **LanceDB**, enabling queries on customer opinions and potential areas for improvement 💬 | [][sentiment_analysis_github] <br>[][sentiment_analysis_colab] <br>[][sentiment_analysis_ghost] |
|
||||||
| **Vector Arithmetic with LanceDB ⚖️** | Unlock powerful semantic search capabilities by performing vector arithmetic on embeddings, enabling complex relationships and nuances in data to be captured, and simplifying the process of retrieving semantically similar results 📊 | [][arithmetic_github] <br>[][arithmetic_colab] <br>[][arithmetic_ghost] |
|
| **Vector Arithmetic with LanceDB ⚖️** | Perform **vector arithmetic** on embeddings, enabling complex relationships and nuances in data to be captured, and simplifying the process of retrieving semantically similar results 📊 | [][arithmetic_github] <br>[][arithmetic_colab] <br>[][arithmetic_ghost] |
|
||||||
| **Imagebind Demo 🖼️** | Explore the multi-modal capabilities of Imagebind through a Gradio app, leveraging LanceDB API for seamless image search and retrieval experiences 📸 | [][imagebind_github] <br> [][imagebind_huggingface] |
|
| **Imagebind Demo 🖼️** | Explore the multi-modal capabilities of **Imagebind** through a Gradio app, use **LanceDB API** for seamless image search and retrieval experiences 📸 | [][imagebind_github] <br> [][imagebind_huggingface] |
|
||||||
| **Search Engine using SAM & CLIP 🔍** | Build a search engine within an image using SAM and CLIP models, enabling object-level search and retrieval, with LanceDB indexing and search capabilities to find the closest match between image embeddings and user queries 📸 | [][swi_github] <br>[][swi_colab] <br>[][swi_ghost] |
|
| **Search Engine using SAM & CLIP 🔍** | Build a search engine within an image using **SAM** and **CLIP** models, enabling object-level search and retrieval, with LanceDB indexing and search capabilities to find the closest match between image embeddings and user queries 📸 | [][swi_github] <br>[][swi_colab] <br>[][swi_ghost] |
|
||||||
| **Zero Shot Object Localization and Detection with CLIP 🔎** | Perform object detection on images using OpenAI's CLIP, enabling zero-shot localization and detection of objects, with capabilities to split images into patches, parse with CLIP, and plot bounding boxes 📊 | [][zsod_github] <br>[][zsod_colab] |
|
| **Zero Shot Object Localization and Detection with CLIP 🔎** | Perform object detection on images using **OpenAI's CLIP**, enabling zero-shot localization and detection of objects, with capabilities to split images into patches, parse with CLIP, and plot bounding boxes 📊 | [][zsod_github] <br>[][zsod_colab] |
|
||||||
| **Accelerate Vector Search with OpenVINO 🚀** | Boost vector search applications using OpenVINO, achieving significant speedups with CLIP for text-to-image and image-to-image searching, through PyTorch model optimization, FP16 and INT8 format conversion, and quantization with OpenVINO NNCF 📈 | [][openvino_github] <br>[][openvino_colab] <br>[][openvino_ghost] |
|
| **Accelerate Vector Search with OpenVINO 🚀** | Boost vector search applications using **OpenVINO**, achieving significant speedups with **CLIP** for text-to-image and image-to-image searching, through PyTorch model optimization, FP16 and INT8 format conversion, and quantization with **OpenVINO NNCF** 📈 | [][openvino_github] <br>[][openvino_colab] <br>[][openvino_ghost] |
|
||||||
| **Zero-Shot Image Classification with CLIP and LanceDB 📸** | Achieve zero-shot image classification using CLIP and LanceDB, enabling models to classify images without prior training on specific use cases, unlocking flexible and adaptable image classification capabilities 🔓 | [][zsic_github] <br>[][zsic_colab] <br>[][zsic_ghost] |
|
| **Zero-Shot Image Classification with CLIP and LanceDB 📸** | Achieve zero-shot image classification using **CLIP** and **LanceDB**, enabling models to classify images without prior training on specific use cases, unlocking flexible and adaptable image classification capabilities 🔓 | [][zsic_github] <br>[][zsic_colab] <br>[][zsic_ghost] |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@@ -37,16 +37,16 @@ LanceDB implements vector search algorithms for efficient document retrieval and
|
|||||||
[NER_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/tutorials/NER-powered-Semantic-Search/NER_powered_Semantic_Search_with_LanceDB.ipynb
|
[NER_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/tutorials/NER-powered-Semantic-Search/NER_powered_Semantic_Search_with_LanceDB.ipynb
|
||||||
[NER_ghost]: https://blog.lancedb.com/ner-powered-semantic-search-using-lancedb-51051dc3e493
|
[NER_ghost]: https://blog.lancedb.com/ner-powered-semantic-search-using-lancedb-51051dc3e493
|
||||||
|
|
||||||
[audio_search_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/audio_search
|
[audio_search_github]: https://github.com/lancedb/vectordb-recipes/tree/main/examples/archived_examples/audio_search
|
||||||
[audio_search_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/audio_search/main.ipynb
|
[audio_search_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/archived_examples/audio_search/main.ipynb
|
||||||
[audio_search_python]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/audio_search/main.py
|
[audio_search_python]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/archived_examples/audio_search/main.py
|
||||||
|
|
||||||
[mls_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/multi-lingual-wiki-qa
|
[mls_github]: https://github.com/lancedb/vectordb-recipes/tree/main/examples/archived_examples/multi-lingual-wiki-qa
|
||||||
[mls_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/multi-lingual-wiki-qa/main.ipynb
|
[mls_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/archived_examples/multi-lingual-wiki-qa/main.ipynb
|
||||||
[mls_python]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/multi-lingual-wiki-qa/main.py
|
[mls_python]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/archived_examples/multi-lingual-wiki-qa/main.py
|
||||||
|
|
||||||
[fr_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/facial_recognition
|
[fr_github]: https://github.com/lancedb/vectordb-recipes/tree/main/examples/archived_examples/facial_recognition
|
||||||
[fr_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/facial_recognition/main.ipynb
|
[fr_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/archived_examples/facial_recognition/main.ipynb
|
||||||
|
|
||||||
[sentiment_analysis_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/Sentiment-Analysis-Analyse-Hotel-Reviews
|
[sentiment_analysis_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/Sentiment-Analysis-Analyse-Hotel-Reviews
|
||||||
[sentiment_analysis_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/Sentiment-Analysis-Analyse-Hotel-Reviews/Sentiment_Analysis_using_LanceDB.ipynb
|
[sentiment_analysis_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/Sentiment-Analysis-Analyse-Hotel-Reviews/Sentiment_Analysis_using_LanceDB.ipynb
|
||||||
@@ -70,8 +70,8 @@ LanceDB implements vector search algorithms for efficient document retrieval and
|
|||||||
[openvino_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/Accelerate-Vector-Search-Applications-Using-OpenVINO/clip_text_image_search.ipynb
|
[openvino_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/Accelerate-Vector-Search-Applications-Using-OpenVINO/clip_text_image_search.ipynb
|
||||||
[openvino_ghost]: https://blog.lancedb.com/accelerate-vector-search-applications-using-openvino-lancedb/
|
[openvino_ghost]: https://blog.lancedb.com/accelerate-vector-search-applications-using-openvino-lancedb/
|
||||||
|
|
||||||
[zsic_github]: https://github.com/lancedb/vectordb-recipes/blob/main/examples/zero-shot-image-classification
|
[zsic_github]: https://github.com/lancedb/vectordb-recipes/tree/main/examples/archived_examples/zero-shot-image-classification
|
||||||
[zsic_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/zero-shot-image-classification/main.ipynb
|
[zsic_colab]: https://colab.research.google.com/github/lancedb/vectordb-recipes/blob/main/examples/archived_examples/zero-shot-image-classification/main.ipynb
|
||||||
[zsic_ghost]: https://blog.lancedb.com/zero-shot-image-classification-with-vector-search/
|
[zsic_ghost]: https://blog.lancedb.com/zero-shot-image-classification-with-vector-search/
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -25,8 +25,8 @@ s3://eto-public/datasets/sift/vec_data.lance
|
|||||||
Then, we can write a quick Python script to populate our LanceDB Table:
|
Then, we can write a quick Python script to populate our LanceDB Table:
|
||||||
|
|
||||||
```python
|
```python
|
||||||
import pylance
|
import lance
|
||||||
sift_dataset = pylance.dataset("/path/to/local/vec_data.lance")
|
sift_dataset = lance.dataset("/path/to/local/vec_data.lance")
|
||||||
df = sift_dataset.to_table().to_pandas()
|
df = sift_dataset.to_table().to_pandas()
|
||||||
|
|
||||||
import lancedb
|
import lancedb
|
||||||
|
|||||||
1
docs/src/extra_js/reo.js
Normal file
1
docs/src/extra_js/reo.js
Normal file
@@ -0,0 +1 @@
|
|||||||
|
!function(){var e,t,n;e="9627b71b382d201",t=function(){Reo.init({clientID:"9627b71b382d201"})},(n=document.createElement("script")).src="https://static.reo.dev/"+e+"/reo.js",n.defer=!0,n.onload=t,document.head.appendChild(n)}();
|
||||||
266
docs/src/fts.md
266
docs/src/fts.md
@@ -1,48 +1,28 @@
|
|||||||
# Full-text search
|
# Full-text search (Native FTS)
|
||||||
|
|
||||||
LanceDB provides support for full-text search via Lance (before via [Tantivy](https://github.com/quickwit-oss/tantivy) (Python only)), allowing you to incorporate keyword-based search (based on BM25) in your retrieval solutions.
|
LanceDB provides support for full-text search via Lance, allowing you to incorporate keyword-based search (based on BM25) in your retrieval solutions.
|
||||||
|
|
||||||
Currently, the Lance full text search is missing some features that are in the Tantivy full text search. This includes phrase queries, re-ranking, and customizing the tokenizer. Thus, in Python, Tantivy is still the default way to do full text search and many of the instructions below apply just to Tantivy-based indices.
|
|
||||||
|
|
||||||
|
|
||||||
## Installation (Only for Tantivy-based FTS)
|
|
||||||
|
|
||||||
!!! note
|
!!! note
|
||||||
No need to install the tantivy dependency if using native FTS
|
The Python SDK uses tantivy-based FTS by default, need to pass `use_tantivy=False` to use native FTS.
|
||||||
|
|
||||||
To use full-text search, install the dependency [`tantivy-py`](https://github.com/quickwit-oss/tantivy-py):
|
|
||||||
|
|
||||||
```sh
|
|
||||||
# Say you want to use tantivy==0.20.1
|
|
||||||
pip install tantivy==0.20.1
|
|
||||||
```
|
|
||||||
|
|
||||||
## Example
|
## Example
|
||||||
|
|
||||||
Consider that we have a LanceDB table named `my_table`, whose string column `text` we want to index and query via keyword search, the FTS index must be created before you can search via keywords.
|
Consider that we have a LanceDB table named `my_table`, whose string column `text` we want to index and query via keyword search, the FTS index must be created before you can search via keywords.
|
||||||
|
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
```python
|
```python
|
||||||
import lancedb
|
--8<-- "python/python/tests/docs/test_search.py:import-lancedb"
|
||||||
|
--8<-- "python/python/tests/docs/test_search.py:import-lancedb-fts"
|
||||||
|
--8<-- "python/python/tests/docs/test_search.py:basic_fts"
|
||||||
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
uri = "data/sample-lancedb"
|
```python
|
||||||
db = lancedb.connect(uri)
|
--8<-- "python/python/tests/docs/test_search.py:import-lancedb"
|
||||||
|
--8<-- "python/python/tests/docs/test_search.py:import-lancedb-fts"
|
||||||
table = db.create_table(
|
--8<-- "python/python/tests/docs/test_search.py:basic_fts_async"
|
||||||
"my_table",
|
|
||||||
data=[
|
|
||||||
{"vector": [3.1, 4.1], "text": "Frodo was a happy puppy"},
|
|
||||||
{"vector": [5.9, 26.5], "text": "There are several kittens playing"},
|
|
||||||
],
|
|
||||||
)
|
|
||||||
|
|
||||||
# passing `use_tantivy=False` to use lance FTS index
|
|
||||||
# `use_tantivy=True` by default
|
|
||||||
table.create_fts_index("text")
|
|
||||||
table.search("puppy").limit(10).select(["text"]).to_list()
|
|
||||||
# [{'text': 'Frodo was a happy puppy', '_score': 0.6931471824645996}]
|
|
||||||
# ...
|
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "TypeScript"
|
=== "TypeScript"
|
||||||
@@ -62,7 +42,7 @@ Consider that we have a LanceDB table named `my_table`, whose string column `tex
|
|||||||
});
|
});
|
||||||
|
|
||||||
await tbl
|
await tbl
|
||||||
.search("puppy")
|
.search("puppy", "fts")
|
||||||
.select(["text"])
|
.select(["text"])
|
||||||
.limit(10)
|
.limit(10)
|
||||||
.toArray();
|
.toArray();
|
||||||
@@ -93,56 +73,102 @@ Consider that we have a LanceDB table named `my_table`, whose string column `tex
|
|||||||
```
|
```
|
||||||
|
|
||||||
It would search on all indexed columns by default, so it's useful when there are multiple indexed columns.
|
It would search on all indexed columns by default, so it's useful when there are multiple indexed columns.
|
||||||
For now, this is supported in tantivy way only.
|
|
||||||
|
|
||||||
Passing `fts_columns="text"` if you want to specify the columns to search, but it's not available for Tantivy-based full text search.
|
Passing `fts_columns="text"` if you want to specify the columns to search.
|
||||||
|
|
||||||
!!! note
|
!!! note
|
||||||
LanceDB automatically searches on the existing FTS index if the input to the search is of type `str`. If you provide a vector as input, LanceDB will search the ANN index instead.
|
LanceDB automatically searches on the existing FTS index if the input to the search is of type `str`. If you provide a vector as input, LanceDB will search the ANN index instead.
|
||||||
|
|
||||||
## Tokenization
|
## Tokenization
|
||||||
By default the text is tokenized by splitting on punctuation and whitespaces and then removing tokens that are longer than 40 chars. For more language specific tokenization then provide the argument tokenizer_name with the 2 letter language code followed by "_stem". So for english it would be "en_stem".
|
By default the text is tokenized by splitting on punctuation and whitespaces, and would filter out words that are with length greater than 40, and lowercase all words.
|
||||||
|
|
||||||
For now, only the Tantivy-based FTS index supports to specify the tokenizer, so it's only available in Python with `use_tantivy=True`.
|
Stemming is useful for improving search results by reducing words to their root form, e.g. "running" to "run". LanceDB supports stemming for multiple languages, you can specify the tokenizer name to enable stemming by the pattern `tokenizer_name="{language_code}_stem"`, e.g. `en_stem` for English.
|
||||||
|
|
||||||
=== "use_tantivy=True"
|
For example, to enable stemming for English:
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
```python
|
```python
|
||||||
table.create_fts_index("text", use_tantivy=True, tokenizer_name="en_stem")
|
--8<-- "python/python/tests/docs/test_search.py:fts_config_stem"
|
||||||
```
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
=== "use_tantivy=False"
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_search.py:fts_config_stem_async"
|
||||||
[**Not supported yet**](https://github.com/lancedb/lance/issues/1195)
|
```
|
||||||
|
|
||||||
the following [languages](https://docs.rs/tantivy/latest/tantivy/tokenizer/enum.Language.html) are currently supported.
|
the following [languages](https://docs.rs/tantivy/latest/tantivy/tokenizer/enum.Language.html) are currently supported.
|
||||||
|
|
||||||
## Index multiple columns
|
The tokenizer is customizable, you can specify how the tokenizer splits the text, and how it filters out words, etc.
|
||||||
|
|
||||||
If you have multiple string columns to index, there's no need to combine them manually -- simply pass them all as a list to `create_fts_index`:
|
For example, for language with accents, you can specify the tokenizer to use `ascii_folding` to remove accents, e.g. 'é' to 'e':
|
||||||
|
=== "Sync API"
|
||||||
=== "use_tantivy=True"
|
|
||||||
|
|
||||||
```python
|
```python
|
||||||
table.create_fts_index(["text1", "text2"])
|
--8<-- "python/python/tests/docs/test_search.py:fts_config_folding"
|
||||||
```
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
=== "use_tantivy=False"
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_search.py:fts_config_folding_async"
|
||||||
[**Not supported yet**](https://github.com/lancedb/lance/issues/1195)
|
```
|
||||||
|
|
||||||
Note that the search API call does not change - you can search over all indexed columns at once.
|
|
||||||
|
|
||||||
## Filtering
|
## Filtering
|
||||||
|
|
||||||
Currently the LanceDB full text search feature supports *post-filtering*, meaning filters are
|
LanceDB full text search supports to filter the search results by a condition, both pre-filtering and post-filtering are supported.
|
||||||
applied on top of the full text search results. This can be invoked via the familiar
|
|
||||||
`where` syntax:
|
|
||||||
|
|
||||||
|
This can be invoked via the familiar `where` syntax.
|
||||||
|
|
||||||
|
With pre-filtering:
|
||||||
=== "Python"
|
=== "Python"
|
||||||
|
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
```python
|
```python
|
||||||
table.search("puppy").limit(10).where("meta='foo'").to_list()
|
--8<-- "python/python/tests/docs/test_search.py:fts_prefiltering"
|
||||||
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_search.py:fts_prefiltering_async"
|
||||||
|
```
|
||||||
|
|
||||||
|
=== "TypeScript"
|
||||||
|
|
||||||
|
```typescript
|
||||||
|
await tbl
|
||||||
|
.search("puppy")
|
||||||
|
.select(["id", "doc"])
|
||||||
|
.limit(10)
|
||||||
|
.where("meta='foo'")
|
||||||
|
.prefilter(true)
|
||||||
|
.toArray();
|
||||||
|
```
|
||||||
|
|
||||||
|
=== "Rust"
|
||||||
|
|
||||||
|
```rust
|
||||||
|
table
|
||||||
|
.query()
|
||||||
|
.full_text_search(FullTextSearchQuery::new("puppy".to_owned()))
|
||||||
|
.select(lancedb::query::Select::Columns(vec!["doc".to_owned()]))
|
||||||
|
.limit(10)
|
||||||
|
.only_if("meta='foo'")
|
||||||
|
.execute()
|
||||||
|
.await?;
|
||||||
|
```
|
||||||
|
|
||||||
|
With post-filtering:
|
||||||
|
=== "Python"
|
||||||
|
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_search.py:fts_postfiltering"
|
||||||
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_search.py:fts_postfiltering_async"
|
||||||
```
|
```
|
||||||
|
|
||||||
=== "TypeScript"
|
=== "TypeScript"
|
||||||
@@ -153,6 +179,7 @@ applied on top of the full text search results. This can be invoked via the fami
|
|||||||
.select(["id", "doc"])
|
.select(["id", "doc"])
|
||||||
.limit(10)
|
.limit(10)
|
||||||
.where("meta='foo'")
|
.where("meta='foo'")
|
||||||
|
.prefilter(false)
|
||||||
.toArray();
|
.toArray();
|
||||||
```
|
```
|
||||||
|
|
||||||
@@ -163,104 +190,69 @@ applied on top of the full text search results. This can be invoked via the fami
|
|||||||
.query()
|
.query()
|
||||||
.full_text_search(FullTextSearchQuery::new(words[0].to_owned()))
|
.full_text_search(FullTextSearchQuery::new(words[0].to_owned()))
|
||||||
.select(lancedb::query::Select::Columns(vec!["doc".to_owned()]))
|
.select(lancedb::query::Select::Columns(vec!["doc".to_owned()]))
|
||||||
|
.postfilter()
|
||||||
.limit(10)
|
.limit(10)
|
||||||
.only_if("meta='foo'")
|
.only_if("meta='foo'")
|
||||||
.execute()
|
.execute()
|
||||||
.await?;
|
.await?;
|
||||||
```
|
```
|
||||||
|
|
||||||
## Sorting
|
|
||||||
|
|
||||||
!!! warning "Warn"
|
|
||||||
Sorting is available for only Tantivy-based FTS
|
|
||||||
|
|
||||||
You can pre-sort the documents by specifying `ordering_field_names` when
|
|
||||||
creating the full-text search index. Once pre-sorted, you can then specify
|
|
||||||
`ordering_field_name` while searching to return results sorted by the given
|
|
||||||
field. For example,
|
|
||||||
|
|
||||||
```python
|
|
||||||
table.create_fts_index(["text_field"], use_tantivy=True, ordering_field_names=["sort_by_field"])
|
|
||||||
|
|
||||||
(table.search("terms", ordering_field_name="sort_by_field")
|
|
||||||
.limit(20)
|
|
||||||
.to_list())
|
|
||||||
```
|
|
||||||
|
|
||||||
!!! note
|
|
||||||
If you wish to specify an ordering field at query time, you must also
|
|
||||||
have specified it during indexing time. Otherwise at query time, an
|
|
||||||
error will be raised that looks like `ValueError: The field does not exist: xxx`
|
|
||||||
|
|
||||||
!!! note
|
|
||||||
The fields to sort on must be of typed unsigned integer, or else you will see
|
|
||||||
an error during indexing that looks like
|
|
||||||
`TypeError: argument 'value': 'float' object cannot be interpreted as an integer`.
|
|
||||||
|
|
||||||
!!! note
|
|
||||||
You can specify multiple fields for ordering at indexing time.
|
|
||||||
But at query time only one ordering field is supported.
|
|
||||||
|
|
||||||
|
|
||||||
## Phrase queries vs. terms queries
|
## Phrase queries vs. terms queries
|
||||||
|
|
||||||
!!! warning "Warn"
|
!!! warning "Warn"
|
||||||
Phrase queries are available for only Tantivy-based FTS
|
Lance-based FTS doesn't support queries using boolean operators `OR`, `AND`.
|
||||||
|
|
||||||
For full-text search you can specify either a **phrase** query like `"the old man and the sea"`,
|
For full-text search you can specify either a **phrase** query like `"the old man and the sea"`,
|
||||||
or a **terms** search query like `"(Old AND Man) AND Sea"`. For more details on the terms
|
or a **terms** search query like `old man sea`. For more details on the terms
|
||||||
query syntax, see Tantivy's [query parser rules](https://docs.rs/tantivy/latest/tantivy/query/struct.QueryParser.html).
|
query syntax, see Tantivy's [query parser rules](https://docs.rs/tantivy/latest/tantivy/query/struct.QueryParser.html).
|
||||||
|
|
||||||
!!! tip "Note"
|
To search for a phrase, the index must be created with `with_position=True`:
|
||||||
The query parser will raise an exception on queries that are ambiguous. For example, in the query `they could have been dogs OR cats`, `OR` is capitalized so it's considered a keyword query operator. But it's ambiguous how the left part should be treated. So if you submit this search query as is, you'll get `Syntax Error: they could have been dogs OR cats`.
|
=== "Sync API"
|
||||||
|
|
||||||
```py
|
```python
|
||||||
# This raises a syntax error
|
--8<-- "python/python/tests/docs/test_search.py:fts_with_position"
|
||||||
table.search("they could have been dogs OR cats")
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_search.py:fts_with_position_async"
|
||||||
|
```
|
||||||
|
This will allow you to search for phrases, but it will also significantly increase the index size and indexing time.
|
||||||
|
|
||||||
|
|
||||||
|
## Incremental indexing
|
||||||
|
|
||||||
|
LanceDB supports incremental indexing, which means you can add new records to the table without reindexing the entire table.
|
||||||
|
|
||||||
|
This can make the query more efficient, especially when the table is large and the new records are relatively small.
|
||||||
|
|
||||||
|
=== "Python"
|
||||||
|
|
||||||
|
=== "Sync API"
|
||||||
|
|
||||||
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_search.py:fts_incremental_index"
|
||||||
|
```
|
||||||
|
=== "Async API"
|
||||||
|
|
||||||
|
```python
|
||||||
|
--8<-- "python/python/tests/docs/test_search.py:fts_incremental_index_async"
|
||||||
```
|
```
|
||||||
|
|
||||||
On the other hand, lowercasing `OR` to `or` will work, because there are no capitalized logical operators and
|
=== "TypeScript"
|
||||||
the query is treated as a phrase query.
|
|
||||||
|
|
||||||
```py
|
```typescript
|
||||||
# This works!
|
await tbl.add([{ vector: [3.1, 4.1], text: "Frodo was a happy puppy" }]);
|
||||||
table.search("they could have been dogs or cats")
|
await tbl.optimize();
|
||||||
```
|
```
|
||||||
|
|
||||||
It can be cumbersome to have to remember what will cause a syntax error depending on the type of
|
=== "Rust"
|
||||||
query you want to perform. To make this simpler, when you want to perform a phrase query, you can
|
|
||||||
enforce it in one of two ways:
|
|
||||||
|
|
||||||
1. Place the double-quoted query inside single quotes. For example, `table.search('"they could have been dogs OR cats"')` is treated as
|
```rust
|
||||||
a phrase query.
|
let more_data: Box<dyn RecordBatchReader + Send> = create_some_records()?;
|
||||||
1. Explicitly declare the `phrase_query()` method. This is useful when you have a phrase query that
|
tbl.add(more_data).execute().await?;
|
||||||
itself contains double quotes. For example, `table.search('the cats OR dogs were not really "pets" at all').phrase_query()`
|
tbl.optimize(OptimizeAction::All).execute().await?;
|
||||||
is treated as a phrase query.
|
```
|
||||||
|
!!! note
|
||||||
|
|
||||||
In general, a query that's declared as a phrase query will be wrapped in double quotes during parsing, with nested
|
New data added after creating the FTS index will appear in search results while incremental index is still progress, but with increased latency due to a flat search on the unindexed portion. LanceDB Cloud automates this merging process, minimizing the impact on search speed.
|
||||||
double quotes replaced by single quotes.
|
|
||||||
|
|
||||||
|
|
||||||
## Configurations (Only for Tantivy-based FTS)
|
|
||||||
|
|
||||||
By default, LanceDB configures a 1GB heap size limit for creating the index. You can
|
|
||||||
reduce this if running on a smaller node, or increase this for faster performance while
|
|
||||||
indexing a larger corpus.
|
|
||||||
|
|
||||||
```python
|
|
||||||
# configure a 512MB heap size
|
|
||||||
heap = 1024 * 1024 * 512
|
|
||||||
table.create_fts_index(["text1", "text2"], writer_heap_size=heap, replace=True)
|
|
||||||
```
|
|
||||||
|
|
||||||
## Current limitations
|
|
||||||
|
|
||||||
For that Tantivy-based FTS:
|
|
||||||
|
|
||||||
1. Currently we do not yet support incremental writes.
|
|
||||||
If you add data after FTS index creation, it won't be reflected
|
|
||||||
in search results until you do a full reindex.
|
|
||||||
|
|
||||||
2. We currently only support local filesystem paths for the FTS index.
|
|
||||||
This is a tantivy limitation. We've implemented an object store plugin
|
|
||||||
but there's no way in tantivy-py to specify to use it.
|
|
||||||
160
docs/src/fts_tantivy.md
Normal file
160
docs/src/fts_tantivy.md
Normal file
@@ -0,0 +1,160 @@
|
|||||||
|
# Full-text search (Tantivy-based FTS)
|
||||||
|
|
||||||
|
LanceDB also provides support for full-text search via [Tantivy](https://github.com/quickwit-oss/tantivy), allowing you to incorporate keyword-based search (based on BM25) in your retrieval solutions.
|
||||||
|
|
||||||
|
The tantivy-based FTS is only available in Python synchronous APIs and does not support building indexes on object storage or incremental indexing. If you need these features, try native FTS [native FTS](fts.md).
|
||||||
|
|
||||||
|
## Installation
|
||||||
|
|
||||||
|
To use full-text search, install the dependency [`tantivy-py`](https://github.com/quickwit-oss/tantivy-py):
|
||||||
|
|
||||||
|
```sh
|
||||||
|
# Say you want to use tantivy==0.20.1
|
||||||
|
pip install tantivy==0.20.1
|
||||||
|
```
|
||||||
|
|
||||||
|
## Example
|
||||||
|
|
||||||
|
Consider that we have a LanceDB table named `my_table`, whose string column `content` we want to index and query via keyword search, the FTS index must be created before you can search via keywords.
|
||||||
|
|
||||||
|
```python
|
||||||
|
import lancedb
|
||||||
|
|
||||||
|
uri = "data/sample-lancedb"
|
||||||
|
db = lancedb.connect(uri)
|
||||||
|
|
||||||
|
table = db.create_table(
|
||||||
|
"my_table",
|
||||||
|
data=[
|
||||||
|
{"id": 1, "vector": [3.1, 4.1], "title": "happy puppy", "content": "Frodo was a happy puppy", "meta": "foo"},
|
||||||
|
{"id": 2, "vector": [5.9, 26.5], "title": "playing kittens", "content": "There are several kittens playing around the puppy", "meta": "bar"},
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
|
# passing `use_tantivy=False` to use lance FTS index
|
||||||
|
# `use_tantivy=True` by default
|
||||||
|
table.create_fts_index("content", use_tantivy=True)
|
||||||
|
table.search("puppy").limit(10).select(["content"]).to_list()
|
||||||
|
# [{'text': 'Frodo was a happy puppy', '_score': 0.6931471824645996}]
|
||||||
|
# ...
|
||||||
|
```
|
||||||
|
|
||||||
|
It would search on all indexed columns by default, so it's useful when there are multiple indexed columns.
|
||||||
|
|
||||||
|
!!! note
|
||||||
|
LanceDB automatically searches on the existing FTS index if the input to the search is of type `str`. If you provide a vector as input, LanceDB will search the ANN index instead.
|
||||||
|
|
||||||
|
## Tokenization
|
||||||
|
By default the text is tokenized by splitting on punctuation and whitespaces and then removing tokens that are longer than 40 chars. For more language specific tokenization then provide the argument tokenizer_name with the 2 letter language code followed by "_stem". So for english it would be "en_stem".
|
||||||
|
|
||||||
|
```python
|
||||||
|
table.create_fts_index("content", use_tantivy=True, tokenizer_name="en_stem", replace=True)
|
||||||
|
```
|
||||||
|
|
||||||
|
the following [languages](https://docs.rs/tantivy/latest/tantivy/tokenizer/enum.Language.html) are currently supported.
|
||||||
|
|
||||||
|
## Index multiple columns
|
||||||
|
|
||||||
|
If you have multiple string columns to index, there's no need to combine them manually -- simply pass them all as a list to `create_fts_index`:
|
||||||
|
|
||||||
|
```python
|
||||||
|
table.create_fts_index(["title", "content"], use_tantivy=True, replace=True)
|
||||||
|
```
|
||||||
|
|
||||||
|
Note that the search API call does not change - you can search over all indexed columns at once.
|
||||||
|
|
||||||
|
## Filtering
|
||||||
|
|
||||||
|
Currently the LanceDB full text search feature supports *post-filtering*, meaning filters are
|
||||||
|
applied on top of the full text search results (see [native FTS](fts.md) if you need pre-filtering). This can be invoked via the familiar
|
||||||
|
`where` syntax:
|
||||||
|
|
||||||
|
```python
|
||||||
|
table.search("puppy").limit(10).where("meta='foo'").to_list()
|
||||||
|
```
|
||||||
|
|
||||||
|
## Sorting
|
||||||
|
|
||||||
|
You can pre-sort the documents by specifying `ordering_field_names` when
|
||||||
|
creating the full-text search index. Once pre-sorted, you can then specify
|
||||||
|
`ordering_field_name` while searching to return results sorted by the given
|
||||||
|
field. For example,
|
||||||
|
|
||||||
|
```python
|
||||||
|
table.create_fts_index(["content"], use_tantivy=True, ordering_field_names=["id"], replace=True)
|
||||||
|
|
||||||
|
(table.search("puppy", ordering_field_name="id")
|
||||||
|
.limit(20)
|
||||||
|
.to_list())
|
||||||
|
```
|
||||||
|
|
||||||
|
!!! note
|
||||||
|
If you wish to specify an ordering field at query time, you must also
|
||||||
|
have specified it during indexing time. Otherwise at query time, an
|
||||||
|
error will be raised that looks like `ValueError: The field does not exist: xxx`
|
||||||
|
|
||||||
|
!!! note
|
||||||
|
The fields to sort on must be of typed unsigned integer, or else you will see
|
||||||
|
an error during indexing that looks like
|
||||||
|
`TypeError: argument 'value': 'float' object cannot be interpreted as an integer`.
|
||||||
|
|
||||||
|
!!! note
|
||||||
|
You can specify multiple fields for ordering at indexing time.
|
||||||
|
But at query time only one ordering field is supported.
|
||||||
|
|
||||||
|
|
||||||
|
## Phrase queries vs. terms queries
|
||||||
|
|
||||||
|
For full-text search you can specify either a **phrase** query like `"the old man and the sea"`,
|
||||||
|
or a **terms** search query like `"(Old AND Man) AND Sea"`. For more details on the terms
|
||||||
|
query syntax, see Tantivy's [query parser rules](https://docs.rs/tantivy/latest/tantivy/query/struct.QueryParser.html).
|
||||||
|
|
||||||
|
!!! tip "Note"
|
||||||
|
The query parser will raise an exception on queries that are ambiguous. For example, in the query `they could have been dogs OR cats`, `OR` is capitalized so it's considered a keyword query operator. But it's ambiguous how the left part should be treated. So if you submit this search query as is, you'll get `Syntax Error: they could have been dogs OR cats`.
|
||||||
|
|
||||||
|
```py
|
||||||
|
# This raises a syntax error
|
||||||
|
table.search("they could have been dogs OR cats")
|
||||||
|
```
|
||||||
|
|
||||||
|
On the other hand, lowercasing `OR` to `or` will work, because there are no capitalized logical operators and
|
||||||
|
the query is treated as a phrase query.
|
||||||
|
|
||||||
|
```py
|
||||||
|
# This works!
|
||||||
|
table.search("they could have been dogs or cats")
|
||||||
|
```
|
||||||
|
|
||||||
|
It can be cumbersome to have to remember what will cause a syntax error depending on the type of
|
||||||
|
query you want to perform. To make this simpler, when you want to perform a phrase query, you can
|
||||||
|
enforce it in one of two ways:
|
||||||
|
|
||||||
|
1. Place the double-quoted query inside single quotes. For example, `table.search('"they could have been dogs OR cats"')` is treated as
|
||||||
|
a phrase query.
|
||||||
|
1. Explicitly declare the `phrase_query()` method. This is useful when you have a phrase query that
|
||||||
|
itself contains double quotes. For example, `table.search('the cats OR dogs were not really "pets" at all').phrase_query()`
|
||||||
|
is treated as a phrase query.
|
||||||
|
|
||||||
|
In general, a query that's declared as a phrase query will be wrapped in double quotes during parsing, with nested
|
||||||
|
double quotes replaced by single quotes.
|
||||||
|
|
||||||
|
|
||||||
|
## Configurations
|
||||||
|
|
||||||
|
By default, LanceDB configures a 1GB heap size limit for creating the index. You can
|
||||||
|
reduce this if running on a smaller node, or increase this for faster performance while
|
||||||
|
indexing a larger corpus.
|
||||||
|
|
||||||
|
```python
|
||||||
|
# configure a 512MB heap size
|
||||||
|
heap = 1024 * 1024 * 512
|
||||||
|
table.create_fts_index(["title", "content"], use_tantivy=True, writer_heap_size=heap, replace=True)
|
||||||
|
```
|
||||||
|
|
||||||
|
## Current limitations
|
||||||
|
|
||||||
|
1. New data added after creating the FTS index will appear in search results, but with increased latency due to a flat search on the unindexed portion. Re-indexing with `create_fts_index` will reduce latency. LanceDB Cloud automates this merging process, minimizing the impact on search speed.
|
||||||
|
|
||||||
|
2. We currently only support local filesystem paths for the FTS index.
|
||||||
|
This is a tantivy limitation. We've implemented an object store plugin
|
||||||
|
but there's no way in tantivy-py to specify to use it.
|
||||||
85
docs/src/guides/multi-vector.md
Normal file
85
docs/src/guides/multi-vector.md
Normal file
@@ -0,0 +1,85 @@
|
|||||||
|
# Late interaction & MultiVector embedding type
|
||||||
|
Late interaction is a technique used in retrieval that calculates the relevance of a query to a document by comparing their multi-vector representations. The key difference between late interaction and other popular methods:
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
|
||||||
|
[ Illustration from https://jina.ai/news/what-is-colbert-and-late-interaction-and-why-they-matter-in-search/]
|
||||||
|
|
||||||
|
<b>No interaction:</b> Refers to independently embedding the query and document, that are compared to calcualte similarity without any interaction between them. This is typically used in vector search operations.
|
||||||
|
|
||||||
|
<b>Partial interaction</b> Refers to a specific approach where the similarity computation happens primarily between query vectors and document vectors, without extensive interaction between individual components of each. An example of this is dual-encoder models like BERT.
|
||||||
|
|
||||||
|
<b>Early full interaction</b> Refers to techniques like cross-encoders that process query and docs in pairs with full interaction across various stages of encoding. This is a powerful, but relatively slower technique. Because it requires processing query and docs in pairs, doc embeddings can't be pre-computed for fast retrieval. This is why cross encoders are typically used as reranking models combined with vector search. Learn more about [LanceDB Reranking support](https://lancedb.github.io/lancedb/reranking/).
|
||||||
|
|
||||||
|
<b>Late interaction</b> Late interaction is a technique that calculates the doc and query similarity independently and then the interaction or evaluation happens during the retrieval process. This is typically used in retrieval models like ColBERT. Unlike early interaction, It allows speeding up the retrieval process without compromising the depth of semantic analysis.
|
||||||
|
|
||||||
|
## Internals of ColBERT
|
||||||
|
Let's take a look at the steps involved in performing late interaction based retrieval using ColBERT:
|
||||||
|
|
||||||
|
• ColBERT employs BERT-based encoders for both queries `(fQ)` and documents `(fD)`
|
||||||
|
• A single BERT model is shared between query and document encoders and special tokens distinguish input types: `[Q]` for queries and `[D]` for documents
|
||||||
|
|
||||||
|
**Query Encoder (fQ):**
|
||||||
|
• Query q is tokenized into WordPiece tokens: `q1, q2, ..., ql`. `[Q]` token is prepended right after BERT's `[CLS]` token
|
||||||
|
• If query length < Nq, it's padded with [MASK] tokens up to Nq.
|
||||||
|
• The padded sequence goes through BERT's transformer architecture
|
||||||
|
• Final embeddings are L2-normalized.
|
||||||
|
|
||||||
|
**Document Encoder (fD):**
|
||||||
|
• Document d is tokenized into tokens `d1, d2, ..., dm`. `[D]` token is prepended after `[CLS]` token
|
||||||
|
• Unlike queries, documents are NOT padded with `[MASK]` tokens
|
||||||
|
• Document tokens are processed through BERT and the same linear layer
|
||||||
|
|
||||||
|
**Late Interaction:**
|
||||||
|
• Late interaction estimates relevance score `S(q,d)` using embedding `Eq` and `Ed`. Late interaction happens after independent encoding
|
||||||
|
• For each query embedding, maximum similarity is computed against all document embeddings
|
||||||
|
• The similarity measure can be cosine similarity or squared L2 distance
|
||||||
|
|
||||||
|
**MaxSim Calculation:**
|
||||||
|
```
|
||||||
|
S(q,d) := Σ max(Eqi⋅EdjT)
|
||||||
|
i∈|Eq| j∈|Ed|
|
||||||
|
```
|
||||||
|
• This finds the best matching document embedding for each query embedding
|
||||||
|
• Captures relevance based on strongest local matches between contextual embeddings
|
||||||
|
|
||||||
|
## LanceDB MultiVector type
|
||||||
|
LanceDB supports multivector type, this is useful when you have multiple vectors for a single item (e.g. with ColBert and ColPali).
|
||||||
|
|
||||||
|
You can index on a column with multivector type and search on it, the query can be single vector or multiple vectors. For now, only cosine metric is supported for multivector search. The vector value type can be float16, float32 or float64. LanceDB integrateds [ConteXtualized Token Retriever(XTR)](https://arxiv.org/abs/2304.01982), which introduces a simple, yet novel, objective function that encourages the model to retrieve the most important document tokens first.
|
||||||
|
|
||||||
|
```python
|
||||||
|
import lancedb
|
||||||
|
import numpy as np
|
||||||
|
import pyarrow as pa
|
||||||
|
|
||||||
|
db = lancedb.connect("data/multivector_demo")
|
||||||
|
schema = pa.schema(
|
||||||
|
[
|
||||||
|
pa.field("id", pa.int64()),
|
||||||
|
# float16, float32, and float64 are supported
|
||||||
|
pa.field("vector", pa.list_(pa.list_(pa.float32(), 256))),
|
||||||
|
]
|
||||||
|
)
|
||||||
|
data = [
|
||||||
|
{
|
||||||
|
"id": i,
|
||||||
|
"vector": np.random.random(size=(2, 256)).tolist(),
|
||||||
|
}
|
||||||
|
for i in range(1024)
|
||||||
|
]
|
||||||
|
tbl = db.create_table("my_table", data=data, schema=schema)
|
||||||
|
|
||||||
|
# only cosine similarity is supported for multi-vectors
|
||||||
|
tbl.create_index(metric="cosine")
|
||||||
|
|
||||||
|
# query with single vector
|
||||||
|
query = np.random.random(256).astype(np.float16)
|
||||||
|
tbl.search(query).to_arrow()
|
||||||
|
|
||||||
|
# query with multiple vectors
|
||||||
|
query = np.random.random(size=(2, 256))
|
||||||
|
tbl.search(query).to_arrow()
|
||||||
|
```
|
||||||
|
Find more about vector search in LanceDB [here](https://lancedb.github.io/lancedb/search/#multivector-type).
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user