@hedgehog-finance/hedgehog-plugin 1.0.19 → 1.0.21

package/src/features/profileLibrary/schema.ts

@@ -0,0 +1,35 @@
+import { z } from "openclaw/plugin-sdk/zod";
+
+export const AddProfileLibraryParamsSchema = z.object({
+	id: z.string().trim().min(1).optional().describe("资料库 ID，不传则自动生成"),
+	title: z.string().trim().min(1).describe("资料库标题")
+});
+export type AddProfileLibraryParams = z.infer<typeof AddProfileLibraryParamsSchema>;
+
+export const DeleteProfileLibraryParamsSchema = z.object({
+	id: z.string().trim().min(1).describe("资料库 ID")
+});
+export type DeleteProfileLibraryParams = z.infer<typeof DeleteProfileLibraryParamsSchema>;
+
+export const GetProfileLibraryByIdParamsSchema = z.object({
+	id: z.string().trim().min(1).describe("资料库 ID")
+});
+export type GetProfileLibraryByIdParams = z.infer<typeof GetProfileLibraryByIdParamsSchema>;
+
+export const QueryProfileLibrariesParamsSchema = z.object({
+	keyword: z.string().trim().optional().describe("模糊查询关键词，可匹配标题或 ID"),
+	page: z.number().int().min(1).optional().describe("页码，从 1 开始，默认 1"),
+	pageSize: z.number().int().min(1).max(100).optional().describe("每页数量，默认 20，最大 100")
+});
+export type QueryProfileLibrariesParams = z.infer<typeof QueryProfileLibrariesParamsSchema>;
+
+export const GetProfileLibrariesParamsSchema = z.object({
+	page: z.number().int().min(1).optional().describe("页码，从 1 开始，默认 1"),
+	pageSize: z.number().int().min(1).max(100).optional().describe("每页数量，默认 20，最大 100")
+});
+export type GetProfileLibrariesParams = z.infer<typeof GetProfileLibrariesParamsSchema>;
+
+export interface ProfileLibraryRow {
+	id: string;
+	title: string;
+}

package/src/features/profileLibrary/tools.ts

