ai/byre
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 7 Wiki Activity

Keep active task transcripts in memory for instant task switching

Browse Source
This commit is contained in:
Ian Gulliver
2026-10-16 10:45:38 -07:00
parent 8200ffd17e
commit 91d95d0581
17 changed files with 252 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files
web/src/api.ts
+5
View File
@@ -123,6 +123,7 @@ export function subscribe(
filter: { task_id?: string; lead?: boolean },
onEvent: (ev: ByreEvent) => void,
onState: (connected: boolean) => void,
onReady: () => void,
): () => void {
let last = after;
let heard = Date.now();
@@ -155,6 +156,10 @@ export function subscribe(
s.addEventListener("ping", () => {
heard = Date.now();
});
s.addEventListener("ready", () => {
heard = Date.now();
onReady();
});
s.onmessage = (msg: MessageEvent<string>) => {
heard = Date.now();
const ev = JSON.parse(msg.data) as ByreEvent;
web/src/app.ts
+10 -4
View File
@@ -78,10 +78,16 @@ async function main(): Promise<void> {
{ capture: true },
);
subscribe(status.last_event_seq, {}, handleEvent, (connected) => {
state.connected = connected;
notify("connection");
});
subscribe(
status.last_event_seq,
{},
handleEvent,
(connected) => {
state.connected = connected;
notify("connection");
},
() => undefined,
);
addEventListener("hashchange", route);
route();
web/src/lead.ts
+3 -1
View File
@@ -1,11 +1,12 @@
import { post } from "./api.js";
import { h } from "./dom.js";
import { currentLead, state } from "./state.js";
import { currentLead, leadLoading, state } from "./state.js";
import { compactButton, mountChat, type ChatSource, type Head } from "./task.js";
const leadSource: ChatSource = {
key: () => "lead",
events: () => state.leadTranscript,
loading: leadLoading,
session: currentLead,
root: () => "",
taskId: () => undefined,
@@ -13,6 +14,7 @@ const leadSource: ChatSource = {
refreshTopics: ["lead.sessions"],
transcriptTopic: "lead.transcript",
streamTopic: "lead.stream",
readyTopic: "lead.ready",
empty: "",
canSend: () => true,
placeholder: "Message your lead. Paste or drop images to attach them.",
web/src/state.ts
+101 -35
View File
@@ -17,7 +17,9 @@ export type Topic =
| "focus"
| "lead.sessions"
| "lead.transcript"
| "lead.stream";
| "lead.stream"
| "ready"
| "lead.ready";
export const tabs: TaskTab[] = ["transcript", "changes", "files", "terminal", "inbox", "reviews", "project"];
@@ -130,6 +132,12 @@ export async function loadProjects(): Promise<void> {
export async function loadTasks(projectId: string): Promise<void> {
const tasks = await get("/api/projects/{projectId}/tasks", { path: { projectId } });
state.tasks.set(projectId, tasks);
for (const t of tasks) {
if (t.status === "active") {
taskFeed(t.id);
}
}
dropClosedFeeds();
await Promise.all(tasks.filter((t) => t.status === "active").map((t) => loadSessions(t.id)));
notify("tasks");
}
@@ -193,7 +201,8 @@ export function focusTab(tab: TaskTab): void {
}
class Feed {
events: ByreEvent[] = [];
readonly events: ByreEvent[] = [];
ready = false;
private last = 0;
private buffer: ByreEvent[] = [];
private timer = 0;
@@ -201,26 +210,34 @@ class Feed {
constructor(
private readonly transcriptTopic: Topic,
private readonly streamTopic: Topic,
private readonly store: (events: ByreEvent[]) => void,
private readonly readyTopic: Topic,
private readonly shown: () => boolean,
) {}
reset(): void {
clearTimeout(this.timer);
this.timer = 0;
this.buffer = [];
this.events = [];
this.last = 0;
this.store(this.events);
}
readonly push = (ev: ByreEvent): void => {
if (ev.seq === 0 && (!this.ready || !this.shown())) {
return;
}
this.buffer.push(ev);
if (!this.timer) {
if (this.ready && !this.timer) {
this.timer = setTimeout(() => this.flush(), 0);
}
};
readonly load = (history: ByreEvent[]): void => {
if (this.ready) {
return;
}
this.ready = true;
this.buffer = [...history, ...this.buffer];
this.flush();
if (this.shown()) {
notify(this.readyTopic);
}
};
private flush(): void {
clearTimeout(this.timer);
const events = this.buffer;
this.buffer = [];
this.timer = 0;
@@ -232,30 +249,84 @@ class Feed {
}
this.append(stored);
stored = [];
notify(this.streamTopic, ev);
if (this.shown()) {
notify(this.streamTopic, ev);
}
}
this.append(stored);
}
private append(events: ByreEvent[]): void {
const fresh = events.filter((ev) => ev.seq > this.last);
const fresh: ByreEvent[] = [];
for (const ev of events) {
if (ev.seq > this.last) {
fresh.push(ev);
this.last = ev.seq;
}
}
if (fresh.length === 0) {
return;
}
this.events.push(...fresh);
this.last = fresh.at(-1)!.seq;
notify(this.transcriptTopic, fresh);
if (this.shown()) {
notify(this.transcriptTopic, fresh);
}
}
}
const taskFeed = new Feed("transcript", "stream", (events) => {
state.transcript = events;
});
const leadFeed = new Feed("lead.transcript", "lead.stream", (events) => {
state.leadTranscript = events;
});
const historyPage = 1000;
let closeTaskStream: (() => void) | undefined;
const taskFeeds = new Map<string, Feed>();
const leadFeed = new Feed("lead.transcript", "lead.stream", "lead.ready", () => true);
state.leadTranscript = leadFeed.events;
async function loadHistory(taskId: string, feed: Feed): Promise<void> {
const events: ByreEvent[] = [];
for (;;) {
const page = await get("/api/tasks/{taskId}/events", { path: { taskId }, query: { after: events.at(-1)?.seq ?? 0, limit: historyPage } });
events.push(...page);
if (page.length < historyPage) {
break;
}
}
feed.load(events);
}
const historyLoads = new Set<Feed>();
function taskFeed(taskId: string): Feed {
let feed = taskFeeds.get(taskId);
if (!feed) {
feed = new Feed("transcript", "stream", "ready", () => state.selectedTaskId === taskId);
taskFeeds.set(taskId, feed);
}
if (feed.ready || historyLoads.has(feed)) {
return feed;
}
const loading = feed;
historyLoads.add(loading);
loadHistory(taskId, loading)
.catch((err: unknown) => console.error("load task history", taskId, err))
.finally(() => historyLoads.delete(loading));
return feed;
}
function dropClosedFeeds(): void {
for (const taskId of taskFeeds.keys()) {
const task = findTask(taskId);
if (taskId !== state.selectedTaskId && task && task.status !== "active") {
taskFeeds.delete(taskId);
}
}
}
export function transcriptLoading(): boolean {
return state.selectedTaskId !== undefined && taskFeeds.get(state.selectedTaskId)?.ready !== true;
}
export function leadLoading(): boolean {
return !leadFeed.ready;
}
const visitsKey = "byre.visits";
@@ -273,27 +344,22 @@ function markVisited(taskId: string | undefined): void {
export function selectTask(taskId: string | undefined): void {
markVisited(state.selectedTaskId);
markVisited(taskId);
closeTaskStream?.();
closeTaskStream = undefined;
taskFeed.reset();
state.selectedTaskId = taskId;
state.transcript = taskId ? taskFeed(taskId).events : [];
state.pinned = undefined;
state.reviewed = undefined;
dropClosedFeeds();
notify("selection");
if (!taskId) {
return;
}
closeTaskStream = subscribe(0, { task_id: taskId }, taskFeed.push, () => undefined);
}
export function startLeadStream(): void {
leadFeed.reset();
subscribe(0, { lead: true }, leadFeed.push, () => undefined);
subscribe(0, { lead: true }, leadFeed.push, () => undefined, () => leadFeed.load([]));
}
export function handleEvent(ev: ByreEvent): void {
if (ev.task_id) {
taskFeeds.get(ev.task_id)?.push(ev);
}
if (ev.seq === 0) {
notify("event", ev);
return;
web/src/task.ts
+10 -1
View File
@@ -4,7 +4,7 @@ import { markdown } from "./markdown.js";
import { previewURL } from "./preview.js";
import { findingCount, reviewCounts, severityCounts, unignoredFindings } from "./reviews.js";
import { composerText, endSend, flushDrafts, forgetLostImages, lostImageCount, setComposerImages, setComposerText, startSend } from "./drafts.js";
import { currentSession, findTask, focusTab, focusTask, on, onEach, pickReview, pinSnapshot, setFocus, showReview, state, type Topic } from "./state.js";
import { currentSession, findTask, focusTab, focusTask, on, onEach, pickReview, pinSnapshot, setFocus, showReview, state, transcriptLoading, type Topic } from "./state.js";
import { taskStyle } from "./taskcolor.js";
import { fillTodoLinks } from "./todolink.js";
@@ -15,6 +15,7 @@ type Review = Schemas["Review"];
export type ChatSource = {
key: () => string | undefined;
events: () => ByreEvent[];
loading: () => boolean;
session: () => AgentSession | undefined;
root: () => string;
taskId: () => string | undefined;
@@ -22,6 +23,7 @@ export type ChatSource = {
refreshTopics: Topic[];
transcriptTopic: Topic;
streamTopic: Topic;
readyTopic: Topic;
empty: string;
canSend: (sessionState: string) => boolean;
placeholder: string;
@@ -84,6 +86,7 @@ type Ctx = {
const taskSource: ChatSource = {
key: () => state.selectedTaskId,
events: () => state.transcript,
loading: transcriptLoading,
session: () => (state.selectedTaskId ? currentSession(state.selectedTaskId) : undefined),
root: () => (state.selectedTaskId ? (findTask(state.selectedTaskId)?.worktree_path ?? "") : ""),
taskId: () => state.selectedTaskId,
@@ -91,6 +94,7 @@ const taskSource: ChatSource = {
refreshTopics: ["tasks", "sessions"],
transcriptTopic: "transcript",
streamTopic: "stream",
readyTopic: "ready",
empty: "No task selected. Ask your lead, or pick one with ⌘K.",
canSend: () => true,
placeholder: "Message the agent. Paste or drop images to attach them.",
@@ -235,6 +239,7 @@ export function mountChat(root: HTMLElement, src: ChatSource, makeHead: (rerende
const headImpl = makeHead(() => renderHead());
const scanner = h("div", { class: "scanner", hidden: true });
const loading = h("div", { class: "empty" }, "loading…");
const setRunning = (running: boolean) => {
transcript.classList.toggle("running", running);
scanner.hidden = !running;
@@ -541,6 +546,9 @@ export function mountChat(root: HTMLElement, src: ChatSource, makeHead: (rerende
return;
}
replace(transcript, scanner);
if (src.loading()) {
transcript.prepend(loading);
}
appendEvents(src.events());
renderComposer();
};
@@ -553,6 +561,7 @@ export function mountChat(root: HTMLElement, src: ChatSource, makeHead: (rerende
}),
onEach([src.transcriptTopic], (fresh) => appendEvents(fresh as ByreEvent[])),
onEach([src.streamTopic], (ev) => appendDelta(ev as ByreEvent)),
onEach([src.readyTopic], () => loading.remove()),
onEach(["tasks", "sessions"], () => fillTodoLinks(transcript, models[0]!)),
];