@llamaindex/llama-cloud-mcp 1.5.0 → 1.7.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.
Files changed (131) hide show
  1. package/auth.d.mts +6 -0
  2. package/auth.d.mts.map +1 -0
  3. package/auth.d.ts +6 -0
  4. package/auth.d.ts.map +1 -0
  5. package/{headers.mjs → auth.js} +18 -2
  6. package/auth.js.map +1 -0
  7. package/{headers.js → auth.mjs} +13 -6
  8. package/auth.mjs.map +1 -0
  9. package/code-tool-paths.cjs +6 -0
  10. package/code-tool-paths.cjs.map +1 -0
  11. package/code-tool-paths.d.cts +2 -0
  12. package/code-tool-paths.d.cts.map +1 -0
  13. package/code-tool-types.d.mts.map +1 -1
  14. package/code-tool-types.d.ts.map +1 -1
  15. package/code-tool-worker.d.mts +5 -0
  16. package/code-tool-worker.d.mts.map +1 -0
  17. package/code-tool-worker.d.ts +5 -0
  18. package/code-tool-worker.d.ts.map +1 -0
  19. package/code-tool-worker.js +345 -0
  20. package/code-tool-worker.js.map +1 -0
  21. package/code-tool-worker.mjs +340 -0
  22. package/code-tool-worker.mjs.map +1 -0
  23. package/code-tool.d.mts +8 -2
  24. package/code-tool.d.mts.map +1 -1
  25. package/code-tool.d.ts +8 -2
  26. package/code-tool.d.ts.map +1 -1
  27. package/code-tool.js +264 -42
  28. package/code-tool.js.map +1 -1
  29. package/code-tool.mjs +229 -43
  30. package/code-tool.mjs.map +1 -1
  31. package/docs-search-tool.d.mts +9 -3
  32. package/docs-search-tool.d.mts.map +1 -1
  33. package/docs-search-tool.d.ts +9 -3
  34. package/docs-search-tool.d.ts.map +1 -1
  35. package/docs-search-tool.js +27 -4
  36. package/docs-search-tool.js.map +1 -1
  37. package/docs-search-tool.mjs +27 -4
  38. package/docs-search-tool.mjs.map +1 -1
  39. package/http.d.mts +3 -5
  40. package/http.d.mts.map +1 -1
  41. package/http.d.ts +3 -5
  42. package/http.d.ts.map +1 -1
  43. package/http.js +66 -43
  44. package/http.js.map +1 -1
  45. package/http.mjs +66 -43
  46. package/http.mjs.map +1 -1
  47. package/index.js +13 -12
  48. package/index.js.map +1 -1
  49. package/index.mjs +13 -12
  50. package/index.mjs.map +1 -1
  51. package/instructions.d.mts +2 -0
  52. package/instructions.d.mts.map +1 -0
  53. package/instructions.d.ts +2 -0
  54. package/instructions.d.ts.map +1 -0
  55. package/instructions.js +56 -0
  56. package/instructions.js.map +1 -0
  57. package/instructions.mjs +53 -0
  58. package/instructions.mjs.map +1 -0
  59. package/logger.d.mts +7 -0
  60. package/logger.d.mts.map +1 -0
  61. package/logger.d.ts +7 -0
  62. package/logger.d.ts.map +1 -0
  63. package/logger.js +29 -0
  64. package/logger.js.map +1 -0
  65. package/logger.mjs +22 -0
  66. package/logger.mjs.map +1 -0
  67. package/methods.d.mts.map +1 -1
  68. package/methods.d.ts.map +1 -1
  69. package/methods.js +1 -0
  70. package/methods.js.map +1 -1
  71. package/methods.mjs +1 -0
  72. package/methods.mjs.map +1 -1
  73. package/options.d.mts +5 -0
  74. package/options.d.mts.map +1 -1
  75. package/options.d.ts +5 -0
  76. package/options.d.ts.map +1 -1
  77. package/options.js +31 -0
  78. package/options.js.map +1 -1
  79. package/options.mjs +31 -0
  80. package/options.mjs.map +1 -1
  81. package/package.json +62 -15
  82. package/server.d.mts +8 -9
  83. package/server.d.mts.map +1 -1
  84. package/server.d.ts +8 -9
  85. package/server.d.ts.map +1 -1
  86. package/server.js +73 -75
  87. package/server.js.map +1 -1
  88. package/server.mjs +72 -71
  89. package/server.mjs.map +1 -1
  90. package/src/{headers.ts → auth.ts} +16 -1
  91. package/src/code-tool-paths.cts +3 -0
  92. package/src/code-tool-types.ts +1 -0
  93. package/src/code-tool-worker.ts +391 -0
  94. package/src/code-tool.ts +309 -56
  95. package/src/docs-search-tool.ts +46 -6
  96. package/src/http.ts +74 -45
  97. package/src/index.ts +15 -13
  98. package/src/instructions.ts +75 -0
  99. package/src/logger.ts +28 -0
  100. package/src/methods.ts +2 -0
  101. package/src/options.ts +43 -0
  102. package/src/server.ts +85 -92
  103. package/src/stdio.ts +4 -3
  104. package/src/types.ts +12 -4
  105. package/src/util.ts +25 -0
  106. package/stdio.d.mts.map +1 -1
  107. package/stdio.d.ts.map +1 -1
  108. package/stdio.js +4 -3
  109. package/stdio.js.map +1 -1
  110. package/stdio.mjs +4 -3
  111. package/stdio.mjs.map +1 -1
  112. package/types.d.mts +8 -1
  113. package/types.d.mts.map +1 -1
  114. package/types.d.ts +8 -1
  115. package/types.d.ts.map +1 -1
  116. package/types.js.map +1 -1
  117. package/types.mjs.map +1 -1
  118. package/util.d.mts +4 -0
  119. package/util.d.mts.map +1 -0
  120. package/util.d.ts +4 -0
  121. package/util.d.ts.map +1 -0
  122. package/util.js +30 -0
  123. package/util.js.map +1 -0
  124. package/util.mjs +24 -0
  125. package/util.mjs.map +1 -0
  126. package/headers.d.mts +0 -4
  127. package/headers.d.mts.map +0 -1
  128. package/headers.d.ts +0 -4
  129. package/headers.d.ts.map +0 -1
  130. package/headers.js.map +0 -1
  131. package/headers.mjs.map +0 -1
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@llamaindex/llama-cloud-mcp",
3
- "version": "1.5.0",
3
+ "version": "1.7.0",
4
4
  "description": "The official MCP Server for the Llama Cloud API",
