fi-pool-server 0.1.0

package/dist/services/llm.d.ts

@@ -0,0 +1,64 @@
+/**
+ * LLM 推理服务 — LM Studio (OpenAI 兼容 API)
+ *
+ * 通过 LM Studio 的 OpenAI 兼容 HTTP 接口调用本地 LLM 模型。
+ * 支持聊天补全、模型列表查询和连接检查。
+ *
+ * @module services/llm
+ */
+/**
+ * 调用 LM Studio 聊天补全 API。
+ *
+ * 向本地 LLM 发送消息列表并获取回复文本。
+ * 内置超时（60s）和重试（2 次）机制。
+ *
+ * @param params.model     - 模型名称，不传则使用环境变量 LLM_MODEL
+ * @param params.messages  - 对话消息列表（role: system/user/assistant）
+ * @param params.maxTokens - 最大生成 token 数，不传则由模型决定
+ * @param params.temperature - 采样温度，默认 0.7
+ * @param params.sessionId   - 会话 ID（透传，暂未用于 LM Studio 本身）
+ * @returns LLM 回复文本
+ *
+ * @throws 当 LM Studio 不可达、返回错误或结果为空白时抛出
+ *
+ * @example
+ * const reply = await chatCompletion({
+ *   messages: [{ role: 'user', content: '分析贵州茅台' }],
+ *   maxTokens: 500,
+ * });
+ */
+export declare function chatCompletion(params: {
+    model?: string;
+    messages: {
+        role: string;
+        content: string;
+    }[];
+    maxTokens?: number;
+    temperature?: number;
+    sessionId?: string;
+}): Promise<string>;
+/**
+ * 列出 LM Studio 中可用的模型。
+ *
+ * @returns 模型 ID 数组
+ *
+ * @example
+ * const models = await listModels();
+ * // ['qwen/qwen3.5-9b', '...']
+ */
+export declare function listModels(): Promise<string[]>;
+/**
+ * 检查 LM Studio 服务是否可达。
+ *
+ * 通过调用 listModels() 验证连接，不抛出异常即为可达。
+ *
+ * @returns true 表示连接正常，false 表示不可达
+ */
+export declare function checkConnection(): Promise<boolean>;
+/**
+ * 获取 LLM 上下文窗口限制。
+ *
+ * @returns 环境变量 LLM_CONTEXT_LIMIT 的值（默认 262144）
+ */
+export declare function getContextLimit(): number;
+//# sourceMappingURL=llm.d.ts.map

package/dist/services/llm.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm.d.ts","sourceRoot":"","sources":["../../src/services/llm.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAoDH;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,wBAAsB,cAAc,CAClC,MAAM,EAAE;IACN,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAC9C,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,GACA,OAAO,CAAC,MAAM,CAAC,CA0CjB;AAED;;;;;;;;GAQG;AACH,wBAAsB,UAAU,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC,CAUpD;AAED;;;;;;GAMG;AACH,wBAAsB,eAAe,IAAI,OAAO,CAAC,OAAO,CAAC,CAOxD;AAED;;;;GAIG;AACH,wBAAgB,eAAe,IAAI,MAAM,CAExC"}

package/dist/services/llm.js

