531 lines
17 KiB
JavaScript
531 lines
17 KiB
JavaScript
const form = document.getElementById("chat-form");
|
|
const input = document.getElementById("message-input");
|
|
const messages = document.getElementById("messages");
|
|
const statusEl = document.getElementById("status");
|
|
const pendingEl = document.getElementById("pending-actions");
|
|
const warningEl = document.getElementById("warning");
|
|
const memoryInspectorEl = document.getElementById("memory-inspector");
|
|
const memoryRefreshButton = document.getElementById("memory-refresh");
|
|
const memoryClearButton = document.getElementById("memory-clear");
|
|
|
|
let ollamaOnline = true;
|
|
|
|
function addMessage(role, text) {
|
|
const node = document.createElement("div");
|
|
node.className = `message ${role}`;
|
|
setMessageMarkdown(node, text);
|
|
messages.appendChild(node);
|
|
messages.scrollTop = messages.scrollHeight;
|
|
return node;
|
|
}
|
|
|
|
function setMessageMarkdown(node, text) {
|
|
const body = node.querySelector(".message-body") || node;
|
|
body.innerHTML = renderMarkdown(text);
|
|
}
|
|
|
|
function setMessageActivity(node, text, active = false) {
|
|
const activity = node.querySelector(".message-activity");
|
|
if (!activity) return;
|
|
const phase = activity.querySelector(".message-phase");
|
|
phase.innerHTML = "";
|
|
if (text) {
|
|
const label = document.createElement("span");
|
|
label.textContent = text;
|
|
phase.appendChild(label);
|
|
}
|
|
if (active) {
|
|
const dots = document.createElement("span");
|
|
dots.className = "working-dots";
|
|
dots.innerHTML = "<i></i><i></i><i></i>";
|
|
phase.appendChild(dots);
|
|
}
|
|
}
|
|
|
|
function setMessageMetrics(node, metrics) {
|
|
const metricsEl = node.querySelector(".message-metrics");
|
|
if (!metricsEl) return;
|
|
metricsEl.textContent = metrics || "";
|
|
}
|
|
|
|
function ensureStreamingChrome(node) {
|
|
if (node.querySelector(".message-activity")) return;
|
|
node.innerHTML = "";
|
|
const activity = document.createElement("div");
|
|
activity.className = "message-activity";
|
|
const phase = document.createElement("span");
|
|
phase.className = "message-phase";
|
|
const metrics = document.createElement("span");
|
|
metrics.className = "message-metrics";
|
|
const body = document.createElement("div");
|
|
body.className = "message-body";
|
|
activity.append(phase, metrics);
|
|
node.append(activity, body);
|
|
}
|
|
|
|
function renderMarkdown(text) {
|
|
const lines = text.replace(/\r\n/g, "\n").split(/\n/);
|
|
const output = [];
|
|
let inList = false;
|
|
let inOrderedList = false;
|
|
let inCode = false;
|
|
let codeLines = [];
|
|
|
|
const closeLists = () => {
|
|
if (inList) {
|
|
output.push("</ul>");
|
|
inList = false;
|
|
}
|
|
if (inOrderedList) {
|
|
output.push("</ol>");
|
|
inOrderedList = false;
|
|
}
|
|
};
|
|
|
|
const flushCode = () => {
|
|
output.push(`<pre><code>${escapeHtml(codeLines.join("\n"))}</code></pre>`);
|
|
codeLines = [];
|
|
inCode = false;
|
|
};
|
|
|
|
const isTableAt = (index) => {
|
|
if (index + 1 >= lines.length) return false;
|
|
return isTableRow(lines[index]) && isTableDivider(lines[index + 1]);
|
|
};
|
|
|
|
for (let index = 0; index < lines.length; index += 1) {
|
|
const line = lines[index];
|
|
const trimmed = line.trim();
|
|
|
|
if (/^```/.test(trimmed)) {
|
|
if (inCode) {
|
|
flushCode();
|
|
} else {
|
|
closeLists();
|
|
inCode = true;
|
|
}
|
|
continue;
|
|
}
|
|
|
|
if (inCode) {
|
|
codeLines.push(line);
|
|
continue;
|
|
}
|
|
|
|
if (isTableAt(index)) {
|
|
closeLists();
|
|
const headers = splitTableRow(lines[index]);
|
|
const aligns = splitTableRow(lines[index + 1]).map((cell) => tableAlignment(cell));
|
|
const rows = [];
|
|
index += 2;
|
|
while (index < lines.length && isTableRow(lines[index])) {
|
|
rows.push(splitTableRow(lines[index]));
|
|
index += 1;
|
|
}
|
|
index -= 1;
|
|
output.push(renderTable(headers, aligns, rows));
|
|
continue;
|
|
}
|
|
|
|
const unorderedItem = line.match(/^(\s*)[-*+]\s+(.+)$/);
|
|
if (unorderedItem) {
|
|
if (inOrderedList) {
|
|
output.push("</ol>");
|
|
inOrderedList = false;
|
|
}
|
|
if (!inList) {
|
|
output.push("<ul>");
|
|
inList = true;
|
|
}
|
|
const nestedClass = unorderedItem[1].length >= 2 ? ' class="nested"' : "";
|
|
output.push(`<li${nestedClass}>${inlineMarkdown(unorderedItem[2])}</li>`);
|
|
continue;
|
|
}
|
|
|
|
const orderedItem = line.match(/^(\s*)\d+\.\s+(.+)$/);
|
|
if (orderedItem) {
|
|
if (inList) {
|
|
output.push("</ul>");
|
|
inList = false;
|
|
}
|
|
if (!inOrderedList) {
|
|
output.push("<ol>");
|
|
inOrderedList = true;
|
|
}
|
|
const nestedClass = orderedItem[1].length >= 2 ? ' class="nested"' : "";
|
|
output.push(`<li${nestedClass}>${inlineMarkdown(orderedItem[2])}</li>`);
|
|
continue;
|
|
}
|
|
|
|
closeLists();
|
|
|
|
if (/^>\s?/.test(trimmed)) {
|
|
output.push(`<blockquote>${inlineMarkdown(trimmed.replace(/^>\s?/, ""))}</blockquote>`);
|
|
} else if (/^---+$/.test(trimmed)) {
|
|
output.push("<hr>");
|
|
} else if (/^#{1,6}\s+/.test(trimmed)) {
|
|
const level = Math.min(4, trimmed.match(/^#+/)[0].length + 2);
|
|
output.push(`<h${level}>${inlineMarkdown(trimmed.replace(/^#{1,6}\s+/, ""))}</h${level}>`);
|
|
} else if (trimmed) {
|
|
output.push(`<p>${inlineMarkdown(line)}</p>`);
|
|
} else {
|
|
output.push("<br>");
|
|
}
|
|
}
|
|
|
|
if (inCode) flushCode();
|
|
closeLists();
|
|
return output.join("");
|
|
}
|
|
|
|
function inlineMarkdown(text) {
|
|
return escapeHtml(text)
|
|
.replace(/`([^`]+)`/g, "<code>$1</code>")
|
|
.replace(/\*\*(.+?)\*\*/g, "<strong>$1</strong>")
|
|
.replace(/\*([^*]+)\*/g, "<em>$1</em>")
|
|
.replace(/\[([^\]]+)\]\((https?:\/\/[^)\s]+)\)/g, '<a href="$2" target="_blank" rel="noreferrer">$1</a>');
|
|
}
|
|
|
|
function isTableRow(line) {
|
|
const trimmed = line.trim();
|
|
return trimmed.includes("|") && /^\|?.+\|.+\|?$/.test(trimmed);
|
|
}
|
|
|
|
function isTableDivider(line) {
|
|
return splitTableRow(line).every((cell) => /^:?-{3,}:?$/.test(cell.trim()));
|
|
}
|
|
|
|
function splitTableRow(line) {
|
|
return line
|
|
.trim()
|
|
.replace(/^\|/, "")
|
|
.replace(/\|$/, "")
|
|
.split("|")
|
|
.map((cell) => cell.trim());
|
|
}
|
|
|
|
function tableAlignment(cell) {
|
|
const trimmed = cell.trim();
|
|
if (trimmed.startsWith(":") && trimmed.endsWith(":")) return "center";
|
|
if (trimmed.endsWith(":")) return "right";
|
|
return "left";
|
|
}
|
|
|
|
function renderTable(headers, aligns, rows) {
|
|
const head = headers
|
|
.map((cell, index) => `<th style="text-align:${aligns[index] || "left"}">${inlineMarkdown(cell)}</th>`)
|
|
.join("");
|
|
const body = rows
|
|
.map((row) => {
|
|
const cells = headers
|
|
.map((_, index) => `<td style="text-align:${aligns[index] || "left"}">${inlineMarkdown(row[index] || "")}</td>`)
|
|
.join("");
|
|
return `<tr>${cells}</tr>`;
|
|
})
|
|
.join("");
|
|
return `<div class="table-wrap"><table><thead><tr>${head}</tr></thead><tbody>${body}</tbody></table></div>`;
|
|
}
|
|
|
|
function escapeHtml(text) {
|
|
return text
|
|
.replace(/&/g, "&")
|
|
.replace(/</g, "<")
|
|
.replace(/>/g, ">")
|
|
.replace(/"/g, """)
|
|
.replace(/'/g, "'");
|
|
}
|
|
|
|
function formatMetrics(event) {
|
|
const read = formatTokenMetric(event.reading_tokens, event.reading_tokens_per_second);
|
|
const wrote = formatTokenMetric(event.writing_tokens, event.writing_tokens_per_second);
|
|
return [read && `read ${read}`, wrote && `wrote ${wrote}`].filter(Boolean).join(" | ");
|
|
}
|
|
|
|
function formatTokenMetric(tokens, speed) {
|
|
if (!tokens) return "";
|
|
const speedText = typeof speed === "number" ? ` @ ${speed.toFixed(1)}/s` : "";
|
|
return `${tokens} tok${speedText}`;
|
|
}
|
|
|
|
function setWarning(text) {
|
|
warningEl.hidden = !text;
|
|
warningEl.textContent = text || "";
|
|
}
|
|
|
|
async function checkHealth() {
|
|
try {
|
|
const response = await fetch("/api/health");
|
|
const result = await response.json();
|
|
const health = result.ollama || {};
|
|
ollamaOnline = Boolean(health.online);
|
|
if (!ollamaOnline) {
|
|
statusEl.textContent = "Offline";
|
|
setWarning(health.message || "Ollama is offline. Start Ollama before chatting.");
|
|
return false;
|
|
}
|
|
if (health.model_available === false) {
|
|
setWarning(`Ollama is online, but model "${health.model}" is not pulled. Run: ollama pull ${health.model}`);
|
|
} else {
|
|
setWarning("");
|
|
}
|
|
statusEl.textContent = "Ready";
|
|
return true;
|
|
} catch (error) {
|
|
ollamaOnline = false;
|
|
statusEl.textContent = "Offline";
|
|
setWarning(`Could not check Ollama health: ${error.message}`);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
function renderPending(actions) {
|
|
pendingEl.innerHTML = "";
|
|
if (!actions.length) {
|
|
pendingEl.className = "pending-empty";
|
|
pendingEl.textContent = "No pending actions";
|
|
return;
|
|
}
|
|
pendingEl.className = "";
|
|
for (const action of actions) {
|
|
const card = document.createElement("div");
|
|
card.className = "pending-card";
|
|
const title = document.createElement("strong");
|
|
title.textContent = action.label;
|
|
const endpoint = document.createElement("p");
|
|
endpoint.className = "muted";
|
|
endpoint.textContent = action.endpoint;
|
|
const payload = document.createElement("pre");
|
|
payload.textContent = JSON.stringify(action.payload, null, 2);
|
|
const approve = document.createElement("button");
|
|
approve.textContent = "Approve";
|
|
approve.addEventListener("click", () => approveAction(action.id));
|
|
const decline = document.createElement("button");
|
|
decline.className = "decline-button";
|
|
decline.textContent = "Decline";
|
|
decline.addEventListener("click", () => declineAction(action.id));
|
|
const controls = document.createElement("div");
|
|
controls.className = "pending-controls";
|
|
controls.append(decline, approve);
|
|
card.append(title, endpoint, payload, controls);
|
|
pendingEl.appendChild(card);
|
|
}
|
|
}
|
|
|
|
async function approveAction(id) {
|
|
statusEl.textContent = "Sending";
|
|
try {
|
|
const response = await fetch(`/api/approve/${id}`, { method: "POST" });
|
|
const result = await response.json();
|
|
addMessage("assistant", `Approval result:\n${JSON.stringify(result, null, 2)}`);
|
|
await refreshPending();
|
|
} catch (error) {
|
|
addMessage("assistant", `Approval failed: ${error.message}`);
|
|
} finally {
|
|
statusEl.textContent = "Ready";
|
|
}
|
|
}
|
|
|
|
async function declineAction(id) {
|
|
statusEl.textContent = "Declining";
|
|
try {
|
|
const response = await fetch(`/api/decline/${id}`, { method: "POST" });
|
|
const result = await response.json();
|
|
if (result.error) {
|
|
addMessage("assistant warning-message", `Decline failed: ${result.error}`);
|
|
} else {
|
|
addMessage("assistant", `Declined pending action: ${result.pending_action?.label || id}`);
|
|
}
|
|
await refreshPending();
|
|
} catch (error) {
|
|
addMessage("assistant warning-message", `Decline failed: ${error.message}`);
|
|
} finally {
|
|
statusEl.textContent = "Ready";
|
|
}
|
|
}
|
|
|
|
async function refreshPending() {
|
|
const response = await fetch("/api/pending-actions");
|
|
const result = await response.json();
|
|
renderPending(result.pending_actions || []);
|
|
}
|
|
|
|
function renderMemory(data) {
|
|
memoryInspectorEl.innerHTML = "";
|
|
const counts = document.createElement("div");
|
|
counts.className = "memory-counts";
|
|
counts.textContent = `${data.memories.length} memories | ${data.conversations.length} chat rows | ${data.profile.length} profile keys | ${data.scheduled_jobs.length} jobs`;
|
|
memoryInspectorEl.appendChild(counts);
|
|
|
|
const path = document.createElement("div");
|
|
path.className = "memory-path";
|
|
path.textContent = data.path;
|
|
memoryInspectorEl.appendChild(path);
|
|
|
|
memoryInspectorEl.appendChild(memoryGroup("Memories", data.memories, (item) => `${item.kind} (${item.importance})\n${item.content}`));
|
|
memoryInspectorEl.appendChild(memoryGroup("Profile", data.profile, (item) => `${item.key}\n${JSON.stringify(item.value, null, 2)}`));
|
|
memoryInspectorEl.appendChild(memoryGroup("Recent Chat", data.conversations, (item) => `${item.created_at} ${item.role}\n${item.content}`));
|
|
memoryInspectorEl.appendChild(memoryGroup("Wake Jobs", data.scheduled_jobs, (item) => `${item.id}\n${item.trigger_type}: ${item.trigger_value}\n${item.prompt}`));
|
|
}
|
|
|
|
function memoryGroup(title, items, formatter) {
|
|
const group = document.createElement("details");
|
|
group.className = "memory-group";
|
|
const summary = document.createElement("summary");
|
|
summary.textContent = `${title} (${items.length})`;
|
|
group.appendChild(summary);
|
|
if (!items.length) {
|
|
const empty = document.createElement("p");
|
|
empty.className = "muted";
|
|
empty.textContent = "Empty";
|
|
group.appendChild(empty);
|
|
return group;
|
|
}
|
|
for (const item of items.slice(0, 20)) {
|
|
const entry = document.createElement("pre");
|
|
entry.textContent = formatter(item);
|
|
group.appendChild(entry);
|
|
}
|
|
return group;
|
|
}
|
|
|
|
async function refreshMemory() {
|
|
try {
|
|
const response = await fetch("/api/memory?limit=50");
|
|
const data = await response.json();
|
|
renderMemory(data);
|
|
} catch (error) {
|
|
memoryInspectorEl.textContent = `Memory load failed: ${error.message}`;
|
|
}
|
|
}
|
|
|
|
async function clearMemory() {
|
|
const payload = {
|
|
include_memories: document.getElementById("clear-memories").checked,
|
|
include_conversations: document.getElementById("clear-conversations").checked,
|
|
include_profile: document.getElementById("clear-profile").checked,
|
|
include_jobs: document.getElementById("clear-jobs").checked,
|
|
include_outbox: document.getElementById("clear-outbox").checked,
|
|
};
|
|
if (!Object.values(payload).some(Boolean)) return;
|
|
const confirmed = window.confirm("Clear the selected TraderAI memory? This cannot be undone.");
|
|
if (!confirmed) return;
|
|
try {
|
|
const response = await fetch("/api/memory/clear", {
|
|
method: "POST",
|
|
headers: { "Content-Type": "application/json" },
|
|
body: JSON.stringify(payload),
|
|
});
|
|
const result = await response.json();
|
|
renderMemory(result.memory);
|
|
addMessage("assistant", `Memory cleared:\n${JSON.stringify(result.deleted, null, 2)}`);
|
|
} catch (error) {
|
|
addMessage("assistant warning-message", `Memory clear failed: ${error.message}`);
|
|
}
|
|
}
|
|
|
|
async function pollNotifications() {
|
|
try {
|
|
const response = await fetch("/api/notifications");
|
|
const result = await response.json();
|
|
for (const notification of result.notifications || []) {
|
|
addMessage("assistant", notification.content);
|
|
}
|
|
} catch {
|
|
// Notification polling should never interrupt chat.
|
|
}
|
|
}
|
|
|
|
form.addEventListener("submit", async (event) => {
|
|
event.preventDefault();
|
|
await sendMessage();
|
|
});
|
|
|
|
input.addEventListener("keydown", async (event) => {
|
|
if (event.key === "Enter" && !event.shiftKey) {
|
|
event.preventDefault();
|
|
await sendMessage();
|
|
}
|
|
});
|
|
|
|
memoryRefreshButton.addEventListener("click", refreshMemory);
|
|
memoryClearButton.addEventListener("click", clearMemory);
|
|
|
|
async function sendMessage() {
|
|
const message = input.value.trim();
|
|
if (!message || input.disabled) return;
|
|
const healthy = await checkHealth();
|
|
if (!healthy) {
|
|
addMessage("assistant warning-message", "Ollama is offline. Start Ollama, then try again.");
|
|
return;
|
|
}
|
|
input.value = "";
|
|
input.disabled = true;
|
|
addMessage("user", message);
|
|
const assistantNode = addMessage("assistant streaming", "");
|
|
ensureStreamingChrome(assistantNode);
|
|
let assistantText = "";
|
|
statusEl.textContent = "Working";
|
|
setMessageActivity(assistantNode, "Thinking", true);
|
|
setMessageMetrics(assistantNode, "");
|
|
try {
|
|
const response = await fetch("/api/chat/stream", {
|
|
method: "POST",
|
|
headers: { "Content-Type": "application/json" },
|
|
body: JSON.stringify({ message }),
|
|
});
|
|
if (!response.ok || !response.body) {
|
|
throw new Error(`HTTP ${response.status}`);
|
|
}
|
|
const reader = response.body.getReader();
|
|
const decoder = new TextDecoder();
|
|
let buffer = "";
|
|
while (true) {
|
|
const { value, done } = await reader.read();
|
|
if (done) break;
|
|
buffer += decoder.decode(value, { stream: true });
|
|
const events = buffer.split("\n\n");
|
|
buffer = events.pop() || "";
|
|
for (const rawEvent of events) {
|
|
const line = rawEvent.split("\n").find((entry) => entry.startsWith("data: "));
|
|
if (!line) continue;
|
|
const event = JSON.parse(line.slice(6));
|
|
if (event.type === "status") {
|
|
setMessageActivity(assistantNode, event.message, true);
|
|
} else if (event.type === "metrics") {
|
|
setMessageMetrics(assistantNode, formatMetrics(event));
|
|
} else if (event.type === "warning") {
|
|
setWarning(event.message);
|
|
assistantText += event.message;
|
|
setMessageMarkdown(assistantNode, assistantText);
|
|
} else if (event.type === "token") {
|
|
assistantText += event.content;
|
|
setMessageMarkdown(assistantNode, assistantText);
|
|
messages.scrollTop = messages.scrollHeight;
|
|
} else if (event.type === "done") {
|
|
renderPending(event.pending_actions || []);
|
|
}
|
|
}
|
|
}
|
|
} catch (error) {
|
|
const message = error.message.includes("503")
|
|
? "Ollama is offline or unreachable. Start Ollama, then try again."
|
|
: `Chat failed: ${error.message}`;
|
|
setWarning(message);
|
|
setMessageMarkdown(assistantNode, message);
|
|
} finally {
|
|
assistantNode.classList.remove("streaming");
|
|
input.disabled = false;
|
|
input.focus();
|
|
statusEl.textContent = "Ready";
|
|
setMessageActivity(assistantNode, "");
|
|
}
|
|
}
|
|
|
|
addMessage("assistant", "Tell me what to find or draft on UEX. I will ask for approval before sending anything.");
|
|
refreshPending();
|
|
refreshMemory();
|
|
pollNotifications();
|
|
checkHealth();
|
|
setInterval(checkHealth, 30000);
|
|
setInterval(pollNotifications, 15000);
|