MCP Avantage

Instructions

Implementation Reference

• src/index.ts:628-642 (registration)

  Registers the 'economicIndicators_inflation' MCP tool with name, description, input schema, and execute handler that delegates to the generic executeAvantageTool function calling the AVantage library's economicIndicators.inflation method.

  server.addTool({
    name: "economicIndicators_inflation",
    description: "Retrieves US inflation rate data.",
    parameters: schemas.EconomicIndicatorsDataTypeParamSchema,
    execute: (
      args,
      context // Let type be inferred
    ) =>
      executeAvantageTool(
        "economicIndicators_inflation",
        args,
        context,
        (av, params) => av.economicIndicators.inflation(params)
      ),
  });

• src/schemas.ts:124-127 (schema)

  Defines the Zod input parameters schema for the tool, allowing an optional 'datatype' field (json or csv).

  export const EconomicIndicatorsDataTypeParamSchema = z.object({
    datatype: DatatypeSchema.default('json').optional(),
  }).describe('Common parameter schema accepting only datatype.')

• src/index.ts:38-115 (helper)

  Generic helper function that implements the core execution logic for all AVantage-based tools: resolves API key, manages AVantage client instance, invokes the provided library method (av.economicIndicators.inflation for this tool), handles errors, and returns JSON-stringified data.

  async function executeAvantageTool<TArgs, TResult>(
    toolName: string,
    args: TArgs,
    context: Context<Record<string, unknown> | undefined>, // Use the imported Context type directly
    avantageMethod: (
      av: AVantage,
      args: TArgs
    ) => Promise<{ error?: boolean; reason?: string; data?: TResult }>
  ): Promise<string> {
    logger.info(`Executing '${toolName}' tool for request ID: ${context}`);
    logger.debug(`Args for ${toolName}: ${JSON.stringify(args)}`);
  
    // --- Authentication & Resource Management ---
    // Access extraArgs safely - it might be null or undefined
    const extraArgsApiKey = context.extraArgs?.apiKey as string | undefined;
    const apiKey = extraArgsApiKey || config.apiKey;
  
    if (!apiKey) {
      logger.error(`'${toolName}' failed: Alpha Vantage API key missing.`);
      throw new UserError(apiKeyErrorMessage);
    }
    logger.debug(
      `Using AV API key (source: ${extraArgsApiKey ? "extraArgs" : "environment"}) for ${toolName}`
    );
  
    try {
      // Get or create AVantage instance managed by ResourceManager
      const av = await resourceManager.getResource<AVantage>(
        apiKey, // Cache key is the resolved API key
        "avantage_client", // Type identifier for logging
        async (key) => {
          // Factory Function
          logger.info(
            `Creating new AVantage instance for key ending ...${key.slice(-4)}`
          );
          // AVantage library reads AV_PREMIUM from process.env internally
          return new AVantage(key);
        },
        async (avInstance) => {
          // Cleanup Function (no-op needed for AVantage)
          logger.debug(`Destroying AVantage instance (no-op)`);
        }
      );
  
      // --- Library Call ---
      const result = await avantageMethod(av, args);
  
      // --- Response Handling ---
      if (result.error) {
        logger.warn(
          `'${toolName}' failed. Reason from avantage: ${result.reason}`
        );
        throw new UserError(result.reason || `Tool '${toolName}' failed.`);
      }
  
      if (result.data === undefined || result.data === null) {
        logger.warn(`'${toolName}' completed successfully but returned no data.`);
        return "null"; // Return string "null" for empty data
      }
  
      logger.info(`'${toolName}' completed successfully.`);
      // Stringify the data part of the response
      return JSON.stringify(result.data);
    } catch (error: any) {
      logger.error(
        `Error during execution of '${toolName}': ${error.message}`,
        error
      );
      // If it's already a UserError, rethrow it
      if (error instanceof UserError) {
        throw error;
      }
      // Otherwise, wrap it in a UserError
      throw new UserError(
        `An unexpected error occurred while executing tool '${toolName}': ${error.message}`
      );
    }
  }