@rpcbase/agents-sdk 0.2.0 → 0.3.0

package/dist/{handler-ImeP5Thm.js → handler-ClWOGjB_.js}

@@ -1,64 +1,64 @@
-import { r as e } from "./openai-reasoning-ARz6ZvZ_.js";
+import { r as e } from "./openai-reasoning-kFq60GX9.js";
 import { t } from "./apiErrors-CFUx-UrE.js";
 import { i as n, r, t as i } from "./chatCtx-DSgD1fix.js";
-import { _ as a, a as o, g as s, h as c, l, m as u, n as d, p as f, s as p, v as m } from "./chatDb-CpCFaAxK.js";
-import { n as h, t as g } from "./resumableSse-BRFonw_y.js";
-import { Output as _, convertToModelMessages as ee, createUIMessageStream as te, extractReasoningMiddleware as v, generateText as ne, smoothStream as y, stepCountIs as re, 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 O } from "mongodb";
+import { _ as a, a as o, b as s, g as c, h as l, l as u, m as d, n as f, p, s as m, v as h, y as g } from "./chatDb-CcAjAGX7.js";
+import { n as _, t as v } from "./resumableSse-BRFonw_y.js";
+import { Output as y, convertToModelMessages as ee, createUIMessageStream as b, extractReasoningMiddleware as x, generateText as te, smoothStream as ne, stepCountIs as re, streamObject as S, streamText as C, tool as w, wrapLanguageModel as ie } from "ai";
+import { getTenantFilesystemDb as T } from "@rpcbase/db";
+import { createHash as E, randomUUID as D } from "node:crypto";
+import { GridFSBucket as O, ObjectId as ae } from "mongodb";
 import { createGoogleGenerativeAI as k } from "@ai-sdk/google";
 import { openai as A } from "@ai-sdk/openai";
-import { createXai as ie } from "@ai-sdk/xai";
-import { z as j } from "zod";
+import { createXai as j } from "@ai-sdk/xai";
+import { z as M } from "zod";
 //#region src/api/chat/models.ts
