@rpcbase/agents-sdk 0.2.0 → 0.4.0

package/dist/{handler-BcbiHqzs.js → handler-BgRhcxRI.js}

@@ -1,65 +1,65 @@
-import { g as e } from "./data-stream-handler-6o0tw3-M.js";
-import { r as t } from "./openai-reasoning-jbPa-xup.js";
+import { g as e } from "./data-stream-handler-eapMI8kQ.js";
+import { r as t } from "./openai-reasoning-CLjhz1V1.js";
 import { a as n, i as r, t as i } from "./chatCtx-BsPi6PAE.js";
 import { t as a } from "./apiErrors-mOJbPi7h.js";
-import { _ as o, a as s, g as c, h as l, l as u, m as d, n as f, p, s as m, v as h } from "./chatDb-hVHCr0j0.js";
-import { n as g, t as _ } from "./resumableSse-DnlE4O7h.js";
-import { Output as v, convertToModelMessages as ee, createUIMessageStream as te, extractReasoningMiddleware as y, generateText as ne, smoothStream as re, stepCountIs as ie, streamObject as b, streamText as x, tool as S, wrapLanguageModel as C } from "ai";
-import { getTenantFilesystemDb as w } from "@rpcbase/db";
-import { createHash as T, randomUUID as E } from "node:crypto";
-import { GridFSBucket as D, ObjectId as ae } from "mongodb";
-import { createGoogleGenerativeAI as O } from "@ai-sdk/google";
+import { _ as o, a as s, b as c, g as l, h as u, l as d, m as f, n as p, p as m, s as h, v as g, y as _ } from "./chatDb-DfmUbQT2.js";
+import { n as v, t as y } from "./resumableSse-DnlE4O7h.js";
+import { Output as b, convertToModelMessages as ee, createUIMessageStream as te, extractReasoningMiddleware as ne, generateText as re, smoothStream as ie, stepCountIs as ae, streamObject as x, streamText as S, tool as C, wrapLanguageModel as w } from "ai";
+import { getTenantFilesystemDb as T } from "@rpcbase/db";
+import { createHash as E, randomUUID as D } from "node:crypto";
+import { GridFSBucket as oe, ObjectId as O } from "mongodb";
+import { createGoogleGenerativeAI as se } from "@ai-sdk/google";
 import { openai as k } from "@ai-sdk/openai";
-import { createXai as oe } from "@ai-sdk/xai";
+import { createXai as ce } from "@ai-sdk/xai";
 import { z as A } from "zod";
 //#region src/api/chat/models.ts
