@llamaindex/llama-cloud-mcp 2.0.0 → 2.2.0

package/src/code-tool-worker.ts

@@ -7,6 +7,10 @@ import ts from 'typescript';
 import { WorkerOutput } from './code-tool-types';
 import { LlamaCloud, ClientOptions } from '@llamaindex/llama-cloud';
 
+async function tseval(code: string) {
+  return import('data:application/typescript;charset=utf-8;base64,' + Buffer.from(code).toString('base64'));
+}
+
 function getRunFunctionSource(code: string): {
   type: 'declaration' | 'expression';
   client: string | undefined;
@@ -125,6 +129,11 @@ const fuse = new Fuse(
     'client.classify.create',
     'client.classify.get',
     'client.classify.list',
+    'client.configurations.create',
+    'client.configurations.delete',
+    'client.configurations.list',
+    'client.configurations.retrieve',
+    'client.configurations.update',
     'client.projects.get',
     'client.projects.list',
     'client.dataSinks.create',
@@ -349,7 +358,9 @@ const fetch = async (req: Request): Promise<Response> => {
 
   const log_lines: string[] = [];
   const err_lines: string[] = [];
-  const console = {
+  const originalConsole = globalThis.console;
+  globalThis.console = {
+    ...originalConsole,
     log: (...args: unknown[]) => {
       log_lines.push(util.format(...args));
     },
@@ -359,7 +370,7 @@ const fetch = async (req: Request): Promise<Response> => {
   };
   try {
     let run_ = async (client: any) => {};
-    eval(`${code}\nrun_ = run;`);
+    run_ = (await tseval(`${code}\nexport default run;`)).default;
     const result = await run_(makeSdkProxy(client, { path: ['client'] }));
     return Response.json({
       is_error: false,
@@ -377,6 +388,8 @@ const fetch = async (req: Request): Promise<Response> => {
       } satisfies WorkerOutput,
       { status: 400, statusText: 'Code execution error' },
     );
+  } finally {
+    globalThis.console = originalConsole;
   }
 };
 

package/src/http.ts

@@ -69,6 +69,11 @@ const newServer = async ({
     }
   }
 
+  const mcpClientInfo =
+    typeof req.body?.params?.clientInfo?.name === 'string' ?
+      { name: req.body.params.clientInfo.name, version: String(req.body.params.clientInfo.version ?? '') }
+    : undefined;
+
   await initMcpServer({
     server: server,
     mcpOptions: effectiveMcpOptions,
@@ -79,12 +84,13 @@ const newServer = async ({
     stainlessApiKey: stainlessApiKey,
     upstreamClientEnvs,
     mcpSessionId: (req as any).mcpSessionId,
-    mcpClientInfo:
-      typeof req.body?.params?.clientInfo?.name === 'string' ?
-        { name: req.body.params.clientInfo.name, version: String(req.body.params.clientInfo.version ?? '') }
-      : undefined,
+    mcpClientInfo,
   });
 
+  if (mcpClientInfo) {
+    getLogger().info({ mcpSessionId: (req as any).mcpSessionId, mcpClientInfo }, 'MCP client connected');
+  }
+
   return server;
 };
 
@@ -154,6 +160,9 @@ export const streamableHTTPApp = ({
   app.use(
     pinoHttp({
       logger: getLogger(),
+      customProps: (req) => ({
+        mcpSessionId: (req as any).mcpSessionId,
+      }),
       customLogLevel: (req, res) => {
         if (res.statusCode >= 500) {
           return 'error';