@twentyfourg/chat-kit 1.0.0-beta.3 → 1.0.0-beta.4

package/dist/chat-kit.js

@@ -717,7 +717,7 @@ function Qe(e) {
 			inputMode: i
 		});
 		o.value.push(u);
-		let d = b({
+		let m = b({
 			id: K(),
 			role: "assistant",
 			text: "",
@@ -731,64 +731,71 @@ function Qe(e) {
 			isReceiving: !1,
 			failed: !1
 		});
-		p.value = d;
-		let m = new AbortController();
-		g = m;
-		let h = new Ke((e) => {
-			d.html = e;
+		p.value = m;
+		let h = new AbortController();
+		g = h;
+		let _ = new Ke((e) => {
+			m.html = e;
 		}, 25, e.showMath);
-		function _() {
-			s.value !== "streaming" && (s.value = "streaming", d.isStreaming = !0, d.isReceiving = !0, o.value.push(d));
+		function v() {
+			s.value !== "streaming" && (s.value = "streaming", m.isStreaming = !0, m.isReceiving = !0, o.value.push(m));
 		}
-		let v = new Ye((e) => {
-			_(), d.text += e, h.addChunk(e);
-		}, e.typewriter ?? !0), y = null;
+		let y = new Ye((e) => {
+			v(), m.text += e, _.addChunk(e);
+		}, e.typewriter ?? !0), x = null, S = null;
 		try {
 			let o = await D();
 			if (await U(await t.sendMessage(r, o, {
 				message: n,
 				stream: !0,
 				...i === "voice" && { inputMode: "dictation" }
-			}, { signal: m.signal }), {
+			}, { signal: h.signal }), {
 				onChunk: (e) => {
-					v.push(e);
+					y.push(e);
 				},
 				onComplete: () => {},
 				onError: (e) => {
-					y = e;
+					x = e;
 				},
 				onMetadata: (e) => {
-					e.sources && (d.sources = e.sources);
+					e.sources && (m.sources = e.sources);
+				},
+				onRateLimit: ({ limit: e, remaining: t, resetSeconds: n }) => {
+					S = {
+						limit: e,
+						remaining: t,
+						resetAt: Number.isFinite(n) ? (/* @__PURE__ */ new Date(n * 1e3)).toISOString() : ""
+					};
 				},
 				onThinkingChunk: (e) => {
-					d.thinking += e;
+					m.thinking += e;
 				}
 			}, (e, t) => {
-				if (e === "citations" && Array.isArray(t)) return d.citations = t, !0;
+				if (e === "citations" && Array.isArray(t)) return m.citations = t, !0;
 				let n = t ?? {};
-				return e === "message_id" && typeof n.messageId == "string" ? (u.basecampMessageId = n.messageId, !0) : e === "assistant_message_id" && typeof n.messageId == "string" ? (d.basecampMessageId = n.messageId, !0) : e === "used_sources" && Array.isArray(t) ? (d.sources = t, !0) : e === "token_usage" && n.thread ? (l.value = n.thread, !0) : !1;
-			}), y) throw y;
-			if (m.signal.aborted) {
-				v.flush(), await h.complete(), d.isReceiving = !1, d.isStreaming = !1;
+				return e === "message_id" && typeof n.messageId == "string" ? (u.basecampMessageId = n.messageId, !0) : e === "assistant_message_id" && typeof n.messageId == "string" ? (m.basecampMessageId = n.messageId, !0) : e === "used_sources" && Array.isArray(t) ? (m.sources = t, !0) : e === "token_usage" && n.thread ? (l.value = n.thread, !0) : !1;
+			}), x) throw x;
+			if (h.signal.aborted) {
+				y.flush(), await _.complete(), m.isReceiving = !1, m.isStreaming = !1;
 				return;
 			}
-			if (d.isReceiving = !1, await v.finish(), !d.text) throw Error("The response ended without any content.");
-			await h.complete(), d.isOutOfScope = e.isOutOfScope?.(d.text) ?? !1, d.isStreaming = !1, w({
+			if (m.isReceiving = !1, await y.finish(), !m.text) throw Error("The response ended without any content.");
+			await _.complete(), m.isOutOfScope = e.isOutOfScope?.(m.text) ?? !1, m.isStreaming = !1, w({
 				type: "turn",
 				query: n,
 				inputMode: i,
-				response: d.text,
-				isOutOfScope: d.isOutOfScope,
-				citations: d.citations,
-				sources: d.sources,
+				response: m.text,
+				isOutOfScope: m.isOutOfScope,
+				citations: m.citations,
+				sources: m.sources,
 				threadId: a.value,
 				userMessageId: u.basecampMessageId,
-				assistantMessageId: d.basecampMessageId
+				assistantMessageId: m.basecampMessageId
 			});
 		} catch (t) {
-			v.flush(), t?.name !== "AbortError" && (d.text ? d.failed = !0 : c.value = t instanceof Error ? t : Error(String(t))), h.stop(), d.isReceiving = !1, d.isStreaming = !1, d.text && (d.html = await W(d.text, { showMath: e.showMath }));
+			y.flush(), t?.name !== "AbortError" && (m.text ? m.failed = !0 : c.value = t instanceof Error ? t : Error(String(t))), _.stop(), m.isReceiving = !1, m.isStreaming = !1, m.text && (m.html = await W(m.text, { showMath: e.showMath }));
 		} finally {
-			v.destroy(), g = null, p.value = null, s.value = "idle", f();
+			y.destroy(), g = null, p.value = null, s.value = "idle", S ? d.value = S : f();
 		}
 	}
 	async function A() {
@@ -1034,9 +1041,22 @@ var J = (e) => new Promise((t) => setTimeout(t, e)), et = class {
 		} });
 		return new Response(m, {
 			status: 200,
-			headers: { "Content-Type": "text/event-stream" }
+			headers: {
+				"Content-Type": "text/event-stream",
+				...this.rateLimitHeaders()
+			}
 		});
 	}
+	rateLimitHeaders() {
+		let e = this.options.messageLimit;
+		if (e === void 0) return { "X-RateLimit-Limit": "unlimited" };
+		let t = Math.ceil(this.messageCount / 2);
+		return {
+			"X-RateLimit-Limit": String(e),
+			"X-RateLimit-Remaining": String(Math.max(0, e - t)),
+			"X-RateLimit-Reset": String(Math.floor(new Date(this.quotaResetAt).getTime() / 1e3))
+		};
+	}
 	async stopMessage() {
 		return {
 			success: !0,
@@ -1730,6 +1750,7 @@ var Mt = { class: "citation-popover__source" }, Nt = ["src"], Pt = { class: "cit
 			e.message.isOutOfScope ? (v(), a("div", Nn, [l(Z, { name: "warning" }), o("span", null, E(D(d).outOfScope.label), 1)])) : i("", !0),
 			e.showThinking && e.message.thinking ? (v(), r(On, {
 				key: 1,
+				class: "assistant-message__reasoning",
 				text: e.message.thinking
 			}, null, 8, ["text"])) : i("", !0),
 			o("div", Pn, [o("div", {
@@ -1771,7 +1792,7 @@ var Mt = { class: "citation-popover__source" }, Nt = ["src"], Pt = { class: "cit
 			w(n.$slots, "message-footer", { message: e.message }, void 0, !0)
 		]));
 	}
-}), [["__scopeId", "data-v-466b7413"]]), Un = { class: "thinking" }, Wn = {
+}), [["__scopeId", "data-v-0cff2e6c"]]), Un = { class: "thinking" }, Wn = {
 	key: 0,
 	class: "thinking__label"
 }, Gn = 2e3, Kn = /*#__PURE__*/ Y(/* @__PURE__ */ u({