1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
import type {
ChatDeltaMessage,
ChatErrorMessage,
ChatSendMessage,
} from "@dispatch/transport-contract";
import type { ChatMessage } from "@dispatch/wire";
import type { RenderedChunk, TranscriptState } from "../../core/chunks";
import {
appendUserMessage,
applyHistory,
foldEvent,
initialState,
selectChunks,
selectMessages,
} from "../../core/chunks";
import type { ConversationCache } from "../conversation-cache";
import type { ChatTransport, HistorySync } from "./ports";
export interface ChatStoreDependencies {
readonly conversationId: string;
readonly model?: string;
readonly transport: ChatTransport;
readonly historySync: HistorySync;
readonly cache: ConversationCache;
}
export interface ChatStore {
readonly messages: readonly ChatMessage[];
readonly chunks: readonly RenderedChunk[];
readonly pendingSync: boolean;
readonly error: string | null;
readonly model: string | undefined;
handleDelta(msg: ChatDeltaMessage | ChatErrorMessage): void;
send(text: string): void;
setModel(model: string): void;
load(): Promise<void>;
dispose(): void;
}
export function createChatStore(deps: ChatStoreDependencies): ChatStore {
let transcript = $state<TranscriptState>(initialState());
let _pendingSync = $state(false);
let _error = $state<string | null>(null);
let _model = $state<string | undefined>(deps.model);
let disposed = false;
async function syncTail(): Promise<void> {
if (disposed || _pendingSync) return;
_pendingSync = true;
try {
const since = await deps.cache.sinceSeq(deps.conversationId);
const res = await deps.historySync(deps.conversationId, since);
const merged = await deps.cache.commit(deps.conversationId, res.chunks);
transcript = applyHistory(transcript, merged);
_error = null;
} catch (err) {
_error = err instanceof Error ? err.message : String(err);
} finally {
_pendingSync = false;
}
}
return {
get messages(): readonly ChatMessage[] {
return selectMessages(transcript);
},
get chunks(): readonly RenderedChunk[] {
return selectChunks(transcript);
},
get pendingSync(): boolean {
return _pendingSync;
},
get error(): string | null {
return _error;
},
get model(): string | undefined {
return _model;
},
handleDelta(msg: ChatDeltaMessage | ChatErrorMessage): void {
if (msg.type === "chat.error") {
if (msg.conversationId !== undefined && msg.conversationId !== deps.conversationId) {
return;
}
_error = msg.message;
return;
}
if (msg.event.conversationId !== deps.conversationId) {
return;
}
transcript = foldEvent(transcript, msg.event);
if (transcript.sealedTurnId !== null) {
void syncTail();
}
},
send(text: string): void {
transcript = appendUserMessage(transcript, text);
const msg: ChatSendMessage = {
type: "chat.send",
conversationId: deps.conversationId,
message: text,
...(_model !== undefined ? { model: _model } : {}),
};
deps.transport.send(msg);
},
setModel(model: string): void {
_model = model;
},
async load(): Promise<void> {
const cached = await deps.cache.load(deps.conversationId);
if (cached.length > 0) {
transcript = applyHistory(transcript, cached);
}
await syncTail();
},
dispose(): void {
disposed = true;
},
};
}
|
