@chnxthr/knowledge-mcp 0.0.0-stage → 0.3.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/LICENSE +21 -0
- package/README.md +37 -2
- package/dist/bootstrap.d.ts +1 -0
- package/dist/bootstrap.js +18 -0
- package/dist/bootstrap.js.map +1 -0
- package/dist/cli.d.ts +2 -0
- package/dist/cli.js +5 -0
- package/dist/cli.js.map +1 -0
- package/dist/core/auth.d.ts +38 -0
- package/dist/core/auth.js +93 -0
- package/dist/core/auth.js.map +1 -0
- package/dist/core/config.d.ts +259 -0
- package/dist/core/config.js +172 -0
- package/dist/core/config.js.map +1 -0
- package/dist/core/context.d.ts +25 -0
- package/dist/core/context.js +37 -0
- package/dist/core/context.js.map +1 -0
- package/dist/core/env.d.ts +2 -0
- package/dist/core/env.js +5 -0
- package/dist/core/env.js.map +1 -0
- package/dist/core/log_mask.d.ts +10 -0
- package/dist/core/log_mask.js +88 -0
- package/dist/core/log_mask.js.map +1 -0
- package/dist/core/oauth.d.ts +46 -0
- package/dist/core/oauth.js +507 -0
- package/dist/core/oauth.js.map +1 -0
- package/dist/core/paths.d.ts +5 -0
- package/dist/core/paths.js +47 -0
- package/dist/core/paths.js.map +1 -0
- package/dist/core/scope.d.ts +28 -0
- package/dist/core/scope.js +54 -0
- package/dist/core/scope.js.map +1 -0
- package/dist/core/upstream.d.ts +30 -0
- package/dist/core/upstream.js +164 -0
- package/dist/core/upstream.js.map +1 -0
- package/dist/index.d.ts +23 -0
- package/dist/index.js +21 -0
- package/dist/index.js.map +1 -0
- package/dist/plugin/define.d.ts +144 -0
- package/dist/plugin/define.js +44 -0
- package/dist/plugin/define.js.map +1 -0
- package/dist/plugin/load.d.ts +2 -0
- package/dist/plugin/load.js +41 -0
- package/dist/plugin/load.js.map +1 -0
- package/dist/plugin/ping.d.ts +9 -0
- package/dist/plugin/ping.js +29 -0
- package/dist/plugin/ping.js.map +1 -0
- package/dist/plugin/registry.d.ts +62 -0
- package/dist/plugin/registry.js +248 -0
- package/dist/plugin/registry.js.map +1 -0
- package/dist/plugin/result.d.ts +20 -0
- package/dist/plugin/result.js +59 -0
- package/dist/plugin/result.js.map +1 -0
- package/dist/plugins/drive/chunker.d.ts +28 -0
- package/dist/plugins/drive/chunker.js +356 -0
- package/dist/plugins/drive/chunker.js.map +1 -0
- package/dist/plugins/drive/clean_text.d.ts +13 -0
- package/dist/plugins/drive/clean_text.js +31 -0
- package/dist/plugins/drive/clean_text.js.map +1 -0
- package/dist/plugins/drive/client.d.ts +25 -0
- package/dist/plugins/drive/client.js +62 -0
- package/dist/plugins/drive/client.js.map +1 -0
- package/dist/plugins/drive/embeddings.d.ts +15 -0
- package/dist/plugins/drive/embeddings.js +118 -0
- package/dist/plugins/drive/embeddings.js.map +1 -0
- package/dist/plugins/drive/exclude.d.ts +11 -0
- package/dist/plugins/drive/exclude.js +36 -0
- package/dist/plugins/drive/exclude.js.map +1 -0
- package/dist/plugins/drive/extract_text.d.ts +26 -0
- package/dist/plugins/drive/extract_text.js +108 -0
- package/dist/plugins/drive/extract_text.js.map +1 -0
- package/dist/plugins/drive/get_token.d.ts +5 -0
- package/dist/plugins/drive/get_token.js +113 -0
- package/dist/plugins/drive/get_token.js.map +1 -0
- package/dist/plugins/drive/index.d.ts +5 -0
- package/dist/plugins/drive/index.js +72 -0
- package/dist/plugins/drive/index.js.map +1 -0
- package/dist/plugins/drive/indexer.d.ts +69 -0
- package/dist/plugins/drive/indexer.js +204 -0
- package/dist/plugins/drive/indexer.js.map +1 -0
- package/dist/plugins/drive/mathpix.d.ts +4 -0
- package/dist/plugins/drive/mathpix.js +102 -0
- package/dist/plugins/drive/mathpix.js.map +1 -0
- package/dist/plugins/drive/path.d.ts +40 -0
- package/dist/plugins/drive/path.js +87 -0
- package/dist/plugins/drive/path.js.map +1 -0
- package/dist/plugins/drive/qdrant.d.ts +50 -0
- package/dist/plugins/drive/qdrant.js +61 -0
- package/dist/plugins/drive/qdrant.js.map +1 -0
- package/dist/plugins/drive/reindex.d.ts +2 -0
- package/dist/plugins/drive/reindex.js +263 -0
- package/dist/plugins/drive/reindex.js.map +1 -0
- package/dist/plugins/drive/settings.d.ts +11 -0
- package/dist/plugins/drive/settings.js +13 -0
- package/dist/plugins/drive/settings.js.map +1 -0
- package/dist/plugins/drive/sync.d.ts +16 -0
- package/dist/plugins/drive/sync.js +139 -0
- package/dist/plugins/drive/sync.js.map +1 -0
- package/dist/plugins/drive/sync_state.d.ts +3 -0
- package/dist/plugins/drive/sync_state.js +26 -0
- package/dist/plugins/drive/sync_state.js.map +1 -0
- package/dist/plugins/drive/tools/list_drive_folder.d.ts +104 -0
- package/dist/plugins/drive/tools/list_drive_folder.js +221 -0
- package/dist/plugins/drive/tools/list_drive_folder.js.map +1 -0
- package/dist/plugins/drive/tools/read_drive_file.d.ts +66 -0
- package/dist/plugins/drive/tools/read_drive_file.js +206 -0
- package/dist/plugins/drive/tools/read_drive_file.js.map +1 -0
- package/dist/plugins/drive/tools/search_drive.d.ts +52 -0
- package/dist/plugins/drive/tools/search_drive.js +78 -0
- package/dist/plugins/drive/tools/search_drive.js.map +1 -0
- package/dist/plugins/drive/tools/semantic_search_drive.d.ts +109 -0
- package/dist/plugins/drive/tools/semantic_search_drive.js +228 -0
- package/dist/plugins/drive/tools/semantic_search_drive.js.map +1 -0
- package/dist/plugins/github/index.d.ts +1 -0
- package/dist/plugins/github/index.js +27 -0
- package/dist/plugins/github/index.js.map +1 -0
- package/dist/plugins/github/repo_tree.d.ts +57 -0
- package/dist/plugins/github/repo_tree.js +123 -0
- package/dist/plugins/github/repo_tree.js.map +1 -0
- package/dist/plugins/index.d.ts +11 -0
- package/dist/plugins/index.js +7 -0
- package/dist/plugins/index.js.map +1 -0
- package/dist/plugins/railway/index.d.ts +1 -0
- package/dist/plugins/railway/index.js +21 -0
- package/dist/plugins/railway/index.js.map +1 -0
- package/dist/plugins/railway/tools.d.ts +114 -0
- package/dist/plugins/railway/tools.js +165 -0
- package/dist/plugins/railway/tools.js.map +1 -0
- package/dist/plugins/slack/api.d.ts +63 -0
- package/dist/plugins/slack/api.js +280 -0
- package/dist/plugins/slack/api.js.map +1 -0
- package/dist/plugins/slack/index.d.ts +1 -0
- package/dist/plugins/slack/index.js +80 -0
- package/dist/plugins/slack/index.js.map +1 -0
- package/dist/plugins/slack/tools/slack_files.d.ts +87 -0
- package/dist/plugins/slack/tools/slack_files.js +204 -0
- package/dist/plugins/slack/tools/slack_files.js.map +1 -0
- package/dist/plugins/slack/tools/slack_list_channels.d.ts +46 -0
- package/dist/plugins/slack/tools/slack_list_channels.js +197 -0
- package/dist/plugins/slack/tools/slack_list_channels.js.map +1 -0
- package/dist/plugins/slack/tools/slack_read_channel.d.ts +56 -0
- package/dist/plugins/slack/tools/slack_read_channel.js +150 -0
- package/dist/plugins/slack/tools/slack_read_channel.js.map +1 -0
- package/dist/plugins/slack/tools/slack_search.d.ts +53 -0
- package/dist/plugins/slack/tools/slack_search.js +132 -0
- package/dist/plugins/slack/tools/slack_search.js.map +1 -0
- package/dist/run.d.ts +8 -0
- package/dist/run.js +270 -0
- package/dist/run.js.map +1 -0
- package/dist/server/http.d.ts +5 -0
- package/dist/server/http.js +406 -0
- package/dist/server/http.js.map +1 -0
- package/dist/server/identity.d.ts +17 -0
- package/dist/server/identity.js +47 -0
- package/dist/server/identity.js.map +1 -0
- package/dist/server/mcp.d.ts +7 -0
- package/dist/server/mcp.js +55 -0
- package/dist/server/mcp.js.map +1 -0
- package/dist/server/stdio.d.ts +2 -0
- package/dist/server/stdio.js +40 -0
- package/dist/server/stdio.js.map +1 -0
- package/dist/store/index.d.ts +11 -0
- package/dist/store/index.js +31 -0
- package/dist/store/index.js.map +1 -0
- package/dist/store/keys.d.ts +9 -0
- package/dist/store/keys.js +32 -0
- package/dist/store/keys.js.map +1 -0
- package/dist/store/sqlite.d.ts +38 -0
- package/dist/store/sqlite.js +316 -0
- package/dist/store/sqlite.js.map +1 -0
- package/dist/store/supabase.d.ts +35 -0
- package/dist/store/supabase.js +269 -0
- package/dist/store/supabase.js.map +1 -0
- package/dist/store/supabase_client.d.ts +10 -0
- package/dist/store/supabase_client.js +31 -0
- package/dist/store/supabase_client.js.map +1 -0
- package/dist/store/tool_log.d.ts +12 -0
- package/dist/store/tool_log.js +8 -0
- package/dist/store/tool_log.js.map +1 -0
- package/dist/store/types.d.ts +96 -0
- package/dist/store/types.js +2 -0
- package/dist/store/types.js.map +1 -0
- package/dist/version.d.ts +1 -0
- package/dist/version.js +11 -0
- package/dist/version.js.map +1 -0
- package/package.json +84 -4
- package/sql/supabase.sql +185 -0
|
@@ -0,0 +1,221 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* list_drive_folder — Drive 폴더 자식 list (배치 + 재귀 depth).
|
|
3
|
+
*
|
|
4
|
+
* 배치/재귀: 폴더 N개·하위 트리를 1회 호출로. 에이전트가 폴더마다 순차 호출(= per-call
|
|
5
|
+
* 네트워크 floor + 그 사이 LLM 추론 턴 ×N)하던 것을 1회로 접어 체감 지연을 N→1 로 줄인다.
|
|
6
|
+
* read_drive_file 의 fileIds 배치와 같은 패턴(서버가 Drive I/O 를 자기↔구글 빠른 망에서 병렬).
|
|
7
|
+
*
|
|
8
|
+
* - folderNames[] / folderIds[] : 서로 다른 위치의 폴더 여러 개를 병렬 list.
|
|
9
|
+
* - depth (1~3) : 한 폴더의 하위 트리를 N단계까지 한 번에.
|
|
10
|
+
* 예) 고객사 10곳 + 각 폴더 문서 → list_drive_folder({ folderName: "01_고객사", depth: 2 }).
|
|
11
|
+
*
|
|
12
|
+
* 보안: 시작 폴더는 색인 root 하위이고 제외 폴더(PII·계약·HR 등)를 거치지 않을 때만 연다(read_drive_file 과 동일 가드).
|
|
13
|
+
* 이름으로 찾을 때도 root 하위 후보만 고른다. 재귀 walk 는 제외 폴더로 확산하지 않는다
|
|
14
|
+
* (해당 폴더는 이름만 표기하고 자식 미전개 — excluded:true).
|
|
15
|
+
*
|
|
16
|
+
* 안전장치: 총 폴더 방문(=Drive list 호출) 상한 MAX_FOLDER_VISITS — 깊고 넓은 트리의 호출 폭발 방지.
|
|
17
|
+
*/
|
|
18
|
+
import { z } from 'zod';
|
|
19
|
+
import { defineTool } from '../../../plugin/define.js';
|
|
20
|
+
import { getDriveClient, escapeDriveQuery } from '../client.js';
|
|
21
|
+
import { getAuth } from '../../../core/context.js';
|
|
22
|
+
import { isExcludedName } from '../exclude.js';
|
|
23
|
+
import { allowedDriveFolders, isPathInScope } from '../../../core/scope.js';
|
|
24
|
+
import { classifyFolder } from '../path.js';
|
|
25
|
+
const FOLDER_MIME = 'application/vnd.google-apps.folder';
|
|
26
|
+
const MAX_FOLDER_VISITS = 120; // 재귀 폭발 방지 — 총 폴더 방문(list 호출) 상한
|
|
27
|
+
const DESCRIPTION = 'Google Drive 폴더의 자식 파일·폴더 list. folderId/folderName(단일) 또는 folderIds/folderNames(여러 개 한 번에). ' +
|
|
28
|
+
'**"고객사 전부/명단", "X 폴더에 뭐 있어" 같은 전수·구조 질문은 semantic_search 가 아니라 이 도구.** ' +
|
|
29
|
+
'**여러 폴더거나 하위까지 봐야 하면 폴더마다 따로 호출하지 말고 한 번에 묶어라:** ' +
|
|
30
|
+
'folderNames=[...] 로 여러 폴더를 병렬, 또는 depth(1~3)로 하위 트리까지 한 번에 ' +
|
|
31
|
+
'(예: 고객사 10곳 + 각 폴더 문서 → folderName="01_고객사", depth=2). ' +
|
|
32
|
+
'folderName 은 이름으로 ID 를 자동 resolve. 결과의 id 로 read_drive_file 호출해 원문 read.';
|
|
33
|
+
const inputSchema = z
|
|
34
|
+
.object({
|
|
35
|
+
folderId: z.string().min(1).optional().describe('Drive folder ID (단일). "root" = My Drive 최상위.'),
|
|
36
|
+
folderName: z.string().min(1).optional().describe('폴더 이름 (단일, 예: "01_고객사"). ID 를 몰라도 이름으로 조회.'),
|
|
37
|
+
folderIds: z
|
|
38
|
+
.array(z.string().min(1))
|
|
39
|
+
.min(1)
|
|
40
|
+
.max(20)
|
|
41
|
+
.optional()
|
|
42
|
+
.describe('여러 folder ID 를 한 번에 병렬 list (1~20). 순차 호출보다 훨씬 빠름.'),
|
|
43
|
+
folderNames: z
|
|
44
|
+
.array(z.string().min(1))
|
|
45
|
+
.min(1)
|
|
46
|
+
.max(20)
|
|
47
|
+
.optional()
|
|
48
|
+
.describe('여러 폴더 이름을 한 번에 병렬 list (1~20). 순차 호출보다 훨씬 빠름.'),
|
|
49
|
+
depth: z
|
|
50
|
+
.number()
|
|
51
|
+
.int()
|
|
52
|
+
.min(1)
|
|
53
|
+
.max(3)
|
|
54
|
+
.default(1)
|
|
55
|
+
.describe('재귀 깊이 1~3. default 1(직속 자식만). 2=손자까지, 3=증손자까지 — 하위 트리를 1회 호출로 받는다.'),
|
|
56
|
+
pageSize: z.number().int().min(1).max(200).default(50).describe('폴더당 최대 자식 수 (1-200). default 50.'),
|
|
57
|
+
})
|
|
58
|
+
.refine((v) => v.folderId || v.folderName || v.folderIds?.length || v.folderNames?.length, { message: 'folderId / folderName / folderIds / folderNames 중 하나는 필수' });
|
|
59
|
+
export const listDriveFolderTool = defineTool({
|
|
60
|
+
name: 'list_drive_folder',
|
|
61
|
+
description: DESCRIPTION,
|
|
62
|
+
input: inputSchema,
|
|
63
|
+
annotations: { readOnlyHint: true },
|
|
64
|
+
run: (args) => handleListDriveFolder(args),
|
|
65
|
+
});
|
|
66
|
+
const ALL_DRIVES = { supportsAllDrives: true, includeItemsFromAllDrives: true };
|
|
67
|
+
/**
|
|
68
|
+
* folderName → 색인 root 하위의 열 수 있는 폴더 (Shared Drive 포함). 못 찾으면 null.
|
|
69
|
+
* 같은 이름 폴더가 여럿이면 root 밖·제외 경로 후보를 건너뛰고 처음 'ok' 인 것을 고른다.
|
|
70
|
+
*/
|
|
71
|
+
async function resolveFolderId(drive, name) {
|
|
72
|
+
const res = await drive.files.list({
|
|
73
|
+
q: `name = '${escapeDriveQuery(name)}' and mimeType = '${FOLDER_MIME}' and trashed = false`,
|
|
74
|
+
fields: 'files(id, name)',
|
|
75
|
+
pageSize: 10,
|
|
76
|
+
...ALL_DRIVES,
|
|
77
|
+
});
|
|
78
|
+
for (const f of res.data.files ?? []) {
|
|
79
|
+
if (!f.id)
|
|
80
|
+
continue;
|
|
81
|
+
const c = await classifyFolder(drive, f.id);
|
|
82
|
+
if (c.cls === 'ok')
|
|
83
|
+
return { id: f.id, path: c.path };
|
|
84
|
+
}
|
|
85
|
+
return null;
|
|
86
|
+
}
|
|
87
|
+
/**
|
|
88
|
+
* 폴더 fid 의 자식을 list 하고, depthLeft 가 남으면 (제외 폴더를 제외하고) 자식 폴더를 병렬 재귀.
|
|
89
|
+
* 같은 레벨 폴더는 Promise.all 로 동시에 펼친다. budget 으로 총 방문 수를 캡.
|
|
90
|
+
*/
|
|
91
|
+
async function walkFolder(drive, fid, depthLeft, pageSize, budget) {
|
|
92
|
+
const res = await drive.files.list({
|
|
93
|
+
q: `'${escapeDriveQuery(fid)}' in parents and trashed = false`,
|
|
94
|
+
fields: 'nextPageToken, files(id, name, mimeType, modifiedTime, size)',
|
|
95
|
+
pageSize,
|
|
96
|
+
orderBy: 'folder, name',
|
|
97
|
+
...ALL_DRIVES,
|
|
98
|
+
});
|
|
99
|
+
// pageSize 초과 자식이 더 있으면 hasMore — 예전엔 조용히 잘려 "전수 목록" 을 오인할 수 있었다.
|
|
100
|
+
const hasMore = !!res.data.nextPageToken;
|
|
101
|
+
const nodes = (res.data.files || []).map((f) => ({
|
|
102
|
+
id: f.id,
|
|
103
|
+
name: f.name,
|
|
104
|
+
mimeType: f.mimeType,
|
|
105
|
+
isFolder: f.mimeType === FOLDER_MIME,
|
|
106
|
+
modifiedTime: f.modifiedTime,
|
|
107
|
+
size: f.size,
|
|
108
|
+
}));
|
|
109
|
+
if (depthLeft <= 1)
|
|
110
|
+
return { children: nodes, hasMore };
|
|
111
|
+
// 자식 폴더 중 제외(PII/계약/HR/deprecated 등) 아닌 것만 재귀. 제외 폴더는 이름만 남기고 미전개.
|
|
112
|
+
const toRecurse = nodes.filter((n) => {
|
|
113
|
+
if (!n.isFolder)
|
|
114
|
+
return false;
|
|
115
|
+
if (isExcludedName(n.name)) {
|
|
116
|
+
n.excluded = true;
|
|
117
|
+
return false;
|
|
118
|
+
}
|
|
119
|
+
return true;
|
|
120
|
+
});
|
|
121
|
+
await Promise.all(toRecurse.map(async (n) => {
|
|
122
|
+
// check + increment 는 동기(중간 await 없음) → 병렬 콜백 간에도 cap 정확.
|
|
123
|
+
if (budget.visits >= MAX_FOLDER_VISITS) {
|
|
124
|
+
n.truncated = true;
|
|
125
|
+
return;
|
|
126
|
+
}
|
|
127
|
+
budget.visits += 1;
|
|
128
|
+
const sub = await walkFolder(drive, n.id, depthLeft - 1, pageSize, budget);
|
|
129
|
+
n.children = sub.children;
|
|
130
|
+
if (sub.hasMore)
|
|
131
|
+
n.childrenTruncated = true;
|
|
132
|
+
}));
|
|
133
|
+
return { children: nodes, hasMore };
|
|
134
|
+
}
|
|
135
|
+
/** 재귀 트리를 직원별 Drive 폴더 scope 로 후처리 필터. 폴더는 자식이 scope 에 걸리면 유지(하위 도달 보장). */
|
|
136
|
+
function filterTreeByScope(nodes, parentPath, scope) {
|
|
137
|
+
return nodes.filter((n) => {
|
|
138
|
+
const p = parentPath ? `${parentPath}/${n.name ?? ''}` : n.name ?? '';
|
|
139
|
+
if (n.isFolder) {
|
|
140
|
+
if (n.children)
|
|
141
|
+
n.children = filterTreeByScope(n.children, p, scope);
|
|
142
|
+
return isPathInScope(p, scope) || (n.children !== undefined && n.children.length > 0);
|
|
143
|
+
}
|
|
144
|
+
return isPathInScope(p, scope);
|
|
145
|
+
});
|
|
146
|
+
}
|
|
147
|
+
export async function handleListDriveFolder(args) {
|
|
148
|
+
const auth = getAuth();
|
|
149
|
+
const { folderId, folderName, folderIds, folderNames, depth, pageSize } = args;
|
|
150
|
+
const drive = await getDriveClient();
|
|
151
|
+
const folderScope = allowedDriveFolders(auth);
|
|
152
|
+
// 단수 + 배치 입력을 루트 명세로 통합.
|
|
153
|
+
const rootSpecs = [];
|
|
154
|
+
if (folderId)
|
|
155
|
+
rootSpecs.push({ id: folderId });
|
|
156
|
+
if (folderName)
|
|
157
|
+
rootSpecs.push({ name: folderName });
|
|
158
|
+
for (const id of folderIds ?? [])
|
|
159
|
+
rootSpecs.push({ id });
|
|
160
|
+
for (const name of folderNames ?? [])
|
|
161
|
+
rootSpecs.push({ name });
|
|
162
|
+
const budget = { visits: rootSpecs.length }; // 루트도 방문 1회로 카운트
|
|
163
|
+
try {
|
|
164
|
+
const roots = await Promise.all(rootSpecs.map(async (spec) => {
|
|
165
|
+
let fid;
|
|
166
|
+
let rootPath;
|
|
167
|
+
if (spec.id) {
|
|
168
|
+
// 시작 폴더 가드 — 색인 root 하위 + 제외 폴더 미경유일 때만 연다.
|
|
169
|
+
const c = await classifyFolder(drive, spec.id);
|
|
170
|
+
if (c.cls !== 'ok') {
|
|
171
|
+
return { folderId: spec.id, folderName: null, error: '[forbidden] 접근 범위 밖 폴더입니다 (색인 대상 root 밖 / 민감 폴더).' };
|
|
172
|
+
}
|
|
173
|
+
fid = spec.id;
|
|
174
|
+
rootPath = c.path;
|
|
175
|
+
}
|
|
176
|
+
else {
|
|
177
|
+
const resolved = await resolveFolderId(drive, spec.name);
|
|
178
|
+
if (!resolved) {
|
|
179
|
+
return { folderId: null, folderName: spec.name, error: `폴더 "${spec.name}" 을 찾지 못함 (색인 대상 root 하위만 조회)` };
|
|
180
|
+
}
|
|
181
|
+
fid = resolved.id;
|
|
182
|
+
rootPath = resolved.path;
|
|
183
|
+
}
|
|
184
|
+
const rootWalk = await walkFolder(drive, fid, depth, pageSize, budget);
|
|
185
|
+
let items = rootWalk.children;
|
|
186
|
+
// 직원별 Drive 폴더 scope 후처리 (admin·미설정 = 전체). root 기준 경로로 트리 필터.
|
|
187
|
+
if (folderScope)
|
|
188
|
+
items = filterTreeByScope(items, rootPath, folderScope);
|
|
189
|
+
return {
|
|
190
|
+
folderId: fid,
|
|
191
|
+
folderName: spec.name ?? null,
|
|
192
|
+
count: items.length,
|
|
193
|
+
childrenTruncated: rootWalk.hasMore || undefined,
|
|
194
|
+
items,
|
|
195
|
+
};
|
|
196
|
+
}));
|
|
197
|
+
const truncated = budget.visits >= MAX_FOLDER_VISITS;
|
|
198
|
+
// 하위호환: 단일 루트 + depth=1 + 배치 미사용 → 기존 응답 형식 그대로(기존 호출자 무손상).
|
|
199
|
+
const isLegacy = !folderIds && !folderNames && depth === 1 && rootSpecs.length === 1;
|
|
200
|
+
if (isLegacy) {
|
|
201
|
+
const r = roots[0];
|
|
202
|
+
if ('error' in r && r.error) {
|
|
203
|
+
return { content: [{ type: 'text', text: `[error] ${r.error}` }], isError: true };
|
|
204
|
+
}
|
|
205
|
+
return { content: [{ type: 'text', text: JSON.stringify(r, null, 2) }] };
|
|
206
|
+
}
|
|
207
|
+
return {
|
|
208
|
+
content: [
|
|
209
|
+
{
|
|
210
|
+
type: 'text',
|
|
211
|
+
text: JSON.stringify({ rootCount: roots.length, depth, folderVisits: budget.visits, truncated, roots }, null, 2),
|
|
212
|
+
},
|
|
213
|
+
],
|
|
214
|
+
};
|
|
215
|
+
}
|
|
216
|
+
catch (e) {
|
|
217
|
+
const msg = e instanceof Error ? e.message : String(e);
|
|
218
|
+
return { content: [{ type: 'text', text: `[error] ${msg}` }], isError: true };
|
|
219
|
+
}
|
|
220
|
+
}
|
|
221
|
+
//# sourceMappingURL=list_drive_folder.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"list_drive_folder.js","sourceRoot":"","sources":["../../../../src/plugins/drive/tools/list_drive_folder.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AACH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAChE,OAAO,EAAE,OAAO,EAAE,MAAM,0BAA0B,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EAAE,mBAAmB,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAC5E,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAE5C,MAAM,WAAW,GAAG,oCAAoC,CAAC;AACzD,MAAM,iBAAiB,GAAG,GAAG,CAAC,CAAC,iCAAiC;AAEhE,MAAM,WAAW,GACf,+FAA+F;IAC/F,yEAAyE;IACzE,mDAAmD;IACnD,6DAA6D;IAC7D,yDAAyD;IACzD,0EAA0E,CAAC;AAE7E,MAAM,WAAW,GAAG,CAAC;KAClB,MAAM,CAAC;IACN,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8CAA8C,CAAC;IAC/F,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4CAA4C,CAAC;IAC/F,SAAS,EAAE,CAAC;SACT,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;SACxB,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,EAAE,CAAC;SACP,QAAQ,EAAE;SACV,QAAQ,CAAC,oDAAoD,CAAC;IACjE,WAAW,EAAE,CAAC;SACX,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;SACxB,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,EAAE,CAAC;SACP,QAAQ,EAAE;SACV,QAAQ,CAAC,+CAA+C,CAAC;IAC5D,KAAK,EAAE,CAAC;SACL,MAAM,EAAE;SACR,GAAG,EAAE;SACL,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,CAAC,CAAC;SACN,OAAO,CAAC,CAAC,CAAC;SACV,QAAQ,CAAC,oEAAoE,CAAC;IACjF,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,kCAAkC,CAAC;CACpG,CAAC;KACD,MAAM,CACL,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,SAAS,EAAE,MAAM,IAAI,CAAC,CAAC,WAAW,EAAE,MAAM,EACjF,EAAE,OAAO,EAAE,0DAA0D,EAAE,CACxE,CAAC;AAEJ,MAAM,CAAC,MAAM,mBAAmB,GAAG,UAAU,CAAC;IAC5C,IAAI,EAAE,mBAAmB;IACzB,WAAW,EAAE,WAAW;IACxB,KAAK,EAAE,WAAW;IAClB,WAAW,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE;IACnC,GAAG,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC;CAC3C,CAAC,CAAC;AAGH,MAAM,UAAU,GAAG,EAAE,iBAAiB,EAAE,IAAI,EAAE,yBAAyB,EAAE,IAAI,EAAW,CAAC;AAiBzF;;;GAGG;AACH,KAAK,UAAU,eAAe,CAAC,KAAkB,EAAE,IAAY;IAC7D,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;QACjC,CAAC,EAAE,WAAW,gBAAgB,CAAC,IAAI,CAAC,qBAAqB,WAAW,uBAAuB;QAC3F,MAAM,EAAE,iBAAiB;QACzB,QAAQ,EAAE,EAAE;QACZ,GAAG,UAAU;KACd,CAAC,CAAC;IACH,KAAK,MAAM,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,EAAE,CAAC;QACrC,IAAI,CAAC,CAAC,CAAC,EAAE;YAAE,SAAS;QACpB,MAAM,CAAC,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,CAAC,GAAG,KAAK,IAAI;YAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IACxD,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;GAGG;AACH,KAAK,UAAU,UAAU,CACvB,KAAkB,EAClB,GAAW,EACX,SAAiB,EACjB,QAAgB,EAChB,MAAc;IAEd,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;QACjC,CAAC,EAAE,IAAI,gBAAgB,CAAC,GAAG,CAAC,kCAAkC;QAC9D,MAAM,EAAE,8DAA8D;QACtE,QAAQ;QACR,OAAO,EAAE,cAAc;QACvB,GAAG,UAAU;KACd,CAAC,CAAC;IAEH,kEAAkE;IAClE,MAAM,OAAO,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC;IACzC,MAAM,KAAK,GAAe,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAC3D,EAAE,EAAE,CAAC,CAAC,EAAE;QACR,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,QAAQ,EAAE,CAAC,CAAC,QAAQ;QACpB,QAAQ,EAAE,CAAC,CAAC,QAAQ,KAAK,WAAW;QACpC,YAAY,EAAE,CAAC,CAAC,YAAY;QAC5B,IAAI,EAAE,CAAC,CAAC,IAAI;KACb,CAAC,CAAC,CAAC;IAEJ,IAAI,SAAS,IAAI,CAAC;QAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;IAExD,mEAAmE;IACnE,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;QACnC,IAAI,CAAC,CAAC,CAAC,QAAQ;YAAE,OAAO,KAAK,CAAC;QAC9B,IAAI,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3B,CAAC,CAAC,QAAQ,GAAG,IAAI,CAAC;YAClB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IAEH,MAAM,OAAO,CAAC,GAAG,CACf,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;QACxB,0DAA0D;QAC1D,IAAI,MAAM,CAAC,MAAM,IAAI,iBAAiB,EAAE,CAAC;YACvC,CAAC,CAAC,SAAS,GAAG,IAAI,CAAC;YACnB,OAAO;QACT,CAAC;QACD,MAAM,CAAC,MAAM,IAAI,CAAC,CAAC;QACnB,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC,EAAG,EAAE,SAAS,GAAG,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC5E,CAAC,CAAC,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC;QAC1B,IAAI,GAAG,CAAC,OAAO;YAAE,CAAC,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAC9C,CAAC,CAAC,CACH,CAAC;IAEF,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;AACtC,CAAC;AAED,4EAA4E;AAC5E,SAAS,iBAAiB,CAAC,KAAiB,EAAE,UAAkB,EAAE,KAAe;IAC/E,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;QACxB,MAAM,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,GAAG,UAAU,IAAI,CAAC,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;QACtE,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;YACf,IAAI,CAAC,CAAC,QAAQ;gBAAE,CAAC,CAAC,QAAQ,GAAG,iBAAiB,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC;YACrE,OAAO,aAAa,CAAC,CAAC,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,KAAK,SAAS,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACxF,CAAC;QACD,OAAO,aAAa,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,IAAiC;IAC3E,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;IACvB,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;IAC/E,MAAM,KAAK,GAAG,MAAM,cAAc,EAAE,CAAC;IACrC,MAAM,WAAW,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAE9C,yBAAyB;IACzB,MAAM,SAAS,GAA0C,EAAE,CAAC;IAC5D,IAAI,QAAQ;QAAE,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC/C,IAAI,UAAU;QAAE,SAAS,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;IACrD,KAAK,MAAM,EAAE,IAAI,SAAS,IAAI,EAAE;QAAE,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;IACzD,KAAK,MAAM,IAAI,IAAI,WAAW,IAAI,EAAE;QAAE,SAAS,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;IAE/D,MAAM,MAAM,GAAW,EAAE,MAAM,EAAE,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,iBAAiB;IAEtE,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,GAAG,CAC7B,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;YAC3B,IAAI,GAAW,CAAC;YAChB,IAAI,QAAgB,CAAC;YACrB,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;gBACZ,4CAA4C;gBAC5C,MAAM,CAAC,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC/C,IAAI,CAAC,CAAC,GAAG,KAAK,IAAI,EAAE,CAAC;oBACnB,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,EAAE,mDAAmD,EAAE,CAAC;gBAC7G,CAAC;gBACD,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC;gBACd,QAAQ,GAAG,CAAC,CAAC,IAAI,CAAC;YACpB,CAAC;iBAAM,CAAC;gBACN,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,KAAK,EAAE,IAAI,CAAC,IAAK,CAAC,CAAC;gBAC1D,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,OAAO,IAAI,CAAC,IAAI,+BAA+B,EAAE,CAAC;gBAC3G,CAAC;gBACD,GAAG,GAAG,QAAQ,CAAC,EAAE,CAAC;gBAClB,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC;YAC3B,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,UAAU,CAAC,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;YACvE,IAAI,KAAK,GAAG,QAAQ,CAAC,QAAQ,CAAC;YAC9B,8DAA8D;YAC9D,IAAI,WAAW;gBAAE,KAAK,GAAG,iBAAiB,CAAC,KAAK,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC;YACzE,OAAO;gBACL,QAAQ,EAAE,GAAG;gBACb,UAAU,EAAE,IAAI,CAAC,IAAI,IAAI,IAAI;gBAC7B,KAAK,EAAE,KAAK,CAAC,MAAM;gBACnB,iBAAiB,EAAE,QAAQ,CAAC,OAAO,IAAI,SAAS;gBAChD,KAAK;aACN,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;QAEF,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,IAAI,iBAAiB,CAAC;QAErD,6DAA6D;QAC7D,MAAM,QAAQ,GAAG,CAAC,SAAS,IAAI,CAAC,WAAW,IAAI,KAAK,KAAK,CAAC,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC;QACrF,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACnB,IAAI,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,CAAC;gBAC5B,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAC7F,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;QACpF,CAAC;QAED,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAClB,EAAE,SAAS,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,EACjF,IAAI,EACJ,CAAC,CACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,CAAU,EAAE,CAAC;QACpB,MAAM,GAAG,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACvD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,WAAW,GAAG,EAAE,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IACzF,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* read_drive_file(fileId | fileIds, maxChars?) — Drive 파일 본문 read.
|
|
3
|
+
*
|
|
4
|
+
* Markdown / text 는 raw media. Google Docs/Sheets/Slides 는 export.
|
|
5
|
+
*
|
|
6
|
+
* 배치: fileIds 로 여러 파일을 한 번에 **병렬** read.
|
|
7
|
+
* 순차 호출(파일당 MCP 왕복 + Drive API 2회)을 1회 호출로 합쳐 per-call 네트워크
|
|
8
|
+
* floor 를 N→1 로 줄이고 Drive I/O 도 동시 처리. 원문 여러 개를 비교/요약할 때 핵심.
|
|
9
|
+
*/
|
|
10
|
+
import { z } from 'zod';
|
|
11
|
+
declare const inputSchema: z.ZodEffects<z.ZodObject<{
|
|
12
|
+
fileId: z.ZodOptional<z.ZodString>;
|
|
13
|
+
fileIds: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
|
|
14
|
+
maxChars: z.ZodDefault<z.ZodNumber>;
|
|
15
|
+
}, "strip", z.ZodTypeAny, {
|
|
16
|
+
maxChars: number;
|
|
17
|
+
fileId?: string | undefined;
|
|
18
|
+
fileIds?: string[] | undefined;
|
|
19
|
+
}, {
|
|
20
|
+
fileId?: string | undefined;
|
|
21
|
+
fileIds?: string[] | undefined;
|
|
22
|
+
maxChars?: number | undefined;
|
|
23
|
+
}>, {
|
|
24
|
+
maxChars: number;
|
|
25
|
+
fileId?: string | undefined;
|
|
26
|
+
fileIds?: string[] | undefined;
|
|
27
|
+
}, {
|
|
28
|
+
fileId?: string | undefined;
|
|
29
|
+
fileIds?: string[] | undefined;
|
|
30
|
+
maxChars?: number | undefined;
|
|
31
|
+
}>;
|
|
32
|
+
export declare const readDriveFileTool: import("../../../plugin/define.js").Tool<z.ZodEffects<z.ZodObject<{
|
|
33
|
+
fileId: z.ZodOptional<z.ZodString>;
|
|
34
|
+
fileIds: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
|
|
35
|
+
maxChars: z.ZodDefault<z.ZodNumber>;
|
|
36
|
+
}, "strip", z.ZodTypeAny, {
|
|
37
|
+
maxChars: number;
|
|
38
|
+
fileId?: string | undefined;
|
|
39
|
+
fileIds?: string[] | undefined;
|
|
40
|
+
}, {
|
|
41
|
+
fileId?: string | undefined;
|
|
42
|
+
fileIds?: string[] | undefined;
|
|
43
|
+
maxChars?: number | undefined;
|
|
44
|
+
}>, {
|
|
45
|
+
maxChars: number;
|
|
46
|
+
fileId?: string | undefined;
|
|
47
|
+
fileIds?: string[] | undefined;
|
|
48
|
+
}, {
|
|
49
|
+
fileId?: string | undefined;
|
|
50
|
+
fileIds?: string[] | undefined;
|
|
51
|
+
maxChars?: number | undefined;
|
|
52
|
+
}>>;
|
|
53
|
+
export declare function handleReadDriveFile(args: z.infer<typeof inputSchema>): Promise<{
|
|
54
|
+
content: {
|
|
55
|
+
type: "text";
|
|
56
|
+
text: string;
|
|
57
|
+
}[];
|
|
58
|
+
isError?: undefined;
|
|
59
|
+
} | {
|
|
60
|
+
content: {
|
|
61
|
+
type: "text";
|
|
62
|
+
text: string;
|
|
63
|
+
}[];
|
|
64
|
+
isError: boolean;
|
|
65
|
+
}>;
|
|
66
|
+
export {};
|
|
@@ -0,0 +1,206 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* read_drive_file(fileId | fileIds, maxChars?) — Drive 파일 본문 read.
|
|
3
|
+
*
|
|
4
|
+
* Markdown / text 는 raw media. Google Docs/Sheets/Slides 는 export.
|
|
5
|
+
*
|
|
6
|
+
* 배치: fileIds 로 여러 파일을 한 번에 **병렬** read.
|
|
7
|
+
* 순차 호출(파일당 MCP 왕복 + Drive API 2회)을 1회 호출로 합쳐 per-call 네트워크
|
|
8
|
+
* floor 를 N→1 로 줄이고 Drive I/O 도 동시 처리. 원문 여러 개를 비교/요약할 때 핵심.
|
|
9
|
+
*/
|
|
10
|
+
import { z } from 'zod';
|
|
11
|
+
import { defineTool } from '../../../plugin/define.js';
|
|
12
|
+
import { getDriveClient } from '../client.js';
|
|
13
|
+
import { getAuth } from '../../../core/context.js';
|
|
14
|
+
import { allowedDriveFolders, isPathInScope } from '../../../core/scope.js';
|
|
15
|
+
import { classifyAndPath } from '../path.js';
|
|
16
|
+
import { extractText } from '../extract_text.js';
|
|
17
|
+
import { getQdrantClient, DRIVE_COLLECTION } from '../qdrant.js';
|
|
18
|
+
import { MAX_FILE_BYTES } from '../settings.js';
|
|
19
|
+
const DESCRIPTION = 'Google Drive 파일 본문 read. markdown/text 는 raw, Google Docs/Sheets/Slides 는 export, ' +
|
|
20
|
+
'**PDF·Word(docx)·PowerPoint(pptx)·Excel(xlsx)·이미지는 본문 텍스트로 추출**(스캔 PDF·이미지는 OCR). ' +
|
|
21
|
+
'search_drive / semantic_search_drive 응답의 id (File ID) 사용. ' +
|
|
22
|
+
'semantic_search_drive 로 관련 청크를 찾은 뒤 원문 전체 맥락이 필요할 때 호출하는 2단계 검색의 2단계. ' +
|
|
23
|
+
'**여러 문서를 읽어야 하면 fileId 를 하나씩 여러 번 호출하지 말고 fileIds=[...] 로 한 번에 (병렬 — 순차보다 훨씬 빠름).** ' +
|
|
24
|
+
'큰 파일은 maxChars 로 truncate.';
|
|
25
|
+
const inputSchema = z
|
|
26
|
+
.object({
|
|
27
|
+
fileId: z.string().min(1).optional().describe('Drive file ID (단일). fileIds 와 둘 중 하나.'),
|
|
28
|
+
fileIds: z
|
|
29
|
+
.array(z.string().min(1))
|
|
30
|
+
.min(1)
|
|
31
|
+
.max(10)
|
|
32
|
+
.optional()
|
|
33
|
+
.describe('여러 Drive file ID 를 한 번에 병렬 read (1~10개). 순차 호출보다 훨씬 빠름. fileId 와 둘 중 하나.'),
|
|
34
|
+
maxChars: z.number().int().min(1000).max(200000).default(50000).describe('파일당 최대 반환 문자 수 (1000-200000). default 50000.'),
|
|
35
|
+
})
|
|
36
|
+
.refine((d) => d.fileId || (d.fileIds && d.fileIds.length > 0), {
|
|
37
|
+
message: 'fileId 또는 fileIds 중 하나는 필수',
|
|
38
|
+
});
|
|
39
|
+
export const readDriveFileTool = defineTool({
|
|
40
|
+
name: 'read_drive_file',
|
|
41
|
+
description: DESCRIPTION,
|
|
42
|
+
input: inputSchema,
|
|
43
|
+
annotations: { readOnlyHint: true },
|
|
44
|
+
run: (args) => handleReadDriveFile(args),
|
|
45
|
+
});
|
|
46
|
+
const GOOGLE_DOC_EXPORT = {
|
|
47
|
+
'application/vnd.google-apps.document': 'text/markdown',
|
|
48
|
+
'application/vnd.google-apps.spreadsheet': 'text/csv',
|
|
49
|
+
'application/vnd.google-apps.presentation': 'text/plain',
|
|
50
|
+
};
|
|
51
|
+
/** raw text 로 읽으면 깨지는 바이너리 — extract_text 파이프라인으로 본문 추출해야 하는 형식. */
|
|
52
|
+
function isExtractableBinary(mime) {
|
|
53
|
+
return (mime === 'application/pdf' ||
|
|
54
|
+
mime.includes('wordprocessingml') || // .docx
|
|
55
|
+
mime.includes('presentationml') || // .pptx
|
|
56
|
+
mime.includes('spreadsheetml') || // .xlsx
|
|
57
|
+
mime.startsWith('image/'));
|
|
58
|
+
}
|
|
59
|
+
/**
|
|
60
|
+
* 색인된 파일이면 Qdrant 청크를 chunk_index 순으로 재조립해 원문 복원 — 색인 시 이미 추출(스캔본은 OCR)한
|
|
61
|
+
* 텍스트를 재사용하므로 재다운로드·재OCR 비용 0(스캔 PDF 를 읽을 때마다 Mathpix 를 다시 태우지 않음).
|
|
62
|
+
* 미색인이거나 조회 실패면 null(→ extract_text 폴백). 청크 overlap 접두("…") 는 best-effort 제거.
|
|
63
|
+
*/
|
|
64
|
+
async function reassembleFromIndex(fileId) {
|
|
65
|
+
try {
|
|
66
|
+
const res = await (await getQdrantClient()).scroll(DRIVE_COLLECTION, {
|
|
67
|
+
filter: { must: [{ key: 'file_id', match: { value: fileId } }] },
|
|
68
|
+
limit: 1000,
|
|
69
|
+
with_payload: true,
|
|
70
|
+
with_vector: false,
|
|
71
|
+
});
|
|
72
|
+
const chunks = (res.points ?? [])
|
|
73
|
+
.map((pt) => {
|
|
74
|
+
const p = pt.payload;
|
|
75
|
+
return {
|
|
76
|
+
content: typeof p?.content === 'string' ? p.content : null,
|
|
77
|
+
idx: typeof p?.chunk_index === 'number' ? p.chunk_index : 0,
|
|
78
|
+
};
|
|
79
|
+
})
|
|
80
|
+
.filter((c) => c.content !== null)
|
|
81
|
+
.sort((a, b) => a.idx - b.idx);
|
|
82
|
+
if (chunks.length === 0)
|
|
83
|
+
return null;
|
|
84
|
+
return chunks
|
|
85
|
+
.map((c, i) => {
|
|
86
|
+
if (i === 0 || !c.content.startsWith('…'))
|
|
87
|
+
return c.content;
|
|
88
|
+
const nl = c.content.indexOf('\n\n'); // overlap 접두(…tail\n\n) 제거
|
|
89
|
+
return nl >= 0 ? c.content.slice(nl + 2) : c.content;
|
|
90
|
+
})
|
|
91
|
+
.join('\n\n');
|
|
92
|
+
}
|
|
93
|
+
catch {
|
|
94
|
+
return null;
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
/** 파일 1개 read → 구조화 객체 (배치에서 파일별 에러 격리용으로 throw 대신 error 필드 반환). */
|
|
98
|
+
async function readOne(drive, fileId, maxChars, folderScope) {
|
|
99
|
+
try {
|
|
100
|
+
const meta = await drive.files.get({
|
|
101
|
+
fileId,
|
|
102
|
+
fields: 'id, name, mimeType, parents, size, modifiedTime, webViewLink',
|
|
103
|
+
});
|
|
104
|
+
const m = meta.data;
|
|
105
|
+
const mimeType = m.mimeType || '';
|
|
106
|
+
if (mimeType === 'application/vnd.google-apps.folder') {
|
|
107
|
+
return { fileId, error: 'fileId is a folder. use list_drive_folder instead.' };
|
|
108
|
+
}
|
|
109
|
+
// 보안: dev/biz root 하위 + 제외 폴더(PII·계약·HR·deprecated) 미경유 + 직원 폴더 scope.
|
|
110
|
+
// search_drive / semantic_search_drive 와 동일 가드 — 임의 File ID 직접 입력으로
|
|
111
|
+
// 색인 대상 밖·민감 폴더·권한 밖 파일을 읽는 우회를 차단한다.
|
|
112
|
+
const { cls, path } = await classifyAndPath(drive, m.parents ?? undefined);
|
|
113
|
+
if (cls !== 'ok' || !isPathInScope(path, folderScope)) {
|
|
114
|
+
return {
|
|
115
|
+
fileId,
|
|
116
|
+
error: '[forbidden] 접근 범위 밖 파일입니다 (색인 대상 root 밖 / 민감 폴더 / 직원 권한 밖).',
|
|
117
|
+
};
|
|
118
|
+
}
|
|
119
|
+
// 크기 상한 — 파일을 통째로 메모리에 올리므로 거대한 파일(영상·대용량 PDF)이 서버를 죽이지 않게 막는다.
|
|
120
|
+
// 색인 재조립본이 있으면 그것으로 읽는다(재다운로드 없음). Google 문서는 size 가 없어 해당 없음.
|
|
121
|
+
const tooLarge = !!m.size && Number(m.size) > MAX_FILE_BYTES;
|
|
122
|
+
const tooLargeError = {
|
|
123
|
+
fileId,
|
|
124
|
+
error: `파일이 너무 커서 읽지 않습니다 (${m.size}B > ${MAX_FILE_BYTES}B, DRIVE_MAX_FILE_BYTES).`,
|
|
125
|
+
};
|
|
126
|
+
let body;
|
|
127
|
+
let extractMethod;
|
|
128
|
+
if (GOOGLE_DOC_EXPORT[mimeType]) {
|
|
129
|
+
const res = await drive.files.export({ fileId, mimeType: GOOGLE_DOC_EXPORT[mimeType] }, { responseType: 'text' });
|
|
130
|
+
body = String(res.data ?? '');
|
|
131
|
+
}
|
|
132
|
+
else if (isExtractableBinary(mimeType)) {
|
|
133
|
+
// PDF·docx·pptx·xlsx·image. 색인된 파일이면 Qdrant 청크 재조립(색인 시 추출·OCR 한 텍스트 재사용
|
|
134
|
+
// → 재다운로드·재OCR 0)을 우선. 미색인이면 extract_text 파이프라인(pdf-parse·mammoth·Mathpix OCR 등).
|
|
135
|
+
// 예전엔 바이너리를 String() 해 깨진 텍스트를 반환했다(검색 챗봇 PDF 읽기 실패의 원인).
|
|
136
|
+
const reassembled = await reassembleFromIndex(fileId);
|
|
137
|
+
if (reassembled && reassembled.trim()) {
|
|
138
|
+
body = reassembled;
|
|
139
|
+
extractMethod = 'index-reassembled';
|
|
140
|
+
}
|
|
141
|
+
else {
|
|
142
|
+
if (tooLarge)
|
|
143
|
+
return tooLargeError;
|
|
144
|
+
const r = await extractText(drive, fileId, mimeType);
|
|
145
|
+
if (r.needsOcr || !r.text.trim()) {
|
|
146
|
+
return {
|
|
147
|
+
fileId,
|
|
148
|
+
error: `본문 추출 불가/미지원입니다 (mimeType=${mimeType}). 스캔 품질이 낮거나 지원하지 않는 형식일 수 있습니다.`,
|
|
149
|
+
};
|
|
150
|
+
}
|
|
151
|
+
body = r.text;
|
|
152
|
+
extractMethod = r.method;
|
|
153
|
+
}
|
|
154
|
+
}
|
|
155
|
+
else {
|
|
156
|
+
if (tooLarge)
|
|
157
|
+
return tooLargeError;
|
|
158
|
+
const res = await drive.files.get({ fileId, alt: 'media' }, { responseType: 'text' });
|
|
159
|
+
body = String(res.data ?? '');
|
|
160
|
+
}
|
|
161
|
+
let truncated = false;
|
|
162
|
+
if (body.length > maxChars) {
|
|
163
|
+
body = body.slice(0, maxChars);
|
|
164
|
+
truncated = true;
|
|
165
|
+
}
|
|
166
|
+
return {
|
|
167
|
+
meta: {
|
|
168
|
+
id: m.id,
|
|
169
|
+
name: m.name,
|
|
170
|
+
mimeType: m.mimeType,
|
|
171
|
+
size: m.size,
|
|
172
|
+
modifiedTime: m.modifiedTime,
|
|
173
|
+
viewUrl: m.webViewLink,
|
|
174
|
+
...(extractMethod ? { extractMethod } : {}),
|
|
175
|
+
},
|
|
176
|
+
truncated,
|
|
177
|
+
charCount: body.length,
|
|
178
|
+
body,
|
|
179
|
+
};
|
|
180
|
+
}
|
|
181
|
+
catch (e) {
|
|
182
|
+
return { fileId, error: e instanceof Error ? e.message : String(e) };
|
|
183
|
+
}
|
|
184
|
+
}
|
|
185
|
+
export async function handleReadDriveFile(args) {
|
|
186
|
+
const auth = getAuth();
|
|
187
|
+
const { fileId, fileIds, maxChars } = args;
|
|
188
|
+
const drive = await getDriveClient();
|
|
189
|
+
const folderScope = allowedDriveFolders(auth);
|
|
190
|
+
// 배치: fileIds 면 병렬 read (per-call floor 1회 + Drive I/O 동시).
|
|
191
|
+
if (fileIds && fileIds.length > 0) {
|
|
192
|
+
const files = await Promise.all(fileIds.map((id) => readOne(drive, id, maxChars, folderScope)));
|
|
193
|
+
return {
|
|
194
|
+
content: [
|
|
195
|
+
{ type: 'text', text: JSON.stringify({ count: files.length, files }, null, 2) },
|
|
196
|
+
],
|
|
197
|
+
};
|
|
198
|
+
}
|
|
199
|
+
// 단일 (하위호환): 기존 응답 형식 유지.
|
|
200
|
+
const result = await readOne(drive, fileId, maxChars, folderScope);
|
|
201
|
+
if ('error' in result) {
|
|
202
|
+
return { content: [{ type: 'text', text: `[error] ${result.error}` }], isError: true };
|
|
203
|
+
}
|
|
204
|
+
return { content: [{ type: 'text', text: JSON.stringify(result, null, 2) }] };
|
|
205
|
+
}
|
|
206
|
+
//# sourceMappingURL=read_drive_file.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"read_drive_file.js","sourceRoot":"","sources":["../../../../src/plugins/drive/tools/read_drive_file.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAE,MAAM,0BAA0B,CAAC;AACnD,OAAO,EAAE,mBAAmB,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAC5E,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AACjE,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAEhD,MAAM,WAAW,GACf,oFAAoF;IACpF,oFAAoF;IACpF,4DAA4D;IAC5D,wEAAwE;IACxE,sFAAsF;IACtF,4BAA4B,CAAC;AAE/B,MAAM,WAAW,GAAG,CAAC;KAClB,MAAM,CAAC;IACN,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uCAAuC,CAAC;IACtF,OAAO,EAAE,CAAC;SACP,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;SACxB,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,EAAE,CAAC;SACP,QAAQ,EAAE;SACV,QAAQ,CAAC,0EAA0E,CAAC;IACvF,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,8CAA8C,CAAC;CACzH,CAAC;KACD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE;IAC9D,OAAO,EAAE,4BAA4B;CACtC,CAAC,CAAC;AAEL,MAAM,CAAC,MAAM,iBAAiB,GAAG,UAAU,CAAC;IAC1C,IAAI,EAAE,iBAAiB;IACvB,WAAW,EAAE,WAAW;IACxB,KAAK,EAAE,WAAW;IAClB,WAAW,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE;IACnC,GAAG,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC;CACzC,CAAC,CAAC;AAEH,MAAM,iBAAiB,GAA2B;IAChD,sCAAsC,EAAE,eAAe;IACvD,yCAAyC,EAAE,UAAU;IACrD,0CAA0C,EAAE,YAAY;CACzD,CAAC;AAEF,oEAAoE;AACpE,SAAS,mBAAmB,CAAC,IAAY;IACvC,OAAO,CACL,IAAI,KAAK,iBAAiB;QAC1B,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,IAAI,QAAQ;QAC7C,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,QAAQ;QAC3C,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,IAAI,QAAQ;QAC1C,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAC1B,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,KAAK,UAAU,mBAAmB,CAAC,MAAc;IAC/C,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,eAAe,EAAE,CAAC,CAAC,MAAM,CAAC,gBAAgB,EAAE;YACnE,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE;YAChE,KAAK,EAAE,IAAI;YACX,YAAY,EAAE,IAAI;YAClB,WAAW,EAAE,KAAK;SACnB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,CAAC,GAAG,CAAC,MAAM,IAAI,EAAE,CAAC;aAC9B,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE;YACV,MAAM,CAAC,GAAG,EAAE,CAAC,OAA8C,CAAC;YAC5D,OAAO;gBACL,OAAO,EAAE,OAAO,CAAC,EAAE,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAE,CAAC,CAAC,OAAkB,CAAC,CAAC,CAAC,IAAI;gBACtE,GAAG,EAAE,OAAO,CAAC,EAAE,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAE,CAAC,CAAC,WAAsB,CAAC,CAAC,CAAC,CAAC;aACxE,CAAC;QACJ,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,CAAC,EAAyC,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,IAAI,CAAC;aACxE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QACrC,OAAO,MAAM;aACV,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACZ,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,OAAO,CAAC,CAAC,OAAO,CAAC;YAC5D,MAAM,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,2BAA2B;YACjE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;QACvD,CAAC,CAAC;aACD,IAAI,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAID,oEAAoE;AACpE,KAAK,UAAU,OAAO,CACpB,KAAkB,EAClB,MAAc,EACd,QAAgB,EAChB,WAA4B;IAE5B,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC;YACjC,MAAM;YACN,MAAM,EAAE,8DAA8D;SACvE,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC;QACpB,MAAM,QAAQ,GAAG,CAAC,CAAC,QAAQ,IAAI,EAAE,CAAC;QAElC,IAAI,QAAQ,KAAK,oCAAoC,EAAE,CAAC;YACtD,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,oDAAoD,EAAE,CAAC;QACjF,CAAC;QAED,uEAAuE;QACvE,oEAAoE;QACpE,sCAAsC;QACtC,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,IAAI,SAAS,CAAC,CAAC;QAC3E,IAAI,GAAG,KAAK,IAAI,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,WAAW,CAAC,EAAE,CAAC;YACtD,OAAO;gBACL,MAAM;gBACN,KAAK,EAAE,6DAA6D;aACrE,CAAC;QACJ,CAAC;QAED,gEAAgE;QAChE,+DAA+D;QAC/D,MAAM,QAAQ,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC;QAC7D,MAAM,aAAa,GAAG;YACpB,MAAM;YACN,KAAK,EAAE,sBAAsB,CAAC,CAAC,IAAI,OAAO,cAAc,2BAA2B;SACpF,CAAC;QAEF,IAAI,IAAY,CAAC;QACjB,IAAI,aAAiC,CAAC;QACtC,IAAI,iBAAiB,CAAC,QAAQ,CAAC,EAAE,CAAC;YAChC,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,MAAM,CAClC,EAAE,MAAM,EAAE,QAAQ,EAAE,iBAAiB,CAAC,QAAQ,CAAC,EAAE,EACjD,EAAE,YAAY,EAAE,MAAM,EAAE,CACzB,CAAC;YACF,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;QAChC,CAAC;aAAM,IAAI,mBAAmB,CAAC,QAAQ,CAAC,EAAE,CAAC;YACzC,yEAAyE;YACzE,kFAAkF;YAClF,0DAA0D;YAC1D,MAAM,WAAW,GAAG,MAAM,mBAAmB,CAAC,MAAM,CAAC,CAAC;YACtD,IAAI,WAAW,IAAI,WAAW,CAAC,IAAI,EAAE,EAAE,CAAC;gBACtC,IAAI,GAAG,WAAW,CAAC;gBACnB,aAAa,GAAG,mBAAmB,CAAC;YACtC,CAAC;iBAAM,CAAC;gBACN,IAAI,QAAQ;oBAAE,OAAO,aAAa,CAAC;gBACnC,MAAM,CAAC,GAAG,MAAM,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;gBACrD,IAAI,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;oBACjC,OAAO;wBACL,MAAM;wBACN,KAAK,EAAE,6BAA6B,QAAQ,mCAAmC;qBAChF,CAAC;gBACJ,CAAC;gBACD,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC;gBACd,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC;YAC3B,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,QAAQ;gBAAE,OAAO,aAAa,CAAC;YACnC,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC,CAAC;YACtF,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;QAChC,CAAC;QAED,IAAI,SAAS,GAAG,KAAK,CAAC;QACtB,IAAI,IAAI,CAAC,MAAM,GAAG,QAAQ,EAAE,CAAC;YAC3B,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;YAC/B,SAAS,GAAG,IAAI,CAAC;QACnB,CAAC;QAED,OAAO;YACL,IAAI,EAAE;gBACJ,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,QAAQ,EAAE,CAAC,CAAC,QAAQ;gBACpB,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,YAAY,EAAE,CAAC,CAAC,YAAY;gBAC5B,OAAO,EAAE,CAAC,CAAC,WAAW;gBACtB,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC5C;YACD,SAAS;YACT,SAAS,EAAE,IAAI,CAAC,MAAM;YACtB,IAAI;SACL,CAAC;IACJ,CAAC;IAAC,OAAO,CAAU,EAAE,CAAC;QACpB,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;IACvE,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,IAAiC;IACzE,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;IACvB,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;IAC3C,MAAM,KAAK,GAAG,MAAM,cAAc,EAAE,CAAC;IACrC,MAAM,WAAW,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAE9C,4DAA4D;IAC5D,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAClC,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,GAAG,CAC7B,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC,CAC/D,CAAC;QACF,OAAO;YACL,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE;aACzF;SACF,CAAC;IACJ,CAAC;IAED,0BAA0B;IAC1B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,EAAE,MAAO,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC;IACpE,IAAI,OAAO,IAAI,MAAM,EAAE,CAAC;QACtB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,WAAW,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAClG,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;AACzF,CAAC"}
|
|
@@ -0,0 +1,52 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* search_drive(query, mode?, mimeType?, pageSize?) — Google Drive 파일 검색.
|
|
3
|
+
*
|
|
4
|
+
* 권한 그룹: drive. 폴더 범위는 app_users.allowed_drive_folders. Drive 접근은 서버 자격증명.
|
|
5
|
+
*/
|
|
6
|
+
import { z } from 'zod';
|
|
7
|
+
declare const inputSchema: z.ZodObject<{
|
|
8
|
+
query: z.ZodString;
|
|
9
|
+
mode: z.ZodDefault<z.ZodEnum<["title", "fullText"]>>;
|
|
10
|
+
mimeType: z.ZodOptional<z.ZodString>;
|
|
11
|
+
pageSize: z.ZodDefault<z.ZodNumber>;
|
|
12
|
+
}, "strip", z.ZodTypeAny, {
|
|
13
|
+
query: string;
|
|
14
|
+
mode: "title" | "fullText";
|
|
15
|
+
pageSize: number;
|
|
16
|
+
mimeType?: string | undefined;
|
|
17
|
+
}, {
|
|
18
|
+
query: string;
|
|
19
|
+
mimeType?: string | undefined;
|
|
20
|
+
mode?: "title" | "fullText" | undefined;
|
|
21
|
+
pageSize?: number | undefined;
|
|
22
|
+
}>;
|
|
23
|
+
export declare const searchDriveTool: import("../../../plugin/define.js").Tool<z.ZodObject<{
|
|
24
|
+
query: z.ZodString;
|
|
25
|
+
mode: z.ZodDefault<z.ZodEnum<["title", "fullText"]>>;
|
|
26
|
+
mimeType: z.ZodOptional<z.ZodString>;
|
|
27
|
+
pageSize: z.ZodDefault<z.ZodNumber>;
|
|
28
|
+
}, "strip", z.ZodTypeAny, {
|
|
29
|
+
query: string;
|
|
30
|
+
mode: "title" | "fullText";
|
|
31
|
+
pageSize: number;
|
|
32
|
+
mimeType?: string | undefined;
|
|
33
|
+
}, {
|
|
34
|
+
query: string;
|
|
35
|
+
mimeType?: string | undefined;
|
|
36
|
+
mode?: "title" | "fullText" | undefined;
|
|
37
|
+
pageSize?: number | undefined;
|
|
38
|
+
}>>;
|
|
39
|
+
export declare function handleSearchDrive(args: z.infer<typeof inputSchema>): Promise<{
|
|
40
|
+
content: {
|
|
41
|
+
type: "text";
|
|
42
|
+
text: string;
|
|
43
|
+
}[];
|
|
44
|
+
isError?: undefined;
|
|
45
|
+
} | {
|
|
46
|
+
content: {
|
|
47
|
+
type: "text";
|
|
48
|
+
text: string;
|
|
49
|
+
}[];
|
|
50
|
+
isError: boolean;
|
|
51
|
+
}>;
|
|
52
|
+
export {};
|
|
@@ -0,0 +1,78 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* search_drive(query, mode?, mimeType?, pageSize?) — Google Drive 파일 검색.
|
|
3
|
+
*
|
|
4
|
+
* 권한 그룹: drive. 폴더 범위는 app_users.allowed_drive_folders. Drive 접근은 서버 자격증명.
|
|
5
|
+
*/
|
|
6
|
+
import { z } from 'zod';
|
|
7
|
+
import { defineTool } from '../../../plugin/define.js';
|
|
8
|
+
import { getDriveClient, escapeDriveQuery } from '../client.js';
|
|
9
|
+
import { getAuth } from '../../../core/context.js';
|
|
10
|
+
import { allowedDriveFolders, isPathInScope } from '../../../core/scope.js';
|
|
11
|
+
import { classifyAndPath } from '../path.js';
|
|
12
|
+
const DESCRIPTION = 'Google Drive 에서 파일 검색. mode="title" 은 파일명, "fullText" 는 본문까지. ' +
|
|
13
|
+
'Unified search across your indexed Drive assets. ' +
|
|
14
|
+
'검색 범위는 설정한 Drive root 하위로 한정되며, 설정의 제외 폴더(excludeFolders)를 거치는 파일은 결과에서 빠진다. ' +
|
|
15
|
+
'search → read_drive_file 순서로 사용.';
|
|
16
|
+
const inputSchema = z.object({
|
|
17
|
+
query: z.string().min(1).describe('검색어'),
|
|
18
|
+
mode: z.enum(['title', 'fullText']).default('title').describe('"title" 또는 "fullText". default "title" (신뢰성 ↑).'),
|
|
19
|
+
mimeType: z.string().optional().describe('선택. 예: "text/markdown" 또는 "application/vnd.google-apps.folder"'),
|
|
20
|
+
pageSize: z.number().int().min(1).max(50).default(10).describe('최대 결과 수 (1-50). default 10.'),
|
|
21
|
+
});
|
|
22
|
+
export const searchDriveTool = defineTool({
|
|
23
|
+
name: 'search_drive',
|
|
24
|
+
description: DESCRIPTION,
|
|
25
|
+
input: inputSchema,
|
|
26
|
+
annotations: { readOnlyHint: true },
|
|
27
|
+
run: (args) => handleSearchDrive(args),
|
|
28
|
+
});
|
|
29
|
+
export async function handleSearchDrive(args) {
|
|
30
|
+
const auth = getAuth();
|
|
31
|
+
const { query, mode, mimeType, pageSize } = args;
|
|
32
|
+
const drive = await getDriveClient();
|
|
33
|
+
const escaped = escapeDriveQuery(query);
|
|
34
|
+
let q = mode === 'fullText' ? `fullText contains '${escaped}'` : `name contains '${escaped}'`;
|
|
35
|
+
if (mimeType) {
|
|
36
|
+
q += ` and mimeType = '${escapeDriveQuery(mimeType)}'`;
|
|
37
|
+
}
|
|
38
|
+
q += ' and trashed = false';
|
|
39
|
+
try {
|
|
40
|
+
const res = await drive.files.list({
|
|
41
|
+
q,
|
|
42
|
+
fields: 'files(id, name, mimeType, parents, modifiedTime, size, webViewLink, owners(emailAddress))',
|
|
43
|
+
pageSize,
|
|
44
|
+
orderBy: 'modifiedTime desc',
|
|
45
|
+
});
|
|
46
|
+
const raw = (res.data.files || []).map((f) => ({
|
|
47
|
+
id: f.id,
|
|
48
|
+
name: f.name,
|
|
49
|
+
mimeType: f.mimeType,
|
|
50
|
+
parents: f.parents ?? undefined,
|
|
51
|
+
modifiedTime: f.modifiedTime,
|
|
52
|
+
size: f.size,
|
|
53
|
+
viewUrl: f.webViewLink,
|
|
54
|
+
owners: (f.owners || []).map((o) => o.emailAddress).filter(Boolean),
|
|
55
|
+
}));
|
|
56
|
+
// 보안: 민감/제외 폴더(PII·계약·HR·deprecated)·색인 대상 밖 제거 + 직원별 Drive 폴더 scope.
|
|
57
|
+
const folderScope = allowedDriveFolders(auth);
|
|
58
|
+
const cps = await Promise.all(raw.map((f) => classifyAndPath(drive, f.parents)));
|
|
59
|
+
const files = raw.filter((_, i) => cps[i].cls === 'ok' && (!folderScope || isPathInScope(cps[i].path, folderScope)));
|
|
60
|
+
const filtered = raw.length - files.length;
|
|
61
|
+
return {
|
|
62
|
+
content: [
|
|
63
|
+
{
|
|
64
|
+
type: 'text',
|
|
65
|
+
text: JSON.stringify({ count: files.length, mode, query, filteredOut: filtered, files }, null, 2),
|
|
66
|
+
},
|
|
67
|
+
],
|
|
68
|
+
};
|
|
69
|
+
}
|
|
70
|
+
catch (e) {
|
|
71
|
+
const msg = e instanceof Error ? e.message : String(e);
|
|
72
|
+
return {
|
|
73
|
+
content: [{ type: 'text', text: `[error] ${msg}` }],
|
|
74
|
+
isError: true,
|
|
75
|
+
};
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
//# sourceMappingURL=search_drive.js.map
|