5
5
  "author": "Llama Cloud <>",
6
6
  "types": "./index.d.ts",
@@ -27,17 +27,18 @@
27
27
  "fix": "eslint --fix --ext ts,js ."
28
28
  },
29
29
  "dependencies": {
30
- "@llamaindex/llama-cloud": "^1.5.0",
30
+ "@llamaindex/llama-cloud": "^1.7.0",
31
31
  "@cloudflare/cabidela": "^0.2.4",
32
- "@modelcontextprotocol/sdk": "^1.25.2",
32
+ "@modelcontextprotocol/sdk": "^1.26.0",
33
33
  "@valtown/deno-http-worker": "^0.0.21",
34
34
  "cookie-parser": "^1.4.6",
35
35
  "cors": "^2.8.5",
36
36
  "express": "^5.1.0",
37
37
  "fuse.js": "^7.1.0",
38
38
  "jq-web": "https://github.com/stainless-api/jq-web/releases/download/v0.8.8/jq-web.tar.gz",
39
- "morgan": "^1.10.0",
40
- "morgan-body": "^2.6.9",
39
+ "pino": "^10.3.1",
40
+ "pino-http": "^11.0.0",
41
+ "pino-pretty": "^13.1.3",
41
42
  "qs": "^6.14.1",
42
43
  "typescript": "5.8.3",
43
44
  "yargs": "^17.7.2",
@@ -61,6 +62,22 @@
61
62
  "types": "./index.d.mts",
62
63
  "default": "./index.mjs"
63
64
  },