@@ -0,0 +1,194 @@
+import { randomUUID } from "node:crypto";
+import { getDB } from "../../core/database.js";
+import {
+	AddProfileLibraryParams,
+	AddProfileLibraryParamsSchema,
+	DeleteProfileLibraryParams,
+	DeleteProfileLibraryParamsSchema,
+	GetProfileLibraryByIdParams,
+	GetProfileLibraryByIdParamsSchema,
+	GetProfileLibrariesParams,
+	GetProfileLibrariesParamsSchema,
+	ProfileLibraryRow,
+	QueryProfileLibrariesParams,
+	QueryProfileLibrariesParamsSchema
+} from "./schema.js";
+
+interface RuntimeTool {
+	name: string;
+	description: string;
+	parameters: unknown;
+	registerTool?: boolean;
+	execute(params: unknown, ctx: { userId: string }): Promise<string>;
+}
+
+function escapeLikePattern(value: string): string {
+	return value.replace(/[\\%_]/g, (char) => `\\${char}`);
+}
+
+function normalizePagination(args: { page?: number; pageSize?: number }) {
+	const page = args.page ?? 1;
+	const pageSize = args.pageSize ?? 20;
+	return {
+		page,
+		pageSize,
+		offset: (page - 1) * pageSize
+	};
+}
+
+function paginatedResponse(rows: ProfileLibraryRow[], page: number, pageSize: number, total: number) {
+	return JSON.stringify({
+		success: true,
+		data: rows,
+		pagination: {
+			page,
+			pageSize,
+			total,
+			totalPages: Math.ceil(total / pageSize)
+		}
+	});
+}
+
+async function getProfileLibraries(args: GetProfileLibrariesParams = {}, ctx: { userId: string }) {
+	try {
+		const db = getDB();
+		const uId = String(ctx.userId);
+		const { page, pageSize, offset } = normalizePagination(args);
+		const total = (db.prepare(`
+			SELECT COUNT(*) AS count
+			FROM profile_libraries
+			WHERE userId = ?
+		`).get(uId) as { count: number }).count;
+		const rows = db.prepare(`
+			SELECT id, title
+			FROM profile_libraries
+			WHERE userId = ?
+			ORDER BY updatedAt DESC, createdAt DESC
+			LIMIT ? OFFSET ?
+		`).all(uId, pageSize, offset) as ProfileLibraryRow[];
+
+		return paginatedResponse(rows, page, pageSize, total);
+	} catch (e: any) {
+		return JSON.stringify({ success: false, error: e.message });
+	}
+}
+
+async function queryProfileLibraries(args: QueryProfileLibrariesParams = {}, ctx: { userId: string }) {
+	try {
+		const db = getDB();
+		const uId = String(ctx.userId);
+		const keyword = args.keyword?.trim();
+		const { page, pageSize, offset } = normalizePagination(args);
+
+		if (keyword) {
+			const pattern = `%${escapeLikePattern(keyword)}%`;
+			const total = (db.prepare(`
+				SELECT COUNT(*) AS count
+				FROM profile_libraries
+				WHERE userId = ?
+				  AND (title LIKE ? ESCAPE '\\' OR id LIKE ? ESCAPE '\\')
+			`).get(uId, pattern, pattern) as { count: number }).count;
+			const rows = db.prepare(`
+				SELECT id, title
+				FROM profile_libraries
+				WHERE userId = ?
+				  AND (title LIKE ? ESCAPE '\\' OR id LIKE ? ESCAPE '\\')
+				ORDER BY updatedAt DESC, createdAt DESC
+				LIMIT ? OFFSET ?
+			`).all(uId, pattern, pattern, pageSize, offset) as ProfileLibraryRow[];
+			return paginatedResponse(rows, page, pageSize, total);
+		}
+
+		return getProfileLibraries(args, ctx);
+	} catch (e: any) {
+		return JSON.stringify({ success: false, error: e.message });
+	}
+}
+
+export const profileLibraryTools: Record<string, RuntimeTool> = {
+	add_profile_library: {
+		name: "add_profile_library",
+		description: "新增个人资料库",
+		parameters: AddProfileLibraryParamsSchema,
+		registerTool: false,
+		execute: async (args: AddProfileLibraryParams, ctx: { userId: string }) => {
+			try {
+				const db = getDB();
+				const uId = String(ctx.userId);
+				const id = args.id?.trim() || randomUUID();
+				const title = args.title.trim();
+
+				db.prepare(`
+					INSERT INTO profile_libraries (id, userId, title)
+					VALUES (?, ?, ?)
+					ON CONFLICT(id, userId) DO UPDATE SET
+						title = excluded.title,
+						updatedAt = STRFTIME('%Y-%m-%dT%H:%M:%fZ', 'NOW')
+				`).run(id, uId, title);
+
+				return JSON.stringify({ success: true, data: { id, title } });
+			} catch (e: any) {
+				return JSON.stringify({ success: false, error: e.message });
+			}
+		}
+	},
+
+	delete_profile_library: {
+		name: "delete_profile_library",
+		description: "删除个人资料库",
+		parameters: DeleteProfileLibraryParamsSchema,
+		registerTool: false,
+		execute: async (args: DeleteProfileLibraryParams, ctx: { userId: string }) => {
+			try {
+				const db = getDB();
+				const uId = String(ctx.userId);
+				const info = db.prepare(`
+					DELETE FROM profile_libraries
+					WHERE id = ? AND userId = ?
+				`).run(args.id.trim(), uId);
+
+				return JSON.stringify({ success: info.changes > 0 });
+			} catch (e: any) {
+				return JSON.stringify({ success: false, error: e.message });
+			}
+		}
+	},
+
+	get_profile_library_by_id: {
+		name: "get_profile_library_by_id",
+		description: "根据 ID 查询个人资料库",
+		parameters: GetProfileLibraryByIdParamsSchema,
+		registerTool: false,
+		execute: async (args: GetProfileLibraryByIdParams, ctx: { userId: string }) => {
+			try {
+				const db = getDB();
+				const uId = String(ctx.userId);
+				const row = db.prepare(`
+					SELECT id, title
+					FROM profile_libraries
+					WHERE id = ? AND userId = ?
+				`).get(args.id.trim(), uId) as ProfileLibraryRow | undefined;
+
+				return JSON.stringify({ success: true, data: row ?? null });
+			} catch (e: any) {
+				return JSON.stringify({ success: false, error: e.message });
+			}
+		}
+	},
+
+	query_profile_libraries: {
+		name: "query_profile_libraries",
+		description: "分页查询个人资料库，支持按标题或 ID 模糊查询",
+		parameters: QueryProfileLibrariesParamsSchema,
+		registerTool: false,
+		execute: queryProfileLibraries
+	},
+
+	get_profile_libraries: {
+		name: "get_profile_libraries",
+		description: "分页获取个人资料库列表",
+		parameters: GetProfileLibrariesParamsSchema,
+		registerTool: false,
+		execute: getProfileLibraries
+	}
+};

