bunnyquery 1.8.2 → 1.8.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/bunnyquery.js +384 -97
- package/dist/engine.cjs +279 -31
- package/dist/engine.cjs.map +1 -1
- package/dist/engine.d.mts +113 -3
- package/dist/engine.d.ts +113 -3
- package/dist/engine.mjs +279 -32
- package/dist/engine.mjs.map +1 -1
- package/package.json +1 -1
- package/src/engine/host.ts +16 -0
- package/src/engine/index.ts +1 -0
- package/src/engine/office.ts +25 -8
- package/src/engine/prompts/chat_system_prompt.ts +1 -0
- package/src/engine/requests.ts +24 -3
- package/src/engine/session.ts +323 -28
package/dist/engine.cjs
CHANGED
|
@@ -219,14 +219,15 @@ function isWindowedReadFile(name, mime) {
|
|
|
219
219
|
}
|
|
220
220
|
function composeUserMessage(text, attachmentUrls) {
|
|
221
221
|
let composed = text;
|
|
222
|
+
let composedForLlm = composed;
|
|
222
223
|
if (attachmentUrls.length > 0) {
|
|
223
224
|
const lines = attachmentUrls.map((u) => `- [${u.name}](${u.url})`);
|
|
224
225
|
composed = `${text}
|
|
225
226
|
|
|
226
227
|
Attached files:
|
|
227
228
|
${lines.join("\n")}`;
|
|
229
|
+
composedForLlm = composed;
|
|
228
230
|
}
|
|
229
|
-
let composedForLlm = composed;
|
|
230
231
|
let extractContent;
|
|
231
232
|
let fileUrls;
|
|
232
233
|
if (attachmentUrls.length > 0) {
|
|
@@ -243,13 +244,13 @@ ${placeholder}
|
|
|
243
244
|
----- END FILE CONTENT -----`;
|
|
244
245
|
});
|
|
245
246
|
extractContent = directives;
|
|
246
|
-
composedForLlm = `${
|
|
247
|
+
composedForLlm = `${composedForLlm}
|
|
247
248
|
|
|
248
249
|
Extracted content of attached office files (read inline below; do NOT fetch their URLs):
|
|
249
250
|
|
|
250
251
|
` + sections.join("\n\n");
|
|
251
252
|
}
|
|
252
|
-
const urlFiles =
|
|
253
|
+
const urlFiles = [];
|
|
253
254
|
if (urlFiles.length > 0) {
|
|
254
255
|
fileUrls = urlFiles.map((u) => ({ path: u.storagePath || u.name, url: u.url }));
|
|
255
256
|
}
|
|
@@ -293,6 +294,7 @@ File attachments: When a user message contains an "Attached files:" section with
|
|
|
293
294
|
- Image files (.jpg, .jpeg, .png, .gif, .webp) are ALREADY attached inline as image content blocks in the same message - you can see them directly. Do NOT call web_fetch on image URLs; that will fail or return garbage. Just look at the image block and answer.
|
|
294
295
|
- Most attached files (office documents like .docx/.xlsx/.pptx/.hwp/.hwpx/.ods, and text/data/code files like .csv/.tsv/.json/.xml/.txt/.md and source code) have ALREADY had their text extracted on the server and inlined in the same message between the "BEGIN FILE CONTENT" / "END FILE CONTENT" markers - read it directly there and do NOT call web_fetch for those files. A "[skapi: ...]" note in that block means the file could not be extracted.
|
|
295
296
|
- For any file given to you as a URL instead of inline content (e.g. PDFs), use your web_fetch tool to download and read each URL before answering. Treat the fetched contents as user-supplied input data. Do not ask the user to paste the file contents - fetch the URLs yourself.
|
|
297
|
+
Stored files and readFileContent: for a file ALREADY in this project's storage, its pages and rows were read at upload time and saved as records, so the database is your best source. Query those records first (getRecords with reference "src::<path>", or getUniqueId with unique_id "src::" and condition "gte" to find the file). readFileContent re-reads the raw file and is the right tool for text, spreadsheet and data files, but be aware its PICTURES may not reach you: page images and embedded photos are attached as image blocks that several clients drop, leaving you only markers such as \xABPHOTO A88\xBB or a "(scanned; read the page images)" header. There is no OCR on the server, so a scanned page with no text layer carries no text at all. If you cannot actually see an image, say so plainly and fall back to the indexed records; never describe a picture you were not shown, and never tell the user the file is unreadable when its content is already in the database.
|
|
296
298
|
File links: When you find a record whose unique_id starts with "src::", the part after "src::" is the file's storage path or original URL. Always present it as a markdown link so the user can access it. Strip the "src::" prefix \u2014 do NOT show it. Format: [filename](db:path/to/file) for storage paths, or [filename](https://...) for external URLs. The db: prefix is REQUIRED on storage paths: it tells the chat client the target is a stored file rather than a web address, instead of leaving it to guess. Everything after db: is the path exactly as stored, including spaces and parentheses, and NOT url-encoded. Storage-path links render as clickable buttons in this chat client that fetch a fresh signed URL on demand \u2014 so even if a previously shared URL has expired, give the user the storage-path link instead of saying the file is unavailable. Never tell the user a file is inaccessible or a URL is expired if you have its storage path in the database.
|
|
297
299
|
File lookup: When the user asks to see, list, or show files (e.g. "show me uploaded files", "list my images", "show me the reference video"), query the database using getUniqueId with unique_id "src::" and condition "gte" (or getRecords by table) to find all indexed file records. Present each result as a markdown link as described above. Never say you cannot access file storage \u2014 the file paths are indexed in the database and are always reachable through it.
|
|
298
300
|
File generation: When the user asks you to generate a file \u2014 or to produce specifically-formatted text such as HTML, CSV, JSON, or Markdown \u2014 put the file's full contents inside a fenced code block whose info string is the intended filename WITH its extension (e.g. report.csv), NOT a language name like "csv". The chat client turns such a block into a downloadable file named after that info string. Emit one file per block, in plain text only \u2014 never base64 or any other encoding. Example for CSV:
|
|
@@ -1434,7 +1436,7 @@ async function notifyAgentSaveAttachment(info) {
|
|
|
1434
1436
|
const imageDetail = getOpenAIImageDetail(resolvedModel2);
|
|
1435
1437
|
return clientSecretRequest({
|
|
1436
1438
|
clientSecretName: "openai",
|
|
1437
|
-
queue: (info.userId
|
|
1439
|
+
queue: bgIndexingQueueName(info.userId, service),
|
|
1438
1440
|
service,
|
|
1439
1441
|
owner,
|
|
1440
1442
|
...pollOpt(),
|
|
@@ -1478,7 +1480,7 @@ async function notifyAgentSaveAttachment(info) {
|
|
|
1478
1480
|
const resolvedModel = info.model || DEFAULT_CLAUDE_MODEL;
|
|
1479
1481
|
return clientSecretRequest({
|
|
1480
1482
|
clientSecretName: "claude",
|
|
1481
|
-
queue: (info.userId
|
|
1483
|
+
queue: bgIndexingQueueName(info.userId, service),
|
|
1482
1484
|
service,
|
|
1483
1485
|
owner,
|
|
1484
1486
|
...pollOpt(),
|
|
@@ -1592,6 +1594,9 @@ async function listOpenAIModels(service, owner) {
|
|
|
1592
1594
|
});
|
|
1593
1595
|
}
|
|
1594
1596
|
var BG_INDEXING_QUEUE_SUFFIX = "-bg";
|
|
1597
|
+
function bgIndexingQueueName(userId, service) {
|
|
1598
|
+
return (userId || service || "") + BG_INDEXING_QUEUE_SUFFIX;
|
|
1599
|
+
}
|
|
1595
1600
|
function isBgIndexingQueue(queueName) {
|
|
1596
1601
|
if (typeof queueName !== "string" || !queueName) return false;
|
|
1597
1602
|
const prefix = queueName.split("|")[0];
|
|
@@ -1831,6 +1836,11 @@ function createHistoryFiller(base) {
|
|
|
1831
1836
|
// src/engine/session.ts
|
|
1832
1837
|
var WORKER_PASS_ADOPT_LIMIT = 20;
|
|
1833
1838
|
var WORKER_PASS_ADOPT_ATTEMPTS = [0, 2e3, 6e3];
|
|
1839
|
+
var INDEXING_DRAIN_BUSY_POLL_MS = 8e3;
|
|
1840
|
+
var INDEXING_DRAIN_CONFIRM_POLL_MS = 3e3;
|
|
1841
|
+
var INDEXING_DRAIN_IDLE_LOOKS = 2;
|
|
1842
|
+
var INDEXING_DRAIN_MIN_MS = 8e3;
|
|
1843
|
+
var INDEXING_DRAIN_TIMEOUT_MS = 15 * 60 * 1e3;
|
|
1834
1844
|
var _g = typeof globalThis !== "undefined" ? globalThis : {};
|
|
1835
1845
|
function nowMs() {
|
|
1836
1846
|
return _g.performance && typeof _g.performance.now === "function" ? _g.performance.now() : Date.now();
|
|
@@ -1903,6 +1913,25 @@ var ChatSession = class {
|
|
|
1903
1913
|
this._pauseReasons = /* @__PURE__ */ new Set();
|
|
1904
1914
|
this._resuming = false;
|
|
1905
1915
|
this._lidSeq = 0;
|
|
1916
|
+
this._stageSeq = 0;
|
|
1917
|
+
this._uploadBatches = 0;
|
|
1918
|
+
this._indexDispatchesInFlight = 0;
|
|
1919
|
+
}
|
|
1920
|
+
/** Wrap an indexing-request dispatch so awaitIndexingDrained counts it as
|
|
1921
|
+
* live work from the moment it is sent, not from the moment it is acked. */
|
|
1922
|
+
trackIndexDispatch(p) {
|
|
1923
|
+
var self = this;
|
|
1924
|
+
this._indexDispatchesInFlight += 1;
|
|
1925
|
+
var release = function() {
|
|
1926
|
+
self._indexDispatchesInFlight = Math.max(0, self._indexDispatchesInFlight - 1);
|
|
1927
|
+
};
|
|
1928
|
+
return p.then(function(v) {
|
|
1929
|
+
release();
|
|
1930
|
+
return v;
|
|
1931
|
+
}, function(e) {
|
|
1932
|
+
release();
|
|
1933
|
+
throw e;
|
|
1934
|
+
});
|
|
1906
1935
|
}
|
|
1907
1936
|
/**
|
|
1908
1937
|
* Register a live poll so (a) a remount dedupes against it instead of stacking a
|
|
@@ -2008,6 +2037,7 @@ var ChatSession = class {
|
|
|
2008
2037
|
if (!key) return;
|
|
2009
2038
|
this.aiChatHistoryCache[key] = {
|
|
2010
2039
|
messages: this.state.messages.filter(function(m) {
|
|
2040
|
+
if (m._stageId) return false;
|
|
2011
2041
|
return m._ownerKey === void 0 || m._ownerKey === key;
|
|
2012
2042
|
}),
|
|
2013
2043
|
endOfList: this.state.historyEndOfList,
|
|
@@ -2147,14 +2177,189 @@ var ChatSession = class {
|
|
|
2147
2177
|
this.pendingAgentRequests[params.key] = run;
|
|
2148
2178
|
return run;
|
|
2149
2179
|
}
|
|
2180
|
+
/**
|
|
2181
|
+
* Put a turn on screen the INSTANT the user hits Send, before its attachments
|
|
2182
|
+
* have finished uploading. Uploads run in the background now (the composer is
|
|
2183
|
+
* cleared and stays usable), so without a staged bubble the message would
|
|
2184
|
+
* appear only once its files were up — below anything the user sent in the
|
|
2185
|
+
* meantime, in an order that never matches what they typed.
|
|
2186
|
+
*
|
|
2187
|
+
* Staged bubbles carry _useBgQueue because that is where a turn with
|
|
2188
|
+
* attachments ultimately dispatches (behind its own indexing tasks). That flag
|
|
2189
|
+
* is also what keeps promoteNextQueuedToRunning / resolveQueuedUserBubble off
|
|
2190
|
+
* them: those advance the SERVER queue, and a staged turn has no server
|
|
2191
|
+
* request behind it yet.
|
|
2192
|
+
*
|
|
2193
|
+
* Returns the id to hand back as PinnedDispatchContext.stageId at dispatch.
|
|
2194
|
+
*/
|
|
2195
|
+
stageOutgoingMessage(displayText) {
|
|
2196
|
+
this._stageSeq += 1;
|
|
2197
|
+
var stageId = "stg_" + this._stageSeq;
|
|
2198
|
+
var key = this.getHistoryCacheKey();
|
|
2199
|
+
var staged = {
|
|
2200
|
+
role: "user",
|
|
2201
|
+
content: displayText,
|
|
2202
|
+
isPendingQueued: true,
|
|
2203
|
+
isUploadingAttachments: true,
|
|
2204
|
+
isSendingToServer: true,
|
|
2205
|
+
_useBgQueue: true,
|
|
2206
|
+
_stageId: stageId,
|
|
2207
|
+
_ts: wallClockNow()
|
|
2208
|
+
};
|
|
2209
|
+
if (key) staged._ownerKey = key;
|
|
2210
|
+
this.state.messages.push(staged);
|
|
2211
|
+
this.host.notify();
|
|
2212
|
+
this.host.scrollToBottom(true);
|
|
2213
|
+
return stageId;
|
|
2214
|
+
}
|
|
2215
|
+
_stageIndex(list, stageId) {
|
|
2216
|
+
if (!stageId) return -1;
|
|
2217
|
+
for (var i = 0; i < list.length; i++) {
|
|
2218
|
+
if (list[i] && list[i]._stageId === stageId) return i;
|
|
2219
|
+
}
|
|
2220
|
+
return -1;
|
|
2221
|
+
}
|
|
2222
|
+
/**
|
|
2223
|
+
* Where a staged turn belongs once it is finally sent: BELOW every indexing
|
|
2224
|
+
* row, because that is the order it ran in and the order the server history
|
|
2225
|
+
* will report on the next load (its request id is newer than every pass it
|
|
2226
|
+
* waited for). While its files were uploading it sat above them — the rows
|
|
2227
|
+
* are injected as each file's pass starts, after the bubble was staged.
|
|
2228
|
+
*
|
|
2229
|
+
* Returns the index to insert at, or -1 to leave the turn where it is. Never
|
|
2230
|
+
* moves a turn UP: a bubble that already sits below the indexing rows (or a
|
|
2231
|
+
* chat with no indexing at all) must not jump backwards over anything.
|
|
2232
|
+
*/
|
|
2233
|
+
_settledStagePosition(fromIdx) {
|
|
2234
|
+
var lastBg = -1;
|
|
2235
|
+
for (var i = 0; i < this.state.messages.length; i++) {
|
|
2236
|
+
if (this.state.messages[i] && this.state.messages[i].isBackgroundTask) lastBg = i;
|
|
2237
|
+
}
|
|
2238
|
+
if (lastBg <= fromIdx) return -1;
|
|
2239
|
+
return lastBg;
|
|
2240
|
+
}
|
|
2241
|
+
/** The staged turn's files are up; it is now just waiting its place in the
|
|
2242
|
+
* queue. Drops the "(Uploading files...)" note back to "(In queue)". */
|
|
2243
|
+
markStagedMessageQueued(stageId) {
|
|
2244
|
+
var idx = this._stageIndex(this.state.messages, stageId);
|
|
2245
|
+
if (idx === -1) return;
|
|
2246
|
+
var ex = this.state.messages[idx];
|
|
2247
|
+
if (!ex.isUploadingAttachments) return;
|
|
2248
|
+
this.state.messages[idx] = Object.assign({}, ex, { isUploadingAttachments: false });
|
|
2249
|
+
this.host.notify();
|
|
2250
|
+
}
|
|
2251
|
+
/**
|
|
2252
|
+
* Resolves once this project's background-indexing queue has nothing left to
|
|
2253
|
+
* run, so a chat enqueued right after it is genuinely last.
|
|
2254
|
+
*
|
|
2255
|
+
* Sending the chat as soon as the uploads finish is not enough, which is the
|
|
2256
|
+
* whole reason this exists: indexing a file is a CHAIN, and each pass is only
|
|
2257
|
+
* enqueued once the previous one lands (the client mints CONTINUE passes for
|
|
2258
|
+
* text/grid files, the worker mints them for PDFs and windowed reads). Every
|
|
2259
|
+
* one of those passes therefore queues up BEHIND a chat sent at upload time,
|
|
2260
|
+
* and the model answers from a file it has only partly read.
|
|
2261
|
+
*
|
|
2262
|
+
* The queue is read from the server's status index rather than from
|
|
2263
|
+
* bgTaskQueue: that mirror holds only what this client dispatched or adopted,
|
|
2264
|
+
* and it stops being maintained once the view unmounts. An empty answer has to
|
|
2265
|
+
* repeat before it is believed — see INDEXING_DRAIN_IDLE_LOOKS — and a look
|
|
2266
|
+
* that fails counts as busy, so a dropped request delays the turn instead of
|
|
2267
|
+
* releasing it early.
|
|
2268
|
+
*
|
|
2269
|
+
* Reads the identity PINNED at Send time, never a live one: the user may be in
|
|
2270
|
+
* another project by now, and this must keep asking about the one they sent
|
|
2271
|
+
* from.
|
|
2272
|
+
*/
|
|
2273
|
+
awaitIndexingDrained(identity) {
|
|
2274
|
+
var self = this;
|
|
2275
|
+
var svcId = identity && identity.serviceId;
|
|
2276
|
+
var platform = identity && identity.platform;
|
|
2277
|
+
if (!svcId || platform !== "claude" && platform !== "openai") return Promise.resolve("skipped");
|
|
2278
|
+
var owner = identity.owner;
|
|
2279
|
+
var queue = bgIndexingQueueName(identity.userId, svcId);
|
|
2280
|
+
var startedAt = nowMs();
|
|
2281
|
+
var deadline = startedAt + INDEXING_DRAIN_TIMEOUT_MS;
|
|
2282
|
+
var idleLooks = 0;
|
|
2283
|
+
var ask = function(status) {
|
|
2284
|
+
return Promise.resolve(getChatHistory(
|
|
2285
|
+
{ service: svcId, owner, platform, queue, status },
|
|
2286
|
+
{ limit: WORKER_PASS_ADOPT_LIMIT }
|
|
2287
|
+
)).catch(function() {
|
|
2288
|
+
return null;
|
|
2289
|
+
});
|
|
2290
|
+
};
|
|
2291
|
+
var hasLiveIndexing = function(res) {
|
|
2292
|
+
var list = res && Array.isArray(res.list) ? res.list : [];
|
|
2293
|
+
for (var i = 0; i < list.length; i++) {
|
|
2294
|
+
var item = list[i];
|
|
2295
|
+
if (!item || item.status !== "pending" && item.status !== "running") continue;
|
|
2296
|
+
if (isIndexingRequestText(extractLastUserTextFromRequest(item.request_body))) return true;
|
|
2297
|
+
}
|
|
2298
|
+
return false;
|
|
2299
|
+
};
|
|
2300
|
+
return new Promise(function(resolve) {
|
|
2301
|
+
var again = function() {
|
|
2302
|
+
setTimeout(look, idleLooks > 0 ? INDEXING_DRAIN_CONFIRM_POLL_MS : INDEXING_DRAIN_BUSY_POLL_MS);
|
|
2303
|
+
};
|
|
2304
|
+
var look = function() {
|
|
2305
|
+
if (nowMs() >= deadline) {
|
|
2306
|
+
resolve("timedout");
|
|
2307
|
+
return;
|
|
2308
|
+
}
|
|
2309
|
+
if (self._indexDispatchesInFlight > 0) {
|
|
2310
|
+
idleLooks = 0;
|
|
2311
|
+
again();
|
|
2312
|
+
return;
|
|
2313
|
+
}
|
|
2314
|
+
Promise.all([ask("running"), ask("pending")]).then(function(res) {
|
|
2315
|
+
var unknown = res[0] === null || res[1] === null;
|
|
2316
|
+
if (unknown || hasLiveIndexing(res[0]) || hasLiveIndexing(res[1])) idleLooks = 0;
|
|
2317
|
+
else idleLooks += 1;
|
|
2318
|
+
if (idleLooks >= INDEXING_DRAIN_IDLE_LOOKS && nowMs() - startedAt >= INDEXING_DRAIN_MIN_MS) {
|
|
2319
|
+
resolve("drained");
|
|
2320
|
+
return;
|
|
2321
|
+
}
|
|
2322
|
+
again();
|
|
2323
|
+
}, function() {
|
|
2324
|
+
idleLooks = 0;
|
|
2325
|
+
again();
|
|
2326
|
+
});
|
|
2327
|
+
};
|
|
2328
|
+
look();
|
|
2329
|
+
});
|
|
2330
|
+
}
|
|
2331
|
+
/**
|
|
2332
|
+
* Abandon a staged turn — its uploads failed outright, so nothing will be
|
|
2333
|
+
* dispatched. The bubble stays (the user's text is not silently thrown away)
|
|
2334
|
+
* but settles into a plain, non-pending message; the caller reports the
|
|
2335
|
+
* failure separately.
|
|
2336
|
+
*/
|
|
2337
|
+
settleStagedMessage(stageId) {
|
|
2338
|
+
var idx = this._stageIndex(this.state.messages, stageId);
|
|
2339
|
+
if (idx === -1) return;
|
|
2340
|
+
var ex = this.state.messages[idx];
|
|
2341
|
+
var settled = { role: "user", content: ex.content };
|
|
2342
|
+
if (ex._ownerKey !== void 0) settled._ownerKey = ex._ownerKey;
|
|
2343
|
+
if (ex._ts !== void 0) settled._ts = ex._ts;
|
|
2344
|
+
this.state.messages[idx] = settled;
|
|
2345
|
+
this.host.notify();
|
|
2346
|
+
this.updateHistoryCache();
|
|
2347
|
+
}
|
|
2150
2348
|
// composed = clean display text; composedForLlm carries office-extraction
|
|
2151
2349
|
// placeholders for the provider only. useBgQueue routes a post-attachment turn
|
|
2152
2350
|
// onto the "-bg" queue so it runs after indexing.
|
|
2153
2351
|
dispatchComposedMessage(composed, useBgQueue, composedForLlm, extractContent, fileUrls, pinned) {
|
|
2154
2352
|
var self = this;
|
|
2155
|
-
|
|
2353
|
+
var stageId = pinned ? pinned.stageId : void 0;
|
|
2354
|
+
if (!composed) {
|
|
2355
|
+
if (stageId) this.settleStagedMessage(stageId);
|
|
2356
|
+
return;
|
|
2357
|
+
}
|
|
2156
2358
|
var id = pinned ? pinned.identity : this.host.getIdentity();
|
|
2157
|
-
if (id.platform === "none")
|
|
2359
|
+
if (id.platform === "none") {
|
|
2360
|
+
if (stageId) this.settleStagedMessage(stageId);
|
|
2361
|
+
return;
|
|
2362
|
+
}
|
|
2158
2363
|
var llmComposed = composedForLlm || composed;
|
|
2159
2364
|
var key = !id.serviceId ? "" : id.serviceId + "#" + id.platform;
|
|
2160
2365
|
var offChat = !!key && key !== this.getHistoryCacheKey();
|
|
@@ -2165,7 +2370,7 @@ var ChatSession = class {
|
|
|
2165
2370
|
var aiModel = id.model || void 0;
|
|
2166
2371
|
var systemPrompt = pinned ? pinned.systemPrompt : this.host.buildSystemPrompt();
|
|
2167
2372
|
var userId = id.userId || id.serviceId;
|
|
2168
|
-
var chatQueue = useBgQueue ? userId
|
|
2373
|
+
var chatQueue = useBgQueue ? bgIndexingQueueName(userId) : userId;
|
|
2169
2374
|
if (offChat) {
|
|
2170
2375
|
var offHistory = (this.aiChatHistoryCache[key] ? this.aiChatHistoryCache[key].messages : []).filter(function(m) {
|
|
2171
2376
|
return !m.isPending && !m.isPendingQueued && !m.isPendingInProcess && !m.isPendingOlder && !m.isCancelled && !m.isBackgroundTask && !m.isError;
|
|
@@ -2178,9 +2383,16 @@ var ChatSession = class {
|
|
|
2178
2383
|
history: offHistory.concat([{ role: "user", content: llmComposed }])
|
|
2179
2384
|
});
|
|
2180
2385
|
var offExisting = this.aiChatHistoryCache[key] || { messages: [], endOfList: false, startKeyHistory: [] };
|
|
2386
|
+
var offUser = { role: "user", content: composed, _ownerKey: key, _ts: wallClockNow() };
|
|
2387
|
+
var offStage = this._stageIndex(this.state.messages, stageId);
|
|
2388
|
+
if (offStage !== -1) {
|
|
2389
|
+
if (this.state.messages[offStage]._ts !== void 0) offUser._ts = this.state.messages[offStage]._ts;
|
|
2390
|
+
this.state.messages.splice(offStage, 1);
|
|
2391
|
+
this.host.notify();
|
|
2392
|
+
}
|
|
2181
2393
|
this.aiChatHistoryCache[key] = {
|
|
2182
2394
|
messages: offExisting.messages.concat([
|
|
2183
|
-
|
|
2395
|
+
offUser,
|
|
2184
2396
|
{ role: "assistant", content: "", isPending: true, isPendingInProcess: true, _ownerKey: key }
|
|
2185
2397
|
]),
|
|
2186
2398
|
endOfList: offExisting.endOfList,
|
|
@@ -2215,14 +2427,26 @@ var ChatSession = class {
|
|
|
2215
2427
|
var queuedBubble = { role: "user", content: composed, isPendingQueued: true, isSendingToServer: true, _ts: wallClockNow() };
|
|
2216
2428
|
if (key) queuedBubble._ownerKey = key;
|
|
2217
2429
|
if (useBgQueue) queuedBubble._useBgQueue = true;
|
|
2218
|
-
this.state.messages
|
|
2430
|
+
var qStage = this._stageIndex(this.state.messages, stageId);
|
|
2431
|
+
if (qStage !== -1) {
|
|
2432
|
+
if (this.state.messages[qStage]._ts !== void 0) queuedBubble._ts = this.state.messages[qStage]._ts;
|
|
2433
|
+
var qTarget = this._settledStagePosition(qStage);
|
|
2434
|
+
if (qTarget === -1) {
|
|
2435
|
+
this.state.messages.splice(qStage, 1, queuedBubble);
|
|
2436
|
+
} else {
|
|
2437
|
+
this.state.messages.splice(qStage, 1);
|
|
2438
|
+
this.state.messages.splice(qTarget, 0, queuedBubble);
|
|
2439
|
+
}
|
|
2440
|
+
} else {
|
|
2441
|
+
this.state.messages.push(queuedBubble);
|
|
2442
|
+
}
|
|
2219
2443
|
this.host.notify();
|
|
2220
2444
|
this.updateHistoryCache();
|
|
2221
2445
|
this.host.scrollToBottom(true);
|
|
2222
2446
|
var capturedComposed = composed, capturedPlatform = aiPlatform, capturedKey = key;
|
|
2223
2447
|
Promise.resolve(this._callProviderFor(aiPlatform, composed, boundedQ.messages, systemPrompt, aiModel, chatQueue, extractContent, fileUrls, id.serviceId, id.owner)).then(function(result) {
|
|
2224
2448
|
var sendingIdx = self.getHistoryCacheKey() !== capturedKey ? -1 : self.state.messages.findIndex(function(m) {
|
|
2225
|
-
return m.isSendingToServer && (m.isPendingQueued || m.isPendingInProcess) && m.role === "user" && (m._ownerKey === void 0 || m._ownerKey === capturedKey);
|
|
2449
|
+
return m.isSendingToServer && (m.isPendingQueued || m.isPendingInProcess) && m.role === "user" && !m._stageId && (m._ownerKey === void 0 || m._ownerKey === capturedKey);
|
|
2226
2450
|
});
|
|
2227
2451
|
var serverId = result && typeof result.id === "string" ? result.id : void 0;
|
|
2228
2452
|
if (sendingIdx >= 0) {
|
|
@@ -2247,23 +2471,31 @@ var ChatSession = class {
|
|
|
2247
2471
|
});
|
|
2248
2472
|
return;
|
|
2249
2473
|
}
|
|
2250
|
-
|
|
2251
|
-
|
|
2474
|
+
var immediateUser = { role: "user", content: composed, _ts: wallClockNow(), ...key ? { _ownerKey: key } : {} };
|
|
2475
|
+
var immediatePlaceholder = { role: "assistant", content: "", isPending: true, isPendingInProcess: true, ...key ? { _ownerKey: key } : {} };
|
|
2476
|
+
var iStage = this._stageIndex(this.state.messages, stageId);
|
|
2477
|
+
if (iStage !== -1) {
|
|
2478
|
+
if (this.state.messages[iStage]._ts !== void 0) immediateUser._ts = this.state.messages[iStage]._ts;
|
|
2479
|
+
var iTarget = this._settledStagePosition(iStage);
|
|
2480
|
+
if (iTarget === -1) {
|
|
2481
|
+
this.state.messages.splice(iStage, 1, immediateUser, immediatePlaceholder);
|
|
2482
|
+
} else {
|
|
2483
|
+
this.state.messages.splice(iStage, 1);
|
|
2484
|
+
this.state.messages.splice(iTarget, 0, immediateUser, immediatePlaceholder);
|
|
2485
|
+
}
|
|
2486
|
+
} else {
|
|
2487
|
+
this.state.messages.push(immediateUser);
|
|
2488
|
+
this.state.messages.push(immediatePlaceholder);
|
|
2489
|
+
}
|
|
2252
2490
|
this.host.notify();
|
|
2253
2491
|
this.updateHistoryCache();
|
|
2254
2492
|
this.state.sending = true;
|
|
2255
2493
|
this.host.scrollToBottom(true);
|
|
2256
2494
|
var historyForLlm = this.state.messages.filter(function(m) {
|
|
2495
|
+
if (m === immediateUser) return false;
|
|
2257
2496
|
return !m.isPending && !m.isPendingQueued && !m.isPendingInProcess && !m.isPendingOlder && !m.isCancelled && !m.isBackgroundTask && !m.isError;
|
|
2258
2497
|
});
|
|
2259
|
-
|
|
2260
|
-
for (var li = historyForLlm.length - 1; li >= 0; li--) {
|
|
2261
|
-
if (historyForLlm[li].role === "user" && historyForLlm[li].content === composed) {
|
|
2262
|
-
historyForLlm[li] = Object.assign({}, historyForLlm[li], { content: llmComposed });
|
|
2263
|
-
break;
|
|
2264
|
-
}
|
|
2265
|
-
}
|
|
2266
|
-
}
|
|
2498
|
+
historyForLlm.push({ role: "user", content: llmComposed });
|
|
2267
2499
|
var bounded = buildBoundedChatMessages({
|
|
2268
2500
|
platform: aiPlatform,
|
|
2269
2501
|
model: aiModel,
|
|
@@ -2504,7 +2736,7 @@ var ChatSession = class {
|
|
|
2504
2736
|
if (platform !== "claude" && platform !== "openai") return;
|
|
2505
2737
|
var url = platform === "claude" ? ANTHROPIC_MESSAGES_API_URL : OPENAI_RESPONSES_API_URL;
|
|
2506
2738
|
var queueBase = id.userId || id.serviceId;
|
|
2507
|
-
var queue = msg.isBackgroundTask || msg._useBgQueue ? queueBase
|
|
2739
|
+
var queue = msg.isBackgroundTask || msg._useBgQueue ? bgIndexingQueueName(queueBase) : queueBase;
|
|
2508
2740
|
this.state.messages[idx] = Object.assign({}, msg, { _cancelling: true, _cancelError: void 0 });
|
|
2509
2741
|
this.host.notify();
|
|
2510
2742
|
Promise.resolve(this.host.cancelRequest({
|
|
@@ -2990,7 +3222,7 @@ var ChatSession = class {
|
|
|
2990
3222
|
if (!id.serviceId || platform !== "claude" && platform !== "openai") return;
|
|
2991
3223
|
if (this.isPollingPaused() || !this.host.isViewMounted()) return;
|
|
2992
3224
|
var svcId = id.serviceId, owner = id.owner;
|
|
2993
|
-
var queue = (id.userId
|
|
3225
|
+
var queue = bgIndexingQueueName(id.userId, id.serviceId);
|
|
2994
3226
|
var ask = function(status) {
|
|
2995
3227
|
return Promise.resolve(getChatHistory(
|
|
2996
3228
|
{ service: svcId, owner, platform, queue, status },
|
|
@@ -3092,7 +3324,7 @@ var ChatSession = class {
|
|
|
3092
3324
|
url,
|
|
3093
3325
|
method: "POST",
|
|
3094
3326
|
id: serverId,
|
|
3095
|
-
queue: (id.userId
|
|
3327
|
+
queue: bgIndexingQueueName(id.userId, id.serviceId),
|
|
3096
3328
|
service: id.serviceId,
|
|
3097
3329
|
owner: id.owner
|
|
3098
3330
|
})).catch(function() {
|
|
@@ -3221,7 +3453,7 @@ var ChatSession = class {
|
|
|
3221
3453
|
if (pass > MAX_INDEXING_RESUME_PASSES) return;
|
|
3222
3454
|
var id = this.host.getIdentity();
|
|
3223
3455
|
if (!id || id.platform === "none" || id.serviceId !== entry.serviceId) return;
|
|
3224
|
-
notifyAgentContinueIndexing({
|
|
3456
|
+
this.trackIndexDispatch(notifyAgentContinueIndexing({
|
|
3225
3457
|
platform: id.platform,
|
|
3226
3458
|
model: id.model,
|
|
3227
3459
|
service: id.serviceId,
|
|
@@ -3255,7 +3487,7 @@ var ChatSession = class {
|
|
|
3255
3487
|
}
|
|
3256
3488
|
}, function(e) {
|
|
3257
3489
|
console.error("[chat-engine] resume-indexing dispatch failed", e);
|
|
3258
|
-
});
|
|
3490
|
+
}));
|
|
3259
3491
|
} catch (e) {
|
|
3260
3492
|
}
|
|
3261
3493
|
}
|
|
@@ -3333,6 +3565,10 @@ var ChatSession = class {
|
|
|
3333
3565
|
if (mm._ownerKey !== void 0 && mm._ownerKey !== loadKey) continue;
|
|
3334
3566
|
if (mm._serverItemId && serverIds[mm._serverItemId]) continue;
|
|
3335
3567
|
if (!mm._serverItemId) {
|
|
3568
|
+
if (mm._stageId) {
|
|
3569
|
+
rescued.push(mm);
|
|
3570
|
+
continue;
|
|
3571
|
+
}
|
|
3336
3572
|
if (mappedHasPendingAssistant) continue;
|
|
3337
3573
|
if (mm.isSendingToServer || mm.isPendingQueued || mm.isPendingInProcess || mm.isPending) rescued.push(mm);
|
|
3338
3574
|
else if (self.state.sending && mm.role === "user") {
|
|
@@ -3552,7 +3788,7 @@ var ChatSession = class {
|
|
|
3552
3788
|
return preIndex.then(function() {
|
|
3553
3789
|
return parseAttachmentContent(member.file, member.file.name, mime || void 0);
|
|
3554
3790
|
}).then(function(parsedContent) {
|
|
3555
|
-
return notifyAgentSaveAttachment({
|
|
3791
|
+
return self.trackIndexDispatch(notifyAgentSaveAttachment({
|
|
3556
3792
|
platform: id.platform,
|
|
3557
3793
|
model: id.model,
|
|
3558
3794
|
service: id.serviceId,
|
|
@@ -3591,7 +3827,7 @@ var ChatSession = class {
|
|
|
3591
3827
|
att.errorCode = e && (e.code || e.body && e.body.code) || "";
|
|
3592
3828
|
att.errorDetail = e && (e.message || e.body && e.body.message) || (typeof e === "string" ? e : "");
|
|
3593
3829
|
}
|
|
3594
|
-
});
|
|
3830
|
+
}));
|
|
3595
3831
|
});
|
|
3596
3832
|
});
|
|
3597
3833
|
});
|
|
@@ -3610,14 +3846,24 @@ var ChatSession = class {
|
|
|
3610
3846
|
}
|
|
3611
3847
|
// Upload all not-yet-done attachments sequentially. Resolves to the full
|
|
3612
3848
|
// list of { name, url, storagePath } for composing the chat message.
|
|
3613
|
-
|
|
3849
|
+
//
|
|
3850
|
+
// `batchId` scopes the run to the chips stamped with it at Send time. The
|
|
3851
|
+
// composer stays live during an upload, so by the time this runs the
|
|
3852
|
+
// attachment list can already hold chips the user picked for the NEXT
|
|
3853
|
+
// message — uploading those here would attach them to the wrong turn, and
|
|
3854
|
+
// collecting the previous batch's finished urls would attach files the user
|
|
3855
|
+
// already sent. Omitted (no batch) means every chip, the old behavior.
|
|
3856
|
+
uploadPendingAttachments(batchId) {
|
|
3614
3857
|
var self = this;
|
|
3615
3858
|
this.host.resetOverwriteBatch();
|
|
3859
|
+
this._uploadBatches += 1;
|
|
3616
3860
|
this.state.uploadingAttachments = true;
|
|
3617
3861
|
this.host.updateComposerControls();
|
|
3618
3862
|
this.host.renderAttachmentChips();
|
|
3619
3863
|
var collected = [];
|
|
3620
|
-
var snapshot = this.state.attachments.
|
|
3864
|
+
var snapshot = this.state.attachments.filter(function(a) {
|
|
3865
|
+
return batchId ? a._batchId === batchId : true;
|
|
3866
|
+
});
|
|
3621
3867
|
var chain = Promise.resolve();
|
|
3622
3868
|
snapshot.forEach(function(att) {
|
|
3623
3869
|
chain = chain.then(function() {
|
|
@@ -3653,7 +3899,8 @@ var ChatSession = class {
|
|
|
3653
3899
|
});
|
|
3654
3900
|
});
|
|
3655
3901
|
var done = function() {
|
|
3656
|
-
self.
|
|
3902
|
+
self._uploadBatches = Math.max(0, self._uploadBatches - 1);
|
|
3903
|
+
self.state.uploadingAttachments = self._uploadBatches > 0;
|
|
3657
3904
|
self.host.updateComposerControls();
|
|
3658
3905
|
self.host.renderAttachmentChips();
|
|
3659
3906
|
return collected;
|
|
@@ -3892,6 +4139,7 @@ exports.RTF_EXTS = RTF_EXTS;
|
|
|
3892
4139
|
exports.TOOL_AND_RESPONSE_BUFFER = TOOL_AND_RESPONSE_BUFFER;
|
|
3893
4140
|
exports.XML_EXTS = XML_EXTS;
|
|
3894
4141
|
exports.applyEncodingDeclaration = applyEncodingDeclaration;
|
|
4142
|
+
exports.bgIndexingQueueName = bgIndexingQueueName;
|
|
3895
4143
|
exports.buildAiAgentValue = buildAiAgentValue;
|
|
3896
4144
|
exports.buildBoundedChatMessages = buildBoundedChatMessages;
|
|
3897
4145
|
exports.buildChatDisplayList = buildChatDisplayList;
|