getSplitsData.ts•3.52 kB
import { z } from 'zod';
import type { MarketstackClient, MarketstackApiParams } from '../../marketstackClient.js';
// Define the input schema shape for the Splits Data tool
const getSplitsDataInputSchemaShape = {
symbols: z
.string()
.describe(
'Specify one or multiple comma-separated stock symbols (tickers) for your request, e.g. `AAPL` or `AAPL,MSFT`. Each symbol consumes one API request. Maximum: 100 symbols'
),
sort: z
.enum(['DESC', 'ASC'])
.optional()
.default('DESC')
.describe(
'By default, results are sorted by date/time descending. Use this parameter to specify a sorting order. Available values: `DESC` (Default), `ASC`.'
),
date_from: z
.string()
.optional()
.describe(
'Filter results based on a specific timeframe by passing a from-date in `YYYY-MM-DD` format. You can also specify an exact time in ISO-8601 date format, e.g. `2020-05-21T00:00:00+0000`.'
),
date_to: z
.string()
.optional()
.describe(
'Filter results based on a specific timeframe by passing an end-date in `YYYY-MM-DD` format. You can also specify an exact time in ISO-8601 date format, e.g. `2020-05-21T00:00:00+0000`.'
),
limit: z
.number()
.int()
.min(1)
.max(1000)
.optional()
.default(100)
.describe(
'Specify a pagination limit (number of results per page) for your API request. Default limit value is `100`, maximum allowed limit value is `1000`.'
),
offset: z
.number()
.int()
.min(0)
.optional()
.default(0)
.describe(
'Specify a pagination offset value for your API request. Example: An offset value of `100` combined with a limit value of 10 would show results 100-110. Default value is `0`, starting with the first available result.'
),
};
type RawSchemaShape = typeof getSplitsDataInputSchemaShape;
type Input = z.infer<z.ZodObject<RawSchemaShape>>;
type Output = any; // TODO: Define a more specific output type based on Marketstack response
// Define the handler function for the Splits Data tool
const getSplitsDataHandler = async (input: Input, client: MarketstackClient): Promise<Output> => {
try {
const { symbols, sort, date_from, date_to, limit, offset } = input;
const apiRequestParams: MarketstackApiParams = {
endpoint: 'splits',
symbols,
...(sort && { sort }), // Include if sort is provided
...(date_from && { date_from }), // Include if date_from is provided
...(date_to && { date_to }), // Include if date_to is provided
...(limit && { limit }), // Include if limit is provided
...(offset && { offset }), // Include if offset is provided
};
const data = await client.fetchApiData(apiRequestParams);
return data;
} catch (error: unknown) {
console.error('getSplitsData tool error:', error);
const message = error instanceof Error ? error.message : 'An unknown error occurred.';
throw new Error(`getSplitsData tool failed: ${message}`);
}
};
// Define the tool definition object structure
type MarketstackToolDefinition = {
name: string;
description: string;
inputSchemaShape: RawSchemaShape;
handler: (input: Input, client: MarketstackClient) => Promise<Output>;
};
export const getSplitsDataTool: MarketstackToolDefinition = {
name: 'get_splits_data',
description: 'Look up information about the stock splits factor for different symbols.',
inputSchemaShape: getSplitsDataInputSchemaShape,
handler: getSplitsDataHandler,
};