@nirvana-labs/nirvana-mcp 1.20.2 → 1.21.0

package/src/code-tool-worker.ts

@@ -1,11 +1,264 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 import util from 'node:util';
+
+import Fuse from 'fuse.js';
+import ts from 'typescript';
+
 import { WorkerInput, WorkerSuccess, WorkerError } from './code-tool-types';
 import { NirvanaLabs } from '@nirvana-labs/nirvana';
 
+function getRunFunctionNode(
+  code: string,
+): ts.FunctionDeclaration | ts.FunctionExpression | ts.ArrowFunction | null {
+  const sourceFile = ts.createSourceFile('code.ts', code, ts.ScriptTarget.Latest, true);
+
+  for (const statement of sourceFile.statements) {
+    // Check for top-level function declarations
+    if (ts.isFunctionDeclaration(statement)) {
+      if (statement.name?.text === 'run') {
+        return statement;
+      }
+    }
+
+    // Check for variable declarations: const run = () => {} or const run = function() {}
+    if (ts.isVariableStatement(statement)) {
+      for (const declaration of statement.declarationList.declarations) {
+        if (ts.isIdentifier(declaration.name) && declaration.name.text === 'run') {
+          // Check if it's initialized with a function
+          if (
+            declaration.initializer &&
+            (ts.isFunctionExpression(declaration.initializer) || ts.isArrowFunction(declaration.initializer))
+          ) {
+            return declaration.initializer;
+          }
+        }
+      }
+    }
+  }
+
+  return null;
+}
+
+const fuse = new Fuse(
+  [
+    'client.user.get',
+    'client.apiKeys.create',
+    'client.apiKeys.delete',
+    'client.apiKeys.get',
+    'client.apiKeys.list',
+    'client.apiKeys.update',
+    'client.operations.get',
+    'client.operations.list',
+    'client.compute.vms.create',
+    'client.compute.vms.delete',
+    'client.compute.vms.get',
+    'client.compute.vms.list',
+    'client.compute.vms.restart',
+    'client.compute.vms.update',
+    'client.compute.vms.availability.create',
+    'client.compute.vms.availability.update',
+    'client.compute.vms.volumes.list',
+    'client.compute.vms.osImages.list',
+    'client.compute.volumes.create',
+    'client.compute.volumes.delete',
+    'client.compute.volumes.get',
+    'client.compute.volumes.list',
+    'client.compute.volumes.update',
+    'client.compute.volumes.availability.create',
+    'client.compute.volumes.availability.update',
+    'client.networking.vpcs.create',
+    'client.networking.vpcs.delete',
+    'client.networking.vpcs.get',
+    'client.networking.vpcs.list',
+    'client.networking.vpcs.update',
+    'client.networking.vpcs.availability.create',
+    'client.networking.vpcs.availability.update',
+    'client.networking.firewallRules.create',
+    'client.networking.firewallRules.delete',
+    'client.networking.firewallRules.get',
+    'client.networking.firewallRules.list',
+    'client.networking.firewallRules.update',
+    'client.networking.connect.connections.create',
+    'client.networking.connect.connections.delete',
+    'client.networking.connect.connections.get',
+    'client.networking.connect.connections.list',
+    'client.networking.connect.connections.update',
+    'client.networking.connect.routes.list',
+    'client.rpcNodes.flex.create',
+    'client.rpcNodes.flex.delete',
+    'client.rpcNodes.flex.get',
+    'client.rpcNodes.flex.list',
+    'client.rpcNodes.flex.update',
+    'client.rpcNodes.flex.blockchains.list',
+    'client.rpcNodes.dedicated.get',
+    'client.rpcNodes.dedicated.list',
+    'client.rpcNodes.dedicated.blockchains.list',
+    'client.vektor.registry.assets.list',
+    'client.vektor.registry.blockchains.list',
+    'client.vektor.registry.venues.list',
+    'client.vektor.registry.errors.list',
+    'client.vektor.registry.lendMarkets.list',
+    'client.vektor.registry.borrowMarkets.list',
+    'client.vektor.registry.lpPools.list',
+    'client.vektor.balances.list',
+    'client.vektor.balances.listHistorical',
+    'client.vektor.prices.list',
+    'client.vektor.prices.listHistorical',
+    'client.vektor.lend.markets.list',
+    'client.vektor.lend.markets.listHistorical',
+    'client.vektor.lend.positions.list',
+    'client.vektor.lend.positions.listHistorical',
+    'client.vektor.lend.lend.create',
+    'client.vektor.lend.withdraw.create',
+    'client.vektor.lend.setCollateral.create',
+    'client.vektor.borrow.markets.list',
+    'client.vektor.borrow.markets.listHistorical',
+    'client.vektor.borrow.positions.list',
+    'client.vektor.borrow.positions.listHistorical',
+    'client.vektor.borrow.accounts.list',
+    'client.vektor.borrow.accounts.listHistorical',
+    'client.vektor.borrow.borrow.create',
+    'client.vektor.borrow.repay.create',
+    'client.vektor.lp.pools.list',
+    'client.vektor.lp.pools.listHistorical',
+    'client.vektor.lp.positions.list',
+    'client.vektor.lp.positions.listHistorical',
+    'client.vektor.lp.depositQuote.create',
+    'client.vektor.lp.withdrawQuote.create',
+    'client.vektor.buy.quotes.list',
+    'client.vektor.buy.buy.create',
+    'client.vektor.sell.quotes.list',
+    'client.vektor.sell.sell.create',
+    'client.vektor.move.create',
+    'client.vektor.wrap.wrap.create',
+    'client.vektor.wrap.unwrap.create',
+    'client.vektor.bridge.quotes.list',
+    'client.vektor.lock.markets.list',
+    'client.vektor.lock.positions.list',
+    'client.vektor.vote.markets.list',
+    'client.vektor.vote.rewards.list',
+    'client.vektor.incentivize.list',
+    'client.vektor.executions.get',
+    'client.vektor.executions.list',
+    'client.vektor.executions.steps.get',
+    'client.vektor.executions.steps.sign',
+  ],
+  { threshold: 1, shouldSort: true },
+);
+
+function getMethodSuggestions(fullyQualifiedMethodName: string): string[] {
+  return fuse
+    .search(fullyQualifiedMethodName)
+    .map(({ item }) => item)
+    .slice(0, 5);
+}
+
+const proxyToObj = new WeakMap<any, any>();
+const objToProxy = new WeakMap<any, any>();
+
+type ClientProxyConfig = {
+  path: string[];
+  isBelievedBad?: boolean;
+};
+
+function makeSdkProxy<T extends object>(obj: T, { path, isBelievedBad = false }: ClientProxyConfig): T {
+  let proxy: T = objToProxy.get(obj);
+
+  if (!proxy) {
+    proxy = new Proxy(obj, {
+      get(target, prop, receiver) {
+        const propPath = [...path, String(prop)];
+        const value = Reflect.get(target, prop, receiver);
+
+        if (isBelievedBad || (!(prop in target) && value === undefined)) {
+          // If we're accessing a path that doesn't exist, it will probably eventually error.
+          // Let's proxy it and mark it bad so that we can control the error message.
+          // We proxy an empty class so that an invocation or construction attempt is possible.
+          return makeSdkProxy(class {}, { path: propPath, isBelievedBad: true });
+        }
+
+        if (value !== null && (typeof value === 'object' || typeof value === 'function')) {
+          return makeSdkProxy(value, { path: propPath, isBelievedBad });
+        }
+
+        return value;
+      },
+
+      apply(target, thisArg, args) {
+        if (isBelievedBad || typeof target !== 'function') {
+          const fullyQualifiedMethodName = path.join('.');
+          const suggestions = getMethodSuggestions(fullyQualifiedMethodName);
+          throw new Error(
+            `${fullyQualifiedMethodName} is not a function. Did you mean: ${suggestions.join(', ')}`,
+          );
+        }
+
+        return Reflect.apply(target, proxyToObj.get(thisArg) ?? thisArg, args);
+      },
+
+      construct(target, args, newTarget) {
+        if (isBelievedBad || typeof target !== 'function') {
+          const fullyQualifiedMethodName = path.join('.');
+          const suggestions = getMethodSuggestions(fullyQualifiedMethodName);
+          throw new Error(
+            `${fullyQualifiedMethodName} is not a constructor. Did you mean: ${suggestions.join(', ')}`,
+          );
+        }
+
+        return Reflect.construct(target, args, newTarget);
+      },
+    });
+
+    objToProxy.set(obj, proxy);
+    proxyToObj.set(proxy, obj);
+  }
+
+  return proxy;
+}
+
+function parseError(code: string, error: unknown): string | undefined {
+  if (!(error instanceof Error)) return;
+  const message = error.name ? `${error.name}: ${error.message}` : error.message;
+  try {
+    // Deno uses V8; the first "<anonymous>:LINE:COLUMN" is the top of stack.
+    const lineNumber = error.stack?.match(/<anonymous>:([0-9]+):[0-9]+/)?.[1];
+    // -1 for the zero-based indexing
+    const line =
+      lineNumber &&
+      code
+        .split('\n')
+        .at(parseInt(lineNumber, 10) - 1)
+        ?.trim();
+    return line ? `${message}\n  at line ${lineNumber}\n    ${line}` : message;
+  } catch {
+    return message;
+  }
+}
+
 const fetch = async (req: Request): Promise<Response> => {
   const { opts, code } = (await req.json()) as WorkerInput;
+  if (code == null) {
+    return Response.json(
+      {
+        message:
+          'The code param is missing. Provide one containing a top-level `run` function. Write code within this template:\n\n```\nasync function run(client) {\n  // Fill this out\n}\n```',
+      } satisfies WorkerError,
+      { status: 400, statusText: 'Code execution error' },
+    );
+  }
+
+  const runFunctionNode = getRunFunctionNode(code);
+  if (!runFunctionNode) {
+    return Response.json(
+      {
+        message:
+          'The code is missing a top-level `run` function. Write code within this template:\n\n```\nasync function run(client) {\n  // Fill this out\n}\n```',
+      } satisfies WorkerError,
+      { status: 400, statusText: 'Code execution error' },
+    );
+  }
+
   const client = new NirvanaLabs({
     ...opts,
   });
@@ -22,21 +275,17 @@ const fetch = async (req: Request): Promise<Response> => {
   };
   try {
     let run_ = async (client: any) => {};
-    eval(`
-      ${code}
-      run_ = run;
-    `);
-    const result = await run_(client);
+    eval(`${code}\nrun_ = run;`);
+    const result = await run_(makeSdkProxy(client, { path: ['client'] }));
     return Response.json({
       result,
       logLines,
       errLines,
     } satisfies WorkerSuccess);
   } catch (e) {
-    const message = e instanceof Error ? e.message : undefined;
     return Response.json(
       {
-        message,
+        message: parseError(code, e),
       } satisfies WorkerError,
       { status: 400, statusText: 'Code execution error' },
     );

package/src/code-tool.ts

@@ -3,7 +3,7 @@
 import { dirname } from 'node:path';
 import { pathToFileURL } from 'node:url';
 import NirvanaLabs, { ClientOptions } from '@nirvana-labs/nirvana';
-import { Endpoint, ContentBlock, Metadata } from './tools/types';
+import { ContentBlock, Endpoint, Metadata, ToolCallResult } from './tools/types';
 
 import { Tool } from '@modelcontextprotocol/sdk/types.js';
 
@@ -12,7 +12,7 @@ import { WorkerInput, WorkerError, WorkerSuccess } from './code-tool-types';
 /**
  * A tool that runs code against a copy of the SDK.
  *
- * Instead of exposing every endpoint as it's own tool, which uses up too many tokens for LLMs to use at once,
+ * Instead of exposing every endpoint as its own tool, which uses up too many tokens for LLMs to use at once,
  * we expose a single tool that can be used to search for endpoints by name, resource, operation, or tag, and then
  * a generic endpoint that can be used to invoke any endpoint with the provided arguments.
  *
@@ -23,7 +23,7 @@ export async function codeTool(): Promise<Endpoint> {
   const tool: Tool = {
     name: 'execute',
     description:
-      'Runs Typescript code to interact with the API.\nYou are a skilled programmer writing code to interface with the service.\nDefine an async function named "run" that takes a single parameter of an initialized client, and it will be run.\nDo not initialize a client, but instead use the client that you are given as a parameter.\nYou will be returned anything that your function returns, plus the results of any console.log statements.\nIf any code triggers an error, the tool will return an error response, so you do not need to add error handling unless you want to output something more helpful than the raw error.\nIt is not necessary to add comments to code, unless by adding those comments you believe that you can generate better code.\nThis code will run in a container, and you will not be able to use fetch or otherwise interact with the network calls other than through the client you are given.\nAny variables you define won\'t live between successive uses of this call, so make sure to return or log any data you might need later.',
+      'Runs TypeScript code to interact with the API.\n\nYou are a skilled programmer writing code to interface with the service.\nDefine an async function named "run" that takes a single parameter of an initialized client named "client", and it will be run.\nWrite code within this template:\n\n```\nasync function run(client) {\n  // Fill this out\n}\n```\n\nYou will be returned anything that your function returns, plus the results of any console.log statements.\nIf any code triggers an error, the tool will return an error response, so you do not need to add error handling unless you want to output something more helpful than the raw error.\nIt is not necessary to add comments to code, unless by adding those comments you believe that you can generate better code.\nThis code will run in a container, and you will not be able to use fetch or otherwise interact with the network calls other than through the client you are given.\nAny variables you define won\'t live between successive uses of this call, so make sure to return or log any data you might need later.',
     inputSchema: { type: 'object', properties: { code: { type: 'string' } } },
   };
 
@@ -31,7 +31,7 @@ export async function codeTool(): Promise<Endpoint> {
   const { newDenoHTTPWorker } = await import('@valtown/deno-http-worker');
   const { workerPath } = await import('./code-tool-paths.cjs');
 
-  const handler = async (client: NirvanaLabs, args: unknown) => {
+  const handler = async (client: NirvanaLabs, args: unknown): Promise<ToolCallResult> => {
     const baseURLHostname = new URL(client.baseURL).hostname;
     const { code } = args as { code: string };
 
@@ -97,7 +97,7 @@ export async function codeTool(): Promise<Endpoint> {
         } satisfies WorkerInput);
 
         req.write(body, (err) => {
-          if (err !== null && err !== undefined) {
+          if (err != null) {
             reject(err);
           }
         });
@@ -108,12 +108,12 @@ export async function codeTool(): Promise<Endpoint> {
       if (resp.status === 200) {
         const { result, logLines, errLines } = (await resp.json()) as WorkerSuccess;
         const returnOutput: ContentBlock | null =
-          result === null ? null
-          : result === undefined ? null
-          : {
+          result == null ? null : (
+            {
               type: 'text',
-              text: typeof result === 'string' ? (result as string) : JSON.stringify(result),
-            };
+              text: typeof result === 'string' ? result : JSON.stringify(result),
+            }
+          );
         const logOutput: ContentBlock | null =
           logLines.length === 0 ?
             null
@@ -133,10 +133,11 @@ export async function codeTool(): Promise<Endpoint> {
         };
       } else {
         const { message } = (await resp.json()) as WorkerError;
-        throw new Error(message);
+        return {
+          content: message == null ? [] : [{ type: 'text', text: message }],
+          isError: true,
+        };
       }
-    } catch (e) {
-      throw e;
     } finally {
       worker.terminate();
     }

package/src/dynamic-tools.ts

@@ -14,7 +14,7 @@ function zodToInputSchema(schema: z.ZodSchema) {
 /**
  * A list of tools that expose all the endpoints in the API dynamically.
  *
- * Instead of exposing every endpoint as it's own tool, which uses up too many tokens for LLMs to use at once,
+ * Instead of exposing every endpoint as its own tool, which uses up too many tokens for LLMs to use at once,
  * we expose a single tool that can be used to search for endpoints by name, resource, operation, or tag, and then
  * a generic endpoint that can be used to invoke any endpoint with the provided arguments.
  *

package/src/options.ts

@@ -284,8 +284,10 @@ const coerceArray = <T extends z.ZodTypeAny>(zodType: T) =>
   );
 
 const QueryOptions = z.object({
-  tools: coerceArray(z.enum(['dynamic', 'all', 'docs'])).describe('Use dynamic tools or all tools'),
-  no_tools: coerceArray(z.enum(['dynamic', 'all', 'docs'])).describe('Do not use dynamic tools or all tools'),
+  tools: coerceArray(z.enum(['dynamic', 'all', 'code', 'docs'])).describe('Specify which MCP tools to use'),
+  no_tools: coerceArray(z.enum(['dynamic', 'all', 'code', 'docs'])).describe(
+    'Specify which MCP tools to not use.',
+  ),
   tool: coerceArray(z.string()).describe('Include tools matching the specified names'),
   resource: coerceArray(z.string()).describe('Include tools matching the specified resources'),
   operation: coerceArray(z.enum(['read', 'write'])).describe(
@@ -385,11 +387,16 @@ export function parseQueryOptions(defaultOptions: McpOptions, query: unknown): M
     : queryOptions.tools?.includes('docs') ? true
     : defaultOptions.includeDocsTools;
 
+  let codeTools: boolean | undefined =
+    queryOptions.no_tools && queryOptions.no_tools?.includes('code') ? false
+    : queryOptions.tools?.includes('code') && defaultOptions.includeCodeTools ? true
+    : defaultOptions.includeCodeTools;
+
   return {
     client: queryOptions.client ?? defaultOptions.client,
     includeDynamicTools: dynamicTools,
     includeAllTools: allTools,
-    includeCodeTools: undefined,
+    includeCodeTools: codeTools,
     includeDocsTools: docsTools,
     filters,
     capabilities: clientCapabilities,

package/src/server.ts

@@ -34,7 +34,7 @@ export const newMcpServer = () =>
   new McpServer(
     {
       name: 'nirvana_labs_nirvana_api',
-      version: '1.20.2',
+      version: '1.21.0',
     },
     { capabilities: { tools: {}, logging: {} } },
   );

package/src/tools/api-keys/list-api-keys.ts

@@ -18,10 +18,18 @@ export const metadata: Metadata = {
 export const tool: Tool = {
   name: 'list_api_keys',
   description:
-    "When using this tool, always use the `jq_filter` parameter to reduce the response size and improve performance.\n\nOnly omit if you're sure you don't need the data.\n\nList all API keys for the authenticated user\n\n# Response Schema\n```json\n{\n  $ref: '#/$defs/api_key_list',\n  $defs: {\n    api_key_list: {\n      type: 'object',\n      properties: {\n        items: {\n          type: 'array',\n          items: {\n            $ref: '#/$defs/api_key'\n          }\n        }\n      },\n      required: [        'items'\n      ]\n    },\n    api_key: {\n      type: 'object',\n      description: 'API Key response.',\n      properties: {\n        id: {\n          type: 'string',\n          description: 'API Key ID.'\n        },\n        created_at: {\n          type: 'string',\n          description: 'When the API Key was created.',\n          format: 'date-time'\n        },\n        expires_at: {\n          type: 'string',\n          description: 'When the API Key expires and is no longer valid.',\n          format: 'date-time'\n        },\n        name: {\n          type: 'string',\n          description: 'API Key name.'\n        },\n        status: {\n          type: 'string',\n          description: 'Status of the API Key.',\n          enum: [            'active',\n            'inactive',\n            'expired'\n          ]\n        },\n        tags: {\n          type: 'array',\n          description: 'Tags to attach to the API Key.',\n          items: {\n            type: 'string'\n          }\n        },\n        updated_at: {\n          type: 'string',\n          description: 'When the API Key was updated.',\n          format: 'date-time'\n        },\n        key: {\n          type: 'string',\n          description: 'API Key. Only returned on creation.'\n        },\n        starts_at: {\n          type: 'string',\n          description: 'When the API Key starts to be valid.',\n          format: 'date-time'\n        }\n      },\n      required: [        'id',\n        'created_at',\n        'expires_at',\n        'name',\n        'status',\n        'tags',\n        'updated_at'\n      ]\n    }\n  }\n}\n```",
+    "When using this tool, always use the `jq_filter` parameter to reduce the response size and improve performance.\n\nOnly omit if you're sure you don't need the data.\n\nList all API keys for the authenticated user\n\n# Response Schema\n```json\n{\n  $ref: '#/$defs/api_key_list',\n  $defs: {\n    api_key_list: {\n      type: 'object',\n      properties: {\n        items: {\n          type: 'array',\n          items: {\n            $ref: '#/$defs/api_key'\n          }\n        },\n        pagination: {\n          $ref: '#/$defs/pagination'\n        }\n      },\n      required: [        'items',\n        'pagination'\n      ]\n    },\n    api_key: {\n      type: 'object',\n      description: 'API Key response.',\n      properties: {\n        id: {\n          type: 'string',\n          description: 'API Key ID.'\n        },\n        created_at: {\n          type: 'string',\n          description: 'When the API Key was created.',\n          format: 'date-time'\n        },\n        expires_at: {\n          type: 'string',\n          description: 'When the API Key expires and is no longer valid.',\n          format: 'date-time'\n        },\n        name: {\n          type: 'string',\n          description: 'API Key name.'\n        },\n        status: {\n          type: 'string',\n          description: 'Status of the API Key.',\n          enum: [            'active',\n            'inactive',\n            'expired'\n          ]\n        },\n        tags: {\n          type: 'array',\n          description: 'Tags to attach to the API Key.',\n          items: {\n            type: 'string'\n          }\n        },\n        updated_at: {\n          type: 'string',\n          description: 'When the API Key was updated.',\n          format: 'date-time'\n        },\n        key: {\n          type: 'string',\n          description: 'API Key. Only returned on creation.'\n        },\n        starts_at: {\n          type: 'string',\n          description: 'When the API Key starts to be valid.',\n          format: 'date-time'\n        }\n      },\n      required: [        'id',\n        'created_at',\n        'expires_at',\n        'name',\n        'status',\n        'tags',\n        'updated_at'\n      ]\n    },\n    pagination: {\n      type: 'object',\n      description: 'Pagination response details.',\n      properties: {\n        next_cursor: {\n          type: 'string'\n        },\n        previous_cursor: {\n          type: 'string'\n        },\n        total_count: {\n          type: 'integer'\n        }\n      },\n      required: [        'next_cursor',\n        'previous_cursor',\n        'total_count'\n      ]\n    }\n  }\n}\n```",
   inputSchema: {
     type: 'object',
     properties: {
+      cursor: {
+        type: 'string',
+        description: 'Pagination cursor returned by a previous request',
+      },
+      limit: {
+        type: 'integer',
+        description: 'Maximum number of items to return',
+      },
       jq_filter: {
         type: 'string',
         title: 'jq Filter',
@@ -37,8 +45,9 @@ export const tool: Tool = {
 };
 
 export const handler = async (client: NirvanaLabs, args: Record<string, unknown> | undefined) => {
-  const { jq_filter } = args as any;
-  return asTextContentResult(await maybeFilter(jq_filter, await client.apiKeys.list()));
+  const { jq_filter, ...body } = args as any;
+  const response = await client.apiKeys.list(body).asResponse();
+  return asTextContentResult(await maybeFilter(jq_filter, await response.json()));
 };
 
 export default { metadata, tool, handler };

package/src/tools/compute/vms/list-compute-vms.ts

@@ -18,10 +18,18 @@ export const metadata: Metadata = {
 export const tool: Tool = {
   name: 'list_compute_vms',
   description:
-    "When using this tool, always use the `jq_filter` parameter to reduce the response size and improve performance.\n\nOnly omit if you're sure you don't need the data.\n\nList all VMs\n\n# Response Schema\n```json\n{\n  $ref: '#/$defs/vm_list',\n  $defs: {\n    vm_list: {\n      type: 'object',\n      properties: {\n        items: {\n          type: 'array',\n          items: {\n            $ref: '#/$defs/vm'\n          }\n        }\n      },\n      required: [        'items'\n      ]\n    },\n    vm: {\n      type: 'object',\n      description: 'VM details.',\n      properties: {\n        id: {\n          type: 'string',\n          description: 'Unique identifier for the VM.'\n        },\n        boot_volume_id: {\n          type: 'string',\n          description: 'ID of the boot volume attached to the VM.'\n        },\n        cpu_config: {\n          $ref: '#/$defs/cpu_config'\n        },\n        created_at: {\n          type: 'string',\n          description: 'When the VM was created.',\n          format: 'date-time'\n        },\n        data_volume_ids: {\n          type: 'array',\n          description: 'IDs of the data volumes attached to the VM.',\n          items: {\n            type: 'string'\n          }\n        },\n        memory_config: {\n          $ref: '#/$defs/memory_config'\n        },\n        name: {\n          type: 'string',\n          description: 'Name of the VM.'\n        },\n        private_ip: {\n          type: 'string',\n          description: 'Private IP of the VM.'\n        },\n        public_ip: {\n          type: 'string',\n          description: 'Public IP of the VM.'\n        },\n        public_ip_enabled: {\n          type: 'boolean',\n          description: 'Whether the public IP is enabled for the VM.'\n        },\n        region: {\n          $ref: '#/$defs/region_name'\n        },\n        status: {\n          $ref: '#/$defs/resource_status'\n        },\n        subnet_id: {\n          type: 'string',\n          description: 'ID of the subnet the VM is in.'\n        },\n        tags: {\n          type: 'array',\n          description: 'Tags to attach to the VM.',\n          items: {\n            type: 'string'\n          }\n        },\n        updated_at: {\n          type: 'string',\n          description: 'When the VM was updated.',\n          format: 'date-time'\n        },\n        vpc_id: {\n          type: 'string',\n          description: 'ID of the VPC the VM is in.'\n        },\n        vpc_name: {\n          type: 'string',\n          description: 'Name of the VPC the VM is in.'\n        }\n      },\n      required: [        'id',\n        'boot_volume_id',\n        'cpu_config',\n        'created_at',\n        'data_volume_ids',\n        'memory_config',\n        'name',\n        'private_ip',\n        'public_ip',\n        'public_ip_enabled',\n        'region',\n        'status',\n        'subnet_id',\n        'tags',\n        'updated_at',\n        'vpc_id',\n        'vpc_name'\n      ]\n    },\n    cpu_config: {\n      type: 'object',\n      description: 'CPU configuration for the VM.',\n      properties: {\n        vcpu: {\n          type: 'integer',\n          description: 'Number of virtual CPUs.'\n        }\n      },\n      required: [        'vcpu'\n      ]\n    },\n    memory_config: {\n      type: 'object',\n      description: 'Memory configuration for the VM.',\n      properties: {\n        size: {\n          type: 'integer',\n          description: 'Size of the memory in GB.'\n        }\n      },\n      required: [        'size'\n      ]\n    },\n    region_name: {\n      type: 'string',\n      description: 'Region the resource is in.',\n      enum: [        'us-sea-1',\n        'us-sva-1',\n        'us-chi-1',\n        'us-wdc-1',\n        'eu-frk-1',\n        'ap-sin-1',\n        'ap-seo-1',\n        'ap-tyo-1'\n      ]\n    },\n    resource_status: {\n      type: 'string',\n      description: 'Status of the resource.',\n      enum: [        'pending',\n        'creating',\n        'updating',\n        'ready',\n        'deleting',\n        'deleted',\n        'error'\n      ]\n    }\n  }\n}\n```",
+    "When using this tool, always use the `jq_filter` parameter to reduce the response size and improve performance.\n\nOnly omit if you're sure you don't need the data.\n\nList all VMs\n\n# Response Schema\n```json\n{\n  $ref: '#/$defs/vm_list',\n  $defs: {\n    vm_list: {\n      type: 'object',\n      properties: {\n        items: {\n          type: 'array',\n          items: {\n            $ref: '#/$defs/vm'\n          }\n        },\n        pagination: {\n          $ref: '#/$defs/pagination'\n        }\n      },\n      required: [        'items',\n        'pagination'\n      ]\n    },\n    vm: {\n      type: 'object',\n      description: 'VM details.',\n      properties: {\n        id: {\n          type: 'string',\n          description: 'Unique identifier for the VM.'\n        },\n        boot_volume_id: {\n          type: 'string',\n          description: 'ID of the boot volume attached to the VM.'\n        },\n        cpu_config: {\n          $ref: '#/$defs/cpu_config'\n        },\n        created_at: {\n          type: 'string',\n          description: 'When the VM was created.',\n          format: 'date-time'\n        },\n        data_volume_ids: {\n          type: 'array',\n          description: 'IDs of the data volumes attached to the VM.',\n          items: {\n            type: 'string'\n          }\n        },\n        memory_config: {\n          $ref: '#/$defs/memory_config'\n        },\n        name: {\n          type: 'string',\n          description: 'Name of the VM.'\n        },\n        private_ip: {\n          type: 'string',\n          description: 'Private IP of the VM.'\n        },\n        public_ip: {\n          type: 'string',\n          description: 'Public IP of the VM.'\n        },\n        public_ip_enabled: {\n          type: 'boolean',\n          description: 'Whether the public IP is enabled for the VM.'\n        },\n        region: {\n          $ref: '#/$defs/region_name'\n        },\n        status: {\n          $ref: '#/$defs/resource_status'\n        },\n        subnet_id: {\n          type: 'string',\n          description: 'ID of the subnet the VM is in.'\n        },\n        tags: {\n          type: 'array',\n          description: 'Tags to attach to the VM.',\n          items: {\n            type: 'string'\n          }\n        },\n        updated_at: {\n          type: 'string',\n          description: 'When the VM was updated.',\n          format: 'date-time'\n        },\n        vpc_id: {\n          type: 'string',\n          description: 'ID of the VPC the VM is in.'\n        },\n        vpc_name: {\n          type: 'string',\n          description: 'Name of the VPC the VM is in.'\n        }\n      },\n      required: [        'id',\n        'boot_volume_id',\n        'cpu_config',\n        'created_at',\n        'data_volume_ids',\n        'memory_config',\n        'name',\n        'private_ip',\n        'public_ip',\n        'public_ip_enabled',\n        'region',\n        'status',\n        'subnet_id',\n        'tags',\n        'updated_at',\n        'vpc_id',\n        'vpc_name'\n      ]\n    },\n    cpu_config: {\n      type: 'object',\n      description: 'CPU configuration for the VM.',\n      properties: {\n        vcpu: {\n          type: 'integer',\n          description: 'Number of virtual CPUs.'\n        }\n      },\n      required: [        'vcpu'\n      ]\n    },\n    memory_config: {\n      type: 'object',\n      description: 'Memory configuration for the VM.',\n      properties: {\n        size: {\n          type: 'integer',\n          description: 'Size of the memory in GB.'\n        }\n      },\n      required: [        'size'\n      ]\n    },\n    region_name: {\n      type: 'string',\n      description: 'Region the resource is in.',\n      enum: [        'us-sea-1',\n        'us-sva-1',\n        'us-chi-1',\n        'us-wdc-1',\n        'eu-frk-1',\n        'ap-sin-1',\n        'ap-seo-1',\n        'ap-tyo-1'\n      ]\n    },\n    resource_status: {\n      type: 'string',\n      description: 'Status of the resource.',\n      enum: [        'pending',\n        'creating',\n        'updating',\n        'ready',\n        'deleting',\n        'deleted',\n        'error'\n      ]\n    },\n    pagination: {\n      type: 'object',\n      description: 'Pagination response details.',\n      properties: {\n        next_cursor: {\n          type: 'string'\n        },\n        previous_cursor: {\n          type: 'string'\n        },\n        total_count: {\n          type: 'integer'\n        }\n      },\n      required: [        'next_cursor',\n        'previous_cursor',\n        'total_count'\n      ]\n    }\n  }\n}\n```",
   inputSchema: {
     type: 'object',
     properties: {
+      cursor: {
+        type: 'string',
+        description: 'Pagination cursor returned by a previous request',
+      },
+      limit: {
+        type: 'integer',
+        description: 'Maximum number of items to return',
+      },
       jq_filter: {
         type: 'string',
         title: 'jq Filter',
@@ -37,8 +45,9 @@ export const tool: Tool = {
 };
 
 export const handler = async (client: NirvanaLabs, args: Record<string, unknown> | undefined) => {
-  const { jq_filter } = args as any;
-  return asTextContentResult(await maybeFilter(jq_filter, await client.compute.vms.list()));
+  const { jq_filter, ...body } = args as any;
+  const response = await client.compute.vms.list(body).asResponse();
+  return asTextContentResult(await maybeFilter(jq_filter, await response.json()));
 };
 
 export default { metadata, tool, handler };

package/src/tools/compute/vms/volumes/list-vms-compute-volumes.ts

@@ -18,13 +18,21 @@ export const metadata: Metadata = {
 export const tool: Tool = {
   name: 'list_vms_compute_volumes',
   description:
-    "When using this tool, always use the `jq_filter` parameter to reduce the response size and improve performance.\n\nOnly omit if you're sure you don't need the data.\n\nList VM's Volumes\n\n# Response Schema\n```json\n{\n  $ref: '#/$defs/volume_list',\n  $defs: {\n    volume_list: {\n      type: 'object',\n      properties: {\n        items: {\n          type: 'array',\n          items: {\n            $ref: '#/$defs/volume'\n          }\n        }\n      },\n      required: [        'items'\n      ]\n    },\n    volume: {\n      type: 'object',\n      description: 'Volume details.',\n      properties: {\n        id: {\n          type: 'string',\n          description: 'Unique identifier for the Volume.'\n        },\n        created_at: {\n          type: 'string',\n          description: 'When the Volume was created.',\n          format: 'date-time'\n        },\n        kind: {\n          $ref: '#/$defs/volume_kind'\n        },\n        name: {\n          type: 'string',\n          description: 'Name of the Volume.'\n        },\n        size: {\n          type: 'integer',\n          description: 'Size of the Volume in GB.'\n        },\n        status: {\n          $ref: '#/$defs/resource_status'\n        },\n        tags: {\n          type: 'array',\n          description: 'Tags to attach to the Volume.',\n          items: {\n            type: 'string'\n          }\n        },\n        type: {\n          $ref: '#/$defs/storage_type'\n        },\n        updated_at: {\n          type: 'string',\n          description: 'When the Volume was updated.',\n          format: 'date-time'\n        },\n        vm_id: {\n          type: 'string',\n          description: 'ID of the VM the Volume is attached to.'\n        },\n        vm_name: {\n          type: 'string',\n          description: 'Name of the VM the Volume is attached to.'\n        }\n      },\n      required: [        'id',\n        'created_at',\n        'kind',\n        'name',\n        'size',\n        'status',\n        'tags',\n        'type',\n        'updated_at',\n        'vm_id',\n        'vm_name'\n      ]\n    },\n    volume_kind: {\n      type: 'string',\n      description: 'Volume kind.',\n      enum: [        'boot',\n        'data'\n      ]\n    },\n    resource_status: {\n      type: 'string',\n      description: 'Status of the resource.',\n      enum: [        'pending',\n        'creating',\n        'updating',\n        'ready',\n        'deleting',\n        'deleted',\n        'error'\n      ]\n    },\n    storage_type: {\n      type: 'string',\n      description: 'Storage type the Volume is using.',\n      enum: [        'nvme'\n      ]\n    }\n  }\n}\n```",
+    "When using this tool, always use the `jq_filter` parameter to reduce the response size and improve performance.\n\nOnly omit if you're sure you don't need the data.\n\nList VM's Volumes\n\n# Response Schema\n```json\n{\n  $ref: '#/$defs/volume_list',\n  $defs: {\n    volume_list: {\n      type: 'object',\n      properties: {\n        items: {\n          type: 'array',\n          items: {\n            $ref: '#/$defs/volume'\n          }\n        },\n        pagination: {\n          $ref: '#/$defs/pagination'\n        }\n      },\n      required: [        'items',\n        'pagination'\n      ]\n    },\n    volume: {\n      type: 'object',\n      description: 'Volume details.',\n      properties: {\n        id: {\n          type: 'string',\n          description: 'Unique identifier for the Volume.'\n        },\n        created_at: {\n          type: 'string',\n          description: 'When the Volume was created.',\n          format: 'date-time'\n        },\n        kind: {\n          $ref: '#/$defs/volume_kind'\n        },\n        name: {\n          type: 'string',\n          description: 'Name of the Volume.'\n        },\n        size: {\n          type: 'integer',\n          description: 'Size of the Volume in GB.'\n        },\n        status: {\n          $ref: '#/$defs/resource_status'\n        },\n        tags: {\n          type: 'array',\n          description: 'Tags to attach to the Volume.',\n          items: {\n            type: 'string'\n          }\n        },\n        type: {\n          $ref: '#/$defs/storage_type'\n        },\n        updated_at: {\n          type: 'string',\n          description: 'When the Volume was updated.',\n          format: 'date-time'\n        },\n        vm_id: {\n          type: 'string',\n          description: 'ID of the VM the Volume is attached to.'\n        },\n        vm_name: {\n          type: 'string',\n          description: 'Name of the VM the Volume is attached to.'\n        }\n      },\n      required: [        'id',\n        'created_at',\n        'kind',\n        'name',\n        'size',\n        'status',\n        'tags',\n        'type',\n        'updated_at',\n        'vm_id',\n        'vm_name'\n      ]\n    },\n    volume_kind: {\n      type: 'string',\n      description: 'Volume kind.',\n      enum: [        'boot',\n        'data'\n      ]\n    },\n    resource_status: {\n      type: 'string',\n      description: 'Status of the resource.',\n      enum: [        'pending',\n        'creating',\n        'updating',\n        'ready',\n        'deleting',\n        'deleted',\n        'error'\n      ]\n    },\n    storage_type: {\n      type: 'string',\n      description: 'Storage type the Volume is using.',\n      enum: [        'nvme'\n      ]\n    },\n    pagination: {\n      type: 'object',\n      description: 'Pagination response details.',\n      properties: {\n        next_cursor: {\n          type: 'string'\n        },\n        previous_cursor: {\n          type: 'string'\n        },\n        total_count: {\n          type: 'integer'\n        }\n      },\n      required: [        'next_cursor',\n        'previous_cursor',\n        'total_count'\n      ]\n    }\n  }\n}\n```",
   inputSchema: {
     type: 'object',
     properties: {
       vm_id: {
         type: 'string',
       },
+      cursor: {
+        type: 'string',
+        description: 'Pagination cursor returned by a previous request',
+      },
+      limit: {
+        type: 'integer',
+        description: 'Maximum number of items to return',
+      },
       jq_filter: {
         type: 'string',
         title: 'jq Filter',
@@ -41,7 +49,8 @@ export const tool: Tool = {
 
 export const handler = async (client: NirvanaLabs, args: Record<string, unknown> | undefined) => {
   const { vm_id, jq_filter, ...body } = args as any;
-  return asTextContentResult(await maybeFilter(jq_filter, await client.compute.vms.volumes.list(vm_id)));
+  const response = await client.compute.vms.volumes.list(vm_id, body).asResponse();
+  return asTextContentResult(await maybeFilter(jq_filter, await response.json()));
 };
 
 export default { metadata, tool, handler };

package/src/tools/compute/volumes/list-compute-volumes.ts

@@ -18,10 +18,18 @@ export const metadata: Metadata = {
 export const tool: Tool = {
   name: 'list_compute_volumes',
   description:
-    "When using this tool, always use the `jq_filter` parameter to reduce the response size and improve performance.\n\nOnly omit if you're sure you don't need the data.\n\nList all volumes\n\n# Response Schema\n```json\n{\n  $ref: '#/$defs/volume_list',\n  $defs: {\n    volume_list: {\n      type: 'object',\n      properties: {\n        items: {\n          type: 'array',\n          items: {\n            $ref: '#/$defs/volume'\n          }\n        }\n      },\n      required: [        'items'\n      ]\n    },\n    volume: {\n      type: 'object',\n      description: 'Volume details.',\n      properties: {\n        id: {\n          type: 'string',\n          description: 'Unique identifier for the Volume.'\n        },\n        created_at: {\n          type: 'string',\n          description: 'When the Volume was created.',\n          format: 'date-time'\n        },\n        kind: {\n          $ref: '#/$defs/volume_kind'\n        },\n        name: {\n          type: 'string',\n          description: 'Name of the Volume.'\n        },\n        size: {\n          type: 'integer',\n          description: 'Size of the Volume in GB.'\n        },\n        status: {\n          $ref: '#/$defs/resource_status'\n        },\n        tags: {\n          type: 'array',\n          description: 'Tags to attach to the Volume.',\n          items: {\n            type: 'string'\n          }\n        },\n        type: {\n          $ref: '#/$defs/storage_type'\n        },\n        updated_at: {\n          type: 'string',\n          description: 'When the Volume was updated.',\n          format: 'date-time'\n        },\n        vm_id: {\n          type: 'string',\n          description: 'ID of the VM the Volume is attached to.'\n        },\n        vm_name: {\n          type: 'string',\n          description: 'Name of the VM the Volume is attached to.'\n        }\n      },\n      required: [        'id',\n        'created_at',\n        'kind',\n        'name',\n        'size',\n        'status',\n        'tags',\n        'type',\n        'updated_at',\n        'vm_id',\n        'vm_name'\n      ]\n    },\n    volume_kind: {\n      type: 'string',\n      description: 'Volume kind.',\n      enum: [        'boot',\n        'data'\n      ]\n    },\n    resource_status: {\n      type: 'string',\n      description: 'Status of the resource.',\n      enum: [        'pending',\n        'creating',\n        'updating',\n        'ready',\n        'deleting',\n        'deleted',\n        'error'\n      ]\n    },\n    storage_type: {\n      type: 'string',\n      description: 'Storage type the Volume is using.',\n      enum: [        'nvme'\n      ]\n    }\n  }\n}\n```",
+    "When using this tool, always use the `jq_filter` parameter to reduce the response size and improve performance.\n\nOnly omit if you're sure you don't need the data.\n\nList all volumes\n\n# Response Schema\n```json\n{\n  $ref: '#/$defs/volume_list',\n  $defs: {\n    volume_list: {\n      type: 'object',\n      properties: {\n        items: {\n          type: 'array',\n          items: {\n            $ref: '#/$defs/volume'\n          }\n        },\n        pagination: {\n          $ref: '#/$defs/pagination'\n        }\n      },\n      required: [        'items',\n        'pagination'\n      ]\n    },\n    volume: {\n      type: 'object',\n      description: 'Volume details.',\n      properties: {\n        id: {\n          type: 'string',\n          description: 'Unique identifier for the Volume.'\n        },\n        created_at: {\n          type: 'string',\n          description: 'When the Volume was created.',\n          format: 'date-time'\n        },\n        kind: {\n          $ref: '#/$defs/volume_kind'\n        },\n        name: {\n          type: 'string',\n          description: 'Name of the Volume.'\n        },\n        size: {\n          type: 'integer',\n          description: 'Size of the Volume in GB.'\n        },\n        status: {\n          $ref: '#/$defs/resource_status'\n        },\n        tags: {\n          type: 'array',\n          description: 'Tags to attach to the Volume.',\n          items: {\n            type: 'string'\n          }\n        },\n        type: {\n          $ref: '#/$defs/storage_type'\n        },\n        updated_at: {\n          type: 'string',\n          description: 'When the Volume was updated.',\n          format: 'date-time'\n        },\n        vm_id: {\n          type: 'string',\n          description: 'ID of the VM the Volume is attached to.'\n        },\n        vm_name: {\n          type: 'string',\n          description: 'Name of the VM the Volume is attached to.'\n        }\n      },\n      required: [        'id',\n        'created_at',\n        'kind',\n        'name',\n        'size',\n        'status',\n        'tags',\n        'type',\n        'updated_at',\n        'vm_id',\n        'vm_name'\n      ]\n    },\n    volume_kind: {\n      type: 'string',\n      description: 'Volume kind.',\n      enum: [        'boot',\n        'data'\n      ]\n    },\n    resource_status: {\n      type: 'string',\n      description: 'Status of the resource.',\n      enum: [        'pending',\n        'creating',\n        'updating',\n        'ready',\n        'deleting',\n        'deleted',\n        'error'\n      ]\n    },\n    storage_type: {\n      type: 'string',\n      description: 'Storage type the Volume is using.',\n      enum: [        'nvme'\n      ]\n    },\n    pagination: {\n      type: 'object',\n      description: 'Pagination response details.',\n      properties: {\n        next_cursor: {\n          type: 'string'\n        },\n        previous_cursor: {\n          type: 'string'\n        },\n        total_count: {\n          type: 'integer'\n        }\n      },\n      required: [        'next_cursor',\n        'previous_cursor',\n        'total_count'\n      ]\n    }\n  }\n}\n```",
   inputSchema: {
     type: 'object',
     properties: {
+      cursor: {
+        type: 'string',
+        description: 'Pagination cursor returned by a previous request',
+      },
+      limit: {
+        type: 'integer',
+        description: 'Maximum number of items to return',
+      },
       jq_filter: {
         type: 'string',
         title: 'jq Filter',
@@ -37,8 +45,9 @@ export const tool: Tool = {
 };
 
 export const handler = async (client: NirvanaLabs, args: Record<string, unknown> | undefined) => {
-  const { jq_filter } = args as any;
-  return asTextContentResult(await maybeFilter(jq_filter, await client.compute.volumes.list()));
+  const { jq_filter, ...body } = args as any;
+  const response = await client.compute.volumes.list(body).asResponse();
+  return asTextContentResult(await maybeFilter(jq_filter, await response.json()));
 };
 
 export default { metadata, tool, handler };