package/src/features/stockAnalysis/schema.ts

@@ -0,0 +1,60 @@
+import { z } from "openclaw/plugin-sdk/zod";
+
+export const GetStockAiAnalysisParamsSchema = z.object({
+	stockCode: z.string().trim().min(1).describe("股票代码"),
+	market: z.string().trim().min(1).default("CN").describe("市场类型，默认 CN")
+});
+export type GetStockAiAnalysisParams = z.infer<typeof GetStockAiAnalysisParamsSchema>;
+
+export const QueryStockAiAnalysisHistoryParamsSchema = z.object({
+	stockCode: z.string().trim().min(1).describe("股票代码"),
+	market: z.string().trim().min(1).default("CN").describe("市场类型，默认 CN"),
+	page: z.number().int().min(1).default(1).describe("页码"),
+	pageSize: z.number().int().min(1).max(50).default(20).describe("每页数量")
+});
+export type QueryStockAiAnalysisHistoryParams = z.infer<typeof QueryStockAiAnalysisHistoryParamsSchema>;
+
+export const SaveStockAiAnalysisParamsSchema = z.object({
+	stockCode: z.string().trim().min(1).describe("股票代码"),
+	stockName: z.string().trim().min(1).describe("股票名称"),
+	market: z.string().trim().min(1).default("CN").describe("市场类型，默认 CN"),
+	content: z.string().trim().min(1).describe("AI 分析内容")
+});
+export type SaveStockAiAnalysisParams = z.infer<typeof SaveStockAiAnalysisParamsSchema>;
+
+export interface StockAiAnalysis {
+	id: string;
+	stockCode: string;
+	stockName: string;
+	market: string;
+	content: string;
+	createdAt: string;
+	updatedAt: string;
+}
+
+export const ArticleAiAnalysisKindSchema = z.enum(["verification", "deduction"]);
+
+export const GetArticleAiAnalysisParamsSchema = z.object({
+	id: z.string().trim().min(1).describe("文章来源 ID，例如 news-5、report-5、announce-5"),
+	analysisType: ArticleAiAnalysisKindSchema.describe("分析类型：verification 信息求证，deduction 深度推演"),
+	market: z.string().trim().min(1).default("CN").describe("市场类型，默认 CN")
+});
+export type GetArticleAiAnalysisParams = z.infer<typeof GetArticleAiAnalysisParamsSchema>;
+
+export const SaveArticleAiAnalysisParamsSchema = z.object({
+	id: z.string().trim().min(1).describe("文章来源 ID，例如 news-5、report-5、announce-5"),
+	analysisType: ArticleAiAnalysisKindSchema.describe("分析类型：verification 信息求证，deduction 深度推演"),
+	market: z.string().trim().min(1).default("CN").describe("市场类型，默认 CN"),
+	content: z.string().trim().min(1).describe("AI 分析内容")
+});
+export type SaveArticleAiAnalysisParams = z.infer<typeof SaveArticleAiAnalysisParamsSchema>;
+
+export interface ArticleAiAnalysis {
+	id: string;
+	sourceId: string;
+	analysisType: GetArticleAiAnalysisParams["analysisType"];
+	market: string;
+	content: string;
+	createdAt: string;
+	updatedAt: string;
+}

package/src/features/stockAnalysis/tools.ts

