kiri-mcp-server 0.17.0 → 0.18.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/README.md +13 -6
- package/dist/package.json +4 -2
- package/dist/src/indexer/codeintel/dart/adapter.d.ts +36 -0
- package/dist/src/indexer/codeintel/dart/adapter.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/dart/adapter.js +60 -0
- package/dist/src/indexer/codeintel/dart/adapter.js.map +1 -0
- package/dist/src/indexer/codeintel/dart/index.d.ts +7 -0
- package/dist/src/indexer/codeintel/dart/index.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/dart/index.js +7 -0
- package/dist/src/indexer/codeintel/dart/index.js.map +1 -0
- package/dist/src/indexer/codeintel/index.d.ts +30 -0
- package/dist/src/indexer/codeintel/index.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/index.js +32 -0
- package/dist/src/indexer/codeintel/index.js.map +1 -0
- package/dist/src/indexer/codeintel/java/analyzer.d.ts +22 -0
- package/dist/src/indexer/codeintel/java/analyzer.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/java/analyzer.js +281 -0
- package/dist/src/indexer/codeintel/java/analyzer.js.map +1 -0
- package/dist/src/indexer/codeintel/java/index.d.ts +7 -0
- package/dist/src/indexer/codeintel/java/index.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/java/index.js +7 -0
- package/dist/src/indexer/codeintel/java/index.js.map +1 -0
- package/dist/src/indexer/codeintel/php/analyzer.d.ts +23 -0
- package/dist/src/indexer/codeintel/php/analyzer.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/php/analyzer.js +342 -0
- package/dist/src/indexer/codeintel/php/analyzer.js.map +1 -0
- package/dist/src/indexer/codeintel/php/index.d.ts +7 -0
- package/dist/src/indexer/codeintel/php/index.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/php/index.js +7 -0
- package/dist/src/indexer/codeintel/php/index.js.map +1 -0
- package/dist/src/indexer/codeintel/registry.d.ts +76 -0
- package/dist/src/indexer/codeintel/registry.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/registry.js +127 -0
- package/dist/src/indexer/codeintel/registry.js.map +1 -0
- package/dist/src/indexer/codeintel/rust/analyzer.d.ts +14 -0
- package/dist/src/indexer/codeintel/rust/analyzer.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/rust/analyzer.js +388 -0
- package/dist/src/indexer/codeintel/rust/analyzer.js.map +1 -0
- package/dist/src/indexer/codeintel/rust/index.d.ts +5 -0
- package/dist/src/indexer/codeintel/rust/index.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/rust/index.js +5 -0
- package/dist/src/indexer/codeintel/rust/index.js.map +1 -0
- package/dist/src/indexer/codeintel/swift/analyzer.d.ts +22 -0
- package/dist/src/indexer/codeintel/swift/analyzer.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/swift/analyzer.js +271 -0
- package/dist/src/indexer/codeintel/swift/analyzer.js.map +1 -0
- package/dist/src/indexer/codeintel/swift/index.d.ts +7 -0
- package/dist/src/indexer/codeintel/swift/index.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/swift/index.js +7 -0
- package/dist/src/indexer/codeintel/swift/index.js.map +1 -0
- package/dist/src/indexer/codeintel/types.d.ts +114 -0
- package/dist/src/indexer/codeintel/types.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/types.js +13 -0
- package/dist/src/indexer/codeintel/types.js.map +1 -0
- package/dist/src/indexer/codeintel/typescript/analyzer.d.ts +22 -0
- package/dist/src/indexer/codeintel/typescript/analyzer.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/typescript/analyzer.js +210 -0
- package/dist/src/indexer/codeintel/typescript/analyzer.js.map +1 -0
- package/dist/src/indexer/codeintel/typescript/index.d.ts +7 -0
- package/dist/src/indexer/codeintel/typescript/index.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/typescript/index.js +7 -0
- package/dist/src/indexer/codeintel/typescript/index.js.map +1 -0
- package/dist/src/indexer/codeintel/utils.d.ts +91 -0
- package/dist/src/indexer/codeintel/utils.d.ts.map +1 -0
- package/dist/src/indexer/codeintel/utils.js +145 -0
- package/dist/src/indexer/codeintel/utils.js.map +1 -0
- package/dist/src/indexer/codeintel.d.ts +33 -26
- package/dist/src/indexer/codeintel.d.ts.map +1 -1
- package/dist/src/indexer/codeintel.js +56 -1078
- package/dist/src/indexer/codeintel.js.map +1 -1
- package/package.json +10 -2
- package/dist/src/client/model-cli.d.ts +0 -12
- package/dist/src/client/model-cli.d.ts.map +0 -1
- package/dist/src/client/model-cli.js +0 -247
- package/dist/src/client/model-cli.js.map +0 -1
- package/dist/src/shared/utils/global-dirs.d.ts +0 -64
- package/dist/src/shared/utils/global-dirs.d.ts.map +0 -1
- package/dist/src/shared/utils/global-dirs.js +0 -134
- package/dist/src/shared/utils/global-dirs.js.map +0 -1
- package/dist/src/vss/config.d.ts +0 -66
- package/dist/src/vss/config.d.ts.map +0 -1
- package/dist/src/vss/config.js +0 -233
- package/dist/src/vss/config.js.map +0 -1
- package/dist/src/vss/index.d.ts +0 -9
- package/dist/src/vss/index.d.ts.map +0 -1
- package/dist/src/vss/index.js +0 -9
- package/dist/src/vss/index.js.map +0 -1
- package/dist/src/vss/model-downloader.d.ts +0 -101
- package/dist/src/vss/model-downloader.d.ts.map +0 -1
- package/dist/src/vss/model-downloader.js +0 -320
- package/dist/src/vss/model-downloader.js.map +0 -1
- package/dist/src/vss/provider.d.ts +0 -132
- package/dist/src/vss/provider.d.ts.map +0 -1
- package/dist/src/vss/provider.js +0 -254
- package/dist/src/vss/provider.js.map +0 -1
package/dist/src/vss/config.d.ts
DELETED
|
@@ -1,66 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* VSS (Vector Semantic Search) Configuration
|
|
3
|
-
*
|
|
4
|
-
* Handles loading and validation of VSS settings from config files and environment variables.
|
|
5
|
-
*/
|
|
6
|
-
/**
|
|
7
|
-
* Supported VSS models
|
|
8
|
-
*
|
|
9
|
-
* Model types:
|
|
10
|
-
* - "legacy": Original CodeBERT/UniXcoder models (requires manual ONNX conversion)
|
|
11
|
-
* - "transformers": Models with ONNX support via @huggingface/transformers
|
|
12
|
-
*/
|
|
13
|
-
export declare const SUPPORTED_MODELS: readonly ["codebert-base", "unixcoder-base", "jina-embeddings-v2-base-code", "all-MiniLM-L6-v2"];
|
|
14
|
-
export type SupportedModel = (typeof SUPPORTED_MODELS)[number];
|
|
15
|
-
/**
|
|
16
|
-
* Model metadata including HuggingFace model ID and embedding dimensions
|
|
17
|
-
*/
|
|
18
|
-
export interface ModelMetadata {
|
|
19
|
-
/** HuggingFace model ID */
|
|
20
|
-
hfModelId: string;
|
|
21
|
-
/** Embedding dimensions */
|
|
22
|
-
dims: number;
|
|
23
|
-
/** Whether the model has ONNX weights available */
|
|
24
|
-
hasOnnx: boolean;
|
|
25
|
-
/** Description */
|
|
26
|
-
description: string;
|
|
27
|
-
}
|
|
28
|
-
/**
|
|
29
|
-
* Model registry with metadata
|
|
30
|
-
*/
|
|
31
|
-
export declare const MODEL_METADATA: Record<SupportedModel, ModelMetadata>;
|
|
32
|
-
/**
|
|
33
|
-
* Behavior when model is not found
|
|
34
|
-
*/
|
|
35
|
-
export type OnMissingBehavior = "prompt" | "download" | "error" | "skip";
|
|
36
|
-
/**
|
|
37
|
-
* VSS configuration interface
|
|
38
|
-
*/
|
|
39
|
-
export interface VSSConfig {
|
|
40
|
-
/** Whether VSS is enabled */
|
|
41
|
-
enabled: boolean;
|
|
42
|
-
/** Model name to use */
|
|
43
|
-
model: SupportedModel;
|
|
44
|
-
/** Directory where models are stored */
|
|
45
|
-
modelDir: string;
|
|
46
|
-
/** Behavior when model is not found */
|
|
47
|
-
onMissing: OnMissingBehavior;
|
|
48
|
-
/** Batch size for embedding generation */
|
|
49
|
-
batchSize: number;
|
|
50
|
-
}
|
|
51
|
-
/**
|
|
52
|
-
* Default VSS configuration
|
|
53
|
-
*/
|
|
54
|
-
export declare const DEFAULT_VSS_CONFIG: VSSConfig;
|
|
55
|
-
/**
|
|
56
|
-
* Load complete VSS configuration
|
|
57
|
-
* Priority: Environment variables > Config file > Defaults
|
|
58
|
-
*
|
|
59
|
-
* @param overrides - Optional overrides (e.g., from CLI flags)
|
|
60
|
-
*/
|
|
61
|
-
export declare function loadVSSConfig(overrides?: Partial<VSSConfig>): VSSConfig;
|
|
62
|
-
/**
|
|
63
|
-
* Check if VSS should be used based on configuration and model availability
|
|
64
|
-
*/
|
|
65
|
-
export declare function shouldUseVSS(config: VSSConfig): boolean;
|
|
66
|
-
//# sourceMappingURL=config.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../../src/vss/config.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAQH;;;;;;GAMG;AACH,eAAO,MAAM,gBAAgB,kGAOnB,CAAC;AACX,MAAM,MAAM,cAAc,GAAG,CAAC,OAAO,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC;AAE/D;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,2BAA2B;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,2BAA2B;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,mDAAmD;IACnD,OAAO,EAAE,OAAO,CAAC;IACjB,kBAAkB;IAClB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED;;GAEG;AACH,eAAO,MAAM,cAAc,EAAE,MAAM,CAAC,cAAc,EAAE,aAAa,CAyBhE,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,iBAAiB,GAAG,QAAQ,GAAG,UAAU,GAAG,OAAO,GAAG,MAAM,CAAC;AAEzE;;GAEG;AACH,MAAM,WAAW,SAAS;IACxB,6BAA6B;IAC7B,OAAO,EAAE,OAAO,CAAC;IACjB,wBAAwB;IACxB,KAAK,EAAE,cAAc,CAAC;IACtB,wCAAwC;IACxC,QAAQ,EAAE,MAAM,CAAC;IACjB,uCAAuC;IACvC,SAAS,EAAE,iBAAiB,CAAC;IAC7B,0CAA0C;IAC1C,SAAS,EAAE,MAAM,CAAC;CACnB;AAED;;GAEG;AACH,eAAO,MAAM,kBAAkB,EAAE,SAMhC,CAAC;AA6KF;;;;;GAKG;AACH,wBAAgB,aAAa,CAAC,SAAS,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,GAAG,SAAS,CAUvE;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,MAAM,EAAE,SAAS,GAAG,OAAO,CAEvD"}
|
package/dist/src/vss/config.js
DELETED
|
@@ -1,233 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* VSS (Vector Semantic Search) Configuration
|
|
3
|
-
*
|
|
4
|
-
* Handles loading and validation of VSS settings from config files and environment variables.
|
|
5
|
-
*/
|
|
6
|
-
import { existsSync, readFileSync } from "node:fs";
|
|
7
|
-
import { join, dirname } from "node:path";
|
|
8
|
-
import { fileURLToPath } from "node:url";
|
|
9
|
-
import { expandTilde, getModelsDir } from "../shared/utils/global-dirs.js";
|
|
10
|
-
/**
|
|
11
|
-
* Supported VSS models
|
|
12
|
-
*
|
|
13
|
-
* Model types:
|
|
14
|
-
* - "legacy": Original CodeBERT/UniXcoder models (requires manual ONNX conversion)
|
|
15
|
-
* - "transformers": Models with ONNX support via @huggingface/transformers
|
|
16
|
-
*/
|
|
17
|
-
export const SUPPORTED_MODELS = [
|
|
18
|
-
// Legacy models (CodeBERT/UniXcoder - no official ONNX)
|
|
19
|
-
"codebert-base",
|
|
20
|
-
"unixcoder-base",
|
|
21
|
-
// Transformers.js compatible models (with ONNX)
|
|
22
|
-
"jina-embeddings-v2-base-code",
|
|
23
|
-
"all-MiniLM-L6-v2",
|
|
24
|
-
];
|
|
25
|
-
/**
|
|
26
|
-
* Model registry with metadata
|
|
27
|
-
*/
|
|
28
|
-
export const MODEL_METADATA = {
|
|
29
|
-
"codebert-base": {
|
|
30
|
-
hfModelId: "microsoft/codebert-base",
|
|
31
|
-
dims: 768,
|
|
32
|
-
hasOnnx: false,
|
|
33
|
-
description: "Microsoft CodeBERT (no official ONNX)",
|
|
34
|
-
},
|
|
35
|
-
"unixcoder-base": {
|
|
36
|
-
hfModelId: "microsoft/unixcoder-base",
|
|
37
|
-
dims: 768,
|
|
38
|
-
hasOnnx: false,
|
|
39
|
-
description: "Microsoft UniXcoder (no official ONNX)",
|
|
40
|
-
},
|
|
41
|
-
"jina-embeddings-v2-base-code": {
|
|
42
|
-
hfModelId: "jinaai/jina-embeddings-v2-base-code",
|
|
43
|
-
dims: 768,
|
|
44
|
-
hasOnnx: true,
|
|
45
|
-
description: "Jina code embedding model (ONNX ready, 30+ languages)",
|
|
46
|
-
},
|
|
47
|
-
"all-MiniLM-L6-v2": {
|
|
48
|
-
hfModelId: "Xenova/all-MiniLM-L6-v2",
|
|
49
|
-
dims: 384,
|
|
50
|
-
hasOnnx: true,
|
|
51
|
-
description: "Lightweight general embedding model (ONNX ready)",
|
|
52
|
-
},
|
|
53
|
-
};
|
|
54
|
-
/**
|
|
55
|
-
* Default VSS configuration
|
|
56
|
-
*/
|
|
57
|
-
export const DEFAULT_VSS_CONFIG = {
|
|
58
|
-
enabled: false,
|
|
59
|
-
model: "jina-embeddings-v2-base-code", // Default to ONNX-ready model
|
|
60
|
-
modelDir: getModelsDir(),
|
|
61
|
-
onMissing: "prompt",
|
|
62
|
-
batchSize: 32,
|
|
63
|
-
};
|
|
64
|
-
/**
|
|
65
|
-
* Environment variable names for VSS configuration
|
|
66
|
-
*/
|
|
67
|
-
const ENV_VARS = {
|
|
68
|
-
enabled: "KIRI_VSS_ENABLED",
|
|
69
|
-
model: "KIRI_VSS_MODEL",
|
|
70
|
-
modelDir: "KIRI_VSS_MODEL_DIR",
|
|
71
|
-
onMissing: "KIRI_VSS_ON_MISSING",
|
|
72
|
-
batchSize: "KIRI_VSS_BATCH_SIZE",
|
|
73
|
-
};
|
|
74
|
-
/**
|
|
75
|
-
* Parse a boolean environment variable
|
|
76
|
-
*/
|
|
77
|
-
function parseEnvBool(value, defaultValue) {
|
|
78
|
-
if (value === undefined)
|
|
79
|
-
return defaultValue;
|
|
80
|
-
return value === "1" || value.toLowerCase() === "true";
|
|
81
|
-
}
|
|
82
|
-
/**
|
|
83
|
-
* Parse an integer environment variable
|
|
84
|
-
*/
|
|
85
|
-
function parseEnvInt(value, defaultValue) {
|
|
86
|
-
if (value === undefined)
|
|
87
|
-
return defaultValue;
|
|
88
|
-
const parsed = parseInt(value, 10);
|
|
89
|
-
return Number.isNaN(parsed) ? defaultValue : parsed;
|
|
90
|
-
}
|
|
91
|
-
/**
|
|
92
|
-
* Validate model name
|
|
93
|
-
*/
|
|
94
|
-
function validateModel(model) {
|
|
95
|
-
if (SUPPORTED_MODELS.includes(model)) {
|
|
96
|
-
return model;
|
|
97
|
-
}
|
|
98
|
-
console.warn(`Unknown VSS model "${model}", falling back to "${DEFAULT_VSS_CONFIG.model}". ` +
|
|
99
|
-
`Supported models: ${SUPPORTED_MODELS.join(", ")}`);
|
|
100
|
-
return DEFAULT_VSS_CONFIG.model;
|
|
101
|
-
}
|
|
102
|
-
/**
|
|
103
|
-
* Validate on_missing behavior
|
|
104
|
-
*/
|
|
105
|
-
function validateOnMissing(behavior) {
|
|
106
|
-
const valid = ["prompt", "download", "error", "skip"];
|
|
107
|
-
if (valid.includes(behavior)) {
|
|
108
|
-
return behavior;
|
|
109
|
-
}
|
|
110
|
-
console.warn(`Unknown on_missing behavior "${behavior}", falling back to "${DEFAULT_VSS_CONFIG.onMissing}". ` +
|
|
111
|
-
`Valid values: ${valid.join(", ")}`);
|
|
112
|
-
return DEFAULT_VSS_CONFIG.onMissing;
|
|
113
|
-
}
|
|
114
|
-
/**
|
|
115
|
-
* Simple YAML parser for VSS config section
|
|
116
|
-
* Note: This is a minimal parser for the vss section only
|
|
117
|
-
*/
|
|
118
|
-
function parseVSSFromYaml(content) {
|
|
119
|
-
const result = {};
|
|
120
|
-
const lines = content.split("\n");
|
|
121
|
-
let inVSSSection = false;
|
|
122
|
-
for (const line of lines) {
|
|
123
|
-
// Check for vss: section start
|
|
124
|
-
if (/^vss:\s*$/.test(line)) {
|
|
125
|
-
inVSSSection = true;
|
|
126
|
-
continue;
|
|
127
|
-
}
|
|
128
|
-
// Check if we've left the vss section
|
|
129
|
-
if (inVSSSection && /^\S/.test(line) && !line.startsWith("#")) {
|
|
130
|
-
inVSSSection = false;
|
|
131
|
-
continue;
|
|
132
|
-
}
|
|
133
|
-
if (!inVSSSection)
|
|
134
|
-
continue;
|
|
135
|
-
// Skip comments and empty lines
|
|
136
|
-
if (/^\s*#/.test(line) || /^\s*$/.test(line))
|
|
137
|
-
continue;
|
|
138
|
-
// Parse key-value pairs
|
|
139
|
-
const match = line.match(/^\s{2}(\w+):\s*"?([^"#\n]+)"?\s*(?:#.*)?$/);
|
|
140
|
-
if (match) {
|
|
141
|
-
const [, key, value] = match;
|
|
142
|
-
const trimmedValue = value?.trim();
|
|
143
|
-
switch (key) {
|
|
144
|
-
case "enabled":
|
|
145
|
-
result.enabled = trimmedValue === "true";
|
|
146
|
-
break;
|
|
147
|
-
case "model":
|
|
148
|
-
result.model = validateModel(trimmedValue ?? "");
|
|
149
|
-
break;
|
|
150
|
-
case "model_dir":
|
|
151
|
-
result.modelDir = expandTilde(trimmedValue ?? "");
|
|
152
|
-
break;
|
|
153
|
-
case "on_missing":
|
|
154
|
-
result.onMissing = validateOnMissing(trimmedValue ?? "");
|
|
155
|
-
break;
|
|
156
|
-
case "batch_size":
|
|
157
|
-
result.batchSize = parseInt(trimmedValue ?? "32", 10);
|
|
158
|
-
break;
|
|
159
|
-
}
|
|
160
|
-
}
|
|
161
|
-
}
|
|
162
|
-
return result;
|
|
163
|
-
}
|
|
164
|
-
/**
|
|
165
|
-
* Load VSS configuration from config file
|
|
166
|
-
*/
|
|
167
|
-
function loadFromConfigFile() {
|
|
168
|
-
const __dirname = dirname(fileURLToPath(import.meta.url));
|
|
169
|
-
// Try multiple config file locations
|
|
170
|
-
const configPaths = [
|
|
171
|
-
process.env.KIRI_CONFIG,
|
|
172
|
-
join(process.cwd(), "config", "default.yml"),
|
|
173
|
-
join(__dirname, "../../config/default.yml"),
|
|
174
|
-
join(__dirname, "../../../config/default.yml"),
|
|
175
|
-
].filter((p) => p !== undefined);
|
|
176
|
-
for (const configPath of configPaths) {
|
|
177
|
-
if (existsSync(configPath)) {
|
|
178
|
-
try {
|
|
179
|
-
const content = readFileSync(configPath, "utf-8");
|
|
180
|
-
return parseVSSFromYaml(content);
|
|
181
|
-
}
|
|
182
|
-
catch {
|
|
183
|
-
// Continue to next config file
|
|
184
|
-
}
|
|
185
|
-
}
|
|
186
|
-
}
|
|
187
|
-
return {};
|
|
188
|
-
}
|
|
189
|
-
/**
|
|
190
|
-
* Load VSS configuration from environment variables
|
|
191
|
-
*/
|
|
192
|
-
function loadFromEnv() {
|
|
193
|
-
const result = {};
|
|
194
|
-
if (process.env[ENV_VARS.enabled] !== undefined) {
|
|
195
|
-
result.enabled = parseEnvBool(process.env[ENV_VARS.enabled], DEFAULT_VSS_CONFIG.enabled);
|
|
196
|
-
}
|
|
197
|
-
if (process.env[ENV_VARS.model] !== undefined) {
|
|
198
|
-
result.model = validateModel(process.env[ENV_VARS.model] ?? "");
|
|
199
|
-
}
|
|
200
|
-
if (process.env[ENV_VARS.modelDir] !== undefined) {
|
|
201
|
-
result.modelDir = expandTilde(process.env[ENV_VARS.modelDir] ?? "");
|
|
202
|
-
}
|
|
203
|
-
if (process.env[ENV_VARS.onMissing] !== undefined) {
|
|
204
|
-
result.onMissing = validateOnMissing(process.env[ENV_VARS.onMissing] ?? "");
|
|
205
|
-
}
|
|
206
|
-
if (process.env[ENV_VARS.batchSize] !== undefined) {
|
|
207
|
-
result.batchSize = parseEnvInt(process.env[ENV_VARS.batchSize], DEFAULT_VSS_CONFIG.batchSize);
|
|
208
|
-
}
|
|
209
|
-
return result;
|
|
210
|
-
}
|
|
211
|
-
/**
|
|
212
|
-
* Load complete VSS configuration
|
|
213
|
-
* Priority: Environment variables > Config file > Defaults
|
|
214
|
-
*
|
|
215
|
-
* @param overrides - Optional overrides (e.g., from CLI flags)
|
|
216
|
-
*/
|
|
217
|
-
export function loadVSSConfig(overrides) {
|
|
218
|
-
const fromFile = loadFromConfigFile();
|
|
219
|
-
const fromEnv = loadFromEnv();
|
|
220
|
-
return {
|
|
221
|
-
...DEFAULT_VSS_CONFIG,
|
|
222
|
-
...fromFile,
|
|
223
|
-
...fromEnv,
|
|
224
|
-
...overrides,
|
|
225
|
-
};
|
|
226
|
-
}
|
|
227
|
-
/**
|
|
228
|
-
* Check if VSS should be used based on configuration and model availability
|
|
229
|
-
*/
|
|
230
|
-
export function shouldUseVSS(config) {
|
|
231
|
-
return config.enabled;
|
|
232
|
-
}
|
|
233
|
-
//# sourceMappingURL=config.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"config.js","sourceRoot":"","sources":["../../../src/vss/config.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAEzC,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,gCAAgC,CAAC;AAE3E;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC9B,wDAAwD;IACxD,eAAe;IACf,gBAAgB;IAChB,gDAAgD;IAChD,8BAA8B;IAC9B,kBAAkB;CACV,CAAC;AAiBX;;GAEG;AACH,MAAM,CAAC,MAAM,cAAc,GAA0C;IACnE,eAAe,EAAE;QACf,SAAS,EAAE,yBAAyB;QACpC,IAAI,EAAE,GAAG;QACT,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,uCAAuC;KACrD;IACD,gBAAgB,EAAE;QAChB,SAAS,EAAE,0BAA0B;QACrC,IAAI,EAAE,GAAG;QACT,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,wCAAwC;KACtD;IACD,8BAA8B,EAAE;QAC9B,SAAS,EAAE,qCAAqC;QAChD,IAAI,EAAE,GAAG;QACT,OAAO,EAAE,IAAI;QACb,WAAW,EAAE,uDAAuD;KACrE;IACD,kBAAkB,EAAE;QAClB,SAAS,EAAE,yBAAyB;QACpC,IAAI,EAAE,GAAG;QACT,OAAO,EAAE,IAAI;QACb,WAAW,EAAE,kDAAkD;KAChE;CACF,CAAC;AAuBF;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAc;IAC3C,OAAO,EAAE,KAAK;IACd,KAAK,EAAE,8BAA8B,EAAE,8BAA8B;IACrE,QAAQ,EAAE,YAAY,EAAE;IACxB,SAAS,EAAE,QAAQ;IACnB,SAAS,EAAE,EAAE;CACd,CAAC;AAEF;;GAEG;AACH,MAAM,QAAQ,GAAG;IACf,OAAO,EAAE,kBAAkB;IAC3B,KAAK,EAAE,gBAAgB;IACvB,QAAQ,EAAE,oBAAoB;IAC9B,SAAS,EAAE,qBAAqB;IAChC,SAAS,EAAE,qBAAqB;CACxB,CAAC;AAEX;;GAEG;AACH,SAAS,YAAY,CAAC,KAAyB,EAAE,YAAqB;IACpE,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,YAAY,CAAC;IAC7C,OAAO,KAAK,KAAK,GAAG,IAAI,KAAK,CAAC,WAAW,EAAE,KAAK,MAAM,CAAC;AACzD,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,KAAyB,EAAE,YAAoB;IAClE,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,YAAY,CAAC;IAC7C,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACnC,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC;AACtD,CAAC;AAED;;GAEG;AACH,SAAS,aAAa,CAAC,KAAa;IAClC,IAAI,gBAAgB,CAAC,QAAQ,CAAC,KAAuB,CAAC,EAAE,CAAC;QACvD,OAAO,KAAuB,CAAC;IACjC,CAAC;IACD,OAAO,CAAC,IAAI,CACV,sBAAsB,KAAK,uBAAuB,kBAAkB,CAAC,KAAK,KAAK;QAC7E,qBAAqB,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACrD,CAAC;IACF,OAAO,kBAAkB,CAAC,KAAK,CAAC;AAClC,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,QAAgB;IACzC,MAAM,KAAK,GAAwB,CAAC,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAC3E,IAAI,KAAK,CAAC,QAAQ,CAAC,QAA6B,CAAC,EAAE,CAAC;QAClD,OAAO,QAA6B,CAAC;IACvC,CAAC;IACD,OAAO,CAAC,IAAI,CACV,gCAAgC,QAAQ,uBAAuB,kBAAkB,CAAC,SAAS,KAAK;QAC9F,iBAAiB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACtC,CAAC;IACF,OAAO,kBAAkB,CAAC,SAAS,CAAC;AACtC,CAAC;AAED;;;GAGG;AACH,SAAS,gBAAgB,CAAC,OAAe;IACvC,MAAM,MAAM,GAAuB,EAAE,CAAC;IACtC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAElC,IAAI,YAAY,GAAG,KAAK,CAAC;IAEzB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,+BAA+B;QAC/B,IAAI,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3B,YAAY,GAAG,IAAI,CAAC;YACpB,SAAS;QACX,CAAC;QAED,sCAAsC;QACtC,IAAI,YAAY,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YAC9D,YAAY,GAAG,KAAK,CAAC;YACrB,SAAS;QACX,CAAC;QAED,IAAI,CAAC,YAAY;YAAE,SAAS;QAE5B,gCAAgC;QAChC,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;YAAE,SAAS;QAEvD,wBAAwB;QACxB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,2CAA2C,CAAC,CAAC;QACtE,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,CAAC,EAAE,GAAG,EAAE,KAAK,CAAC,GAAG,KAAK,CAAC;YAC7B,MAAM,YAAY,GAAG,KAAK,EAAE,IAAI,EAAE,CAAC;YAEnC,QAAQ,GAAG,EAAE,CAAC;gBACZ,KAAK,SAAS;oBACZ,MAAM,CAAC,OAAO,GAAG,YAAY,KAAK,MAAM,CAAC;oBACzC,MAAM;gBACR,KAAK,OAAO;oBACV,MAAM,CAAC,KAAK,GAAG,aAAa,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;oBACjD,MAAM;gBACR,KAAK,WAAW;oBACd,MAAM,CAAC,QAAQ,GAAG,WAAW,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;oBAClD,MAAM;gBACR,KAAK,YAAY;oBACf,MAAM,CAAC,SAAS,GAAG,iBAAiB,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;oBACzD,MAAM;gBACR,KAAK,YAAY;oBACf,MAAM,CAAC,SAAS,GAAG,QAAQ,CAAC,YAAY,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;oBACtD,MAAM;YACV,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAS,kBAAkB;IACzB,MAAM,SAAS,GAAG,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAE1D,qCAAqC;IACrC,MAAM,WAAW,GAAG;QAClB,OAAO,CAAC,GAAG,CAAC,WAAW;QACvB,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,aAAa,CAAC;QAC5C,IAAI,CAAC,SAAS,EAAE,0BAA0B,CAAC;QAC3C,IAAI,CAAC,SAAS,EAAE,6BAA6B,CAAC;KAC/C,CAAC,MAAM,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;IAE9C,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;QACrC,IAAI,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;gBAClD,OAAO,gBAAgB,CAAC,OAAO,CAAC,CAAC;YACnC,CAAC;YAAC,MAAM,CAAC;gBACP,+BAA+B;YACjC,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,EAAE,CAAC;AACZ,CAAC;AAED;;GAEG;AACH,SAAS,WAAW;IAClB,MAAM,MAAM,GAAuB,EAAE,CAAC;IAEtC,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,SAAS,EAAE,CAAC;QAChD,MAAM,CAAC,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,kBAAkB,CAAC,OAAO,CAAC,CAAC;IAC3F,CAAC;IAED,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,SAAS,EAAE,CAAC;QAC9C,MAAM,CAAC,KAAK,GAAG,aAAa,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IAClE,CAAC;IAED,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,SAAS,EAAE,CAAC;QACjD,MAAM,CAAC,QAAQ,GAAG,WAAW,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;IACtE,CAAC;IAED,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,KAAK,SAAS,EAAE,CAAC;QAClD,MAAM,CAAC,SAAS,GAAG,iBAAiB,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;IAC9E,CAAC;IAED,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,KAAK,SAAS,EAAE,CAAC;QAClD,MAAM,CAAC,SAAS,GAAG,WAAW,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,kBAAkB,CAAC,SAAS,CAAC,CAAC;IAChG,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,aAAa,CAAC,SAA8B;IAC1D,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;IACtC,MAAM,OAAO,GAAG,WAAW,EAAE,CAAC;IAE9B,OAAO;QACL,GAAG,kBAAkB;QACrB,GAAG,QAAQ;QACX,GAAG,OAAO;QACV,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY,CAAC,MAAiB;IAC5C,OAAO,MAAM,CAAC,OAAO,CAAC;AACxB,CAAC"}
|
package/dist/src/vss/index.d.ts
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/vss/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,cAAc,aAAa,CAAC;AAC5B,cAAc,eAAe,CAAC;AAC9B,cAAc,uBAAuB,CAAC"}
|
package/dist/src/vss/index.js
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/vss/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,cAAc,aAAa,CAAC;AAC5B,cAAc,eAAe,CAAC;AAC9B,cAAc,uBAAuB,CAAC"}
|
|
@@ -1,101 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* VSS Model Downloader
|
|
3
|
-
*
|
|
4
|
-
* Downloads and manages VSS models from HuggingFace Hub.
|
|
5
|
-
* Models are stored in the global ~/.kiri/models/ directory.
|
|
6
|
-
*/
|
|
7
|
-
import { type SupportedModel } from "./config.js";
|
|
8
|
-
/**
|
|
9
|
-
* Model metadata for supported models
|
|
10
|
-
*/
|
|
11
|
-
interface ModelInfo {
|
|
12
|
-
/** HuggingFace repository ID */
|
|
13
|
-
repo: string;
|
|
14
|
-
/** Files to download */
|
|
15
|
-
files: string[];
|
|
16
|
-
/** Expected total size in bytes (approximate) */
|
|
17
|
-
sizeBytes: number;
|
|
18
|
-
/** Human-readable size */
|
|
19
|
-
sizeHuman: string;
|
|
20
|
-
/** Model description */
|
|
21
|
-
description: string;
|
|
22
|
-
}
|
|
23
|
-
/**
|
|
24
|
-
* Model registry with download information
|
|
25
|
-
*
|
|
26
|
-
* Note: ONNX-ready models (jina-embeddings, all-MiniLM) are auto-downloaded
|
|
27
|
-
* by @huggingface/transformers and don't need manual download via this registry.
|
|
28
|
-
*/
|
|
29
|
-
export declare const MODEL_REGISTRY: Record<SupportedModel, ModelInfo>;
|
|
30
|
-
/**
|
|
31
|
-
* Download progress callback
|
|
32
|
-
*/
|
|
33
|
-
export type ProgressCallback = (progress: DownloadProgress) => void;
|
|
34
|
-
/**
|
|
35
|
-
* Download progress information
|
|
36
|
-
*/
|
|
37
|
-
export interface DownloadProgress {
|
|
38
|
-
/** Current file being downloaded */
|
|
39
|
-
file: string;
|
|
40
|
-
/** Bytes downloaded so far */
|
|
41
|
-
downloaded: number;
|
|
42
|
-
/** Total bytes to download (may be 0 if unknown) */
|
|
43
|
-
total: number;
|
|
44
|
-
/** Progress percentage (0-100) */
|
|
45
|
-
percentage: number;
|
|
46
|
-
/** Current phase */
|
|
47
|
-
phase: "downloading" | "verifying" | "complete" | "error";
|
|
48
|
-
/** Error message if phase is "error" */
|
|
49
|
-
error?: string;
|
|
50
|
-
}
|
|
51
|
-
/**
|
|
52
|
-
* Download result
|
|
53
|
-
*/
|
|
54
|
-
export interface DownloadResult {
|
|
55
|
-
success: boolean;
|
|
56
|
-
modelPath: string;
|
|
57
|
-
error?: string;
|
|
58
|
-
}
|
|
59
|
-
/**
|
|
60
|
-
* Get model information
|
|
61
|
-
*/
|
|
62
|
-
export declare function getModelInfo(model: SupportedModel): ModelInfo | undefined;
|
|
63
|
-
/**
|
|
64
|
-
* List all available models
|
|
65
|
-
*/
|
|
66
|
-
export declare function listAvailableModels(): Array<{
|
|
67
|
-
name: SupportedModel;
|
|
68
|
-
info: ModelInfo;
|
|
69
|
-
installed: boolean;
|
|
70
|
-
}>;
|
|
71
|
-
/**
|
|
72
|
-
* Download a model from HuggingFace
|
|
73
|
-
*/
|
|
74
|
-
export declare function downloadModel(model: SupportedModel, onProgress?: ProgressCallback): Promise<DownloadResult>;
|
|
75
|
-
/**
|
|
76
|
-
* Remove an installed model
|
|
77
|
-
*/
|
|
78
|
-
export declare function removeModel(model: SupportedModel): Promise<{
|
|
79
|
-
success: boolean;
|
|
80
|
-
error?: string;
|
|
81
|
-
}>;
|
|
82
|
-
/**
|
|
83
|
-
* Get model status information
|
|
84
|
-
*/
|
|
85
|
-
export interface ModelStatus {
|
|
86
|
-
name: SupportedModel;
|
|
87
|
-
installed: boolean;
|
|
88
|
-
path: string;
|
|
89
|
-
sizeOnDisk?: number;
|
|
90
|
-
files?: string[];
|
|
91
|
-
}
|
|
92
|
-
/**
|
|
93
|
-
* Get status of a specific model
|
|
94
|
-
*/
|
|
95
|
-
export declare function getModelStatus(model: SupportedModel): ModelStatus;
|
|
96
|
-
/**
|
|
97
|
-
* Format bytes to human-readable string
|
|
98
|
-
*/
|
|
99
|
-
export declare function formatBytes(bytes: number): string;
|
|
100
|
-
export {};
|
|
101
|
-
//# sourceMappingURL=model-downloader.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"model-downloader.d.ts","sourceRoot":"","sources":["../../../src/vss/model-downloader.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAUH,OAAO,EAAE,KAAK,cAAc,EAAoB,MAAM,aAAa,CAAC;AAEpE;;GAEG;AACH,UAAU,SAAS;IACjB,gCAAgC;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,wBAAwB;IACxB,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,iDAAiD;IACjD,SAAS,EAAE,MAAM,CAAC;IAClB,0BAA0B;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,wBAAwB;IACxB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED;;;;;GAKG;AACH,eAAO,MAAM,cAAc,EAAE,MAAM,CAAC,cAAc,EAAE,SAAS,CA8B5D,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,gBAAgB,GAAG,CAAC,QAAQ,EAAE,gBAAgB,KAAK,IAAI,CAAC;AAEpE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,oCAAoC;IACpC,IAAI,EAAE,MAAM,CAAC;IACb,8BAA8B;IAC9B,UAAU,EAAE,MAAM,CAAC;IACnB,oDAAoD;IACpD,KAAK,EAAE,MAAM,CAAC;IACd,kCAAkC;IAClC,UAAU,EAAE,MAAM,CAAC;IACnB,oBAAoB;IACpB,KAAK,EAAE,aAAa,GAAG,WAAW,GAAG,UAAU,GAAG,OAAO,CAAC;IAC1D,wCAAwC;IACxC,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,OAAO,EAAE,OAAO,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,KAAK,EAAE,cAAc,GAAG,SAAS,GAAG,SAAS,CAEzE;AAED;;GAEG;AACH,wBAAgB,mBAAmB,IAAI,KAAK,CAAC;IAC3C,IAAI,EAAE,cAAc,CAAC;IACrB,IAAI,EAAE,SAAS,CAAC;IAChB,SAAS,EAAE,OAAO,CAAC;CACpB,CAAC,CAMD;AAgHD;;GAEG;AACH,wBAAsB,aAAa,CACjC,KAAK,EAAE,cAAc,EACrB,UAAU,CAAC,EAAE,gBAAgB,GAC5B,OAAO,CAAC,cAAc,CAAC,CAoGzB;AAED;;GAEG;AACH,wBAAsB,WAAW,CAC/B,KAAK,EAAE,cAAc,GACpB,OAAO,CAAC;IAAE,OAAO,EAAE,OAAO,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAmB/C;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,cAAc,CAAC;IACrB,SAAS,EAAE,OAAO,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;CAClB;AAED;;GAEG;AACH,wBAAgB,cAAc,CAAC,KAAK,EAAE,cAAc,GAAG,WAAW,CA2BjE;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAOjD"}
|