@llmindset/hf-mcp-server 0.2.3 → 0.2.4
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/server/mcp-server.d.ts.map +1 -1
- package/dist/server/mcp-server.js +6 -0
- package/dist/server/mcp-server.js.map +1 -1
- package/dist/server/utils/hf-dataset-transport.d.ts +8 -6
- package/dist/server/utils/hf-dataset-transport.d.ts.map +1 -1
- package/dist/server/utils/hf-dataset-transport.js +64 -25
- package/dist/server/utils/hf-dataset-transport.js.map +1 -1
- package/dist/server/utils/query-logger.d.ts +15 -0
- package/dist/server/utils/query-logger.d.ts.map +1 -0
- package/dist/server/utils/query-logger.js +61 -0
- package/dist/server/utils/query-logger.js.map +1 -0
- package/dist/server/utils/query-logging-wrapper.d.ts +2 -0
- package/dist/server/utils/query-logging-wrapper.d.ts.map +1 -0
- package/dist/server/utils/query-logging-wrapper.js +28 -0
- package/dist/server/utils/query-logging-wrapper.js.map +1 -0
- package/package.json +2 -2
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"mcp-server.d.ts","sourceRoot":"","sources":["../../src/server/mcp-server.ts"],"names":[],"mappings":"AAgDA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AACnE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;
|
|
1
|
+
{"version":3,"file":"mcp-server.d.ts","sourceRoot":"","sources":["../../src/server/mcp-server.ts"],"names":[],"mappings":"AAgDA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AACnE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAGjD,OAAO,EAAuB,KAAK,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAK9E,eAAO,MAAM,gBAAgB,EAAE,WAG9B,CAAC;AAGF,eAAO,MAAM,oBAAoB,EAAE,WAGlC,CAAC;AAQF,eAAO,MAAM,mBAAmB,GAAI,oBAAoB,SAAS,EAAE,iBAAiB,YAAY,KAAG,aAoXlG,CAAC"}
|
|
@@ -3,6 +3,7 @@ import { createRequire } from 'module';
|
|
|
3
3
|
import { whoAmI } from '@huggingface/hub';
|
|
4
4
|
import { SpaceSearchTool, formatSearchResults, SEMANTIC_SEARCH_TOOL_CONFIG, ModelSearchTool, MODEL_SEARCH_TOOL_CONFIG, ModelDetailTool, MODEL_DETAIL_TOOL_CONFIG, PaperSearchTool, PAPER_SEARCH_TOOL_CONFIG, DatasetSearchTool, DATASET_SEARCH_TOOL_CONFIG, DatasetDetailTool, DATASET_DETAIL_TOOL_CONFIG, DuplicateSpaceTool, formatDuplicateResult, SpaceInfoTool, formatSpaceInfoResult, SpaceFilesTool, UserSummaryPrompt, USER_SUMMARY_PROMPT_CONFIG, PaperSummaryPrompt, PAPER_SUMMARY_PROMPT_CONFIG, CONFIG_GUIDANCE, TOOL_ID_GROUPS, DOCS_SEMANTIC_SEARCH_CONFIG, DocSearchTool, DOC_FETCH_CONFIG, DocFetchTool, } from '@llmindset/hf-mcp';
|
|
5
5
|
import { logger } from './utils/logger.js';
|
|
6
|
+
import { logSearchQuery } from './utils/query-logger.js';
|
|
6
7
|
import { DEFAULT_SPACE_TOOLS } from '../shared/settings.js';
|
|
7
8
|
import { extractAuthBouquetAndMix } from './utils/auth-utils.js';
|
|
8
9
|
import { ToolSelectionStrategy } from './utils/tool-selection-strategy.js';
|
|
@@ -95,6 +96,7 @@ export const createServerFactory = (_webServerInstance, sharedApiClient) => {
|
|
|
95
96
|
toolInstances[SEMANTIC_SEARCH_TOOL_CONFIG.name] = server.tool(SEMANTIC_SEARCH_TOOL_CONFIG.name, SEMANTIC_SEARCH_TOOL_CONFIG.description, SEMANTIC_SEARCH_TOOL_CONFIG.schema.shape, SEMANTIC_SEARCH_TOOL_CONFIG.annotations, async (params) => {
|
|
96
97
|
const semanticSearch = new SpaceSearchTool(hfToken);
|
|
97
98
|
const searchResult = await semanticSearch.search(params.query, params.limit, params.mcp);
|
|
99
|
+
logSearchQuery(SEMANTIC_SEARCH_TOOL_CONFIG.name, params.query, { limit: params.limit, mcp: params.mcp });
|
|
98
100
|
return {
|
|
99
101
|
content: [
|
|
100
102
|
{ type: 'text', text: formatSearchResults(params.query, searchResult.results, searchResult.totalCount) },
|
|
@@ -104,6 +106,7 @@ export const createServerFactory = (_webServerInstance, sharedApiClient) => {
|
|
|
104
106
|
toolInstances[MODEL_SEARCH_TOOL_CONFIG.name] = server.tool(MODEL_SEARCH_TOOL_CONFIG.name, MODEL_SEARCH_TOOL_CONFIG.description, MODEL_SEARCH_TOOL_CONFIG.schema.shape, MODEL_SEARCH_TOOL_CONFIG.annotations, async (params) => {
|
|
105
107
|
const modelSearch = new ModelSearchTool(hfToken);
|
|
106
108
|
const results = await modelSearch.searchWithParams(params);
|
|
109
|
+
logSearchQuery(MODEL_SEARCH_TOOL_CONFIG.name, params.query || `sort:${params.sort || 'trendingScore'}`, params);
|
|
107
110
|
return {
|
|
108
111
|
content: [{ type: 'text', text: results }],
|
|
109
112
|
};
|
|
@@ -117,6 +120,7 @@ export const createServerFactory = (_webServerInstance, sharedApiClient) => {
|
|
|
117
120
|
});
|
|
118
121
|
toolInstances[PAPER_SEARCH_TOOL_CONFIG.name] = server.tool(PAPER_SEARCH_TOOL_CONFIG.name, PAPER_SEARCH_TOOL_CONFIG.description, PAPER_SEARCH_TOOL_CONFIG.schema.shape, PAPER_SEARCH_TOOL_CONFIG.annotations, async (params) => {
|
|
119
122
|
const results = await new PaperSearchTool(hfToken).search(params.query, params.results_limit, params.concise_only);
|
|
123
|
+
logSearchQuery(PAPER_SEARCH_TOOL_CONFIG.name, params.query, { results_limit: params.results_limit, concise_only: params.concise_only });
|
|
120
124
|
return {
|
|
121
125
|
content: [{ type: 'text', text: results }],
|
|
122
126
|
};
|
|
@@ -124,6 +128,7 @@ export const createServerFactory = (_webServerInstance, sharedApiClient) => {
|
|
|
124
128
|
toolInstances[DATASET_SEARCH_TOOL_CONFIG.name] = server.tool(DATASET_SEARCH_TOOL_CONFIG.name, DATASET_SEARCH_TOOL_CONFIG.description, DATASET_SEARCH_TOOL_CONFIG.schema.shape, DATASET_SEARCH_TOOL_CONFIG.annotations, async (params) => {
|
|
125
129
|
const datasetSearch = new DatasetSearchTool(hfToken);
|
|
126
130
|
const results = await datasetSearch.searchWithParams(params);
|
|
131
|
+
logSearchQuery(DATASET_SEARCH_TOOL_CONFIG.name, params.query || `sort:${params.sort || 'trendingScore'}`, params);
|
|
127
132
|
return {
|
|
128
133
|
content: [{ type: 'text', text: results }],
|
|
129
134
|
};
|
|
@@ -138,6 +143,7 @@ export const createServerFactory = (_webServerInstance, sharedApiClient) => {
|
|
|
138
143
|
toolInstances[DOCS_SEMANTIC_SEARCH_CONFIG.name] = server.tool(DOCS_SEMANTIC_SEARCH_CONFIG.name, DOCS_SEMANTIC_SEARCH_CONFIG.description, DOCS_SEMANTIC_SEARCH_CONFIG.schema.shape, DOCS_SEMANTIC_SEARCH_CONFIG.annotations, async (params) => {
|
|
139
144
|
const docSearch = new DocSearchTool(hfToken);
|
|
140
145
|
const results = await docSearch.search(params);
|
|
146
|
+
logSearchQuery(DOCS_SEMANTIC_SEARCH_CONFIG.name, params.query, { product: params.product });
|
|
141
147
|
return {
|
|
142
148
|
content: [{ type: 'text', text: results }],
|
|
143
149
|
};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"mcp-server.js","sourceRoot":"","sources":["../../src/server/mcp-server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEpE,OAAO,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAC;AACvC,OAAO,EAAE,MAAM,EAAe,MAAM,kBAAkB,CAAC;AAEvD,OAAO,EACN,eAAe,EACf,mBAAmB,EACnB,2BAA2B,EAE3B,eAAe,EACf,wBAAwB,EAExB,eAAe,EACf,wBAAwB,EAExB,eAAe,EACf,wBAAwB,EACxB,iBAAiB,EACjB,0BAA0B,EAE1B,iBAAiB,EACjB,0BAA0B,EAE1B,kBAAkB,EAClB,qBAAqB,EAErB,aAAa,EACb,qBAAqB,EACrB,cAAc,EAGd,iBAAiB,EACjB,0BAA0B,EAE1B,kBAAkB,EAClB,2BAA2B,EAE3B,eAAe,EACf,cAAc,EACd,2BAA2B,EAC3B,aAAa,EAEb,gBAAgB,EAChB,YAAY,GAEZ,MAAM,mBAAmB,CAAC;AAK3B,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,mBAAmB,EAAoB,MAAM,uBAAuB,CAAC;AAC9E,OAAO,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC;AACjE,OAAO,EAAE,qBAAqB,EAA6B,MAAM,oCAAoC,CAAC;AAGtG,MAAM,CAAC,MAAM,gBAAgB,GAAgB;IAC5C,YAAY,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,CAAC;IACxC,UAAU,EAAE,mBAAmB;CAC/B,CAAC;AAGF,MAAM,CAAC,MAAM,oBAAoB,GAAgB;IAChD,YAAY,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,CAAC;IACxC,UAAU,EAAE,mBAAmB;CAC/B,CAAC;AAQF,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,kBAA6B,EAAE,eAA6B,EAAiB,EAAE;IAClH,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/C,MAAM,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,oBAAoB,CAAwB,CAAC;IAEzE,OAAO,KAAK,EACX,OAAsC,EACtC,YAA0B,EAC1B,UAAoB,EACC,EAAE;QACvB,MAAM,CAAC,KAAK,CAAC,0CAA0C,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QAEzE,MAAM,EAAE,OAAO,EAAE,GAAG,wBAAwB,CAAC,OAAO,CAAC,CAAC;QAGtD,MAAM,qBAAqB,GAAG,IAAI,qBAAqB,CAAC,eAAe,CAAC,CAAC;QAEzE,IAAI,QAAQ,GACX,2EAA2E;YAC3E,0FAA0F;YAC1F,4DAA4D,CAAC;QAC9D,IAAI,QAA4B,CAAC;QACjC,IAAI,WAA+B,CAAC;QAEpC,IAAI,OAAO,EAAE,CAAC;YACb,IAAI,CAAC;gBACJ,WAAW,GAAG,MAAM,MAAM,CAAC,EAAE,WAAW,EAAE,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;gBACtE,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC;gBAC5B,QAAQ,GAAG,4DAA4D,WAAW,CAAC,IAAI,GAAG,CAAC;YAC5F,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAEhB,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,EAAE,8CAA8C,CAAC,CAAC;YAClG,CAAC;QACF,CAAC;QAMD,MAAM,MAAM,GAAG,IAAI,SAAS,CAC3B;YACC,IAAI,EAAE,2BAA2B;YACjC,OAAO,EAAE,OAAO;SAChB,EACD;YACC,YAAY,EACX,gFAAgF;gBAChF,0FAA0F;gBAC1F,4FAA4F;gBAC5F,mFAAmF;gBACnF,0EAA0E;gBAC1E,QAAQ;SACT,CACD,CAAC;QAQF,MAAM,aAAa,GAA6B,EAAE,CAAC;QAEnD,MAAM,cAAc,GAAG,WAAW;YACjC,CAAC,CAAC,4DAA4D,QAAQ,GAAG;YACzE,CAAC,CAAC,uIAAuI,CAAC;QAE3I,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,4BAA4B,QAAQ,IAAI,SAAS,GAAG,CAAC,CAAC,CAAC,eAAe,CAAC;QACtG,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,wBAAwB,EAAE,EAAE,GAAG,EAAE;YACtF,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAGH,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,KAAK,MAAM,EAAE,CAAC;YAC9C,MAAM,CAAC,IAAI,CACV,cAAc,EACd,gCAAgC,EAChC,EAAE,EACF,EAAE,KAAK,EAAE,6BAA6B,EAAE,EACxC,GAAG,EAAE;gBACJ,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qCAAqC,EAAE,CAAC,EAAE,CAAC;YACrF,CAAC,CACD,CAAC;QACH,CAAC;QAED,MAAM,CAAC,MAAM,CACZ,0BAA0B,CAAC,IAAI,EAC/B,0BAA0B,CAAC,WAAW,EACtC,0BAA0B,CAAC,MAAM,CAAC,KAAK,EACvC,KAAK,EAAE,MAAyB,EAAE,EAAE;YACnC,MAAM,WAAW,GAAG,IAAI,iBAAiB,CAAC,OAAO,CAAC,CAAC;YACnD,MAAM,WAAW,GAAG,MAAM,WAAW,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;YAE9D,OAAO;gBACN,WAAW,EAAE,oBAAoB,MAAM,CAAC,OAAO,EAAE;gBACjD,QAAQ,EAAE;oBACT;wBACC,IAAI,EAAE,MAAe;wBACrB,OAAO,EAAE;4BACR,IAAI,EAAE,MAAe;4BACrB,IAAI,EAAE,WAAW;yBACjB;qBACD;iBACD;aACD,CAAC;QACH,CAAC,CACD,CAAC;QAEF,MAAM,CAAC,MAAM,CACZ,2BAA2B,CAAC,IAAI,EAChC,2BAA2B,CAAC,WAAW,EACvC,2BAA2B,CAAC,MAAM,CAAC,KAAK,EACxC,KAAK,EAAE,MAA0B,EAAE,EAAE;YACpC,MAAM,YAAY,GAAG,IAAI,kBAAkB,CAAC,OAAO,CAAC,CAAC;YACrD,MAAM,WAAW,GAAG,MAAM,YAAY,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;YAE/D,OAAO;gBACN,WAAW,EAAE,qBAAqB,MAAM,CAAC,QAAQ,EAAE;gBACnD,QAAQ,EAAE;oBACT;wBACC,IAAI,EAAE,MAAe;wBACrB,OAAO,EAAE;4BACR,IAAI,EAAE,MAAe;4BACrB,IAAI,EAAE,WAAW;yBACjB;qBACD;iBACD;aACD,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,2BAA2B,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CAC5D,2BAA2B,CAAC,IAAI,EAChC,2BAA2B,CAAC,WAAW,EACvC,2BAA2B,CAAC,MAAM,CAAC,KAAK,EACxC,2BAA2B,CAAC,WAAW,EACvC,KAAK,EAAE,MAAoB,EAAE,EAAE;YAC9B,MAAM,cAAc,GAAG,IAAI,eAAe,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,YAAY,GAAG,MAAM,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC;YACzF,OAAO;gBACN,OAAO,EAAE;oBACR,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,CAAC,MAAM,CAAC,KAAK,EAAE,YAAY,CAAC,OAAO,EAAE,YAAY,CAAC,UAAU,CAAC,EAAE;iBACxG;aACD,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,wBAAwB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACzD,wBAAwB,CAAC,IAAI,EAC7B,wBAAwB,CAAC,WAAW,EACpC,wBAAwB,CAAC,MAAM,CAAC,KAAK,EACrC,wBAAwB,CAAC,WAAW,EACpC,KAAK,EAAE,MAAyB,EAAE,EAAE;YACnC,MAAM,WAAW,GAAG,IAAI,eAAe,CAAC,OAAO,CAAC,CAAC;YACjD,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC3D,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,wBAAwB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACzD,wBAAwB,CAAC,IAAI,EAC7B,wBAAwB,CAAC,WAAW,EACpC,wBAAwB,CAAC,MAAM,CAAC,KAAK,EACrC,wBAAwB,CAAC,WAAW,EACpC,KAAK,EAAE,MAAyB,EAAE,EAAE;YACnC,MAAM,WAAW,GAAG,IAAI,eAAe,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAC5D,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC9D,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,wBAAwB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACzD,wBAAwB,CAAC,IAAI,EAC7B,wBAAwB,CAAC,WAAW,EACpC,wBAAwB,CAAC,MAAM,CAAC,KAAK,EACrC,wBAAwB,CAAC,WAAW,EACpC,KAAK,EAAE,MAAuD,EAAE,EAAE;YACjE,MAAM,OAAO,GAAG,MAAM,IAAI,eAAe,CAAC,OAAO,CAAC,CAAC,MAAM,CACxD,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,aAAa,EACpB,MAAM,CAAC,YAAY,CACnB,CAAC;YACF,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,0BAA0B,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CAC3D,0BAA0B,CAAC,IAAI,EAC/B,0BAA0B,CAAC,WAAW,EACtC,0BAA0B,CAAC,MAAM,CAAC,KAAK,EACvC,0BAA0B,CAAC,WAAW,EACtC,KAAK,EAAE,MAA2B,EAAE,EAAE;YACrC,MAAM,aAAa,GAAG,IAAI,iBAAiB,CAAC,OAAO,CAAC,CAAC;YACrD,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC7D,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,0BAA0B,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CAC3D,0BAA0B,CAAC,IAAI,EAC/B,0BAA0B,CAAC,WAAW,EACtC,0BAA0B,CAAC,MAAM,CAAC,KAAK,EACvC,0BAA0B,CAAC,WAAW,EACtC,KAAK,EAAE,MAA2B,EAAE,EAAE;YACrC,MAAM,aAAa,GAAG,IAAI,iBAAiB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAChE,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAClE,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,2BAA2B,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CAC5D,2BAA2B,CAAC,IAAI,EAChC,2BAA2B,CAAC,WAAW,EACvC,2BAA2B,CAAC,MAAM,CAAC,KAAK,EACxC,2BAA2B,CAAC,WAAW,EACvC,KAAK,EAAE,MAAuB,EAAE,EAAE;YACjC,MAAM,SAAS,GAAG,IAAI,aAAa,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC/C,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,gBAAgB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACjD,gBAAgB,CAAC,IAAI,EACrB,gBAAgB,CAAC,WAAW,EAC5B,gBAAgB,CAAC,MAAM,CAAC,KAAK,EAC7B,gBAAgB,CAAC,WAAW,EAC5B,KAAK,EAAE,MAAsB,EAAE,EAAE;YAChC,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;YACpC,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC7C,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC1E,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACpD,mBAAmB,CAAC,IAAI,EACxB,mBAAmB,CAAC,WAAW,EAC/B,mBAAmB,CAAC,MAAM,CAAC,KAAK,EAChC,mBAAmB,CAAC,WAAW,EAC/B,KAAK,EAAE,MAA4B,EAAE,EAAE;YACtC,MAAM,cAAc,GAAG,IAAI,kBAAkB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YACjE,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACtD,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAqB,CAAC,MAAM,CAAC,EAAE,CAAC;aAChE,CAAC;QACH,CAAC,CACD,CAAC;QAEF,MAAM,mBAAmB,GAAG,aAAa,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACrE,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACpD,mBAAmB,CAAC,IAAI,EACxB,mBAAmB,CAAC,WAAW,EAC/B,mBAAmB,CAAC,MAAM,CAAC,KAAK,EAChC,mBAAmB,CAAC,WAAW,EAC/B,KAAK,EAAE,MAAuB,EAAE,EAAE;YACjC,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC3D,MAAM,MAAM,GAAG,MAAM,qBAAqB,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;YAClE,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;aACzC,CAAC;QACH,CAAC,CACD,CAAC;QAEF,MAAM,oBAAoB,GAAG,cAAc,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACvE,aAAa,CAAC,oBAAoB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACrD,oBAAoB,CAAC,IAAI,EACzB,oBAAoB,CAAC,WAAW,EAChC,oBAAoB,CAAC,MAAM,CAAC,KAAK,EACjC,oBAAoB,CAAC,WAAW,EAChC,KAAK,EAAE,MAAwB,EAAE,EAAE;YAClC,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC7D,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACtD,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;aACzC,CAAC;QACH,CAAC,CACD,CAAC;QAQF,MAAM,eAAe,GAAG,KAAK,IAAI,EAAE;YAClC,MAAM,OAAO,GAAyB;gBACrC,OAAO;gBACP,YAAY;gBACZ,OAAO;aACP,CAAC;YAEF,MAAM,aAAa,GAAG,MAAM,qBAAqB,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAEvE,MAAM,CAAC,IAAI,CACV;gBACC,IAAI,EAAE,aAAa,CAAC,IAAI;gBACxB,MAAM,EAAE,aAAa,CAAC,MAAM;gBAC5B,YAAY,EAAE,aAAa,CAAC,cAAc,CAAC,MAAM;gBACjD,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,MAAM;gBAC7C,YAAY,EAAE,aAAa,CAAC,YAAY;aACxC,EACD,iCAAiC,CACjC,CAAC;YAGF,KAAK,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;gBACtE,IAAI,aAAa,CAAC,cAAc,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACrD,YAAY,CAAC,MAAM,EAAE,CAAC;gBACvB,CAAC;qBAAM,CAAC;oBACP,YAAY,CAAC,OAAO,EAAE,CAAC;gBACxB,CAAC;YACF,CAAC;QACF,CAAC,CAAC;QAGF,MAAM,aAAa,GAAG,eAAe,CAAC,gBAAgB,EAAE,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,oBAAoB,CAAC;YAClC,KAAK,EAAE;gBACN,WAAW,EAAE,CAAC,aAAa,EAAE,mBAAmB;aAChD;YACD,OAAO,EAAE;gBACR,WAAW,EAAE,KAAK;aAClB;SACD,CAAC,CAAC;QAEH,IAAI,CAAC,UAAU,EAAE,CAAC;YACjB,KAAK,eAAe,EAAE,CAAC;YAEvB,IAAI,CAAC,aAAa,EAAE,mBAAmB,IAAI,CAAC,aAAa,EAAE,eAAe,EAAE,CAAC;gBAE5E,MAAM,sBAAsB,GAAG,CAAC,MAAc,EAAE,OAAgB,EAAE,EAAE;oBACnE,MAAM,YAAY,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;oBAC3C,IAAI,YAAY,EAAE,CAAC;wBAClB,IAAI,OAAO,EAAE,CAAC;4BACb,YAAY,CAAC,MAAM,EAAE,CAAC;wBACvB,CAAC;6BAAM,CAAC;4BACP,YAAY,CAAC,OAAO,EAAE,CAAC;wBACxB,CAAC;wBACD,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,kCAAkC,CAAC,CAAC;oBACvE,CAAC;gBACF,CAAC,CAAC;gBAEF,eAAe,CAAC,EAAE,CAAC,iBAAiB,EAAE,sBAAsB,CAAC,CAAC;gBAG9D,MAAM,CAAC,MAAM,CAAC,OAAO,GAAG,GAAG,EAAE;oBAC5B,eAAe,CAAC,cAAc,CAAC,iBAAiB,EAAE,sBAAsB,CAAC,CAAC;oBAC1E,MAAM,CAAC,KAAK,CAAC,oDAAoD,CAAC,CAAC;gBACpE,CAAC,CAAC;YACH,CAAC;QACF,CAAC;QACD,OAAO,MAAM,CAAC;IACf,CAAC,CAAC;AACH,CAAC,CAAC"}
|
|
1
|
+
{"version":3,"file":"mcp-server.js","sourceRoot":"","sources":["../../src/server/mcp-server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEpE,OAAO,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAC;AACvC,OAAO,EAAE,MAAM,EAAe,MAAM,kBAAkB,CAAC;AAEvD,OAAO,EACN,eAAe,EACf,mBAAmB,EACnB,2BAA2B,EAE3B,eAAe,EACf,wBAAwB,EAExB,eAAe,EACf,wBAAwB,EAExB,eAAe,EACf,wBAAwB,EACxB,iBAAiB,EACjB,0BAA0B,EAE1B,iBAAiB,EACjB,0BAA0B,EAE1B,kBAAkB,EAClB,qBAAqB,EAErB,aAAa,EACb,qBAAqB,EACrB,cAAc,EAGd,iBAAiB,EACjB,0BAA0B,EAE1B,kBAAkB,EAClB,2BAA2B,EAE3B,eAAe,EACf,cAAc,EACd,2BAA2B,EAC3B,aAAa,EAEb,gBAAgB,EAChB,YAAY,GAEZ,MAAM,mBAAmB,CAAC;AAK3B,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,mBAAmB,EAAoB,MAAM,uBAAuB,CAAC;AAC9E,OAAO,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC;AACjE,OAAO,EAAE,qBAAqB,EAA6B,MAAM,oCAAoC,CAAC;AAGtG,MAAM,CAAC,MAAM,gBAAgB,GAAgB;IAC5C,YAAY,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,CAAC;IACxC,UAAU,EAAE,mBAAmB;CAC/B,CAAC;AAGF,MAAM,CAAC,MAAM,oBAAoB,GAAgB;IAChD,YAAY,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,CAAC;IACxC,UAAU,EAAE,mBAAmB;CAC/B,CAAC;AAQF,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,kBAA6B,EAAE,eAA6B,EAAiB,EAAE;IAClH,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/C,MAAM,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,oBAAoB,CAAwB,CAAC;IAEzE,OAAO,KAAK,EACX,OAAsC,EACtC,YAA0B,EAC1B,UAAoB,EACC,EAAE;QACvB,MAAM,CAAC,KAAK,CAAC,0CAA0C,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QAEzE,MAAM,EAAE,OAAO,EAAE,GAAG,wBAAwB,CAAC,OAAO,CAAC,CAAC;QAGtD,MAAM,qBAAqB,GAAG,IAAI,qBAAqB,CAAC,eAAe,CAAC,CAAC;QAEzE,IAAI,QAAQ,GACX,2EAA2E;YAC3E,0FAA0F;YAC1F,4DAA4D,CAAC;QAC9D,IAAI,QAA4B,CAAC;QACjC,IAAI,WAA+B,CAAC;QAEpC,IAAI,OAAO,EAAE,CAAC;YACb,IAAI,CAAC;gBACJ,WAAW,GAAG,MAAM,MAAM,CAAC,EAAE,WAAW,EAAE,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;gBACtE,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC;gBAC5B,QAAQ,GAAG,4DAA4D,WAAW,CAAC,IAAI,GAAG,CAAC;YAC5F,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAEhB,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,EAAE,8CAA8C,CAAC,CAAC;YAClG,CAAC;QACF,CAAC;QAMD,MAAM,MAAM,GAAG,IAAI,SAAS,CAC3B;YACC,IAAI,EAAE,2BAA2B;YACjC,OAAO,EAAE,OAAO;SAChB,EACD;YACC,YAAY,EACX,gFAAgF;gBAChF,0FAA0F;gBAC1F,4FAA4F;gBAC5F,mFAAmF;gBACnF,0EAA0E;gBAC1E,QAAQ;SACT,CACD,CAAC;QAQF,MAAM,aAAa,GAA6B,EAAE,CAAC;QAEnD,MAAM,cAAc,GAAG,WAAW;YACjC,CAAC,CAAC,4DAA4D,QAAQ,GAAG;YACzE,CAAC,CAAC,uIAAuI,CAAC;QAE3I,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,4BAA4B,QAAQ,IAAI,SAAS,GAAG,CAAC,CAAC,CAAC,eAAe,CAAC;QACtG,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,wBAAwB,EAAE,EAAE,GAAG,EAAE;YACtF,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAGH,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,KAAK,MAAM,EAAE,CAAC;YAC9C,MAAM,CAAC,IAAI,CACV,cAAc,EACd,gCAAgC,EAChC,EAAE,EACF,EAAE,KAAK,EAAE,6BAA6B,EAAE,EACxC,GAAG,EAAE;gBACJ,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qCAAqC,EAAE,CAAC,EAAE,CAAC;YACrF,CAAC,CACD,CAAC;QACH,CAAC;QAED,MAAM,CAAC,MAAM,CACZ,0BAA0B,CAAC,IAAI,EAC/B,0BAA0B,CAAC,WAAW,EACtC,0BAA0B,CAAC,MAAM,CAAC,KAAK,EACvC,KAAK,EAAE,MAAyB,EAAE,EAAE;YACnC,MAAM,WAAW,GAAG,IAAI,iBAAiB,CAAC,OAAO,CAAC,CAAC;YACnD,MAAM,WAAW,GAAG,MAAM,WAAW,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;YAE9D,OAAO;gBACN,WAAW,EAAE,oBAAoB,MAAM,CAAC,OAAO,EAAE;gBACjD,QAAQ,EAAE;oBACT;wBACC,IAAI,EAAE,MAAe;wBACrB,OAAO,EAAE;4BACR,IAAI,EAAE,MAAe;4BACrB,IAAI,EAAE,WAAW;yBACjB;qBACD;iBACD;aACD,CAAC;QACH,CAAC,CACD,CAAC;QAEF,MAAM,CAAC,MAAM,CACZ,2BAA2B,CAAC,IAAI,EAChC,2BAA2B,CAAC,WAAW,EACvC,2BAA2B,CAAC,MAAM,CAAC,KAAK,EACxC,KAAK,EAAE,MAA0B,EAAE,EAAE;YACpC,MAAM,YAAY,GAAG,IAAI,kBAAkB,CAAC,OAAO,CAAC,CAAC;YACrD,MAAM,WAAW,GAAG,MAAM,YAAY,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;YAE/D,OAAO;gBACN,WAAW,EAAE,qBAAqB,MAAM,CAAC,QAAQ,EAAE;gBACnD,QAAQ,EAAE;oBACT;wBACC,IAAI,EAAE,MAAe;wBACrB,OAAO,EAAE;4BACR,IAAI,EAAE,MAAe;4BACrB,IAAI,EAAE,WAAW;yBACjB;qBACD;iBACD;aACD,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,2BAA2B,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CAC5D,2BAA2B,CAAC,IAAI,EAChC,2BAA2B,CAAC,WAAW,EACvC,2BAA2B,CAAC,MAAM,CAAC,KAAK,EACxC,2BAA2B,CAAC,WAAW,EACvC,KAAK,EAAE,MAAoB,EAAE,EAAE;YAC9B,MAAM,cAAc,GAAG,IAAI,eAAe,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,YAAY,GAAG,MAAM,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC;YACzF,cAAc,CAAC,2BAA2B,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC;YACzG,OAAO;gBACN,OAAO,EAAE;oBACR,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,CAAC,MAAM,CAAC,KAAK,EAAE,YAAY,CAAC,OAAO,EAAE,YAAY,CAAC,UAAU,CAAC,EAAE;iBACxG;aACD,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,wBAAwB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACzD,wBAAwB,CAAC,IAAI,EAC7B,wBAAwB,CAAC,WAAW,EACpC,wBAAwB,CAAC,MAAM,CAAC,KAAK,EACrC,wBAAwB,CAAC,WAAW,EACpC,KAAK,EAAE,MAAyB,EAAE,EAAE;YACnC,MAAM,WAAW,GAAG,IAAI,eAAe,CAAC,OAAO,CAAC,CAAC;YACjD,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC3D,cAAc,CAAC,wBAAwB,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,IAAI,QAAQ,MAAM,CAAC,IAAI,IAAI,eAAe,EAAE,EAAE,MAAM,CAAC,CAAC;YAChH,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,wBAAwB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACzD,wBAAwB,CAAC,IAAI,EAC7B,wBAAwB,CAAC,WAAW,EACpC,wBAAwB,CAAC,MAAM,CAAC,KAAK,EACrC,wBAAwB,CAAC,WAAW,EACpC,KAAK,EAAE,MAAyB,EAAE,EAAE;YACnC,MAAM,WAAW,GAAG,IAAI,eAAe,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAC5D,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC9D,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,wBAAwB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACzD,wBAAwB,CAAC,IAAI,EAC7B,wBAAwB,CAAC,WAAW,EACpC,wBAAwB,CAAC,MAAM,CAAC,KAAK,EACrC,wBAAwB,CAAC,WAAW,EACpC,KAAK,EAAE,MAAuD,EAAE,EAAE;YACjE,MAAM,OAAO,GAAG,MAAM,IAAI,eAAe,CAAC,OAAO,CAAC,CAAC,MAAM,CACxD,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,aAAa,EACpB,MAAM,CAAC,YAAY,CACnB,CAAC;YACF,cAAc,CAAC,wBAAwB,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,EAAE,EAAE,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;YACxI,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,0BAA0B,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CAC3D,0BAA0B,CAAC,IAAI,EAC/B,0BAA0B,CAAC,WAAW,EACtC,0BAA0B,CAAC,MAAM,CAAC,KAAK,EACvC,0BAA0B,CAAC,WAAW,EACtC,KAAK,EAAE,MAA2B,EAAE,EAAE;YACrC,MAAM,aAAa,GAAG,IAAI,iBAAiB,CAAC,OAAO,CAAC,CAAC;YACrD,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC7D,cAAc,CAAC,0BAA0B,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,IAAI,QAAQ,MAAM,CAAC,IAAI,IAAI,eAAe,EAAE,EAAE,MAAM,CAAC,CAAC;YAClH,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,0BAA0B,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CAC3D,0BAA0B,CAAC,IAAI,EAC/B,0BAA0B,CAAC,WAAW,EACtC,0BAA0B,CAAC,MAAM,CAAC,KAAK,EACvC,0BAA0B,CAAC,WAAW,EACtC,KAAK,EAAE,MAA2B,EAAE,EAAE;YACrC,MAAM,aAAa,GAAG,IAAI,iBAAiB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAChE,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAClE,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,2BAA2B,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CAC5D,2BAA2B,CAAC,IAAI,EAChC,2BAA2B,CAAC,WAAW,EACvC,2BAA2B,CAAC,MAAM,CAAC,KAAK,EACxC,2BAA2B,CAAC,WAAW,EACvC,KAAK,EAAE,MAAuB,EAAE,EAAE;YACjC,MAAM,SAAS,GAAG,IAAI,aAAa,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC/C,cAAc,CAAC,2BAA2B,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;YAC5F,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,gBAAgB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACjD,gBAAgB,CAAC,IAAI,EACrB,gBAAgB,CAAC,WAAW,EAC5B,gBAAgB,CAAC,MAAM,CAAC,KAAK,EAC7B,gBAAgB,CAAC,WAAW,EAC5B,KAAK,EAAE,MAAsB,EAAE,EAAE;YAChC,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;YACpC,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC7C,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC1E,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACpD,mBAAmB,CAAC,IAAI,EACxB,mBAAmB,CAAC,WAAW,EAC/B,mBAAmB,CAAC,MAAM,CAAC,KAAK,EAChC,mBAAmB,CAAC,WAAW,EAC/B,KAAK,EAAE,MAA4B,EAAE,EAAE;YACtC,MAAM,cAAc,GAAG,IAAI,kBAAkB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YACjE,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACtD,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAqB,CAAC,MAAM,CAAC,EAAE,CAAC;aAChE,CAAC;QACH,CAAC,CACD,CAAC;QAEF,MAAM,mBAAmB,GAAG,aAAa,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACrE,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACpD,mBAAmB,CAAC,IAAI,EACxB,mBAAmB,CAAC,WAAW,EAC/B,mBAAmB,CAAC,MAAM,CAAC,KAAK,EAChC,mBAAmB,CAAC,WAAW,EAC/B,KAAK,EAAE,MAAuB,EAAE,EAAE;YACjC,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC3D,MAAM,MAAM,GAAG,MAAM,qBAAqB,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;YAClE,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;aACzC,CAAC;QACH,CAAC,CACD,CAAC;QAEF,MAAM,oBAAoB,GAAG,cAAc,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACvE,aAAa,CAAC,oBAAoB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACrD,oBAAoB,CAAC,IAAI,EACzB,oBAAoB,CAAC,WAAW,EAChC,oBAAoB,CAAC,MAAM,CAAC,KAAK,EACjC,oBAAoB,CAAC,WAAW,EAChC,KAAK,EAAE,MAAwB,EAAE,EAAE;YAClC,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC7D,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACtD,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;aACzC,CAAC;QACH,CAAC,CACD,CAAC;QAQF,MAAM,eAAe,GAAG,KAAK,IAAI,EAAE;YAClC,MAAM,OAAO,GAAyB;gBACrC,OAAO;gBACP,YAAY;gBACZ,OAAO;aACP,CAAC;YAEF,MAAM,aAAa,GAAG,MAAM,qBAAqB,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAEvE,MAAM,CAAC,IAAI,CACV;gBACC,IAAI,EAAE,aAAa,CAAC,IAAI;gBACxB,MAAM,EAAE,aAAa,CAAC,MAAM;gBAC5B,YAAY,EAAE,aAAa,CAAC,cAAc,CAAC,MAAM;gBACjD,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,MAAM;gBAC7C,YAAY,EAAE,aAAa,CAAC,YAAY;aACxC,EACD,iCAAiC,CACjC,CAAC;YAGF,KAAK,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;gBACtE,IAAI,aAAa,CAAC,cAAc,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACrD,YAAY,CAAC,MAAM,EAAE,CAAC;gBACvB,CAAC;qBAAM,CAAC;oBACP,YAAY,CAAC,OAAO,EAAE,CAAC;gBACxB,CAAC;YACF,CAAC;QACF,CAAC,CAAC;QAGF,MAAM,aAAa,GAAG,eAAe,CAAC,gBAAgB,EAAE,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,oBAAoB,CAAC;YAClC,KAAK,EAAE;gBACN,WAAW,EAAE,CAAC,aAAa,EAAE,mBAAmB;aAChD;YACD,OAAO,EAAE;gBACR,WAAW,EAAE,KAAK;aAClB;SACD,CAAC,CAAC;QAEH,IAAI,CAAC,UAAU,EAAE,CAAC;YACjB,KAAK,eAAe,EAAE,CAAC;YAEvB,IAAI,CAAC,aAAa,EAAE,mBAAmB,IAAI,CAAC,aAAa,EAAE,eAAe,EAAE,CAAC;gBAE5E,MAAM,sBAAsB,GAAG,CAAC,MAAc,EAAE,OAAgB,EAAE,EAAE;oBACnE,MAAM,YAAY,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;oBAC3C,IAAI,YAAY,EAAE,CAAC;wBAClB,IAAI,OAAO,EAAE,CAAC;4BACb,YAAY,CAAC,MAAM,EAAE,CAAC;wBACvB,CAAC;6BAAM,CAAC;4BACP,YAAY,CAAC,OAAO,EAAE,CAAC;wBACxB,CAAC;wBACD,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,kCAAkC,CAAC,CAAC;oBACvE,CAAC;gBACF,CAAC,CAAC;gBAEF,eAAe,CAAC,EAAE,CAAC,iBAAiB,EAAE,sBAAsB,CAAC,CAAC;gBAG9D,MAAM,CAAC,MAAM,CAAC,OAAO,GAAG,GAAG,EAAE;oBAC5B,eAAe,CAAC,cAAc,CAAC,iBAAiB,EAAE,sBAAsB,CAAC,CAAC;oBAC1E,MAAM,CAAC,KAAK,CAAC,oDAAoD,CAAC,CAAC;gBACpE,CAAC,CAAC;YACH,CAAC;QACF,CAAC;QACD,OAAO,MAAM,CAAC;IACf,CAAC,CAAC;AACH,CAAC,CAAC"}
|
|
@@ -6,6 +6,12 @@ export interface HfDatasetTransportOptions {
|
|
|
6
6
|
batchSize?: number;
|
|
7
7
|
flushInterval?: number;
|
|
8
8
|
uploadFunction?: typeof UploadFileFunction;
|
|
9
|
+
logType?: string;
|
|
10
|
+
}
|
|
11
|
+
export interface HfTransportOptions {
|
|
12
|
+
batchSize?: number;
|
|
13
|
+
flushInterval?: number;
|
|
14
|
+
logType?: string;
|
|
9
15
|
}
|
|
10
16
|
export interface LogEntry {
|
|
11
17
|
level: number;
|
|
@@ -25,6 +31,7 @@ export declare class HfDatasetLogger {
|
|
|
25
31
|
private sessionId;
|
|
26
32
|
private uploadFunction;
|
|
27
33
|
private readonly maxBufferSize;
|
|
34
|
+
private logType;
|
|
28
35
|
constructor(options: HfDatasetTransportOptions);
|
|
29
36
|
processLog(logEntry: LogEntry): void;
|
|
30
37
|
private startFlushTimer;
|
|
@@ -38,10 +45,5 @@ export declare class HfDatasetLogger {
|
|
|
38
45
|
sessionId: string;
|
|
39
46
|
};
|
|
40
47
|
}
|
|
41
|
-
|
|
42
|
-
batchSize?: number;
|
|
43
|
-
flushInterval?: number;
|
|
44
|
-
}
|
|
45
|
-
export default function (opts?: TransportOptions): Promise<Transform>;
|
|
46
|
-
export {};
|
|
48
|
+
export default function (opts?: HfTransportOptions): Promise<Transform>;
|
|
47
49
|
//# sourceMappingURL=hf-dataset-transport.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"hf-dataset-transport.d.ts","sourceRoot":"","sources":["../../../src/server/utils/hf-dataset-transport.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,UAAU,IAAI,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAEzE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAG7C,MAAM,WAAW,yBAAyB;IACzC,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,cAAc,CAAC,EAAE,OAAO,kBAAkB,CAAC;
|
|
1
|
+
{"version":3,"file":"hf-dataset-transport.d.ts","sourceRoot":"","sources":["../../../src/server/utils/hf-dataset-transport.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,UAAU,IAAI,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAEzE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAG7C,MAAM,WAAW,yBAAyB;IACzC,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,cAAc,CAAC,EAAE,OAAO,kBAAkB,CAAC;IAC3C,OAAO,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,kBAAkB;IAClC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,OAAO,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,QAAQ;IACxB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;IACZ,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACvB;AAED,qBAAa,eAAe;IAC3B,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,SAAS,CAAkB;IACnC,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,UAAU,CAAC,CAAiB;IACpC,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,cAAc,CAA4B;IAClD,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAgB;IAC9C,OAAO,CAAC,OAAO,CAAS;gBAEZ,OAAO,EAAE,yBAAyB;IAmB9C,UAAU,CAAC,QAAQ,EAAE,QAAQ,GAAG,IAAI;IAmBpC,OAAO,CAAC,eAAe;YAST,KAAK;YAwBL,UAAU;IA8BxB,OAAO,CAAC,wBAAwB;IAoB1B,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAoB9B,SAAS,IAAI;QACZ,UAAU,EAAE,MAAM,CAAC;QACnB,gBAAgB,EAAE,OAAO,CAAC;QAC1B,SAAS,EAAE,MAAM,CAAC;KAClB;CAOD;AA8DD,yBAA+B,IAAI,GAAE,kBAAuB,GAAG,OAAO,CAAC,SAAS,CAAC,CAgEhF"}
|
|
@@ -14,17 +14,18 @@ export class HfDatasetLogger {
|
|
|
14
14
|
sessionId;
|
|
15
15
|
uploadFunction;
|
|
16
16
|
maxBufferSize = 1000;
|
|
17
|
+
logType;
|
|
17
18
|
constructor(options) {
|
|
18
19
|
this.loggingToken = options.loggingToken;
|
|
19
20
|
this.datasetId = options.datasetId;
|
|
20
21
|
this.batchSize = options.batchSize || 100;
|
|
21
|
-
|
|
22
|
-
this.flushInterval = isTest ? options.flushInterval || 1000 : Math.max(options.flushInterval || 300000, 300000);
|
|
22
|
+
this.flushInterval = calculateFlushInterval(options.flushInterval);
|
|
23
23
|
this.sessionId = randomUUID();
|
|
24
24
|
this.uploadFunction = options.uploadFunction || uploadFile;
|
|
25
|
+
this.logType = options.logType || 'Logs';
|
|
25
26
|
this.startFlushTimer();
|
|
26
27
|
this.registerShutdownHandlers();
|
|
27
|
-
console.log(`[HF Dataset
|
|
28
|
+
console.log(`[HF Dataset ${this.logType}] Initialized - Dataset: ${this.datasetId}, Session: ${this.sessionId}, FlushInterval: ${this.flushInterval}ms, BatchSize: ${this.batchSize}`);
|
|
28
29
|
}
|
|
29
30
|
processLog(logEntry) {
|
|
30
31
|
try {
|
|
@@ -32,34 +33,39 @@ export class HfDatasetLogger {
|
|
|
32
33
|
this.logBuffer.shift();
|
|
33
34
|
}
|
|
34
35
|
this.logBuffer.push(logEntry);
|
|
36
|
+
console.log(`[HF Dataset ${this.logType}] Log received - Buffer: ${this.logBuffer.length}/${this.batchSize}`);
|
|
35
37
|
if (this.logBuffer.length >= this.batchSize) {
|
|
38
|
+
console.log(`[HF Dataset ${this.logType}] Triggering flush due to batch size (${this.batchSize})`);
|
|
36
39
|
void this.flush();
|
|
37
40
|
}
|
|
38
41
|
}
|
|
39
42
|
catch (error) {
|
|
40
|
-
console.error(
|
|
43
|
+
console.error(`[HF Dataset ${this.logType}] Error processing log:`, error);
|
|
41
44
|
}
|
|
42
45
|
}
|
|
43
46
|
startFlushTimer() {
|
|
44
47
|
this.flushTimer = setInterval(() => {
|
|
45
48
|
if (this.logBuffer.length > 0) {
|
|
49
|
+
console.log(`[HF Dataset ${this.logType}] Timer flush triggered - Buffer: ${this.logBuffer.length} logs`);
|
|
46
50
|
void this.flush();
|
|
47
51
|
}
|
|
48
52
|
}, this.flushInterval);
|
|
49
53
|
}
|
|
50
54
|
async flush() {
|
|
55
|
+
console.log(`[HF Dataset ${this.logType}] Flush called - Buffer: ${this.logBuffer.length}, InProgress: ${this.uploadInProgress}`);
|
|
51
56
|
if (this.uploadInProgress || this.logBuffer.length === 0) {
|
|
52
57
|
return;
|
|
53
58
|
}
|
|
54
59
|
const logsToUpload = [...this.logBuffer];
|
|
55
60
|
this.logBuffer = [];
|
|
56
61
|
this.uploadInProgress = true;
|
|
62
|
+
console.log(`[HF Dataset ${this.logType}] Starting upload of ${logsToUpload.length} logs`);
|
|
57
63
|
try {
|
|
58
64
|
await this.uploadLogs(logsToUpload);
|
|
59
|
-
console.log(`[HF Dataset
|
|
65
|
+
console.log(`[HF Dataset ${this.logType}] ✅ Uploaded ${logsToUpload.length} logs to ${this.datasetId}`);
|
|
60
66
|
}
|
|
61
67
|
catch (error) {
|
|
62
|
-
console.error(
|
|
68
|
+
console.error(`[HF Dataset ${this.logType}] ❌ Upload failed, dropping logs:`, error);
|
|
63
69
|
}
|
|
64
70
|
finally {
|
|
65
71
|
this.uploadInProgress = false;
|
|
@@ -71,7 +77,7 @@ export class HfDatasetLogger {
|
|
|
71
77
|
const dateFolder = new Date().toISOString().split('T')[0];
|
|
72
78
|
const pathInRepo = `logs/${dateFolder}/${filename}`;
|
|
73
79
|
const logData = logs
|
|
74
|
-
.map((log) => safeStringifyLog(log, this.sessionId))
|
|
80
|
+
.map((log) => safeStringifyLog(log, this.sessionId, this.logType))
|
|
75
81
|
.filter(Boolean)
|
|
76
82
|
.join('\n');
|
|
77
83
|
const uploadTimeout = 30000;
|
|
@@ -109,7 +115,7 @@ export class HfDatasetLogger {
|
|
|
109
115
|
if (this.isShuttingDown)
|
|
110
116
|
return;
|
|
111
117
|
this.isShuttingDown = true;
|
|
112
|
-
console.log(
|
|
118
|
+
console.log(`[HF Dataset ${this.logType}] Shutting down...`);
|
|
113
119
|
if (this.flushTimer) {
|
|
114
120
|
clearInterval(this.flushTimer);
|
|
115
121
|
}
|
|
@@ -117,7 +123,7 @@ export class HfDatasetLogger {
|
|
|
117
123
|
await this.flush();
|
|
118
124
|
}
|
|
119
125
|
catch (error) {
|
|
120
|
-
console.error(
|
|
126
|
+
console.error(`[HF Dataset ${this.logType}] Error during final log flush:`, error);
|
|
121
127
|
}
|
|
122
128
|
}
|
|
123
129
|
getStatus() {
|
|
@@ -128,42 +134,75 @@ export class HfDatasetLogger {
|
|
|
128
134
|
};
|
|
129
135
|
}
|
|
130
136
|
}
|
|
131
|
-
function
|
|
132
|
-
|
|
137
|
+
function calculateFlushInterval(optionsInterval) {
|
|
138
|
+
const envInterval = parseInt(process.env.LOGGING_FLUSH_INTERVAL || '300000', 10);
|
|
139
|
+
const interval = optionsInterval || envInterval;
|
|
140
|
+
const isTest = process.env.NODE_ENV === 'test' || process.env.VITEST === 'true';
|
|
141
|
+
const isDev = process.env.NODE_ENV === 'development';
|
|
142
|
+
if (isTest)
|
|
143
|
+
return interval || 1000;
|
|
144
|
+
if (isDev)
|
|
145
|
+
return interval;
|
|
146
|
+
return Math.max(interval, 300000);
|
|
147
|
+
}
|
|
148
|
+
function createNoOpTransport(reason, logType = 'Logs') {
|
|
149
|
+
console.warn(`[HF Dataset ${logType}] Dataset logging disabled: ${reason}`);
|
|
133
150
|
return build(function (source) {
|
|
134
151
|
source.on('data', function (_obj) {
|
|
135
152
|
});
|
|
136
153
|
});
|
|
137
154
|
}
|
|
138
|
-
function safeStringifyLog(log, sessionId) {
|
|
155
|
+
function safeStringifyLog(log, sessionId, logType) {
|
|
139
156
|
if (!log)
|
|
140
157
|
return '';
|
|
141
|
-
|
|
142
|
-
|
|
158
|
+
if (logType === 'Query') {
|
|
159
|
+
return safeStringify.default({
|
|
160
|
+
...log,
|
|
161
|
+
timestamp: new Date(log.time || Date.now()).toISOString(),
|
|
162
|
+
sessionId,
|
|
163
|
+
});
|
|
164
|
+
}
|
|
165
|
+
const standardizedLog = {
|
|
166
|
+
message: log.msg || 'No message',
|
|
167
|
+
level: log.level,
|
|
143
168
|
timestamp: new Date(log.time || Date.now()).toISOString(),
|
|
144
169
|
sessionId,
|
|
170
|
+
data: (() => {
|
|
171
|
+
const { msg: _msg, level: _level, time: _time, pid: _pid, hostname: _hostname, ...extraData } = log;
|
|
172
|
+
return Object.keys(extraData).length > 0 ? JSON.stringify(extraData) : undefined;
|
|
173
|
+
})(),
|
|
174
|
+
};
|
|
175
|
+
Object.keys(standardizedLog).forEach(key => {
|
|
176
|
+
if (standardizedLog[key] === undefined) {
|
|
177
|
+
delete standardizedLog[key];
|
|
178
|
+
}
|
|
145
179
|
});
|
|
180
|
+
return safeStringify.default(standardizedLog);
|
|
146
181
|
}
|
|
147
182
|
export default async function (opts = {}) {
|
|
183
|
+
const logType = opts.logType || 'Logs';
|
|
148
184
|
if (process.env.NODE_ENV === 'test' || process.env.VITEST === 'true') {
|
|
149
|
-
return createNoOpTransport('disabled during tests');
|
|
185
|
+
return createNoOpTransport('disabled during tests', logType);
|
|
150
186
|
}
|
|
151
|
-
const datasetId =
|
|
187
|
+
const datasetId = logType === 'Query'
|
|
188
|
+
? process.env.QUERY_DATASET_ID
|
|
189
|
+
: process.env.LOGGING_DATASET_ID;
|
|
152
190
|
if (!datasetId) {
|
|
153
|
-
return createNoOpTransport('no dataset ID configured');
|
|
191
|
+
return createNoOpTransport('no dataset ID configured', logType);
|
|
154
192
|
}
|
|
155
193
|
const loggingToken = process.env.LOGGING_HF_TOKEN || process.env.DEFAULT_HF_TOKEN;
|
|
156
194
|
if (!loggingToken) {
|
|
157
|
-
console.warn(
|
|
158
|
-
return createNoOpTransport('no HF token available');
|
|
195
|
+
console.warn(`[HF Dataset ${logType}] No HF token available (LOGGING_HF_TOKEN or DEFAULT_HF_TOKEN). Dataset logging disabled.`);
|
|
196
|
+
return createNoOpTransport('no HF token available', logType);
|
|
159
197
|
}
|
|
160
|
-
console.log(`[HF Dataset
|
|
198
|
+
console.log(`[HF Dataset ${logType}] Logging to dataset: ${datasetId}`);
|
|
161
199
|
try {
|
|
162
200
|
const hfLogger = new HfDatasetLogger({
|
|
163
201
|
loggingToken,
|
|
164
202
|
datasetId,
|
|
165
203
|
batchSize: opts.batchSize || parseInt(process.env.LOGGING_BATCH_SIZE || '100', 10),
|
|
166
|
-
flushInterval: opts.flushInterval
|
|
204
|
+
flushInterval: calculateFlushInterval(opts.flushInterval),
|
|
205
|
+
logType,
|
|
167
206
|
});
|
|
168
207
|
return build(async function (source) {
|
|
169
208
|
for await (const obj of source) {
|
|
@@ -171,7 +210,7 @@ export default async function (opts = {}) {
|
|
|
171
210
|
hfLogger.processLog(obj);
|
|
172
211
|
}
|
|
173
212
|
catch (error) {
|
|
174
|
-
console.error(
|
|
213
|
+
console.error(`[HF Dataset ${logType}] Transport error (ignoring):`, error);
|
|
175
214
|
}
|
|
176
215
|
}
|
|
177
216
|
}, {
|
|
@@ -180,14 +219,14 @@ export default async function (opts = {}) {
|
|
|
180
219
|
await hfLogger.destroy();
|
|
181
220
|
}
|
|
182
221
|
catch (error) {
|
|
183
|
-
console.error(
|
|
222
|
+
console.error(`[HF Dataset ${logType}] Error during close (ignoring):`, error);
|
|
184
223
|
}
|
|
185
224
|
},
|
|
186
225
|
});
|
|
187
226
|
}
|
|
188
227
|
catch (error) {
|
|
189
|
-
console.error(
|
|
190
|
-
return createNoOpTransport('initialization failed');
|
|
228
|
+
console.error(`[HF Dataset ${logType}] Failed to initialize, falling back to no-op:`, error);
|
|
229
|
+
return createNoOpTransport('initialization failed', logType);
|
|
191
230
|
}
|
|
192
231
|
}
|
|
193
232
|
//# sourceMappingURL=hf-dataset-transport.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"hf-dataset-transport.js","sourceRoot":"","sources":["../../../src/server/utils/hf-dataset-transport.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,yBAAyB,CAAC;AAC5C,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE9C,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,aAAa,MAAM,qBAAqB,CAAC;
|
|
1
|
+
{"version":3,"file":"hf-dataset-transport.js","sourceRoot":"","sources":["../../../src/server/utils/hf-dataset-transport.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,yBAAyB,CAAC;AAC5C,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE9C,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,aAAa,MAAM,qBAAqB,CAAC;AAwBhD,MAAM,OAAO,eAAe;IACnB,YAAY,CAAS;IACrB,SAAS,CAAS;IAClB,SAAS,GAAe,EAAE,CAAC;IAC3B,SAAS,CAAS;IAClB,aAAa,CAAS;IACtB,UAAU,CAAkB;IAC5B,cAAc,GAAG,KAAK,CAAC;IACvB,gBAAgB,GAAG,KAAK,CAAC;IACzB,SAAS,CAAS;IAClB,cAAc,CAA4B;IACjC,aAAa,GAAW,IAAI,CAAC;IACtC,OAAO,CAAS;IAExB,YAAY,OAAkC;QAC7C,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,GAAG,CAAC;QAC1C,IAAI,CAAC,aAAa,GAAG,sBAAsB,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACnE,IAAI,CAAC,SAAS,GAAG,UAAU,EAAE,CAAC;QAC9B,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,UAAU,CAAC;QAC3D,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,MAAM,CAAC;QAGzC,IAAI,CAAC,eAAe,EAAE,CAAC;QAGvB,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAGhC,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,CAAC,OAAO,4BAA4B,IAAI,CAAC,SAAS,cAAc,IAAI,CAAC,SAAS,oBAAoB,IAAI,CAAC,aAAa,kBAAkB,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;IACxL,CAAC;IAED,UAAU,CAAC,QAAkB;QAC5B,IAAI,CAAC;YACJ,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;gBACjD,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;YACxB,CAAC;YAED,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAE9B,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,CAAC,OAAO,4BAA4B,IAAI,CAAC,SAAS,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YAE9G,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC7C,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,CAAC,OAAO,yCAAyC,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC;gBACnG,KAAK,IAAI,CAAC,KAAK,EAAE,CAAC;YACnB,CAAC;QACF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,KAAK,CAAC,eAAe,IAAI,CAAC,OAAO,yBAAyB,EAAE,KAAK,CAAC,CAAC;QAC5E,CAAC;IACF,CAAC;IAEO,eAAe;QACtB,IAAI,CAAC,UAAU,GAAG,WAAW,CAAC,GAAG,EAAE;YAClC,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC/B,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,CAAC,OAAO,qCAAqC,IAAI,CAAC,SAAS,CAAC,MAAM,OAAO,CAAC,CAAC;gBAC1G,KAAK,IAAI,CAAC,KAAK,EAAE,CAAC;YACnB,CAAC;QACF,CAAC,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;IACxB,CAAC;IAEO,KAAK,CAAC,KAAK;QAClB,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,CAAC,OAAO,4BAA4B,IAAI,CAAC,SAAS,CAAC,MAAM,iBAAiB,IAAI,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAElI,IAAI,IAAI,CAAC,gBAAgB,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1D,OAAO;QACR,CAAC;QAED,MAAM,YAAY,GAAG,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QACzC,IAAI,CAAC,SAAS,GAAG,EAAE,CAAC;QACpB,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAE7B,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,CAAC,OAAO,wBAAwB,YAAY,CAAC,MAAM,OAAO,CAAC,CAAC;QAE3F,IAAI,CAAC;YACJ,MAAM,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;YACpC,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,CAAC,OAAO,gBAAgB,YAAY,CAAC,MAAM,YAAY,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QACzG,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEhB,OAAO,CAAC,KAAK,CAAC,eAAe,IAAI,CAAC,OAAO,mCAAmC,EAAE,KAAK,CAAC,CAAC;QACtF,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC;QAC/B,CAAC;IACF,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,IAAgB;QACxC,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QACjE,MAAM,QAAQ,GAAG,QAAQ,SAAS,IAAI,IAAI,CAAC,SAAS,QAAQ,CAAC;QAE7D,MAAM,UAAU,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,UAAU,GAAG,QAAQ,UAAU,IAAI,QAAQ,EAAE,CAAC;QAGpD,MAAM,OAAO,GAAG,IAAI;aAClB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,gBAAgB,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;aACjE,MAAM,CAAC,OAAO,CAAC;aACf,IAAI,CAAC,IAAI,CAAC,CAAC;QAGb,MAAM,aAAa,GAAG,KAAK,CAAC;QAC5B,MAAM,OAAO,CAAC,IAAI,CAAC;YAClB,IAAI,CAAC,cAAc,CAAC;gBACnB,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,EAAE;gBAC/C,IAAI,EAAE;oBACL,IAAI,EAAE,UAAU;oBAChB,OAAO,EAAE,IAAI,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC;iBAC9D;gBACD,WAAW,EAAE,IAAI,CAAC,YAAY;gBAC9B,WAAW,EAAE,OAAO,IAAI,CAAC,MAAM,cAAc;gBAC7C,iBAAiB,EAAE,YAAY,IAAI,CAAC,SAAS,WAAW,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE;aAClF,CAAC;YACF,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC;SAChG,CAAC,CAAC;IACJ,CAAC;IAEO,wBAAwB;QAC/B,MAAM,eAAe,GAAG,KAAK,IAAI,EAAE;YAClC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACtB,CAAC,CAAC;QAGF,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,CAAC;QAChF,IAAI,CAAC,MAAM,EAAE,CAAC;YACb,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,eAAe,CAAC,CAAC;YAC5C,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;YACxC,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;YACzC,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE;gBAEzB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBACrB,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBAChC,CAAC;YACF,CAAC,CAAC,CAAC;QACJ,CAAC;IACF,CAAC;IAED,KAAK,CAAC,OAAO;QACZ,IAAI,IAAI,CAAC,cAAc;YAAE,OAAO;QAChC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAE3B,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,CAAC,OAAO,oBAAoB,CAAC,CAAC;QAG7D,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAChC,CAAC;QAGD,IAAI,CAAC;YACJ,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,KAAK,CAAC,eAAe,IAAI,CAAC,OAAO,iCAAiC,EAAE,KAAK,CAAC,CAAC;QACpF,CAAC;IACF,CAAC;IAGD,SAAS;QAKR,OAAO;YACN,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM;YACjC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,SAAS,EAAE,IAAI,CAAC,SAAS;SACzB,CAAC;IACH,CAAC;CACD;AAID,SAAS,sBAAsB,CAAC,eAAwB;IACvD,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,IAAI,QAAQ,EAAE,EAAE,CAAC,CAAC;IACjF,MAAM,QAAQ,GAAG,eAAe,IAAI,WAAW,CAAC;IAChD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,CAAC;IAChF,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,aAAa,CAAC;IAErD,IAAI,MAAM;QAAE,OAAO,QAAQ,IAAI,IAAI,CAAC;IACpC,IAAI,KAAK;QAAE,OAAO,QAAQ,CAAC;IAC3B,OAAO,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;AACnC,CAAC;AAGD,SAAS,mBAAmB,CAAC,MAAc,EAAE,OAAO,GAAG,MAAM;IAC5D,OAAO,CAAC,IAAI,CAAC,eAAe,OAAO,+BAA+B,MAAM,EAAE,CAAC,CAAC;IAC5E,OAAO,KAAK,CAAC,UAAU,MAAM;QAC5B,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,UAAU,IAAa;QAEzC,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;AACJ,CAAC;AAGD,SAAS,gBAAgB,CAAC,GAAa,EAAE,SAAiB,EAAE,OAAe;IAC1E,IAAI,CAAC,GAAG;QAAE,OAAO,EAAE,CAAC;IAEpB,IAAI,OAAO,KAAK,OAAO,EAAE,CAAC;QAEzB,OAAO,aAAa,CAAC,OAAO,CAAC;YAC5B,GAAG,GAAG;YACN,SAAS,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,WAAW,EAAE;YACzD,SAAS;SACT,CAAC,CAAC;IACJ,CAAC;IAGD,MAAM,eAAe,GAAG;QACvB,OAAO,EAAE,GAAG,CAAC,GAAG,IAAI,YAAY;QAChC,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,SAAS,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,WAAW,EAAE;QACzD,SAAS;QACT,IAAI,EAAE,CAAC,GAAG,EAAE;YAEX,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,SAAS,EAAE,GAAG,GAAG,CAAC;YACpG,OAAO,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAClF,CAAC,CAAC,EAAE;KACJ,CAAC;IAGF,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;QAC1C,IAAI,eAAe,CAAC,GAAmC,CAAC,KAAK,SAAS,EAAE,CAAC;YACxE,OAAO,eAAe,CAAC,GAAmC,CAAC,CAAC;QAC7D,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,OAAO,aAAa,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;AAC/C,CAAC;AAGD,MAAM,CAAC,OAAO,CAAC,KAAK,WAAW,OAA2B,EAAE;IAC3D,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,MAAM,CAAC;IAGvC,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;QACtE,OAAO,mBAAmB,CAAC,uBAAuB,EAAE,OAAO,CAAC,CAAC;IAC9D,CAAC;IAGD,MAAM,SAAS,GAAG,OAAO,KAAK,OAAO;QACpC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB;QAC9B,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;IAClC,IAAI,CAAC,SAAS,EAAE,CAAC;QAChB,OAAO,mBAAmB,CAAC,0BAA0B,EAAE,OAAO,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAClF,IAAI,CAAC,YAAY,EAAE,CAAC;QACnB,OAAO,CAAC,IAAI,CACX,eAAe,OAAO,2FAA2F,CACjH,CAAC;QACF,OAAO,mBAAmB,CAAC,uBAAuB,EAAE,OAAO,CAAC,CAAC;IAC9D,CAAC;IAGD,OAAO,CAAC,GAAG,CAAC,eAAe,OAAO,yBAAyB,SAAS,EAAE,CAAC,CAAC;IAExE,IAAI,CAAC;QAEJ,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC;YACpC,YAAY;YACZ,SAAS;YACT,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,KAAK,EAAE,EAAE,CAAC;YAClF,aAAa,EAAE,sBAAsB,CAAC,IAAI,CAAC,aAAa,CAAC;YACzD,OAAO;SACP,CAAC,CAAC;QAEH,OAAO,KAAK,CACX,KAAK,WAAW,MAAM;YACrB,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;gBAEhC,IAAI,CAAC;oBACJ,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;gBAC1B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBAEhB,OAAO,CAAC,KAAK,CAAC,eAAe,OAAO,+BAA+B,EAAE,KAAK,CAAC,CAAC;gBAC7E,CAAC;YACF,CAAC;QACF,CAAC,EACD;YACC,KAAK,CAAC,KAAK,CAAC,IAAW;gBAEtB,IAAI,CAAC;oBACJ,MAAM,QAAQ,CAAC,OAAO,EAAE,CAAC;gBAC1B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBAChB,OAAO,CAAC,KAAK,CAAC,eAAe,OAAO,kCAAkC,EAAE,KAAK,CAAC,CAAC;gBAChF,CAAC;YACF,CAAC;SACD,CACD,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,OAAO,CAAC,KAAK,CAAC,eAAe,OAAO,gDAAgD,EAAE,KAAK,CAAC,CAAC;QAC7F,OAAO,mBAAmB,CAAC,uBAAuB,EAAE,OAAO,CAAC,CAAC;IAC9D,CAAC;AACF,CAAC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import { type Logger } from 'pino';
|
|
2
|
+
export interface QueryLogEntry {
|
|
3
|
+
query: string;
|
|
4
|
+
toolName: string;
|
|
5
|
+
parameters: string;
|
|
6
|
+
sessionId: string;
|
|
7
|
+
timestamp: string;
|
|
8
|
+
status: 'success' | 'error';
|
|
9
|
+
error?: string;
|
|
10
|
+
}
|
|
11
|
+
declare const queryLogger: Logger | null;
|
|
12
|
+
export declare function logQuery(entry: Omit<QueryLogEntry, 'sessionId' | 'timestamp'>): void;
|
|
13
|
+
export declare function logSearchQuery(toolName: string, query: string, data: Record<string, unknown>): void;
|
|
14
|
+
export { queryLogger };
|
|
15
|
+
//# sourceMappingURL=query-logger.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"query-logger.d.ts","sourceRoot":"","sources":["../../../src/server/utils/query-logger.ts"],"names":[],"mappings":"AAAA,OAAO,EAAQ,KAAK,MAAM,EAAE,MAAM,MAAM,CAAC;AAezC,MAAM,WAAW,aAAa;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,SAAS,GAAG,OAAO,CAAC;IAC5B,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AA4CD,QAAA,MAAM,WAAW,EAAE,MAAM,GAAG,IAA0B,CAAC;AAKvD,wBAAgB,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC,aAAa,EAAE,WAAW,GAAG,WAAW,CAAC,GAAG,IAAI,CAYpF;AAKD,wBAAgB,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAOnG;AAED,OAAO,EAAE,WAAW,EAAE,CAAC"}
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
import { pino } from 'pino';
|
|
2
|
+
import { fileURLToPath } from 'url';
|
|
3
|
+
import { dirname, join } from 'path';
|
|
4
|
+
const queryLoggingEnabled = !!process.env.QUERY_DATASET_ID;
|
|
5
|
+
const __filename = fileURLToPath(import.meta.url);
|
|
6
|
+
const __dirname = dirname(__filename);
|
|
7
|
+
function createQueryLogger() {
|
|
8
|
+
if (process.env.NODE_ENV === 'test' || process.env.VITEST === 'true') {
|
|
9
|
+
return null;
|
|
10
|
+
}
|
|
11
|
+
if (!queryLoggingEnabled) {
|
|
12
|
+
return null;
|
|
13
|
+
}
|
|
14
|
+
const datasetId = process.env.QUERY_DATASET_ID;
|
|
15
|
+
const hfToken = process.env.LOGGING_HF_TOKEN || process.env.DEFAULT_HF_TOKEN;
|
|
16
|
+
if (!hfToken) {
|
|
17
|
+
console.warn('[Query Logger] Query logging disabled: No HF token found (set LOGGING_HF_TOKEN or DEFAULT_HF_TOKEN)');
|
|
18
|
+
return null;
|
|
19
|
+
}
|
|
20
|
+
console.log(`[Query Logger] Query logging enabled for dataset: ${datasetId}`);
|
|
21
|
+
try {
|
|
22
|
+
const transportPath = join(__dirname, 'hf-dataset-transport.js');
|
|
23
|
+
const baseOptions = {
|
|
24
|
+
level: 'info',
|
|
25
|
+
timestamp: pino.stdTimeFunctions.isoTime,
|
|
26
|
+
};
|
|
27
|
+
return pino({
|
|
28
|
+
...baseOptions,
|
|
29
|
+
transport: {
|
|
30
|
+
target: transportPath,
|
|
31
|
+
options: { sync: false, logType: 'Query' },
|
|
32
|
+
},
|
|
33
|
+
});
|
|
34
|
+
}
|
|
35
|
+
catch (error) {
|
|
36
|
+
console.error('[Query Logger] Failed to setup query logging transport:', error);
|
|
37
|
+
return null;
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
const queryLogger = createQueryLogger();
|
|
41
|
+
export function logQuery(entry) {
|
|
42
|
+
if (!queryLogger) {
|
|
43
|
+
return;
|
|
44
|
+
}
|
|
45
|
+
const logEntry = {
|
|
46
|
+
...entry,
|
|
47
|
+
sessionId: crypto.randomUUID(),
|
|
48
|
+
timestamp: new Date().toISOString(),
|
|
49
|
+
};
|
|
50
|
+
queryLogger.info(logEntry);
|
|
51
|
+
}
|
|
52
|
+
export function logSearchQuery(toolName, query, data) {
|
|
53
|
+
logQuery({
|
|
54
|
+
query,
|
|
55
|
+
toolName,
|
|
56
|
+
parameters: JSON.stringify(data),
|
|
57
|
+
status: 'success',
|
|
58
|
+
});
|
|
59
|
+
}
|
|
60
|
+
export { queryLogger };
|
|
61
|
+
//# sourceMappingURL=query-logger.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"query-logger.js","sourceRoot":"","sources":["../../../src/server/utils/query-logger.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAe,MAAM,MAAM,CAAC;AAEzC,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAGrC,MAAM,mBAAmB,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;AAG3D,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AAetC,SAAS,iBAAiB;IAEzB,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;QACtE,OAAO,IAAI,CAAC;IACb,CAAC;IAED,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC;IACb,CAAC;IAED,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAC/C,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAE7E,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,CAAC,IAAI,CAAC,qGAAqG,CAAC,CAAC;QACpH,OAAO,IAAI,CAAC;IACb,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,qDAAqD,SAAS,EAAE,CAAC,CAAC;IAE9E,IAAI,CAAC;QACJ,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,EAAE,yBAAyB,CAAC,CAAC;QAEjE,MAAM,WAAW,GAAkB;YAClC,KAAK,EAAE,MAAM;YACb,SAAS,EAAE,IAAI,CAAC,gBAAgB,CAAC,OAAO;SACxC,CAAC;QAGF,OAAO,IAAI,CAAC;YACX,GAAG,WAAW;YACd,SAAS,EAAE;gBACV,MAAM,EAAE,aAAa;gBACrB,OAAO,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE;aAC1C;SACD,CAAC,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,OAAO,CAAC,KAAK,CAAC,yDAAyD,EAAE,KAAK,CAAC,CAAC;QAChF,OAAO,IAAI,CAAC;IACb,CAAC;AACF,CAAC;AAED,MAAM,WAAW,GAAkB,iBAAiB,EAAE,CAAC;AAKvD,MAAM,UAAU,QAAQ,CAAC,KAAqD;IAC7E,IAAI,CAAC,WAAW,EAAE,CAAC;QAClB,OAAO;IACR,CAAC;IAED,MAAM,QAAQ,GAAkB;QAC/B,GAAG,KAAK;QACR,SAAS,EAAE,MAAM,CAAC,UAAU,EAAE;QAC9B,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACnC,CAAC;IAEF,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AAC5B,CAAC;AAKD,MAAM,UAAU,cAAc,CAAC,QAAgB,EAAE,KAAa,EAAE,IAA6B;IAC5F,QAAQ,CAAC;QACR,KAAK;QACL,QAAQ;QACR,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;QAChC,MAAM,EAAE,SAAS;KACjB,CAAC,CAAC;AACJ,CAAC;AAED,OAAO,EAAE,WAAW,EAAE,CAAC"}
|
|
@@ -0,0 +1,2 @@
|
|
|
1
|
+
export declare function withQueryLogging<T extends unknown[], R>(originalMethod: (...args: T) => Promise<R>, toolName: string, extractQuery: (...args: T) => string, extractParams: (...args: T) => Record<string, unknown>): (...args: T) => Promise<R>;
|
|
2
|
+
//# sourceMappingURL=query-logging-wrapper.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"query-logging-wrapper.d.ts","sourceRoot":"","sources":["../../../src/server/utils/query-logging-wrapper.ts"],"names":[],"mappings":"AAKA,wBAAgB,gBAAgB,CAAC,CAAC,SAAS,OAAO,EAAE,EAAE,CAAC,EACtD,cAAc,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,EAC1C,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,KAAK,MAAM,EACpC,aAAa,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAExC,GAAG,MAAM,CAAC,KAAG,OAAO,CAAC,CAAC,CAAC,CA6BrC"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import { logQuery } from './query-logger.js';
|
|
2
|
+
export function withQueryLogging(originalMethod, toolName, extractQuery, extractParams) {
|
|
3
|
+
return async (...args) => {
|
|
4
|
+
const query = extractQuery(...args);
|
|
5
|
+
const params = extractParams(...args);
|
|
6
|
+
try {
|
|
7
|
+
const result = await originalMethod(...args);
|
|
8
|
+
logQuery({
|
|
9
|
+
query,
|
|
10
|
+
toolName,
|
|
11
|
+
parameters: JSON.stringify(params),
|
|
12
|
+
status: 'success',
|
|
13
|
+
});
|
|
14
|
+
return result;
|
|
15
|
+
}
|
|
16
|
+
catch (error) {
|
|
17
|
+
logQuery({
|
|
18
|
+
query,
|
|
19
|
+
toolName,
|
|
20
|
+
parameters: JSON.stringify(params),
|
|
21
|
+
status: 'error',
|
|
22
|
+
error: error instanceof Error ? error.message : String(error),
|
|
23
|
+
});
|
|
24
|
+
throw error;
|
|
25
|
+
}
|
|
26
|
+
};
|
|
27
|
+
}
|
|
28
|
+
//# sourceMappingURL=query-logging-wrapper.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"query-logging-wrapper.js","sourceRoot":"","sources":["../../../src/server/utils/query-logging-wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAK7C,MAAM,UAAU,gBAAgB,CAC/B,cAA0C,EAC1C,QAAgB,EAChB,YAAoC,EACpC,aAAsD;IAEtD,OAAO,KAAK,EAAE,GAAG,IAAO,EAAc,EAAE;QACvC,MAAM,KAAK,GAAG,YAAY,CAAC,GAAG,IAAI,CAAC,CAAC;QACpC,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,IAAI,CAAC,CAAC;QAEtC,IAAI,CAAC;YACJ,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,GAAG,IAAI,CAAC,CAAC;YAG7C,QAAQ,CAAC;gBACR,KAAK;gBACL,QAAQ;gBACR,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;gBAClC,MAAM,EAAE,SAAS;aACjB,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QACf,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEhB,QAAQ,CAAC;gBACR,KAAK;gBACL,QAAQ;gBACR,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;gBAClC,MAAM,EAAE,OAAO;gBACf,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC7D,CAAC,CAAC;YAEH,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC,CAAC;AACH,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@llmindset/hf-mcp-server",
|
|
3
|
-
"version": "0.2.
|
|
3
|
+
"version": "0.2.4",
|
|
4
4
|
"description": "Official Hugging Face MCP Server",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "./dist/server/streamableHttp.js",
|
|
@@ -56,7 +56,7 @@
|
|
|
56
56
|
"tailwind-merge": "^3.2.0",
|
|
57
57
|
"tailwindcss": "^4.1.5",
|
|
58
58
|
"zod": "^3.24.4",
|
|
59
|
-
"@llmindset/hf-mcp": "^0.2.
|
|
59
|
+
"@llmindset/hf-mcp": "^0.2.4"
|
|
60
60
|
},
|
|
61
61
|
"devDependencies": {
|
|
62
62
|
"@eslint/js": "^9.25.0",
|