@@ -0,0 +1,192 @@
+import { randomUUID } from "node:crypto";
+import { getDB } from "../../core/database.js";
+import {
+	ArticleAiAnalysis,
+	GetArticleAiAnalysisParamsSchema,
+	GetStockAiAnalysisParamsSchema,
+	QueryStockAiAnalysisHistoryParamsSchema,
+	SaveArticleAiAnalysisParamsSchema,
+	SaveStockAiAnalysisParamsSchema,
+	StockAiAnalysis
+} from "./schema.js";
+
+interface RuntimeTool {
+	name: string;
+	description: string;
+	parameters: unknown;
+	registerTool?: boolean;
+	execute(params: unknown, ctx: { userId: string }): Promise<string>;
+}
+
+export function normalizeStockCode(stockCode: string): string {
+	return stockCode.trim().toUpperCase().replace(/\.SS$/i, ".SH");
+}
+
+function selectLatestStockAnalysis(
+	db: ReturnType<typeof getDB>,
+	userId: string,
+	stockCode: string
+): StockAiAnalysis | undefined {
+	return db.prepare(`
+		SELECT id, stockCode, stockName, market, content, createdAt, updatedAt
+		FROM stock_ai_analysis
+		WHERE userId = ? AND stockCode = ?
+		ORDER BY updatedAt DESC, createdAt DESC
+		LIMIT 1
+	`).get(userId, stockCode) as StockAiAnalysis | undefined;
+}
+
+export function saveStockAiAnalysisRecord(
+	db: ReturnType<typeof getDB>,
+	userId: string,
+	args: {
+		stockCode: string;
+		stockName: string;
+		market: string;
+		content: string;
+	}
+): StockAiAnalysis {
+	const stockCode = normalizeStockCode(args.stockCode);
+	const id = randomUUID();
+
+	db.prepare(`
+		INSERT INTO stock_ai_analysis (id, userId, stockCode, stockName, market, content)
+		VALUES (?, ?, ?, ?, ?, ?)
+	`).run(id, userId, stockCode, args.stockName, args.market, args.content);
+
+	return db.prepare(`
+		SELECT id, stockCode, stockName, market, content, createdAt, updatedAt
+		FROM stock_ai_analysis
+		WHERE userId = ? AND id = ?
+	`).get(userId, id) as StockAiAnalysis;
+}
+
+function selectLatestArticleAnalysis(
+	db: ReturnType<typeof getDB>,
+	userId: string,
+	sourceId: string,
+	analysisType: ArticleAiAnalysis["analysisType"],
+	market: string
+): ArticleAiAnalysis | undefined {
+	return db.prepare(`
+		SELECT id, sourceId, analysisType, market, content, createdAt, updatedAt
+		FROM article_ai_analysis
+		WHERE userId = ? AND sourceId = ? AND analysisType = ? AND market = ?
+		ORDER BY updatedAt DESC, createdAt DESC
+		LIMIT 1
+	`).get(userId, sourceId, analysisType, market) as ArticleAiAnalysis | undefined;
+}
+
+function saveArticleAiAnalysisRecord(
+	db: ReturnType<typeof getDB>,
+	userId: string,
+	args: {
+		sourceId: string;
+		analysisType: ArticleAiAnalysis["analysisType"];
+		market: string;
+		content: string;
+	}
+): ArticleAiAnalysis {
+	const id = randomUUID();
+
+	db.prepare(`
+		INSERT INTO article_ai_analysis (id, sourceId, userId, analysisType, market, content)
+		VALUES (?, ?, ?, ?, ?, ?)
+		ON CONFLICT(sourceId, userId, analysisType, market) DO UPDATE SET
+			content = excluded.content,
+			updatedAt = STRFTIME('%Y-%m-%dT%H:%M:%fZ', 'NOW')
+	`).run(id, args.sourceId, userId, args.analysisType, args.market, args.content);
+
+	return db.prepare(`
+		SELECT id, sourceId, analysisType, market, content, createdAt, updatedAt
+		FROM article_ai_analysis
+		WHERE userId = ? AND sourceId = ? AND analysisType = ? AND market = ?
+	`).get(userId, args.sourceId, args.analysisType, args.market) as ArticleAiAnalysis;
+}
+
+export const stockAnalysisTools: Record<string, RuntimeTool> = {
+	get_stock_ai_analysis: {
+		name: "get_stock_ai_analysis",
+		description: "读取股票 AI 分析的最新一条历史记录；不触发模型分析。",
+		parameters: GetStockAiAnalysisParamsSchema,
+		registerTool: false,
+		async execute(params, ctx) {
+			const args = GetStockAiAnalysisParamsSchema.parse(params);
+			const db = getDB();
+			const data = selectLatestStockAnalysis(db, ctx.userId, normalizeStockCode(args.stockCode));
+			return JSON.stringify({ success: true, data: data || null });
+		}
+	},
+	query_stock_ai_analysis_history: {
+		name: "query_stock_ai_analysis_history",
+		description: "分页读取股票 AI 分析历史记录；不触发模型分析。",
+		parameters: QueryStockAiAnalysisHistoryParamsSchema,
+		registerTool: false,
+		async execute(params, ctx) {
+			const args = QueryStockAiAnalysisHistoryParamsSchema.parse(params);
+			const db = getDB();
+			const stockCode = normalizeStockCode(args.stockCode);
+			const offset = (args.page - 1) * args.pageSize;
+			const rows = db.prepare(`
+				SELECT id, stockCode, stockName, market, content, createdAt, updatedAt
+				FROM stock_ai_analysis
+				WHERE userId = ? AND stockCode = ? AND market = ?
+				ORDER BY updatedAt DESC, createdAt DESC
+				LIMIT ? OFFSET ?
+			`).all(ctx.userId, stockCode, args.market, args.pageSize, offset) as StockAiAnalysis[];
+			const countRow = db.prepare(`
+				SELECT COUNT(*) AS total
+				FROM stock_ai_analysis
+				WHERE userId = ? AND stockCode = ? AND market = ?
+			`).get(ctx.userId, stockCode, args.market) as { total: number };
+			const total = countRow.total || 0;
+
+			return JSON.stringify({
+				success: true,
+				data: rows,
+				pagination: {
+					page: args.page,
+					pageSize: args.pageSize,
+					total,
+					totalPages: Math.ceil(total / args.pageSize)
+				}
+			});
+		}
+	},
+	save_stock_ai_analysis: {
+		name: "save_stock_ai_analysis",
+		description: "追加保存一条股票 AI 分析历史记录。",
+		parameters: SaveStockAiAnalysisParamsSchema,
+		registerTool: false,
+		async execute(params, ctx) {
+			const args = SaveStockAiAnalysisParamsSchema.parse(params);
+			const db = getDB();
+			const data = saveStockAiAnalysisRecord(db, ctx.userId, args);
+			return JSON.stringify({ success: true, data });
+		}
+	},
+	get_article_ai_analysis: {
+		name: "get_article_ai_analysis",
+		description: "读取文章 AI 分析结果；支持信息求证与深度推演，不触发模型分析。",
+		parameters: GetArticleAiAnalysisParamsSchema,
+		registerTool: false,
+		async execute(params, ctx) {
+			const args = GetArticleAiAnalysisParamsSchema.parse(params);
+			const db = getDB();
+			const data = selectLatestArticleAnalysis(db, ctx.userId, args.id, args.analysisType, args.market);
+			return JSON.stringify({ success: true, data: data || null });
+		}
+	},
+	save_article_ai_analysis: {
+		name: "save_article_ai_analysis",
+		description: "保存文章 AI 分析结果；支持信息求证与深度推演。",
+		parameters: SaveArticleAiAnalysisParamsSchema,
+		registerTool: false,
+		async execute(params, ctx) {
+			const args = SaveArticleAiAnalysisParamsSchema.parse(params);
+			const db = getDB();
+			const data = saveArticleAiAnalysisRecord(db, ctx.userId, { ...args, sourceId: args.id });
+			return JSON.stringify({ success: true, data });
+		}
+	}
+};

