@g99/lightrag-mcp-server 1.0.4 → 1.0.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/index.js +34 -18
- package/package.json +1 -1
package/index.js
CHANGED
|
@@ -37,7 +37,7 @@ const httpClient = axios.create({
|
|
|
37
37
|
const server = new Server(
|
|
38
38
|
{
|
|
39
39
|
name: '@g99/lightrag-mcp-server',
|
|
40
|
-
version: '1.0.
|
|
40
|
+
version: '1.0.5',
|
|
41
41
|
},
|
|
42
42
|
{
|
|
43
43
|
capabilities: {
|
|
@@ -366,20 +366,20 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
|
|
366
366
|
switch (name) {
|
|
367
367
|
// DOCUMENT MANAGEMENT
|
|
368
368
|
case 'insert_text':
|
|
369
|
-
response = await httpClient.post('/
|
|
369
|
+
response = await httpClient.post('/documents/text', {
|
|
370
370
|
text: args.text,
|
|
371
371
|
description: args.description
|
|
372
372
|
});
|
|
373
373
|
break;
|
|
374
374
|
|
|
375
375
|
case 'insert_texts':
|
|
376
|
-
response = await httpClient.post('/
|
|
376
|
+
response = await httpClient.post('/documents/texts', {
|
|
377
377
|
texts: args.texts
|
|
378
378
|
});
|
|
379
379
|
break;
|
|
380
380
|
|
|
381
381
|
case 'upload_document':
|
|
382
|
-
response = await httpClient.post('/upload', {
|
|
382
|
+
response = await httpClient.post('/documents/upload', {
|
|
383
383
|
file_path: args.file_path,
|
|
384
384
|
chunk_size: args.chunk_size,
|
|
385
385
|
chunk_overlap: args.chunk_overlap
|
|
@@ -387,13 +387,13 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
|
|
387
387
|
break;
|
|
388
388
|
|
|
389
389
|
case 'upload_documents':
|
|
390
|
-
response = await httpClient.post('/
|
|
390
|
+
response = await httpClient.post('/documents/upload/batch', {
|
|
391
391
|
file_paths: args.file_paths
|
|
392
392
|
});
|
|
393
393
|
break;
|
|
394
394
|
|
|
395
395
|
case 'scan_documents':
|
|
396
|
-
response = await httpClient.post('/scan');
|
|
396
|
+
response = await httpClient.post('/documents/scan');
|
|
397
397
|
break;
|
|
398
398
|
|
|
399
399
|
case 'get_documents':
|
|
@@ -401,7 +401,9 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
|
|
401
401
|
break;
|
|
402
402
|
|
|
403
403
|
case 'get_documents_paginated':
|
|
404
|
-
response = await httpClient.get(
|
|
404
|
+
response = await httpClient.get('/documents/paginated', {
|
|
405
|
+
params: { page: args.page, page_size: args.page_size }
|
|
406
|
+
});
|
|
405
407
|
break;
|
|
406
408
|
|
|
407
409
|
case 'delete_document':
|
|
@@ -413,7 +415,11 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
|
|
413
415
|
break;
|
|
414
416
|
|
|
415
417
|
case 'document_status':
|
|
416
|
-
|
|
418
|
+
if (args.document_id) {
|
|
419
|
+
response = await httpClient.get(`/documents/${args.document_id}/status`);
|
|
420
|
+
} else {
|
|
421
|
+
response = await httpClient.get('/documents/status');
|
|
422
|
+
}
|
|
417
423
|
break;
|
|
418
424
|
|
|
419
425
|
// QUERY OPERATIONS
|
|
@@ -421,22 +427,24 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
|
|
421
427
|
response = await httpClient.post('/query', {
|
|
422
428
|
query: args.query,
|
|
423
429
|
mode: args.mode || 'hybrid',
|
|
424
|
-
only_need_context: args.only_need_context,
|
|
425
|
-
top_k: args.top_k
|
|
430
|
+
only_need_context: args.only_need_context || false,
|
|
431
|
+
top_k: args.top_k || 60
|
|
426
432
|
});
|
|
427
433
|
break;
|
|
428
434
|
|
|
429
435
|
case 'query_text_stream':
|
|
430
|
-
response = await httpClient.post('/query
|
|
436
|
+
response = await httpClient.post('/query', {
|
|
431
437
|
query: args.query,
|
|
432
|
-
mode: args.mode || 'hybrid'
|
|
438
|
+
mode: args.mode || 'hybrid',
|
|
439
|
+
stream: true
|
|
433
440
|
});
|
|
434
441
|
break;
|
|
435
442
|
|
|
436
443
|
case 'query_with_citation':
|
|
437
|
-
response = await httpClient.post('/query
|
|
444
|
+
response = await httpClient.post('/query', {
|
|
438
445
|
query: args.query,
|
|
439
|
-
mode: args.mode || 'hybrid'
|
|
446
|
+
mode: args.mode || 'hybrid',
|
|
447
|
+
with_citation: true
|
|
440
448
|
});
|
|
441
449
|
break;
|
|
442
450
|
|
|
@@ -450,19 +458,27 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
|
|
450
458
|
break;
|
|
451
459
|
|
|
452
460
|
case 'get_entities':
|
|
453
|
-
response = await httpClient.get(
|
|
461
|
+
response = await httpClient.get('/graph/entities', {
|
|
462
|
+
params: args.limit ? { limit: args.limit } : {}
|
|
463
|
+
});
|
|
454
464
|
break;
|
|
455
465
|
|
|
456
466
|
case 'get_relations':
|
|
457
|
-
response = await httpClient.get(
|
|
467
|
+
response = await httpClient.get('/graph/relations', {
|
|
468
|
+
params: args.limit ? { limit: args.limit } : {}
|
|
469
|
+
});
|
|
458
470
|
break;
|
|
459
471
|
|
|
460
472
|
case 'check_entity_exists':
|
|
461
|
-
response = await httpClient.get(
|
|
473
|
+
response = await httpClient.get('/graph/entity/exists', {
|
|
474
|
+
params: { name: args.entity_name }
|
|
475
|
+
});
|
|
462
476
|
break;
|
|
463
477
|
|
|
464
478
|
case 'update_entity':
|
|
465
|
-
response = await httpClient.put(`/graph/entity/${args.entity_id}`,
|
|
479
|
+
response = await httpClient.put(`/graph/entity/${args.entity_id}`, {
|
|
480
|
+
properties: args.properties
|
|
481
|
+
});
|
|
466
482
|
break;
|
|
467
483
|
|
|
468
484
|
case 'delete_entity':
|
package/package.json
CHANGED