Initial commit
This commit is contained in:
Z User
264
skills/stock-analysis-skill/src/analyzer.ts
Executable file
264
skills/stock-analysis-skill/src/analyzer.ts
Executable file
@@ -0,0 +1,264 @@
|
||||
/**
|
||||
* analyzer.ts — LLM/VLM 分析层
|
||||
* 七段式决策仪表盘 + 美股可附加股息分析
|
||||
*/
|
||||
|
||||
import ZAI from "z-ai-web-dev-sdk";
|
||||
import { StockData, AnalysisResult, OutputFormat, Market, Verdict, PositionInfo } from "./types";
|
||||
import { validateStockData } from "./dataFetcher";
|
||||
import { analyzeDividend, formatDividendMarkdown } from "./dividend";
|
||||
|
||||
const MARKET_LABEL: Record<Market, string> = { CN: "A股", HK: "港股", US: "美股" };
|
||||
|
||||
// ── 仪表盘 Prompt ─────────────────────────────────────────
|
||||
|
||||
function buildDashboardPrompt(
|
||||
data: StockData,
|
||||
position: PositionInfo | undefined,
|
||||
warnings: string[]
|
||||
): string {
|
||||
const warningBlock = warnings.length > 0
|
||||
? `⚠️ 数据预警（必须在报告中体现）：\n${warnings.map((w) => `- ${w}`).join("\n")}\n\n`
|
||||
: "";
|
||||
|
||||
const positionBlock = position
|
||||
? position.status === "holding"
|
||||
? `用户持仓：持仓中，成本价 ${position.cost ?? "未知"} 元${position.shares ? `，${position.shares} 股` : ""}。请给出盈亏分析和针对性建议。`
|
||||
: `用户持仓：当前空仓。`
|
||||
: `用户持仓：未提供（请同时给出空仓者和持仓者两套建议）。`;
|
||||
|
||||
return `${warningBlock}${positionBlock}
|
||||
|
||||
股票数据：
|
||||
\`\`\`json
|
||||
${JSON.stringify(data, null, 2)}
|
||||
\`\`\`
|
||||
|
||||
请输出以下格式的完整决策仪表盘（严格按结构，不增删章节）：
|
||||
|
||||
---
|
||||
|
||||
## 决策仪表盘 · {名称}({代码}) · {市场}
|
||||
|
||||
---
|
||||
|
||||
### 📰 重要信息速览
|
||||
**💭 舆情情绪：** 一句话描述
|
||||
**📊 业绩预期：** 结合 PE/ROE/行业简述，数据缺失标"暂缺"
|
||||
**🚨 风险警报：**
|
||||
- 风险1（技术面或宏观）
|
||||
- 风险2（基本面或行业）
|
||||
**✨ 利好催化：**
|
||||
- 利好1（技术面）
|
||||
- 利好2（基本面或行业）
|
||||
**📢 最新动态：** 结合行业背景，补充1条关键信息
|
||||
|
||||
---
|
||||
|
||||
### 📌 核心结论
|
||||
**[emoji] 结论：强烈买入 / 买入 / 观望 / 卖出**（四选一，乖离率>5%不得为买入）
|
||||
**💬 一句话决策：** 核心逻辑
|
||||
**⏰ 时效性：** 立即行动 / 今日内 / 不急
|
||||
|
||||
（根据持仓状态输出）
|
||||
- **🆕 空仓者：** 是否进场、建仓点位、仓位比例
|
||||
- **💼 持仓者：** 持有/加仓/减仓/止损建议${position?.status === "holding" && position.cost ? "，含成本盈亏分析" : ""}
|
||||
|
||||
---
|
||||
|
||||
### 📈 当日行情
|
||||
列出：收盘价、昨收、开盘、最高、最低、涨跌幅、涨跌额、振幅、成交量、成交额（缺失标"暂缺"）
|
||||
|
||||
---
|
||||
|
||||
### 📊 数据透视
|
||||
|
||||
**技术面：**
|
||||
表格（指标 | 数值 | 解读）：MA5、MA10、MA20、乖离率(BIAS20)、RSI（如有）、支撑位、压力位
|
||||
结论：均线状态 + 趋势强度（xx/100）
|
||||
|
||||
**基本面（注明报告期）：**
|
||||
表格（指标 | 数值 | 行业对比）：ROE、毛利率、净利率、资产负债率、PE、PB
|
||||
数据缺失标"暂缺"，不得捏造
|
||||
|
||||
**资金面：**（A股/港股适用，美股可略）
|
||||
- 主力净流入：金额（占比%），一句话解读
|
||||
- 筹码：获利比例 | 平均成本 | 集中度
|
||||
|
||||
---
|
||||
|
||||
### 🎯 作战计划
|
||||
|
||||
| 点位类型 | 价格 | 说明 |
|
||||
|---------|------|------|
|
||||
| 🎯 理想买入 | xxx | |
|
||||
| 🔵 次优买入 | xxx | |
|
||||
| 🛑 止损位 | xxx | |
|
||||
| 🎊 目标位 | xxx | |
|
||||
|
||||
**💰 仓位建议：** x成
|
||||
**建仓策略：** 分批策略
|
||||
**风控策略：** 止损纪律
|
||||
|
||||
---
|
||||
|
||||
### ✅ 检查清单
|
||||
- ✅/⚠️/❌ 均线状态
|
||||
- ✅/⚠️/❌ 乖离率安全（<5%）
|
||||
- ✅/⚠️/❌ 量能配合
|
||||
- ✅/⚠️/❌ 估值合理
|
||||
- ✅/⚠️/❌ 资金流向
|
||||
- ✅/⚠️/❌ 筹码健康
|
||||
|
||||
**综合结论：** 一句话总结当前状态和建议。
|
||||
|
||||
---
|
||||
*以上分析仅供参考，不构成投资建议，据此操作风险自担。*`;
|
||||
}
|
||||
|
||||
// ── 研报 Prompt（PDF/Word）──────────────────────────────
|
||||
|
||||
function buildReportPrompt(data: StockData, position: PositionInfo | undefined): string {
|
||||
const positionBlock = position?.status === "holding"
|
||||
? `用户持仓成本：${position.cost ?? "未知"}`
|
||||
: "用户当前空仓";
|
||||
|
||||
return `${positionBlock}
|
||||
|
||||
股票数据：
|
||||
\`\`\`json
|
||||
${JSON.stringify(data, null, 2)}
|
||||
\`\`\`
|
||||
|
||||
请生成结构化研报：
|
||||
|
||||
【研究报告】{名称}({代码}) · {市场} · {日期}
|
||||
|
||||
一、投资结论（买入/强烈买入/观望/卖出，含目标价、止损价，分空仓/持仓两套建议）
|
||||
二、重要信息速览（舆情/业绩预期/风险/利好/最新动态）
|
||||
三、数据透视（技术面/基本面/资金面）
|
||||
四、作战计划（点位表/仓位/持仓周期）
|
||||
五、风险提示（2-3条）
|
||||
|
||||
免责声明：本报告由AI辅助生成，仅供参考，不构成投资建议，据此操作风险自担。`;
|
||||
}
|
||||
|
||||
// ── 提取结论 ──────────────────────────────────────────────
|
||||
|
||||
function extractVerdict(text: string): Verdict {
|
||||
const patterns = [
|
||||
/结论[：:]\s*[💚🟢🟡🔴⚪]?\s*(强烈买入|买入|观望|卖出)/,
|
||||
/核心结论[：:]\s*[💚🟢🟡🔴⚪]?\s*(强烈买入|买入|观望|卖出)/,
|
||||
/\*\*(强烈买入|买入|观望|卖出)\*\*/,
|
||||
];
|
||||
for (const p of patterns) {
|
||||
const m = text.match(p);
|
||||
if (m) return m[1] as Verdict;
|
||||
}
|
||||
return "观望";
|
||||
}
|
||||
|
||||
// ── 核心分析 ──────────────────────────────────────────────
|
||||
|
||||
export async function analyzeStock(
|
||||
data: StockData,
|
||||
outputFormat: OutputFormat = "markdown",
|
||||
position?: PositionInfo,
|
||||
includeDividend = false
|
||||
): Promise<AnalysisResult> {
|
||||
const { valid, warnings } = validateStockData(data);
|
||||
const name = data.name ?? data.code;
|
||||
|
||||
if (!valid) {
|
||||
return {
|
||||
code: data.code, market: data.market, name,
|
||||
verdict: "观望",
|
||||
analysis: `## ⚠️ 数据获取失败\n\n${data.code} 数据无法获取（${data.error ?? "未知错误"}），建议手动核实。`,
|
||||
warnings, outputFormat,
|
||||
generatedAt: new Date().toISOString(),
|
||||
};
|
||||
}
|
||||
|
||||
const zai = await ZAI.create();
|
||||
const userPrompt = outputFormat === "markdown"
|
||||
? buildDashboardPrompt(data, position, warnings)
|
||||
: buildReportPrompt(data, position);
|
||||
|
||||
let analysisText = "⚠️ LLM 未返回内容，请重试。";
|
||||
try {
|
||||
const completion = await zai.chat.completions.create({
|
||||
messages: [
|
||||
{ role: "system", content: `你是一位资深${MARKET_LABEL[data.market]}股票分析师。数据缺失标"暂缺"，严禁捏造。乖离率>5%不得建议买入。结论四选一：强烈买入/买入/观望/卖出。输出语言：中文。` },
|
||||
{ role: "user", content: userPrompt },
|
||||
],
|
||||
thinking: { type: "disabled" },
|
||||
});
|
||||
analysisText = completion.choices[0]?.message?.content ?? analysisText;
|
||||
} catch (err: any) {
|
||||
analysisText = `## ⚠️ 分析失败\n\nLLM 调用出错：${err.message}`;
|
||||
}
|
||||
|
||||
// 美股附加股息分析
|
||||
if (includeDividend && data.market === "US" && outputFormat === "markdown") {
|
||||
try {
|
||||
const dividend = await analyzeDividend(data.code);
|
||||
const dividendMd = formatDividendMarkdown(dividend);
|
||||
analysisText += `\n\n${dividendMd}`;
|
||||
} catch {}
|
||||
}
|
||||
|
||||
return {
|
||||
code: data.code, market: data.market, name,
|
||||
verdict: extractVerdict(analysisText),
|
||||
analysis: analysisText,
|
||||
warnings, outputFormat,
|
||||
generatedAt: new Date().toISOString(),
|
||||
};
|
||||
}
|
||||
|
||||
// ── 批量分析 ──────────────────────────────────────────────
|
||||
|
||||
export async function analyzeMultipleStocks(
|
||||
stockDataList: StockData[],
|
||||
outputFormat: OutputFormat = "markdown",
|
||||
positions?: Record<string, PositionInfo>,
|
||||
includeDividend = false
|
||||
): Promise<AnalysisResult[]> {
|
||||
const results: AnalysisResult[] = [];
|
||||
for (const data of stockDataList) {
|
||||
const position = positions?.[data.code];
|
||||
results.push(await analyzeStock(data, outputFormat, position, includeDividend));
|
||||
}
|
||||
return results;
|
||||
}
|
||||
|
||||
// ── K线图分析（VLM）──────────────────────────────────────
|
||||
|
||||
export async function analyzeChartImage(
|
||||
imageUrlOrBase64: string,
|
||||
stockCode: string,
|
||||
isBase64 = false
|
||||
): Promise<string> {
|
||||
try {
|
||||
const zai = await ZAI.create();
|
||||
const imageContent = isBase64
|
||||
? { type: "base64" as const, data: imageUrlOrBase64, mediaType: "image/png" as const }
|
||||
: { type: "url" as const, url: imageUrlOrBase64 };
|
||||
|
||||
const completion = await zai.chat.completions.create({
|
||||
messages: [
|
||||
{ role: "system", content: "你是技术分析专家，擅长K线形态识别。请用中文回答。" },
|
||||
{
|
||||
role: "user",
|
||||
content: [
|
||||
{ type: "image", image: imageContent },
|
||||
{ type: "text", text: `这是 ${stockCode} 的K线图，请分析：\n1. 当前K线形态\n2. 趋势方向\n3. 关键支撑位和压力位\n4. 成交量配合\n5. 短期操作建议` },
|
||||
],
|
||||
},
|
||||
],
|
||||
});
|
||||
return completion.choices[0]?.message?.content ?? "⚠️ VLM 未返回内容";
|
||||
} catch (err: any) {
|
||||
return `K线图分析失败：${err.message}`;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user