@delali/narsil 0.3.1-next.73 → 0.3.1-next.74
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/{capabilities-CYeAWYxQ.d.ts → capabilities-CAMzmkAg.d.ts} +2 -2
- package/dist/{chunk-PU43WDY4.mjs → chunk-2E35UM3G.mjs} +1 -13
- package/dist/chunk-3DDUNI6N.mjs +48 -0
- package/dist/{chunk-THMSFOJO.mjs → chunk-BVBIQQKG.mjs} +576 -547
- package/dist/{chunk-ZULSEZCE.mjs → chunk-CTOSGRM7.mjs} +100 -19
- package/dist/{chunk-CVFAFZSN.mjs → chunk-CUZOX6CN.mjs} +23 -15
- package/dist/chunk-I23G3IZC.mjs +593 -0
- package/dist/chunk-KEHK2O6P.mjs +13 -0
- package/dist/{chunk-AFKH4ZK3.mjs → chunk-LX4JERXI.mjs} +1 -1
- package/dist/{chunk-LPBWCHBE.mjs → chunk-NOZUQM7J.mjs} +2 -2
- package/dist/{chunk-WVDFG45M.mjs → chunk-PO53FV4Q.mjs} +2 -1
- package/dist/chunk-T5ZWEWVD.mjs +658 -0
- package/dist/{chunk-2WLD2NFW.mjs → chunk-TVFLHM7M.mjs} +2 -2
- package/dist/{chunk-3YKWSO54.mjs → chunk-VKX2BPXD.mjs} +147 -647
- package/dist/{chunk-AK3BLGCI.mjs → chunk-Z3DQPOJQ.mjs} +15 -48
- package/dist/{client-D698E5ge.d.ts → client-C-I70ArH.d.ts} +3 -3
- package/dist/client.d.ts +8 -8
- package/dist/{config-Bv8etwUZ.d.ts → config-C6Iw5rCX.d.ts} +2 -2
- package/dist/{config-QXU3HMpq.d.ts → config-DJPH3IQB.d.ts} +2 -2
- package/dist/direct-executor-KFAGILJC.mjs +17 -0
- package/dist/distribution/coordinator/etcd.d.ts +2 -2
- package/dist/distribution/coordinator/in-memory.d.ts +2 -2
- package/dist/distribution.d.ts +8 -8
- package/dist/distribution.mjs +19 -9
- package/dist/{engine-D_OzKZ-6.d.ts → engine-B8eRPO7m.d.ts} +2 -2
- package/dist/index.browser.mjs +1200 -808
- package/dist/index.d.ts +27 -8
- package/dist/index.mjs +12 -8
- package/dist/{memory-B4jMmygw.d.ts → memory-DJEwWoCi.d.ts} +1205 -1180
- package/dist/partial-words-DTIHO8G4.d.ts +19 -0
- package/dist/partial-words.d.ts +22 -0
- package/dist/partial-words.mjs +11 -0
- package/dist/persistence/durability/checkpoint-worker.d.ts +33 -13
- package/dist/persistence/durability/checkpoint-worker.mjs +12 -8
- package/dist/react.d.ts +7 -7
- package/dist/{schema-BBOCdhND.d.ts → schema-DgyDijhf.d.ts} +6 -2
- package/dist/server.d.ts +10 -10
- package/dist/server.mjs +8 -6
- package/dist/{tasks-DpgCpW1x.d.ts → tasks-BVmAKYBh.d.ts} +1 -1
- package/dist/{thread-3LNVA43V.mjs → thread-W6DPKEF6.mjs} +8 -5
- package/dist/{types-BiUmaiWT.d.ts → types-ChrVy4lH.d.ts} +1 -1
- package/dist/vector/search-worker.d.ts +1 -1
- package/dist/vector/search-worker.mjs +3 -2
- package/dist/workers/entry.mjs +3 -3
- package/package.json +15 -7
- package/dist/direct-executor-5SRDFZAS.mjs +0 -13
|
@@ -0,0 +1,593 @@
|
|
|
1
|
+
import { isWordIndexedBase, flattenSchema } from './chunk-Z3DQPOJQ.mjs';
|
|
2
|
+
import { createNarsilError, ErrorCodes } from './chunk-I64LXKO3.mjs';
|
|
3
|
+
|
|
4
|
+
// src/core/tokenizer/constants.ts
|
|
5
|
+
var TOKEN_CACHE_SIZE_FLOOR = 5e4;
|
|
6
|
+
var TOKEN_CACHE_SIZE_CEILING = 2e6;
|
|
7
|
+
var TOKEN_CACHE_BYTES_PER_ENTRY = 200;
|
|
8
|
+
var NODE_TOKEN_CACHE_ENTRIES = 1e6;
|
|
9
|
+
var UNKNOWN_MEMORY_TOKEN_CACHE_ENTRIES = 2e5;
|
|
10
|
+
var CONSTRAINED_MEMORY_TOKEN_CACHE_FRACTION = 0.05;
|
|
11
|
+
var SMALL_DEVICE_MEMORY_GB = 1;
|
|
12
|
+
var MEDIUM_DEVICE_MEMORY_GB = 4;
|
|
13
|
+
var SMALL_DEVICE_TOKEN_CACHE_ENTRIES = 1e5;
|
|
14
|
+
var MEDIUM_DEVICE_TOKEN_CACHE_ENTRIES = 25e4;
|
|
15
|
+
var LARGE_DEVICE_TOKEN_CACHE_ENTRIES = 5e5;
|
|
16
|
+
var DEFAULT_MIN_TOKEN_LENGTH = 1;
|
|
17
|
+
|
|
18
|
+
// src/core/tokenizer/normalize.ts
|
|
19
|
+
var DOTTED_CAPITAL_I = "\u0130";
|
|
20
|
+
var COMBINING_DOT_ABOVE = "\u0307";
|
|
21
|
+
var ARMENIAN_MARKS = /[\u055B-\u055F]/;
|
|
22
|
+
var FOLDED_FORMS = /\u0307|[\u0130\u055B-\u055F\u2018\u2019\u02BC\u02BB]/g;
|
|
23
|
+
var WIDTH_FORMS = /[\uFF01-\uFF5E\uFF61-\uFF9F]/;
|
|
24
|
+
var WIDTH_FORMS_GLOBAL = /[\uFF01-\uFF5E\uFF61-\uFF9F]/g;
|
|
25
|
+
var FULLWIDTH_ASCII_END = 65374;
|
|
26
|
+
var FULLWIDTH_ASCII_OFFSET = 65248;
|
|
27
|
+
var HALFWIDTH_KANA_START = 65377;
|
|
28
|
+
var HALFWIDTH_VOICED_MARK = 65438;
|
|
29
|
+
var HALFWIDTH_SEMI_VOICED_MARK = 65439;
|
|
30
|
+
var COMBINING_VOICED_MARK = "\u3099";
|
|
31
|
+
var COMBINING_SEMI_VOICED_MARK = "\u309A";
|
|
32
|
+
var HALFWIDTH_KANA_TARGETS = "\u3002\u300C\u300D\u3001\u30FB\u30F2\u30A1\u30A3\u30A5\u30A7\u30A9\u30E3\u30E5\u30E7\u30C3\u30FC\u30A2\u30A4\u30A6\u30A8\u30AA\u30AB\u30AD\u30AF\u30B1\u30B3\u30B5\u30B7\u30B9\u30BB\u30BD\u30BF\u30C1\u30C4\u30C6\u30C8\u30CA\u30CB\u30CC\u30CD\u30CE\u30CF\u30D2\u30D5\u30D8\u30DB\u30DE\u30DF\u30E0\u30E1\u30E2\u30E4\u30E6\u30E8\u30E9\u30EA\u30EB\u30EC\u30ED\u30EF\u30F3";
|
|
33
|
+
function foldForm(match) {
|
|
34
|
+
if (match === DOTTED_CAPITAL_I) return "i";
|
|
35
|
+
if (match === COMBINING_DOT_ABOVE || ARMENIAN_MARKS.test(match)) return "";
|
|
36
|
+
return "'";
|
|
37
|
+
}
|
|
38
|
+
function foldWidth(match) {
|
|
39
|
+
const codePoint = match.charCodeAt(0);
|
|
40
|
+
if (codePoint <= FULLWIDTH_ASCII_END) return String.fromCharCode(codePoint - FULLWIDTH_ASCII_OFFSET);
|
|
41
|
+
if (codePoint === HALFWIDTH_VOICED_MARK) return COMBINING_VOICED_MARK;
|
|
42
|
+
if (codePoint === HALFWIDTH_SEMI_VOICED_MARK) return COMBINING_SEMI_VOICED_MARK;
|
|
43
|
+
return HALFWIDTH_KANA_TARGETS[codePoint - HALFWIDTH_KANA_START];
|
|
44
|
+
}
|
|
45
|
+
function isAscii(text) {
|
|
46
|
+
for (let i = 0; i < text.length; i++) {
|
|
47
|
+
if (text.charCodeAt(i) > 127) return false;
|
|
48
|
+
}
|
|
49
|
+
return true;
|
|
50
|
+
}
|
|
51
|
+
function foldWidthForms(text) {
|
|
52
|
+
if (!WIDTH_FORMS.test(text)) return text;
|
|
53
|
+
return text.replace(WIDTH_FORMS_GLOBAL, foldWidth);
|
|
54
|
+
}
|
|
55
|
+
function normalizeForSplitting(text) {
|
|
56
|
+
if (isAscii(text)) return text.toLowerCase();
|
|
57
|
+
return foldWidthForms(text).normalize("NFC").replace(FOLDED_FORMS, foldForm).toLowerCase();
|
|
58
|
+
}
|
|
59
|
+
function stripDiacritics(text) {
|
|
60
|
+
return text.normalize("NFD").replace(/[\u0300-\u036f]/g, "");
|
|
61
|
+
}
|
|
62
|
+
|
|
63
|
+
// src/core/tokenizer/cache.ts
|
|
64
|
+
function clampCacheSize(entries) {
|
|
65
|
+
return Math.max(TOKEN_CACHE_SIZE_FLOOR, Math.min(entries, TOKEN_CACHE_SIZE_CEILING));
|
|
66
|
+
}
|
|
67
|
+
function computeDefaultCacheSize() {
|
|
68
|
+
try {
|
|
69
|
+
if (typeof process !== "undefined" && typeof process.versions?.node === "string" && typeof window === "undefined") {
|
|
70
|
+
const constrainedMemory = typeof process.constrainedMemory === "function" ? process.constrainedMemory() : 0;
|
|
71
|
+
if (constrainedMemory > 0 && constrainedMemory < Number.MAX_SAFE_INTEGER) {
|
|
72
|
+
const budget = constrainedMemory * CONSTRAINED_MEMORY_TOKEN_CACHE_FRACTION;
|
|
73
|
+
return clampCacheSize(Math.floor(budget / TOKEN_CACHE_BYTES_PER_ENTRY));
|
|
74
|
+
}
|
|
75
|
+
return clampCacheSize(NODE_TOKEN_CACHE_ENTRIES);
|
|
76
|
+
}
|
|
77
|
+
if (typeof navigator !== "undefined") {
|
|
78
|
+
const mem = navigator.deviceMemory;
|
|
79
|
+
if (typeof mem === "number" && mem > 0) {
|
|
80
|
+
let entries;
|
|
81
|
+
if (mem <= SMALL_DEVICE_MEMORY_GB) entries = SMALL_DEVICE_TOKEN_CACHE_ENTRIES;
|
|
82
|
+
else if (mem <= MEDIUM_DEVICE_MEMORY_GB) entries = MEDIUM_DEVICE_TOKEN_CACHE_ENTRIES;
|
|
83
|
+
else entries = LARGE_DEVICE_TOKEN_CACHE_ENTRIES;
|
|
84
|
+
return clampCacheSize(entries);
|
|
85
|
+
}
|
|
86
|
+
}
|
|
87
|
+
} catch {
|
|
88
|
+
return clampCacheSize(UNKNOWN_MEMORY_TOKEN_CACHE_ENTRIES);
|
|
89
|
+
}
|
|
90
|
+
return clampCacheSize(UNKNOWN_MEMORY_TOKEN_CACHE_ENTRIES);
|
|
91
|
+
}
|
|
92
|
+
var normalizationCache = /* @__PURE__ */ new Map();
|
|
93
|
+
var normalizationCacheSize = 0;
|
|
94
|
+
var maxCacheSize = computeDefaultCacheSize();
|
|
95
|
+
var cachedLangName = "";
|
|
96
|
+
var cachedFlags = "";
|
|
97
|
+
var cachedBucket = null;
|
|
98
|
+
function configureNormalizationCache(maxSize) {
|
|
99
|
+
if (!Number.isFinite(maxSize)) {
|
|
100
|
+
throw createNarsilError(ErrorCodes.CONFIG_INVALID, "tokenizerCacheSize must be a finite number", {
|
|
101
|
+
received: maxSize
|
|
102
|
+
});
|
|
103
|
+
}
|
|
104
|
+
if (maxSize < 0) {
|
|
105
|
+
throw createNarsilError(ErrorCodes.CONFIG_INVALID, "tokenizerCacheSize must not be negative", {
|
|
106
|
+
received: maxSize
|
|
107
|
+
});
|
|
108
|
+
}
|
|
109
|
+
if (maxSize === 0) {
|
|
110
|
+
throw createNarsilError(
|
|
111
|
+
ErrorCodes.CONFIG_INVALID,
|
|
112
|
+
"tokenizerCacheSize must be greater than zero; the normalization cache cannot be disabled",
|
|
113
|
+
{
|
|
114
|
+
received: maxSize
|
|
115
|
+
}
|
|
116
|
+
);
|
|
117
|
+
}
|
|
118
|
+
maxCacheSize = Math.max(TOKEN_CACHE_SIZE_FLOOR, Math.min(Math.floor(maxSize), TOKEN_CACHE_SIZE_CEILING));
|
|
119
|
+
if (normalizationCacheSize > maxCacheSize) {
|
|
120
|
+
evictOldestEntries(normalizationCacheSize - maxCacheSize);
|
|
121
|
+
}
|
|
122
|
+
}
|
|
123
|
+
function getCacheBucket(language, flags) {
|
|
124
|
+
const langKey = `${language.name}@${language.revision}`;
|
|
125
|
+
if (cachedBucket && langKey === cachedLangName && flags === cachedFlags) return cachedBucket;
|
|
126
|
+
const key = `${langKey}:${flags}`;
|
|
127
|
+
let bucket = normalizationCache.get(key);
|
|
128
|
+
if (!bucket) {
|
|
129
|
+
bucket = /* @__PURE__ */ new Map();
|
|
130
|
+
normalizationCache.set(key, bucket);
|
|
131
|
+
}
|
|
132
|
+
cachedLangName = langKey;
|
|
133
|
+
cachedFlags = flags;
|
|
134
|
+
cachedBucket = bucket;
|
|
135
|
+
return bucket;
|
|
136
|
+
}
|
|
137
|
+
function evictOldestEntries(count) {
|
|
138
|
+
let remaining = count;
|
|
139
|
+
for (const [key, bucket] of normalizationCache) {
|
|
140
|
+
if (remaining <= 0) break;
|
|
141
|
+
if (bucket.size <= remaining) {
|
|
142
|
+
remaining -= bucket.size;
|
|
143
|
+
normalizationCacheSize -= bucket.size;
|
|
144
|
+
normalizationCache.delete(key);
|
|
145
|
+
if (bucket === cachedBucket) cachedBucket = null;
|
|
146
|
+
} else {
|
|
147
|
+
let deleted = 0;
|
|
148
|
+
for (const raw of bucket.keys()) {
|
|
149
|
+
if (deleted >= remaining) break;
|
|
150
|
+
bucket.delete(raw);
|
|
151
|
+
deleted++;
|
|
152
|
+
}
|
|
153
|
+
normalizationCacheSize -= deleted;
|
|
154
|
+
remaining = 0;
|
|
155
|
+
}
|
|
156
|
+
}
|
|
157
|
+
}
|
|
158
|
+
function transformToken(raw, language, stem, removeDiacritics) {
|
|
159
|
+
const flags = (stem ? "s" : "") + (removeDiacritics ? "d" : "");
|
|
160
|
+
let bucket = getCacheBucket(language, flags);
|
|
161
|
+
const cached = bucket.get(raw);
|
|
162
|
+
if (cached !== void 0) return cached;
|
|
163
|
+
let normalized = raw;
|
|
164
|
+
if (language.normalizer) {
|
|
165
|
+
normalized = language.normalizer(normalized);
|
|
166
|
+
}
|
|
167
|
+
if (removeDiacritics) {
|
|
168
|
+
normalized = stripDiacritics(normalized);
|
|
169
|
+
}
|
|
170
|
+
if (stem && language.stemmer) {
|
|
171
|
+
normalized = language.stemmer(normalized);
|
|
172
|
+
}
|
|
173
|
+
if (normalizationCacheSize >= maxCacheSize) {
|
|
174
|
+
evictOldestEntries(Math.max(1, maxCacheSize >>> 2));
|
|
175
|
+
bucket = getCacheBucket(language, flags);
|
|
176
|
+
}
|
|
177
|
+
bucket.set(raw, normalized);
|
|
178
|
+
normalizationCacheSize++;
|
|
179
|
+
return normalized;
|
|
180
|
+
}
|
|
181
|
+
function clearNormalizationCache() {
|
|
182
|
+
normalizationCache.clear();
|
|
183
|
+
normalizationCacheSize = 0;
|
|
184
|
+
cachedBucket = null;
|
|
185
|
+
}
|
|
186
|
+
function resetNormalizationCache() {
|
|
187
|
+
clearNormalizationCache();
|
|
188
|
+
maxCacheSize = computeDefaultCacheSize();
|
|
189
|
+
}
|
|
190
|
+
function getNormalizationCacheSize() {
|
|
191
|
+
return normalizationCacheSize;
|
|
192
|
+
}
|
|
193
|
+
|
|
194
|
+
// src/core/tokenizer/ngrams.ts
|
|
195
|
+
var OTHER_SCRIPT = 0;
|
|
196
|
+
var HAN_SCRIPT = 1;
|
|
197
|
+
var HIRAGANA_SCRIPT = 2;
|
|
198
|
+
var KATAKANA_SCRIPT = 3;
|
|
199
|
+
var HANGUL_SCRIPT = 4;
|
|
200
|
+
var THAI_SCRIPT = 5;
|
|
201
|
+
var LAO_SCRIPT = 6;
|
|
202
|
+
var KHMER_SCRIPT = 7;
|
|
203
|
+
var MYANMAR_SCRIPT = 8;
|
|
204
|
+
var PROLONGED_SOUND_MARK = 12540;
|
|
205
|
+
var IDEOGRAPHIC_ITERATION_MARK = 12293;
|
|
206
|
+
var LOWEST_NGRAM_CODE_POINT = 3584;
|
|
207
|
+
var LOWEST_MARK_CODE_POINT = 768;
|
|
208
|
+
var THAI_BLOCK_END = 3711;
|
|
209
|
+
var LAO_BLOCK_END = 3839;
|
|
210
|
+
var MYANMAR_BLOCK_START = 4096;
|
|
211
|
+
var MYANMAR_BLOCK_END = 4255;
|
|
212
|
+
var KHMER_BLOCK_START = 6016;
|
|
213
|
+
var KHMER_BLOCK_END = 6655;
|
|
214
|
+
var SOUTHEAST_ASIAN_BLOCKS_END = 6656;
|
|
215
|
+
var HIRAGANA_BLOCK_START = 12352;
|
|
216
|
+
var HIRAGANA_BLOCK_END = 12447;
|
|
217
|
+
var COMBINING_VOICED_MARK2 = 12441;
|
|
218
|
+
var COMBINING_SEMI_VOICED_MARK2 = 12442;
|
|
219
|
+
var KATAKANA_BLOCK_START = 12448;
|
|
220
|
+
var KATAKANA_BLOCK_END = 12543;
|
|
221
|
+
var HAN_EXTENSION_A_START = 13312;
|
|
222
|
+
var HAN_EXTENSION_A_END = 19903;
|
|
223
|
+
var HAN_UNIFIED_START = 19968;
|
|
224
|
+
var HAN_UNIFIED_END = 40959;
|
|
225
|
+
var HANGUL_SYLLABLE_START = 44032;
|
|
226
|
+
var HANGUL_SYLLABLE_END = 55203;
|
|
227
|
+
var HAN_COMPATIBILITY_START = 63744;
|
|
228
|
+
var HAN_COMPATIBILITY_END = 64255;
|
|
229
|
+
var SUPPLEMENTARY_PLANE_START = 65536;
|
|
230
|
+
var MARK = /\p{M}/uy;
|
|
231
|
+
var HAN = /\p{Script=Han}/uy;
|
|
232
|
+
var HIRAGANA = /\p{Script=Hiragana}/uy;
|
|
233
|
+
var KATAKANA = /\p{Script=Katakana}/uy;
|
|
234
|
+
var HANGUL = /\p{Script=Hangul}/uy;
|
|
235
|
+
var THAI = /\p{Script=Thai}/uy;
|
|
236
|
+
var LAO = /\p{Script=Lao}/uy;
|
|
237
|
+
var KHMER = /\p{Script=Khmer}/uy;
|
|
238
|
+
var MYANMAR = /\p{Script=Myanmar}/uy;
|
|
239
|
+
function matchesAt(pattern, part, index) {
|
|
240
|
+
pattern.lastIndex = index;
|
|
241
|
+
return pattern.test(part);
|
|
242
|
+
}
|
|
243
|
+
function isMark(part, index, codePoint) {
|
|
244
|
+
if (codePoint < LOWEST_MARK_CODE_POINT) return false;
|
|
245
|
+
if (codePoint >= HAN_UNIFIED_START && codePoint <= HAN_UNIFIED_END) return false;
|
|
246
|
+
if (codePoint >= HIRAGANA_BLOCK_START && codePoint <= KATAKANA_BLOCK_END) {
|
|
247
|
+
return codePoint === COMBINING_VOICED_MARK2 || codePoint === COMBINING_SEMI_VOICED_MARK2;
|
|
248
|
+
}
|
|
249
|
+
if (codePoint >= HANGUL_SYLLABLE_START && codePoint <= HANGUL_SYLLABLE_END) return false;
|
|
250
|
+
if (codePoint >= HAN_EXTENSION_A_START && codePoint <= HAN_EXTENSION_A_END) return false;
|
|
251
|
+
return matchesAt(MARK, part, index);
|
|
252
|
+
}
|
|
253
|
+
function classifyOutsideCommonBlocks(part, index) {
|
|
254
|
+
if (matchesAt(HAN, part, index)) return HAN_SCRIPT;
|
|
255
|
+
if (matchesAt(HIRAGANA, part, index)) return HIRAGANA_SCRIPT;
|
|
256
|
+
if (matchesAt(KATAKANA, part, index)) return KATAKANA_SCRIPT;
|
|
257
|
+
if (matchesAt(HANGUL, part, index)) return HANGUL_SCRIPT;
|
|
258
|
+
if (matchesAt(THAI, part, index)) return THAI_SCRIPT;
|
|
259
|
+
if (matchesAt(LAO, part, index)) return LAO_SCRIPT;
|
|
260
|
+
if (matchesAt(KHMER, part, index)) return KHMER_SCRIPT;
|
|
261
|
+
if (matchesAt(MYANMAR, part, index)) return MYANMAR_SCRIPT;
|
|
262
|
+
return OTHER_SCRIPT;
|
|
263
|
+
}
|
|
264
|
+
function classifySoutheastAsian(part, index, codePoint) {
|
|
265
|
+
if (codePoint <= THAI_BLOCK_END) return matchesAt(THAI, part, index) ? THAI_SCRIPT : OTHER_SCRIPT;
|
|
266
|
+
if (codePoint <= LAO_BLOCK_END) return matchesAt(LAO, part, index) ? LAO_SCRIPT : OTHER_SCRIPT;
|
|
267
|
+
if (codePoint >= MYANMAR_BLOCK_START && codePoint <= MYANMAR_BLOCK_END) {
|
|
268
|
+
return matchesAt(MYANMAR, part, index) ? MYANMAR_SCRIPT : OTHER_SCRIPT;
|
|
269
|
+
}
|
|
270
|
+
if (codePoint >= KHMER_BLOCK_START && codePoint <= KHMER_BLOCK_END) {
|
|
271
|
+
return matchesAt(KHMER, part, index) ? KHMER_SCRIPT : OTHER_SCRIPT;
|
|
272
|
+
}
|
|
273
|
+
return classifyOutsideCommonBlocks(part, index);
|
|
274
|
+
}
|
|
275
|
+
function classify(part, index, codePoint, current) {
|
|
276
|
+
if (codePoint < LOWEST_NGRAM_CODE_POINT) return OTHER_SCRIPT;
|
|
277
|
+
if (codePoint < SOUTHEAST_ASIAN_BLOCKS_END) return classifySoutheastAsian(part, index, codePoint);
|
|
278
|
+
if (codePoint === PROLONGED_SOUND_MARK) {
|
|
279
|
+
return current === HIRAGANA_SCRIPT ? HIRAGANA_SCRIPT : KATAKANA_SCRIPT;
|
|
280
|
+
}
|
|
281
|
+
if (codePoint >= HIRAGANA_BLOCK_START && codePoint <= HIRAGANA_BLOCK_END) return HIRAGANA_SCRIPT;
|
|
282
|
+
if (codePoint >= KATAKANA_BLOCK_START && codePoint <= KATAKANA_BLOCK_END) return KATAKANA_SCRIPT;
|
|
283
|
+
if (codePoint >= HAN_UNIFIED_START && codePoint <= HAN_UNIFIED_END) return HAN_SCRIPT;
|
|
284
|
+
if (codePoint >= HAN_EXTENSION_A_START && codePoint <= HAN_EXTENSION_A_END) return HAN_SCRIPT;
|
|
285
|
+
if (codePoint >= HANGUL_SYLLABLE_START && codePoint <= HANGUL_SYLLABLE_END) return HANGUL_SCRIPT;
|
|
286
|
+
if (codePoint >= HAN_COMPATIBILITY_START && codePoint <= HAN_COMPATIBILITY_END) return HAN_SCRIPT;
|
|
287
|
+
if (codePoint === IDEOGRAPHIC_ITERATION_MARK) return HAN_SCRIPT;
|
|
288
|
+
return classifyOutsideCommonBlocks(part, index);
|
|
289
|
+
}
|
|
290
|
+
function emitRun(part, offsets, kind, size, out) {
|
|
291
|
+
const count = offsets.length - 1;
|
|
292
|
+
if (count < 1) return;
|
|
293
|
+
if (kind === OTHER_SCRIPT || count <= size) {
|
|
294
|
+
out.push(part.slice(offsets[0], offsets[count]));
|
|
295
|
+
return;
|
|
296
|
+
}
|
|
297
|
+
for (let start = 0; start + size <= count; start++) {
|
|
298
|
+
out.push(part.slice(offsets[start], offsets[start + size]));
|
|
299
|
+
}
|
|
300
|
+
}
|
|
301
|
+
function expandNgrams(part, size, out) {
|
|
302
|
+
const length = part.length;
|
|
303
|
+
if (length === 0) return;
|
|
304
|
+
const offsets = [];
|
|
305
|
+
let kind = OTHER_SCRIPT;
|
|
306
|
+
let index = 0;
|
|
307
|
+
while (index < length) {
|
|
308
|
+
const codePoint = part.codePointAt(index);
|
|
309
|
+
if (codePoint === void 0) break;
|
|
310
|
+
const width = codePoint >= SUPPLEMENTARY_PLANE_START ? 2 : 1;
|
|
311
|
+
if (offsets.length > 0 && isMark(part, index, codePoint)) {
|
|
312
|
+
index += width;
|
|
313
|
+
continue;
|
|
314
|
+
}
|
|
315
|
+
const next = classify(part, index, codePoint, kind);
|
|
316
|
+
if (offsets.length > 0 && next !== kind) {
|
|
317
|
+
offsets.push(index);
|
|
318
|
+
emitRun(part, offsets, kind, size, out);
|
|
319
|
+
offsets.length = 0;
|
|
320
|
+
}
|
|
321
|
+
kind = next;
|
|
322
|
+
offsets.push(index);
|
|
323
|
+
index += width;
|
|
324
|
+
}
|
|
325
|
+
offsets.push(length);
|
|
326
|
+
emitRun(part, offsets, kind, size, out);
|
|
327
|
+
}
|
|
328
|
+
|
|
329
|
+
// src/core/tokenizer/tokenize.ts
|
|
330
|
+
function producesSurfaceForms(language, options) {
|
|
331
|
+
if (!options?.collectSurfaces || options.customTokenizer) return false;
|
|
332
|
+
return (options.stem ?? true) && language.stemmer !== null;
|
|
333
|
+
}
|
|
334
|
+
var DEFAULT_SPLIT_PATTERN = /[^\p{L}\p{M}\p{N}_'-]+/u;
|
|
335
|
+
function resolveStopWords(language, override) {
|
|
336
|
+
if (!override) return language.stopWords;
|
|
337
|
+
if (override instanceof Set) return override;
|
|
338
|
+
return override(language.stopWords);
|
|
339
|
+
}
|
|
340
|
+
function splitText(text, language) {
|
|
341
|
+
const pattern = language.tokenizer?.splitPattern ?? DEFAULT_SPLIT_PATTERN;
|
|
342
|
+
const parts = text.split(pattern);
|
|
343
|
+
const ngramSize = language.tokenizer?.ngramSize;
|
|
344
|
+
if (ngramSize === void 0 || ngramSize < 1) return parts;
|
|
345
|
+
const size = Math.floor(ngramSize);
|
|
346
|
+
const expanded = [];
|
|
347
|
+
for (const part of parts) {
|
|
348
|
+
expandNgrams(part, size, expanded);
|
|
349
|
+
}
|
|
350
|
+
return expanded;
|
|
351
|
+
}
|
|
352
|
+
var CHAR_APOSTROPHE = 39;
|
|
353
|
+
var CHAR_S = 115;
|
|
354
|
+
function stripPossessive(token) {
|
|
355
|
+
const len = token.length;
|
|
356
|
+
if (len >= 2 && token.charCodeAt(len - 1) === CHAR_S && token.charCodeAt(len - 2) === CHAR_APOSTROPHE) {
|
|
357
|
+
return token.slice(0, -2);
|
|
358
|
+
}
|
|
359
|
+
if (len >= 1 && token.charCodeAt(len - 1) === CHAR_APOSTROPHE) {
|
|
360
|
+
return token.slice(0, -1);
|
|
361
|
+
}
|
|
362
|
+
return token;
|
|
363
|
+
}
|
|
364
|
+
function visitTokens(text, language, options, visit) {
|
|
365
|
+
const {
|
|
366
|
+
stem = true,
|
|
367
|
+
removeStopWords = true,
|
|
368
|
+
removeDiacritics = false,
|
|
369
|
+
collectSurfaces = false,
|
|
370
|
+
customTokenizer,
|
|
371
|
+
stopWordOverride
|
|
372
|
+
} = options ?? {};
|
|
373
|
+
if (customTokenizer) {
|
|
374
|
+
for (const entry of customTokenizer.tokenize(text)) visit(entry.token, entry.position, void 0, entry.token);
|
|
375
|
+
return;
|
|
376
|
+
}
|
|
377
|
+
const rawParts = splitText(normalizeForSplitting(text), language);
|
|
378
|
+
const minLength = language.tokenizer?.minTokenLength ?? DEFAULT_MIN_TOKEN_LENGTH;
|
|
379
|
+
const effectiveDiacritics = removeDiacritics || (language.tokenizer?.normalizeDiacritics ?? false);
|
|
380
|
+
const wantSurfaces = collectSurfaces && stem && language.stemmer !== null;
|
|
381
|
+
const stopWords = removeStopWords ? resolveStopWords(language, stopWordOverride) : null;
|
|
382
|
+
const stripPossessives = language.tokenizer?.stripPossessive ?? false;
|
|
383
|
+
for (let position = 0; position < rawParts.length; position++) {
|
|
384
|
+
const part = rawParts[position];
|
|
385
|
+
const candidate = stripPossessives ? stripPossessive(part) : part;
|
|
386
|
+
if (candidate.length < minLength || stopWords?.has(candidate)) continue;
|
|
387
|
+
const processed = transformToken(candidate, language, stem, effectiveDiacritics);
|
|
388
|
+
if (processed.length === 0) continue;
|
|
389
|
+
if (!wantSurfaces) {
|
|
390
|
+
visit(processed, position, void 0, part);
|
|
391
|
+
continue;
|
|
392
|
+
}
|
|
393
|
+
const surface = effectiveDiacritics ? transformToken(candidate, language, false, true) : candidate;
|
|
394
|
+
visit(processed, position, surface === processed ? void 0 : surface, part);
|
|
395
|
+
}
|
|
396
|
+
}
|
|
397
|
+
function tokenize(text, language, options) {
|
|
398
|
+
if (options?.customTokenizer) {
|
|
399
|
+
const customResult = options.customTokenizer.tokenize(text);
|
|
400
|
+
return { tokens: customResult, originalTokens: customResult.map((entry) => entry.token) };
|
|
401
|
+
}
|
|
402
|
+
const tokens = [];
|
|
403
|
+
const originalTokens = [];
|
|
404
|
+
const surfaces = producesSurfaceForms(language, options) ? [] : void 0;
|
|
405
|
+
visitTokens(text, language, options, (token, position, surface, original) => {
|
|
406
|
+
tokens.push({ token, position });
|
|
407
|
+
originalTokens.push(original);
|
|
408
|
+
surfaces?.push(surface);
|
|
409
|
+
});
|
|
410
|
+
return { tokens, originalTokens, surfaces };
|
|
411
|
+
}
|
|
412
|
+
function* tokenizeIterator(text, language, options) {
|
|
413
|
+
const {
|
|
414
|
+
stem = true,
|
|
415
|
+
removeStopWords = true,
|
|
416
|
+
removeDiacritics = false,
|
|
417
|
+
collectSurfaces = false,
|
|
418
|
+
customTokenizer,
|
|
419
|
+
stopWordOverride
|
|
420
|
+
} = options ?? {};
|
|
421
|
+
if (customTokenizer) {
|
|
422
|
+
const customResult = customTokenizer.tokenize(text);
|
|
423
|
+
for (const entry of customResult) {
|
|
424
|
+
yield entry;
|
|
425
|
+
}
|
|
426
|
+
return;
|
|
427
|
+
}
|
|
428
|
+
const normalized = normalizeForSplitting(text);
|
|
429
|
+
const rawParts = splitText(normalized, language);
|
|
430
|
+
const minLength = language.tokenizer?.minTokenLength ?? DEFAULT_MIN_TOKEN_LENGTH;
|
|
431
|
+
const effectiveDiacritics = removeDiacritics || (language.tokenizer?.normalizeDiacritics ?? false);
|
|
432
|
+
const stemsTokens = stem && language.stemmer !== null;
|
|
433
|
+
const wantSurfaces = collectSurfaces && stemsTokens;
|
|
434
|
+
const stopWords = removeStopWords ? resolveStopWords(language, stopWordOverride) : /* @__PURE__ */ new Set();
|
|
435
|
+
const stripPossessives = language.tokenizer?.stripPossessive ?? false;
|
|
436
|
+
let position = 0;
|
|
437
|
+
for (const part of rawParts) {
|
|
438
|
+
const candidate = stripPossessives ? stripPossessive(part) : part;
|
|
439
|
+
if (candidate.length < minLength) {
|
|
440
|
+
position++;
|
|
441
|
+
continue;
|
|
442
|
+
}
|
|
443
|
+
if (stopWords.has(candidate)) {
|
|
444
|
+
position++;
|
|
445
|
+
continue;
|
|
446
|
+
}
|
|
447
|
+
const processed = transformToken(candidate, language, stem, effectiveDiacritics);
|
|
448
|
+
if (processed.length > 0) {
|
|
449
|
+
if (wantSurfaces) {
|
|
450
|
+
const surface = effectiveDiacritics ? transformToken(candidate, language, false, true) : candidate;
|
|
451
|
+
yield { token: processed, position, surface: surface === processed ? void 0 : surface };
|
|
452
|
+
} else {
|
|
453
|
+
yield { token: processed, position };
|
|
454
|
+
}
|
|
455
|
+
}
|
|
456
|
+
position++;
|
|
457
|
+
}
|
|
458
|
+
}
|
|
459
|
+
|
|
460
|
+
// src/core/fuzzy.ts
|
|
461
|
+
var FIRST_SURROGATE = 55296;
|
|
462
|
+
var LAST_SURROGATE = 57343;
|
|
463
|
+
var LAST_BASIC_PLANE_CODE_POINT = 65535;
|
|
464
|
+
function hasSurrogate(text) {
|
|
465
|
+
for (let index = 0; index < text.length; index++) {
|
|
466
|
+
const unit = text.charCodeAt(index);
|
|
467
|
+
if (unit >= FIRST_SURROGATE && unit <= LAST_SURROGATE) return true;
|
|
468
|
+
}
|
|
469
|
+
return false;
|
|
470
|
+
}
|
|
471
|
+
function fuzzyPrefixOf(queryTerm, prefixLength) {
|
|
472
|
+
let end = 0;
|
|
473
|
+
for (let counted = 0; counted < prefixLength && end < queryTerm.length; counted++) {
|
|
474
|
+
const codePoint = queryTerm.codePointAt(end);
|
|
475
|
+
end += codePoint !== void 0 && codePoint > LAST_BASIC_PLANE_CODE_POINT ? 2 : 1;
|
|
476
|
+
}
|
|
477
|
+
return queryTerm.slice(0, end);
|
|
478
|
+
}
|
|
479
|
+
function fuzzyTermMatches(queryTerm, indexTerm, tolerance, prefixLength) {
|
|
480
|
+
if (queryTerm === indexTerm) return true;
|
|
481
|
+
if (tolerance <= 0) return false;
|
|
482
|
+
if (!indexTerm.startsWith(fuzzyPrefixOf(queryTerm, prefixLength))) return false;
|
|
483
|
+
return boundedLevenshtein(queryTerm, indexTerm, tolerance).withinTolerance;
|
|
484
|
+
}
|
|
485
|
+
function boundedLevenshtein(a, b, tolerance) {
|
|
486
|
+
if (tolerance < 0) return { distance: -1, withinTolerance: false };
|
|
487
|
+
if (a === b) return { distance: 0, withinTolerance: true };
|
|
488
|
+
if (hasSurrogate(a) || hasSurrogate(b)) return bandedDistance(Array.from(a), Array.from(b), tolerance);
|
|
489
|
+
return bandedDistance(a, b, tolerance);
|
|
490
|
+
}
|
|
491
|
+
function bandedDistance(a, b, tolerance) {
|
|
492
|
+
const m = a.length;
|
|
493
|
+
const n = b.length;
|
|
494
|
+
const beyond = tolerance + 1;
|
|
495
|
+
if (Math.abs(m - n) > tolerance) return { distance: beyond, withinTolerance: false };
|
|
496
|
+
let previous = new Uint32Array(n + 1).fill(beyond);
|
|
497
|
+
let current = new Uint32Array(n + 1).fill(beyond);
|
|
498
|
+
const firstRowEnd = Math.min(n, tolerance);
|
|
499
|
+
for (let j = 0; j <= firstRowEnd; j++) previous[j] = j;
|
|
500
|
+
for (let i = 1; i <= m; i++) {
|
|
501
|
+
const low = Math.max(1, i - tolerance);
|
|
502
|
+
const high = Math.min(n, i + tolerance);
|
|
503
|
+
current[low - 1] = low === 1 ? i : beyond;
|
|
504
|
+
let rowMin = current[low - 1];
|
|
505
|
+
const character = a[i - 1];
|
|
506
|
+
for (let j = low; j <= high; j++) {
|
|
507
|
+
const substitution = previous[j - 1] + (character === b[j - 1] ? 0 : 1);
|
|
508
|
+
const cell = Math.min(previous[j] + 1, current[j - 1] + 1, substitution, beyond);
|
|
509
|
+
current[j] = cell;
|
|
510
|
+
if (cell < rowMin) rowMin = cell;
|
|
511
|
+
}
|
|
512
|
+
if (rowMin > tolerance) return { distance: beyond, withinTolerance: false };
|
|
513
|
+
const finished = previous;
|
|
514
|
+
previous = current;
|
|
515
|
+
current = finished;
|
|
516
|
+
}
|
|
517
|
+
const distance = previous[n];
|
|
518
|
+
return { distance, withinTolerance: distance <= tolerance };
|
|
519
|
+
}
|
|
520
|
+
|
|
521
|
+
// src/core/partition/utils.ts
|
|
522
|
+
function getOrCreateFieldNameIndex(table, fieldName) {
|
|
523
|
+
const existing = table.indexMap.get(fieldName);
|
|
524
|
+
if (existing !== void 0) return existing;
|
|
525
|
+
const idx = table.names.length;
|
|
526
|
+
table.names.push(fieldName);
|
|
527
|
+
table.indexMap.set(fieldName, idx);
|
|
528
|
+
return idx;
|
|
529
|
+
}
|
|
530
|
+
var VECTOR_PATTERN = /^vector\[(\d+)]$/;
|
|
531
|
+
function getNestedValue(obj, path) {
|
|
532
|
+
const parts = path.split(".");
|
|
533
|
+
let current = obj;
|
|
534
|
+
for (let i = 0; i < parts.length; i++) {
|
|
535
|
+
if (current === null || current === void 0 || typeof current !== "object") return void 0;
|
|
536
|
+
if (!Object.hasOwn(current, parts[i])) return void 0;
|
|
537
|
+
current = current[parts[i]];
|
|
538
|
+
}
|
|
539
|
+
return current;
|
|
540
|
+
}
|
|
541
|
+
function stringArraysEqual(a, b) {
|
|
542
|
+
if (a === b) return true;
|
|
543
|
+
if (!a || !b) return false;
|
|
544
|
+
if (a.length !== b.length) return false;
|
|
545
|
+
for (let i = 0; i < a.length; i++) {
|
|
546
|
+
if (a[i] !== b[i]) return false;
|
|
547
|
+
}
|
|
548
|
+
return true;
|
|
549
|
+
}
|
|
550
|
+
function textFieldsChanged(oldDoc, newDoc, fields) {
|
|
551
|
+
for (const { path, base } of fields) {
|
|
552
|
+
if (!isWordIndexedBase(base)) continue;
|
|
553
|
+
const oldVal = getNestedValue(oldDoc, path);
|
|
554
|
+
const newVal = getNestedValue(newDoc, path);
|
|
555
|
+
if (base === "string") {
|
|
556
|
+
if (oldVal !== newVal) return true;
|
|
557
|
+
} else {
|
|
558
|
+
if (!stringArraysEqual(oldVal, newVal)) return true;
|
|
559
|
+
}
|
|
560
|
+
}
|
|
561
|
+
return false;
|
|
562
|
+
}
|
|
563
|
+
function parseVectorDimension(fieldType) {
|
|
564
|
+
const match = VECTOR_PATTERN.exec(fieldType);
|
|
565
|
+
return match ? Number.parseInt(match[1], 10) : null;
|
|
566
|
+
}
|
|
567
|
+
function tokenizeOptions(options) {
|
|
568
|
+
return {
|
|
569
|
+
stem: true,
|
|
570
|
+
removeStopWords: true,
|
|
571
|
+
collectSurfaces: options?.collectSurfaces === true,
|
|
572
|
+
stopWordOverride: options?.stopWordOverride,
|
|
573
|
+
customTokenizer: options?.customTokenizer
|
|
574
|
+
};
|
|
575
|
+
}
|
|
576
|
+
function getFlatSchema(state, schema) {
|
|
577
|
+
if (state.lastSchemaRef === schema && state.flatSchemaCache) return state.flatSchemaCache;
|
|
578
|
+
state.flatSchemaCache = flattenSchema(schema);
|
|
579
|
+
state.lastSchemaRef = schema;
|
|
580
|
+
return state.flatSchemaCache;
|
|
581
|
+
}
|
|
582
|
+
function getFieldValueForDoc(docStore, docId, fieldPath) {
|
|
583
|
+
const stored = docStore.get(docId);
|
|
584
|
+
if (!stored) return void 0;
|
|
585
|
+
return getNestedValue(stored.fields, fieldPath);
|
|
586
|
+
}
|
|
587
|
+
function getFieldValueByInternalId(docStore, internalId, fieldPath) {
|
|
588
|
+
const externalId = docStore.getExternalId(internalId);
|
|
589
|
+
if (externalId === void 0) return void 0;
|
|
590
|
+
return getFieldValueForDoc(docStore, externalId, fieldPath);
|
|
591
|
+
}
|
|
592
|
+
|
|
593
|
+
export { boundedLevenshtein, clearNormalizationCache, configureNormalizationCache, fuzzyPrefixOf, fuzzyTermMatches, getFieldValueByInternalId, getFieldValueForDoc, getFlatSchema, getNestedValue, getNormalizationCacheSize, getOrCreateFieldNameIndex, parseVectorDimension, resetNormalizationCache, textFieldsChanged, tokenize, tokenizeIterator, tokenizeOptions, visitTokens };
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
// src/vector/hnsw/constants.ts
|
|
2
|
+
var MAX_LAYER_CAP = 32;
|
|
3
|
+
var MAX_M = 512;
|
|
4
|
+
var MIN_M = 2;
|
|
5
|
+
var EARLIER_RELEASE_MIN_M = 1;
|
|
6
|
+
var INITIAL_UPPER_SLOTS = 64;
|
|
7
|
+
var INITIAL_LIST_CAPACITY = 64;
|
|
8
|
+
var PLACEMENT_ROOM_REQUESTS = 8;
|
|
9
|
+
var COMPACTION_TOMBSTONE_RATIO = 0.1;
|
|
10
|
+
var COMPACTION_ABSOLUTE_THRESHOLD = 1e3;
|
|
11
|
+
var DEFAULT_EF_SEARCH = 50;
|
|
12
|
+
|
|
13
|
+
export { COMPACTION_ABSOLUTE_THRESHOLD, COMPACTION_TOMBSTONE_RATIO, DEFAULT_EF_SEARCH, EARLIER_RELEASE_MIN_M, INITIAL_LIST_CAPACITY, INITIAL_UPPER_SLOTS, MAX_LAYER_CAP, MAX_M, MIN_M, PLACEMENT_ROOM_REQUESTS };
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { requireDocumentWithinSizeLimit, validateRequiredFields } from './chunk-
|
|
1
|
+
import { requireDocumentWithinSizeLimit, validateRequiredFields } from './chunk-Z3DQPOJQ.mjs';
|
|
2
2
|
import { ErrorCodes, NarsilError } from './chunk-I64LXKO3.mjs';
|
|
3
3
|
|
|
4
4
|
// src/narsil/internals.ts
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { MAX_SORT_FIELDS } from './chunk-
|
|
2
|
-
import { nativeSearchCoreIsEnabled, loadNativeCore } from './chunk-
|
|
1
|
+
import { MAX_SORT_FIELDS } from './chunk-3DDUNI6N.mjs';
|
|
2
|
+
import { nativeSearchCoreIsEnabled, loadNativeCore } from './chunk-2E35UM3G.mjs';
|
|
3
3
|
import { NarsilError, ServerErrorCodes, ErrorCodes } from './chunk-I64LXKO3.mjs';
|
|
4
4
|
import { mkdtemp, open, rm } from 'fs/promises';
|
|
5
5
|
import { tmpdir } from 'os';
|
|
@@ -1,5 +1,6 @@
|
|
|
1
1
|
import { OSQ_STEPS, OSQ_GRID, OSQ_REFINE_ROUNDS, OSQ_REFINE_TOLERANCE, OSQ4_NARROW_DIMENSION_LIMIT, DEFAULT_OSQ4_NARROW_OVERSAMPLE, DEFAULT_OVERSAMPLE_BY_BITS, OSQ_QUERY_BITS, OSQ_LAMBDA } from './chunk-ZJ4CSAB2.mjs';
|
|
2
|
-
import {
|
|
2
|
+
import { nativeSearchCoreIsEnabled, loadNativeCore, stopUsingTheNativeSearchCore, noteFallback } from './chunk-2E35UM3G.mjs';
|
|
3
|
+
import { COMPACTION_TOMBSTONE_RATIO, COMPACTION_ABSOLUTE_THRESHOLD, MAX_LAYER_CAP, PLACEMENT_ROOM_REQUESTS, INITIAL_LIST_CAPACITY, DEFAULT_EF_SEARCH, INITIAL_UPPER_SLOTS } from './chunk-KEHK2O6P.mjs';
|
|
3
4
|
import { compareCodePoints } from './chunk-LJYTW7GT.mjs';
|
|
4
5
|
import { NarsilError, ErrorCodes } from './chunk-I64LXKO3.mjs';
|
|
5
6
|
import { readSync, closeSync, openSync } from 'fs';
|