65
+ "./auth": {
66
+ "import": "./auth.mjs",
67
+ "require": "./auth.js"
68
+ },
69
+ "./auth.js": {
70
+ "default": "./auth.js"
71
+ },
72
+ "./auth.mjs": {
73
+ "default": "./auth.mjs"
74
+ },
75
+ "./code-tool-paths": {
76
+ "require": "./code-tool-paths.cjs"
77
+ },
78
+ "./code-tool-paths.cjs": {
79
+ "default": "./code-tool-paths.cjs"
80
+ },
64
81
  "./code-tool-types": {
65
82
  "import": "./code-tool-types.mjs",
66
83
  "require": "./code-tool-types.js"
@@ -71,6 +88,16 @@
71
88
  "./code-tool-types.mjs": {
72
89
  "default": "./code-tool-types.mjs"
73
90
  },
91
+ "./code-tool-worker": {
92
+ "import": "./code-tool-worker.mjs",
93
+ "require": "./code-tool-worker.js"
94
+ },
95
+ "./code-tool-worker.js": {
96
+ "default": "./code-tool-worker.js"
97
+ },
98
+ "./code-tool-worker.mjs": {
99
+ "default": "./code-tool-worker.mjs"
100
+ },
74
101
  "./code-tool": {
75
102
  "import": "./code-tool.mjs",
76
103
  "require": "./code-tool.js"
@@ -91,16 +118,6 @@
91
118
  "./docs-search-tool.mjs": {
92
119
  "default": "./docs-search-tool.mjs"
93
120
  },
94
- "./headers": {
95
- "import": "./headers.mjs",
96
- "require": "./headers.js"
97
- },
98
- "./headers.js": {
99
- "default": "./headers.js"
100
- },
101
- "./headers.mjs": {
102
- "default": "./headers.mjs"
103
- },
104
121
  "./http": {
105
122
  "import": "./http.mjs",
106
123
  "require": "./http.js"
@@ -121,6 +138,26 @@
121
138
  "./index.mjs": {
122
139
  "default": "./index.mjs"
123
140
  },
141
+ "./instructions": {
142
+ "import": "./instructions.mjs",
143
+ "require": "./instructions.js"
144
+ },
145
+ "./instructions.js": {
146
+ "default": "./instructions.js"
147
+ },
148
+ "./instructions.mjs": {
149
+ "default": "./instructions.mjs"
150
+ },
151
+ "./logger": {
152
+ "import": "./logger.mjs",
153
+ "require": "./logger.js"
154
+ },
155
+ "./logger.js": {
156
+ "default": "./logger.js"
157
+ },
158
+ "./logger.mjs": {
159
+ "default": "./logger.mjs"
160
+ },
124
161
  "./methods": {
125
162
  "import": "./methods.mjs",
126
163
  "require": "./methods.js"
@@ -170,6 +207,16 @@
170
207
  },
171
208
  "./types.mjs": {
172
209
  "default": "./types.mjs"
210
+ },
211
+ "./util": {
212
+ "import": "./util.mjs",
213
+ "require": "./util.js"
214
+ },
215
+ "./util.js": {
216
+ "default": "./util.js"
217
+ },
218
+ "./util.mjs": {
219
+ "default": "./util.mjs"
173
220
  }
174
221
  }
175
222
  }
package/server.d.mts CHANGED
@@ -1,12 +1,9 @@
1
1
  import { Server } from '@modelcontextprotocol/sdk/server/index.js';
2
2
  import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
3
3
  import { ClientOptions } from '@llamaindex/llama-cloud';
4
- import LlamaCloud from '@llamaindex/llama-cloud';
5
4
  import { McpOptions } from "./options.mjs";
6
- import { HandlerFunction, McpTool } from "./types.mjs";
7
- export { McpOptions } from "./options.mjs";
8
- export { ClientOptions } from '@llamaindex/llama-cloud';
9
- export declare const newMcpServer: () => Promise<McpServer>;
5
+ import { HandlerFunction, McpRequestContext, ToolCallResult, McpTool } from "./types.mjs";
6
+ export declare const newMcpServer: (stainlessApiKey: string | undefined) => Promise<McpServer>;
10
7
  /**
11
8
  * Initializes the provided MCP Server with the given tools and handlers.
12
9
  * If not provided, the default client, tools and handlers will be used.
@@ -15,6 +12,7 @@ export declare function initMcpServer(params: {
15
12
  server: Server | McpServer;
16
13
  clientOptions?: ClientOptions;
17
14
  mcpOptions?: McpOptions;
15
+ stainlessApiKey?: string | undefined;
18
16
  }): Promise<void>;
19
17
  /**
20
18
  * Selects the tools to include in the MCP Server based on the provided options.
@@ -23,8 +21,9 @@ export declare function selectTools(options?: McpOptions): McpTool[];
23
21
  /**
24
22
  * Runs the provided handler with the given client and arguments.
25
23
  */
26
- export declare function executeHandler(handler: HandlerFunction, client: LlamaCloud, args: Record<string, unknown> | undefined): Promise<import("./types").ToolCallResult>;
27
- export declare const readEnv: (env: string) => string | undefined;
28
- export declare const readEnvOrError: (env: string) => string;
29
- export declare const requireValue: <T>(value: T | undefined, description: string) => T;
24
+ export declare function executeHandler({ handler, reqContext, args, }: {
25
+ handler: HandlerFunction;
26
+ reqContext: McpRequestContext;
27
+ args: Record<string, unknown> | undefined;
28
+ }): Promise<ToolCallResult>;
30
29
  //# sourceMappingURL=server.d.mts.map
package/server.d.mts.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"server.d.mts","sourceRoot":"","sources":["src/server.ts"],"names":[],"mappings":"OAEO,EAAE,MAAM,EAAE,MAAM,2CAA2C;OAC3D,EAAE,SAAS,EAAE,MAAM,yCAAyC;OAM5D,EAAE,aAAa,EAAE,MAAM,yBAAyB;OAChD,UAAU,MAAM,yBAAyB;OAGzC,EAAE,UAAU,EAAE;OAEd,EAAE,eAAe,EAAE,OAAO,EAAE;OAE5B,EAAE,UAAU,EAAE;OACd,EAAE,aAAa,EAAE,MAAM,yBAAyB;AAsCvD,eAAO,MAAM,YAAY,0BAUtB,CAAC;AAEJ;;;GAGG;AACH,wBAAsB,aAAa,CAAC,MAAM,EAAE;IAC1C,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,UAAU,CAAC,EAAE,UAAU,CAAC;CACzB,iBAoEA;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,OAAO,CAAC,EAAE,UAAU,GAAG,OAAO,EAAE,CAU3D;AAED;;GAEG;AACH,wBAAsB,cAAc,CAClC,OAAO,EAAE,eAAe,EACxB,MAAM,EAAE,UAAU,EAClB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,6CAG1C;AAED,eAAO,MAAM,OAAO,GAAI,KAAK,MAAM,KAAG,MAAM,GAAG,SAO9C,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,KAAK,MAAM,KAAG,MAM5C,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,CAAC,EAAE,OAAO,CAAC,GAAG,SAAS,EAAE,aAAa,MAAM,KAAG,CAK3E,CAAC"}
1
+ {"version":3,"file":"server.d.mts","sourceRoot":"","sources":["src/server.ts"],"names":[],"mappings":"OAEO,EAAE,MAAM,EAAE,MAAM,2CAA2C;OAC3D,EAAE,SAAS,EAAE,MAAM,yCAAyC;OAM5D,EAAE,aAAa,EAAE,MAAM,yBAAyB;OAKhD,EAAE,UAAU,EAAE;OAEd,EAAE,eAAe,EAAE,iBAAiB,EAAE,cAAc,EAAE,OAAO,EAAE;AAEtE,eAAO,MAAM,YAAY,GAAU,iBAAiB,MAAM,GAAG,SAAS,uBAUnE,CAAC;AAEJ;;;GAGG;AACH,wBAAsB,aAAa,CAAC,MAAM,EAAE;IAC1C,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACtC,iBAiHA;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,OAAO,CAAC,EAAE,UAAU,GAAG,OAAO,EAAE,CAe3D;AAED;;GAEG;AACH,wBAAsB,cAAc,CAAC,EACnC,OAAO,EACP,UAAU,EACV,IAAI,GACL,EAAE;IACD,OAAO,EAAE,eAAe,CAAC;IACzB,UAAU,EAAE,iBAAiB,CAAC;IAC9B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;CAC3C,GAAG,OAAO,CAAC,cAAc,CAAC,CAE1B"}
package/server.d.ts CHANGED
@@ -1,12 +1,9 @@
1
1
  import { Server } from '@modelcontextprotocol/sdk/server/index.js';
