HadTavern 0.01: Gemini/Claude fixes; UI _origId reuse; docs; .bat open
This commit is contained in:
247
static/js/serialization.js
Normal file
247
static/js/serialization.js
Normal file
@@ -0,0 +1,247 @@
|
||||
/* global window, document, requestAnimationFrame */
|
||||
// AgentUI serialization helpers extracted from editor.html
|
||||
// Exposes window.AgentUISer.{toPipelineJSON, fromPipelineJSON}
|
||||
// Depends on globals defined by editor.html: editor, NODE_IO, addNode, applyNodeDefaults, status
|
||||
(function (w) {
|
||||
'use strict';
|
||||
|
||||
function ensureDeps() {
|
||||
if (!w || !w.editor) throw new Error('AgentUISer: global editor is not available');
|
||||
if (!w.NODE_IO) throw new Error('AgentUISer: global NODE_IO is not available');
|
||||
}
|
||||
|
||||
// Drawflow -> pipeline JSON
|
||||
function toPipelineJSON() {
|
||||
ensureDeps();
|
||||
const editor = w.editor;
|
||||
const NODE_IO = w.NODE_IO;
|
||||
|
||||
const data = editor.export();
|
||||
const nodes = [];
|
||||
const idMap = {}; // drawflow id -> generated id like n1, n2
|
||||
|
||||
const dfNodes = (data && data.drawflow && data.drawflow.Home && data.drawflow.Home.data) ? data.drawflow.Home.data : {};
|
||||
|
||||
// 1) Собираем ноды
|
||||
let idx = 1;
|
||||
for (const id in dfNodes) {
|
||||
const df = dfNodes[id];
|
||||
const genId = `n${idx++}`;
|
||||
idMap[id] = genId;
|
||||
const el = document.querySelector(`#node-${id}`);
|
||||
// Берём источник правды из DOM.__data (куда жмём «Сохранить параметры») или из drawflow.data
|
||||
const datacopySrc = el && el.__data ? el.__data : (df.data || {});
|
||||
const datacopy = typeof w.applyNodeDefaults === 'function'
|
||||
? w.applyNodeDefaults(df.name, JSON.parse(JSON.stringify(datacopySrc)))
|
||||
: (JSON.parse(JSON.stringify(datacopySrc)));
|
||||
nodes.push({
|
||||
id: genId,
|
||||
type: df.name,
|
||||
pos_x: df.pos_x,
|
||||
pos_y: df.pos_y,
|
||||
config: datacopy,
|
||||
in: {}
|
||||
});
|
||||
}
|
||||
|
||||
// 2) Восстанавливаем связи по входам (inputs)
|
||||
// В Drawflow v0.0.55 inputs/outputs — это объекты вида input_1/output_1
|
||||
for (const id in dfNodes) {
|
||||
const df = dfNodes[id];
|
||||
const targetNode = nodes.find(n => n.id === idMap[id]);
|
||||
if (!targetNode) continue;
|
||||
const io = NODE_IO[targetNode.type] || { inputs: [], outputs: [] };
|
||||
|
||||
for (let i = 0; i < io.inputs.length; i++) {
|
||||
const inputKey = `input_${i + 1}`;
|
||||
const input = df.inputs && df.inputs[inputKey];
|
||||
if (!input || !Array.isArray(input.connections) || input.connections.length === 0) continue;
|
||||
|
||||
// Собираем все связи этого входа и сохраняем строку либо массив строк (для depends поддерживаем мульти-коннекты)
|
||||
const refs = [];
|
||||
for (const conn of (input.connections || [])) {
|
||||
if (!conn) continue;
|
||||
const sourceDfId = String(conn.node);
|
||||
const outKey = String(conn.output ?? '');
|
||||
|
||||
// conn.output может быть "output_1", "1" (строкой), либо числом 1
|
||||
let sourceOutIdx = -1;
|
||||
let m = outKey.match(/output_(\d+)/);
|
||||
if (m) {
|
||||
sourceOutIdx = parseInt(m[1], 10) - 1;
|
||||
} else if (/^\d+$/.test(outKey)) {
|
||||
sourceOutIdx = parseInt(outKey, 10) - 1;
|
||||
} else if (typeof conn.output === 'number') {
|
||||
sourceOutIdx = conn.output - 1;
|
||||
}
|
||||
if (!(sourceOutIdx >= 0)) sourceOutIdx = 0; // safety
|
||||
|
||||
const sourceNode = nodes.find(n => n.id === idMap[sourceDfId]);
|
||||
if (!sourceNode) continue;
|
||||
const sourceIo = NODE_IO[sourceNode.type] || { outputs: [] };
|
||||
const sourceOutName = (sourceIo.outputs && sourceIo.outputs[sourceOutIdx] != null)
|
||||
? sourceIo.outputs[sourceOutIdx]
|
||||
: `out${sourceOutIdx}`;
|
||||
refs.push(`${sourceNode.id}.${sourceOutName}`);
|
||||
}
|
||||
|
||||
// Каноничное имя входа: по NODE_IO, иначе in{0-based}
|
||||
const targetInName = (io.inputs && io.inputs[i] != null)
|
||||
? io.inputs[i]
|
||||
: `in${i}`;
|
||||
|
||||
if (!targetNode.in) targetNode.in = {};
|
||||
targetNode.in[targetInName] = (refs.length <= 1 ? refs[0] : refs);
|
||||
}
|
||||
}
|
||||
|
||||
return { id: 'pipeline_editor', name: 'Edited Pipeline', nodes };
|
||||
}
|
||||
|
||||
// pipeline JSON -> Drawflow
|
||||
async function fromPipelineJSON(p) {
|
||||
ensureDeps();
|
||||
const editor = w.editor;
|
||||
const NODE_IO = w.NODE_IO;
|
||||
|
||||
editor.clear();
|
||||
let x = 100; let y = 120; // Fallback
|
||||
const idMap = {}; // pipeline id -> drawflow id
|
||||
const logs = [];
|
||||
const $ = (sel) => document.querySelector(sel);
|
||||
|
||||
const resolveOutIdx = (type, outName) => {
|
||||
const outs = ((NODE_IO[type] && NODE_IO[type].outputs) || []);
|
||||
let idx = outs.indexOf(outName);
|
||||
if (idx < 0 && typeof outName === 'string') {
|
||||
// поддержка: out-1, out_1, output_1, out1, out0
|
||||
const s = String(outName);
|
||||
let m = s.match(/^out(?:put)?[_-]?(\d+)$/);
|
||||
if (m) {
|
||||
const n = parseInt(m[1], 10);
|
||||
idx = n > 0 ? n - 1 : 0;
|
||||
} else {
|
||||
m = s.match(/^out(\d+)$/); // совместимость со старым out0
|
||||
if (m) idx = parseInt(m[1], 10) | 0;
|
||||
}
|
||||
}
|
||||
return idx;
|
||||
};
|
||||
const resolveInIdx = (type, inName) => {
|
||||
const ins = ((NODE_IO[type] && NODE_IO[type].inputs) || []);
|
||||
let idx = ins.indexOf(inName);
|
||||
if (idx < 0 && typeof inName === 'string') {
|
||||
// поддержка: in-1, in_1, in1, in0
|
||||
const s = String(inName);
|
||||
let m = s.match(/^in[_-]?(\d+)$/);
|
||||
if (m) {
|
||||
const n = parseInt(m[1], 10);
|
||||
idx = n > 0 ? n - 1 : 0;
|
||||
} else {
|
||||
m = s.match(/^in(\d+)$/); // совместимость со старым in0
|
||||
if (m) idx = parseInt(m[1], 10) | 0;
|
||||
}
|
||||
}
|
||||
return idx;
|
||||
};
|
||||
|
||||
// Ожидание появления порта в DOM (устранение гонки рендера)
|
||||
async function waitForPort(dfid, kind, idx, tries = 60, delay = 16) {
|
||||
// Drawflow создаёт DOM-узел с id="node-${dfid}"
|
||||
const sel = `#node-${dfid} .${kind}_${idx}`;
|
||||
for (let i = 0; i < tries; i++) {
|
||||
if ($(sel)) return true;
|
||||
await new Promise(r => setTimeout(r, delay));
|
||||
}
|
||||
logs.push(`port missing: #${dfid} ${kind}_${idx}`);
|
||||
return false;
|
||||
}
|
||||
// Повторные попытки соединить порты, пока DOM не готов
|
||||
async function connectWithRetry(srcDfId, tgtDfId, outNum, inNum, tries = 120, delay = 25) {
|
||||
const outClass = `output_${outNum}`;
|
||||
const inClass = `input_${inNum}`;
|
||||
for (let i = 0; i < tries; i++) {
|
||||
const okOut = await waitForPort(srcDfId, 'output', outNum, 1, delay);
|
||||
const okIn = await waitForPort(tgtDfId, 'input', inNum, 1, delay);
|
||||
if (okOut && okIn) {
|
||||
try {
|
||||
editor.addConnection(srcDfId, tgtDfId, outClass, inClass);
|
||||
return true;
|
||||
} catch (e) {
|
||||
// retry on next loop
|
||||
}
|
||||
}
|
||||
await new Promise(r => setTimeout(r, delay));
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
// 1) Создаём ноды
|
||||
for (const n of p.nodes) {
|
||||
const pos = { x: n.pos_x || x, y: n.pos_y || y };
|
||||
const data = { ...(n.config || {}), _origId: n.id };
|
||||
const dfid = typeof w.addNode === 'function'
|
||||
? w.addNode(n.type, pos, data)
|
||||
: (function () { throw new Error('AgentUISer: addNode() is not defined'); })();
|
||||
idMap[n.id] = dfid;
|
||||
if (!n.pos_x) x += 260; // раскладываем по горизонтали, если нет сохраненной позиции
|
||||
}
|
||||
|
||||
// 2) Дождёмся полного рендера DOM
|
||||
await new Promise(r => setTimeout(r, 0));
|
||||
if (typeof requestAnimationFrame === 'function') {
|
||||
await new Promise(r => requestAnimationFrame(() => r()));
|
||||
await new Promise(r => requestAnimationFrame(() => r())); // двойной rAF для надежности
|
||||
} else {
|
||||
await new Promise(r => setTimeout(r, 32));
|
||||
}
|
||||
// 3) Проставляем связи из in (поддержка строк и массивов ссылок)
|
||||
for (const n of p.nodes) {
|
||||
if (!n.in) continue;
|
||||
const targetDfId = idMap[n.id];
|
||||
const targetIo = NODE_IO[n.type] || { inputs: [] };
|
||||
for (const [inName, ref] of Object.entries(n.in)) {
|
||||
const refs = Array.isArray(ref) ? ref : [ref];
|
||||
for (const oneRef of refs) {
|
||||
if (!oneRef || typeof oneRef !== 'string' || !oneRef.includes('.')) continue;
|
||||
const [srcId, outName] = oneRef.split('.');
|
||||
const sourceDfId = idMap[srcId];
|
||||
if (!sourceDfId) { logs.push(`skip: src ${srcId} not found`); continue; }
|
||||
const srcType = p.nodes.find(nn => nn.id === srcId)?.type;
|
||||
|
||||
let outIdx = resolveOutIdx(srcType, outName);
|
||||
let inIdx = resolveInIdx(n.type, inName);
|
||||
|
||||
// Fallback на первый порт, если неизвестные имена, но порт существует
|
||||
if (outIdx < 0) outIdx = 0;
|
||||
if (inIdx < 0) inIdx = 0;
|
||||
|
||||
const ok = await connectWithRetry(sourceDfId, targetDfId, outIdx + 1, inIdx + 1, 200, 25);
|
||||
if (ok) {
|
||||
logs.push(`connect: ${srcId}.${outName} (#${sourceDfId}.output_${outIdx + 1}) -> ${n.id}.${inName} (#${targetDfId}.input_${inIdx + 1})`);
|
||||
} else {
|
||||
logs.push(`skip connect (ports not ready after retries): ${srcId}.${outName} -> ${n.id}.${inName}`);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 4) Обновим линии и выведем лог
|
||||
try {
|
||||
Object.values(idMap).forEach((dfid) => {
|
||||
if (editor.updateConnectionNodes) {
|
||||
editor.updateConnectionNodes(`node-${dfid}`);
|
||||
}
|
||||
});
|
||||
} catch {}
|
||||
if (logs.length) {
|
||||
try { (typeof w.status === 'function') && w.status('Загружено (links):\n' + logs.join('\n')); } catch {}
|
||||
try { console.debug('[fromPipelineJSON]', logs); } catch {}
|
||||
}
|
||||
}
|
||||
|
||||
w.AgentUISer = {
|
||||
toPipelineJSON,
|
||||
fromPipelineJSON,
|
||||
};
|
||||
})(window);
|
||||
Reference in New Issue
Block a user