moneyFlow.ts•10.8 kB
import { TUSHARE_CONFIG } from '../config.js';
export const moneyFlow = {
name: "money_flow",
description: "获取个股和大盘资金流向数据，包括主力资金、超大单、大单、中单、小单的净流入净额和净占比数据",
parameters: {
type: "object",
properties: {
ts_code: {
type: "string",
description: "股票代码，如'000001.SZ'表示平安银行个股资金流向。不填写则查询大盘资金流向数据"
},
start_date: {
type: "string",
description: "起始日期，格式为YYYYMMDD，如'20240901'"
},
end_date: {
type: "string",
description: "结束日期，格式为YYYYMMDD，如'20240930'"
}
},
required: ["start_date", "end_date"]
},
async run(args: {
ts_code?: string;
start_date: string;
end_date: string;
}) {
try {
console.log('资金流向数据查询参数:', args);
const TUSHARE_API_KEY = TUSHARE_CONFIG.API_TOKEN;
const TUSHARE_API_URL = TUSHARE_CONFIG.API_URL;
if (!TUSHARE_API_KEY) {
throw new Error('请配置TUSHARE_TOKEN环境变量');
}
// 判断查询类型：个股 or 大盘
const isMarketFlow = !args.ts_code || args.ts_code.trim() === '';
let result;
if (isMarketFlow) {
// 查询大盘资金流向
result = await fetchMarketMoneyFlow(
args.start_date,
args.end_date,
TUSHARE_API_KEY,
TUSHARE_API_URL
);
} else {
// 查询个股资金流向
result = await fetchStockMoneyFlow(
args.ts_code!,
args.start_date,
args.end_date,
TUSHARE_API_KEY,
TUSHARE_API_URL
);
}
if (!result.data || result.data.length === 0) {
const target = isMarketFlow ? '大盘' : `股票${args.ts_code}`;
throw new Error(`未找到${target}在指定时间范围内的资金流向数据`);
}
// 格式化输出
const formattedOutput = formatMoneyFlowData(result.data, result.fields, isMarketFlow, args.ts_code);
return {
content: [{ type: "text", text: formattedOutput }]
};
} catch (error) {
console.error('资金流向数据查询错误:', error);
return {
content: [{
type: "text",
text: `查询资金流向数据时发生错误: ${error instanceof Error ? error.message : '未知错误'}`
}]
};
}
}
};
// 获取大盘资金流向数据
async function fetchMarketMoneyFlow(
startDate: string,
endDate: string,
apiKey: string,
apiUrl: string
) {
const params = {
api_name: "moneyflow_mkt_dc",
token: apiKey,
params: {
start_date: startDate,
end_date: endDate
},
fields: "trade_date,close_sh,pct_change_sh,close_sz,pct_change_sz,net_amount,net_amount_rate,buy_elg_amount,buy_elg_amount_rate,buy_lg_amount,buy_lg_amount_rate,buy_md_amount,buy_md_amount_rate,buy_sm_amount,buy_sm_amount_rate"
};
return await callTushareAPI(params, apiUrl);
}
// 获取个股资金流向数据
async function fetchStockMoneyFlow(
tsCode: string,
startDate: string,
endDate: string,
apiKey: string,
apiUrl: string
) {
const params = {
api_name: "moneyflow_dc",
token: apiKey,
params: {
ts_code: tsCode,
start_date: startDate,
end_date: endDate
},
fields: "ts_code,trade_date,close,pct_change,net_amount,net_amount_rate,buy_elg_amount,buy_elg_amount_rate,buy_lg_amount,buy_lg_amount_rate,buy_md_amount,buy_md_amount_rate,buy_sm_amount,buy_sm_amount_rate"
};
return await callTushareAPI(params, apiUrl);
}
// 调用Tushare API的通用函数
async function callTushareAPI(params: any, apiUrl: string) {
const controller = new AbortController();
const timeoutId = setTimeout(() => controller.abort(), TUSHARE_CONFIG.TIMEOUT);
try {
console.log(`请求Tushare API: ${params.api_name}，参数:`, params.params);
const response = await fetch(apiUrl, {
method: "POST",
headers: {
"Content-Type": "application/json"
},
body: JSON.stringify(params),
signal: controller.signal
});
if (!response.ok) {
throw new Error(`Tushare API请求失败: ${response.status}`);
}
const data = await response.json();
if (data.code !== 0) {
throw new Error(`Tushare API错误: ${data.msg}`);
}
if (!data.data || !data.data.items) {
throw new Error(`未找到资金流向数据`);
}
// 获取字段名
const fields = data.data.fields;
// 将数据转换为对象数组
const convertedData = data.data.items.map((item: any) => {
const result: Record<string, any> = {};
fields.forEach((field: string, index: number) => {
result[field] = item[index];
});
return result;
});
console.log(`成功获取到${convertedData.length}条资金流向数据记录`);
return {
data: convertedData,
fields: fields
};
} finally {
clearTimeout(timeoutId);
}
}
// 格式化资金流向数据输出
function formatMoneyFlowData(data: any[], fields: string[], isMarketFlow: boolean, tsCode?: string): string {
// 按交易日期倒序排列（最新在前）
const sortedData = data.sort((a, b) => (b.trade_date || '').localeCompare(a.trade_date || ''));
const target = isMarketFlow ? '大盘' : `个股 ${tsCode}`;
let output = `# 💰 ${target}资金流向数据\n\n`;
// 数据统计摘要
const totalDays = sortedData.length;
const netInflowDays = sortedData.filter(item => (parseFloat(item.net_amount) || 0) > 0).length;
const netOutflowDays = totalDays - netInflowDays;
// 计算累计净流入金额
const totalNetAmount = sortedData.reduce((sum, item) => sum + (parseFloat(item.net_amount) || 0), 0);
output += `## 📊 统计摘要\n\n`;
output += `- 查询时间范围: ${sortedData[sortedData.length - 1]?.trade_date} 至 ${sortedData[0]?.trade_date}\n`;
output += `- 交易天数: ${totalDays} 天\n`;
output += `- 净流入天数: ${netInflowDays} 天 (${((netInflowDays/totalDays)*100).toFixed(1)}%)\n`;
output += `- 净流出天数: ${netOutflowDays} 天 (${((netOutflowDays/totalDays)*100).toFixed(1)}%)\n`;
output += `- 累计净流入: ${formatMoney(totalNetAmount)}\n\n`;
// 构建数据表格
if (isMarketFlow) {
output += formatMarketFlowTable(sortedData);
} else {
output += formatStockFlowTable(sortedData);
}
// 最近5个交易日资金流向趋势
const recentData = sortedData.slice(0, Math.min(5, sortedData.length));
output += `\n## 📈 最近资金流向趋势\n\n`;
recentData.forEach(item => {
const netAmount = parseFloat(item.net_amount) || 0;
const netAmountRate = parseFloat(item.net_amount_rate) || 0;
const trend = netAmount > 0 ? '🟢' : '🔴';
const direction = netAmount > 0 ? '净流入' : '净流出';
output += `${item.trade_date} ${trend} 主力${direction} ${formatMoney(Math.abs(netAmount))} (${Math.abs(netAmountRate).toFixed(2)}%)\n`;
});
output += `\n---\n*数据来源: [Tushare](https://tushare.pro)*`;
return output;
}
// 格式化大盘资金流向表格
function formatMarketFlowTable(data: any[]): string {
let output = `## 📋 大盘资金流向明细\n\n`;
output += `| 交易日期 | 上证收盘 | 上证涨跌% | 深证收盘 | 深证涨跌% | 主力净流入(万元) | 净占比% | 超大单净流入(万元) | 大单净流入(万元) |\n`;
output += `|---------|---------|---------|---------|---------|------------|--------|------------|----------|\n`;
data.forEach(item => {
const netAmount = parseFloat(item.net_amount) || 0;
const netAmountRate = parseFloat(item.net_amount_rate) || 0;
const elgAmount = parseFloat(item.buy_elg_amount) || 0;
const lgAmount = parseFloat(item.buy_lg_amount) || 0;
const netFlowIcon = netAmount > 0 ? '🟢' : '🔴';
output += `| ${item.trade_date} `;
output += `| ${formatNumber(item.close_sh)} `;
output += `| ${formatPercent(item.pct_change_sh)} `;
output += `| ${formatNumber(item.close_sz)} `;
output += `| ${formatPercent(item.pct_change_sz)} `;
output += `| ${netFlowIcon} ${formatMoney(netAmount)} `;
output += `| ${formatPercent(netAmountRate)} `;
output += `| ${formatMoney(elgAmount)} `;
output += `| ${formatMoney(lgAmount)} |\n`;
});
return output;
}
// 格式化个股资金流向表格
function formatStockFlowTable(data: any[]): string {
let output = `## 📋 个股资金流向明细\n\n`;
output += `| 交易日期 | 收盘价 | 涨跌% | 主力净流入(万元) | 净占比% | 超大单净流入(万元) | 大单净流入(万元) | 中单净流入(万元) | 小单净流入(万元) |\n`;
output += `|---------|--------|------|------------|--------|------------|------------|------------|------------|\n`;
data.forEach(item => {
const netAmount = parseFloat(item.net_amount) || 0;
const netAmountRate = parseFloat(item.net_amount_rate) || 0;
const elgAmount = parseFloat(item.buy_elg_amount) || 0;
const lgAmount = parseFloat(item.buy_lg_amount) || 0;
const mdAmount = parseFloat(item.buy_md_amount) || 0;
const smAmount = parseFloat(item.buy_sm_amount) || 0;
const netFlowIcon = netAmount > 0 ? '🟢' : '🔴';
output += `| ${item.trade_date} `;
output += `| ${formatNumber(item.close)} `;
output += `| ${formatPercent(item.pct_change)} `;
output += `| ${netFlowIcon} ${formatMoney(netAmount)} `;
output += `| ${formatPercent(netAmountRate)} `;
output += `| ${formatMoney(elgAmount)} `;
output += `| ${formatMoney(lgAmount)} `;
output += `| ${formatMoney(mdAmount)} `;
output += `| ${formatMoney(smAmount)} |\n`;
});
return output;
}
// 格式化金额（万元）
function formatMoney(amount: number): string {
if (amount === 0) return '0.00万';
const amountInWan = amount / 10000;
if (Math.abs(amountInWan) >= 100000) {
return (amountInWan / 10000).toFixed(2) + '亿';
}
return amountInWan.toFixed(2) + '万';
}
// 格式化数字
function formatNumber(num: any): string {
if (num === null || num === undefined || num === '' || isNaN(parseFloat(num))) {
return 'N/A';
}
const number = parseFloat(num);
return number.toLocaleString('zh-CN', { maximumFractionDigits: 2 });
}
// 格式化百分比
function formatPercent(num: any): string {
if (num === null || num === undefined || num === '' || isNaN(parseFloat(num))) {
return 'N/A';
}
const number = parseFloat(num);
const sign = number > 0 ? '+' : '';
return `${sign}${number.toFixed(2)}%`;
}