2
2
  import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
3
3
  import { ClientOptions } from '@llamaindex/llama-cloud';
4
- import LlamaCloud from '@llamaindex/llama-cloud';
5
4
  import { McpOptions } from "./options.js";
6
- import { HandlerFunction, McpTool } from "./types.js";
7
- export { McpOptions } from "./options.js";
8
- export { ClientOptions } from '@llamaindex/llama-cloud';
9
- export declare const newMcpServer: () => Promise<McpServer>;
5
+ import { HandlerFunction, McpRequestContext, ToolCallResult, McpTool } from "./types.js";
6
+ export declare const newMcpServer: (stainlessApiKey: string | undefined) => Promise<McpServer>;
10
7
  /**
11
8
  * Initializes the provided MCP Server with the given tools and handlers.
12
9
  * If not provided, the default client, tools and handlers will be used.
@@ -15,6 +12,7 @@ export declare function initMcpServer(params: {
15
12
  server: Server | McpServer;
16
13
  clientOptions?: ClientOptions;
17
14
  mcpOptions?: McpOptions;
15
+ stainlessApiKey?: string | undefined;
18
16
  }): Promise<void>;
19
17
  /**
20
18
  * Selects the tools to include in the MCP Server based on the provided options.
@@ -23,8 +21,9 @@ export declare function selectTools(options?: McpOptions): McpTool[];
23
21
  /**
24
22
  * Runs the provided handler with the given client and arguments.
25
23
  */
26
- export declare function executeHandler(handler: HandlerFunction, client: LlamaCloud, args: Record<string, unknown> | undefined): Promise<import("./types").ToolCallResult>;
27
- export declare const readEnv: (env: string) => string | undefined;
28
- export declare const readEnvOrError: (env: string) => string;
29
- export declare const requireValue: <T>(value: T | undefined, description: string) => T;
24
+ export declare function executeHandler({ handler, reqContext, args, }: {
25
+ handler: HandlerFunction;
26
+ reqContext: McpRequestContext;
27
+ args: Record<string, unknown> | undefined;
28
+ }): Promise<ToolCallResult>;
30
29
  //# sourceMappingURL=server.d.ts.map
package/server.d.ts.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"server.d.ts","sourceRoot":"","sources":["src/server.ts"],"names":[],"mappings":"OAEO,EAAE,MAAM,EAAE,MAAM,2CAA2C;OAC3D,EAAE,SAAS,EAAE,MAAM,yCAAyC;OAM5D,EAAE,aAAa,EAAE,MAAM,yBAAyB;OAChD,UAAU,MAAM,yBAAyB;OAGzC,EAAE,UAAU,EAAE;OAEd,EAAE,eAAe,EAAE,OAAO,EAAE;OAE5B,EAAE,UAAU,EAAE;OACd,EAAE,aAAa,EAAE,MAAM,yBAAyB;AAsCvD,eAAO,MAAM,YAAY,0BAUtB,CAAC;AAEJ;;;GAGG;AACH,wBAAsB,aAAa,CAAC,MAAM,EAAE;IAC1C,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,UAAU,CAAC,EAAE,UAAU,CAAC;CACzB,iBAoEA;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,OAAO,CAAC,EAAE,UAAU,GAAG,OAAO,EAAE,CAU3D;AAED;;GAEG;AACH,wBAAsB,cAAc,CAClC,OAAO,EAAE,eAAe,EACxB,MAAM,EAAE,UAAU,EAClB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,6CAG1C;AAED,eAAO,MAAM,OAAO,GAAI,KAAK,MAAM,KAAG,MAAM,GAAG,SAO9C,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,KAAK,MAAM,KAAG,MAM5C,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,CAAC,EAAE,OAAO,CAAC,GAAG,SAAS,EAAE,aAAa,MAAM,KAAG,CAK3E,CAAC"}
1
+ {"version":3,"file":"server.d.ts","sourceRoot":"","sources":["src/server.ts"],"names":[],"mappings":"OAEO,EAAE,MAAM,EAAE,MAAM,2CAA2C;OAC3D,EAAE,SAAS,EAAE,MAAM,yCAAyC;OAM5D,EAAE,aAAa,EAAE,MAAM,yBAAyB;OAKhD,EAAE,UAAU,EAAE;OAEd,EAAE,eAAe,EAAE,iBAAiB,EAAE,cAAc,EAAE,OAAO,EAAE;AAEtE,eAAO,MAAM,YAAY,GAAU,iBAAiB,MAAM,GAAG,SAAS,uBAUnE,CAAC;AAEJ;;;GAGG;AACH,wBAAsB,aAAa,CAAC,MAAM,EAAE;IAC1C,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACtC,iBAiHA;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,OAAO,CAAC,EAAE,UAAU,GAAG,OAAO,EAAE,CAe3D;AAED;;GAEG;AACH,wBAAsB,cAAc,CAAC,EACnC,OAAO,EACP,UAAU,EACV,IAAI,GACL,EAAE;IACD,OAAO,EAAE,eAAe,CAAC;IACzB,UAAU,EAAE,iBAAiB,CAAC;IAC9B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;CAC3C,GAAG,OAAO,CAAC,cAAc,CAAC,CAE1B"}
package/server.js CHANGED
@@ -4,7 +4,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
4
4
  return (mod && mod.__esModule) ? mod : { "default": mod };
