@roarkanalytics/sdk-mcp 2.23.0 → 2.24.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/code-tool.js +2 -2
- package/code-tool.mjs +2 -2
- package/http.d.mts.map +1 -1
- package/http.d.ts.map +1 -1
- package/http.js +10 -22
- package/http.js.map +1 -1
- package/http.mjs +10 -22
- package/http.mjs.map +1 -1
- package/index.js +1 -1
- package/index.js.map +1 -1
- package/index.mjs +1 -1
- package/index.mjs.map +1 -1
- package/instructions.d.mts +2 -0
- package/instructions.d.mts.map +1 -0
- package/instructions.d.ts +2 -0
- package/instructions.d.ts.map +1 -0
- package/instructions.js +55 -0
- package/instructions.js.map +1 -0
- package/instructions.mjs +52 -0
- package/instructions.mjs.map +1 -0
- package/methods.d.mts.map +1 -1
- package/methods.d.ts.map +1 -1
- package/methods.js +0 -54
- package/methods.js.map +1 -1
- package/methods.mjs +0 -54
- package/methods.mjs.map +1 -1
- package/package.json +13 -3
- package/server.d.mts.map +1 -1
- package/server.d.ts.map +1 -1
- package/server.js +55 -42
- package/server.js.map +1 -1
- package/server.mjs +54 -41
- package/server.mjs.map +1 -1
- package/src/code-tool.ts +2 -2
- package/src/http.ts +10 -21
- package/src/index.ts +1 -1
- package/src/instructions.ts +74 -0
- package/src/methods.ts +0 -54
- package/src/server.ts +53 -51
package/server.js
CHANGED
|
@@ -13,39 +13,13 @@ const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
|
|
|
13
13
|
const sdk_1 = __importDefault(require("@roarkanalytics/sdk"));
|
|
14
14
|
const code_tool_1 = require("./code-tool.js");
|
|
15
15
|
const docs_search_tool_1 = __importDefault(require("./docs-search-tool.js"));
|
|
16
|
+
const instructions_1 = require("./instructions.js");
|
|
16
17
|
const methods_1 = require("./methods.js");
|
|
17
|
-
const util_1 = require("./util.js");
|
|
18
|
-
async function getInstructions(stainlessApiKey) {
|
|
19
|
-
// Setting the stainless API key is optional, but may be required
|
|
20
|
-
// to authenticate requests to the Stainless API.
|
|
21
|
-
const response = await fetch((0, util_1.readEnv)('CODE_MODE_INSTRUCTIONS_URL') ?? 'https://api.stainless.com/api/ai/instructions/roark-analytics', {
|
|
22
|
-
method: 'GET',
|
|
23
|
-
headers: { ...(stainlessApiKey && { Authorization: stainlessApiKey }) },
|
|
24
|
-
});
|
|
25
|
-
let instructions;
|
|
26
|
-
if (!response.ok) {
|
|
27
|
-
console.warn('Warning: failed to retrieve MCP server instructions. Proceeding with default instructions...');
|
|
28
|
-
instructions = `
|
|
29
|
-
This is the roark-analytics MCP server. You will use Code Mode to help the user perform
|
|
30
|
-
actions. You can use search_docs tool to learn about how to take action with this server. Then,
|
|
31
|
-
you will write TypeScript code using the execute tool take action. It is CRITICAL that you be
|
|
32
|
-
thoughtful and deliberate when executing code. Always try to entirely solve the problem in code
|
|
33
|
-
block: it can be as long as you need to get the job done!
|
|
34
|
-
`;
|
|
35
|
-
}
|
|
36
|
-
instructions ??= (await response.json()).instructions;
|
|
37
|
-
instructions = `
|
|
38
|
-
The current time in Unix timestamps is ${Date.now()}.
|
|
39
|
-
|
|
40
|
-
${instructions}
|
|
41
|
-
`;
|
|
42
|
-
return instructions;
|
|
43
|
-
}
|
|
44
18
|
const newMcpServer = async (stainlessApiKey) => new mcp_js_1.McpServer({
|
|
45
19
|
name: 'roarkanalytics_sdk_api',
|
|
46
|
-
version: '2.
|
|
20
|
+
version: '2.24.0',
|
|
47
21
|
}, {
|
|
48
|
-
instructions: await getInstructions(stainlessApiKey),
|
|
22
|
+
instructions: await (0, instructions_1.getInstructions)(stainlessApiKey),
|
|
49
23
|
capabilities: { tools: {}, logging: {} },
|
|
50
24
|
});
|
|
51
25
|
exports.newMcpServer = newMcpServer;
|
|
@@ -67,14 +41,33 @@ async function initMcpServer(params) {
|
|
|
67
41
|
warn: logAtLevel('warning'),
|
|
68
42
|
error: logAtLevel('error'),
|
|
69
43
|
};
|
|
70
|
-
let
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
44
|
+
let _client;
|
|
45
|
+
let _clientError;
|
|
46
|
+
let _logLevel;
|
|
47
|
+
const getClient = () => {
|
|
48
|
+
if (_clientError)
|
|
49
|
+
throw _clientError;
|
|
50
|
+
if (!_client) {
|
|
51
|
+
try {
|
|
52
|
+
_client = new sdk_1.default({
|
|
53
|
+
logger,
|
|
54
|
+
...params.clientOptions,
|
|
55
|
+
defaultHeaders: {
|
|
56
|
+
...params.clientOptions?.defaultHeaders,
|
|
57
|
+
'X-Stainless-MCP': 'true',
|
|
58
|
+
},
|
|
59
|
+
});
|
|
60
|
+
if (_logLevel) {
|
|
61
|
+
_client = _client.withOptions({ logLevel: _logLevel });
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
catch (e) {
|
|
65
|
+
_clientError = e instanceof Error ? e : new Error(String(e));
|
|
66
|
+
throw _clientError;
|
|
67
|
+
}
|
|
68
|
+
}
|
|
69
|
+
return _client;
|
|
70
|
+
};
|
|
78
71
|
const providedTools = selectTools(params.mcpOptions);
|
|
79
72
|
const toolMap = Object.fromEntries(providedTools.map((mcpTool) => [mcpTool.tool.name, mcpTool]));
|
|
80
73
|
server.setRequestHandler(types_js_1.ListToolsRequestSchema, async () => {
|
|
@@ -88,6 +81,21 @@ async function initMcpServer(params) {
|
|
|
88
81
|
if (!mcpTool) {
|
|
89
82
|
throw new Error(`Unknown tool: ${name}`);
|
|
90
83
|
}
|
|
84
|
+
let client;
|
|
85
|
+
try {
|
|
86
|
+
client = getClient();
|
|
87
|
+
}
|
|
88
|
+
catch (error) {
|
|
89
|
+
return {
|
|
90
|
+
content: [
|
|
91
|
+
{
|
|
92
|
+
type: 'text',
|
|
93
|
+
text: `Failed to initialize client: ${error instanceof Error ? error.message : String(error)}`,
|
|
94
|
+
},
|
|
95
|
+
],
|
|
96
|
+
isError: true,
|
|
97
|
+
};
|
|
98
|
+
}
|
|
91
99
|
return executeHandler({
|
|
92
100
|
handler: mcpTool.handler,
|
|
93
101
|
reqContext: {
|
|
@@ -99,24 +107,29 @@ async function initMcpServer(params) {
|
|
|
99
107
|
});
|
|
100
108
|
server.setRequestHandler(types_js_1.SetLevelRequestSchema, async (request) => {
|
|
101
109
|
const { level } = request.params;
|
|
110
|
+
let logLevel;
|
|
102
111
|
switch (level) {
|
|
103
112
|
case 'debug':
|
|
104
|
-
|
|
113
|
+
logLevel = 'debug';
|
|
105
114
|
break;
|
|
106
115
|
case 'info':
|
|
107
|
-
|
|
116
|
+
logLevel = 'info';
|
|
108
117
|
break;
|
|
109
118
|
case 'notice':
|
|
110
119
|
case 'warning':
|
|
111
|
-
|
|
120
|
+
logLevel = 'warn';
|
|
112
121
|
break;
|
|
113
122
|
case 'error':
|
|
114
|
-
|
|
123
|
+
logLevel = 'error';
|
|
115
124
|
break;
|
|
116
125
|
default:
|
|
117
|
-
|
|
126
|
+
logLevel = 'off';
|
|
118
127
|
break;
|
|
119
128
|
}
|
|
129
|
+
_logLevel = logLevel;
|
|
130
|
+
if (_client) {
|
|
131
|
+
_client = _client.withOptions({ logLevel });
|
|
132
|
+
}
|
|
120
133
|
return {};
|
|
121
134
|
});
|
|
122
135
|
}
|
package/server.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"server.js","sourceRoot":"","sources":["src/server.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;;;;
|
|
1
|
+
{"version":3,"file":"server.js","sourceRoot":"","sources":["src/server.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;;;;AAkCtF,sCAsHC;AAKD,kCAUC;AAKD,wCAUC;AAnLD,oEAAoE;AACpE,iEAI4C;AAE5C,8DAAwC;AACxC,8CAAuC;AACvC,6EAAgD;AAChD,oDAAiD;AAEjD,0CAAsD;AAG/C,MAAM,YAAY,GAAG,KAAK,EAAE,eAAmC,EAAE,EAAE,CACxE,IAAI,kBAAS,CACX;IACE,IAAI,EAAE,wBAAwB;IAC9B,OAAO,EAAE,QAAQ;CAClB,EACD;IACE,YAAY,EAAE,MAAM,IAAA,8BAAe,EAAC,eAAe,CAAC;IACpD,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;CACzC,CACF,CAAC;AAVS,QAAA,YAAY,gBAUrB;AAEJ;;;GAGG;AACI,KAAK,UAAU,aAAa,CAAC,MAKnC;IACC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,YAAY,kBAAS,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;IAEzF,MAAM,UAAU,GACd,CAAC,KAA6C,EAAE,EAAE,CAClD,CAAC,OAAe,EAAE,GAAG,IAAe,EAAE,EAAE;QACtC,KAAK,MAAM,CAAC,kBAAkB,CAAC;YAC7B,KAAK;YACL,IAAI,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;SACxB,CAAC,CAAC;IACL,CAAC,CAAC;IACJ,MAAM,MAAM,GAAG;QACb,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC;QAC1B,IAAI,EAAE,UAAU,CAAC,MAAM,CAAC;QACxB,IAAI,EAAE,UAAU,CAAC,SAAS,CAAC;QAC3B,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC;KAC3B,CAAC;IAEF,IAAI,OAA0B,CAAC;IAC/B,IAAI,YAA+B,CAAC;IACpC,IAAI,SAAkE,CAAC;IAEvE,MAAM,SAAS,GAAG,GAAU,EAAE;QAC5B,IAAI,YAAY;YAAE,MAAM,YAAY,CAAC;QACrC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,IAAI,CAAC;gBACH,OAAO,GAAG,IAAI,aAAK,CAAC;oBAClB,MAAM;oBACN,GAAG,MAAM,CAAC,aAAa;oBACvB,cAAc,EAAE;wBACd,GAAG,MAAM,CAAC,aAAa,EAAE,cAAc;wBACvC,iBAAiB,EAAE,MAAM;qBAC1B;iBACF,CAAC,CAAC;gBACH,IAAI,SAAS,EAAE,CAAC;oBACd,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;gBACzD,CAAC;YACH,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,YAAY,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC7D,MAAM,YAAY,CAAC;YACrB,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC,CAAC;IAEF,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IACrD,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;IAEjG,MAAM,CAAC,iBAAiB,CAAC,iCAAsB,EAAE,KAAK,IAAI,EAAE;QAC1D,OAAO;YACL,KAAK,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;SACpD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,iBAAiB,CAAC,gCAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAChE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QACjD,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,iBAAiB,IAAI,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,MAAa,CAAC;QAClB,IAAI,CAAC;YACH,MAAM,GAAG,SAAS,EAAE,CAAC;QACvB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,gCAAgC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;qBAC/F;iBACF;gBACD,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;QAED,OAAO,cAAc,CAAC;YACpB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,UAAU,EAAE;gBACV,MAAM;gBACN,eAAe,EAAE,MAAM,CAAC,eAAe,IAAI,MAAM,CAAC,UAAU,EAAE,eAAe;aAC9E;YACD,IAAI;SACL,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,iBAAiB,CAAC,gCAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAChE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QACjC,IAAI,QAAqD,CAAC;QAC1D,QAAQ,KAAK,EAAE,CAAC;YACd,KAAK,OAAO;gBACV,QAAQ,GAAG,OAAO,CAAC;gBACnB,MAAM;YACR,KAAK,MAAM;gBACT,QAAQ,GAAG,MAAM,CAAC;gBAClB,MAAM;YACR,KAAK,QAAQ,CAAC;YACd,KAAK,SAAS;gBACZ,QAAQ,GAAG,MAAM,CAAC;gBAClB,MAAM;YACR,KAAK,OAAO;gBACV,QAAQ,GAAG,OAAO,CAAC;gBACnB,MAAM;YACR;gBACE,QAAQ,GAAG,KAAK,CAAC;gBACjB,MAAM;QACV,CAAC;QACD,SAAS,GAAG,QAAQ,CAAC;QACrB,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC9C,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,SAAgB,WAAW,CAAC,OAAoB;IAC9C,MAAM,aAAa,GAAG;QACpB,IAAA,oBAAQ,EAAC;YACP,cAAc,EAAE,IAAA,mCAAyB,EAAC,OAAO,CAAC;SACnD,CAAC;KACH,CAAC;IACF,IAAI,OAAO,EAAE,gBAAgB,IAAI,IAAI,EAAE,CAAC;QACtC,aAAa,CAAC,IAAI,CAAC,0BAAc,CAAC,CAAC;IACrC,CAAC;IACD,OAAO,aAAa,CAAC;AACvB,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,cAAc,CAAC,EACnC,OAAO,EACP,UAAU,EACV,IAAI,GAKL;IACC,OAAO,MAAM,OAAO,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;AACzD,CAAC"}
|
package/server.mjs
CHANGED
|
@@ -4,37 +4,11 @@ import { CallToolRequestSchema, ListToolsRequestSchema, SetLevelRequestSchema, }
|
|
|
4
4
|
import Roark from '@roarkanalytics/sdk';
|
|
5
5
|
import { codeTool } from "./code-tool.mjs";
|
|
6
6
|
import docsSearchTool from "./docs-search-tool.mjs";
|
|
7
|
+
import { getInstructions } from "./instructions.mjs";
|
|
7
8
|
import { blockedMethodsForCodeTool } from "./methods.mjs";
|
|
8
|
-
import { readEnv } from "./util.mjs";
|
|
9
|
-
async function getInstructions(stainlessApiKey) {
|
|
10
|
-
// Setting the stainless API key is optional, but may be required
|
|
11
|
-
// to authenticate requests to the Stainless API.
|
|
12
|
-
const response = await fetch(readEnv('CODE_MODE_INSTRUCTIONS_URL') ?? 'https://api.stainless.com/api/ai/instructions/roark-analytics', {
|
|
13
|
-
method: 'GET',
|
|
14
|
-
headers: { ...(stainlessApiKey && { Authorization: stainlessApiKey }) },
|
|
15
|
-
});
|
|
16
|
-
let instructions;
|
|
17
|
-
if (!response.ok) {
|
|
18
|
-
console.warn('Warning: failed to retrieve MCP server instructions. Proceeding with default instructions...');
|
|
19
|
-
instructions = `
|
|
20
|
-
This is the roark-analytics MCP server. You will use Code Mode to help the user perform
|
|
21
|
-
actions. You can use search_docs tool to learn about how to take action with this server. Then,
|
|
22
|
-
you will write TypeScript code using the execute tool take action. It is CRITICAL that you be
|
|
23
|
-
thoughtful and deliberate when executing code. Always try to entirely solve the problem in code
|
|
24
|
-
block: it can be as long as you need to get the job done!
|
|
25
|
-
`;
|
|
26
|
-
}
|
|
27
|
-
instructions ??= (await response.json()).instructions;
|
|
28
|
-
instructions = `
|
|
29
|
-
The current time in Unix timestamps is ${Date.now()}.
|
|
30
|
-
|
|
31
|
-
${instructions}
|
|
32
|
-
`;
|
|
33
|
-
return instructions;
|
|
34
|
-
}
|
|
35
9
|
export const newMcpServer = async (stainlessApiKey) => new McpServer({
|
|
36
10
|
name: 'roarkanalytics_sdk_api',
|
|
37
|
-
version: '2.
|
|
11
|
+
version: '2.24.0',
|
|
38
12
|
}, {
|
|
39
13
|
instructions: await getInstructions(stainlessApiKey),
|
|
40
14
|
capabilities: { tools: {}, logging: {} },
|
|
@@ -57,14 +31,33 @@ export async function initMcpServer(params) {
|
|
|
57
31
|
warn: logAtLevel('warning'),
|
|
58
32
|
error: logAtLevel('error'),
|
|
59
33
|
};
|
|
60
|
-
let
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
34
|
+
let _client;
|
|
35
|
+
let _clientError;
|
|
36
|
+
let _logLevel;
|
|
37
|
+
const getClient = () => {
|
|
38
|
+
if (_clientError)
|
|
39
|
+
throw _clientError;
|
|
40
|
+
if (!_client) {
|
|
41
|
+
try {
|
|
42
|
+
_client = new Roark({
|
|
43
|
+
logger,
|
|
44
|
+
...params.clientOptions,
|
|
45
|
+
defaultHeaders: {
|
|
46
|
+
...params.clientOptions?.defaultHeaders,
|
|
47
|
+
'X-Stainless-MCP': 'true',
|
|
48
|
+
},
|
|
49
|
+
});
|
|
50
|
+
if (_logLevel) {
|
|
51
|
+
_client = _client.withOptions({ logLevel: _logLevel });
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
catch (e) {
|
|
55
|
+
_clientError = e instanceof Error ? e : new Error(String(e));
|
|
56
|
+
throw _clientError;
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
return _client;
|
|
60
|
+
};
|
|
68
61
|
const providedTools = selectTools(params.mcpOptions);
|
|
69
62
|
const toolMap = Object.fromEntries(providedTools.map((mcpTool) => [mcpTool.tool.name, mcpTool]));
|
|
70
63
|
server.setRequestHandler(ListToolsRequestSchema, async () => {
|
|
@@ -78,6 +71,21 @@ export async function initMcpServer(params) {
|
|
|
78
71
|
if (!mcpTool) {
|
|
79
72
|
throw new Error(`Unknown tool: ${name}`);
|
|
80
73
|
}
|
|
74
|
+
let client;
|
|
75
|
+
try {
|
|
76
|
+
client = getClient();
|
|
77
|
+
}
|
|
78
|
+
catch (error) {
|
|
79
|
+
return {
|
|
80
|
+
content: [
|
|
81
|
+
{
|
|
82
|
+
type: 'text',
|
|
83
|
+
text: `Failed to initialize client: ${error instanceof Error ? error.message : String(error)}`,
|
|
84
|
+
},
|
|
85
|
+
],
|
|
86
|
+
isError: true,
|
|
87
|
+
};
|
|
88
|
+
}
|
|
81
89
|
return executeHandler({
|
|
82
90
|
handler: mcpTool.handler,
|
|
83
91
|
reqContext: {
|
|
@@ -89,24 +97,29 @@ export async function initMcpServer(params) {
|
|
|
89
97
|
});
|
|
90
98
|
server.setRequestHandler(SetLevelRequestSchema, async (request) => {
|
|
91
99
|
const { level } = request.params;
|
|
100
|
+
let logLevel;
|
|
92
101
|
switch (level) {
|
|
93
102
|
case 'debug':
|
|
94
|
-
|
|
103
|
+
logLevel = 'debug';
|
|
95
104
|
break;
|
|
96
105
|
case 'info':
|
|
97
|
-
|
|
106
|
+
logLevel = 'info';
|
|
98
107
|
break;
|
|
99
108
|
case 'notice':
|
|
100
109
|
case 'warning':
|
|
101
|
-
|
|
110
|
+
logLevel = 'warn';
|
|
102
111
|
break;
|
|
103
112
|
case 'error':
|
|
104
|
-
|
|
113
|
+
logLevel = 'error';
|
|
105
114
|
break;
|
|
106
115
|
default:
|
|
107
|
-
|
|
116
|
+
logLevel = 'off';
|
|
108
117
|
break;
|
|
109
118
|
}
|
|
119
|
+
_logLevel = logLevel;
|
|
120
|
+
if (_client) {
|
|
121
|
+
_client = _client.withOptions({ logLevel });
|
|
122
|
+
}
|
|
110
123
|
return {};
|
|
111
124
|
});
|
|
112
125
|
}
|
package/server.mjs.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"server.mjs","sourceRoot":"","sources":["src/server.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAG/E,EAAE,SAAS,EAAE,MAAM,yCAAyC;OAC5D,EACL,qBAAqB,EACrB,sBAAsB,EACtB,qBAAqB,GACtB,MAAM,oCAAoC;OAEpC,KAAK,MAAM,qBAAqB;OAChC,EAAE,QAAQ,EAAE;OACZ,cAAc;
|
|
1
|
+
{"version":3,"file":"server.mjs","sourceRoot":"","sources":["src/server.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAG/E,EAAE,SAAS,EAAE,MAAM,yCAAyC;OAC5D,EACL,qBAAqB,EACrB,sBAAsB,EACtB,qBAAqB,GACtB,MAAM,oCAAoC;OAEpC,KAAK,MAAM,qBAAqB;OAChC,EAAE,QAAQ,EAAE;OACZ,cAAc;OACd,EAAE,eAAe,EAAE;OAEnB,EAAE,yBAAyB,EAAE;AAGpC,MAAM,CAAC,MAAM,YAAY,GAAG,KAAK,EAAE,eAAmC,EAAE,EAAE,CACxE,IAAI,SAAS,CACX;IACE,IAAI,EAAE,wBAAwB;IAC9B,OAAO,EAAE,QAAQ;CAClB,EACD;IACE,YAAY,EAAE,MAAM,eAAe,CAAC,eAAe,CAAC;IACpD,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;CACzC,CACF,CAAC;AAEJ;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,MAKnC;IACC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,YAAY,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;IAEzF,MAAM,UAAU,GACd,CAAC,KAA6C,EAAE,EAAE,CAClD,CAAC,OAAe,EAAE,GAAG,IAAe,EAAE,EAAE;QACtC,KAAK,MAAM,CAAC,kBAAkB,CAAC;YAC7B,KAAK;YACL,IAAI,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;SACxB,CAAC,CAAC;IACL,CAAC,CAAC;IACJ,MAAM,MAAM,GAAG;QACb,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC;QAC1B,IAAI,EAAE,UAAU,CAAC,MAAM,CAAC;QACxB,IAAI,EAAE,UAAU,CAAC,SAAS,CAAC;QAC3B,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC;KAC3B,CAAC;IAEF,IAAI,OAA0B,CAAC;IAC/B,IAAI,YAA+B,CAAC;IACpC,IAAI,SAAkE,CAAC;IAEvE,MAAM,SAAS,GAAG,GAAU,EAAE;QAC5B,IAAI,YAAY;YAAE,MAAM,YAAY,CAAC;QACrC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,IAAI,CAAC;gBACH,OAAO,GAAG,IAAI,KAAK,CAAC;oBAClB,MAAM;oBACN,GAAG,MAAM,CAAC,aAAa;oBACvB,cAAc,EAAE;wBACd,GAAG,MAAM,CAAC,aAAa,EAAE,cAAc;wBACvC,iBAAiB,EAAE,MAAM;qBAC1B;iBACF,CAAC,CAAC;gBACH,IAAI,SAAS,EAAE,CAAC;oBACd,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;gBACzD,CAAC;YACH,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,YAAY,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC7D,MAAM,YAAY,CAAC;YACrB,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC,CAAC;IAEF,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IACrD,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;IAEjG,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QAC1D,OAAO;YACL,KAAK,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;SACpD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAChE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QACjD,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,iBAAiB,IAAI,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,MAAa,CAAC;QAClB,IAAI,CAAC;YACH,MAAM,GAAG,SAAS,EAAE,CAAC;QACvB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,gCAAgC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;qBAC/F;iBACF;gBACD,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;QAED,OAAO,cAAc,CAAC;YACpB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,UAAU,EAAE;gBACV,MAAM;gBACN,eAAe,EAAE,MAAM,CAAC,eAAe,IAAI,MAAM,CAAC,UAAU,EAAE,eAAe;aAC9E;YACD,IAAI;SACL,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAChE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QACjC,IAAI,QAAqD,CAAC;QAC1D,QAAQ,KAAK,EAAE,CAAC;YACd,KAAK,OAAO;gBACV,QAAQ,GAAG,OAAO,CAAC;gBACnB,MAAM;YACR,KAAK,MAAM;gBACT,QAAQ,GAAG,MAAM,CAAC;gBAClB,MAAM;YACR,KAAK,QAAQ,CAAC;YACd,KAAK,SAAS;gBACZ,QAAQ,GAAG,MAAM,CAAC;gBAClB,MAAM;YACR,KAAK,OAAO;gBACV,QAAQ,GAAG,OAAO,CAAC;gBACnB,MAAM;YACR;gBACE,QAAQ,GAAG,KAAK,CAAC;gBACjB,MAAM;QACV,CAAC;QACD,SAAS,GAAG,QAAQ,CAAC;QACrB,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC9C,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,WAAW,CAAC,OAAoB;IAC9C,MAAM,aAAa,GAAG;QACpB,QAAQ,CAAC;YACP,cAAc,EAAE,yBAAyB,CAAC,OAAO,CAAC;SACnD,CAAC;KACH,CAAC;IACF,IAAI,OAAO,EAAE,gBAAgB,IAAI,IAAI,EAAE,CAAC;QACtC,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACrC,CAAC;IACD,OAAO,aAAa,CAAC;AACvB,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,EACnC,OAAO,EACP,UAAU,EACV,IAAI,GAKL;IACC,OAAO,MAAM,OAAO,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;AACzD,CAAC"}
|
package/src/code-tool.ts
CHANGED
|
@@ -21,9 +21,9 @@ For example:
|
|
|
21
21
|
|
|
22
22
|
\`\`\`
|
|
23
23
|
async function run(client) {
|
|
24
|
-
const
|
|
24
|
+
const health = await client.health.get();
|
|
25
25
|
|
|
26
|
-
console.log(
|
|
26
|
+
console.log(health.data);
|
|
27
27
|
}
|
|
28
28
|
\`\`\`
|
|
29
29
|
|
package/src/http.ts
CHANGED
|
@@ -24,28 +24,17 @@ const newServer = async ({
|
|
|
24
24
|
const stainlessApiKey = getStainlessApiKey(req, mcpOptions);
|
|
25
25
|
const server = await newMcpServer(stainlessApiKey);
|
|
26
26
|
|
|
27
|
-
|
|
28
|
-
const authOptions = parseClientAuthHeaders(req, false);
|
|
27
|
+
const authOptions = parseClientAuthHeaders(req, false);
|
|
29
28
|
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
} catch (error) {
|
|
40
|
-
res.status(401).json({
|
|
41
|
-
jsonrpc: '2.0',
|
|
42
|
-
error: {
|
|
43
|
-
code: -32000,
|
|
44
|
-
message: `Unauthorized: ${error instanceof Error ? error.message : error}`,
|
|
45
|
-
},
|
|
46
|
-
});
|
|
47
|
-
return null;
|
|
48
|
-
}
|
|
29
|
+
await initMcpServer({
|
|
30
|
+
server: server,
|
|
31
|
+
mcpOptions: mcpOptions,
|
|
32
|
+
clientOptions: {
|
|
33
|
+
...clientOptions,
|
|
34
|
+
...authOptions,
|
|
35
|
+
},
|
|
36
|
+
stainlessApiKey: stainlessApiKey,
|
|
37
|
+
});
|
|
49
38
|
|
|
50
39
|
return server;
|
|
51
40
|
};
|
package/src/index.ts
CHANGED
|
@@ -0,0 +1,74 @@
|
|
|
1
|
+
// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
|
|
2
|
+
|
|
3
|
+
import { readEnv } from './util';
|
|
4
|
+
|
|
5
|
+
const INSTRUCTIONS_CACHE_TTL_MS = 15 * 60 * 1000; // 15 minutes
|
|
6
|
+
|
|
7
|
+
interface InstructionsCacheEntry {
|
|
8
|
+
fetchedInstructions: string;
|
|
9
|
+
fetchedAt: number;
|
|
10
|
+
}
|
|
11
|
+
|
|
12
|
+
const instructionsCache = new Map<string, InstructionsCacheEntry>();
|
|
13
|
+
|
|
14
|
+
// Periodically evict stale entries so the cache doesn't grow unboundedly.
|
|
15
|
+
const _cacheCleanupInterval = setInterval(() => {
|
|
16
|
+
const now = Date.now();
|
|
17
|
+
for (const [key, entry] of instructionsCache) {
|
|
18
|
+
if (now - entry.fetchedAt > INSTRUCTIONS_CACHE_TTL_MS) {
|
|
19
|
+
instructionsCache.delete(key);
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
}, INSTRUCTIONS_CACHE_TTL_MS);
|
|
23
|
+
|
|
24
|
+
// Don't keep the process alive just for cleanup.
|
|
25
|
+
_cacheCleanupInterval.unref();
|
|
26
|
+
|
|
27
|
+
export async function getInstructions(stainlessApiKey: string | undefined): Promise<string> {
|
|
28
|
+
const cacheKey = stainlessApiKey ?? '';
|
|
29
|
+
const cached = instructionsCache.get(cacheKey);
|
|
30
|
+
|
|
31
|
+
if (cached && Date.now() - cached.fetchedAt <= INSTRUCTIONS_CACHE_TTL_MS) {
|
|
32
|
+
return cached.fetchedInstructions;
|
|
33
|
+
}
|
|
34
|
+
|
|
35
|
+
const fetchedInstructions = await fetchLatestInstructions(stainlessApiKey);
|
|
36
|
+
instructionsCache.set(cacheKey, { fetchedInstructions, fetchedAt: Date.now() });
|
|
37
|
+
return fetchedInstructions;
|
|
38
|
+
}
|
|
39
|
+
|
|
40
|
+
async function fetchLatestInstructions(stainlessApiKey: string | undefined): Promise<string> {
|
|
41
|
+
// Setting the stainless API key is optional, but may be required
|
|
42
|
+
// to authenticate requests to the Stainless API.
|
|
43
|
+
const response = await fetch(
|
|
44
|
+
readEnv('CODE_MODE_INSTRUCTIONS_URL') ?? 'https://api.stainless.com/api/ai/instructions/roark-analytics',
|
|
45
|
+
{
|
|
46
|
+
method: 'GET',
|
|
47
|
+
headers: { ...(stainlessApiKey && { Authorization: stainlessApiKey }) },
|
|
48
|
+
},
|
|
49
|
+
);
|
|
50
|
+
|
|
51
|
+
let instructions: string | undefined;
|
|
52
|
+
if (!response.ok) {
|
|
53
|
+
console.warn(
|
|
54
|
+
'Warning: failed to retrieve MCP server instructions. Proceeding with default instructions...',
|
|
55
|
+
);
|
|
56
|
+
|
|
57
|
+
instructions = `
|
|
58
|
+
This is the roark-analytics MCP server. You will use Code Mode to help the user perform
|
|
59
|
+
actions. You can use search_docs tool to learn about how to take action with this server. Then,
|
|
60
|
+
you will write TypeScript code using the execute tool take action. It is CRITICAL that you be
|
|
61
|
+
thoughtful and deliberate when executing code. Always try to entirely solve the problem in code
|
|
62
|
+
block: it can be as long as you need to get the job done!
|
|
63
|
+
`;
|
|
64
|
+
}
|
|
65
|
+
|
|
66
|
+
instructions ??= ((await response.json()) as { instructions: string }).instructions;
|
|
67
|
+
instructions = `
|
|
68
|
+
If needed, you can get the current time by executing Date.now().
|
|
69
|
+
|
|
70
|
+
${instructions}
|
|
71
|
+
`;
|
|
72
|
+
|
|
73
|
+
return instructions;
|
|
74
|
+
}
|
package/src/methods.ts
CHANGED
|
@@ -16,48 +16,6 @@ export const sdkMethods: SdkMethod[] = [
|
|
|
16
16
|
httpMethod: 'get',
|
|
17
17
|
httpPath: '/health',
|
|
18
18
|
},
|
|
19
|
-
{
|
|
20
|
-
clientCallName: 'client.evaluation.createEvaluator',
|
|
21
|
-
fullyQualifiedName: 'evaluation.createEvaluator',
|
|
22
|
-
httpMethod: 'post',
|
|
23
|
-
httpPath: '/v1/evaluation/evaluators',
|
|
24
|
-
},
|
|
25
|
-
{
|
|
26
|
-
clientCallName: 'client.evaluation.createJob',
|
|
27
|
-
fullyQualifiedName: 'evaluation.createJob',
|
|
28
|
-
httpMethod: 'post',
|
|
29
|
-
httpPath: '/v1/evaluation/job',
|
|
30
|
-
},
|
|
31
|
-
{
|
|
32
|
-
clientCallName: 'client.evaluation.getEvaluatorByID',
|
|
33
|
-
fullyQualifiedName: 'evaluation.getEvaluatorByID',
|
|
34
|
-
httpMethod: 'get',
|
|
35
|
-
httpPath: '/v1/evaluation/evaluators/{evaluatorId}',
|
|
36
|
-
},
|
|
37
|
-
{
|
|
38
|
-
clientCallName: 'client.evaluation.getJob',
|
|
39
|
-
fullyQualifiedName: 'evaluation.getJob',
|
|
40
|
-
httpMethod: 'get',
|
|
41
|
-
httpPath: '/v1/evaluation/job/{jobId}',
|
|
42
|
-
},
|
|
43
|
-
{
|
|
44
|
-
clientCallName: 'client.evaluation.listEvaluators',
|
|
45
|
-
fullyQualifiedName: 'evaluation.listEvaluators',
|
|
46
|
-
httpMethod: 'get',
|
|
47
|
-
httpPath: '/v1/evaluation/evaluators',
|
|
48
|
-
},
|
|
49
|
-
{
|
|
50
|
-
clientCallName: 'client.evaluation.listJobRuns',
|
|
51
|
-
fullyQualifiedName: 'evaluation.listJobRuns',
|
|
52
|
-
httpMethod: 'get',
|
|
53
|
-
httpPath: '/v1/evaluation/job/{jobId}/runs',
|
|
54
|
-
},
|
|
55
|
-
{
|
|
56
|
-
clientCallName: 'client.evaluation.updateEvaluator',
|
|
57
|
-
fullyQualifiedName: 'evaluation.updateEvaluator',
|
|
58
|
-
httpMethod: 'put',
|
|
59
|
-
httpPath: '/v1/evaluation/evaluators/{evaluatorId}',
|
|
60
|
-
},
|
|
61
19
|
{
|
|
62
20
|
clientCallName: 'client.call.create',
|
|
63
21
|
fullyQualifiedName: 'call.create',
|
|
@@ -106,18 +64,6 @@ export const sdkMethods: SdkMethod[] = [
|
|
|
106
64
|
httpMethod: 'get',
|
|
107
65
|
httpPath: '/v1/metric/definitions',
|
|
108
66
|
},
|
|
109
|
-
{
|
|
110
|
-
clientCallName: 'client.integrations.createRetellCall',
|
|
111
|
-
fullyQualifiedName: 'integrations.createRetellCall',
|
|
112
|
-
httpMethod: 'post',
|
|
113
|
-
httpPath: '/v1/retell/call',
|
|
114
|
-
},
|
|
115
|
-
{
|
|
116
|
-
clientCallName: 'client.integrations.createVapiCall',
|
|
117
|
-
fullyQualifiedName: 'integrations.createVapiCall',
|
|
118
|
-
httpMethod: 'post',
|
|
119
|
-
httpPath: '/v1/vapi/call',
|
|
120
|
-
},
|
|
121
67
|
{
|
|
122
68
|
clientCallName: 'client.simulationJob.getByID',
|
|
123
69
|
fullyQualifiedName: 'simulationJob.getByID',
|