-var M = /-thinking$/, N = "openai/", P = "google/", F = "xai/", ae = new Set([
+var N = /-thinking$/, P = "openai/", F = "google/", I = "xai/", oe = new Set([
 	"openai/gpt-5.4-mini",
 	"openai/gpt-5.5",
 	"google/gemini-3.5-flash",
 	"xai/grok-4.3-thinking"
-]), I = (e) => e.startsWith(N), L = (e) => e.startsWith(P), R = (e) => e.startsWith(F), z = (e) => e.endsWith("-thinking"), B = ie({
+]), se = (e) => e.startsWith(P), ce = (e) => e.startsWith(F), le = (e) => e.startsWith(I), L = (e) => e.endsWith("-thinking"), R = j({
 	baseURL: process.env.XAI_BASE_URL ?? "https://api.x.ai/v1",
 	apiKey: process.env.XAI_API_KEY
-}), V = k({ apiKey: process.env.GOOGLE_GENERATIVE_AI_API_KEY ?? process.env.GOOGLE_API_KEY }), oe = (e) => {
-	if (I(e)) return A.responses(e.slice(N.length));
-	if (L(e)) {
-		let t = e.slice(P.length).replace(M, ""), n = V.chat(t);
-		return z(e) ? C({
+}), z = k({ apiKey: process.env.GOOGLE_GENERATIVE_AI_API_KEY ?? process.env.GOOGLE_API_KEY }), ue = (e) => {
+	if (se(e)) return A.responses(e.slice(P.length));
+	if (ce(e)) {
+		let t = e.slice(F.length).replace(N, ""), n = z.chat(t);
+		return L(e) ? ie({
 			model: n,
-			middleware: v({ tagName: "thinking" })
+			middleware: x({ tagName: "thinking" })
 		}) : n;
 	}
-	if (R(e)) {
-		let t = e.slice(F.length).replace(M, "");
-		return z(e) ? B.responses(t) : B.chat(t);
+	if (le(e)) {
+		let t = e.slice(I.length).replace(N, "");
+		return L(e) ? R.responses(t) : R.chat(t);
 	}
 	throw Error(`Unsupported modelId: ${e}`);
-}, se = () => A.chat("gpt-5.4-mini"), H = () => A.chat(process.env.AGENT_SDK_ARTIFACT_MODEL ?? "gpt-5.4-mini"), U = "\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", W = "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.", ce = (e) => `\
+}, de = () => A.chat("gpt-5.4-mini"), B = () => A.chat(process.env.AGENT_SDK_ARTIFACT_MODEL ?? "gpt-5.4-mini"), V = "\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", H = "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.", U = (e) => `\
 About the origin of user's request:
 - lat: ${e.latitude ?? "unknown"}
 - lon: ${e.longitude ?? "unknown"}
 - city: ${e.city ?? "unknown"}
 - country: ${e.country ?? "unknown"}
-`, le = ({ selectedChatModel: e, requestHints: t, agentInstructions: n, knowledge: r }) => {
-	let i = ce(t), a = [n?.trim() ? `Agent instructions:\n${n.trim()}` : "", r?.trim() ? `Knowledge context:\n${r.trim()}` : ""].filter(Boolean).join("\n\n");
+`, fe = ({ selectedChatModel: e, requestHints: t, agentInstructions: n, knowledge: r }) => {
+	let i = U(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") ? `${W}\n\nWhen you reason, put your reasoning inside <thinking>...</thinking> tags. Then provide your final answer outside of those tags.\n\n${i}` : `${W}\n\n${i}`;
+		let t = e.startsWith("google/") && e.includes("thinking") ? `${H}\n\nWhen you reason, put your reasoning inside <thinking>...</thinking> tags. Then provide your final answer outside of those tags.\n\n${i}` : `${H}\n\n${i}`;
 		return a ? `${t}\n\n${a}` : t;
 	}
-	let o = `${W}\n\n${i}\n\n${U}`;
+	let o = `${H}\n\n${i}\n\n${V}`;
 	return a ? `${o}\n\n${a}` : o;
-}, ue = "\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", de = "\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", G = (e, t) => {
+}, pe = "\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", me = "\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", W = (e, t) => {
 	let n = "document";
 	return t === "code" ? n = "script" : t === "sheet" && (n = "spreadsheet"), `Rewrite the following ${n} based on the given prompt.
 
 ${e}`;
-}, fe = "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", K = [
+}, he = "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", ge = [
 	"text",
 	"code",
 	"sheet"
-], pe = (e) => e ? e === "1" || e.toLowerCase() === "true" || e.toLowerCase() === "yes" : !1, q = () => !pe(process.env.AGENT_SDK_LIVE_TOOLS) && (pe(process.env.PLAYWRIGHT) || process.env.NODE_ENV === "test"), me = async ({ system: e, prompt: t, dataStream: n, providerOptions: r }) => {
-	if (q()) {
+], _e = (e) => e ? e === "1" || e.toLowerCase() === "true" || e.toLowerCase() === "yes" : !1, G = () => !_e(process.env.AGENT_SDK_LIVE_TOOLS) && (_e(process.env.PLAYWRIGHT) || process.env.NODE_ENV === "test"), ve = async ({ system: e, prompt: t, dataStream: n, providerOptions: r }) => {
+	if (G()) {
 		let e = `Stub document: ${t}`;
 		return n.write({
 			type: "data-textDelta",
@@ -66,10 +66,10 @@ ${e}`;
 			transient: !0
 		}), e;
 	}
-	let i = "", { fullStream: a } = x({
-		model: H(),
+	let i = "", { fullStream: a } = C({
+		model: B(),
 		system: e,
-		experimental_transform: y({ chunking: "word" }),
+		experimental_transform: ne({ chunking: "word" }),
 		prompt: t,
 		providerOptions: r
 	});
@@ -79,8 +79,8 @@ ${e}`;
 		transient: !0
 	}));
 	return i;
-}, he = async ({ system: e, prompt: t, dataStream: n }) => {
-	if (q()) {
+}, ye = async ({ system: e, prompt: t, dataStream: n }) => {
+	if (G()) {
 		let e = `print("Stub code for: ${t}")\n`;
 		return n.write({
 			type: "data-codeDelta",
@@ -88,11 +88,11 @@ ${e}`;
 			transient: !0
 		}), e;
 	}
-	let r = "", { fullStream: i } = b({
-		model: H(),
+	let r = "", { fullStream: i } = S({
+		model: B(),
 		system: e,
 		prompt: t,
-		schema: j.object({ code: j.string() })
+		schema: M.object({ code: M.string() })
 	});
 	for await (let e of i) {
 		if (e.type !== "object") continue;
@@ -104,8 +104,8 @@ ${e}`;
 		}), r = t);
 	}
 	return r;
-}, ge = async ({ system: e, prompt: t, dataStream: n }) => {
-	if (q()) {
+}, be = async ({ system: e, prompt: t, dataStream: n }) => {
+	if (G()) {
 		let e = `name,value\n${t.replaceAll(",", " ")},1\n`;
 		return n.write({
 			type: "data-sheetDelta",
@@ -113,11 +113,11 @@ ${e}`;
 			transient: !0
 		}), e;
 	}
-	let r = "", { fullStream: i } = b({
-		model: H(),
+	let r = "", { fullStream: i } = S({
+		model: B(),
 		system: e,
 		prompt: t,
-		schema: j.object({ csv: j.string().describe("CSV data") })
+		schema: M.object({ csv: M.string().describe("CSV data") })
 	});
 	for await (let e of i) {
 		if (e.type !== "object") continue;
@@ -133,7 +133,7 @@ ${e}`;
 		data: r,
 		transient: !0
 	}), r;
-}, _e = async (e) => {
+}, xe = 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;
@@ -147,22 +147,22 @@ ${e}`;
 	} catch {
 		return null;
 	}
-}, ve = ({ ctx: e, dataStream: t, requestHints: n }) => {
+}, Se = ({ ctx: e, dataStream: t, requestHints: n }) => {
 	let r = i(e), a = async (t) => {
-		let n = await p(e, t);
+		let n = await m(e, t);
 		return n.length > 0 ? n.at(-1) ?? null : null;
-	}, o = S({
+	}, o = w({
 		description: "Get the current weather at a location. You can provide either coordinates or a city name.",
-		inputSchema: j.object({
-			latitude: j.number().optional(),
-			longitude: j.number().optional(),
-			city: j.string().describe("City name (e.g., 'San Francisco', 'New York', 'London')").optional()
+		inputSchema: M.object({
+			latitude: M.number().optional(),
+			longitude: M.number().optional(),
+			city: M.string().describe("City name (e.g., 'San Francisco', 'New York', 'London')").optional()
 		}),
 		needsApproval: !0,
 		execute: async (e) => {
 			let t, n;
 			if (e.city) {
-				let r = await _e(e.city);
+				let r = await xe(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;
@@ -170,14 +170,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;
 		}
-	}), c = S({
+	}), s = w({
 		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: j.object({
-			title: j.string().describe("The title of the artifact"),
-			kind: j.enum(K).describe("Use 'code' for programming, 'text' for prose, and 'sheet' for CSV/tabular data")
+		inputSchema: M.object({
+			title: M.string().describe("The title of the artifact"),
+			kind: M.enum(ge).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,
@@ -196,19 +196,19 @@ ${e}`;
 				transient: !0
 			});
 			let a = "";
-			return r === "text" ? a = await me({
+			return r === "text" ? a = await ve({
 				system: "Write about the given topic. Markdown is supported. Use headings wherever appropriate.",
 				prompt: n,
 				dataStream: t
-			}) : r === "code" ? a = await he({
-				system: ue,
+			}) : r === "code" ? a = await ye({
+				system: pe,
 				prompt: n,
 				dataStream: t
-			}) : r === "sheet" && (a = await ge({
-				system: de,
+			}) : r === "sheet" && (a = await be({
+				system: me,
 				prompt: n,
 				dataStream: t
-			})), await u(e, {
+			})), await d(e, {
 				id: i,
 				title: n,
 				kind: r,
@@ -224,40 +224,40 @@ ${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({
+	}), l = w({
 		description: "Fully rewrite an existing artifact with the given description. Use editDocument instead for targeted find-and-replace changes.",
-		inputSchema: j.object({
-			id: j.string().describe("The ID of the artifact to rewrite"),
-			description: j.string().default("Improve the content").describe("The description of changes that need to be made")
+		inputSchema: M.object({
+			id: M.string().describe("The ID of the artifact to rewrite"),
+			description: M.string().default("Improve the content").describe("The description of changes that need to be made")
 		}),
 		execute: async ({ id: n, description: r }) => {
 			let i = await a(n);
 			if (!i) return { error: "Document not found" };
 			let o = i.kind;
-			if (!K.includes(o)) return { error: `Unsupported document kind: ${i.kind}` };
+			if (!ge.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 me({
-				system: G(i.content, i.kind),
+			return o === "text" ? s = await ve({
+				system: W(i.content, i.kind),
 				prompt: r,
 				dataStream: t,
 				providerOptions: { openai: { prediction: {
 					type: "content",
 					content: i.content
 				} } }
-			}) : o === "code" ? s = await he({
-				system: G(i.content, i.kind),
+			}) : o === "code" ? s = await ye({
+				system: W(i.content, i.kind),
 				prompt: r,
 				dataStream: t
-			}) : o === "sheet" && (s = await ge({
-				system: G(i.content, i.kind),
+			}) : o === "sheet" && (s = await be({
+				system: W(i.content, i.kind),
 				prompt: r,
 				dataStream: t
-			})), await u(e, {
+			})), await d(e, {
 				id: i.id,
 				title: i.title,
 				kind: i.kind,
@@ -276,24 +276,24 @@ ${e}`;
 	});
 	return {
 		getWeather: o,
-		createDocument: c,
-		editDocument: S({
+		createDocument: s,
+		editDocument: w({
 			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: j.object({
-				id: j.string().describe("The ID of the artifact to edit"),
-				old_string: j.string().min(1).describe("Exact string to find. Include enough surrounding context to make the match unique."),
-				new_string: j.string().describe("Replacement string"),
-				replace_all: j.boolean().optional().describe("Replace all occurrences instead of only the first one")
+			inputSchema: M.object({
+				id: M.string().describe("The ID of the artifact to edit"),
+				old_string: M.string().min(1).describe("Exact string to find. Include enough surrounding context to make the match unique."),
+				new_string: M.string().describe("Replacement string"),
+				replace_all: M.boolean().optional().describe("Replace all occurrences instead of only the first one")
 			}),
 			execute: async ({ id: n, old_string: r, new_string: i, replace_all: o }) => {
 				let s = await a(n);
 				if (!s) return { error: "Document not found" };
 				if (!s.content) return { error: "Document has no content" };
 				let c = s.kind;
-				if (!K.includes(c)) return { error: `Unsupported document kind: ${s.kind}` };
+				if (!ge.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 u(e, {
+				return await d(e, {
 					id: s.id,
 					title: s.title,
 					kind: s.kind,
@@ -327,18 +327,18 @@ ${e}`;
 			}
 		}),
 		updateDocument: l,
-		requestSuggestions: S({
+		requestSuggestions: w({
 			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: j.object({ documentId: j.string().describe("The UUID of an existing document artifact that was previously created with createDocument") }),
+			inputSchema: M.object({ documentId: M.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 (q()) {
+				if (G()) {
 					let a = {
 						originalText: "Stub original sentence.",
 						suggestedText: "Stub suggested sentence.",
 						description: "Stub suggestion",
-						id: E(),
+						id: D(),
 						documentId: n,
 						documentCreatedAt: i.createdAt,
 						isResolved: !1,
@@ -349,7 +349,7 @@ ${e}`;
 						type: "data-suggestion",
 						data: a,
 						transient: !0
-					}), await s(e, {
+					}), await c(e, {
 						documentId: n,
 						suggestions: [a]
 					}), {
@@ -359,24 +359,24 @@ ${e}`;
 						message: "Suggestions have been added to the document"
 					};
 				}
-				let o = [], { partialOutputStream: c } = x({
-					model: H(),
+				let o = [], { partialOutputStream: s } = C({
+					model: B(),
 					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: _.array({ element: j.object({
-						originalSentence: j.string().describe("The original sentence"),
-						suggestedSentence: j.string().describe("The suggested sentence"),
-						description: j.string().describe("The description of the suggestion")
+					output: y.array({ element: M.object({
+						originalSentence: M.string().describe("The original sentence"),
+						suggestedSentence: M.string().describe("The suggested sentence"),
+						description: M.string().describe("The description of the suggestion")
 					}) })
 				}), l = 0;
-				for await (let e of c) if (e) for (let a = l; a < e.length; a++) {
+				for await (let e of s) if (e) for (let a = l; a < e.length; a++) {
 					let s = e[a];
 					if (!s?.originalSentence || !s?.suggestedSentence || !s?.description) continue;
 					let c = {
 						originalText: s.originalSentence,
 						suggestedText: s.suggestedSentence,
 						description: s.description,
-						id: E(),
+						id: D(),
 						documentId: n,
 						documentCreatedAt: i.createdAt,
 						isResolved: !1,
@@ -389,7 +389,7 @@ ${e}`;
 						transient: !0
 					}), o.push(c), l++;
 				}
-				return o.length > 0 && await s(e, {
+				return o.length > 0 && await c(e, {
 					documentId: n,
 					suggestions: o
 				}), {
@@ -410,34 +410,34 @@ ${e}`;
 			}
 		})
 	};
-}, ye = (e) => e ? !e.trim() : !0, J = (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(), be = (e) => {
+}, Ce = (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(), we = (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;
-}, xe = (e) => {
+}, Te = (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";
-}, Y = (e) => {
+}, q = (e) => {
 	let t = e.trim();
-	return !t || xe(t) ? "" : be(t.replaceAll("\n", " ").replaceAll("\r", " ").split(/\s+/).filter((e) => e.trim() !== "").slice(0, 5).join(" "));
-}, X = "/api/rb/agents/chat";
-j.any(), j.any();
-var Se = 1024 * 1024 * 20, Ce = 1e5, we = /^\/api\/rb\/files\/([^/]+)\/?$/, Te = "test-model", Z = async (e) => new Promise((t) => setTimeout(t, e)), Q = async (e, t) => {
-	let n = h(t);
-	n && !n.isDone || await a(e, {
+	return !t || Te(t) ? "" : we(t.replaceAll("\n", " ").replaceAll("\r", " ").split(/\s+/).filter((e) => e.trim() !== "").slice(0, 5).join(" "));
+}, J = "/api/rb/agents/chat";
+M.any(), M.any();
+var Ee = 1024 * 1024 * 20, De = 1e5, Oe = /^\/api\/rb\/files\/([^/]+)\/?$/, ke = "test-model", Ae = "openai/gpt-5.4-mini", Y = async (e) => new Promise((t) => setTimeout(t, e)), X = async (e, t) => {
+	let n = _(t);
+	n && !n.isDone || await g(e, {
 		chatId: t,
 		isStreaming: !1
 	});
-}, Ee = (e) => e ? e === "1" || e.toLowerCase() === "true" || e.toLowerCase() === "yes" : !1, $ = (e) => typeof e == "string" && e.trim() || null, De = (e) => {
+}, je = (e) => e ? e === "1" || e.toLowerCase() === "true" || e.toLowerCase() === "yes" : !1, Z = (e) => typeof e == "string" && e.trim() || null, Me = (e) => {
 	if (typeof e == "boolean") return e;
-	let t = $(e)?.toLowerCase();
+	let t = Z(e)?.toLowerCase();
 	return t ? t === "true" ? !0 : t === "false" ? !1 : null : null;
-}, Oe = (e) => {
+}, Ne = (e) => {
 	let t = e.req?.session?.user?.currentTenantId;
-	return $(t);
-}, ke = (e) => {
+	return Z(t);
+}, Pe = (e) => {
 	let t = e.req?.session?.user?.id;
-	return $(t);
-}, Ae = () => $(process.env.RB_FILESYSTEM_BUCKET_NAME) ?? "fs", je = (e) => T("sha256").update(Buffer.from(e)).digest("hex"), Me = (e) => {
+	return Z(t);
+}, Fe = () => Z(process.env.RB_FILESYSTEM_BUCKET_NAME) ?? "fs", Ie = (e) => E("sha256").update(Buffer.from(e)).digest("hex"), Le = (e) => {
 	let t = e.trim();
 	if (!t) return null;
 	let n;
@@ -450,36 +450,36 @@ var Se = 1024 * 1024 * 20, Ce = 1e5, we = /^\/api\/rb\/files\/([^/]+)\/?$/, Te =
 			return null;
 		}
 	}
-	let r = we.exec(n.pathname);
+	let r = Oe.exec(n.pathname);
 	if (!r) return null;
-	let i = $(decodeURIComponent(r[1]));
+	let i = Z(decodeURIComponent(r[1]));
 	return i ? {
 		fileId: i,
-		uploadKey: $(n.searchParams.get("rb-upload-key"))
+		uploadKey: Z(n.searchParams.get("rb-upload-key"))
 	} : null;
-}, Ne = async ({ ctx: e, fileId: t, uploadKey: n }) => {
-	let r = Oe(e);
+}, Re = async ({ ctx: e, fileId: t, uploadKey: n }) => {
+	let r = Ne(e);
 	if (!r) throw Error("tenant_missing");
 	let i;
 	try {
-		i = new O(t);
+		i = new ae(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: Ae() }), [s] = await o.find({ _id: i }).limit(1).toArray();
+	let o = new O(a, { bucketName: Fe() }), [s] = await o.find({ _id: i }).limit(1).toArray();
 	if (!s) throw Error("file_not_found");
-	let c = De(s?.metadata?.isPublic) ?? !1, l = ke(e), u = $(s?.metadata?.userId), d = $(s?.metadata?.ownerKeyHash);
+	let c = Me(s?.metadata?.isPublic) ?? !1, l = Pe(e), u = Z(s?.metadata?.userId), d = Z(s?.metadata?.ownerKeyHash);
 	if (!c) {
-		let e = !!(l && u && l === u), t = !!(n && d && je(n) === d);
+		let e = !!(l && u && l === u), t = !!(n && d && Ie(n) === d);
 		if (!e && !t) throw Error("unauthorized");
 	}
-	let f = $(s?.metadata?.mimeType), p = [], m = 0;
+	let f = Z(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 > Se) {
+			if (m += e.length, m > Ee) {
 				n.destroy(/* @__PURE__ */ Error("attachment_too_large"));
 				return;
 			}
@@ -489,16 +489,16 @@ var Se = 1024 * 1024 * 20, Ce = 1e5, we = /^\/api\/rb\/files\/([^/]+)\/?$/, Te =
 		data: Buffer.concat(p, m),
 		mimeType: f
 	};
-}, Pe = async (e, t) => {
+}, ze = 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 = Me(t.data);
+			let n = Le(t.data);
 			n && r.push((async () => {
-				let { data: r, mimeType: i } = await Ne({
+				let { data: r, mimeType: i } = await Re({
 					ctx: e,
 					...n
 				});
@@ -507,7 +507,7 @@ var Se = 1024 * 1024 * 20, Ce = 1e5, we = /^\/api\/rb\/files\/([^/]+)\/?$/, Te =
 		}
 	}
 	r.length > 0 && await Promise.all(r);
-}, Fe = (e) => {
+}, Be = (e) => {
 	let t = Array.isArray(e) ? e : [];
 	for (let e of t) {
 		if (!e || typeof e != "object" || !("role" in e) || e.role !== "user") continue;
@@ -515,12 +515,12 @@ var Se = 1024 * 1024 * 20, Ce = 1e5, we = /^\/api\/rb\/files\/([^/]+)\/?$/, Te =
 		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 = $(n.mediaType);
+			let r = Z(n.mediaType);
 			if (r?.startsWith("image/") || r === "application/pdf") continue;
-			let i = $(n.filename) ?? $(n.name) ?? `attachment-${e + 1}`, a = n.data;
+			let i = Z(n.filename) ?? Z(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 > Ce && (n = `${n.slice(0, Ce)}\n\n[Truncated]`);
+				n.length > De && (n = `${n.slice(0, De)}\n\n[Truncated]`);
 				let o = r === "text/csv" || i.toLowerCase().endsWith(".csv") ? "csv" : "";
 				t[e] = {
 					type: "text",
@@ -535,66 +535,66 @@ var Se = 1024 * 1024 * 20, Ce = 1e5, we = /^\/api\/rb\/files\/([^/]+)\/?$/, Te =
 			};
 		}
 	}
-}, Ie = async (e, t, n) => {
-	if (e.message?.role === "user") return J(e.message.parts);
+}, Ve = async (e, t, n) => {
+	if (e.message?.role === "user") return K(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 = J(r.parts);
+			let i = K(r.parts);
 			if (i) {
-				if (Y(i)) return i;
+				if (q(i)) return i;
 				t ??= i;
 			}
 		}
 		if (t) return t;
 	}
-	let r = await l(t, n), i = null;
+	let r = await u(t, n), i = null;
 	for (let e = r.length - 1; e >= 0; --e) {
 		let t = r[e];
 		if (t.role !== "user") continue;
-		let n = J(t.parts);
+		let n = K(t.parts);
 		if (n) {
-			if (Y(n)) return n;
+			if (q(n)) return n;
 			i ??= n;
 		}
 	}
 	return i ?? "";
-}, Le = (e) => {
+}, He = (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();
-}, Re = (e) => {
+}, Ue = (e) => {
 	let t = e.id;
 	return typeof t == "string" && t.trim() !== "" ? t : null;
-}, ze = (e) => {
+}, We = (e) => {
 	let t = e.selectedChatModel;
 	return typeof t == "string" && t.trim() !== "" ? t : null;
-}, Be = (e) => {
+}, Ge = (e) => {
 	let t = e.openaiReasoningEffort;
 	return typeof t == "string" && t.trim() !== "" ? t.trim() : null;
-}, Ve = (e) => {
+}, Ke = (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");
-}, He = (e) => {
+}, qe = (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) => Ve(e)));
-}, Ue = (e) => {
+	return t.length === 0 || t.at(-1)?.role === "user" ? !1 : t.some((e) => (Array.isArray(e?.parts) ? e.parts : []).some((e) => Ke(e)));
+}, Je = (e) => {
 	let t = /* @__PURE__ */ new Map();
 	for (let n of e) {
 		let e = Array.isArray(n?.parts) ? n.parts : [];
-		for (let n of e) Ve(n) && t.set(n.toolCallId, n);
+		for (let n of e) Ke(n) && t.set(n.toolCallId, n);
 	}
 	return t;
-}, We = (e, t) => {
+}, Ye = (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 = $(e.toolCallId);
+				let n = Z(e.toolCallId);
 				if (!n) return e;
 				let i = t.get(n);
 				return i ? (r = !0, {
@@ -610,23 +610,78 @@ var Se = 1024 * 1024 * 20, Ce = 1e5, we = /^\/api\/rb\/files\/([^/]+)\/?$/, Te =
 			return n.push(a), a;
 		})
 	};
-}, Ge = (e) => {
+}, Xe = (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;
-}, Ke = (e) => ({
+}, Ze = (e) => ({
 	latitude: null,
 	longitude: null,
 	city: null,
 	country: null
-}), qe = (e) => `agents-sdk:chat:${e}:stream`, Je = async ({ ctx: e, chatId: t, selectedChatModel: i }) => await r().resolveAgent?.({
+}), Qe = (e) => {
+	let [t] = e.split("/");
+	return t && t !== e ? t : "unknown";
+}, Q = (e) => typeof e == "number" && Number.isFinite(e) ? e : void 0, $e = (e) => e instanceof Error && e.message.trim() ? e.message : "Unknown error", et = (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: Q(t.inputTokens),
+		outputTokens: Q(t.outputTokens),
+		totalTokens: Q(t.totalTokens),
+		cachedInputTokens: Q(n.cacheReadTokens) ?? Q(t.cachedInputTokens),
+		reasoningTokens: Q(r.reasoningTokens) ?? Q(t.reasoningTokens),
+		rawUsage: Object.keys(t).length > 0 ? t.raw ?? t : void 0
+	};
+}, tt = () => ({
+	inputTokens: 0,
+	outputTokens: 0,
+	totalTokens: 0,
+	cachedInputTokens: 0,
+	reasoningTokens: 0
+}), nt = async ({ ctx: e, chatId: t, agent: n, selectedChatModel: i, mode: a }) => {
+	try {
+		return { workspaceId: Z((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 };
+	}
+}, $ = async ({ ctx: e, usageContext: t, agent: n, chatId: r, mode: i, feature: o, modelId: s, status: c, billable: l, usage: u, finishReason: d, error: f, startedAt: p, finishedAt: m = /* @__PURE__ */ new Date() }) => {
+	let h = u ? et(u) : {}, g = {
+		id: D(),
+		workspaceId: t.workspaceId,
+		agentId: n.id,
+		chatId: r,
+		mode: i,
+		feature: o,
+		provider: Qe(s),
+		model: s,
+		billable: l,
+		status: c,
+		finishReason: Z(d) ?? void 0,
+		errorMessage: f ? $e(f) : void 0,
+		...h,
+		startedAt: p,
+		finishedAt: m,
+		createdAt: /* @__PURE__ */ new Date()
+	};
+	try {
+		await a(e, g);
+	} catch (e) {
+		console.error("Failed to save agents usage event.", e);
+	}
+}, rt = (e) => `agents-sdk:chat:${e}:stream`, it = async ({ ctx: e, chatId: t, selectedChatModel: i }) => await r().resolveAgent?.({
 	ctx: e,
 	chatId: t,
 	selectedChatModel: i
-}) ?? n(), Ye = ({ ctx: e, chatId: t, mode: n, streamTransport: r }) => {
-	let a = i(e), o = qe(t), s = g({
+}) ?? n(), at = ({ ctx: e, chatId: t, mode: n, streamTransport: r }) => {
+	let a = i(e), o = rt(t), s = v({
 		chatId: t,
 		abortWhenNoSubscribersMs: n === "headless" || r ? 0 : void 0
 	});
@@ -654,93 +709,104 @@ var Se = 1024 * 1024 * 20, Ce = 1e5, we = /^\/api\/rb\/files\/([^/]+)\/?$/, Te =
 			});
 		}
 	};
-}, Xe = ({ ctx: e, chatId: t, streamTransport: n, message: r }) => {
+}, ot = ({ ctx: e, chatId: t, streamTransport: n, message: r }) => {
 	if (!n) return;
 	let a = i(e);
 	n.publish({
 		tenantId: a.tenantId,
 		userId: a.userId,
-		channel: qe(t),
+		channel: rt(t),
 		payload: {
 			chatId: t,
 			message: r
 		}
 	});
-}, Ze = async ({ payload: n, ctx: s, mode: u = "interactive", streamTransport: d = null }) => {
-	let p = Re(n);
-	if (!p) return t(s, "bad_request:api", "Parameter id is required.");
-	let h = n, g = ze(h), _ = Ee(process.env.PLAYWRIGHT), v = !Ee(process.env.AGENT_SDK_LIVE_CHAT) && _ || process.env.NODE_ENV === "test";
-	if (!g) return t(s, "bad_request:api", "Parameter selectedChatModel is required.");
-	if (g === Te && (v = !0), g !== Te && !ae.has(g)) return t(s, "bad_request:api", `Unsupported selectedChatModel: "${g}".`);
-	let y = process.env.OPENAI_API_KEY, b = process.env.GOOGLE_GENERATIVE_AI_API_KEY ?? process.env.GOOGLE_API_KEY, S = process.env.XAI_API_KEY, C = I(g), w = L(g), T = R(g), D = Be(h), O = D && D !== "default" && D !== "off" ? D : null, k = e(g);
-	if (O && !k.some((e) => e === O)) return t(s, "bad_request:api", `Unsupported reasoning effort "${O}" for selectedChatModel: "${g}".`);
-	if (!v && C && !y) return t(s, "bad_request:provider", "Missing OPENAI_API_KEY.");
-	if (!v && w && !b) return t(s, "bad_request:provider", "Missing GOOGLE_GENERATIVE_AI_API_KEY (or GOOGLE_API_KEY).");
-	if (!v && T && !S) return t(s, "bad_request:provider", "Missing XAI_API_KEY.");
-	if (!v && !C && !w && !T) return t(s, "bad_request:provider", `Unsupported selectedChatModel: "${g}".`);
-	let A = await Je({
-		ctx: s,
-		chatId: p,
-		selectedChatModel: g
-	}), ie = await r().resolveKnowledge?.({
-		ctx: s,
-		chatId: p,
-		agent: A,
-		selectedChatModel: g
-	}) ?? null, j = i(s);
-	await o(s, p) || await f(s, {
-		id: p,
+}, st = async ({ payload: n, ctx: a, mode: c = "interactive", streamTransport: d = null }) => {
+	let f = Ue(n);
+	if (!f) return t(a, "bad_request:api", "Parameter id is required.");
+	let m = n, _ = We(m), v = je(process.env.PLAYWRIGHT), y = !je(process.env.AGENT_SDK_LIVE_CHAT) && v || process.env.NODE_ENV === "test";
+	if (!_) return t(a, "bad_request:api", "Parameter selectedChatModel is required.");
+	if (_ === ke && (y = !0), _ !== ke && !oe.has(_)) return t(a, "bad_request:api", `Unsupported selectedChatModel: "${_}".`);
+	let x = process.env.OPENAI_API_KEY, ne = process.env.GOOGLE_GENERATIVE_AI_API_KEY ?? process.env.GOOGLE_API_KEY, S = process.env.XAI_API_KEY, w = se(_), ie = ce(_), T = le(_), E = Ge(m), O = E && E !== "default" && E !== "off" ? E : null, ae = e(_);
+	if (O && !ae.some((e) => e === O)) return t(a, "bad_request:api", `Unsupported reasoning effort "${O}" for selectedChatModel: "${_}".`);
+	if (!y && w && !x) return t(a, "bad_request:provider", "Missing OPENAI_API_KEY.");
+	if (!y && ie && !ne) return t(a, "bad_request:provider", "Missing GOOGLE_GENERATIVE_AI_API_KEY (or GOOGLE_API_KEY).");
+	if (!y && T && !S) return t(a, "bad_request:provider", "Missing XAI_API_KEY.");
+	if (!y && !w && !ie && !T) return t(a, "bad_request:provider", `Unsupported selectedChatModel: "${_}".`);
+	let k = await it({
+		ctx: a,
+		chatId: f,
+		selectedChatModel: _
+	}), A = await r().resolveKnowledge?.({
+		ctx: a,
+		chatId: f,
+		agent: k,
+		selectedChatModel: _
+	}) ?? null, j = await nt({
+		ctx: a,
+		chatId: f,
+		agent: k,
+		selectedChatModel: _,
+		mode: c
+	}), M = i(a), N = await o(a, f, { includeDeleted: !0 });
+	if (N?.deletedAt) return t(a, "not_found:chat");
+	N ? N.selectedChatModel !== _ && await h(a, {
+		chatId: f,
+		selectedChatModel: _
+	}) : await p(a, {
+		id: f,
 		createdAt: /* @__PURE__ */ new Date(),
-		userId: j.userId,
+		userId: M.userId,
 		title: "",
 		isStreaming: !1,
+		selectedChatModel: _,
 		lastContext: null
 	});
-	let M = h.message, N = typeof M?.id == "string" ? M.id : E(), P = Array.isArray(M?.parts) ? M?.parts : [];
-	if ((typeof M?.role == "string" ? M.role : null) === "user") {
+	let P = m.message, F = typeof P?.id == "string" ? P.id : D(), I = Array.isArray(P?.parts) ? P?.parts : [];
+	if ((typeof P?.role == "string" ? P.role : null) === "user") {
 		let e = /* @__PURE__ */ new Date();
-		await c(s, [{
-			id: N,
-			chatId: p,
+		await l(a, [{
+			id: F,
+			chatId: f,
 			role: "user",
-			parts: P,
+			parts: I,
 			attachments: [],
 			createdAt: e
-		}]), Xe({
-			ctx: s,
-			chatId: p,
+		}]), ot({
+			ctx: a,
+			chatId: f,
 			streamTransport: d,
 			message: {
-				id: N,
+				id: F,
 				role: "user",
-				parts: P,
+				parts: I,
 				metadata: { createdAt: e.toISOString() }
 			}
 		});
 	}
-	if (v) {
-		await a(s, {
-			chatId: p,
+	if (y) {
+		await g(a, {
+			chatId: f,
 			isStreaming: !0
 		});
-		let e = Ye({
-			ctx: s,
-			chatId: p,
-			mode: u,
+		let e = at({
+			ctx: a,
+			chatId: f,
+			mode: c,
 			streamTransport: d
 		});
 		try {
-			let t = Y(await Ie(h, s, p)), n = await o(s, p);
-			n && ye(n.title) && t && t !== n.title && (await m(s, {
-				chatId: p,
+			let t = q(await Ve(m, a, f)), n = await o(a, f);
+			n && Ce(n.title) && t && t !== n.title && (await s(a, {
+				chatId: f,
 				title: t
 			}), e.publishJson({
 				type: "data-chat-title",
 				data: t,
 				transient: !0
 			}));
-			let r = E();
-			_ && !e.abortController.signal.aborted && await Z(600), e.publishJson({
+			let r = D();
+			v && !e.abortController.signal.aborted && await Y(600), e.publishJson({
 				type: "start",
 				messageId: r
 			});
@@ -749,18 +815,18 @@ var Se = 1024 * 1024 * 20, Ce = 1e5, we = /^\/api\/rb\/files\/([^/]+)\/?$/, Te =
 				type: "text-start",
 				id: i
 			});
-			let a = Le(h), l = a ? `Echo: ${a}` : "Hello from agents-sdk", u = Date.now(), d = l.split(/(\s+)/).filter((e) => e.length > 0);
-			for (let t of d) {
+			let u = He(m), d = u ? `Echo: ${u}` : "Hello from agents-sdk", p = Date.now(), h = d.split(/(\s+)/).filter((e) => e.length > 0);
+			for (let t of h) {
 				if (e.abortController.signal.aborted) break;
 				e.publishJson({
 					type: "text-delta",
 					id: i,
 					delta: t
-				}), await Z(_ ? 120 : 80);
+				}), await Y(v ? 120 : 80);
 			}
-			if (_ && !e.abortController.signal.aborted) {
-				let e = Date.now() - u, t = 1200;
-				e < t && await Z(t - e);
+			if (v && !e.abortController.signal.aborted) {
+				let e = Date.now() - p, t = 1200;
+				e < t && await Y(t - e);
 			}
 			return e.abortController.signal.aborted ? {} : (e.publishJson({
 				type: "text-end",
@@ -768,118 +834,160 @@ var Se = 1024 * 1024 * 20, Ce = 1e5, we = /^\/api\/rb\/files\/([^/]+)\/?$/, Te =
 			}), e.publishJson({
 				type: "finish",
 				finishReason: "stop"
-			}), e.publishDone(), await c(s, [{
+			}), e.publishDone(), await l(a, [{
 				id: r,
-				chatId: p,
+				chatId: f,
 				role: "assistant",
 				parts: [{
 					type: "text",
-					text: l
+					text: d
 				}],
 				attachments: [],
 				createdAt: /* @__PURE__ */ new Date()
-			}]), {});
+			}]), await $({
+				ctx: a,
+				usageContext: j,
+				agent: k,
+				chatId: f,
+				mode: c,
+				feature: "chat",
+				modelId: _,
+				status: "success",
+				billable: !1,
+				usage: tt(),
+				finishReason: "stop",
+				startedAt: new Date(p)
+			}), {});
 		} finally {
-			await Q(s, p);
+			await X(a, f);
 		}
 	}
-	let F = Ke(s), B = He(h);
-	if (Array.isArray(h.messages) && !B) {
-		let e = h.messages.flatMap((e) => {
+	let R = Ze(a), z = qe(m);
+	if (Array.isArray(m.messages) && !z) {
+		let e = m.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,
-				chatId: p,
+				chatId: f,
 				role: n,
 				parts: r,
 				attachments: [],
-				createdAt: Ge(e?.metadata) ?? /* @__PURE__ */ new Date()
+				createdAt: Xe(e?.metadata) ?? /* @__PURE__ */ new Date()
 			}];
 		});
-		e.length > 0 && await c(s, e);
+		e.length > 0 && await l(a, e);
 	}
-	let V;
-	if (Array.isArray(h.messages) && h.messages.length > 0 && B) {
-		let e = await l(s, p);
-		if (e.length === 0) V = h.messages;
+	let B;
+	if (Array.isArray(m.messages) && m.messages.length > 0 && z) {
+		let e = await u(a, f);
+		if (e.length === 0) B = m.messages;
 		else {
-			let { changedMessages: t, updatedMessages: n } = We(e, Ue(h.messages));
-			t.length > 0 && await c(s, t), V = n.map((e) => ({
+			let { changedMessages: t, updatedMessages: n } = Ye(e, Je(m.messages));
+			t.length > 0 && await l(a, t), B = n.map((e) => ({
 				id: e.id,
 				role: e.role,
 				parts: e.parts,
 				metadata: { createdAt: e.createdAt.toISOString() }
 			}));
 		}
-	} else V = Array.isArray(h.messages) && h.messages.length > 0 ? h.messages : (await l(s, p)).map((e) => ({
+	} else B = Array.isArray(m.messages) && m.messages.length > 0 ? m.messages : (await u(a, f)).map((e) => ({
 		id: e.id,
 		role: e.role,
 		parts: e.parts,
 		metadata: { createdAt: e.createdAt.toISOString() }
 	}));
-	await a(s, {
-		chatId: p,
+	await g(a, {
+		chatId: f,
 		isStreaming: !0
 	});
-	let H = Ye({
-		ctx: s,
-		chatId: p,
-		mode: u,
+	let V = at({
+		ctx: a,
+		chatId: f,
+		mode: c,
 		streamTransport: d
-	}), U = "", W = "";
+	}), H = "", U = "";
 	try {
-		U = await Ie(h, s, p), W = Y(U);
-		let e = await o(s, p);
-		e && ye(e.title) && W && W !== e.title && (await m(s, {
-			chatId: p,
-			title: W
-		}), H.publishJson({
+		H = await Ve(m, a, f), U = q(H);
+		let e = await o(a, f);
+		e && Ce(e.title) && U && U !== e.title && (await s(a, {
+			chatId: f,
+			title: U
+		}), V.publishJson({
 			type: "data-chat-title",
-			data: W,
+			data: U,
 			transient: !0
 		}));
 	} catch (e) {
-		console.error(e), H.publishJson({
+		console.error(e), V.publishJson({
 			type: "error",
 			errorText: "Something went wrong."
-		}), H.publishDone();
+		}), V.publishDone();
 		try {
-			await Q(s, p);
+			await X(a, f);
 		} catch (e) {
 			console.error("Failed to update chat streaming state.", e);
 		}
 		return {};
 	}
-	U && W && y && (async () => {
+	H && U && x && (async () => {
+		let e = /* @__PURE__ */ new Date();
 		try {
-			let { text: e } = await ne({
-				model: se(),
-				system: fe,
-				prompt: U
-			}), t = be(e);
-			if (!t) return;
-			let n = await o(s, p);
+			let { text: t } = await te({
+				model: de(),
+				system: he,
+				prompt: H,
+				onFinish: async (t) => {
+					await $({
+						ctx: a,
+						usageContext: j,
+						agent: k,
+						chatId: f,
+						mode: c,
+						feature: "title",
+						modelId: Ae,
+						status: "success",
+						billable: !0,
+						usage: t.totalUsage ?? t.usage,
+						finishReason: t.finishReason,
+						startedAt: e
+					});
+				}
+			}), n = we(t);
 			if (!n) return;
-			n.title !== t && (await m(s, {
-				chatId: p,
-				title: t
-			}), H.publishJson({
+			let r = await o(a, f);
+			if (!r) return;
+			r.title !== n && (await s(a, {
+				chatId: f,
+				title: n
+			}), V.publishJson({
 				type: "data-chat-title",
-				data: t,
+				data: n,
 				transient: !0
 			}));
-		} catch (e) {
-			console.warn("Failed to generate chat title.", e);
+		} catch (t) {
+			await $({
+				ctx: a,
+				usageContext: j,
+				agent: k,
+				chatId: f,
+				mode: c,
+				feature: "title",
+				modelId: Ae,
+				status: "error",
+				billable: !1,
+				error: t,
+				startedAt: e
+			}), console.warn("Failed to generate chat title.", t);
 		}
 	})();
-	let ce = te({
-		originalMessages: V,
+	let pe = b({
+		originalMessages: B,
 		onError: (e) => (console.error(e), "Something went wrong."),
 		onFinish: async ({ responseMessage: e }) => {
 			let t = Array.isArray(e.parts) ? e.parts : [];
-			await c(s, [{
-				id: e.id ?? E(),
-				chatId: p,
+			await l(a, [{
+				id: e.id ?? D(),
+				chatId: f,
 				role: e.role ?? "assistant",
 				parts: t,
 				attachments: [],
@@ -887,10 +995,10 @@ var Se = 1024 * 1024 * 20, Ce = 1e5, we = /^\/api\/rb\/files\/([^/]+)\/?$/, Te =
 			}]);
 		},
 		execute: async ({ writer: e }) => {
-			let t = ve({
-				ctx: s,
+			let t = Se({
+				ctx: a,
 				dataStream: e,
-				requestHints: F
+				requestHints: R
 			}), n = [
 				"getWeather",
 				"createDocument",
@@ -904,87 +1012,123 @@ var Se = 1024 * 1024 * 20, Ce = 1e5, we = /^\/api\/rb\/files\/([^/]+)\/?$/, Te =
 				"editDocument",
 				"updateDocument",
 				"requestSuggestions"
-			], a = z(g), o = a ? [] : C ? n : i, c = t, l = o, u = await r().resolveTools?.({
-				ctx: s,
+			], o = L(_), s = o ? [] : w ? n : i, l = t, u = s, d = await r().resolveTools?.({
+				ctx: a,
 				dataStream: e,
-				requestHints: F,
-				agent: A,
-				selectedChatModel: g,
+				requestHints: R,
+				agent: k,
+				selectedChatModel: _,
 				availableTools: t,
-				defaultActiveTools: o
+				defaultActiveTools: s
 			});
-			u && (c = u.tools, l = u.activeTools ?? Object.keys(u.tools));
-			let d = await ee(V.map((e) => ({
+			d && (l = d.tools, u = d.activeTools ?? Object.keys(d.tools));
+			let p = await ee(B.map((e) => ({
 				role: e?.role,
 				parts: e?.parts,
 				metadata: e?.metadata
 			})), {
-				tools: c,
+				tools: l,
 				ignoreIncompleteToolCalls: !0
 			});
-			await Pe(s, d), C && Fe(d);
-			let f = typeof h.openaiReasoningSummary == "string" ? h.openaiReasoningSummary : null, p;
-			C && O ? p = { openai: {
+			await ze(a, p), w && Be(p);
+			let h = typeof m.openaiReasoningSummary == "string" ? m.openaiReasoningSummary : null, g;
+			w && O ? g = { openai: {
 				reasoningEffort: O,
-				reasoningSummary: f ?? void 0
-			} } : T && a && O && (p = { xai: { reasoningEffort: O } });
-			let m = x({
-				model: oe(g),
-				system: le({
-					selectedChatModel: g,
-					requestHints: F,
-					agentInstructions: A.instructions,
-					knowledge: ie
+				reasoningSummary: h ?? void 0
+			} } : T && o && O && (g = { xai: { reasoningEffort: O } });
+			let v = !1, y = /* @__PURE__ */ new Date(), b = async ({ status: e, usage: t, finishReason: n, error: r }) => {
+				v || (v = !0, await $({
+					ctx: a,
+					usageContext: j,
+					agent: k,
+					chatId: f,
+					mode: c,
+					feature: "chat",
+					modelId: _,
+					status: e,
+					billable: e === "success",
+					usage: t,
+					finishReason: n,
+					error: r,
+					startedAt: y
+				}));
+			}, x = C({
+				model: ue(_),
+				system: fe({
+					selectedChatModel: _,
+					requestHints: R,
+					agentInstructions: k.instructions,
+					knowledge: A
 				}),
-				messages: d,
-				tools: c,
+				messages: p,
+				tools: l,
 				stopWhen: re(5),
-				abortSignal: H.abortController.signal,
-				providerOptions: p,
-				activeTools: l
+				abortSignal: V.abortController.signal,
+				providerOptions: g,
+				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(m.toUIMessageStream({
-				originalMessages: V,
-				sendReasoning: a || !!(C && O)
+			e.merge(x.toUIMessageStream({
+				originalMessages: B,
+				sendReasoning: o || !!(w && O)
 			}));
 		}
 	});
 	return (async () => {
 		let e = null;
 		try {
-			let t = ce.getReader();
+			let t = pe.getReader();
 			for (e = () => t.releaseLock();;) {
 				let { value: e, done: n } = await t.read();
 				if (n) break;
-				H.publishJson(e);
+				V.publishJson(e);
 			}
-			H.publishDone();
+			V.publishDone();
 		} catch (e) {
-			console.error(e), H.publishJson({
+			console.error(e), V.publishJson({
 				type: "error",
 				errorText: "Something went wrong."
-			}), H.publishDone();
+			}), V.publishDone();
 		} finally {
 			e?.();
 			try {
-				await Q(s, p);
+				await X(a, f);
 			} catch (e) {
 				console.error("Failed to update chat streaming state.", e);
 			}
 		}
 	})(), {};
-}, Qe = async (e, t) => Ze({
+}, ct = async (e, t) => st({
 	payload: e,
 	ctx: t,
 	mode: "interactive"
-}), $e = async (e, n) => {
+}), lt = async (e, n) => {
 	let r = n.req.query.id, i = Array.isArray(r) ? r[0] : r;
-	return typeof i != "string" || i.trim() === "" ? t(n, "bad_request:api") : d(n, i);
-}, et = (e) => {
-	e.get(X, (async (e, n) => {
+	return typeof i != "string" || i.trim() === "" ? t(n, "bad_request:api") : f(n, i);
+}, ut = (e) => {
+	e.get(J, (async (e, n) => {
 		let r = n.req.query.id, i = Array.isArray(r) ? r[0] : r;
 		return typeof i != "string" || i.trim() === "" ? t(n, "bad_request:api", "Parameter id is required.") : await o(n, i) || t(n, "not_found:chat");
-	})), e.post(X, Qe), e.delete(X, $e);
+	})), e.post(J, ct), e.delete(J, lt);
 };
 //#endregion
-export { et as default, Ze as runAgentLoop };
+export { ut as default, st as runAgentLoop };