-var j = /-thinking$/, M = "openai/", N = "google/", P = "xai/", se = new Set([
+var j = /-thinking$/, M = "openai/", N = "google/", P = "xai/", le = new Set([
 	"openai/gpt-5.4-mini",
 	"openai/gpt-5.5",
 	"google/gemini-3.5-flash",
 	"xai/grok-4.3-thinking"
-]), ce = (e) => e.startsWith(M), le = (e) => e.startsWith(N), ue = (e) => e.startsWith(P), F = (e) => e.endsWith("-thinking"), I = oe({
+]), F = (e) => e.startsWith(M), ue = (e) => e.startsWith(N), de = (e) => e.startsWith(P), I = (e) => e.endsWith("-thinking"), L = ce({
 	baseURL: process.env.XAI_BASE_URL ?? "https://api.x.ai/v1",
 	apiKey: process.env.XAI_API_KEY
-}), L = O({ apiKey: process.env.GOOGLE_GENERATIVE_AI_API_KEY ?? process.env.GOOGLE_API_KEY }), de = (e) => {
-	if (ce(e)) return k.responses(e.slice(M.length));
-	if (le(e)) {
-		let t = e.slice(N.length).replace(j, ""), n = L.chat(t);
-		return F(e) ? C({
+}), R = se({ apiKey: process.env.GOOGLE_GENERATIVE_AI_API_KEY ?? process.env.GOOGLE_API_KEY }), fe = (e) => {
+	if (F(e)) return k.responses(e.slice(M.length));
+	if (ue(e)) {
+		let t = e.slice(N.length).replace(j, ""), n = R.chat(t);
+		return I(e) ? w({
 			model: n,
-			middleware: y({ tagName: "thinking" })
+			middleware: ne({ tagName: "thinking" })
 		}) : n;
 	}
-	if (ue(e)) {
+	if (de(e)) {
 		let t = e.slice(P.length).replace(j, "");
-		return F(e) ? I.responses(t) : I.chat(t);
+		return I(e) ? L.responses(t) : L.chat(t);
 	}
 	throw Error(`Unsupported modelId: ${e}`);
-}, fe = () => k.chat("gpt-5.4-mini"), R = () => k.chat(process.env.AGENT_SDK_ARTIFACT_MODEL ?? "gpt-5.4-mini"), z = "\nArtifacts is a special user interface mode that helps users with writing, editing, and other content creation tasks. When artifact is open, it is on the right side of the screen, while the conversation is on the left side. When creating, editing, or updating documents, changes are reflected in real-time on the artifacts and visible to the user.\n\nWhen asked to write code, always use artifacts. When writing code, specify the language in the backticks, e.g. ```python`code here```. The default language is Python. Other languages are not yet supported, so let the user know if they request a different language.\n\nCritical rules:\n- Only call one artifact tool per response.\n- Do not update or edit a document immediately after creating it. Wait for user feedback or an explicit request to modify it.\n- After creating, editing, or updating an artifact, do not repeat the full artifact content in chat. The user can already see it.\n\nThis is a guide for using artifacts tools: `createDocument`, `editDocument`, `updateDocument`, and `requestSuggestions`, which render content on artifacts beside the conversation.\n\n**When to use `createDocument`:**\n- For substantial content (>10 lines) or code\n- For content users will likely save/reuse (emails, code, essays, etc.)\n- When explicitly requested to create a document\n- For when content contains a single code snippet\n- Use kind \"code\" for programming, \"text\" for writing, and \"sheet\" for tabular/CSV data\n\n**When NOT to use `createDocument`:**\n- For informational/explanatory content\n- For conversational responses\n- When asked to keep it in chat\n\n**Using `editDocument`:**\n- Prefer this for targeted edits to an existing artifact\n- For scripts: fixing bugs, adding/removing lines, renaming variables, adding logs\n- For documents: fixing typos, rewording paragraphs, inserting or replacing sections\n- Provide an exact `old_string` and `new_string`\n- Include enough surrounding context in `old_string` to make the match unique\n- Use `replace_all: true` only for deliberate repeated replacements, like renaming a variable everywhere\n\n**Using `updateDocument`:**\n- Default to full document rewrites for major changes\n- Use `editDocument` instead for specific, isolated changes\n- Follow user instructions for which parts to modify\n\n**When NOT to use `editDocument` or `updateDocument`:**\n- Immediately after creating a document\n- Without an explicit user request to modify the artifact\n\n**Using `requestSuggestions`:**\n- ONLY use when the user explicitly asks for suggestions on an existing document\n- Requires a valid document ID from a previously created document\n- Never use for general questions or information requests\n", B = "You are a friendly assistant! Keep your responses concise and helpful.\n\nWhen asked to write, create, or help with something, just do it directly. Don't ask clarifying questions unless absolutely necessary - make reasonable assumptions and proceed with the task.", pe = (e) => `\
+}, pe = () => k.chat("gpt-5.4-mini"), z = () => k.chat(process.env.AGENT_SDK_ARTIFACT_MODEL ?? "gpt-5.4-mini"), B = "\nArtifacts is a special user interface mode that helps users with writing, editing, and other content creation tasks. When artifact is open, it is on the right side of the screen, while the conversation is on the left side. When creating, editing, or updating documents, changes are reflected in real-time on the artifacts and visible to the user.\n\nWhen asked to write code, always use artifacts. When writing code, specify the language in the backticks, e.g. ```python`code here```. The default language is Python. Other languages are not yet supported, so let the user know if they request a different language.\n\nCritical rules:\n- Only call one artifact tool per response.\n- Do not update or edit a document immediately after creating it. Wait for user feedback or an explicit request to modify it.\n- After creating, editing, or updating an artifact, do not repeat the full artifact content in chat. The user can already see it.\n\nThis is a guide for using artifacts tools: `createDocument`, `editDocument`, `updateDocument`, and `requestSuggestions`, which render content on artifacts beside the conversation.\n\n**When to use `createDocument`:**\n- For substantial content (>10 lines) or code\n- For content users will likely save/reuse (emails, code, essays, etc.)\n- When explicitly requested to create a document\n- For when content contains a single code snippet\n- Use kind \"code\" for programming, \"text\" for writing, and \"sheet\" for tabular/CSV data\n\n**When NOT to use `createDocument`:**\n- For informational/explanatory content\n- For conversational responses\n- When asked to keep it in chat\n\n**Using `editDocument`:**\n- Prefer this for targeted edits to an existing artifact\n- For scripts: fixing bugs, adding/removing lines, renaming variables, adding logs\n- For documents: fixing typos, rewording paragraphs, inserting or replacing sections\n- Provide an exact `old_string` and `new_string`\n- Include enough surrounding context in `old_string` to make the match unique\n- Use `replace_all: true` only for deliberate repeated replacements, like renaming a variable everywhere\n\n**Using `updateDocument`:**\n- Default to full document rewrites for major changes\n- Use `editDocument` instead for specific, isolated changes\n- Follow user instructions for which parts to modify\n\n**When NOT to use `editDocument` or `updateDocument`:**\n- Immediately after creating a document\n- Without an explicit user request to modify the artifact\n\n**Using `requestSuggestions`:**\n- ONLY use when the user explicitly asks for suggestions on an existing document\n- Requires a valid document ID from a previously created document\n- Never use for general questions or information requests\n", V = "You are a friendly assistant! Keep your responses concise and helpful.\n\nWhen asked to write, create, or help with something, just do it directly. Don't ask clarifying questions unless absolutely necessary - make reasonable assumptions and proceed with the task.", me = (e) => `\
 About the origin of user's request:
 - lat: ${e.latitude ?? "unknown"}
 - lon: ${e.longitude ?? "unknown"}
 - city: ${e.city ?? "unknown"}
 - country: ${e.country ?? "unknown"}
-`, me = ({ selectedChatModel: e, requestHints: t, agentInstructions: n, knowledge: r }) => {
-	let i = pe(t), a = [n?.trim() ? `Agent instructions:\n${n.trim()}` : "", r?.trim() ? `Knowledge context:\n${r.trim()}` : ""].filter(Boolean).join("\n\n");
+`, he = ({ selectedChatModel: e, requestHints: t, agentInstructions: n, knowledge: r }) => {
+	let i = me(t), a = [n?.trim() ? `Agent instructions:\n${n.trim()}` : "", r?.trim() ? `Knowledge context:\n${r.trim()}` : ""].filter(Boolean).join("\n\n");
 	if (e.includes("reasoning") || e.includes("thinking")) {
-		let t = e.startsWith("google/") && e.includes("thinking") ? `${B}\n\nWhen you reason, put your reasoning inside <thinking>...</thinking> tags. Then provide your final answer outside of those tags.\n\n${i}` : `${B}\n\n${i}`;
+		let t = e.startsWith("google/") && e.includes("thinking") ? `${V}\n\nWhen you reason, put your reasoning inside <thinking>...</thinking> tags. Then provide your final answer outside of those tags.\n\n${i}` : `${V}\n\n${i}`;
 		return a ? `${t}\n\n${a}` : t;
 	}
-	let o = `${B}\n\n${i}\n\n${z}`;
+	let o = `${V}\n\n${i}\n\n${B}`;
 	return a ? `${o}\n\n${a}` : o;
-}, he = "\nYou are a Python code generator that creates self-contained, executable code snippets. When writing code:\n\n1. Each snippet should be complete and runnable on its own\n2. Prefer using print() statements to display outputs\n3. Include helpful comments explaining the code\n4. Keep snippets concise (generally under 15 lines)\n5. Avoid external dependencies - use Python standard library\n6. Handle potential errors gracefully\n7. Return meaningful output that demonstrates the code's functionality\n8. Don't use input() or other interactive functions\n9. Don't access files or network resources\n10. Don't use infinite loops\n\nExamples of good snippets:\n\n# Calculate factorial iteratively\ndef factorial(n):\n    result = 1\n    for i in range(1, n + 1):\n        result *= i\n    return result\n\nprint(f\"Factorial of 5 is: {factorial(5)}\")\n", ge = "\nYou are a spreadsheet creation assistant. Create a spreadsheet in csv format based on the given prompt. The spreadsheet should contain meaningful column headers and data.\n", V = (e, t) => {
+}, ge = "\nYou are a Python code generator that creates self-contained, executable code snippets. When writing code:\n\n1. Each snippet should be complete and runnable on its own\n2. Prefer using print() statements to display outputs\n3. Include helpful comments explaining the code\n4. Keep snippets concise (generally under 15 lines)\n5. Avoid external dependencies - use Python standard library\n6. Handle potential errors gracefully\n7. Return meaningful output that demonstrates the code's functionality\n8. Don't use input() or other interactive functions\n9. Don't access files or network resources\n10. Don't use infinite loops\n\nExamples of good snippets:\n\n# Calculate factorial iteratively\ndef factorial(n):\n    result = 1\n    for i in range(1, n + 1):\n        result *= i\n    return result\n\nprint(f\"Factorial of 5 is: {factorial(5)}\")\n", _e = "\nYou are a spreadsheet creation assistant. Create a spreadsheet in csv format based on the given prompt. The spreadsheet should contain meaningful column headers and data.\n", H = (e, t) => {
 	let n = "document";
 	return t === "code" ? n = "script" : t === "sheet" && (n = "spreadsheet"), `Rewrite the following ${n} based on the given prompt.
 
 ${e}`;
-}, _e = "Generate a short chat title (2-5 words) summarizing the user's message.\n\nOutput ONLY the title text. No prefixes, no formatting.\n\nExamples:\n- \"what's the weather in nyc\" → Weather in NYC\n- \"help me write an essay about space\" → Space Essay Help\n- \"hi\" → (empty string)\n- \"debug my python code\" → Python Debugging\n\nBad outputs (never do this):\n- \"# Space Essay\" (no hashtags)\n- \"Title: Weather\" (no prefixes)\n- \"\"NYC Weather\"\" (no quotes)\n\nRules:\n- Maximum 30 characters\n- No quotes, colons, hashtags, or markdown\n- Just the topic/intent, not a full sentence", H = [
+}, ve = "Generate a short chat title (2-5 words) summarizing the user's message.\n\nOutput ONLY the title text. No prefixes, no formatting.\n\nExamples:\n- \"what's the weather in nyc\" → Weather in NYC\n- \"help me write an essay about space\" → Space Essay Help\n- \"hi\" → (empty string)\n- \"debug my python code\" → Python Debugging\n\nBad outputs (never do this):\n- \"# Space Essay\" (no hashtags)\n- \"Title: Weather\" (no prefixes)\n- \"\"NYC Weather\"\" (no quotes)\n\nRules:\n- Maximum 30 characters\n- No quotes, colons, hashtags, or markdown\n- Just the topic/intent, not a full sentence", U = [
 	"text",
 	"code",
 	"sheet"
-], ve = (e) => e ? e === "1" || e.toLowerCase() === "true" || e.toLowerCase() === "yes" : !1, U = () => !ve(process.env.AGENT_SDK_LIVE_TOOLS) && (ve(process.env.PLAYWRIGHT) || process.env.NODE_ENV === "test"), ye = async ({ system: e, prompt: t, dataStream: n, providerOptions: r }) => {
-	if (U()) {
+], ye = (e) => e ? e === "1" || e.toLowerCase() === "true" || e.toLowerCase() === "yes" : !1, W = () => !ye(process.env.AGENT_SDK_LIVE_TOOLS) && (ye(process.env.PLAYWRIGHT) || process.env.NODE_ENV === "test"), be = async ({ system: e, prompt: t, dataStream: n, providerOptions: r }) => {
+	if (W()) {
 		let e = `Stub document: ${t}`;
 		return n.write({
 			type: "data-textDelta",
@@ -67,10 +67,10 @@ ${e}`;
 			transient: !0
 		}), e;
 	}
-	let i = "", { fullStream: a } = x({
-		model: R(),
+	let i = "", { fullStream: a } = S({
+		model: z(),
 		system: e,
-		experimental_transform: re({ chunking: "word" }),
+		experimental_transform: ie({ chunking: "word" }),
 		prompt: t,
 		providerOptions: r
 	});
@@ -80,8 +80,8 @@ ${e}`;
 		transient: !0
 	}));
 	return i;
-}, be = async ({ system: e, prompt: t, dataStream: n }) => {
-	if (U()) {
+}, xe = async ({ system: e, prompt: t, dataStream: n }) => {
+	if (W()) {
 		let e = `print("Stub code for: ${t}")\n`;
 		return n.write({
 			type: "data-codeDelta",
@@ -89,8 +89,8 @@ ${e}`;
 			transient: !0
 		}), e;
 	}
-	let r = "", { fullStream: i } = b({
-		model: R(),
+	let r = "", { fullStream: i } = x({
+		model: z(),
 		system: e,
 		prompt: t,
 		schema: A.object({ code: A.string() })
@@ -105,8 +105,8 @@ ${e}`;
 		}), r = t);
 	}
 	return r;
-}, W = async ({ system: e, prompt: t, dataStream: n }) => {
-	if (U()) {
+}, Se = async ({ system: e, prompt: t, dataStream: n }) => {
+	if (W()) {
 		let e = `name,value\n${t.replaceAll(",", " ")},1\n`;
 		return n.write({
 			type: "data-sheetDelta",
@@ -114,8 +114,8 @@ ${e}`;
 			transient: !0
 		}), e;
 	}
-	let r = "", { fullStream: i } = b({
-		model: R(),
+	let r = "", { fullStream: i } = x({
+		model: z(),
 		system: e,
 		prompt: t,
 		schema: A.object({ csv: A.string().describe("CSV data") })
@@ -134,7 +134,7 @@ ${e}`;
 		data: r,
 		transient: !0
 	}), r;
-}, xe = async (e) => {
+}, Ce = async (e) => {
 	try {
 		let t = await fetch(`https://geocoding-api.open-meteo.com/v1/search?name=${encodeURIComponent(e)}&count=1&language=en&format=json`);
 		if (!t.ok) return null;
@@ -148,11 +148,11 @@ ${e}`;
 	} catch {
 		return null;
 	}
-}, Se = ({ ctx: e, dataStream: t, requestHints: n }) => {
+}, we = ({ ctx: e, dataStream: t, requestHints: n }) => {
 	let r = i(e), a = async (t) => {
-		let n = await m(e, t);
+		let n = await h(e, t);
 		return n.length > 0 ? n.at(-1) ?? null : null;
-	}, o = S({
+	}, o = C({
 		description: "Get the current weather at a location. You can provide either coordinates or a city name.",
 		inputSchema: A.object({
 			latitude: A.number().optional(),
@@ -163,7 +163,7 @@ ${e}`;
 		execute: async (e) => {
 			let t, n;
 			if (e.city) {
-				let r = await xe(e.city);
+				let r = await Ce(e.city);
 				if (!r) return { error: `Could not find coordinates for "${e.city}". Please check the city name.` };
 				t = r.latitude, n = r.longitude;
 			} else if (e.latitude !== void 0 && e.longitude !== void 0) t = e.latitude, n = e.longitude;
@@ -171,14 +171,14 @@ ${e}`;
 			let r = await (await fetch(`https://api.open-meteo.com/v1/forecast?latitude=${t}&longitude=${n}&current=temperature_2m&hourly=temperature_2m&daily=sunrise,sunset&timezone=auto`)).json();
 			return "city" in e && (r.cityName = e.city), r;
 		}
-	}), s = S({
+	}), s = C({
 		description: "Create an artifact for writing or content creation. Use kind 'code' for programming, 'text' for writing, and 'sheet' for spreadsheets or tabular data.",
 		inputSchema: A.object({
 			title: A.string().describe("The title of the artifact"),
-			kind: A.enum(H).describe("Use 'code' for programming, 'text' for prose, and 'sheet' for CSV/tabular data")
+			kind: A.enum(U).describe("Use 'code' for programming, 'text' for prose, and 'sheet' for CSV/tabular data")
 		}),
 		execute: async ({ title: n, kind: r }) => {
-			let i = E();
+			let i = D();
 			t.write({
 				type: "data-kind",
 				data: r,
@@ -197,19 +197,19 @@ ${e}`;
 				transient: !0
 			});
 			let a = "";
-			return r === "text" ? a = await ye({
+			return r === "text" ? a = await be({
 				system: "Write about the given topic. Markdown is supported. Use headings wherever appropriate.",
 				prompt: n,
 				dataStream: t
-			}) : r === "code" ? a = await be({
-				system: he,
+			}) : r === "code" ? a = await xe({
+				system: ge,
 				prompt: n,
 				dataStream: t
-			}) : r === "sheet" && (a = await W({
-				system: ge,
+			}) : r === "sheet" && (a = await Se({
+				system: _e,
 				prompt: n,
 				dataStream: t
-			})), await d(e, {
+			})), await f(e, {
 				id: i,
 				title: n,
 				kind: r,
@@ -225,7 +225,7 @@ ${e}`;
 				content: r === "code" ? "A script was created and is now visible to the user." : "A document was created and is now visible to the user."
 			};
 		}
-	}), l = S({
+	}), c = C({
 		description: "Fully rewrite an existing artifact with the given description. Use editDocument instead for targeted find-and-replace changes.",
 		inputSchema: A.object({
 			id: A.string().describe("The ID of the artifact to rewrite"),
@@ -235,30 +235,30 @@ ${e}`;
 			let i = await a(n);
 			if (!i) return { error: "Document not found" };
 			let o = i.kind;
-			if (!H.includes(o)) return { error: `Unsupported document kind: ${i.kind}` };
+			if (!U.includes(o)) return { error: `Unsupported document kind: ${i.kind}` };
 			t.write({
 				type: "data-clear",
 				data: null,
 				transient: !0
 			});
 			let s = "";
-			return o === "text" ? s = await ye({
-				system: V(i.content, i.kind),
+			return o === "text" ? s = await be({
+				system: H(i.content, i.kind),
 				prompt: r,
 				dataStream: t,
 				providerOptions: { openai: { prediction: {
 					type: "content",
 					content: i.content
 				} } }
-			}) : o === "code" ? s = await be({
-				system: V(i.content, i.kind),
+			}) : o === "code" ? s = await xe({
+				system: H(i.content, i.kind),
 				prompt: r,
 				dataStream: t
-			}) : o === "sheet" && (s = await W({
-				system: V(i.content, i.kind),
+			}) : o === "sheet" && (s = await Se({
+				system: H(i.content, i.kind),
 				prompt: r,
 				dataStream: t
-			})), await d(e, {
+			})), await f(e, {
 				id: i.id,
 				title: i.title,
 				kind: i.kind,
@@ -278,7 +278,7 @@ ${e}`;
 	return {
 		getWeather: o,
 		createDocument: s,
-		editDocument: S({
+		editDocument: C({
 			description: "Make a targeted edit to an existing artifact by replacing an exact string. Prefer this over updateDocument for small changes, fixes, renames, and isolated edits.",
 			inputSchema: A.object({
 				id: A.string().describe("The ID of the artifact to edit"),
@@ -291,10 +291,10 @@ ${e}`;
 				if (!s) return { error: "Document not found" };
 				if (!s.content) return { error: "Document has no content" };
 				let c = s.kind;
-				if (!H.includes(c)) return { error: `Unsupported document kind: ${s.kind}` };
+				if (!U.includes(c)) return { error: `Unsupported document kind: ${s.kind}` };
 				if (!s.content.includes(r)) return { error: "old_string not found in document" };
 				let l = o ? s.content.replaceAll(r, i) : s.content.replace(r, i);
-				return await d(e, {
+				return await f(e, {
 					id: s.id,
 					title: s.title,
 					kind: s.kind,
@@ -327,19 +327,19 @@ ${e}`;
 				};
 			}
 		}),
-		updateDocument: l,
-		requestSuggestions: S({
+		updateDocument: c,
+		requestSuggestions: C({
 			description: "Request writing suggestions for an existing document artifact. Only use this when the user explicitly asks to improve or get suggestions for a document they have already created. Never use for general questions.",
 			inputSchema: A.object({ documentId: A.string().describe("The UUID of an existing document artifact that was previously created with createDocument") }),
 			execute: async ({ documentId: n }) => {
 				let i = await a(n);
 				if (!i?.content) return { error: "Document not found" };
-				if (U()) {
+				if (W()) {
 					let a = {
 						originalText: "Stub original sentence.",
 						suggestedText: "Stub suggested sentence.",
 						description: "Stub suggestion",
-						id: E(),
+						id: D(),
 						documentId: n,
 						documentCreatedAt: i.createdAt,
 						isResolved: !1,
@@ -350,7 +350,7 @@ ${e}`;
 						type: "data-suggestion",
 						data: a,
 						transient: !0
-					}), await c(e, {
+					}), await l(e, {
 						documentId: n,
 						suggestions: [a]
 					}), {
@@ -360,24 +360,24 @@ ${e}`;
 						message: "Suggestions have been added to the document"
 					};
 				}
-				let o = [], { partialOutputStream: s } = x({
-					model: R(),
+				let o = [], { partialOutputStream: s } = S({
+					model: z(),
 					system: "You are a help writing assistant. Given a piece of writing, please offer suggestions to improve the piece of writing and describe the change. It is very important for the edits to contain full sentences instead of just words. Max 5 suggestions.",
 					prompt: i.content,
-					output: v.array({ element: A.object({
+					output: b.array({ element: A.object({
 						originalSentence: A.string().describe("The original sentence"),
 						suggestedSentence: A.string().describe("The suggested sentence"),
 						description: A.string().describe("The description of the suggestion")
 					}) })
-				}), l = 0;
-				for await (let e of s) if (e) for (let a = l; a < e.length; a++) {
+				}), c = 0;
+				for await (let e of s) if (e) for (let a = c; a < e.length; a++) {
 					let s = e[a];
 					if (!s?.originalSentence || !s?.suggestedSentence || !s?.description) continue;
-					let c = {
+					let l = {
 						originalText: s.originalSentence,
 						suggestedText: s.suggestedSentence,
 						description: s.description,
-						id: E(),
+						id: D(),
 						documentId: n,
 						documentCreatedAt: i.createdAt,
 						isResolved: !1,
@@ -386,11 +386,11 @@ ${e}`;
 					};
 					t.write({
 						type: "data-suggestion",
-						data: c,
+						data: l,
 						transient: !0
-					}), o.push(c), l++;
+					}), o.push(l), c++;
 				}
-				return o.length > 0 && await c(e, {
+				return o.length > 0 && await l(e, {
 					documentId: n,
 					suggestions: o
 				}), {
@@ -411,39 +411,39 @@ ${e}`;
 			}
 		})
 	};
-}, G = (e) => e ? !e.trim() : !0, K = (e) => (Array.isArray(e) ? e : []).filter((e) => !!e && typeof e == "object" && !!e && "type" in e && "text" in e && e.type === "text" && typeof e.text == "string").map((e) => e.text).join("").trim(), Ce = (e) => {
+}, Te = (e) => e ? !e.trim() : !0, G = (e) => (Array.isArray(e) ? e : []).filter((e) => !!e && typeof e == "object" && !!e && "type" in e && "text" in e && e.type === "text" && typeof e.text == "string").map((e) => e.text).join("").trim(), Ee = (e) => {
 	let t = e.trim();
 	return t = t.replaceAll("\n", " "), t = t.replaceAll("\r", " "), t = t.replace(/\s+/g, " ").trim(), t = t.replace(/^(?:chat\s+)?title\s*[:\-–—]\s*/i, ""), t = t.replace(/^[#*\-•>"'`\s]+/, ""), t = t.replace(/[#:]/g, " "), t = t.replace(/[`"'*_~[\]()\\]/g, ""), t = t.replace(/\s+/g, " ").trim(), t.length > 30 && (t = t.slice(0, 30).trimEnd()), t;
-}, we = (e) => {
+}, De = (e) => {
 	let t = e.trim().toLowerCase().replace(/[^\p{L}\p{N}\s]/gu, "").replace(/\s+/g, " ").trim();
 	return t === "hi" || t === "hello" || t === "hey" || t === "salut" || t === "bonjour";
-}, q = (e) => {
+}, K = (e) => {
 	let t = e.trim();
-	return !t || we(t) ? "" : Ce(t.replaceAll("\n", " ").replaceAll("\r", " ").split(/\s+/).filter((e) => e.trim() !== "").slice(0, 5).join(" "));
-}, J = "/api/rb/agents/chat";
+	return !t || De(t) ? "" : Ee(t.replaceAll("\n", " ").replaceAll("\r", " ").split(/\s+/).filter((e) => e.trim() !== "").slice(0, 5).join(" "));
+}, q = "/api/rb/agents/chat";
 A.any(), A.any();
 //#endregion
 //#region src/api/chat/handler.ts
-var Te = /* @__PURE__ */ e({
-	default: () => nt,
+var Oe = /* @__PURE__ */ e({
+	default: () => ft,
 	runAgentLoop: () => $
-}), Ee = 1024 * 1024 * 20, De = 1e5, Oe = /^\/api\/rb\/files\/([^/]+)\/?$/, Y = "test-model", X = async (e) => new Promise((t) => setTimeout(t, e)), Z = async (e, t) => {
-	let n = g(t);
-	n && !n.isDone || await o(e, {
+}), ke = 1024 * 1024 * 20, Ae = 1e5, je = /^\/api\/rb\/files\/([^/]+)\/?$/, Me = "test-model", Ne = "openai/gpt-5.4-mini", J = async (e) => new Promise((t) => setTimeout(t, e)), Y = async (e, t) => {
+	let n = v(t);
+	n && !n.isDone || await _(e, {
 		chatId: t,
 		isStreaming: !1
 	});
-}, ke = (e) => e ? e === "1" || e.toLowerCase() === "true" || e.toLowerCase() === "yes" : !1, Q = (e) => typeof e == "string" && e.trim() || null, Ae = (e) => {
+}, Pe = (e) => e ? e === "1" || e.toLowerCase() === "true" || e.toLowerCase() === "yes" : !1, X = (e) => typeof e == "string" && e.trim() || null, Fe = (e) => {
 	if (typeof e == "boolean") return e;
-	let t = Q(e)?.toLowerCase();
+	let t = X(e)?.toLowerCase();
 	return t ? t === "true" ? !0 : t === "false" ? !1 : null : null;
-}, je = (e) => {
+}, Ie = (e) => {
 	let t = e.req?.session?.user?.currentTenantId;
-	return Q(t);
-}, Me = (e) => {
+	return X(t);
+}, Le = (e) => {
 	let t = e.req?.session?.user?.id;
-	return Q(t);
-}, Ne = () => Q(process.env.RB_FILESYSTEM_BUCKET_NAME) ?? "fs", Pe = (e) => T("sha256").update(Buffer.from(e)).digest("hex"), Fe = (e) => {
+	return X(t);
+}, Re = () => X(process.env.RB_FILESYSTEM_BUCKET_NAME) ?? "fs", ze = (e) => E("sha256").update(Buffer.from(e)).digest("hex"), Be = (e) => {
 	let t = e.trim();
 	if (!t) return null;
 	let n;
@@ -456,36 +456,36 @@ var Te = /* @__PURE__ */ e({
 			return null;
 		}
 	}
-	let r = Oe.exec(n.pathname);
+	let r = je.exec(n.pathname);
 	if (!r) return null;
-	let i = Q(decodeURIComponent(r[1]));
+	let i = X(decodeURIComponent(r[1]));
 	return i ? {
 		fileId: i,
-		uploadKey: Q(n.searchParams.get("rb-upload-key"))
+		uploadKey: X(n.searchParams.get("rb-upload-key"))
 	} : null;
-}, Ie = async ({ ctx: e, fileId: t, uploadKey: n }) => {
-	let r = je(e);
+}, Ve = async ({ ctx: e, fileId: t, uploadKey: n }) => {
+	let r = Ie(e);
 	if (!r) throw Error("tenant_missing");
 	let i;
 	try {
-		i = new ae(t);
+		i = new O(t);
 	} catch {
 		throw Error("invalid_file_id");
 	}
-	let a = (await w(r)).db;
+	let a = (await T(r)).db;
 	if (!a) throw Error("filesystem_db_unavailable");
-	let o = new D(a, { bucketName: Ne() }), [s] = await o.find({ _id: i }).limit(1).toArray();
+	let o = new oe(a, { bucketName: Re() }), [s] = await o.find({ _id: i }).limit(1).toArray();
 	if (!s) throw Error("file_not_found");
-	let c = Ae(s?.metadata?.isPublic) ?? !1, l = Me(e), u = Q(s?.metadata?.userId), d = Q(s?.metadata?.ownerKeyHash);
+	let c = Fe(s?.metadata?.isPublic) ?? !1, l = Le(e), u = X(s?.metadata?.userId), d = X(s?.metadata?.ownerKeyHash);
 	if (!c) {
-		let e = !!(l && u && l === u), t = !!(n && d && Pe(n) === d);
+		let e = !!(l && u && l === u), t = !!(n && d && ze(n) === d);
 		if (!e && !t) throw Error("unauthorized");
 	}
-	let f = Q(s?.metadata?.mimeType), p = [], m = 0;
+	let f = X(s?.metadata?.mimeType), p = [], m = 0;
 	return await new Promise((e, t) => {
 		let n = o.openDownloadStream(i);
 		n.on("data", (e) => {
-			if (m += e.length, m > Ee) {
+			if (m += e.length, m > ke) {
 				n.destroy(/* @__PURE__ */ Error("attachment_too_large"));
 				return;
 			}
@@ -495,16 +495,16 @@ var Te = /* @__PURE__ */ e({
 		data: Buffer.concat(p, m),
 		mimeType: f
 	};
-}, Le = async (e, t) => {
+}, He = async (e, t) => {
 	let n = Array.isArray(t) ? t : [], r = [];
 	for (let t of n) {
 		if (!t || typeof t != "object" || !("role" in t) || t.role !== "user") continue;
 		let n = Array.isArray(t.content) ? t.content : null;
 		if (n) for (let t of n) {
 			if (!t || typeof t != "object" || t.type !== "file" || typeof t.data != "string") continue;
-			let n = Fe(t.data);
+			let n = Be(t.data);
 			n && r.push((async () => {
-				let { data: r, mimeType: i } = await Ie({
+				let { data: r, mimeType: i } = await Ve({
 					ctx: e,
 					...n
 				});
@@ -513,7 +513,7 @@ var Te = /* @__PURE__ */ e({
 		}
 	}
 	r.length > 0 && await Promise.all(r);
-}, Re = (e) => {
+}, Ue = (e) => {
 	let t = Array.isArray(e) ? e : [];
 	for (let e of t) {
 		if (!e || typeof e != "object" || !("role" in e) || e.role !== "user") continue;
@@ -521,12 +521,12 @@ var Te = /* @__PURE__ */ e({
 		if (t) for (let e = 0; e < t.length; e += 1) {
 			let n = t[e];
 			if (!n || typeof n != "object" || n.type !== "file") continue;
-			let r = Q(n.mediaType);
+			let r = X(n.mediaType);
 			if (r?.startsWith("image/") || r === "application/pdf") continue;
-			let i = Q(n.filename) ?? Q(n.name) ?? `attachment-${e + 1}`, a = n.data;
+			let i = X(n.filename) ?? X(n.name) ?? `attachment-${e + 1}`, a = n.data;
 			if (r?.startsWith("text/") && a instanceof Uint8Array) {
 				let n = (Buffer.isBuffer(a) ? a : Buffer.from(a)).toString("utf8");
-				n.length > De && (n = `${n.slice(0, De)}\n\n[Truncated]`);
+				n.length > Ae && (n = `${n.slice(0, Ae)}\n\n[Truncated]`);
 				let o = r === "text/csv" || i.toLowerCase().endsWith(".csv") ? "csv" : "";
 				t[e] = {
 					type: "text",
@@ -541,66 +541,66 @@ var Te = /* @__PURE__ */ e({
 			};
 		}
 	}
-}, ze = async (e, t, n) => {
-	if (e.message?.role === "user") return K(e.message.parts);
+}, We = async (e, t, n) => {
+	if (e.message?.role === "user") return G(e.message.parts);
 	if (Array.isArray(e.messages)) {
 		let t = null;
 		for (let n = e.messages.length - 1; n >= 0; --n) {
 			let r = e.messages[n];
 			if (r?.role !== "user") continue;
-			let i = K(r.parts);
+			let i = G(r.parts);
 			if (i) {
-				if (q(i)) return i;
+				if (K(i)) return i;
 				t ??= i;
 			}
 		}
 		if (t) return t;
 	}
-	let r = await u(t, n), i = null;
+	let r = await d(t, n), i = null;
 	for (let e = r.length - 1; e >= 0; --e) {
 		let t = r[e];
 		if (t.role !== "user") continue;
-		let n = K(t.parts);
+		let n = G(t.parts);
 		if (n) {
-			if (q(n)) return n;
+			if (K(n)) return n;
 			i ??= n;
 		}
 	}
 	return i ?? "";
-}, Be = (e) => {
+}, Ge = (e) => {
 	let t = Array.isArray(e.messages) && e.messages.length > 0 ? e.messages.at(-1) : e.message;
 	return !t || t.role !== "user" ? "" : (Array.isArray(t.parts) ? t.parts : []).filter((e) => !!e && typeof e == "object" && !!e && "type" in e && "text" in e && e.type === "text" && typeof e.text == "string").map((e) => e.text).join("\n").trim();
-}, Ve = (e) => {
+}, Ke = (e) => {
 	let t = e.id;
 	return typeof t == "string" && t.trim() !== "" ? t : null;
-}, He = (e) => {
+}, qe = (e) => {
 	let t = e.selectedChatModel;
 	return typeof t == "string" && t.trim() !== "" ? t : null;
-}, Ue = (e) => {
+}, Je = (e) => {
 	let t = e.openaiReasoningEffort;
 	return typeof t == "string" && t.trim() !== "" ? t.trim() : null;
-}, We = (e) => {
+}, Ye = (e) => {
 	if (!e || typeof e != "object") return !1;
 	let t = e.state, n = e.toolCallId;
 	return typeof n == "string" && n.trim() !== "" && (t === "approval-responded" || t === "output-denied");
-}, Ge = (e) => {
+}, Xe = (e) => {
 	let t = Array.isArray(e.messages) ? e.messages : [];
-	return t.length === 0 || t.at(-1)?.role === "user" ? !1 : t.some((e) => (Array.isArray(e?.parts) ? e.parts : []).some((e) => We(e)));
-}, Ke = (e) => {
+	return t.length === 0 || t.at(-1)?.role === "user" ? !1 : t.some((e) => (Array.isArray(e?.parts) ? e.parts : []).some((e) => Ye(e)));
+}, Ze = (e) => {
 	let t = /* @__PURE__ */ new Map();
 	for (let n of e) {
 		let e = Array.isArray(n?.parts) ? n.parts : [];
-		for (let n of e) We(n) && t.set(n.toolCallId, n);
+		for (let n of e) Ye(n) && t.set(n.toolCallId, n);
 	}
 	return t;
-}, qe = (e, t) => {
+}, Qe = (e, t) => {
 	let n = [];
 	return {
 		changedMessages: n,
 		updatedMessages: e.map((e) => {
 			let r = !1, i = e.parts.map((e) => {
 				if (!e || typeof e != "object") return e;
-				let n = Q(e.toolCallId);
+				let n = X(e.toolCallId);
 				if (!n) return e;
 				let i = t.get(n);
 				return i ? (r = !0, {
@@ -616,23 +616,78 @@ var Te = /* @__PURE__ */ e({
 			return n.push(a), a;
 		})
 	};
-}, Je = (e) => {
+}, $e = (e) => {
 	if (!e || typeof e != "object" || !("createdAt" in e)) return null;
 	let t = e.createdAt;
 	if (typeof t != "string") return null;
 	let n = new Date(t);
 	return Number.isNaN(n.getTime()) ? null : n;
-}, Ye = (e) => ({
+}, et = (e) => ({
 	latitude: null,
 	longitude: null,
 	city: null,
 	country: null
-}), Xe = (e) => `agents-sdk:chat:${e}:stream`, Ze = async ({ ctx: e, chatId: t, selectedChatModel: i }) => await r().resolveAgent?.({
+}), tt = (e) => {
+	let [t] = e.split("/");
+	return t && t !== e ? t : "unknown";
+}, Z = (e) => typeof e == "number" && Number.isFinite(e) ? e : void 0, nt = (e) => e instanceof Error && e.message.trim() ? e.message : "Unknown error", rt = (e) => {
+	let t = e && typeof e == "object" ? e : {}, n = t.inputTokenDetails && typeof t.inputTokenDetails == "object" ? t.inputTokenDetails : {}, r = t.outputTokenDetails && typeof t.outputTokenDetails == "object" ? t.outputTokenDetails : {};
+	return {
+		inputTokens: Z(t.inputTokens),
+		outputTokens: Z(t.outputTokens),
+		totalTokens: Z(t.totalTokens),
+		cachedInputTokens: Z(n.cacheReadTokens) ?? Z(t.cachedInputTokens),
+		reasoningTokens: Z(r.reasoningTokens) ?? Z(t.reasoningTokens),
+		rawUsage: Object.keys(t).length > 0 ? t.raw ?? t : void 0
+	};
+}, it = () => ({
+	inputTokens: 0,
+	outputTokens: 0,
+	totalTokens: 0,
+	cachedInputTokens: 0,
+	reasoningTokens: 0
+}), at = async ({ ctx: e, chatId: t, agent: n, selectedChatModel: i, mode: a }) => {
+	try {
+		return { workspaceId: X((await r().resolveUsageContext?.({
+			ctx: e,
+			chatId: t,
+			agent: n,
+			selectedChatModel: i,
+			mode: a
+		}))?.workspaceId) ?? void 0 };
+	} catch (e) {
+		return console.warn("Failed to resolve agent usage context.", e), { workspaceId: void 0 };
+	}
+}, Q = async ({ ctx: e, usageContext: t, agent: n, chatId: r, mode: i, feature: a, modelId: s, status: c, billable: l, usage: u, finishReason: d, error: f, startedAt: p, finishedAt: m = /* @__PURE__ */ new Date() }) => {
+	let h = u ? rt(u) : {}, g = {
+		id: D(),
+		workspaceId: t.workspaceId,
+		agentId: n.id,
+		chatId: r,
+		mode: i,
+		feature: a,
+		provider: tt(s),
+		model: s,
+		billable: l,
+		status: c,
+		finishReason: X(d) ?? void 0,
+		errorMessage: f ? nt(f) : void 0,
+		...h,
+		startedAt: p,
+		finishedAt: m,
+		createdAt: /* @__PURE__ */ new Date()
+	};
+	try {
+		await o(e, g);
+	} catch (e) {
+		console.error("Failed to save agents usage event.", e);
+	}
+}, ot = (e) => `agents-sdk:chat:${e}:stream`, st = async ({ ctx: e, chatId: t, selectedChatModel: i }) => await r().resolveAgent?.({
 	ctx: e,
 	chatId: t,
 	selectedChatModel: i
-}) ?? n(), Qe = ({ ctx: e, chatId: t, mode: n, streamTransport: r }) => {
-	let a = i(e), o = Xe(t), s = _({
+}) ?? n(), ct = ({ ctx: e, chatId: t, mode: n, streamTransport: r }) => {
+	let a = i(e), o = ot(t), s = y({
 		chatId: t,
 		abortWhenNoSubscribersMs: n === "headless" || r ? 0 : void 0
 	});
@@ -660,84 +715,95 @@ var Te = /* @__PURE__ */ e({
 			});
 		}
 	};
-}, $e = ({ ctx: e, chatId: t, streamTransport: n, message: r }) => {
+}, lt = ({ ctx: e, chatId: t, streamTransport: n, message: r }) => {
 	if (!n) return;
 	let a = i(e);
 	n.publish({
 		tenantId: a.tenantId,
 		userId: a.userId,
-		channel: Xe(t),
+		channel: ot(t),
 		payload: {
 			chatId: t,
 			message: r
 		}
 	});
-}, $ = async ({ payload: e, ctx: n, mode: c = "interactive", streamTransport: d = null }) => {
-	let f = Ve(e);
+}, $ = async ({ payload: e, ctx: n, mode: o = "interactive", streamTransport: l = null }) => {
+	let f = Ke(e);
 	if (!f) return a(n, "bad_request:api", "Parameter id is required.");
-	let m = e, g = He(m), _ = ke(process.env.PLAYWRIGHT), v = !ke(process.env.AGENT_SDK_LIVE_CHAT) && _ || process.env.NODE_ENV === "test";
-	if (!g) return a(n, "bad_request:api", "Parameter selectedChatModel is required.");
-	if (g === Y && (v = !0), g !== Y && !se.has(g)) return a(n, "bad_request:api", `Unsupported selectedChatModel: "${g}".`);
-	let y = process.env.OPENAI_API_KEY, re = process.env.GOOGLE_GENERATIVE_AI_API_KEY ?? process.env.GOOGLE_API_KEY, b = process.env.XAI_API_KEY, S = ce(g), C = le(g), w = ue(g), T = Ue(m), D = T && T !== "default" && T !== "off" ? T : null, ae = t(g);
-	if (D && !ae.some((e) => e === D)) return a(n, "bad_request:api", `Unsupported reasoning effort "${D}" for selectedChatModel: "${g}".`);
-	if (!v && S && !y) return a(n, "bad_request:provider", "Missing OPENAI_API_KEY.");
-	if (!v && C && !re) return a(n, "bad_request:provider", "Missing GOOGLE_GENERATIVE_AI_API_KEY (or GOOGLE_API_KEY).");
-	if (!v && w && !b) return a(n, "bad_request:provider", "Missing XAI_API_KEY.");
-	if (!v && !S && !C && !w) return a(n, "bad_request:provider", `Unsupported selectedChatModel: "${g}".`);
-	let O = await Ze({
+	let p = e, h = qe(p), v = Pe(process.env.PLAYWRIGHT), y = !Pe(process.env.AGENT_SDK_LIVE_CHAT) && v || process.env.NODE_ENV === "test";
+	if (!h) return a(n, "bad_request:api", "Parameter selectedChatModel is required.");
+	if (h === Me && (y = !0), h !== Me && !le.has(h)) return a(n, "bad_request:api", `Unsupported selectedChatModel: "${h}".`);
+	let b = process.env.OPENAI_API_KEY, ne = process.env.GOOGLE_GENERATIVE_AI_API_KEY ?? process.env.GOOGLE_API_KEY, ie = process.env.XAI_API_KEY, x = F(h), C = ue(h), w = de(h), T = Je(p), E = T && T !== "default" && T !== "off" ? T : null, oe = t(h);
+	if (E && !oe.some((e) => e === E)) return a(n, "bad_request:api", `Unsupported reasoning effort "${E}" for selectedChatModel: "${h}".`);
+	if (!y && x && !b) return a(n, "bad_request:provider", "Missing OPENAI_API_KEY.");
+	if (!y && C && !ne) return a(n, "bad_request:provider", "Missing GOOGLE_GENERATIVE_AI_API_KEY (or GOOGLE_API_KEY).");
+	if (!y && w && !ie) return a(n, "bad_request:provider", "Missing XAI_API_KEY.");
+	if (!y && !x && !C && !w) return a(n, "bad_request:provider", `Unsupported selectedChatModel: "${h}".`);
+	let O = await st({
 		ctx: n,
 		chatId: f,
-		selectedChatModel: g
-	}), k = await r().resolveKnowledge?.({
+		selectedChatModel: h
+	}), se = await r().resolveKnowledge?.({
+		ctx: n,
+		chatId: f,
+		agent: O,
+		selectedChatModel: h
+	}) ?? null, k = await at({
 		ctx: n,
 		chatId: f,
 		agent: O,
-		selectedChatModel: g
-	}) ?? null, oe = i(n);
-	await s(n, f) || await p(n, {
+		selectedChatModel: h,
+		mode: o
+	}), ce = i(n), A = await s(n, f, { includeDeleted: !0 });
+	if (A?.deletedAt) return a(n, "not_found:chat");
+	A ? A.selectedChatModel !== h && await g(n, {
+		chatId: f,
+		selectedChatModel: h
+	}) : await m(n, {
 		id: f,
 		createdAt: /* @__PURE__ */ new Date(),
-		userId: oe.userId,
+		userId: ce.userId,
 		title: "",
 		isStreaming: !1,
+		selectedChatModel: h,
 		lastContext: null
 	});
-	let A = m.message, j = typeof A?.id == "string" ? A.id : E(), M = Array.isArray(A?.parts) ? A?.parts : [];
-	if ((typeof A?.role == "string" ? A.role : null) === "user") {
+	let j = p.message, M = typeof j?.id == "string" ? j.id : D(), N = Array.isArray(j?.parts) ? j?.parts : [];
+	if ((typeof j?.role == "string" ? j.role : null) === "user") {
 		let e = /* @__PURE__ */ new Date();
-		await l(n, [{
-			id: j,
+		await u(n, [{
+			id: M,
 			chatId: f,
 			role: "user",
-			parts: M,
+			parts: N,
 			attachments: [],
 			createdAt: e
-		}]), $e({
+		}]), lt({
 			ctx: n,
 			chatId: f,
-			streamTransport: d,
+			streamTransport: l,
 			message: {
-				id: j,
+				id: M,
 				role: "user",
-				parts: M,
+				parts: N,
 				metadata: { createdAt: e.toISOString() }
 			}
 		});
 	}
-	if (v) {
-		await o(n, {
+	if (y) {
+		await _(n, {
 			chatId: f,
 			isStreaming: !0
 		});
-		let e = Qe({
+		let e = ct({
 			ctx: n,
 			chatId: f,
-			mode: c,
-			streamTransport: d
+			mode: o,
+			streamTransport: l
 		});
 		try {
-			let t = q(await ze(m, n, f)), r = await s(n, f);
-			r && G(r.title) && t && t !== r.title && (await h(n, {
+			let t = K(await We(p, n, f)), r = await s(n, f);
+			r && Te(r.title) && t && t !== r.title && (await c(n, {
 				chatId: f,
 				title: t
 			}), e.publishJson({
@@ -745,8 +811,8 @@ var Te = /* @__PURE__ */ e({
 				data: t,
 				transient: !0
 			}));
-			let i = E();
-			_ && !e.abortController.signal.aborted && await X(600), e.publishJson({
+			let i = D();
+			v && !e.abortController.signal.aborted && await J(600), e.publishJson({
 				type: "start",
 				messageId: i
 			});
@@ -755,18 +821,18 @@ var Te = /* @__PURE__ */ e({
 				type: "text-start",
 				id: a
 			});
-			let o = Be(m), c = o ? `Echo: ${o}` : "Hello from agents-sdk", u = Date.now(), d = c.split(/(\s+)/).filter((e) => e.length > 0);
-			for (let t of d) {
+			let l = Ge(p), d = l ? `Echo: ${l}` : "Hello from agents-sdk", m = Date.now(), g = d.split(/(\s+)/).filter((e) => e.length > 0);
+			for (let t of g) {
 				if (e.abortController.signal.aborted) break;
 				e.publishJson({
 					type: "text-delta",
 					id: a,
 					delta: t
-				}), await X(_ ? 120 : 80);
+				}), await J(v ? 120 : 80);
 			}
-			if (_ && !e.abortController.signal.aborted) {
-				let e = Date.now() - u, t = 1200;
-				e < t && await X(t - e);
+			if (v && !e.abortController.signal.aborted) {
+				let e = Date.now() - m, t = 1200;
+				e < t && await J(t - e);
 			}
 			return e.abortController.signal.aborted ? {} : (e.publishJson({
 				type: "text-end",
@@ -774,24 +840,37 @@ var Te = /* @__PURE__ */ e({
 			}), e.publishJson({
 				type: "finish",
 				finishReason: "stop"
-			}), e.publishDone(), await l(n, [{
+			}), e.publishDone(), await u(n, [{
 				id: i,
 				chatId: f,
 				role: "assistant",
 				parts: [{
 					type: "text",
-					text: c
+					text: d
 				}],
 				attachments: [],
 				createdAt: /* @__PURE__ */ new Date()
-			}]), {});
+			}]), await Q({
+				ctx: n,
+				usageContext: k,
+				agent: O,
+				chatId: f,
+				mode: o,
+				feature: "chat",
+				modelId: h,
+				status: "success",
+				billable: !1,
+				usage: it(),
+				finishReason: "stop",
+				startedAt: new Date(m)
+			}), {});
 		} finally {
-			await Z(n, f);
+			await Y(n, f);
 		}
 	}
-	let N = Ye(n), P = Ge(m);
-	if (Array.isArray(m.messages) && !P) {
-		let e = m.messages.flatMap((e) => {
+	let P = et(n), L = Xe(p);
+	if (Array.isArray(p.messages) && !L) {
+		let e = p.messages.flatMap((e) => {
 			let t = typeof e?.id == "string" ? e.id : null, n = typeof e?.role == "string" ? e.role : null, r = Array.isArray(e?.parts) ? e.parts : [];
 			return !t || !n ? [] : [{
 				id: t,
@@ -799,92 +878,121 @@ var Te = /* @__PURE__ */ e({
 				role: n,
 				parts: r,
 				attachments: [],
-				createdAt: Je(e?.metadata) ?? /* @__PURE__ */ new Date()
+				createdAt: $e(e?.metadata) ?? /* @__PURE__ */ new Date()
 			}];
 		});
-		e.length > 0 && await l(n, e);
+		e.length > 0 && await u(n, e);
 	}
-	let I;
-	if (Array.isArray(m.messages) && m.messages.length > 0 && P) {
-		let e = await u(n, f);
-		if (e.length === 0) I = m.messages;
+	let R;
+	if (Array.isArray(p.messages) && p.messages.length > 0 && L) {
+		let e = await d(n, f);
+		if (e.length === 0) R = p.messages;
 		else {
-			let { changedMessages: t, updatedMessages: r } = qe(e, Ke(m.messages));
-			t.length > 0 && await l(n, t), I = r.map((e) => ({
+			let { changedMessages: t, updatedMessages: r } = Qe(e, Ze(p.messages));
+			t.length > 0 && await u(n, t), R = r.map((e) => ({
 				id: e.id,
 				role: e.role,
 				parts: e.parts,
 				metadata: { createdAt: e.createdAt.toISOString() }
 			}));
 		}
-	} else I = Array.isArray(m.messages) && m.messages.length > 0 ? m.messages : (await u(n, f)).map((e) => ({
+	} else R = Array.isArray(p.messages) && p.messages.length > 0 ? p.messages : (await d(n, f)).map((e) => ({
 		id: e.id,
 		role: e.role,
 		parts: e.parts,
 		metadata: { createdAt: e.createdAt.toISOString() }
 	}));
-	await o(n, {
+	await _(n, {
 		chatId: f,
 		isStreaming: !0
 	});
-	let L = Qe({
+	let z = ct({
 		ctx: n,
 		chatId: f,
-		mode: c,
-		streamTransport: d
-	}), R = "", z = "";
+		mode: o,
+		streamTransport: l
+	}), B = "", V = "";
 	try {
-		R = await ze(m, n, f), z = q(R);
+		B = await We(p, n, f), V = K(B);
 		let e = await s(n, f);
-		e && G(e.title) && z && z !== e.title && (await h(n, {
+		e && Te(e.title) && V && V !== e.title && (await c(n, {
 			chatId: f,
-			title: z
-		}), L.publishJson({
+			title: V
+		}), z.publishJson({
 			type: "data-chat-title",
-			data: z,
+			data: V,
 			transient: !0
 		}));
 	} catch (e) {
-		console.error(e), L.publishJson({
+		console.error(e), z.publishJson({
 			type: "error",
 			errorText: "Something went wrong."
-		}), L.publishDone();
+		}), z.publishDone();
 		try {
-			await Z(n, f);
+			await Y(n, f);
 		} catch (e) {
 			console.error("Failed to update chat streaming state.", e);
 		}
 		return {};
 	}
-	R && z && y && (async () => {
+	B && V && b && (async () => {
+		let e = /* @__PURE__ */ new Date();
 		try {
-			let { text: e } = await ne({
-				model: fe(),
-				system: _e,
-				prompt: R
-			}), t = Ce(e);
-			if (!t) return;
-			let r = await s(n, f);
+			let { text: t } = await re({
+				model: pe(),
+				system: ve,
+				prompt: B,
+				onFinish: async (t) => {
+					await Q({
+						ctx: n,
+						usageContext: k,
+						agent: O,
+						chatId: f,
+						mode: o,
+						feature: "title",
+						modelId: Ne,
+						status: "success",
+						billable: !0,
+						usage: t.totalUsage ?? t.usage,
+						finishReason: t.finishReason,
+						startedAt: e
+					});
+				}
+			}), r = Ee(t);
 			if (!r) return;
-			r.title !== t && (await h(n, {
+			let i = await s(n, f);
+			if (!i) return;
+			i.title !== r && (await c(n, {
 				chatId: f,
-				title: t
-			}), L.publishJson({
+				title: r
+			}), z.publishJson({
 				type: "data-chat-title",
-				data: t,
+				data: r,
 				transient: !0
 			}));
-		} catch (e) {
-			console.warn("Failed to generate chat title.", e);
+		} catch (t) {
+			await Q({
+				ctx: n,
+				usageContext: k,
+				agent: O,
+				chatId: f,
+				mode: o,
+				feature: "title",
+				modelId: Ne,
+				status: "error",
+				billable: !1,
+				error: t,
+				startedAt: e
+			}), console.warn("Failed to generate chat title.", t);
 		}
 	})();
-	let B = te({
-		originalMessages: I,
+	let me = te({
+		originalMessages: R,
 		onError: (e) => (console.error(e), "Something went wrong."),
 		onFinish: async ({ responseMessage: e }) => {
 			let t = Array.isArray(e.parts) ? e.parts : [];
-			await l(n, [{
-				id: e.id ?? E(),
+			await u(n, [{
+				id: e.id ?? D(),
 				chatId: f,
 				role: e.role ?? "assistant",
 				parts: t,
@@ -893,10 +1001,10 @@ var Te = /* @__PURE__ */ e({
 			}]);
 		},
 		execute: async ({ writer: e }) => {
-			let t = Se({
+			let t = we({
 				ctx: n,
 				dataStream: e,
-				requestHints: N
+				requestHints: P
 			}), i = [
 				"getWeather",
 				"createDocument",
@@ -910,87 +1018,123 @@ var Te = /* @__PURE__ */ e({
 				"editDocument",
 				"updateDocument",
 				"requestSuggestions"
-			], o = F(g), s = o ? [] : S ? i : a, c = t, l = s, u = await r().resolveTools?.({
+			], s = I(h), c = s ? [] : x ? i : a, l = t, u = c, d = await r().resolveTools?.({
 				ctx: n,
 				dataStream: e,
-				requestHints: N,
+				requestHints: P,
 				agent: O,
-				selectedChatModel: g,
+				selectedChatModel: h,
 				availableTools: t,
-				defaultActiveTools: s
+				defaultActiveTools: c
 			});
-			u && (c = u.tools, l = u.activeTools ?? Object.keys(u.tools));
-			let d = await ee(I.map((e) => ({
+			d && (l = d.tools, u = d.activeTools ?? Object.keys(d.tools));
+			let m = await ee(R.map((e) => ({
 				role: e?.role,
 				parts: e?.parts,
 				metadata: e?.metadata
 			})), {
-				tools: c,
+				tools: l,
 				ignoreIncompleteToolCalls: !0
 			});
-			await Le(n, d), S && Re(d);
-			let f = typeof m.openaiReasoningSummary == "string" ? m.openaiReasoningSummary : null, p;
-			S && D ? p = { openai: {
-				reasoningEffort: D,
-				reasoningSummary: f ?? void 0
-			} } : w && o && D && (p = { xai: { reasoningEffort: D } });
-			let h = x({
-				model: de(g),
-				system: me({
-					selectedChatModel: g,
-					requestHints: N,
+			await He(n, m), x && Ue(m);
+			let g = typeof p.openaiReasoningSummary == "string" ? p.openaiReasoningSummary : null, _;
+			x && E ? _ = { openai: {
+				reasoningEffort: E,
+				reasoningSummary: g ?? void 0
+			} } : w && s && E && (_ = { xai: { reasoningEffort: E } });
+			let v = !1, y = /* @__PURE__ */ new Date(), b = async ({ status: e, usage: t, finishReason: r, error: i }) => {
+				v || (v = !0, await Q({
+					ctx: n,
+					usageContext: k,
+					agent: O,
+					chatId: f,
+					mode: o,
+					feature: "chat",
+					modelId: h,
+					status: e,
+					billable: e === "success",
+					usage: t,
+					finishReason: r,
+					error: i,
+					startedAt: y
+				}));
+			}, te = S({
+				model: fe(h),
+				system: he({
+					selectedChatModel: h,
+					requestHints: P,
 					agentInstructions: O.instructions,
-					knowledge: k
+					knowledge: se
 				}),
-				messages: d,
-				tools: c,
-				stopWhen: ie(5),
-				abortSignal: L.abortController.signal,
-				providerOptions: p,
-				activeTools: l
+				messages: m,
+				tools: l,
+				stopWhen: ae(5),
+				abortSignal: z.abortController.signal,
+				providerOptions: _,
+				activeTools: u,
+				onFinish: async (e) => {
+					await b({
+						status: "success",
+						usage: e.totalUsage ?? e.usage,
+						finishReason: e.finishReason
+					});
+				},
+				onError: async (e) => {
+					await b({
+						status: "error",
+						error: e?.error ?? e
+					});
+				},
+				onAbort: async (e) => {
+					await b({
+						status: "aborted",
+						usage: e?.totalUsage ?? e?.usage,
+						finishReason: e?.finishReason
+					});
+				}
 			});
-			e.merge(h.toUIMessageStream({
-				originalMessages: I,
-				sendReasoning: o || !!(S && D)
+			e.merge(te.toUIMessageStream({
+				originalMessages: R,
+				sendReasoning: s || !!(x && E)
 			}));
 		}
 	});
 	return (async () => {
 		let e = null;
 		try {
-			let t = B.getReader();
+			let t = me.getReader();
 			for (e = () => t.releaseLock();;) {
 				let { value: e, done: n } = await t.read();
 				if (n) break;
-				L.publishJson(e);
+				z.publishJson(e);
 			}
-			L.publishDone();
+			z.publishDone();
 		} catch (e) {
-			console.error(e), L.publishJson({
+			console.error(e), z.publishJson({
 				type: "error",
 				errorText: "Something went wrong."
-			}), L.publishDone();
+			}), z.publishDone();
 		} finally {
 			e?.();
 			try {
-				await Z(n, f);
+				await Y(n, f);
 			} catch (e) {
 				console.error("Failed to update chat streaming state.", e);
 			}
 		}
 	})(), {};
-}, et = async (e, t) => $({
+}, ut = async (e, t) => $({
 	payload: e,
 	ctx: t,
 	mode: "interactive"
-}), tt = async (e, t) => {
+}), dt = async (e, t) => {
 	let n = t.req.query.id, r = Array.isArray(n) ? n[0] : n;
-	return typeof r != "string" || r.trim() === "" ? a(t, "bad_request:api") : f(t, r);
-}, nt = (e) => {
-	e.get(J, (async (e, t) => {
+	return typeof r != "string" || r.trim() === "" ? a(t, "bad_request:api") : p(t, r);
+}, ft = (e) => {
+	e.get(q, (async (e, t) => {
 		let n = t.req.query.id, r = Array.isArray(n) ? n[0] : n;
 		return typeof r != "string" || r.trim() === "" ? a(t, "bad_request:api", "Parameter id is required.") : await s(t, r) || a(t, "not_found:chat");
-	})), e.post(J, et), e.delete(J, tt);
+	})), e.post(q, ut), e.delete(q, dt);
 };
 //#endregion
-export { $ as n, Te as t };
+export { $ as n, Oe as t };