codevault 1.0.0
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/.env.example +37 -0
- package/LICENSE +21 -0
- package/README.md +174 -0
- package/dist/chunking/file-grouper.d.ts +39 -0
- package/dist/chunking/file-grouper.d.ts.map +1 -0
- package/dist/chunking/file-grouper.js +181 -0
- package/dist/chunking/file-grouper.js.map +1 -0
- package/dist/chunking/semantic-chunker.d.ts +37 -0
- package/dist/chunking/semantic-chunker.d.ts.map +1 -0
- package/dist/chunking/semantic-chunker.js +172 -0
- package/dist/chunking/semantic-chunker.js.map +1 -0
- package/dist/chunking/token-counter.d.ts +28 -0
- package/dist/chunking/token-counter.d.ts.map +1 -0
- package/dist/chunking/token-counter.js +207 -0
- package/dist/chunking/token-counter.js.map +1 -0
- package/dist/cli/commands/context.d.ts +3 -0
- package/dist/cli/commands/context.d.ts.map +1 -0
- package/dist/cli/commands/context.js +98 -0
- package/dist/cli/commands/context.js.map +1 -0
- package/dist/cli.d.ts +3 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +222 -0
- package/dist/cli.js.map +1 -0
- package/dist/codemap/io.d.ts +5 -0
- package/dist/codemap/io.d.ts.map +1 -0
- package/dist/codemap/io.js +30 -0
- package/dist/codemap/io.js.map +1 -0
- package/dist/context/packs.d.ts +33 -0
- package/dist/context/packs.d.ts.map +1 -0
- package/dist/context/packs.js +180 -0
- package/dist/context/packs.js.map +1 -0
- package/dist/core/indexer.d.ts +3 -0
- package/dist/core/indexer.d.ts.map +1 -0
- package/dist/core/indexer.js +610 -0
- package/dist/core/indexer.js.map +1 -0
- package/dist/core/metadata.d.ts +19 -0
- package/dist/core/metadata.d.ts.map +1 -0
- package/dist/core/metadata.js +161 -0
- package/dist/core/metadata.js.map +1 -0
- package/dist/core/search.d.ts +6 -0
- package/dist/core/search.d.ts.map +1 -0
- package/dist/core/search.js +482 -0
- package/dist/core/search.js.map +1 -0
- package/dist/core/symbol-extractor.d.ts +3 -0
- package/dist/core/symbol-extractor.d.ts.map +1 -0
- package/dist/core/symbol-extractor.js +78 -0
- package/dist/core/symbol-extractor.js.map +1 -0
- package/dist/core/types.d.ts +96 -0
- package/dist/core/types.d.ts.map +1 -0
- package/dist/core/types.js +2 -0
- package/dist/core/types.js.map +1 -0
- package/dist/database/db.d.ts +63 -0
- package/dist/database/db.d.ts.map +1 -0
- package/dist/database/db.js +205 -0
- package/dist/database/db.js.map +1 -0
- package/dist/indexer/merkle.d.ts +13 -0
- package/dist/indexer/merkle.d.ts.map +1 -0
- package/dist/indexer/merkle.js +86 -0
- package/dist/indexer/merkle.js.map +1 -0
- package/dist/indexer/update.d.ts +19 -0
- package/dist/indexer/update.d.ts.map +1 -0
- package/dist/indexer/update.js +40 -0
- package/dist/indexer/update.js.map +1 -0
- package/dist/indexer/watch.d.ts +21 -0
- package/dist/indexer/watch.d.ts.map +1 -0
- package/dist/indexer/watch.js +204 -0
- package/dist/indexer/watch.js.map +1 -0
- package/dist/languages/rules.d.ts +11 -0
- package/dist/languages/rules.d.ts.map +1 -0
- package/dist/languages/rules.js +371 -0
- package/dist/languages/rules.js.map +1 -0
- package/dist/languages/tree-sitter-loader.d.ts +27 -0
- package/dist/languages/tree-sitter-loader.d.ts.map +1 -0
- package/dist/languages/tree-sitter-loader.js +75 -0
- package/dist/languages/tree-sitter-loader.js.map +1 -0
- package/dist/mcp/tools/use-context-pack.d.ts +57 -0
- package/dist/mcp/tools/use-context-pack.d.ts.map +1 -0
- package/dist/mcp/tools/use-context-pack.js +91 -0
- package/dist/mcp/tools/use-context-pack.js.map +1 -0
- package/dist/mcp-server.d.ts +3 -0
- package/dist/mcp-server.d.ts.map +1 -0
- package/dist/mcp-server.js +363 -0
- package/dist/mcp-server.js.map +1 -0
- package/dist/providers/base.d.ts +34 -0
- package/dist/providers/base.d.ts.map +1 -0
- package/dist/providers/base.js +160 -0
- package/dist/providers/base.js.map +1 -0
- package/dist/providers/index.d.ts +6 -0
- package/dist/providers/index.d.ts.map +1 -0
- package/dist/providers/index.js +22 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/providers/ollama.d.ts +13 -0
- package/dist/providers/ollama.d.ts.map +1 -0
- package/dist/providers/ollama.js +50 -0
- package/dist/providers/ollama.js.map +1 -0
- package/dist/providers/openai.d.ts +13 -0
- package/dist/providers/openai.d.ts.map +1 -0
- package/dist/providers/openai.js +59 -0
- package/dist/providers/openai.js.map +1 -0
- package/dist/providers/token-counter.d.ts +2 -0
- package/dist/providers/token-counter.d.ts.map +1 -0
- package/dist/providers/token-counter.js +18 -0
- package/dist/providers/token-counter.js.map +1 -0
- package/dist/ranking/api-reranker.d.ts +18 -0
- package/dist/ranking/api-reranker.d.ts.map +1 -0
- package/dist/ranking/api-reranker.js +148 -0
- package/dist/ranking/api-reranker.js.map +1 -0
- package/dist/ranking/symbol-boost.d.ts +15 -0
- package/dist/ranking/symbol-boost.d.ts.map +1 -0
- package/dist/ranking/symbol-boost.js +154 -0
- package/dist/ranking/symbol-boost.js.map +1 -0
- package/dist/search/bm25.d.ts +17 -0
- package/dist/search/bm25.d.ts.map +1 -0
- package/dist/search/bm25.js +56 -0
- package/dist/search/bm25.js.map +1 -0
- package/dist/search/hybrid.d.ts +21 -0
- package/dist/search/hybrid.d.ts.map +1 -0
- package/dist/search/hybrid.js +50 -0
- package/dist/search/hybrid.js.map +1 -0
- package/dist/search/scope.d.ts +5 -0
- package/dist/search/scope.d.ts.map +1 -0
- package/dist/search/scope.js +107 -0
- package/dist/search/scope.js.map +1 -0
- package/dist/storage/encrypted-chunks.d.ts +40 -0
- package/dist/storage/encrypted-chunks.d.ts.map +1 -0
- package/dist/storage/encrypted-chunks.js +236 -0
- package/dist/storage/encrypted-chunks.js.map +1 -0
- package/dist/symbols/extract.d.ts +15 -0
- package/dist/symbols/extract.d.ts.map +1 -0
- package/dist/symbols/extract.js +187 -0
- package/dist/symbols/extract.js.map +1 -0
- package/dist/symbols/graph.d.ts +3 -0
- package/dist/symbols/graph.d.ts.map +1 -0
- package/dist/symbols/graph.js +89 -0
- package/dist/symbols/graph.js.map +1 -0
- package/dist/types/ast.d.ts +3 -0
- package/dist/types/ast.d.ts.map +1 -0
- package/dist/types/ast.js +2 -0
- package/dist/types/ast.js.map +1 -0
- package/dist/types/codemap.d.ts +58 -0
- package/dist/types/codemap.d.ts.map +1 -0
- package/dist/types/codemap.js +224 -0
- package/dist/types/codemap.js.map +1 -0
- package/dist/types/context-pack.d.ts +47 -0
- package/dist/types/context-pack.d.ts.map +1 -0
- package/dist/types/context-pack.js +44 -0
- package/dist/types/context-pack.js.map +1 -0
- package/dist/types/search.d.ts +15 -0
- package/dist/types/search.d.ts.map +1 -0
- package/dist/types/search.js +11 -0
- package/dist/types/search.js.map +1 -0
- package/dist/utils/rate-limiter.d.ts +31 -0
- package/dist/utils/rate-limiter.d.ts.map +1 -0
- package/dist/utils/rate-limiter.js +168 -0
- package/dist/utils/rate-limiter.js.map +1 -0
- package/package.json +92 -0
|
@@ -0,0 +1,224 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
export const DEFAULT_PATH_WEIGHT = 1;
|
|
3
|
+
export const DEFAULT_SUCCESS_RATE = 0;
|
|
4
|
+
export const CodemapChunkSchema = z.object({
|
|
5
|
+
file: z.string(),
|
|
6
|
+
symbol: z.union([z.string(), z.null()]).optional(),
|
|
7
|
+
sha: z.string(),
|
|
8
|
+
lang: z.string().optional(),
|
|
9
|
+
chunkType: z.string().optional(),
|
|
10
|
+
provider: z.string().optional(),
|
|
11
|
+
dimensions: z.number().optional(),
|
|
12
|
+
hasCodevaultTags: z.boolean().optional(),
|
|
13
|
+
hasIntent: z.boolean().optional(),
|
|
14
|
+
hasDocumentation: z.boolean().optional(),
|
|
15
|
+
variableCount: z.number().optional(),
|
|
16
|
+
synonyms: z.array(z.string()).optional(),
|
|
17
|
+
path_weight: z.number().optional(),
|
|
18
|
+
last_used_at: z.union([z.string(), z.null()]).optional(),
|
|
19
|
+
success_rate: z.number().optional(),
|
|
20
|
+
encrypted: z.boolean().optional(),
|
|
21
|
+
symbol_signature: z.string().optional(),
|
|
22
|
+
symbol_parameters: z.array(z.string()).optional(),
|
|
23
|
+
symbol_return: z.string().optional(),
|
|
24
|
+
symbol_calls: z.array(z.string()).optional(),
|
|
25
|
+
symbol_call_targets: z.array(z.string()).optional(),
|
|
26
|
+
symbol_callers: z.array(z.string()).optional(),
|
|
27
|
+
symbol_neighbors: z.array(z.string()).optional()
|
|
28
|
+
}).passthrough();
|
|
29
|
+
export const CodemapSchema = z.record(z.string(), CodemapChunkSchema);
|
|
30
|
+
const KNOWN_FIELDS = new Set([
|
|
31
|
+
'file',
|
|
32
|
+
'symbol',
|
|
33
|
+
'sha',
|
|
34
|
+
'lang',
|
|
35
|
+
'chunkType',
|
|
36
|
+
'provider',
|
|
37
|
+
'dimensions',
|
|
38
|
+
'hasCodevaultTags',
|
|
39
|
+
'hasIntent',
|
|
40
|
+
'hasDocumentation',
|
|
41
|
+
'variableCount',
|
|
42
|
+
'synonyms',
|
|
43
|
+
'path_weight',
|
|
44
|
+
'last_used_at',
|
|
45
|
+
'success_rate',
|
|
46
|
+
'symbol_signature',
|
|
47
|
+
'symbol_parameters',
|
|
48
|
+
'symbol_return',
|
|
49
|
+
'symbol_calls',
|
|
50
|
+
'symbol_call_targets',
|
|
51
|
+
'symbol_callers',
|
|
52
|
+
'symbol_neighbors',
|
|
53
|
+
'encrypted'
|
|
54
|
+
]);
|
|
55
|
+
function sanitizeStringArray(value, options = {}) {
|
|
56
|
+
if (!Array.isArray(value)) {
|
|
57
|
+
return [];
|
|
58
|
+
}
|
|
59
|
+
const unique = new Set();
|
|
60
|
+
for (const entry of value) {
|
|
61
|
+
if (typeof entry !== 'string') {
|
|
62
|
+
continue;
|
|
63
|
+
}
|
|
64
|
+
let trimmed = entry.trim();
|
|
65
|
+
if (trimmed.length === 0) {
|
|
66
|
+
continue;
|
|
67
|
+
}
|
|
68
|
+
if (options.lowercase) {
|
|
69
|
+
trimmed = trimmed.toLowerCase();
|
|
70
|
+
}
|
|
71
|
+
unique.add(trimmed);
|
|
72
|
+
}
|
|
73
|
+
return Array.from(unique.values());
|
|
74
|
+
}
|
|
75
|
+
function sanitizeOptionalString(value) {
|
|
76
|
+
if (typeof value !== 'string') {
|
|
77
|
+
return undefined;
|
|
78
|
+
}
|
|
79
|
+
const trimmed = value.trim();
|
|
80
|
+
return trimmed.length > 0 ? trimmed : undefined;
|
|
81
|
+
}
|
|
82
|
+
function sanitizePathWeight(value) {
|
|
83
|
+
if (typeof value !== 'number' || !Number.isFinite(value)) {
|
|
84
|
+
return DEFAULT_PATH_WEIGHT;
|
|
85
|
+
}
|
|
86
|
+
if (value < 0) {
|
|
87
|
+
return 0;
|
|
88
|
+
}
|
|
89
|
+
return value;
|
|
90
|
+
}
|
|
91
|
+
function sanitizeSuccessRate(value) {
|
|
92
|
+
if (typeof value !== 'number' || Number.isNaN(value)) {
|
|
93
|
+
return DEFAULT_SUCCESS_RATE;
|
|
94
|
+
}
|
|
95
|
+
if (value < 0) {
|
|
96
|
+
return 0;
|
|
97
|
+
}
|
|
98
|
+
if (value > 1) {
|
|
99
|
+
return 1;
|
|
100
|
+
}
|
|
101
|
+
return value;
|
|
102
|
+
}
|
|
103
|
+
function sanitizeLastUsed(value) {
|
|
104
|
+
if (!value) {
|
|
105
|
+
return undefined;
|
|
106
|
+
}
|
|
107
|
+
const date = typeof value === 'string' ? new Date(value) : value;
|
|
108
|
+
if (!(date instanceof Date) || Number.isNaN(date.valueOf())) {
|
|
109
|
+
return undefined;
|
|
110
|
+
}
|
|
111
|
+
return date.toISOString();
|
|
112
|
+
}
|
|
113
|
+
function sanitizeVariableCount(value) {
|
|
114
|
+
if (typeof value !== 'number' || !Number.isFinite(value)) {
|
|
115
|
+
return 0;
|
|
116
|
+
}
|
|
117
|
+
const rounded = Math.round(value);
|
|
118
|
+
return rounded < 0 ? 0 : rounded;
|
|
119
|
+
}
|
|
120
|
+
function extractExtras(source) {
|
|
121
|
+
const extras = {};
|
|
122
|
+
if (!source || typeof source !== 'object') {
|
|
123
|
+
return extras;
|
|
124
|
+
}
|
|
125
|
+
for (const [key, val] of Object.entries(source)) {
|
|
126
|
+
if (!KNOWN_FIELDS.has(key)) {
|
|
127
|
+
extras[key] = val;
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
return extras;
|
|
131
|
+
}
|
|
132
|
+
function internalNormalize(raw) {
|
|
133
|
+
const fallback = raw && typeof raw === 'object' ? raw : {};
|
|
134
|
+
const parsed = CodemapChunkSchema.safeParse(fallback);
|
|
135
|
+
const data = parsed.success ? parsed.data : fallback;
|
|
136
|
+
const extras = extractExtras(data);
|
|
137
|
+
const file = typeof data.file === 'string' && data.file.trim().length > 0 ? data.file : 'unknown';
|
|
138
|
+
const sha = typeof data.sha === 'string' && data.sha.trim().length > 0 ? data.sha : 'unknown';
|
|
139
|
+
const lang = typeof data.lang === 'string' && data.lang.trim().length > 0 ? data.lang : 'unknown';
|
|
140
|
+
const chunkType = typeof data.chunkType === 'string' && data.chunkType.trim().length > 0 ? data.chunkType : undefined;
|
|
141
|
+
const provider = typeof data.provider === 'string' && data.provider.trim().length > 0 ? data.provider : undefined;
|
|
142
|
+
const dimensions = typeof data.dimensions === 'number' && Number.isFinite(data.dimensions) ? data.dimensions : undefined;
|
|
143
|
+
const hasCodevaultTags = typeof data.hasCodevaultTags === 'boolean' ? data.hasCodevaultTags : false;
|
|
144
|
+
const hasIntent = typeof data.hasIntent === 'boolean' ? data.hasIntent : false;
|
|
145
|
+
const hasDocumentation = typeof data.hasDocumentation === 'boolean' ? data.hasDocumentation : false;
|
|
146
|
+
const variableCount = sanitizeVariableCount(data.variableCount);
|
|
147
|
+
const synonyms = sanitizeStringArray(data.synonyms);
|
|
148
|
+
const pathWeight = sanitizePathWeight(data.path_weight);
|
|
149
|
+
const lastUsed = sanitizeLastUsed(data.last_used_at);
|
|
150
|
+
const successRate = sanitizeSuccessRate(data.success_rate);
|
|
151
|
+
const encrypted = typeof data.encrypted === 'boolean' ? data.encrypted : false;
|
|
152
|
+
const symbolSignature = sanitizeOptionalString(data.symbol_signature);
|
|
153
|
+
const symbolParameters = Array.isArray(data.symbol_parameters)
|
|
154
|
+
? sanitizeStringArray(data.symbol_parameters)
|
|
155
|
+
: [];
|
|
156
|
+
const symbolReturn = sanitizeOptionalString(data.symbol_return);
|
|
157
|
+
const symbolCalls = Array.isArray(data.symbol_calls)
|
|
158
|
+
? sanitizeStringArray(data.symbol_calls)
|
|
159
|
+
: [];
|
|
160
|
+
const symbolCallTargets = Array.isArray(data.symbol_call_targets)
|
|
161
|
+
? sanitizeStringArray(data.symbol_call_targets)
|
|
162
|
+
: [];
|
|
163
|
+
const symbolCallers = Array.isArray(data.symbol_callers)
|
|
164
|
+
? sanitizeStringArray(data.symbol_callers)
|
|
165
|
+
: [];
|
|
166
|
+
const symbolNeighbors = Array.isArray(data.symbol_neighbors)
|
|
167
|
+
? sanitizeStringArray(data.symbol_neighbors)
|
|
168
|
+
: [];
|
|
169
|
+
const symbol = typeof data.symbol === 'string' && data.symbol.trim().length > 0
|
|
170
|
+
? data.symbol
|
|
171
|
+
: null;
|
|
172
|
+
const normalized = {
|
|
173
|
+
...extras,
|
|
174
|
+
file,
|
|
175
|
+
symbol,
|
|
176
|
+
sha,
|
|
177
|
+
lang,
|
|
178
|
+
chunkType,
|
|
179
|
+
provider,
|
|
180
|
+
dimensions,
|
|
181
|
+
hasCodevaultTags,
|
|
182
|
+
hasIntent,
|
|
183
|
+
hasDocumentation,
|
|
184
|
+
variableCount,
|
|
185
|
+
synonyms,
|
|
186
|
+
path_weight: pathWeight,
|
|
187
|
+
success_rate: successRate,
|
|
188
|
+
encrypted,
|
|
189
|
+
symbol_calls: symbolCalls,
|
|
190
|
+
symbol_call_targets: symbolCallTargets,
|
|
191
|
+
symbol_callers: symbolCallers,
|
|
192
|
+
symbol_neighbors: symbolNeighbors
|
|
193
|
+
};
|
|
194
|
+
if (lastUsed) {
|
|
195
|
+
normalized.last_used_at = lastUsed;
|
|
196
|
+
}
|
|
197
|
+
if (symbolSignature) {
|
|
198
|
+
normalized.symbol_signature = symbolSignature;
|
|
199
|
+
}
|
|
200
|
+
if (symbolParameters.length > 0) {
|
|
201
|
+
normalized.symbol_parameters = symbolParameters;
|
|
202
|
+
}
|
|
203
|
+
if (symbolReturn) {
|
|
204
|
+
normalized.symbol_return = symbolReturn;
|
|
205
|
+
}
|
|
206
|
+
return normalized;
|
|
207
|
+
}
|
|
208
|
+
export function normalizeChunkMetadata(raw, previous) {
|
|
209
|
+
const base = previous ? internalNormalize(previous) : undefined;
|
|
210
|
+
const incoming = raw && typeof raw === 'object' ? raw : {};
|
|
211
|
+
const merged = base ? { ...base, ...incoming } : incoming;
|
|
212
|
+
return internalNormalize(merged);
|
|
213
|
+
}
|
|
214
|
+
export function normalizeCodemapRecord(raw) {
|
|
215
|
+
if (!raw || typeof raw !== 'object') {
|
|
216
|
+
return {};
|
|
217
|
+
}
|
|
218
|
+
const entries = Object.entries(raw)
|
|
219
|
+
.filter(([key]) => typeof key === 'string' && key.length > 0)
|
|
220
|
+
.map(([chunkId, value]) => [chunkId, normalizeChunkMetadata(value)]);
|
|
221
|
+
entries.sort(([a], [b]) => a.localeCompare(b));
|
|
222
|
+
return Object.fromEntries(entries);
|
|
223
|
+
}
|
|
224
|
+
//# sourceMappingURL=codemap.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"codemap.js","sourceRoot":"","sources":["../../src/types/codemap.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC;AACrC,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,CAAC;AAEtC,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IACzC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;IAChB,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE;IAClD,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;IACf,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC3B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAChC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC/B,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACjC,gBAAgB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IACxC,SAAS,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IACjC,gBAAgB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IACxC,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACpC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IACxC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE;IACxD,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACnC,SAAS,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IACjC,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACvC,iBAAiB,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IACjD,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACpC,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IAC5C,mBAAmB,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IACnD,cAAc,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IAC9C,gBAAgB,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;CACjD,CAAC,CAAC,WAAW,EAAE,CAAC;AAEjB,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,kBAAkB,CAAC,CAAC;AAKtE,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC;IAC3B,MAAM;IACN,QAAQ;IACR,KAAK;IACL,MAAM;IACN,WAAW;IACX,UAAU;IACV,YAAY;IACZ,kBAAkB;IAClB,WAAW;IACX,kBAAkB;IAClB,eAAe;IACf,UAAU;IACV,aAAa;IACb,cAAc;IACd,cAAc;IACd,kBAAkB;IAClB,mBAAmB;IACnB,eAAe;IACf,cAAc;IACd,qBAAqB;IACrB,gBAAgB;IAChB,kBAAkB;IAClB,WAAW;CACZ,CAAC,CAAC;AAEH,SAAS,mBAAmB,CAAC,KAAU,EAAE,UAAmC,EAAE;IAC5E,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAC;IACjC,KAAK,MAAM,KAAK,IAAI,KAAK,EAAE,CAAC;QAC1B,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,SAAS;QACX,CAAC;QACD,IAAI,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;QAC3B,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,SAAS;QACX,CAAC;QACD,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YACtB,OAAO,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QAClC,CAAC;QACD,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACtB,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;AACrC,CAAC;AAED,SAAS,sBAAsB,CAAC,KAAU;IACxC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IAC7B,OAAO,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;AAClD,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAU;IACpC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACzD,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IACD,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;QACd,OAAO,CAAC,CAAC;IACX,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAU;IACrC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;QACrD,OAAO,oBAAoB,CAAC;IAC9B,CAAC;IACD,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;QACd,OAAO,CAAC,CAAC;IACX,CAAC;IACD,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;QACd,OAAO,CAAC,CAAC;IACX,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAU;IAClC,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,IAAI,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IACjE,IAAI,CAAC,CAAC,IAAI,YAAY,IAAI,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;QAC5D,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC;AAC5B,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAU;IACvC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACzD,OAAO,CAAC,CAAC;IACX,CAAC;IACD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAClC,OAAO,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;AACnC,CAAC;AAED,SAAS,aAAa,CAAC,MAAW;IAChC,MAAM,MAAM,GAAwB,EAAE,CAAC;IACvC,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC1C,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAChD,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAC3B,MAAM,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC;QACpB,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,iBAAiB,CAAC,GAAQ;IACjC,MAAM,QAAQ,GAAG,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3D,MAAM,MAAM,GAAG,kBAAkB,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACtD,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC;IACrD,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;IAEnC,MAAM,IAAI,GAAG,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IAClG,MAAM,GAAG,GAAG,OAAO,IAAI,CAAC,GAAG,KAAK,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9F,MAAM,IAAI,GAAG,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IAClG,MAAM,SAAS,GAAG,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;IACtH,MAAM,QAAQ,GAAG,OAAO,IAAI,CAAC,QAAQ,KAAK,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;IAClH,MAAM,UAAU,GAAG,OAAO,IAAI,CAAC,UAAU,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;IACzH,MAAM,gBAAgB,GAAG,OAAO,IAAI,CAAC,gBAAgB,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAK,CAAC;IACpG,MAAM,SAAS,GAAG,OAAO,IAAI,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC;IAC/E,MAAM,gBAAgB,GAAG,OAAO,IAAI,CAAC,gBAAgB,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAK,CAAC;IACpG,MAAM,aAAa,GAAG,qBAAqB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAChE,MAAM,QAAQ,GAAG,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACpD,MAAM,UAAU,GAAG,kBAAkB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACxD,MAAM,QAAQ,GAAG,gBAAgB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACrD,MAAM,WAAW,GAAG,mBAAmB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC3D,MAAM,SAAS,GAAG,OAAO,IAAI,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC;IAC/E,MAAM,eAAe,GAAG,sBAAsB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACtE,MAAM,gBAAgB,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,CAAC;QAC5D,CAAC,CAAC,mBAAmB,CAAC,IAAI,CAAC,iBAAiB,CAAC;QAC7C,CAAC,CAAC,EAAE,CAAC;IACP,MAAM,YAAY,GAAG,sBAAsB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAChE,MAAM,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC;QAClD,CAAC,CAAC,mBAAmB,CAAC,IAAI,CAAC,YAAY,CAAC;QACxC,CAAC,CAAC,EAAE,CAAC;IACP,MAAM,iBAAiB,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,mBAAmB,CAAC;QAC/D,CAAC,CAAC,mBAAmB,CAAC,IAAI,CAAC,mBAAmB,CAAC;QAC/C,CAAC,CAAC,EAAE,CAAC;IACP,MAAM,aAAa,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC;QACtD,CAAC,CAAC,mBAAmB,CAAC,IAAI,CAAC,cAAc,CAAC;QAC1C,CAAC,CAAC,EAAE,CAAC;IACP,MAAM,eAAe,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC;QAC1D,CAAC,CAAC,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,CAAC;QAC5C,CAAC,CAAC,EAAE,CAAC;IAEP,MAAM,MAAM,GAAG,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC;QAC7E,CAAC,CAAC,IAAI,CAAC,MAAM;QACb,CAAC,CAAC,IAAI,CAAC;IAET,MAAM,UAAU,GAAQ;QACtB,GAAG,MAAM;QACT,IAAI;QACJ,MAAM;QACN,GAAG;QACH,IAAI;QACJ,SAAS;QACT,QAAQ;QACR,UAAU;QACV,gBAAgB;QAChB,SAAS;QACT,gBAAgB;QAChB,aAAa;QACb,QAAQ;QACR,WAAW,EAAE,UAAU;QACvB,YAAY,EAAE,WAAW;QACzB,SAAS;QACT,YAAY,EAAE,WAAW;QACzB,mBAAmB,EAAE,iBAAiB;QACtC,cAAc,EAAE,aAAa;QAC7B,gBAAgB,EAAE,eAAe;KAClC,CAAC;IAEF,IAAI,QAAQ,EAAE,CAAC;QACb,UAAU,CAAC,YAAY,GAAG,QAAQ,CAAC;IACrC,CAAC;IAED,IAAI,eAAe,EAAE,CAAC;QACpB,UAAU,CAAC,gBAAgB,GAAG,eAAe,CAAC;IAChD,CAAC;IAED,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChC,UAAU,CAAC,iBAAiB,GAAG,gBAAgB,CAAC;IAClD,CAAC;IAED,IAAI,YAAY,EAAE,CAAC;QACjB,UAAU,CAAC,aAAa,GAAG,YAAY,CAAC;IAC1C,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,GAAQ,EAAE,QAAuB;IACtE,MAAM,IAAI,GAAG,QAAQ,CAAC,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAChE,MAAM,QAAQ,GAAG,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3D,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,QAAQ,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC;IAC1D,OAAO,iBAAiB,CAAC,MAAM,CAAC,CAAC;AACnC,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,GAAQ;IAC7C,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;QACpC,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC;SAChC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC;SAC5D,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,sBAAsB,CAAC,KAAK,CAAC,CAA2B,CAAC,CAAC;IAEjG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;IAE/C,OAAO,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;AACrC,CAAC"}
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
export declare const ContextPackScopeSchema: z.ZodObject<{
|
|
3
|
+
path_glob: z.ZodOptional<z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>>;
|
|
4
|
+
tags: z.ZodOptional<z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>>;
|
|
5
|
+
lang: z.ZodOptional<z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>>;
|
|
6
|
+
provider: z.ZodOptional<z.ZodString>;
|
|
7
|
+
reranker: z.ZodOptional<z.ZodEnum<{
|
|
8
|
+
off: "off";
|
|
9
|
+
api: "api";
|
|
10
|
+
}>>;
|
|
11
|
+
hybrid: z.ZodOptional<z.ZodUnion<readonly [z.ZodBoolean, z.ZodString]>>;
|
|
12
|
+
bm25: z.ZodOptional<z.ZodUnion<readonly [z.ZodBoolean, z.ZodString]>>;
|
|
13
|
+
symbol_boost: z.ZodOptional<z.ZodUnion<readonly [z.ZodBoolean, z.ZodString]>>;
|
|
14
|
+
}, z.core.$strict>;
|
|
15
|
+
export declare const ContextPackSchema: z.ZodObject<{
|
|
16
|
+
name: z.ZodOptional<z.ZodString>;
|
|
17
|
+
description: z.ZodOptional<z.ZodString>;
|
|
18
|
+
metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
|
|
19
|
+
scope: z.ZodOptional<z.ZodObject<{
|
|
20
|
+
path_glob: z.ZodOptional<z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>>;
|
|
21
|
+
tags: z.ZodOptional<z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>>;
|
|
22
|
+
lang: z.ZodOptional<z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>>;
|
|
23
|
+
provider: z.ZodOptional<z.ZodString>;
|
|
24
|
+
reranker: z.ZodOptional<z.ZodEnum<{
|
|
25
|
+
off: "off";
|
|
26
|
+
api: "api";
|
|
27
|
+
}>>;
|
|
28
|
+
hybrid: z.ZodOptional<z.ZodUnion<readonly [z.ZodBoolean, z.ZodString]>>;
|
|
29
|
+
bm25: z.ZodOptional<z.ZodUnion<readonly [z.ZodBoolean, z.ZodString]>>;
|
|
30
|
+
symbol_boost: z.ZodOptional<z.ZodUnion<readonly [z.ZodBoolean, z.ZodString]>>;
|
|
31
|
+
}, z.core.$strict>>;
|
|
32
|
+
path_glob: z.ZodOptional<z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>>;
|
|
33
|
+
tags: z.ZodOptional<z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>>;
|
|
34
|
+
lang: z.ZodOptional<z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>>;
|
|
35
|
+
provider: z.ZodOptional<z.ZodString>;
|
|
36
|
+
reranker: z.ZodOptional<z.ZodEnum<{
|
|
37
|
+
off: "off";
|
|
38
|
+
api: "api";
|
|
39
|
+
}>>;
|
|
40
|
+
hybrid: z.ZodOptional<z.ZodUnion<readonly [z.ZodBoolean, z.ZodString]>>;
|
|
41
|
+
bm25: z.ZodOptional<z.ZodUnion<readonly [z.ZodBoolean, z.ZodString]>>;
|
|
42
|
+
symbol_boost: z.ZodOptional<z.ZodUnion<readonly [z.ZodBoolean, z.ZodString]>>;
|
|
43
|
+
}, z.core.$strict>;
|
|
44
|
+
export type ContextPackScope = z.infer<typeof ContextPackScopeSchema>;
|
|
45
|
+
export type ContextPack = z.infer<typeof ContextPackSchema>;
|
|
46
|
+
export declare function extractScopeFromPackDefinition(definition: ContextPack): Record<string, any>;
|
|
47
|
+
//# sourceMappingURL=context-pack.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"context-pack.d.ts","sourceRoot":"","sources":["../../src/types/context-pack.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAMxB,eAAO,MAAM,sBAAsB;;;;;;;;;;;;kBASxB,CAAC;AAEZ,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;kBAanB,CAAC;AAEZ,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AACtE,MAAM,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,CAAC;AAE5D,wBAAgB,8BAA8B,CAAC,UAAU,EAAE,WAAW,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAkB3F"}
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import { RERANKER_OPTIONS } from './search.js';
|
|
3
|
+
const stringOrStringArray = z.union([z.string(), z.array(z.string())]);
|
|
4
|
+
const booleanLike = z.union([z.boolean(), z.string()]);
|
|
5
|
+
export const ContextPackScopeSchema = z.object({
|
|
6
|
+
path_glob: stringOrStringArray.optional(),
|
|
7
|
+
tags: stringOrStringArray.optional(),
|
|
8
|
+
lang: stringOrStringArray.optional(),
|
|
9
|
+
provider: z.string().optional(),
|
|
10
|
+
reranker: z.enum(RERANKER_OPTIONS).optional(),
|
|
11
|
+
hybrid: booleanLike.optional(),
|
|
12
|
+
bm25: booleanLike.optional(),
|
|
13
|
+
symbol_boost: booleanLike.optional()
|
|
14
|
+
}).strict();
|
|
15
|
+
export const ContextPackSchema = z.object({
|
|
16
|
+
name: z.string().optional(),
|
|
17
|
+
description: z.string().optional(),
|
|
18
|
+
metadata: z.record(z.string(), z.any()).optional(),
|
|
19
|
+
scope: ContextPackScopeSchema.optional(),
|
|
20
|
+
path_glob: stringOrStringArray.optional(),
|
|
21
|
+
tags: stringOrStringArray.optional(),
|
|
22
|
+
lang: stringOrStringArray.optional(),
|
|
23
|
+
provider: z.string().optional(),
|
|
24
|
+
reranker: z.enum(RERANKER_OPTIONS).optional(),
|
|
25
|
+
hybrid: booleanLike.optional(),
|
|
26
|
+
bm25: booleanLike.optional(),
|
|
27
|
+
symbol_boost: booleanLike.optional()
|
|
28
|
+
}).strict();
|
|
29
|
+
export function extractScopeFromPackDefinition(definition) {
|
|
30
|
+
if (!definition || typeof definition !== 'object') {
|
|
31
|
+
return {};
|
|
32
|
+
}
|
|
33
|
+
const scopeCandidate = definition.scope && typeof definition.scope === 'object'
|
|
34
|
+
? { ...definition.scope }
|
|
35
|
+
: {};
|
|
36
|
+
const scope = { ...scopeCandidate };
|
|
37
|
+
for (const key of ['path_glob', 'tags', 'lang', 'provider', 'reranker', 'hybrid', 'bm25', 'symbol_boost']) {
|
|
38
|
+
if (Object.prototype.hasOwnProperty.call(definition, key) && typeof definition[key] !== 'undefined') {
|
|
39
|
+
scope[key] = definition[key];
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
return scope;
|
|
43
|
+
}
|
|
44
|
+
//# sourceMappingURL=context-pack.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"context-pack.js","sourceRoot":"","sources":["../../src/types/context-pack.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAE/C,MAAM,mBAAmB,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC;AACvE,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;AAEvD,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7C,SAAS,EAAE,mBAAmB,CAAC,QAAQ,EAAE;IACzC,IAAI,EAAE,mBAAmB,CAAC,QAAQ,EAAE;IACpC,IAAI,EAAE,mBAAmB,CAAC,QAAQ,EAAE;IACpC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC/B,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,QAAQ,EAAE;IAC7C,MAAM,EAAE,WAAW,CAAC,QAAQ,EAAE;IAC9B,IAAI,EAAE,WAAW,CAAC,QAAQ,EAAE;IAC5B,YAAY,EAAE,WAAW,CAAC,QAAQ,EAAE;CACrC,CAAC,CAAC,MAAM,EAAE,CAAC;AAEZ,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC;IACxC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC3B,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,QAAQ,EAAE;IAClD,KAAK,EAAE,sBAAsB,CAAC,QAAQ,EAAE;IACxC,SAAS,EAAE,mBAAmB,CAAC,QAAQ,EAAE;IACzC,IAAI,EAAE,mBAAmB,CAAC,QAAQ,EAAE;IACpC,IAAI,EAAE,mBAAmB,CAAC,QAAQ,EAAE;IACpC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC/B,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,QAAQ,EAAE;IAC7C,MAAM,EAAE,WAAW,CAAC,QAAQ,EAAE;IAC9B,IAAI,EAAE,WAAW,CAAC,QAAQ,EAAE;IAC5B,YAAY,EAAE,WAAW,CAAC,QAAQ,EAAE;CACrC,CAAC,CAAC,MAAM,EAAE,CAAC;AAKZ,MAAM,UAAU,8BAA8B,CAAC,UAAuB;IACpE,IAAI,CAAC,UAAU,IAAI,OAAO,UAAU,KAAK,QAAQ,EAAE,CAAC;QAClD,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,cAAc,GAAG,UAAU,CAAC,KAAK,IAAI,OAAO,UAAU,CAAC,KAAK,KAAK,QAAQ;QAC7E,CAAC,CAAC,EAAE,GAAG,UAAU,CAAC,KAAK,EAAE;QACzB,CAAC,CAAC,EAAE,CAAC;IAEP,MAAM,KAAK,GAAG,EAAE,GAAG,cAAc,EAAE,CAAC;IAEpC,KAAK,MAAM,GAAG,IAAI,CAAC,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,cAAc,CAAC,EAAE,CAAC;QAC1G,IAAI,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,CAAC,IAAI,OAAQ,UAAkB,CAAC,GAAG,CAAC,KAAK,WAAW,EAAE,CAAC;YAC5G,KAAa,CAAC,GAAG,CAAC,GAAI,UAAkB,CAAC,GAAG,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
export declare const RERANKER_OPTIONS: readonly ["off", "api"];
|
|
2
|
+
export declare const DEFAULT_RERANKER = "off";
|
|
3
|
+
export type RerankMode = typeof RERANKER_OPTIONS[number];
|
|
4
|
+
export interface ScopeFilters {
|
|
5
|
+
path_glob?: string[];
|
|
6
|
+
tags?: string[];
|
|
7
|
+
lang?: string[];
|
|
8
|
+
provider?: string;
|
|
9
|
+
reranker?: RerankMode;
|
|
10
|
+
hybrid?: boolean;
|
|
11
|
+
bm25?: boolean;
|
|
12
|
+
symbol_boost?: boolean;
|
|
13
|
+
}
|
|
14
|
+
export declare function hasScopeFilters(scope?: ScopeFilters): boolean;
|
|
15
|
+
//# sourceMappingURL=search.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"search.d.ts","sourceRoot":"","sources":["../../src/types/search.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,gBAAgB,yBAA0B,CAAC;AACxD,eAAO,MAAM,gBAAgB,QAAQ,CAAC;AAEtC,MAAM,MAAM,UAAU,GAAG,OAAO,gBAAgB,CAAC,MAAM,CAAC,CAAC;AAEzD,MAAM,WAAW,YAAY;IAC3B,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,UAAU,CAAC;IACtB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,IAAI,CAAC,EAAE,OAAO,CAAC;IACf,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB;AAED,wBAAgB,eAAe,CAAC,KAAK,CAAC,EAAE,YAAY,GAAG,OAAO,CAU7D"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
export const RERANKER_OPTIONS = ['off', 'api'];
|
|
2
|
+
export const DEFAULT_RERANKER = 'off';
|
|
3
|
+
export function hasScopeFilters(scope) {
|
|
4
|
+
if (!scope) {
|
|
5
|
+
return false;
|
|
6
|
+
}
|
|
7
|
+
return Boolean((Array.isArray(scope.path_glob) && scope.path_glob.length > 0) ||
|
|
8
|
+
(Array.isArray(scope.tags) && scope.tags.length > 0) ||
|
|
9
|
+
(Array.isArray(scope.lang) && scope.lang.length > 0));
|
|
10
|
+
}
|
|
11
|
+
//# sourceMappingURL=search.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"search.js","sourceRoot":"","sources":["../../src/types/search.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,KAAK,EAAE,KAAK,CAAU,CAAC;AACxD,MAAM,CAAC,MAAM,gBAAgB,GAAG,KAAK,CAAC;AAetC,MAAM,UAAU,eAAe,CAAC,KAAoB;IAClD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,OAAO,CACZ,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC;QAC9D,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;QACpD,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CACrD,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
export declare class RateLimiter {
|
|
2
|
+
private rpm;
|
|
3
|
+
private tpm;
|
|
4
|
+
private queue;
|
|
5
|
+
private processing;
|
|
6
|
+
private requestTimes;
|
|
7
|
+
private tokenUsage;
|
|
8
|
+
private retryDelays;
|
|
9
|
+
constructor(requestsPerMinute?: number | null, tokensPerMinute?: number | null);
|
|
10
|
+
private getDefaultRPM;
|
|
11
|
+
private getDefaultTPM;
|
|
12
|
+
private canMakeRequest;
|
|
13
|
+
private recordRequest;
|
|
14
|
+
private getDelayUntilNextSlot;
|
|
15
|
+
execute<T>(fn: () => Promise<T>, retryCount?: number, estimatedTokens?: number): Promise<T>;
|
|
16
|
+
private processQueue;
|
|
17
|
+
private isRateLimitError;
|
|
18
|
+
getStats(): {
|
|
19
|
+
rpm: number | null;
|
|
20
|
+
tpm: number | null;
|
|
21
|
+
queueLength: number;
|
|
22
|
+
requestsInLastMinute: number;
|
|
23
|
+
tokensInLastMinute: number;
|
|
24
|
+
isRpmLimited: boolean;
|
|
25
|
+
isTpmLimited: boolean;
|
|
26
|
+
isLimited: boolean;
|
|
27
|
+
};
|
|
28
|
+
reset(): void;
|
|
29
|
+
}
|
|
30
|
+
export declare function createRateLimiter(providerName: string): RateLimiter;
|
|
31
|
+
//# sourceMappingURL=rate-limiter.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rate-limiter.d.ts","sourceRoot":"","sources":["../../src/utils/rate-limiter.ts"],"names":[],"mappings":"AAaA,qBAAa,WAAW;IACtB,OAAO,CAAC,GAAG,CAAgB;IAC3B,OAAO,CAAC,GAAG,CAAgB;IAC3B,OAAO,CAAC,KAAK,CAAwB;IACrC,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,YAAY,CAAgB;IACpC,OAAO,CAAC,UAAU,CAAyB;IAC3C,OAAO,CAAC,WAAW,CAA6B;gBAEpC,iBAAiB,GAAE,MAAM,GAAG,IAAW,EAAE,eAAe,GAAE,MAAM,GAAG,IAAW;IAK1F,OAAO,CAAC,aAAa;IAWrB,OAAO,CAAC,aAAa;IAUrB,OAAO,CAAC,cAAc;IAqBtB,OAAO,CAAC,aAAa;IAYrB,OAAO,CAAC,qBAAqB;IAsBvB,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EAAE,UAAU,SAAI,EAAE,eAAe,SAAI,GAAG,OAAO,CAAC,CAAC,CAAC;YAOzE,YAAY;IAkD1B,OAAO,CAAC,gBAAgB;IAYxB,QAAQ;;;;;;;;;;IAmBR,KAAK,IAAI,IAAI;CAMd;AAED,wBAAgB,iBAAiB,CAAC,YAAY,EAAE,MAAM,GAAG,WAAW,CAanE"}
|
|
@@ -0,0 +1,168 @@
|
|
|
1
|
+
export class RateLimiter {
|
|
2
|
+
rpm;
|
|
3
|
+
tpm;
|
|
4
|
+
queue = [];
|
|
5
|
+
processing = false;
|
|
6
|
+
requestTimes = [];
|
|
7
|
+
tokenUsage = [];
|
|
8
|
+
retryDelays = [1000, 2000, 5000, 10000];
|
|
9
|
+
constructor(requestsPerMinute = null, tokensPerMinute = null) {
|
|
10
|
+
this.rpm = requestsPerMinute ?? this.getDefaultRPM();
|
|
11
|
+
this.tpm = tokensPerMinute ?? this.getDefaultTPM();
|
|
12
|
+
}
|
|
13
|
+
getDefaultRPM() {
|
|
14
|
+
const envVar = process.env.CODEVAULT_RATE_LIMIT_RPM || process.env.CODEVAULT_RATE_LIMIT;
|
|
15
|
+
if (envVar) {
|
|
16
|
+
const limit = parseInt(envVar, 10);
|
|
17
|
+
if (!isNaN(limit) && limit > 0) {
|
|
18
|
+
return limit;
|
|
19
|
+
}
|
|
20
|
+
}
|
|
21
|
+
return null;
|
|
22
|
+
}
|
|
23
|
+
getDefaultTPM() {
|
|
24
|
+
if (process.env.CODEVAULT_RATE_LIMIT_TPM) {
|
|
25
|
+
const limit = parseInt(process.env.CODEVAULT_RATE_LIMIT_TPM, 10);
|
|
26
|
+
if (!isNaN(limit) && limit > 0) {
|
|
27
|
+
return limit;
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
return null;
|
|
31
|
+
}
|
|
32
|
+
canMakeRequest(estimatedTokens = 0) {
|
|
33
|
+
const now = Date.now();
|
|
34
|
+
const oneMinuteAgo = now - 60000;
|
|
35
|
+
this.requestTimes = this.requestTimes.filter(time => time > oneMinuteAgo);
|
|
36
|
+
this.tokenUsage = this.tokenUsage.filter(entry => entry.time > oneMinuteAgo);
|
|
37
|
+
if (this.rpm !== null && this.requestTimes.length >= this.rpm) {
|
|
38
|
+
return false;
|
|
39
|
+
}
|
|
40
|
+
if (this.tpm !== null && estimatedTokens > 0) {
|
|
41
|
+
const tokensInLastMinute = this.tokenUsage.reduce((sum, entry) => sum + entry.tokens, 0);
|
|
42
|
+
if (tokensInLastMinute + estimatedTokens > this.tpm) {
|
|
43
|
+
return false;
|
|
44
|
+
}
|
|
45
|
+
}
|
|
46
|
+
return true;
|
|
47
|
+
}
|
|
48
|
+
recordRequest(tokensUsed = 0) {
|
|
49
|
+
const now = Date.now();
|
|
50
|
+
if (this.rpm !== null) {
|
|
51
|
+
this.requestTimes.push(now);
|
|
52
|
+
}
|
|
53
|
+
if (this.tpm !== null && tokensUsed > 0) {
|
|
54
|
+
this.tokenUsage.push({ time: now, tokens: tokensUsed });
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
getDelayUntilNextSlot(estimatedTokens = 0) {
|
|
58
|
+
const now = Date.now();
|
|
59
|
+
let delay = 0;
|
|
60
|
+
if (this.rpm !== null && this.requestTimes.length > 0) {
|
|
61
|
+
const oldestRequest = this.requestTimes[0];
|
|
62
|
+
const rpmDelay = 60000 - (now - oldestRequest);
|
|
63
|
+
delay = Math.max(delay, rpmDelay);
|
|
64
|
+
}
|
|
65
|
+
if (this.tpm !== null && estimatedTokens > 0 && this.tokenUsage.length > 0) {
|
|
66
|
+
const tokensInLastMinute = this.tokenUsage.reduce((sum, entry) => sum + entry.tokens, 0);
|
|
67
|
+
if (tokensInLastMinute + estimatedTokens > this.tpm) {
|
|
68
|
+
const oldestToken = this.tokenUsage[0];
|
|
69
|
+
const tpmDelay = 60000 - (now - oldestToken.time);
|
|
70
|
+
delay = Math.max(delay, tpmDelay);
|
|
71
|
+
}
|
|
72
|
+
}
|
|
73
|
+
return Math.max(0, delay + 100);
|
|
74
|
+
}
|
|
75
|
+
async execute(fn, retryCount = 0, estimatedTokens = 0) {
|
|
76
|
+
return new Promise((resolve, reject) => {
|
|
77
|
+
this.queue.push({ fn, resolve, reject, retryCount, estimatedTokens });
|
|
78
|
+
void this.processQueue();
|
|
79
|
+
});
|
|
80
|
+
}
|
|
81
|
+
async processQueue() {
|
|
82
|
+
if (this.processing || this.queue.length === 0) {
|
|
83
|
+
return;
|
|
84
|
+
}
|
|
85
|
+
this.processing = true;
|
|
86
|
+
while (this.queue.length > 0) {
|
|
87
|
+
const item = this.queue[0];
|
|
88
|
+
const { fn, resolve, reject, retryCount, estimatedTokens } = item;
|
|
89
|
+
while (!this.canMakeRequest(estimatedTokens || 0)) {
|
|
90
|
+
const delay = this.getDelayUntilNextSlot(estimatedTokens || 0);
|
|
91
|
+
if (delay > 0) {
|
|
92
|
+
await new Promise(r => setTimeout(r, delay));
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
this.queue.shift();
|
|
96
|
+
try {
|
|
97
|
+
const result = await fn();
|
|
98
|
+
const tokensUsed = estimatedTokens || result?.usage?.total_tokens || 0;
|
|
99
|
+
this.recordRequest(tokensUsed);
|
|
100
|
+
resolve(result);
|
|
101
|
+
}
|
|
102
|
+
catch (error) {
|
|
103
|
+
if (this.isRateLimitError(error)) {
|
|
104
|
+
const maxRetries = this.retryDelays.length;
|
|
105
|
+
if (retryCount < maxRetries) {
|
|
106
|
+
const delay = this.retryDelays[retryCount];
|
|
107
|
+
console.warn(`⚠️ Rate limit hit (429). Retrying in ${delay}ms... (attempt ${retryCount + 1}/${maxRetries})`);
|
|
108
|
+
await new Promise(r => setTimeout(r, delay));
|
|
109
|
+
this.queue.unshift({ fn, resolve, reject, retryCount: retryCount + 1, estimatedTokens });
|
|
110
|
+
}
|
|
111
|
+
else {
|
|
112
|
+
reject(new Error(`Rate limit exceeded after ${maxRetries} retries: ${error.message}`));
|
|
113
|
+
}
|
|
114
|
+
}
|
|
115
|
+
else {
|
|
116
|
+
reject(error);
|
|
117
|
+
}
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
this.processing = false;
|
|
121
|
+
}
|
|
122
|
+
isRateLimitError(error) {
|
|
123
|
+
if (!error)
|
|
124
|
+
return false;
|
|
125
|
+
const message = error.message || '';
|
|
126
|
+
const status = error.status || error.statusCode || 0;
|
|
127
|
+
return status === 429 ||
|
|
128
|
+
message.includes('429') ||
|
|
129
|
+
message.includes('rate limit') ||
|
|
130
|
+
message.includes('too many requests');
|
|
131
|
+
}
|
|
132
|
+
getStats() {
|
|
133
|
+
const now = Date.now();
|
|
134
|
+
const oneMinuteAgo = now - 60000;
|
|
135
|
+
const tokensInLastMinute = this.tokenUsage
|
|
136
|
+
.filter(entry => entry.time > oneMinuteAgo)
|
|
137
|
+
.reduce((sum, entry) => sum + entry.tokens, 0);
|
|
138
|
+
return {
|
|
139
|
+
rpm: this.rpm,
|
|
140
|
+
tpm: this.tpm,
|
|
141
|
+
queueLength: this.queue.length,
|
|
142
|
+
requestsInLastMinute: this.requestTimes.filter(t => t > oneMinuteAgo).length,
|
|
143
|
+
tokensInLastMinute,
|
|
144
|
+
isRpmLimited: this.rpm !== null,
|
|
145
|
+
isTpmLimited: this.tpm !== null,
|
|
146
|
+
isLimited: this.rpm !== null || this.tpm !== null
|
|
147
|
+
};
|
|
148
|
+
}
|
|
149
|
+
reset() {
|
|
150
|
+
this.queue = [];
|
|
151
|
+
this.requestTimes = [];
|
|
152
|
+
this.tokenUsage = [];
|
|
153
|
+
this.processing = false;
|
|
154
|
+
}
|
|
155
|
+
}
|
|
156
|
+
export function createRateLimiter(providerName) {
|
|
157
|
+
const defaultLimits = {
|
|
158
|
+
'OpenAI': { rpm: 50, tpm: null },
|
|
159
|
+
'Ollama': { rpm: null, tpm: null },
|
|
160
|
+
'Qwen': { rpm: 10000, tpm: 600000 },
|
|
161
|
+
};
|
|
162
|
+
if (process.env.CODEVAULT_RATE_LIMIT_RPM || process.env.CODEVAULT_RATE_LIMIT || process.env.CODEVAULT_RATE_LIMIT_TPM) {
|
|
163
|
+
return new RateLimiter();
|
|
164
|
+
}
|
|
165
|
+
const limits = defaultLimits[providerName] ?? { rpm: null, tpm: null };
|
|
166
|
+
return new RateLimiter(limits.rpm, limits.tpm);
|
|
167
|
+
}
|
|
168
|
+
//# sourceMappingURL=rate-limiter.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rate-limiter.js","sourceRoot":"","sources":["../../src/utils/rate-limiter.ts"],"names":[],"mappings":"AAaA,MAAM,OAAO,WAAW;IACd,GAAG,CAAgB;IACnB,GAAG,CAAgB;IACnB,KAAK,GAAqB,EAAE,CAAC;IAC7B,UAAU,GAAG,KAAK,CAAC;IACnB,YAAY,GAAa,EAAE,CAAC;IAC5B,UAAU,GAAsB,EAAE,CAAC;IACnC,WAAW,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;IAEhD,YAAY,oBAAmC,IAAI,EAAE,kBAAiC,IAAI;QACxF,IAAI,CAAC,GAAG,GAAG,iBAAiB,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;QACrD,IAAI,CAAC,GAAG,GAAG,eAAe,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;IACrD,CAAC;IAEO,aAAa;QACnB,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;QACxF,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACnC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;gBAC/B,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,aAAa;QACnB,IAAI,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,CAAC;YACzC,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,EAAE,CAAC,CAAC;YACjE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;gBAC/B,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,cAAc,CAAC,eAAe,GAAG,CAAC;QACxC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,YAAY,GAAG,GAAG,GAAG,KAAK,CAAC;QAEjC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,GAAG,YAAY,CAAC,CAAC;QAC1E,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,GAAG,YAAY,CAAC,CAAC;QAE7E,IAAI,IAAI,CAAC,GAAG,KAAK,IAAI,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YAC9D,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,IAAI,CAAC,GAAG,KAAK,IAAI,IAAI,eAAe,GAAG,CAAC,EAAE,CAAC;YAC7C,MAAM,kBAAkB,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACzF,IAAI,kBAAkB,GAAG,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACpD,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,aAAa,CAAC,UAAU,GAAG,CAAC;QAClC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,IAAI,IAAI,CAAC,GAAG,KAAK,IAAI,EAAE,CAAC;YACtB,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC;QAED,IAAI,IAAI,CAAC,GAAG,KAAK,IAAI,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;YACxC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAEO,qBAAqB,CAAC,eAAe,GAAG,CAAC;QAC/C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,IAAI,IAAI,CAAC,GAAG,KAAK,IAAI,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtD,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC3C,MAAM,QAAQ,GAAG,KAAK,GAAG,CAAC,GAAG,GAAG,aAAa,CAAC,CAAC;YAC/C,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QACpC,CAAC;QAED,IAAI,IAAI,CAAC,GAAG,KAAK,IAAI,IAAI,eAAe,GAAG,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3E,MAAM,kBAAkB,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACzF,IAAI,kBAAkB,GAAG,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACpD,MAAM,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;gBACvC,MAAM,QAAQ,GAAG,KAAK,GAAG,CAAC,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;gBAClD,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,GAAG,GAAG,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,OAAO,CAAI,EAAoB,EAAE,UAAU,GAAG,CAAC,EAAE,eAAe,GAAG,CAAC;QACxE,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;YACtE,KAAK,IAAI,CAAC,YAAY,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,YAAY;QACxB,IAAI,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/C,OAAO;QACT,CAAC;QAED,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QAEvB,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAC3B,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,eAAe,EAAE,GAAG,IAAI,CAAC;YAElE,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,eAAe,IAAI,CAAC,CAAC,EAAE,CAAC;gBAClD,MAAM,KAAK,GAAG,IAAI,CAAC,qBAAqB,CAAC,eAAe,IAAI,CAAC,CAAC,CAAC;gBAC/D,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;oBACd,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;gBAC/C,CAAC;YACH,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YAEnB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,EAAE,EAAE,CAAC;gBAE1B,MAAM,UAAU,GAAG,eAAe,IAAK,MAAc,EAAE,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC;gBAChF,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;gBAE/B,OAAO,CAAC,MAAM,CAAC,CAAC;YAClB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;oBACjC,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;oBAE3C,IAAI,UAAU,GAAG,UAAU,EAAE,CAAC;wBAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;wBAC3C,OAAO,CAAC,IAAI,CAAC,yCAAyC,KAAK,kBAAkB,UAAU,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,CAAC;wBAE9G,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;wBAE7C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,GAAG,CAAC,EAAE,eAAe,EAAE,CAAC,CAAC;oBAC3F,CAAC;yBAAM,CAAC;wBACN,MAAM,CAAC,IAAI,KAAK,CAAC,6BAA6B,UAAU,aAAc,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;oBACpG,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,MAAM,CAAC,KAAc,CAAC,CAAC;gBACzB,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;IAC1B,CAAC;IAEO,gBAAgB,CAAC,KAAc;QACrC,IAAI,CAAC,KAAK;YAAE,OAAO,KAAK,CAAC;QAEzB,MAAM,OAAO,GAAI,KAAa,CAAC,OAAO,IAAI,EAAE,CAAC;QAC7C,MAAM,MAAM,GAAI,KAAa,CAAC,MAAM,IAAK,KAAa,CAAC,UAAU,IAAI,CAAC,CAAC;QAEvE,OAAO,MAAM,KAAK,GAAG;YACd,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC;YACvB,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC;YAC9B,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC;IAC/C,CAAC;IAED,QAAQ;QACN,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,YAAY,GAAG,GAAG,GAAG,KAAK,CAAC;QACjC,MAAM,kBAAkB,GAAG,IAAI,CAAC,UAAU;aACvC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,GAAG,YAAY,CAAC;aAC1C,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAEjD,OAAO;YACL,GAAG,EAAE,IAAI,CAAC,GAAG;YACb,GAAG,EAAE,IAAI,CAAC,GAAG;YACb,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM;YAC9B,oBAAoB,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,YAAY,CAAC,CAAC,MAAM;YAC5E,kBAAkB;YAClB,YAAY,EAAE,IAAI,CAAC,GAAG,KAAK,IAAI;YAC/B,YAAY,EAAE,IAAI,CAAC,GAAG,KAAK,IAAI;YAC/B,SAAS,EAAE,IAAI,CAAC,GAAG,KAAK,IAAI,IAAI,IAAI,CAAC,GAAG,KAAK,IAAI;SAClD,CAAC;IACJ,CAAC;IAED,KAAK;QACH,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;QAChB,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;IAC1B,CAAC;CACF;AAED,MAAM,UAAU,iBAAiB,CAAC,YAAoB;IACpD,MAAM,aAAa,GAA+D;QAChF,QAAQ,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE;QAChC,QAAQ,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE;QAClC,MAAM,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE;KACpC,CAAC;IAEF,IAAI,OAAO,CAAC,GAAG,CAAC,wBAAwB,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,CAAC;QACrH,OAAO,IAAI,WAAW,EAAE,CAAC;IAC3B,CAAC;IAED,MAAM,MAAM,GAAG,aAAa,CAAC,YAAY,CAAC,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;IACvE,OAAO,IAAI,WAAW,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC;AACjD,CAAC"}
|