neuroalpha 0.0.0-stage → 0.2.0

package/server/strategist/tools.js

@@ -0,0 +1,302 @@
+import { PARAMS, PARAM_MAP } from '../params-meta.js';
+import { getSettings, updateSettings, validateChanges } from '../settings.js';
+import { getRun, listRuns, addJournal, listJournal, kvGet, kvSet } from '../db.js';
+import { analyzeRun, compareRuns, recentRuns, runSummary, compactMetrics } from './analytics.js';
+import { hub } from '../hub.js';
+import { startModelResearch } from '../model/research.js';
+import { forecastNow } from '../model/forecast.js';
+import { live } from '../model/live.js';
+import { listNews } from '../news/index.js';
+import { plural, fmtBarMsk } from '../util.js';
+
+export const OOS_LIMIT = 3;
+const AI_LOCKED = PARAMS.filter((p) => p.ai === false).map((p) => p.key);
+
+const changesSchema = {
+  type: 'object',
+  description: 'Изменения настроек: ключ — точное название настройки из списка в системной инструкции (например modelHorizon, modelMinEdge), значение — новое значение.',
+  properties: Object.fromEntries(PARAMS.filter((p) => p.ai !== false).map((p) => [p.key, {
+    type: p.type === 'checkbox' ? 'boolean' : p.type === 'select' ? 'string' : 'number',
+    description: `${p.label}${p.min !== undefined ? ` (${p.min}–${p.max})` : ''}`,
+  }])),
+  additionalProperties: false,
+};
+
+function checkChanges(changes) {
+  const locked = Object.keys(changes || {}).filter((k) => AI_LOCKED.includes(k));
+  const { values, errors } = validateChanges(changes);
+  for (const k of locked) { errors.push(`${k}: эту настройку ИИ менять нельзя (она отвечает за честность проверки)`); delete values[k]; }
+  return { values, errors };
+}
+
+const describeChanges = (values) => Object.entries(values).map(([k, v]) => `${PARAM_MAP[k]?.label || k}: ${v}`).join('; ') || 'без изменений';
+const fmtPct = (v) => (v === null || v === undefined ? '—' : `${v > 0 ? '+' : ''}${v}%`);
+const metricLine = (m) => (m && !m.trades ? 'сделок не было' : m ? `${m.trades} ${plural(m.trades, 'сделка', 'сделки', 'сделок')}, итог ${fmtPct(m.returnPct)}, на 1 ₽ убытка ${m.profitFactor ?? '—'} ₽, худшее падение ${m.maxDrawdownPct}%` : 'нет данных');
+
+// Последнее исследование с текущими настройками (база для сравнения)
+function findBaseline(excludeId) {
+  const cur = getSettings();
+  const keys = PARAMS.map((p) => p.key).filter((k) => !['capital', 'paperDailyReview', 'paperPollSec', 'autopilotBudget'].includes(k));
+  return listRuns().find((r) => r.id !== excludeId && r.mode === 'model' && r.status === 'done'
+    && keys.every((k) => JSON.stringify(r.params?.[k] ?? null) === JSON.stringify(cur[k] ?? null))) || null;
+}
+
+// kind: read — без подтверждения; write — подтверждение в режиме «Советник»; danger — подтверждение всегда;
+// confirmAlways — подтверждение во всех режимах (кроме read-режимов, где инструмент недоступен)
+export const TOOLS = {
+  get_settings: {
+    kind: 'read',
+    label: 'Смотрит текущие настройки',
+    description: 'Текущие настройки модели и счёта с пояснениями.',
+    parameters: { type: 'object', properties: {} },
+    async run() {
+      const s = getSettings();
+      return {
+        result: Object.fromEntries(PARAMS.map((p) => [p.key, { значение: s[p.key], что_это: p.label, пояснение: p.hint, ...(p.ai === false ? { менять_нельзя: true } : {}) }])),
+        summary: 'Прочитал текущие настройки',
+      };
+    },
+  },
+
+  list_runs: {
+    kind: 'read',
+    label: 'Смотрит список исследований',
+    description: 'Последние исследования модели и живые проверки с результатами на периоде настройки.',
+    parameters: { type: 'object', properties: { limit: { type: 'number', description: 'сколько последних, по умолчанию 10' } } },
+    async run({ limit = 10 }) {
+      return { result: recentRuns(Math.min(30, limit)), summary: `Просмотрел последние исследования` };
+    },
+  },
+
+  analyze_run: {
+    kind: 'read',
+    label: 'Разбирает исследование',
+    description: 'Подробный разбор исследования: сравнение моделей, сделки торгующей модели по акциям, часу входа, сроку удержания, веса голов внимания. Только период настройки.',
+    parameters: { type: 'object', properties: { run_id: { type: 'number', description: 'номер исследования' } }, required: ['run_id'] },
+    async run({ run_id }) {
+      const r = analyzeRun(Number(run_id));
+      return { result: r, summary: r.ошибка || `Разобрал исследование №${run_id}` };
+    },
+  },
+
+  compare_runs: {
+    kind: 'read',
+    label: 'Сравнивает исследования',
+    description: 'Сравнить два исследования: чем отличаются настройки и результаты на периоде настройки.',
+    parameters: { type: 'object', properties: { a: { type: 'number' }, b: { type: 'number' } }, required: ['a', 'b'] },
+    async run({ a, b }) {
+      return { result: compareRuns(Number(a), Number(b)), summary: `Сравнил исследования №${a} и №${b}` };
+    },
+  },
+
+  run_model_research: {
+    kind: 'write',
+    budget: true,
+    label: 'Исследует модель',
+    description: 'Пошаговая проверка трёх моделей предсказания «следующего состояния рынка» (~100 акций, часовые свечи): своя инерция, кто за кем ходит, внимание через карту рынка (и трансформер, если modelUseTransformer=true — тогда в разы дольше). Модель учится на окне прошлого, прогнозирует следующий месяц, окно сдвигается. Возвращает сравнение моделей на периоде настройки и базу для сравнения. Можно передать изменения настроек (сами настройки не меняются). Около минуты.',
+    parameters: { type: 'object', properties: { changes: changesSchema, note: { type: 'string', description: 'что проверяем (гипотеза)' } }, required: ['note'] },
+    preview(args) {
+      const { values, errors } = checkChanges(args.changes);
+      return `${args.note || 'Исследование модели'}\nИзменения: ${describeChanges(values)}${errors.length ? `\nОшибки: ${errors.join('; ')}` : ''}`;
+    },
+    async run(args, ctx) {
+      const { values, errors } = checkChanges(args.changes);
+      if (errors.length) return { result: { ошибка: errors }, summary: `Ошибка в изменениях: ${errors.join('; ')}`, ok: false };
+      const { runId, done, reused } = startModelResearch({ ...getSettings(), ...values }, { source: 'ai', note: args.note || '' });
+      ctx.onRunStart?.(runId);
+      try { await done; } catch (e) {
+        return { result: { исследование: runId, ошибка: e.message }, summary: `Исследование №${runId}: ${e.message}`, ok: false };
+      }
+      const run = getRun(runId);
+      const base = findBaseline(runId);
+      const c = run.metrics?.model?.comparison?.attention?.inSample;
+      return {
+        result: {
+          ...runSummary(run),
+          изменения: values,
+          ...(reused ? { повтор: `результат взят из исследования №${runId} с теми же настройками` } : {}),
+          база_для_сравнения: base ? runSummary(base) : 'нет исследования с текущими настройками без изменений — для честного сравнения запусти run_model_research без changes',
+        },
+        summary: reused
+          ? `Исследование №${runId} (готовый результат): внимание через карту — совпадение ${c?.ic}, итог ${fmtPct(c?.returnPct)} (без комиссий ${fmtPct(c?.grossReturnPct)}), ${c?.trades} сделок (период настройки)`
+          : `Исследование №${runId}: внимание через карту — совпадение ${c?.ic}, итог ${fmtPct(c?.returnPct)} (без комиссий ${fmtPct(c?.grossReturnPct)}), ${c?.trades} сделок (период настройки)`,
+        runId,
+      };
+    },
+  },
+
+  model_forecast: {
+    kind: 'read',
+    label: 'Смотрит прогноз на ближайшие часы',
+    description: 'Прогноз модели по последней закрытой свече: какие акции, по её мнению, будут сильнее или слабее рынка в ближайшие часы и под чьим влиянием (соседи по карте).',
+    parameters: { type: 'object', properties: { top: { type: 'number', description: 'сколько акций показать, по умолчанию 10' } } },
+    async run({ top = 10 }) {
+      const r = await forecastNow(getSettings());
+      const pick = (x) => ({ акция: x.symbol, ожидание_проц: x.expectedPct, сигнал_покупки: !!x.signal, ...(x.shortSignal !== undefined ? { сигнал_продажи: !!x.shortSignal } : {}), ликвидна: x.liquid, последний_час_проц: x.lastMovePct, влияют: x.influences.map((n) => `${n.symbol} (${n.contribPct > 0 ? '+' : ''}${n.contribPct}%, близость ${n.closeness})`), своё_движение_проц: x.selfPct });
+      return {
+        result: { свеча: r.bar, горизонт_часов: r.horizon, порог_покупки_проц: r.thresholdPct, сильнее_рынка: r.rows.slice(0, top).map(pick), слабее_рынка: r.rows.slice(-Math.min(5, top)).map(pick) },
+        summary: `Прогноз на свече ${r.bar}: сигналов покупки ${r.rows.filter((x) => x.signal).length}`,
+      };
+    },
+  },
+
+  market_neighbors: {
+    kind: 'read',
+    label: 'Смотрит соседей на карте',
+    description: 'Ближайшие соседи акции на карте рынка (кто движется похоже) и её группа — по последнему исследованию.',
+    parameters: { type: 'object', properties: { symbol: { type: 'string' } }, required: ['symbol'] },
+    async run({ symbol }) {
+      const run = listRuns().find((r) => r.mode === 'model' && r.status === 'done');
+      const ex = run && getRun(run.id).extra;
+      if (!ex) return { result: { ошибка: 'нет исследования — запусти run_model_research' }, summary: 'Нет исследования', ok: false };
+      const i = ex.symbols.indexOf(String(symbol).toUpperCase());
+      if (i < 0) return { result: { ошибка: `${symbol} нет в словаре модели` }, summary: `${symbol} нет в словаре`, ok: false };
+      const snap = ex.map.snapshots.at(-1);
+      return {
+        result: { акция: ex.symbols[i], соседи: ex.neighbors[i].map(([k, c]) => ({ акция: ex.symbols[k], близость: c })), группа_на_карте: ex.symbols.filter((_, k) => snap.cl[k] === snap.cl[i]) },
+        summary: `Соседи ${ex.symbols[i]}: ${ex.neighbors[i].slice(0, 4).map(([k]) => ex.symbols[k]).join(', ')}`,
+      };
+    },
+  },
+
+  get_news: {
+    kind: 'read',
+    label: 'Смотрит свежие новости',
+    description: 'Свежие новости и раскрытия эмитентов: каких акций из словаря касаются, тон −1…1 (плохо/хорошо для акции), важность 0…1 и короткая суть. Без symbol — все важные новости за период. Новости есть только «сейчас»: в исторических проверках они не используются.',
+    parameters: {
+      type: 'object',
+      properties: {
+        symbol: { type: 'string', description: 'тикер, например SBER; без него — все важные новости' },
+        hours: { type: 'number', description: 'за сколько часов смотреть, по умолчанию 24' },
+      },
+    },
+    async run({ symbol, hours = 24 }) {
+      const items = listNews({ symbol, hours, limit: 25 });
+      if (!items.length) return { result: { новостей: 'нет' }, summary: `Новостей ${symbol ? `по ${symbol}` : ''} за последние ${hours} ч нет` };
+      return {
+        result: items.map((n) => ({
+          время: fmtBarMsk(n.ts), источник: n.src,
+          ...(n.symbols.length ? { акции: n.symbols.map((s) => `${s.symbol} (тон ${s.tone ?? '—'})`) } : {}),
+          ...(n.imp !== null ? { важность: n.imp } : {}), суть: n.gist || n.title, ссылка: n.link,
+        })),
+        summary: `Новости за ${hours} ч: ${items.length}${symbol ? ` по ${symbol}` : ''}`,
+      };
+    },
+  },
+
+  journal_list: {
+    kind: 'read',
+    label: 'Читает журнал гипотез',
+    description: 'Полный журнал гипотез: что уже проверяли и к каким выводам пришли.',
+    parameters: { type: 'object', properties: {} },
+    async run() {
+      return { result: listJournal(100), summary: 'Прочитал журнал гипотез' };
+    },
+  },
+
+  journal_add: {
+    kind: 'read',
+    label: 'Записывает вывод в журнал',
+    description: 'Записать в журнал гипотез результат эксперимента. Делай это после каждого законченного эксперимента.',
+    parameters: {
+      type: 'object',
+      properties: {
+        hypothesis: { type: 'string', description: 'что предполагали, простыми словами' },
+        change: { type: 'string', description: 'что меняли' },
+        run_ids: { type: 'array', items: { type: 'number' }, description: 'номера исследований: базовое и новое' },
+        result: { type: 'string', description: 'что получилось, с цифрами и числом сделок' },
+        verdict: { type: 'string', enum: ['confirmed', 'rejected', 'unclear'], description: 'подтвердилась / не подтвердилась / неясно' },
+      },
+      required: ['hypothesis', 'result', 'verdict'],
+    },
+    async run(args) {
+      const id = addJournal({ ...args, source: 'ai' });
+      hub.emit({ type: 'journal:update' });
+      const v = { confirmed: 'подтвердилась', rejected: 'не подтвердилась', unclear: 'неясно' }[args.verdict] || args.verdict;
+      return { result: { ok: true, id }, summary: `Записал в журнал: «${args.hypothesis}» — ${v}` };
+    },
+  },
+
+  final_check: {
+    kind: 'read',
+    label: 'Смотрит контрольный период',
+    description: `Открыть результат исследования на КОНТРОЛЬНОМ периоде (данные, которые не используются для подбора). Не больше ${OOS_LIMIT} раз за разговор — только для финальной оценки лучшего варианта против базы.`,
+    parameters: { type: 'object', properties: { run_id: { type: 'number' } }, required: ['run_id'] },
+    async run({ run_id }, ctx) {
+      const key = `oos_used_${ctx.convId}`;
+      const used = kvGet(key, []);
+      if (!used.includes(Number(run_id)) && used.length >= OOS_LIMIT) {
+        return { result: { ошибка: `лимит ${OOS_LIMIT} просмотров контрольного периода за разговор исчерпан` }, summary: 'Лимит просмотров контрольного периода исчерпан', ok: false };
+      }
+      const run = getRun(Number(run_id));
+      if (!run) return { result: { ошибка: 'нет такого исследования' }, summary: 'Нет такого исследования', ok: false };
+      if (!used.includes(Number(run_id))) kvSet(key, [...used, Number(run_id)]);
+      return {
+        result: { ...runSummary(run, { revealOOS: true }), осталось_просмотров: OOS_LIMIT - kvGet(key, []).length },
+        summary: `Контрольный период №${run_id}: ${metricLine(run.metrics?.outOfSample)}`,
+      };
+    },
+  },
+
+  set_settings: {
+    kind: 'write',
+    confirmAlways: true,
+    label: 'Меняет настройки',
+    description: 'Сохранить изменения в рабочие настройки (их видит пользователь и использует живая проверка). Только когда изменение подтверждено исследованиями.',
+    parameters: { type: 'object', properties: { changes: changesSchema, reason: { type: 'string', description: 'почему' } }, required: ['changes', 'reason'] },
+    preview(args) {
+      const { values, errors } = checkChanges(args.changes);
+      return `${args.reason || ''}\nИзменения: ${describeChanges(values)}${errors.length ? `\nОшибки: ${errors.join('; ')}` : ''}`;
+    },
+    async run(args) {
+      const { values, errors } = checkChanges(args.changes);
+      const r = updateSettings(values, { source: 'ai', reason: args.reason });
+      const all = [...errors, ...r.errors];
+      return {
+        result: { изменено: r.changed, ошибки: all.length ? all : undefined },
+        summary: Object.keys(r.changed).length ? `Настройки изменены: ${describeChanges(Object.fromEntries(Object.entries(r.changed).map(([k, v]) => [k, v.стало])))}` : `Настройки не изменились${all.length ? `: ${all.join('; ')}` : ''}`,
+        ok: !all.length,
+      };
+    },
+  },
+
+  paper_start: {
+    kind: 'danger',
+    label: 'Запускает живую проверку',
+    description: 'Запустить живую проверку модели (виртуальная торговля на текущих ценах) с текущими настройками.',
+    parameters: { type: 'object', properties: { reason: { type: 'string' } } },
+    preview: (args) => args.reason || 'Запустить живую проверку с текущими настройками',
+    async run() {
+      if (live.status().running) return { result: { ошибка: 'уже запущена' }, summary: 'Живая проверка уже запущена', ok: false };
+      const runId = await live.start(getSettings(), { source: 'ai' });
+      return { result: { ok: true, проверка: runId }, summary: `Живая проверка №${runId} запущена` };
+    },
+  },
+
+  paper_stop: {
+    kind: 'danger',
+    label: 'Останавливает живую проверку',
+    description: 'Остановить живую проверку.',
+    parameters: { type: 'object', properties: { reason: { type: 'string' } } },
+    preview: (args) => args.reason || 'Остановить живую проверку',
+    async run() {
+      const st = live.status();
+      if (!st.running) return { result: { ошибка: 'не запущена' }, summary: 'Живая проверка не запущена', ok: false };
+      live.stop();
+      return { result: { ok: true, итог: compactMetrics(getRun(st.runId)?.metrics?.full) }, summary: `Живая проверка №${st.runId} остановлена` };
+    },
+  },
+};
+
+export const MODES = {
+  discuss: { label: 'Обсуждение', allowed: (t) => t.kind === 'read' },
+  advisor: { label: 'Советник', allowed: () => true, confirm: (t) => t.kind !== 'read' },
+  autopilot: { label: 'Автопилот', allowed: () => true, confirm: (t) => t.kind === 'danger' || !!t.confirmAlways },
+  review: { label: 'Разбор дня', allowed: (t) => t.kind === 'read' },
+};
+
+export function toolSchemas(mode) {
+  return Object.entries(TOOLS)
+    .filter(([, t]) => MODES[mode].allowed(t))
+    .map(([name, t]) => ({ type: 'function', function: { name, description: t.description, parameters: t.parameters } }));
+}

package/server/util.js

@@ -0,0 +1,9 @@
+export const DAY = 86400000;
+export const MSK = 3 * 3600000;
+export const round = (v, d = 2) => (Number.isFinite(v) ? Math.round(v * 10 ** d) / 10 ** d : v);
+export const fmtDayMsk = (ts) => new Date(ts + MSK).toISOString().slice(0, 10).split('-').reverse().join('.');
+export const fmtBarMsk = (ts) => new Date(ts + MSK).toISOString().slice(0, 16).replace('T', ' ');
+export const plural = (n, one, few, many) => {
+  const m10 = n % 10, m100 = n % 100;
+  return m10 === 1 && m100 !== 11 ? one : m10 >= 2 && m10 <= 4 && (m100 < 10 || m100 >= 20) ? few : many;
+};