OrgTJWater/TJWaterAgent
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(chat): 解决token传输、本地文件存储顺序、读取的问题

Browse Source
This commit is contained in:
Huarch
2026-06-04 18:19:29 +08:00
parent 10c11a5254
commit fc0e76439d
3 changed files with 76 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/routes/chat.ts
+34 -12
View File
@@ -823,12 +823,30 @@ export const buildChatRouter = (
};
activeRuns.set(clientSessionId, activeRun);
lastRunStatuses.set(clientSessionId, "running");
await sessionUiStateStore.write(toSessionUiStateContext(activeSessionRecord), {
const sessionUiStateContext = toSessionUiStateContext(activeSessionRecord);
let persistQueue = sessionUiStateStore.write(sessionUiStateContext, {
sessionId: activeSessionRecord.sessionId,
isTitleManuallyEdited: initialSessionState?.isTitleManuallyEdited ?? false,
messages: initialMessages,
branchGroups,
});
const queueSessionUiStatePersist = () => {
const snapshot = {
sessionId: activeSessionRecord.sessionId,
isTitleManuallyEdited: initialSessionState?.isTitleManuallyEdited ?? false,
messages: activeRun.messages,
branchGroups,
};
persistQueue = persistQueue
.catch((error) => {
logger.warn(
{ err: error, sessionId: clientSessionId },
"failed to persist previous chat stream state",
);
})
.then(() => sessionUiStateStore.write(sessionUiStateContext, snapshot));
return persistQueue;
};
const primarySubscriber: StreamSubscriber = {
write: (event, data) => {
if (!streamClosed && !res.writableEnded && !res.destroyed) {
@@ -850,7 +868,7 @@ export const buildChatRouter = (
req.on("close", handleClientClose);
res.on("close", handleClientClose);
const publish = async (event: string, data: Record<string, unknown>) => {
const publish = (event: string, data: Record<string, unknown>) => {
if (event === "token") {
activeRun.messages = updateLastAssistantMessage(activeRun.messages, (message) => ({
...message,
@@ -887,15 +905,12 @@ export const buildChatRouter = (
}));
}
await sessionUiStateStore.write(toSessionUiStateContext(activeSessionRecord), {
sessionId: activeSessionRecord.sessionId,
isTitleManuallyEdited: initialSessionState?.isTitleManuallyEdited ?? false,
messages: activeRun.messages,
branchGroups,
});
for (const subscriber of activeRun.subscribers) {
subscriber.write(event, data);
}
void queueSessionUiStatePersist().catch((error) => {
logger.warn({ err: error, sessionId: clientSessionId }, "failed to persist chat stream state");
});
};
try {
@@ -920,11 +935,12 @@ export const buildChatRouter = (
projectId: requestContext.projectId,
signal: abortController.signal,
write: (event, data) => {
void publish(event, data).catch((error) => {
logger.warn({ err: error, sessionId: clientSessionId }, "failed to publish chat stream event");
});
publish(event, data);
},
});
await persistQueue.catch((error) => {
logger.warn({ err: error, sessionId: clientSessionId }, "failed to persist chat stream state");
});
if (!streamResult.aborted && !streamResult.failed) {
const messages = await runtime.messages(binding.sessionId, 60);
@@ -965,13 +981,19 @@ export const buildChatRouter = (
sessionTitle &&
sessionTitle !== existingSessionTitle
) {
await publish("session_title", {
publish("session_title", {
session_id: clientSessionId,
title: sessionTitle,
});
await persistQueue.catch((error) => {
logger.warn({ err: error, sessionId: clientSessionId }, "failed to persist chat stream state");
});
}
}
} finally {
await persistQueue.catch((error) => {
logger.warn({ err: error, sessionId: clientSessionId }, "failed to persist chat stream state");
});
sessionBridge.finalizeRequest(clientSessionId);
activeRun.status = abortController.signal.aborted
? activeRun.status === "aborted"