@@ -0,0 +1,109 @@
+/**
+ * LLM 推理服务 — LM Studio (OpenAI 兼容 API)
+ *
+ * 通过 LM Studio 的 OpenAI 兼容 HTTP 接口调用本地 LLM 模型。
+ * 支持聊天补全、模型列表查询和连接检查。
+ *
+ * @module services/llm
+ */
+import { fetchJson, fetchWithRetry } from '../utils/http-client.js';
+// ─── 配置 ────────────────────────────────────────────────────────
+/** LM Studio 服务地址，默认 http://127.0.0.1:1234 */
+const BASE_URL = process.env.LLM_BASE_URL || 'http://127.0.0.1:1234';
+/** 默认使用的模型名称 */
+const DEFAULT_MODEL = process.env.LLM_MODEL || 'qwen/qwen3.5-9b';
+/** LLM 上下文窗口限制（token），默认 262144 */
+const CONTEXT_LIMIT = parseInt(process.env.LLM_CONTEXT_LIMIT || '262144', 10);
+// ─── 公开 API ────────────────────────────────────────────────────
+/**
+ * 调用 LM Studio 聊天补全 API。
+ *
+ * 向本地 LLM 发送消息列表并获取回复文本。
+ * 内置超时（60s）和重试（2 次）机制。
+ *
+ * @param params.model     - 模型名称，不传则使用环境变量 LLM_MODEL
+ * @param params.messages  - 对话消息列表（role: system/user/assistant）
+ * @param params.maxTokens - 最大生成 token 数，不传则由模型决定
+ * @param params.temperature - 采样温度，默认 0.7
+ * @param params.sessionId   - 会话 ID（透传，暂未用于 LM Studio 本身）
+ * @returns LLM 回复文本
+ *
+ * @throws 当 LM Studio 不可达、返回错误或结果为空白时抛出
+ *
+ * @example
+ * const reply = await chatCompletion({
+ *   messages: [{ role: 'user', content: '分析贵州茅台' }],
+ *   maxTokens: 500,
+ * });
+ */
+export async function chatCompletion(params) {
+    const url = `${BASE_URL}/v1/chat/completions`;
+    const body = {
+        model: params.model || DEFAULT_MODEL,
+        messages: params.messages,
+        max_tokens: params.maxTokens,
+        temperature: params.temperature ?? 0.7,
+        stream: false,
+    };
+    // 使用 fetchWithRetry：传入一个返回 Promise 的函数
+    // LLM 推理可能较慢（尤其首次加载模型），超时设为 120s
+    const LLM_TIMEOUT = 120_000;
+    const data = await fetchWithRetry(() => fetchJson(url, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+        },
+        body: JSON.stringify(body),
+    }, undefined, // signal
+    LLM_TIMEOUT), 2, 1000);
+    if (!data.choices || data.choices.length === 0) {
+        throw new Error(`LLM 返回空结果: 无 choices (model=${body.model})`);
+    }
+    const content = data.choices[0].message?.content;
+    if (content === undefined || content === null) {
+        throw new Error(`LLM 返回空消息内容 (model=${body.model})`);
+    }
+    return content;
+}
+/**
+ * 列出 LM Studio 中可用的模型。
+ *
+ * @returns 模型 ID 数组
+ *
+ * @example
+ * const models = await listModels();
+ * // ['qwen/qwen3.5-9b', '...']
+ */
+export async function listModels() {
+    const url = `${BASE_URL}/v1/models`;
+    const data = await fetchJson(url);
+    if (!data.data || !Array.isArray(data.data)) {
+        throw new Error(`模型列表响应格式异常: ${JSON.stringify(data).slice(0, 200)}`);
+    }
+    return data.data.map((m) => m.id);
+}
+/**
+ * 检查 LM Studio 服务是否可达。
+ *
+ * 通过调用 listModels() 验证连接，不抛出异常即为可达。
+ *
+ * @returns true 表示连接正常，false 表示不可达
+ */
+export async function checkConnection() {
+    try {
+        await listModels();
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+/**
+ * 获取 LLM 上下文窗口限制。
+ *
+ * @returns 环境变量 LLM_CONTEXT_LIMIT 的值（默认 262144）
+ */
+export function getContextLimit() {
+    return CONTEXT_LIMIT;
+}
+//# sourceMappingURL=llm.js.map

package/dist/services/llm.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm.js","sourceRoot":"","sources":["../../src/services/llm.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAEpE,kEAAkE;AAElE,8CAA8C;AAC9C,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,uBAAuB,CAAC;AAErE,gBAAgB;AAChB,MAAM,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,SAAS,IAAI,iBAAiB,CAAC;AAEjE,mCAAmC;AACnC,MAAM,aAAa,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,QAAQ,EAAE,EAAE,CAAC,CAAC;AAqC9E,kEAAkE;AAElE;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,MAMC;IAED,MAAM,GAAG,GAAG,GAAG,QAAQ,sBAAsB,CAAC;IAE9C,MAAM,IAAI,GAA0B;QAClC,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,aAAa;QACpC,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,UAAU,EAAE,MAAM,CAAC,SAAS;QAC5B,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,GAAG;QACtC,MAAM,EAAE,KAAK;KACd,CAAC;IAEF,uCAAuC;IACvC,iCAAiC;IACjC,MAAM,WAAW,GAAG,OAAO,CAAC;IAC5B,MAAM,IAAI,GAAG,MAAM,cAAc,CAC/B,GAAG,EAAE,CACH,SAAS,CACP,GAAG,EACH;QACE,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;SACnC;QACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;KAC3B,EACD,SAAS,EAAE,SAAS;IACpB,WAAW,CACZ,EACH,CAAC,EACD,IAAI,CACL,CAAC;IAEF,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/C,MAAM,IAAI,KAAK,CAAC,+BAA+B,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC;IAChE,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC;IACjD,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;QAC9C,MAAM,IAAI,KAAK,CAAC,sBAAsB,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC;IACvD,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,UAAU;IAC9B,MAAM,GAAG,GAAG,GAAG,QAAQ,YAAY,CAAC;IAEpC,MAAM,IAAI,GAAG,MAAM,SAAS,CAAoB,GAAG,CAAC,CAAC;IAErD,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC5C,MAAM,IAAI,KAAK,CAAC,eAAe,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;IACvE,CAAC;IAED,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;AACpC,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe;IACnC,IAAI,CAAC;QACH,MAAM,UAAU,EAAE,CAAC;QACnB,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,eAAe;IAC7B,OAAO,aAAa,CAAC;AACvB,CAAC"}

package/dist/services/pipeline.d.ts

@@ -0,0 +1,161 @@
+/**
+ * 流水线编排器
+ *
+ * 将数据获取、LLM 分析、舆情搜索、多角色讨论和最终报告
+ * 组装为可执行的个股分析流水线。
+ *
+ * @module services/pipeline
+ */
+export interface Stage1Result {
+    date: string;
+    records: number;
+}
+export interface Stage2Result {
+    id: number;
+    summary: string;
+    indicators: string;
+    signals: string;
+}
+export interface Stage3Result {
+    id: number;
+    report: string;
+    sources: string[];
+}
+export interface Stage4Result {
+    roles: {
+        id: number;
+        role: string;
+        report: string;
+        round: number;
+        wordCount: number;
+    }[];
+}
+export interface Stage5Result {
+    id: number;
+    summary: string;
+    fullReport: string;
+    roleSummary: string;
+    pipelineId: string;
+}
+export interface PipelineResult {
+    date: string;
+    pipelineId: string;
+    finalReportId: number;
+}
+export interface LocalAnalysisResult {
+    date: string;
+    analysisReportId: number;
+}
+export declare class Pipeline {
+    code: string;
+    sessionId: string;
+    constructor(code: string);
+    /**
+     * Stage 1 — 数据获取
+     *
+     * 从腾讯财经接口获取最近 60 个交易日数据，
+     * 写入 daily_info 表并更新 stock.current_price。
+     *
+     * @returns 最新交易日日期和记录数
+     *
+     * @throws 股票不存在或数据获取失败时抛出
+     */
+    stage1FetchData(): Promise<Stage1Result>;
+    /**
+     * Stage 2 — 客观报告（LLM）
+     *
+     * 基于最近 60 个交易日的行情数据和技术指标，
+     * 调用本地 LLM 生成客观分析报告。
+     *
+     * 报告包含结构化技术指标和文字摘要，
+     * 写入 daily_analysis_report 表并生成向量嵌入。
+     *
+     * @param date - 报告日期（yyyy-MM-dd）
+     * @returns 创建的客观报告信息
+     */
+    stage2ObjectiveReport(date: string): Promise<Stage2Result>;
+    /**
+     * Stage 3 — 舆情获取
+     *
+     * 调用 DashScope API 搜索股票最近三天的新闻和市场舆情，
+     * 将结果写入 sentiment_report 表。
+     *
+     * @param date - 报告日期（yyyy-MM-dd）
+     * @returns 舆情报告信息
+     */
+    stage3Sentiment(date: string): Promise<Stage3Result>;
+    /**
+     * Stage 4 — 多角色分析
+     *
+     * 4 个角色按顺序发言，每轮每人聚焦各自专业领域。
+     * 第一轮（必须）：技术分析师 → 基本面分析师 → 舆情分析师 → 风控官
+     * 第二轮（可选）：风控官 → 舆情分析师 → 基本面分析师 → 技术分析师（回应分歧）
+     *
+     * 每个角色的发言写入 analysis_roler 表。
+     *
+     * @param date  - 分析日期
+     * @param darId - 客观报告 ID
+     * @param srId  - 舆情报告 ID（可为 null）
+     * @returns 所有角色发言记录
+     */
+    stage4MultiRole(date: string, darId: number, srId: number | null): Promise<Stage4Result>;
+    /**
+     * Stage 5 — 最终报告
+     *
+     * 综合客观报告、舆情报告和多角色分析结果，
+     * 调用 LLM 生成包含概述（overview）和完整报告（full）的最终报告。
+     *
+     * 写入 final_report 表并生成向量嵌入。
+     *
+     * @param date - 报告日期
+     * @returns 最终报告信息
+     */
+    stage5FinalReport(date: string): Promise<Stage5Result>;
+    /**
+     * 运行完整流水线（Stage 1→5）。
+     *
+     * 依次执行数据获取、客观报告、
+     * 舆情获取、多角色分析和最终报告。
+     *
+     * @returns 流水线运行结果（日期、流水线 ID、最终报告 ID）
+     *
+     * @example
+     * const result = await new Pipeline('600519').runFull();
+     * console.log(result.date, result.pipelineId);
+     */
+    runFull(): Promise<PipelineResult>;
+    /**
+     * 运行本地分析（Stage 1→2）。
+     *
+     * 仅执行数据获取和客观分析报告生成，跳过舆情和多角色环节。
+     *
+     * @returns 分析结果（日期和报告 ID）
+     *
+     * @example
+     * const result = await new Pipeline('600519').runLocalAnalysis();
+     * console.log(result.date, result.analysisReportId);
+     */
+    runLocalAnalysis(): Promise<LocalAnalysisResult>;
+}
+/**
+ * Stage 1 — 数据获取（向后兼容包装）
+ *
+ * @param code - 六位股票代码
+ * @returns 最新交易日日期和记录数
+ */
+export declare function stage1FetchData(code: string): Promise<Stage1Result>;
+/**
+ * 运行完整流水线（Stage 1→5，向后兼容包装）。
+ *
+ * @param code - 股票代码
+ * @returns 流水线运行结果（日期、流水线 ID、最终报告 ID）
+ */
+export declare function runFullPipeline(code: string): Promise<PipelineResult>;
+/**
+ * 运行本地分析（Stage 1→2，向后兼容包装）。
+ *
+ * @param code - 股票代码
+ * @returns 分析结果（日期和报告 ID）
+ */
+export declare function runLocalAnalysis(code: string): Promise<LocalAnalysisResult>;
+//# sourceMappingURL=pipeline.d.ts.map

package/dist/services/pipeline.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pipeline.d.ts","sourceRoot":"","sources":["../../src/services/pipeline.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAyBH,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,EAAE,CAAC;CACnB;AAED,MAAM,WAAW,YAAY;IAC3B,KAAK,EAAE;QACL,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,MAAM,EAAE,MAAM,CAAC;QACf,KAAK,EAAE,MAAM,CAAC;QACd,SAAS,EAAE,MAAM,CAAC;KACnB,EAAE,CAAC;CACL;AAED,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,MAAM,CAAC;IACb,gBAAgB,EAAE,MAAM,CAAC;CAC1B;AAoUD,qBAAa,QAAQ;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;gBAEN,IAAI,EAAE,MAAM;IAOxB;;;;;;;;;OASG;IACG,eAAe,IAAI,OAAO,CAAC,YAAY,CAAC;IAuC9C;;;;;;;;;;;OAWG;IACG,qBAAqB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IA2JhE;;;;;;;;OAQG;IACG,eAAe,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAmD1D;;;;;;;;;;;;;OAaG;IACG,eAAe,CACnB,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,IAAI,EAAE,MAAM,GAAG,IAAI,GAClB,OAAO,CAAC,YAAY,CAAC;IA2IxB;;;;;;;;;;OAUG;IACG,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAkK5D;;;;;;;;;;;OAWG;IACG,OAAO,IAAI,OAAO,CAAC,cAAc,CAAC;IA+CxC;;;;;;;;;;OAUG;IACG,gBAAgB,IAAI,OAAO,CAAC,mBAAmB,CAAC;CAsBvD;AAID;;;;;GAKG;AACH,wBAAsB,eAAe,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,CAGzE;AAED;;;;;GAKG;AACH,wBAAsB,eAAe,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,CAAC,CAG3E;AAED;;;;;GAKG;AACH,wBAAsB,gBAAgB,CACpC,IAAI,EAAE,MAAM,GACX,OAAO,CAAC,mBAAmB,CAAC,CAG9B"}