5
5
  };
6
6
  Object.defineProperty(exports, "__esModule", { value: true });
7
- exports.requireValue = exports.readEnvOrError = exports.readEnv = exports.newMcpServer = void 0;
7
+ exports.newMcpServer = void 0;
8
8
  exports.initMcpServer = initMcpServer;
9
9
  exports.selectTools = selectTools;
10
10
  exports.executeHandler = executeHandler;
@@ -13,38 +13,13 @@ const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
13
13
  const llama_cloud_1 = __importDefault(require("@llamaindex/llama-cloud"));
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
- async function getInstructions() {
18
- // This API key is optional; providing it allows the server to fetch instructions for unreleased versions.
19
- const stainlessAPIKey = (0, exports.readEnv)('STAINLESS_API_KEY');
20
- const response = await fetch((0, exports.readEnv)('CODE_MODE_INSTRUCTIONS_URL') ?? 'https://api.stainless.com/api/ai/instructions/llamacloud-prod', {
21
- method: 'GET',
22
- headers: { ...(stainlessAPIKey && { Authorization: stainlessAPIKey }) },
23
- });
24
- let instructions;
25
- if (!response.ok) {
26
- console.warn('Warning: failed to retrieve MCP server instructions. Proceeding with default instructions...');
27
- instructions = `
28
- This is the llamacloud-prod MCP server. You will use Code Mode to help the user perform
29
- actions. You can use search_docs tool to learn about how to take action with this server. Then,
30
- you will write TypeScript code using the execute tool take action. It is CRITICAL that you be
31
- thoughtful and deliberate when executing code. Always try to entirely solve the problem in code
32
- block: it can be as long as you need to get the job done!
33
- `;
34
- }
35
- instructions ??= (await response.json()).instructions;
36
- instructions = `
37
- The current time in Unix timestamps is ${Date.now()}.
38
-
39
- ${instructions}
40
- `;
41
- return instructions;
42
- }
43
- const newMcpServer = async () => new mcp_js_1.McpServer({
18
+ const newMcpServer = async (stainlessApiKey) => new mcp_js_1.McpServer({
44
19
  name: 'llamaindex_llama_cloud_api',
45
- version: '1.5.0',
20
+ version: '1.7.0',
46
21
  }, {
47
- instructions: await getInstructions(),
22
+ instructions: await (0, instructions_1.getInstructions)(stainlessApiKey),
48
23
  capabilities: { tools: {}, logging: {} },
49
24
  });
50
25
  exports.newMcpServer = newMcpServer;
@@ -66,14 +41,33 @@ async function initMcpServer(params) {
66
41
  warn: logAtLevel('warning'),
67
42
  error: logAtLevel('error'),
68
43
  };
69
- let client = new llama_cloud_1.default({
70
- logger,
71
- ...params.clientOptions,
72
- defaultHeaders: {
73
- ...params.clientOptions?.defaultHeaders,
74
- 'X-Stainless-MCP': 'true',
75
- },
76
- });
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 llama_cloud_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
+ };
77
71
  const providedTools = selectTools(params.mcpOptions);
78
72
  const toolMap = Object.fromEntries(providedTools.map((mcpTool) => [mcpTool.tool.name, mcpTool]));
79
73
  server.setRequestHandler(types_js_1.ListToolsRequestSchema, async () => {
@@ -87,28 +81,55 @@ async function initMcpServer(params) {
87
81
  if (!mcpTool) {
88
82
  throw new Error(`Unknown tool: ${name}`);
89
83
  }
90
- return executeHandler(mcpTool.handler, client, args);
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
+ }
99
+ return executeHandler({
100
+ handler: mcpTool.handler,
101
+ reqContext: {
102
+ client,
103
+ stainlessApiKey: params.stainlessApiKey ?? params.mcpOptions?.stainlessApiKey,
104
+ },
105
+ args,
106
+ });
91
107
  });
92
108
  server.setRequestHandler(types_js_1.SetLevelRequestSchema, async (request) => {
93
109
  const { level } = request.params;
110
+ let logLevel;
94
111
  switch (level) {
95
112
  case 'debug':
96
- client = client.withOptions({ logLevel: 'debug' });
113
+ logLevel = 'debug';
97
114
  break;
98
115
  case 'info':
99
- client = client.withOptions({ logLevel: 'info' });
116
+ logLevel = 'info';
100
117
  break;
101
118
  case 'notice':
102
119
  case 'warning':
103
- client = client.withOptions({ logLevel: 'warn' });
120
+ logLevel = 'warn';
104
121
  break;
105
122
  case 'error':
106
- client = client.withOptions({ logLevel: 'error' });
123
+ logLevel = 'error';
107
124
  break;
108
125
  default:
109
- client = client.withOptions({ logLevel: 'off' });
126
+ logLevel = 'off';
110
127
  break;
111
128
  }
129
+ _logLevel = logLevel;
130
+ if (_client) {
131
+ _client = _client.withOptions({ logLevel });
132
+ }
112
133
  return {};
113
134
  });
114
135
  }
