265 lines
9.1 KiB
TypeScript
Executable File
265 lines
9.1 KiB
TypeScript
Executable File
/**
|
||
* 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}`;
|
||
}
|
||
}
|