@unlimiting/qsc 0.1.4 → 0.1.5
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/llm/local.d.ts.map +1 -1
- package/dist/llm/local.js +33 -16
- package/dist/llm/local.js.map +1 -1
- package/dist/llm/openai.d.ts.map +1 -1
- package/dist/llm/openai.js +36 -19
- package/dist/llm/openai.js.map +1 -1
- package/dist/mcp.js +4 -4
- package/package.json +1 -1
package/dist/llm/local.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"local.d.ts","sourceRoot":"","sources":["../../src/llm/local.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,KAAK,EAAE,WAAW,EAAmB,MAAM,YAAY,CAAC;AAE/D;;;;;GAKG;AACH,wBAAgB,cAAc,CAAC,MAAM,EAAE,SAAS,GAAG,WAAW,
|
|
1
|
+
{"version":3,"file":"local.d.ts","sourceRoot":"","sources":["../../src/llm/local.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,KAAK,EAAE,WAAW,EAAmB,MAAM,YAAY,CAAC;AAE/D;;;;;GAKG;AACH,wBAAgB,cAAc,CAAC,MAAM,EAAE,SAAS,GAAG,WAAW,CAqG7D"}
|
package/dist/llm/local.js
CHANGED
|
@@ -49,26 +49,43 @@ export function createLocalLLM(config) {
|
|
|
49
49
|
async rerank(query, documents) {
|
|
50
50
|
if (documents.length === 0)
|
|
51
51
|
return [];
|
|
52
|
-
const
|
|
53
|
-
|
|
54
|
-
|
|
52
|
+
const BATCH_SIZE = 5;
|
|
53
|
+
const batches = [];
|
|
54
|
+
for (let i = 0; i < documents.length; i += BATCH_SIZE) {
|
|
55
|
+
batches.push(documents.slice(i, i + BATCH_SIZE));
|
|
56
|
+
}
|
|
57
|
+
const batchResults = await Promise.all(batches.map(async (batch) => {
|
|
58
|
+
try {
|
|
59
|
+
const snippetParts = batch
|
|
60
|
+
.map((doc, idx) => `Snippet ${idx + 1}:\n\`\`\`\n${doc}\n\`\`\``)
|
|
61
|
+
.join("\n\n");
|
|
62
|
+
const prompt = `You are a relevance scoring system. Given a search query and multiple code snippets, rate the relevance of each snippet to the query on a scale from 0.0 to 1.0.
|
|
55
63
|
|
|
56
64
|
Query: ${query}
|
|
57
65
|
|
|
58
|
-
|
|
59
|
-
\`\`\`
|
|
60
|
-
${doc}
|
|
61
|
-
\`\`\`
|
|
66
|
+
${snippetParts}
|
|
62
67
|
|
|
63
|
-
Respond with ONLY
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
68
|
+
Respond with ONLY the scores as comma-separated decimal numbers in order. Example: 0.8, 0.3, 0.9`;
|
|
69
|
+
const result = await provider.generate(prompt, {
|
|
70
|
+
temperature: 0.0,
|
|
71
|
+
maxTokens: 50,
|
|
72
|
+
});
|
|
73
|
+
const parsed = result.trim().split(",").map((s) => {
|
|
74
|
+
const score = parseFloat(s.trim());
|
|
75
|
+
return Number.isFinite(score) ? Math.max(0, Math.min(1, score)) : 0;
|
|
76
|
+
});
|
|
77
|
+
// Pad with 0 if fewer scores than expected
|
|
78
|
+
while (parsed.length < batch.length) {
|
|
79
|
+
parsed.push(0);
|
|
80
|
+
}
|
|
81
|
+
return parsed.slice(0, batch.length);
|
|
82
|
+
}
|
|
83
|
+
catch {
|
|
84
|
+
// Batch failure: return 0 for all documents in this batch
|
|
85
|
+
return batch.map(() => 0);
|
|
86
|
+
}
|
|
87
|
+
}));
|
|
88
|
+
return batchResults.flat();
|
|
72
89
|
},
|
|
73
90
|
};
|
|
74
91
|
return provider;
|
package/dist/llm/local.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"local.js","sourceRoot":"","sources":["../../src/llm/local.ts"],"names":[],"mappings":"AAGA;;;;;GAKG;AACH,MAAM,UAAU,cAAc,CAAC,MAAiB;IAC9C,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC;IAE/B,8DAA8D;IAC9D,IAAI,WAAW,GAAQ,IAAI,CAAC;IAC5B,8DAA8D;IAC9D,IAAI,WAAW,GAAQ,IAAI,CAAC;IAC5B,8DAA8D;IAC9D,IAAI,aAAa,GAAQ,IAAI,CAAC;IAE9B,KAAK,UAAU,SAAS;QACtB,IAAI,WAAW;YAAE,OAAO,WAAW,CAAC;QACpC,IAAI,CAAC;YACH,4DAA4D;YAC5D,uEAAuE;YACvE,WAAW,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;YAC7C,OAAO,WAAW,CAAC;QACrB,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,KAAK,CACb,8EAA8E,CAC/E,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,UAAU,UAAU;QACvB,IAAI,aAAa;YAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;QAEzE,MAAM,KAAK,GAAG,MAAM,SAAS,EAAE,CAAC;QAChC,MAAM,aAAa,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC7C,WAAW,GAAG,MAAM,aAAa,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;QAC3D,aAAa,GAAG,MAAM,WAAW,CAAC,aAAa,EAAE,CAAC;QAClD,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC;IAC3C,CAAC;IAED,MAAM,QAAQ,GAAgB;QAC5B,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,OAAyB;YACtD,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,UAAU,EAAE,CAAC;YAC9C,MAAM,OAAO,GAAG,IAAI,KAAK,CAAC,gBAAgB,CAAC;gBACzC,eAAe,EAAE,OAAO,CAAC,WAAW,EAAE;aACvC,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE;gBAC5C,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,GAAG;gBACxC,SAAS,EAAE,OAAO,EAAE,SAAS;aAC9B,CAAC,CAAC;YAEH,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,SAAmB;YAC7C,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,EAAE,CAAC;YAEtC,MAAM,MAAM,
|
|
1
|
+
{"version":3,"file":"local.js","sourceRoot":"","sources":["../../src/llm/local.ts"],"names":[],"mappings":"AAGA;;;;;GAKG;AACH,MAAM,UAAU,cAAc,CAAC,MAAiB;IAC9C,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC;IAE/B,8DAA8D;IAC9D,IAAI,WAAW,GAAQ,IAAI,CAAC;IAC5B,8DAA8D;IAC9D,IAAI,WAAW,GAAQ,IAAI,CAAC;IAC5B,8DAA8D;IAC9D,IAAI,aAAa,GAAQ,IAAI,CAAC;IAE9B,KAAK,UAAU,SAAS;QACtB,IAAI,WAAW;YAAE,OAAO,WAAW,CAAC;QACpC,IAAI,CAAC;YACH,4DAA4D;YAC5D,uEAAuE;YACvE,WAAW,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;YAC7C,OAAO,WAAW,CAAC;QACrB,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,KAAK,CACb,8EAA8E,CAC/E,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,UAAU,UAAU;QACvB,IAAI,aAAa;YAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;QAEzE,MAAM,KAAK,GAAG,MAAM,SAAS,EAAE,CAAC;QAChC,MAAM,aAAa,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC7C,WAAW,GAAG,MAAM,aAAa,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;QAC3D,aAAa,GAAG,MAAM,WAAW,CAAC,aAAa,EAAE,CAAC;QAClD,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC;IAC3C,CAAC;IAED,MAAM,QAAQ,GAAgB;QAC5B,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,OAAyB;YACtD,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,UAAU,EAAE,CAAC;YAC9C,MAAM,OAAO,GAAG,IAAI,KAAK,CAAC,gBAAgB,CAAC;gBACzC,eAAe,EAAE,OAAO,CAAC,WAAW,EAAE;aACvC,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE;gBAC5C,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,GAAG;gBACxC,SAAS,EAAE,OAAO,EAAE,SAAS;aAC9B,CAAC,CAAC;YAEH,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,SAAmB;YAC7C,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,EAAE,CAAC;YAEtC,MAAM,UAAU,GAAG,CAAC,CAAC;YACrB,MAAM,OAAO,GAAe,EAAE,CAAC;YAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC;gBACtD,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC;YACnD,CAAC;YAED,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CACpC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,EAAqB,EAAE;gBAC7C,IAAI,CAAC;oBACH,MAAM,YAAY,GAAG,KAAK;yBACvB,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,WAAW,GAAG,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;yBAChE,IAAI,CAAC,MAAM,CAAC,CAAC;oBAEhB,MAAM,MAAM,GAAG;;SAElB,KAAK;;EAEZ,YAAY;;iGAEmF,CAAC;oBAEtF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,MAAM,EAAE;wBAC7C,WAAW,EAAE,GAAG;wBAChB,SAAS,EAAE,EAAE;qBACd,CAAC,CAAC;oBAEH,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;wBAChD,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;wBACnC,OAAO,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;oBACtE,CAAC,CAAC,CAAC;oBAEH,2CAA2C;oBAC3C,OAAO,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;wBACpC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;oBACjB,CAAC;oBAED,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;gBACvC,CAAC;gBAAC,MAAM,CAAC;oBACP,0DAA0D;oBAC1D,OAAO,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC5B,CAAC;YACH,CAAC,CAAC,CACH,CAAC;YAEF,OAAO,YAAY,CAAC,IAAI,EAAE,CAAC;QAC7B,CAAC;KACF,CAAC;IAEF,OAAO,QAAQ,CAAC;AAClB,CAAC"}
|
package/dist/llm/openai.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/llm/openai.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,KAAK,EAAE,WAAW,EAAmB,MAAM,YAAY,CAAC;AAK/D;;;;;GAKG;AACH,wBAAgB,eAAe,CAAC,MAAM,EAAE,SAAS,GAAG,WAAW,
|
|
1
|
+
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/llm/openai.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,KAAK,EAAE,WAAW,EAAmB,MAAM,YAAY,CAAC;AAK/D;;;;;GAKG;AACH,wBAAgB,eAAe,CAAC,MAAM,EAAE,SAAS,GAAG,WAAW,CAwG9D"}
|
package/dist/llm/openai.js
CHANGED
|
@@ -46,29 +46,46 @@ export function createOpenAILLM(config) {
|
|
|
46
46
|
async rerank(query, documents) {
|
|
47
47
|
if (documents.length === 0)
|
|
48
48
|
return [];
|
|
49
|
-
const
|
|
50
|
-
|
|
51
|
-
|
|
49
|
+
const BATCH_SIZE = 5;
|
|
50
|
+
const batches = [];
|
|
51
|
+
for (let i = 0; i < documents.length; i += BATCH_SIZE) {
|
|
52
|
+
batches.push(documents.slice(i, i + BATCH_SIZE));
|
|
53
|
+
}
|
|
54
|
+
const batchResults = await Promise.all(batches.map(async (batch) => {
|
|
55
|
+
try {
|
|
56
|
+
const snippetParts = batch
|
|
57
|
+
.map((doc, idx) => `Snippet ${idx + 1}:\n\`\`\`\n${doc}\n\`\`\``)
|
|
58
|
+
.join("\n\n");
|
|
59
|
+
const prompt = `You are a relevance scoring system. Given a search query and multiple code snippets, rate the relevance of each snippet to the query on a scale from 0.0 to 1.0.
|
|
52
60
|
|
|
53
61
|
Query: ${query}
|
|
54
62
|
|
|
55
|
-
|
|
56
|
-
\`\`\`
|
|
57
|
-
${doc}
|
|
58
|
-
\`\`\`
|
|
63
|
+
${snippetParts}
|
|
59
64
|
|
|
60
|
-
Respond with ONLY
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
65
|
+
Respond with ONLY the scores as comma-separated decimal numbers in order. Example: 0.8, 0.3, 0.9`;
|
|
66
|
+
const response = await callWithRetry(() => client.chat.completions.create({
|
|
67
|
+
model,
|
|
68
|
+
messages: [{ role: "user", content: prompt }],
|
|
69
|
+
temperature: 0.0,
|
|
70
|
+
max_tokens: 50,
|
|
71
|
+
}));
|
|
72
|
+
const text = response.choices[0]?.message?.content?.trim() ?? "";
|
|
73
|
+
const parsed = text.split(",").map((s) => {
|
|
74
|
+
const score = parseFloat(s.trim());
|
|
75
|
+
return Number.isFinite(score) ? Math.max(0, Math.min(1, score)) : 0;
|
|
76
|
+
});
|
|
77
|
+
// Pad with 0 if fewer scores than expected
|
|
78
|
+
while (parsed.length < batch.length) {
|
|
79
|
+
parsed.push(0);
|
|
80
|
+
}
|
|
81
|
+
return parsed.slice(0, batch.length);
|
|
82
|
+
}
|
|
83
|
+
catch {
|
|
84
|
+
// Batch failure: return 0 for all documents in this batch
|
|
85
|
+
return batch.map(() => 0);
|
|
86
|
+
}
|
|
87
|
+
}));
|
|
88
|
+
return batchResults.flat();
|
|
72
89
|
},
|
|
73
90
|
};
|
|
74
91
|
return provider;
|
package/dist/llm/openai.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/llm/openai.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAI5B,MAAM,WAAW,GAAG,CAAC,CAAC;AACtB,MAAM,aAAa,GAAG,IAAI,CAAC;AAE3B;;;;;GAKG;AACH,MAAM,UAAU,eAAe,CAAC,MAAiB;IAC/C,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IAC/C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CACb,0CAA0C,MAAM,CAAC,WAAW,cAAc,CAC3E,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;IACtC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;IAE3B,KAAK,UAAU,aAAa,CAAI,EAAoB;QAClD,IAAI,SAAkB,CAAC;QAEvB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC;YACvD,IAAI,CAAC;gBACH,OAAO,MAAM,EAAE,EAAE,CAAC;YACpB,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACxB,SAAS,GAAG,KAAK,CAAC;gBAClB,MAAM,MAAM,GACV,KAAK,YAAY,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC9D,IAAI,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,MAAM,IAAI,GAAG,CAAC,EAAE,CAAC;oBAC9D,MAAM,KAAK,GAAG,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;oBACnD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;oBAC3D,SAAS;gBACX,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QAED,MAAM,SAAS,CAAC;IAClB,CAAC;IAED,MAAM,QAAQ,GAAgB;QAC5B,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,OAAyB;YACtD,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,GAAG,EAAE,CACxC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBAC7B,KAAK;gBACL,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;gBAC7C,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,GAAG;gBACxC,UAAU,EAAE,OAAO,EAAE,SAAS;aAC/B,CAAC,CACH,CAAC;YAEF,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QACrD,CAAC;QAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,SAAmB;YAC7C,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,EAAE,CAAC;YAEtC,MAAM,MAAM,
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/llm/openai.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAI5B,MAAM,WAAW,GAAG,CAAC,CAAC;AACtB,MAAM,aAAa,GAAG,IAAI,CAAC;AAE3B;;;;;GAKG;AACH,MAAM,UAAU,eAAe,CAAC,MAAiB;IAC/C,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IAC/C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CACb,0CAA0C,MAAM,CAAC,WAAW,cAAc,CAC3E,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;IACtC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;IAE3B,KAAK,UAAU,aAAa,CAAI,EAAoB;QAClD,IAAI,SAAkB,CAAC;QAEvB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC;YACvD,IAAI,CAAC;gBACH,OAAO,MAAM,EAAE,EAAE,CAAC;YACpB,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACxB,SAAS,GAAG,KAAK,CAAC;gBAClB,MAAM,MAAM,GACV,KAAK,YAAY,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC9D,IAAI,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,MAAM,IAAI,GAAG,CAAC,EAAE,CAAC;oBAC9D,MAAM,KAAK,GAAG,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;oBACnD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;oBAC3D,SAAS;gBACX,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QAED,MAAM,SAAS,CAAC;IAClB,CAAC;IAED,MAAM,QAAQ,GAAgB;QAC5B,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,OAAyB;YACtD,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,GAAG,EAAE,CACxC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBAC7B,KAAK;gBACL,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;gBAC7C,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,GAAG;gBACxC,UAAU,EAAE,OAAO,EAAE,SAAS;aAC/B,CAAC,CACH,CAAC;YAEF,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QACrD,CAAC;QAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,SAAmB;YAC7C,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,EAAE,CAAC;YAEtC,MAAM,UAAU,GAAG,CAAC,CAAC;YACrB,MAAM,OAAO,GAAe,EAAE,CAAC;YAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC;gBACtD,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC;YACnD,CAAC;YAED,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CACpC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,EAAqB,EAAE;gBAC7C,IAAI,CAAC;oBACH,MAAM,YAAY,GAAG,KAAK;yBACvB,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,WAAW,GAAG,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;yBAChE,IAAI,CAAC,MAAM,CAAC,CAAC;oBAEhB,MAAM,MAAM,GAAG;;SAElB,KAAK;;EAEZ,YAAY;;iGAEmF,CAAC;oBAEtF,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,GAAG,EAAE,CACxC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;wBAC7B,KAAK;wBACL,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;wBAC7C,WAAW,EAAE,GAAG;wBAChB,UAAU,EAAE,EAAE;qBACf,CAAC,CACH,CAAC;oBAEF,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;oBACjE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;wBACvC,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;wBACnC,OAAO,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;oBACtE,CAAC,CAAC,CAAC;oBAEH,2CAA2C;oBAC3C,OAAO,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;wBACpC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;oBACjB,CAAC;oBAED,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;gBACvC,CAAC;gBAAC,MAAM,CAAC;oBACP,0DAA0D;oBAC1D,OAAO,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC5B,CAAC;YACH,CAAC,CAAC,CACH,CAAC;YAEF,OAAO,YAAY,CAAC,IAAI,EAAE,CAAC;QAC7B,CAAC;KACF,CAAC;IAEF,OAAO,QAAQ,CAAC;AAClB,CAAC"}
|
package/dist/mcp.js
CHANGED
|
@@ -160,7 +160,7 @@ export async function startMcpServer() {
|
|
|
160
160
|
.min(1)
|
|
161
161
|
.max(100)
|
|
162
162
|
.optional()
|
|
163
|
-
.describe("Maximum number of results (default:
|
|
163
|
+
.describe("Maximum number of results (default: 10)"),
|
|
164
164
|
}),
|
|
165
165
|
}, async ({ query, limit }) => {
|
|
166
166
|
if (!query.trim()) {
|
|
@@ -169,7 +169,7 @@ export async function startMcpServer() {
|
|
|
169
169
|
isError: true,
|
|
170
170
|
};
|
|
171
171
|
}
|
|
172
|
-
const results = store.searchBM25(query, limit ??
|
|
172
|
+
const results = store.searchBM25(query, limit ?? 10);
|
|
173
173
|
if (results.length === 0) {
|
|
174
174
|
return { content: [{ type: "text", text: "No results found." }] };
|
|
175
175
|
}
|
|
@@ -193,7 +193,7 @@ export async function startMcpServer() {
|
|
|
193
193
|
.min(1)
|
|
194
194
|
.max(100)
|
|
195
195
|
.optional()
|
|
196
|
-
.describe("Maximum number of results (default:
|
|
196
|
+
.describe("Maximum number of results (default: 10)"),
|
|
197
197
|
expand: z
|
|
198
198
|
.boolean()
|
|
199
199
|
.optional()
|
|
@@ -216,7 +216,7 @@ export async function startMcpServer() {
|
|
|
216
216
|
const shouldRerank = rerank ?? !!llmProvider;
|
|
217
217
|
const response = await sp.search(query, {
|
|
218
218
|
mode,
|
|
219
|
-
limit: limit ??
|
|
219
|
+
limit: limit ?? 10,
|
|
220
220
|
expand: shouldExpand,
|
|
221
221
|
rerank: shouldRerank,
|
|
222
222
|
});
|