@@ -116,11 +137,13 @@ async function initMcpServer(params) {
116
137
  * Selects the tools to include in the MCP Server based on the provided options.
117
138
  */
118
139
  function selectTools(options) {
119
- const includedTools = [
120
- (0, code_tool_1.codeTool)({
140
+ const includedTools = [];
141
+ if (options?.includeCodeTool ?? true) {
142
+ includedTools.push((0, code_tool_1.codeTool)({
121
143
  blockedMethods: (0, methods_1.blockedMethodsForCodeTool)(options),
122
- }),
123
- ];
144
+ codeExecutionMode: options?.codeExecutionMode ?? 'stainless-sandbox',
145
+ }));
146
+ }
124
147
  if (options?.includeDocsTools ?? true) {
125
148
  includedTools.push(docs_search_tool_1.default);
126
149
  }
@@ -129,32 +152,7 @@ function selectTools(options) {
129
152
  /**
130
153
  * Runs the provided handler with the given client and arguments.
131
154
  */
132
- async function executeHandler(handler, client, args) {
133
- return await handler(client, args || {});
155
+ async function executeHandler({ handler, reqContext, args, }) {
156
+ return await handler({ reqContext, args: args || {} });
134
157
  }
135
- const readEnv = (env) => {
136
- if (typeof globalThis.process !== 'undefined') {
137
- return globalThis.process.env?.[env]?.trim();
138
- }
139
- else if (typeof globalThis.Deno !== 'undefined') {
140
- return globalThis.Deno.env?.get?.(env)?.trim();
141
- }
142
- return;
143
- };
144
- exports.readEnv = readEnv;
145
- const readEnvOrError = (env) => {
146
- let envValue = (0, exports.readEnv)(env);
147
- if (envValue === undefined) {
148
- throw new Error(`Environment variable ${env} is not set`);
149
- }
150
- return envValue;
151
- };
152
- exports.readEnvOrError = readEnvOrError;
153
- const requireValue = (value, description) => {
154
- if (value === undefined) {
155
- throw new Error(`Missing required value: ${description}`);
156
- }
157
- return value;
158
- };
159
- exports.requireValue = requireValue;
160
158
  //# sourceMappingURL=server.js.map
package/server.js.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"server.js","sourceRoot":"","sources":["src/server.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;;;;AAwEtF,sCAwEC;AAKD,kCAUC;AAKD,wCAMC;AAvKD,oEAAoE;AACpE,iEAI4C;AAE5C,0EAAiD;AACjD,8CAAuC;AACvC,6EAAgD;AAEhD,0CAAsD;AAMtD,KAAK,UAAU,eAAe;IAC5B,0GAA0G;IAC1G,MAAM,eAAe,GAAG,IAAA,eAAO,EAAC,mBAAmB,CAAC,CAAC;IACrD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,IAAA,eAAO,EAAC,4BAA4B,CAAC,IAAI,+DAA+D,EACxG;QACE,MAAM,EAAE,KAAK;QACb,OAAO,EAAE,EAAE,GAAG,CAAC,eAAe,IAAI,EAAE,aAAa,EAAE,eAAe,EAAE,CAAC,EAAE;KACxE,CACF,CAAC;IAEF,IAAI,YAAgC,CAAC;IACrC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,OAAO,CAAC,IAAI,CACV,8FAA8F,CAC/F,CAAC;QAEF,YAAY,GAAG;;;;;;KAMd,CAAC;IACJ,CAAC;IAED,YAAY,KAAM,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA8B,CAAC,YAAY,CAAC;IACpF,YAAY,GAAG;6CAC4B,IAAI,CAAC,GAAG,EAAE;;MAEjD,YAAY;GACf,CAAC;IAEF,OAAO,YAAY,CAAC;AACtB,CAAC;AAEM,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE,CACrC,IAAI,kBAAS,CACX;IACE,IAAI,EAAE,4BAA4B;IAClC,OAAO,EAAE,OAAO;CACjB,EACD;IACE,YAAY,EAAE,MAAM,eAAe,EAAE;IACrC,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,MAInC;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,MAAM,GAAG,IAAI,qBAAU,CAAC;QAC1B,MAAM;QACN,GAAG,MAAM,CAAC,aAAa;QACvB,cAAc,EAAE;YACd,GAAG,MAAM,CAAC,aAAa,EAAE,cAAc;YACvC,iBAAiB,EAAE,MAAM;SAC1B;KACF,CAAC,CAAC;IAEH,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,OAAO,cAAc,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;IACvD,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,QAAQ,KAAK,EAAE,CAAC;YACd,KAAK,OAAO;gBACV,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;gBACnD,MAAM;YACR,KAAK,MAAM;gBACT,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;gBAClD,MAAM;YACR,KAAK,QAAQ,CAAC;YACd,KAAK,SAAS;gBACZ,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;gBAClD,MAAM;YACR,KAAK,OAAO;gBACV,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;gBACnD,MAAM;YACR;gBACE,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;gBACjD,MAAM;QACV,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,CAClC,OAAwB,EACxB,MAAkB,EAClB,IAAyC;IAEzC,OAAO,MAAM,OAAO,CAAC,MAAM,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;AAC3C,CAAC;AAEM,MAAM,OAAO,GAAG,CAAC,GAAW,EAAsB,EAAE;IACzD,IAAI,OAAQ,UAAkB,CAAC,OAAO,KAAK,WAAW,EAAE,CAAC;QACvD,OAAQ,UAAkB,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC;IACxD,CAAC;SAAM,IAAI,OAAQ,UAAkB,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;QAC3D,OAAQ,UAAkB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC;IAC1D,CAAC;IACD,OAAO;AACT,CAAC,CAAC;AAPW,QAAA,OAAO,WAOlB;AAEK,MAAM,cAAc,GAAG,CAAC,GAAW,EAAU,EAAE;IACpD,IAAI,QAAQ,GAAG,IAAA,eAAO,EAAC,GAAG,CAAC,CAAC;IAC5B,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,wBAAwB,GAAG,aAAa,CAAC,CAAC;IAC5D,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC;AANW,QAAA,cAAc,kBAMzB;AAEK,MAAM,YAAY,GAAG,CAAI,KAAoB,EAAE,WAAmB,EAAK,EAAE;IAC9E,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CAAC,2BAA2B,WAAW,EAAE,CAAC,CAAC;IAC5D,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AALW,QAAA,YAAY,gBAKvB"}
1
+ {"version":3,"file":"server.js","sourceRoot":"","sources":["src/server.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;;;;AAkCtF,sCAsHC;AAKD,kCAeC;AAKD,wCAUC;AAxLD,oEAAoE;AACpE,iEAI4C;AAE5C,0EAAiD;AACjD,8CAAuC;AACvC,6EAAgD;AAChD,oDAAiD;AAEjD,0CAAsD;AAG/C,MAAM,YAAY,GAAG,KAAK,EAAE,eAAmC,EAAE,EAAE,CACxE,IAAI,kBAAS,CACX;IACE,IAAI,EAAE,4BAA4B;IAClC,OAAO,EAAE,OAAO;CACjB,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,OAA+B,CAAC;IACpC,IAAI,YAA+B,CAAC;IACpC,IAAI,SAAkE,CAAC;IAEvE,MAAM,SAAS,GAAG,GAAe,EAAE;QACjC,IAAI,YAAY;YAAE,MAAM,YAAY,CAAC;QACrC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,IAAI,CAAC;gBACH,OAAO,GAAG,IAAI,qBAAU,CAAC;oBACvB,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,MAAkB,CAAC;QACvB,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,EAAE,CAAC;IAEzB,IAAI,OAAO,EAAE,eAAe,IAAI,IAAI,EAAE,CAAC;QACrC,aAAa,CAAC,IAAI,CAChB,IAAA,oBAAQ,EAAC;YACP,cAAc,EAAE,IAAA,mCAAyB,EAAC,OAAO,CAAC;YAClD,iBAAiB,EAAE,OAAO,EAAE,iBAAiB,IAAI,mBAAmB;SACrE,CAAC,CACH,CAAC;IACJ,CAAC;IACD,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,38 +4,13 @@ import { CallToolRequestSchema, ListToolsRequestSchema, SetLevelRequestSchema, }
4
4
  import LlamaCloud from '@llamaindex/llama-cloud';
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
- async function getInstructions() {
9
- // This API key is optional; providing it allows the server to fetch instructions for unreleased versions.
10
- const stainlessAPIKey = readEnv('STAINLESS_API_KEY');
11
- const response = await fetch(readEnv('CODE_MODE_INSTRUCTIONS_URL') ?? 'https://api.stainless.com/api/ai/instructions/llamacloud-prod', {
12
- method: 'GET',
13
- headers: { ...(stainlessAPIKey && { Authorization: stainlessAPIKey }) },
14
- });
15
- let instructions;
16
- if (!response.ok) {
17
- console.warn('Warning: failed to retrieve MCP server instructions. Proceeding with default instructions...');
18
- instructions = `
19
- This is the llamacloud-prod MCP server. You will use Code Mode to help the user perform
20
- actions. You can use search_docs tool to learn about how to take action with this server. Then,
21
- you will write TypeScript code using the execute tool take action. It is CRITICAL that you be
22
- thoughtful and deliberate when executing code. Always try to entirely solve the problem in code
23
- block: it can be as long as you need to get the job done!
24
- `;
25
- }
26
- instructions ??= (await response.json()).instructions;
27
- instructions = `
28
- The current time in Unix timestamps is ${Date.now()}.
29
-
30
- ${instructions}
31
- `;
32
- return instructions;
33
- }
34
- export const newMcpServer = async () => new McpServer({
9
+ export const newMcpServer = async (stainlessApiKey) => new McpServer({
35
10
  name: 'llamaindex_llama_cloud_api',
36
- version: '1.5.0',
11
+ version: '1.7.0',
37
12
  }, {
38
- instructions: await getInstructions(),
13
+ instructions: await getInstructions(stainlessApiKey),
39
14
  capabilities: { tools: {}, logging: {} },
40
15
  });
41
16
  /**
@@ -56,14 +31,33 @@ export async function initMcpServer(params) {
56
31
  warn: logAtLevel('warning'),
57
32
  error: logAtLevel('error'),
58
33
  };
59
- let client = new LlamaCloud({
60
- logger,
61
- ...params.clientOptions,
62
- defaultHeaders: {
63
- ...params.clientOptions?.defaultHeaders,
64
- 'X-Stainless-MCP': 'true',
65
- },
66
- });
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 LlamaCloud({
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
+ };
67
61
  const providedTools = selectTools(params.mcpOptions);
68
62
  const toolMap = Object.fromEntries(providedTools.map((mcpTool) => [mcpTool.tool.name, mcpTool]));
69
63
  server.setRequestHandler(ListToolsRequestSchema, async () => {
@@ -77,28 +71,55 @@ export async function initMcpServer(params) {
77
71
  if (!mcpTool) {
78
72
  throw new Error(`Unknown tool: ${name}`);
79
73
  }
80
- return executeHandler(mcpTool.handler, client, args);
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
+ }
89
+ return executeHandler({
90
+ handler: mcpTool.handler,
91
+ reqContext: {
92
+ client,
93
+ stainlessApiKey: params.stainlessApiKey ?? params.mcpOptions?.stainlessApiKey,
94
+ },
95
+ args,
96
+ });
81
97
  });
82
98
  server.setRequestHandler(SetLevelRequestSchema, async (request) => {
83
99
  const { level } = request.params;
100
+ let logLevel;
84
101
  switch (level) {
85
102
  case 'debug':
86
- client = client.withOptions({ logLevel: 'debug' });
103
+ logLevel = 'debug';
87
104
  break;
88
105
  case 'info':
89
- client = client.withOptions({ logLevel: 'info' });
106
+ logLevel = 'info';
90
107
  break;
91
108
  case 'notice':
92
109
  case 'warning':
93
- client = client.withOptions({ logLevel: 'warn' });
110
+ logLevel = 'warn';
94
111
  break;
95
112
  case 'error':
96
- client = client.withOptions({ logLevel: 'error' });
113
+ logLevel = 'error';
97
114
  break;
98
115
  default:
99
- client = client.withOptions({ logLevel: 'off' });
116
+ logLevel = 'off';
100
117
  break;
101
118
  }
119
+ _logLevel = logLevel;
120
+ if (_client) {
121
+ _client = _client.withOptions({ logLevel });
122
+ }
102
123
  return {};
103
124
  });
104
125
  }
@@ -106,11 +127,13 @@ export async function initMcpServer(params) {
106
127
  * Selects the tools to include in the MCP Server based on the provided options.
107
128
  */
108
129
  export function selectTools(options) {
109
- const includedTools = [
110
- codeTool({
130
+ const includedTools = [];
131
+ if (options?.includeCodeTool ?? true) {
132
+ includedTools.push(codeTool({
111
133
  blockedMethods: blockedMethodsForCodeTool(options),
112
- }),
113
- ];
134
+ codeExecutionMode: options?.codeExecutionMode ?? 'stainless-sandbox',
135
+ }));
136
+ }
114
137
  if (options?.includeDocsTools ?? true) {
115
138
  includedTools.push(docsSearchTool);
116
139
  }
@@ -119,29 +142,7 @@ export function selectTools(options) {
119
142
  /**
120
143
  * Runs the provided handler with the given client and arguments.
121
144
  */
122
- export async function executeHandler(handler, client, args) {
123
- return await handler(client, args || {});
145
+ export async function executeHandler({ handler, reqContext, args, }) {
146
+ return await handler({ reqContext, args: args || {} });
124
147
  }
125
- export const readEnv = (env) => {
126
- if (typeof globalThis.process !== 'undefined') {
127
- return globalThis.process.env?.[env]?.trim();
128
- }
129
- else if (typeof globalThis.Deno !== 'undefined') {
130
- return globalThis.Deno.env?.get?.(env)?.trim();
131
- }
132
- return;
133
- };
134
- export const readEnvOrError = (env) => {
135
- let envValue = readEnv(env);
136
- if (envValue === undefined) {
137
- throw new Error(`Environment variable ${env} is not set`);
138
- }
139
- return envValue;
140
- };
141
- export const requireValue = (value, description) => {
142
- if (value === undefined) {
143
- throw new Error(`Missing required value: ${description}`);
144
- }
145
- return value;
146
- };
147
148
  //# sourceMappingURL=server.mjs.map