package/src/features/watchlist/logic.ts

@@ -3,7 +3,7 @@ import { completeSimple } from "@mariozechner/pi-ai";
 import { PluginRuntime } from "openclaw/plugin-sdk";
 import {
 	extractAssistantText,
-	prepareSimpleCompletionModel
+	prepareSimpleCompletionModelForAgent
 } from "openclaw/plugin-sdk/simple-completion-runtime";
 import { getDB } from "../../core/database.js";
 import { logger } from "../../core/logger.js";
@@ -32,44 +32,6 @@ const CLASSIFIER_SYSTEM_PROMPT = [
 	"只允许根据用户消息中提供的行业/主题分类字典和股票列表输出纯 JSON 数组。"
 ].join("\n");
 
-function resolveModelRef(modelConfig: unknown): string | undefined {
-	if (typeof modelConfig === "string" && modelConfig.trim()) return modelConfig.trim();
-	if (!modelConfig || typeof modelConfig !== "object" || Array.isArray(modelConfig)) return undefined;
-	const primary = (modelConfig as { primary?: unknown }).primary;
-	return typeof primary === "string" && primary.trim() ? primary.trim() : undefined;
-}
-
-function getConfiguredProviderConfig(cfg: any, provider: string): any | undefined {
-	const providerConfigs = cfg.models?.providers;
-	if (!providerConfigs) return undefined;
-	const exact = providerConfigs[provider];
-	if (exact) return exact;
-	const normalized = provider.trim().toLowerCase();
-	return Object.entries(providerConfigs).find(([key]) => key.trim().toLowerCase() === normalized)?.[1];
-}
-
-function resolveClassifierModelSelection(
-	cfg: any,
-	defaultProvider: string,
-	defaultModel: string
-): { provider: string; model: string } {
-	const agentEntry = ((cfg.agents?.list || []) as any[]).find((agent) => agent?.id === MAIN_AGENT_ID);
-	const primary = resolveModelRef(agentEntry?.model)
-		|| resolveModelRef(cfg.agents?.defaults?.model)
-		|| `${defaultProvider}/${defaultModel}`;
-	const slash = primary.indexOf("/");
-	if (slash > 0) {
-		return {
-			provider: primary.slice(0, slash),
-			model: primary.slice(slash + 1)
-		};
-	}
-	return {
-		provider: defaultProvider,
-		model: primary
-	};
-}
-
 function normalizeStockCodeForCache(stockCode: string, exchange?: string): string {
 	const code = String(stockCode || "")
 		.trim()
@@ -445,17 +407,7 @@ export const watchlistLogic = {
 
 	async _callClassifierCompletion(rt: PluginRuntime, sessionId: string, prompt: string, timeoutMs: number): Promise<string> {
 		const cfg = rt.config.loadConfig();
-		const { provider, model } = resolveClassifierModelSelection(cfg, rt.agent.defaults.provider, rt.agent.defaults.model);
 		const maxTokens = resolveClassifierOutputMaxTokens(prompt);
-		const providerAuth = await rt.modelAuth.resolveApiKeyForProvider({ provider, cfg });
-		if (!providerAuth.apiKey) {
-			throw new Error(`No API key found for provider "${provider}".`);
-		}
-		const providerConfigs = cfg.models?.providers;
-		const providerConfig = getConfiguredProviderConfig(cfg, provider);
-		if (!providerConfig) {
-			throw new Error(`No model provider config found for "${provider}".`);
-		}
 		const embeddedCfg = {
 			...cfg,
 			agents: {
@@ -464,27 +416,19 @@ export const watchlistLogic = {
 					...cfg.agents?.defaults,
 					systemPromptOverride: CLASSIFIER_SYSTEM_PROMPT
 				}
-			},
-			models: {
-				...cfg.models,
-				providers: {
-					...providerConfigs,
-					[provider]: {
-						...providerConfig,
-						auth: "api-key" as const,
-						apiKey: providerAuth.apiKey
-					}
-				}
 			}
 		};
-		const prepared = await prepareSimpleCompletionModel({
+		const prepared = await prepareSimpleCompletionModelForAgent({
 			cfg: embeddedCfg,
-			provider,
-			modelId: model
+			agentId: MAIN_AGENT_ID,
+			allowBundledStaticCatalogFallback: true
 		});
 		if ("error" in prepared) {
 			throw new Error(prepared.error);
 		}
+		if (!prepared.auth.apiKey) {
+			throw new Error(`No API key found for provider "${prepared.selection.provider}".`);
+		}
 		const abortController = new AbortController();
 		const abortTimer = setTimeout(() => {
 			abortController.abort(new Error(`classifier completion timed out after ${timeoutMs}ms`));