drizzle-cube 0.4.28 → 0.4.30

package/dist/adapters/dist-De5fzUEM.js

@@ -0,0 +1,581 @@
+//#region node_modules/@google/generative-ai/dist/index.mjs
+var e;
+(function(e) {
+	e.STRING = "string", e.NUMBER = "number", e.INTEGER = "integer", e.BOOLEAN = "boolean", e.ARRAY = "array", e.OBJECT = "object";
+})(e ||= {});
+var t;
+(function(e) {
+	e.LANGUAGE_UNSPECIFIED = "language_unspecified", e.PYTHON = "python";
+})(t ||= {});
+var n;
+(function(e) {
+	e.OUTCOME_UNSPECIFIED = "outcome_unspecified", e.OUTCOME_OK = "outcome_ok", e.OUTCOME_FAILED = "outcome_failed", e.OUTCOME_DEADLINE_EXCEEDED = "outcome_deadline_exceeded";
+})(n ||= {});
+var r = [
+	"user",
+	"model",
+	"function",
+	"system"
+], i;
+(function(e) {
+	e.HARM_CATEGORY_UNSPECIFIED = "HARM_CATEGORY_UNSPECIFIED", e.HARM_CATEGORY_HATE_SPEECH = "HARM_CATEGORY_HATE_SPEECH", e.HARM_CATEGORY_SEXUALLY_EXPLICIT = "HARM_CATEGORY_SEXUALLY_EXPLICIT", e.HARM_CATEGORY_HARASSMENT = "HARM_CATEGORY_HARASSMENT", e.HARM_CATEGORY_DANGEROUS_CONTENT = "HARM_CATEGORY_DANGEROUS_CONTENT", e.HARM_CATEGORY_CIVIC_INTEGRITY = "HARM_CATEGORY_CIVIC_INTEGRITY";
+})(i ||= {});
+var a;
+(function(e) {
+	e.HARM_BLOCK_THRESHOLD_UNSPECIFIED = "HARM_BLOCK_THRESHOLD_UNSPECIFIED", e.BLOCK_LOW_AND_ABOVE = "BLOCK_LOW_AND_ABOVE", e.BLOCK_MEDIUM_AND_ABOVE = "BLOCK_MEDIUM_AND_ABOVE", e.BLOCK_ONLY_HIGH = "BLOCK_ONLY_HIGH", e.BLOCK_NONE = "BLOCK_NONE";
+})(a ||= {});
+var o;
+(function(e) {
+	e.HARM_PROBABILITY_UNSPECIFIED = "HARM_PROBABILITY_UNSPECIFIED", e.NEGLIGIBLE = "NEGLIGIBLE", e.LOW = "LOW", e.MEDIUM = "MEDIUM", e.HIGH = "HIGH";
+})(o ||= {});
+var s;
+(function(e) {
+	e.BLOCKED_REASON_UNSPECIFIED = "BLOCKED_REASON_UNSPECIFIED", e.SAFETY = "SAFETY", e.OTHER = "OTHER";
+})(s ||= {});
+var c;
+(function(e) {
+	e.FINISH_REASON_UNSPECIFIED = "FINISH_REASON_UNSPECIFIED", e.STOP = "STOP", e.MAX_TOKENS = "MAX_TOKENS", e.SAFETY = "SAFETY", e.RECITATION = "RECITATION", e.LANGUAGE = "LANGUAGE", e.BLOCKLIST = "BLOCKLIST", e.PROHIBITED_CONTENT = "PROHIBITED_CONTENT", e.SPII = "SPII", e.MALFORMED_FUNCTION_CALL = "MALFORMED_FUNCTION_CALL", e.OTHER = "OTHER";
+})(c ||= {});
+var l;
+(function(e) {
+	e.TASK_TYPE_UNSPECIFIED = "TASK_TYPE_UNSPECIFIED", e.RETRIEVAL_QUERY = "RETRIEVAL_QUERY", e.RETRIEVAL_DOCUMENT = "RETRIEVAL_DOCUMENT", e.SEMANTIC_SIMILARITY = "SEMANTIC_SIMILARITY", e.CLASSIFICATION = "CLASSIFICATION", e.CLUSTERING = "CLUSTERING";
+})(l ||= {});
+var u;
+(function(e) {
+	e.MODE_UNSPECIFIED = "MODE_UNSPECIFIED", e.AUTO = "AUTO", e.ANY = "ANY", e.NONE = "NONE";
+})(u ||= {});
+var d;
+(function(e) {
+	e.MODE_UNSPECIFIED = "MODE_UNSPECIFIED", e.MODE_DYNAMIC = "MODE_DYNAMIC";
+})(d ||= {});
+var f = class extends Error {
+	constructor(e) {
+		super(`[GoogleGenerativeAI Error]: ${e}`);
+	}
+}, p = class extends f {
+	constructor(e, t) {
+		super(e), this.response = t;
+	}
+}, m = class extends f {
+	constructor(e, t, n, r) {
+		super(e), this.status = t, this.statusText = n, this.errorDetails = r;
+	}
+}, h = class extends f {}, g = class extends f {}, ee = "https://generativelanguage.googleapis.com", _ = "v1beta", v = "0.24.1", te = "genai-js", y;
+(function(e) {
+	e.GENERATE_CONTENT = "generateContent", e.STREAM_GENERATE_CONTENT = "streamGenerateContent", e.COUNT_TOKENS = "countTokens", e.EMBED_CONTENT = "embedContent", e.BATCH_EMBED_CONTENTS = "batchEmbedContents";
+})(y ||= {});
+var b = class {
+	constructor(e, t, n, r, i) {
+		this.model = e, this.task = t, this.apiKey = n, this.stream = r, this.requestOptions = i;
+	}
+	toString() {
+		let e = this.requestOptions?.apiVersion || _, t = `${this.requestOptions?.baseUrl || ee}/${e}/${this.model}:${this.task}`;
+		return this.stream && (t += "?alt=sse"), t;
+	}
+};
+function x(e) {
+	let t = [];
+	return e?.apiClient && t.push(e.apiClient), t.push(`${te}/${v}`), t.join(" ");
+}
+async function S(e) {
+	let t = new Headers();
+	t.append("Content-Type", "application/json"), t.append("x-goog-api-client", x(e.requestOptions)), t.append("x-goog-api-key", e.apiKey);
+	let n = e.requestOptions?.customHeaders;
+	if (n) {
+		if (!(n instanceof Headers)) try {
+			n = new Headers(n);
+		} catch (e) {
+			throw new h(`unable to convert customHeaders value ${JSON.stringify(n)} to Headers: ${e.message}`);
+		}
+		for (let [e, r] of n.entries()) {
+			if (e === "x-goog-api-key") throw new h(`Cannot set reserved header name ${e}`);
+			if (e === "x-goog-api-client") throw new h(`Header name ${e} can only be set using the apiClient field`);
+			t.append(e, r);
+		}
+	}
+	return t;
+}
+async function ne(e, t, n, r, i, a) {
+	let o = new b(e, t, n, r, a);
+	return {
+		url: o.toString(),
+		fetchOptions: Object.assign(Object.assign({}, D(a)), {
+			method: "POST",
+			headers: await S(o),
+			body: i
+		})
+	};
+}
+async function C(e, t, n, r, i, a = {}, o = fetch) {
+	let { url: s, fetchOptions: c } = await ne(e, t, n, r, i, a);
+	return w(s, c, o);
+}
+async function w(e, t, n = fetch) {
+	let r;
+	try {
+		r = await n(e, t);
+	} catch (t) {
+		T(t, e);
+	}
+	return r.ok || await E(r, e), r;
+}
+function T(e, t) {
+	let n = e;
+	throw n.name === "AbortError" ? (n = new g(`Request aborted when fetching ${t.toString()}: ${e.message}`), n.stack = e.stack) : e instanceof m || e instanceof h || (n = new f(`Error fetching from ${t.toString()}: ${e.message}`), n.stack = e.stack), n;
+}
+async function E(e, t) {
+	let n = "", r;
+	try {
+		let t = await e.json();
+		n = t.error.message, t.error.details && (n += ` ${JSON.stringify(t.error.details)}`, r = t.error.details);
+	} catch {}
+	throw new m(`Error fetching from ${t.toString()}: [${e.status} ${e.statusText}] ${n}`, e.status, e.statusText, r);
+}
+function D(e) {
+	let t = {};
+	if (e?.signal !== void 0 || e?.timeout >= 0) {
+		let n = new AbortController();
+		e?.timeout >= 0 && setTimeout(() => n.abort(), e.timeout), e?.signal && e.signal.addEventListener("abort", () => {
+			n.abort();
+		}), t.signal = n.signal;
+	}
+	return t;
+}
+function O(e) {
+	return e.text = () => {
+		if (e.candidates && e.candidates.length > 0) {
+			if (e.candidates.length > 1 && console.warn(`This response had ${e.candidates.length} candidates. Returning text from the first candidate only. Access response.candidates directly to use the other candidates.`), j(e.candidates[0])) throw new p(`${M(e)}`, e);
+			return k(e);
+		} else if (e.promptFeedback) throw new p(`Text not available. ${M(e)}`, e);
+		return "";
+	}, e.functionCall = () => {
+		if (e.candidates && e.candidates.length > 0) {
+			if (e.candidates.length > 1 && console.warn(`This response had ${e.candidates.length} candidates. Returning function calls from the first candidate only. Access response.candidates directly to use the other candidates.`), j(e.candidates[0])) throw new p(`${M(e)}`, e);
+			return console.warn("response.functionCall() is deprecated. Use response.functionCalls() instead."), A(e)[0];
+		} else if (e.promptFeedback) throw new p(`Function call not available. ${M(e)}`, e);
+	}, e.functionCalls = () => {
+		if (e.candidates && e.candidates.length > 0) {
+			if (e.candidates.length > 1 && console.warn(`This response had ${e.candidates.length} candidates. Returning function calls from the first candidate only. Access response.candidates directly to use the other candidates.`), j(e.candidates[0])) throw new p(`${M(e)}`, e);
+			return A(e);
+		} else if (e.promptFeedback) throw new p(`Function call not available. ${M(e)}`, e);
+	}, e;
+}
+function k(e) {
+	let t = [];
+	if (e.candidates?.[0].content?.parts) for (let n of e.candidates?.[0].content?.parts) n.text && t.push(n.text), n.executableCode && t.push("\n```" + n.executableCode.language + "\n" + n.executableCode.code + "\n```\n"), n.codeExecutionResult && t.push("\n```\n" + n.codeExecutionResult.output + "\n```\n");
+	return t.length > 0 ? t.join("") : "";
+}
+function A(e) {
+	let t = [];
+	if (e.candidates?.[0].content?.parts) for (let n of e.candidates?.[0].content?.parts) n.functionCall && t.push(n.functionCall);
+	if (t.length > 0) return t;
+}
+var re = [
+	c.RECITATION,
+	c.SAFETY,
+	c.LANGUAGE
+];
+function j(e) {
+	return !!e.finishReason && re.includes(e.finishReason);
+}
+function M(e) {
+	let t = "";
+	if ((!e.candidates || e.candidates.length === 0) && e.promptFeedback) t += "Response was blocked", e.promptFeedback?.blockReason && (t += ` due to ${e.promptFeedback.blockReason}`), e.promptFeedback?.blockReasonMessage && (t += `: ${e.promptFeedback.blockReasonMessage}`);
+	else if (e.candidates?.[0]) {
+		let n = e.candidates[0];
+		j(n) && (t += `Candidate was blocked due to ${n.finishReason}`, n.finishMessage && (t += `: ${n.finishMessage}`));
+	}
+	return t;
+}
+function N(e) {
+	return this instanceof N ? (this.v = e, this) : new N(e);
+}
+function P(e, t, n) {
+	if (!Symbol.asyncIterator) throw TypeError("Symbol.asyncIterator is not defined.");
+	var r = n.apply(e, t || []), i, a = [];
+	return i = {}, o("next"), o("throw"), o("return"), i[Symbol.asyncIterator] = function() {
+		return this;
+	}, i;
+	function o(e) {
+		r[e] && (i[e] = function(t) {
+			return new Promise(function(n, r) {
+				a.push([
+					e,
+					t,
+					n,
+					r
+				]) > 1 || s(e, t);
+			});
+		});
+	}
+	function s(e, t) {
+		try {
+			c(r[e](t));
+		} catch (e) {
+			d(a[0][3], e);
+		}
+	}
+	function c(e) {
+		e.value instanceof N ? Promise.resolve(e.value.v).then(l, u) : d(a[0][2], e);
+	}
+	function l(e) {
+		s("next", e);
+	}
+	function u(e) {
+		s("throw", e);
+	}
+	function d(e, t) {
+		e(t), a.shift(), a.length && s(a[0][0], a[0][1]);
+	}
+}
+var F = /^data\: (.*)(?:\n\n|\r\r|\r\n\r\n)/;
+function I(e) {
+	let [t, n] = z(e.body.pipeThrough(new TextDecoderStream("utf8", { fatal: !0 }))).tee();
+	return {
+		stream: R(t),
+		response: L(n)
+	};
+}
+async function L(e) {
+	let t = [], n = e.getReader();
+	for (;;) {
+		let { done: e, value: r } = await n.read();
+		if (e) return O(B(t));
+		t.push(r);
+	}
+}
+function R(e) {
+	return P(this, arguments, function* () {
+		let t = e.getReader();
+		for (;;) {
+			let { value: e, done: n } = yield N(t.read());
+			if (n) break;
+			yield yield N(O(e));
+		}
+	});
+}
+function z(e) {
+	let t = e.getReader();
+	return new ReadableStream({ start(e) {
+		let n = "";
+		return r();
+		function r() {
+			return t.read().then(({ value: t, done: i }) => {
+				if (i) {
+					if (n.trim()) {
+						e.error(new f("Failed to parse stream"));
+						return;
+					}
+					e.close();
+					return;
+				}
+				n += t;
+				let a = n.match(F), o;
+				for (; a;) {
+					try {
+						o = JSON.parse(a[1]);
+					} catch {
+						e.error(new f(`Error parsing JSON response: "${a[1]}"`));
+						return;
+					}
+					e.enqueue(o), n = n.substring(a[0].length), a = n.match(F);
+				}
+				return r();
+			}).catch((e) => {
+				let t = e;
+				throw t.stack = e.stack, t = t.name === "AbortError" ? new g("Request aborted when reading from the stream") : new f("Error reading from the stream"), t;
+			});
+		}
+	} });
+}
+function B(e) {
+	let t = { promptFeedback: e[e.length - 1]?.promptFeedback };
+	for (let n of e) {
+		if (n.candidates) {
+			let e = 0;
+			for (let r of n.candidates) if (t.candidates ||= [], t.candidates[e] || (t.candidates[e] = { index: e }), t.candidates[e].citationMetadata = r.citationMetadata, t.candidates[e].groundingMetadata = r.groundingMetadata, t.candidates[e].finishReason = r.finishReason, t.candidates[e].finishMessage = r.finishMessage, t.candidates[e].safetyRatings = r.safetyRatings, r.content && r.content.parts) {
+				t.candidates[e].content || (t.candidates[e].content = {
+					role: r.content.role || "user",
+					parts: []
+				});
+				let n = {};
+				for (let i of r.content.parts) i.text && (n.text = i.text), i.functionCall && (n.functionCall = i.functionCall), i.executableCode && (n.executableCode = i.executableCode), i.codeExecutionResult && (n.codeExecutionResult = i.codeExecutionResult), Object.keys(n).length === 0 && (n.text = ""), t.candidates[e].content.parts.push(n);
+			}
+			e++;
+		}
+		n.usageMetadata && (t.usageMetadata = n.usageMetadata);
+	}
+	return t;
+}
+async function V(e, t, n, r) {
+	return I(await C(t, y.STREAM_GENERATE_CONTENT, e, !0, JSON.stringify(n), r));
+}
+async function H(e, t, n, r) {
+	return { response: O(await (await C(t, y.GENERATE_CONTENT, e, !1, JSON.stringify(n), r)).json()) };
+}
+function U(e) {
+	if (e != null) {
+		if (typeof e == "string") return {
+			role: "system",
+			parts: [{ text: e }]
+		};
+		if (e.text) return {
+			role: "system",
+			parts: [e]
+		};
+		if (e.parts) return e.role ? e : {
+			role: "system",
+			parts: e.parts
+		};
+	}
+}
+function W(e) {
+	let t = [];
+	if (typeof e == "string") t = [{ text: e }];
+	else for (let n of e) typeof n == "string" ? t.push({ text: n }) : t.push(n);
+	return G(t);
+}
+function G(e) {
+	let t = {
+		role: "user",
+		parts: []
+	}, n = {
+		role: "function",
+		parts: []
+	}, r = !1, i = !1;
+	for (let a of e) "functionResponse" in a ? (n.parts.push(a), i = !0) : (t.parts.push(a), r = !0);
+	if (r && i) throw new f("Within a single message, FunctionResponse cannot be mixed with other type of part in the request for sending chat message.");
+	if (!r && !i) throw new f("No content is provided for sending chat message.");
+	return r ? t : n;
+}
+function K(e, t) {
+	let n = {
+		model: t?.model,
+		generationConfig: t?.generationConfig,
+		safetySettings: t?.safetySettings,
+		tools: t?.tools,
+		toolConfig: t?.toolConfig,
+		systemInstruction: t?.systemInstruction,
+		cachedContent: t?.cachedContent?.name,
+		contents: []
+	}, r = e.generateContentRequest != null;
+	if (e.contents) {
+		if (r) throw new h("CountTokensRequest must have one of contents or generateContentRequest, not both.");
+		n.contents = e.contents;
+	} else if (r) n = Object.assign(Object.assign({}, n), e.generateContentRequest);
+	else {
+		let t = W(e);
+		n.contents = [t];
+	}
+	return { generateContentRequest: n };
+}
+function q(e) {
+	let t;
+	return t = e.contents ? e : { contents: [W(e)] }, e.systemInstruction && (t.systemInstruction = U(e.systemInstruction)), t;
+}
+function J(e) {
+	return typeof e == "string" || Array.isArray(e) ? { content: W(e) } : e;
+}
+var Y = [
+	"text",
+	"inlineData",
+	"functionCall",
+	"functionResponse",
+	"executableCode",
+	"codeExecutionResult"
+], ie = {
+	user: ["text", "inlineData"],
+	function: ["functionResponse"],
+	model: [
+		"text",
+		"functionCall",
+		"executableCode",
+		"codeExecutionResult"
+	],
+	system: ["text"]
+};
+function ae(e) {
+	let t = !1;
+	for (let n of e) {
+		let { role: e, parts: i } = n;
+		if (!t && e !== "user") throw new f(`First content should be with role 'user', got ${e}`);
+		if (!r.includes(e)) throw new f(`Each item should include role field. Got ${e} but valid roles are: ${JSON.stringify(r)}`);
+		if (!Array.isArray(i)) throw new f("Content should have 'parts' property with an array of Parts");
+		if (i.length === 0) throw new f("Each Content should have at least one part");
+		let a = {
+			text: 0,
+			inlineData: 0,
+			functionCall: 0,
+			functionResponse: 0,
+			fileData: 0,
+			executableCode: 0,
+			codeExecutionResult: 0
+		};
+		for (let e of i) for (let t of Y) t in e && (a[t] += 1);
+		let o = ie[e];
+		for (let t of Y) if (!o.includes(t) && a[t] > 0) throw new f(`Content with role '${e}' can't contain '${t}' part`);
+		t = !0;
+	}
+}
+function X(e) {
+	if (e.candidates === void 0 || e.candidates.length === 0) return !1;
+	let t = e.candidates[0]?.content;
+	if (t === void 0 || t.parts === void 0 || t.parts.length === 0) return !1;
+	for (let e of t.parts) if (e === void 0 || Object.keys(e).length === 0 || e.text !== void 0 && e.text === "") return !1;
+	return !0;
+}
+var Z = "SILENT_ERROR", Q = class {
+	constructor(e, t, n, r = {}) {
+		this.model = t, this.params = n, this._requestOptions = r, this._history = [], this._sendPromise = Promise.resolve(), this._apiKey = e, n?.history && (ae(n.history), this._history = n.history);
+	}
+	async getHistory() {
+		return await this._sendPromise, this._history;
+	}
+	async sendMessage(e, t = {}) {
+		await this._sendPromise;
+		let n = W(e), r = {
+			safetySettings: this.params?.safetySettings,
+			generationConfig: this.params?.generationConfig,
+			tools: this.params?.tools,
+			toolConfig: this.params?.toolConfig,
+			systemInstruction: this.params?.systemInstruction,
+			cachedContent: this.params?.cachedContent,
+			contents: [...this._history, n]
+		}, i = Object.assign(Object.assign({}, this._requestOptions), t), a;
+		return this._sendPromise = this._sendPromise.then(() => H(this._apiKey, this.model, r, i)).then((e) => {
+			if (X(e.response)) {
+				this._history.push(n);
+				let t = Object.assign({
+					parts: [],
+					role: "model"
+				}, e.response.candidates?.[0].content);
+				this._history.push(t);
+			} else {
+				let t = M(e.response);
+				t && console.warn(`sendMessage() was unsuccessful. ${t}. Inspect response object for details.`);
+			}
+			a = e;
+		}).catch((e) => {
+			throw this._sendPromise = Promise.resolve(), e;
+		}), await this._sendPromise, a;
+	}
+	async sendMessageStream(e, t = {}) {
+		await this._sendPromise;
+		let n = W(e), r = {
+			safetySettings: this.params?.safetySettings,
+			generationConfig: this.params?.generationConfig,
+			tools: this.params?.tools,
+			toolConfig: this.params?.toolConfig,
+			systemInstruction: this.params?.systemInstruction,
+			cachedContent: this.params?.cachedContent,
+			contents: [...this._history, n]
+		}, i = Object.assign(Object.assign({}, this._requestOptions), t), a = V(this._apiKey, this.model, r, i);
+		return this._sendPromise = this._sendPromise.then(() => a).catch((e) => {
+			throw Error(Z);
+		}).then((e) => e.response).then((e) => {
+			if (X(e)) {
+				this._history.push(n);
+				let t = Object.assign({}, e.candidates[0].content);
+				t.role ||= "model", this._history.push(t);
+			} else {
+				let t = M(e);
+				t && console.warn(`sendMessageStream() was unsuccessful. ${t}. Inspect response object for details.`);
+			}
+		}).catch((e) => {
+			e.message !== Z && console.error(e);
+		}), a;
+	}
+};
+async function oe(e, t, n, r) {
+	return (await C(t, y.COUNT_TOKENS, e, !1, JSON.stringify(n), r)).json();
+}
+async function se(e, t, n, r) {
+	return (await C(t, y.EMBED_CONTENT, e, !1, JSON.stringify(n), r)).json();
+}
+async function ce(e, t, n, r) {
+	let i = n.requests.map((e) => Object.assign(Object.assign({}, e), { model: t }));
+	return (await C(t, y.BATCH_EMBED_CONTENTS, e, !1, JSON.stringify({ requests: i }), r)).json();
+}
+var $ = class {
+	constructor(e, t, n = {}) {
+		this.apiKey = e, this._requestOptions = n, t.model.includes("/") ? this.model = t.model : this.model = `models/${t.model}`, this.generationConfig = t.generationConfig || {}, this.safetySettings = t.safetySettings || [], this.tools = t.tools, this.toolConfig = t.toolConfig, this.systemInstruction = U(t.systemInstruction), this.cachedContent = t.cachedContent;
+	}
+	async generateContent(e, t = {}) {
+		let n = q(e), r = Object.assign(Object.assign({}, this._requestOptions), t);
+		return H(this.apiKey, this.model, Object.assign({
+			generationConfig: this.generationConfig,
+			safetySettings: this.safetySettings,
+			tools: this.tools,
+			toolConfig: this.toolConfig,
+			systemInstruction: this.systemInstruction,
+			cachedContent: this.cachedContent?.name
+		}, n), r);
+	}
+	async generateContentStream(e, t = {}) {
+		let n = q(e), r = Object.assign(Object.assign({}, this._requestOptions), t);
+		return V(this.apiKey, this.model, Object.assign({
+			generationConfig: this.generationConfig,
+			safetySettings: this.safetySettings,
+			tools: this.tools,
+			toolConfig: this.toolConfig,
+			systemInstruction: this.systemInstruction,
+			cachedContent: this.cachedContent?.name
+		}, n), r);
+	}
+	startChat(e) {
+		return new Q(this.apiKey, this.model, Object.assign({
+			generationConfig: this.generationConfig,
+			safetySettings: this.safetySettings,
+			tools: this.tools,
+			toolConfig: this.toolConfig,
+			systemInstruction: this.systemInstruction,
+			cachedContent: this.cachedContent?.name
+		}, e), this._requestOptions);
+	}
+	async countTokens(e, t = {}) {
+		let n = K(e, {
+			model: this.model,
+			generationConfig: this.generationConfig,
+			safetySettings: this.safetySettings,
+			tools: this.tools,
+			toolConfig: this.toolConfig,
+			systemInstruction: this.systemInstruction,
+			cachedContent: this.cachedContent
+		}), r = Object.assign(Object.assign({}, this._requestOptions), t);
+		return oe(this.apiKey, this.model, n, r);
+	}
+	async embedContent(e, t = {}) {
+		let n = J(e), r = Object.assign(Object.assign({}, this._requestOptions), t);
+		return se(this.apiKey, this.model, n, r);
+	}
+	async batchEmbedContents(e, t = {}) {
+		let n = Object.assign(Object.assign({}, this._requestOptions), t);
+		return ce(this.apiKey, this.model, e, n);
+	}
+}, le = class {
+	constructor(e) {
+		this.apiKey = e;
+	}
+	getGenerativeModel(e, t) {
+		if (!e.model) throw new f("Must provide a model name. Example: genai.getGenerativeModel({ model: 'my-model-name' })");
+		return new $(this.apiKey, e, t);
+	}
+	getGenerativeModelFromCachedContent(e, t, n) {
+		if (!e.name) throw new h("Cached content must contain a `name` field.");
+		if (!e.model) throw new h("Cached content must contain a `model` field.");
+		for (let n of ["model", "systemInstruction"]) if (t?.[n] && e[n] && t?.[n] !== e[n]) {
+			if (n === "model" && (t.model.startsWith("models/") ? t.model.replace("models/", "") : t.model) === (e.model.startsWith("models/") ? e.model.replace("models/", "") : e.model)) continue;
+			throw new h(`Different value for "${n}" specified in modelParams (${t[n]}) and cachedContent (${e[n]})`);
+		}
+		let r = Object.assign(Object.assign({}, t), {
+			model: e.model,
+			tools: e.tools,
+			toolConfig: e.toolConfig,
+			systemInstruction: e.systemInstruction,
+			cachedContent: e
+		});
+		return new $(this.apiKey, r, n);
+	}
+};
+//#endregion
+export { s as BlockReason, Q as ChatSession, d as DynamicRetrievalMode, t as ExecutableCodeLanguage, c as FinishReason, u as FunctionCallingMode, $ as GenerativeModel, le as GoogleGenerativeAI, g as GoogleGenerativeAIAbortError, f as GoogleGenerativeAIError, m as GoogleGenerativeAIFetchError, h as GoogleGenerativeAIRequestInputError, p as GoogleGenerativeAIResponseError, a as HarmBlockThreshold, i as HarmCategory, o as HarmProbability, n as Outcome, r as POSSIBLE_ROLES, e as SchemaType, l as TaskType };

package/dist/adapters/express/index.cjs

@@ -1,11 +1,3 @@
-"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const w=require("express"),H=require("cors"),c=require("../mcp-transport-DCiSGtp1.cjs"),n=require("../utils.cjs");function $(C){const{cubes:v,drizzle:x,schema:I,extractSecurityContext:y,engineType:P,cors:S,basePath:f="/cubejs-api/v1",jsonLimit:Q="10mb",cache:O,mcp:E={enabled:!0},agent:h}=C;if(!v||v.length===0)throw new Error("At least one cube must be provided in the cubes array");const d=w.Router();S&&d.use(H(S)),d.use(w.json({limit:Q})),d.use(w.urlencoded({extended:!0,limit:Q}));const u=new c.SemanticLayerCompiler({drizzle:x,schema:I,engineType:P,cache:O});if(v.forEach(r=>{u.registerCube(r)}),d.post(`${f}/load`,async(r,t)=>{try{const e=r.body.query||r.body,o=await y(r,t),s=u.validateQuery(e);if(!s.isValid)return t.status(400).json(n.formatErrorResponse(`Query validation failed: ${s.errors.join(", ")}`,400));const a=r.headers["x-cache-control"]==="no-cache",i=await u.executeMultiCubeQuery(e,o,{skipCache:a});t.json(n.formatCubeResponse(e,i,u))}catch(e){console.error("Query execution error:",e),t.status(500).json(n.formatErrorResponse(e instanceof Error?e.message:"Query execution failed",500))}}),d.get(`${f}/load`,async(r,t)=>{try{const e=r.query.query;if(!e)return t.status(400).json(n.formatErrorResponse("Query parameter is required",400));let o;try{o=JSON.parse(e)}catch{return t.status(400).json(n.formatErrorResponse("Invalid JSON in query parameter",400))}const s=await y(r,t),a=u.validateQuery(o);if(!a.isValid)return t.status(400).json(n.formatErrorResponse(`Query validation failed: ${a.errors.join(", ")}`,400));const i=r.headers["x-cache-control"]==="no-cache",m=await u.executeMultiCubeQuery(o,s,{skipCache:i});t.json(n.formatCubeResponse(o,m,u))}catch(e){console.error("Query execution error:",e),t.status(500).json(n.formatErrorResponse(e instanceof Error?e.message:"Query execution failed",500))}}),d.post(`${f}/batch`,async(r,t)=>{try{const{queries:e}=r.body;if(!e||!Array.isArray(e))return t.status(400).json(n.formatErrorResponse('Request body must contain a "queries" array',400));if(e.length===0)return t.status(400).json(n.formatErrorResponse("Queries array cannot be empty",400));const o=await y(r,t),s=r.headers["x-cache-control"]==="no-cache",a=await n.handleBatchRequest(e,o,u,{skipCache:s});t.json(a)}catch(e){console.error("Batch execution error:",e),t.status(500).json(n.formatErrorResponse(e instanceof Error?e.message:"Batch execution failed",500))}}),d.get(`${f}/meta`,(r,t)=>{try{const e=u.getMetadata();t.json(n.formatMetaResponse(e))}catch(e){console.error("Metadata error:",e),t.status(500).json(n.formatErrorResponse(e instanceof Error?e.message:"Failed to fetch metadata",500))}}),d.post(`${f}/sql`,async(r,t)=>{try{const e=r.body,o=await y(r,t),s=u.validateQuery(e);if(!s.isValid)return t.status(400).json(n.formatErrorResponse(`Query validation failed: ${s.errors.join(", ")}`,400));const a=e.measures?.[0]||e.dimensions?.[0];if(!a)return t.status(400).json(n.formatErrorResponse("No measures or dimensions specified",400));const i=a.split(".")[0],m=await u.generateSQL(i,e,o);t.json(n.formatSqlResponse(e,m))}catch(e){console.error("SQL generation error:",e),t.status(500).json(n.formatErrorResponse(e instanceof Error?e.message:"SQL generation failed",500))}}),d.get(`${f}/sql`,async(r,t)=>{try{const e=r.query.query;if(!e)return t.status(400).json(n.formatErrorResponse("Query parameter is required",400));const o=JSON.parse(e),s=await y(r,t),a=u.validateQuery(o);if(!a.isValid)return t.status(400).json(n.formatErrorResponse(`Query validation failed: ${a.errors.join(", ")}`,400));const i=o.measures?.[0]||o.dimensions?.[0];if(!i)return t.status(400).json(n.formatErrorResponse("No measures or dimensions specified",400));const m=i.split(".")[0],R=await u.generateSQL(m,o,s);t.json(n.formatSqlResponse(o,R))}catch(e){console.error("SQL generation error:",e),t.status(500).json(n.formatErrorResponse(e instanceof Error?e.message:"SQL generation failed",500))}}),d.post(`${f}/dry-run`,async(r,t)=>{try{const e=r.body.query||r.body,o=await y(r,t),s=await n.handleDryRun(e,o,u);t.json(s)}catch(e){console.error("Dry-run error:",e),t.status(400).json({error:e instanceof Error?e.message:"Dry-run validation failed",valid:!1})}}),d.get(`${f}/dry-run`,async(r,t)=>{try{const e=r.query.query;if(!e)return t.status(400).json({error:"Query parameter is required",valid:!1});const o=JSON.parse(e),s=await y(r,t),a=await n.handleDryRun(o,s,u);t.json(a)}catch(e){console.error("Dry-run error:",e),t.status(400).json({error:e instanceof Error?e.message:"Dry-run validation failed",valid:!1})}}),d.post(`${f}/explain`,async(r,t)=>{try{const e=r.body.query||r.body,o=r.body.options||{},s=await y(r,t),a=u.validateQuery(e);if(!a.isValid)return t.status(400).json({error:`Query validation failed: ${a.errors.join(", ")}`});const i=await u.explainQuery(e,s,o);t.json(i)}catch(e){console.error("Explain error:",e),t.status(500).json({error:e instanceof Error?e.message:"Explain query failed"})}}),h&&d.post(`${f}/agent/chat`,async(r,t)=>{try{const{handleAgentChat:e}=await Promise.resolve().then(()=>require("../handler-LMRPeTNJ.cjs")),{message:o,sessionId:s,history:a}=r.body;if(!o||typeof o!="string")return t.status(400).json({error:"message is required and must be a string"});let i=(h.apiKey||"").trim();if(h.allowClientApiKey){const p=r.headers["x-agent-api-key"];p&&(i=p.trim())}if(!i)return t.status(401).json({error:"No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header."});const m=h.allowClientApiKey?r.headers["x-agent-provider"]:void 0,R=h.allowClientApiKey?r.headers["x-agent-model"]:void 0,l=h.allowClientApiKey?r.headers["x-agent-provider-endpoint"]:void 0,g=await y(r,t),b=h.buildSystemContext?.(g);t.writeHead(200,{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive"});try{const p=e({message:o,sessionId:s,history:a,semanticLayer:u,securityContext:g,agentConfig:h,apiKey:i,systemContext:b,providerOverride:m,modelOverride:R,baseURLOverride:l});for await(const j of p)t.write(`data: ${JSON.stringify(j)}
+Object.defineProperty(exports,Symbol.toStringTag,{value:`Module`});const e=require(`../utils-CwJplXR5.cjs`),t=require(`../mcp-transport-Dpp6hdZe.cjs`);let n=require(`express`);n=e.g(n);let r=require(`cors`);r=e.g(r);function i(i){let{cubes:a,drizzle:o,schema:s,extractSecurityContext:c,engineType:l,cors:u,basePath:d=`/cubejs-api/v1`,jsonLimit:f=`10mb`,cache:p,mcp:m={enabled:!0},agent:h}=i;if(!a||a.length===0)throw Error(`At least one cube must be provided in the cubes array`);let g=(0,n.Router)();u&&g.use((0,r.default)(u)),g.use(n.default.json({limit:f})),g.use(n.default.urlencoded({extended:!0,limit:f}));let _=new t.h({drizzle:o,schema:s,engineType:l,cache:p});if(a.forEach(e=>{_.registerCube(e)}),g.post(`${d}/load`,async(t,n)=>{try{let r=t.body.query||t.body,i=await c(t,n),a=_.validateQuery(r);if(!a.isValid)return n.status(400).json(e.i(`Query validation failed: ${a.errors.join(`, `)}`,400));let o=t.headers[`x-cache-control`]===`no-cache`,s=await _.executeMultiCubeQuery(r,i,{skipCache:o});n.json(e.r(r,s,_))}catch(t){console.error(`Query execution error:`,t),n.status(500).json(e.i(t instanceof Error?t.message:`Query execution failed`,500))}}),g.get(`${d}/load`,async(t,n)=>{try{let r=t.query.query;if(!r)return n.status(400).json(e.i(`Query parameter is required`,400));let i;try{i=JSON.parse(r)}catch{return n.status(400).json(e.i(`Invalid JSON in query parameter`,400))}let a=await c(t,n),o=_.validateQuery(i);if(!o.isValid)return n.status(400).json(e.i(`Query validation failed: ${o.errors.join(`, `)}`,400));let s=t.headers[`x-cache-control`]===`no-cache`,l=await _.executeMultiCubeQuery(i,a,{skipCache:s});n.json(e.r(i,l,_))}catch(t){console.error(`Query execution error:`,t),n.status(500).json(e.i(t instanceof Error?t.message:`Query execution failed`,500))}}),g.post(`${d}/batch`,async(t,n)=>{try{let{queries:r}=t.body;if(!r||!Array.isArray(r))return n.status(400).json(e.i(`Request body must contain a "queries" array`,400));if(r.length===0)return n.status(400).json(e.i(`Queries array cannot be empty`,400));let i=await e.u(r,await c(t,n),_,{skipCache:t.headers[`x-cache-control`]===`no-cache`});n.json(i)}catch(t){console.error(`Batch execution error:`,t),n.status(500).json(e.i(t instanceof Error?t.message:`Batch execution failed`,500))}}),g.get(`${d}/meta`,(t,n)=>{try{let t=_.getMetadata();n.json(e.a(t))}catch(t){console.error(`Metadata error:`,t),n.status(500).json(e.i(t instanceof Error?t.message:`Failed to fetch metadata`,500))}}),g.post(`${d}/sql`,async(t,n)=>{try{let r=t.body,i=await c(t,n),a=_.validateQuery(r);if(!a.isValid)return n.status(400).json(e.i(`Query validation failed: ${a.errors.join(`, `)}`,400));let o=r.measures?.[0]||r.dimensions?.[0];if(!o)return n.status(400).json(e.i(`No measures or dimensions specified`,400));let s=o.split(`.`)[0],l=await _.generateSQL(s,r,i);n.json(e.o(r,l))}catch(t){console.error(`SQL generation error:`,t),n.status(500).json(e.i(t instanceof Error?t.message:`SQL generation failed`,500))}}),g.get(`${d}/sql`,async(t,n)=>{try{let r=t.query.query;if(!r)return n.status(400).json(e.i(`Query parameter is required`,400));let i=JSON.parse(r),a=await c(t,n),o=_.validateQuery(i);if(!o.isValid)return n.status(400).json(e.i(`Query validation failed: ${o.errors.join(`, `)}`,400));let s=i.measures?.[0]||i.dimensions?.[0];if(!s)return n.status(400).json(e.i(`No measures or dimensions specified`,400));let l=s.split(`.`)[0],u=await _.generateSQL(l,i,a);n.json(e.o(i,u))}catch(t){console.error(`SQL generation error:`,t),n.status(500).json(e.i(t instanceof Error?t.message:`SQL generation failed`,500))}}),g.post(`${d}/dry-run`,async(t,n)=>{try{let r=await e.f(t.body.query||t.body,await c(t,n),_);n.json(r)}catch(e){console.error(`Dry-run error:`,e),n.status(400).json({error:e instanceof Error?e.message:`Dry-run validation failed`,valid:!1})}}),g.get(`${d}/dry-run`,async(t,n)=>{try{let r=t.query.query;if(!r)return n.status(400).json({error:`Query parameter is required`,valid:!1});let i=await e.f(JSON.parse(r),await c(t,n),_);n.json(i)}catch(e){console.error(`Dry-run error:`,e),n.status(400).json({error:e instanceof Error?e.message:`Dry-run validation failed`,valid:!1})}}),g.post(`${d}/explain`,async(e,t)=>{try{let n=e.body.query||e.body,r=e.body.options||{},i=await c(e,t),a=_.validateQuery(n);if(!a.isValid)return t.status(400).json({error:`Query validation failed: ${a.errors.join(`, `)}`});let o=await _.explainQuery(n,i,r);t.json(o)}catch(e){console.error(`Explain error:`,e),t.status(500).json({error:e instanceof Error?e.message:`Explain query failed`})}}),h&&g.post(`${d}/agent/chat`,async(e,t)=>{try{let{handleAgentChat:n}=await Promise.resolve().then(()=>require(`../handler-B8vuFQYP.cjs`)),{message:r,sessionId:i,history:a}=e.body;if(!r||typeof r!=`string`)return t.status(400).json({error:`message is required and must be a string`});let o=(h.apiKey||``).trim();if(h.allowClientApiKey){let t=e.headers[`x-agent-api-key`];t&&(o=t.trim())}if(!o)return t.status(401).json({error:`No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header.`});let s=h.allowClientApiKey?e.headers[`x-agent-provider`]:void 0,l=h.allowClientApiKey?e.headers[`x-agent-model`]:void 0,u=h.allowClientApiKey?e.headers[`x-agent-provider-endpoint`]:void 0,d=await c(e,t),f=h.buildSystemContext?.(d);t.writeHead(200,{"Content-Type":`text/event-stream`,"Cache-Control":`no-cache`,Connection:`keep-alive`});try{let e=n({message:r,sessionId:i,history:a,semanticLayer:_,securityContext:d,agentConfig:h,apiKey:o,systemContext:f,providerOverride:s,modelOverride:l,baseURLOverride:u});for await(let n of e)t.write(`data: ${JSON.stringify(n)}\n\n`)}catch(e){let n={type:`error`,data:{message:e instanceof Error?e.message:`Stream failed`}};t.write(`data: ${JSON.stringify(n)}\n\n`)}finally{t.end()}}catch(e){console.error(`Agent chat error:`,e),t.headersSent||t.status(500).json({error:e instanceof Error?e.message:`Agent chat failed`})}}),m.enabled!==!1){let e=m.basePath??`/mcp`;g.post(`${e}`,async(e,n)=>{let r=t.p(e.headers.origin,m.allowedOrigins?{allowedOrigins:m.allowedOrigins}:{});if(!r.valid)return n.status(403).json(t.n(null,-32600,r.reason));let i=e.headers.accept;if(!t.f(i))return n.status(400).json(t.n(null,-32600,`Accept header must include both application/json and text/event-stream`));let a=t.c(e.headers);if(!a.ok)return n.status(426).json({error:`Unsupported MCP protocol version`,supported:a.supported});let o=t.l(e.body);if(!o)return n.status(400).json(t.n(null,-32600,`Invalid JSON-RPC 2.0 request`));let s=t.m(i),l=o.method===`initialize`;try{let r=await t.i(o.method,o.params,{semanticLayer:_,extractSecurityContext:c,rawRequest:e,rawResponse:n,negotiatedProtocol:a.negotiated});if(t.s(o))return n.status(202).end();let i=l&&r&&typeof r==`object`&&`sessionId`in r?r.sessionId:void 0;i&&n.setHeader(t.t,i);let u=t.r(o.id??null,r);if(s){let e=t.u();return n.status(200),n.setHeader(`Content-Type`,`text/event-stream`),n.setHeader(`Cache-Control`,`no-cache`),n.setHeader(`Connection`,`keep-alive`),n.write(`id: ${e}\n\n`),n.write(t.d(u,e)),n.end()}return n.json(u)}catch(e){if(t.s(o))return console.error(`MCP notification processing error:`,e),n.status(202).end();console.error(`MCP RPC error:`,e);let r=e?.code??-32603,i=e?.data,a=e.message||`MCP request failed`,c=t.n(o.id??null,r,a,i);if(s){let e=t.u();return n.status(200),n.setHeader(`Content-Type`,`text/event-stream`),n.setHeader(`Cache-Control`,`no-cache`),n.setHeader(`Connection`,`keep-alive`),n.write(`id: ${e}\n\n`),n.write(t.d(c,e)),n.end()}return n.status(200).json(c)}}),g.get(`${e}`,async(e,n)=>{let r=t.u();n.status(200),n.setHeader(`Content-Type`,`text/event-stream`),n.setHeader(`Cache-Control`,`no-cache`),n.setHeader(`Connection`,`keep-alive`),n.write(t.d({jsonrpc:`2.0`,method:`mcp/ready`,params:{protocol:`streamable-http`}},r,15e3));let i=setInterval(()=>{n.write(`: keep-alive
 
-`)}catch(p){const j={type:"error",data:{message:p instanceof Error?p.message:"Stream failed"}};t.write(`data: ${JSON.stringify(j)}
-
-`)}finally{t.end()}}catch(e){console.error("Agent chat error:",e),t.headersSent||t.status(500).json({error:e instanceof Error?e.message:"Agent chat failed"})}}),E.enabled!==!1){const r=E.basePath??"/mcp";d.post(`${r}`,async(t,e)=>{const o=c.validateOriginHeader(t.headers.origin,E.allowedOrigins?{allowedOrigins:E.allowedOrigins}:{});if(!o.valid)return e.status(403).json(c.buildJsonRpcError(null,-32600,o.reason));const s=t.headers.accept;if(!c.validateAcceptHeader(s))return e.status(400).json(c.buildJsonRpcError(null,-32600,"Accept header must include both application/json and text/event-stream"));const a=c.negotiateProtocol(t.headers);if(!a.ok)return e.status(426).json({error:"Unsupported MCP protocol version",supported:a.supported});const i=c.parseJsonRpc(t.body);if(!i)return e.status(400).json(c.buildJsonRpcError(null,-32600,"Invalid JSON-RPC 2.0 request"));const m=c.wantsEventStream(s),R=i.method==="initialize";try{const l=await c.dispatchMcpMethod(i.method,i.params,{semanticLayer:u,extractSecurityContext:y,rawRequest:t,rawResponse:e,negotiatedProtocol:a.negotiated});if(c.isNotification(i))return e.status(202).end();const g=R&&l&&typeof l=="object"&&"sessionId"in l?l.sessionId:void 0;g&&e.setHeader(c.MCP_SESSION_ID_HEADER,g);const b=c.buildJsonRpcResult(i.id??null,l);if(m){const p=c.primeEventId();return e.status(200),e.setHeader("Content-Type","text/event-stream"),e.setHeader("Cache-Control","no-cache"),e.setHeader("Connection","keep-alive"),e.write(`id: ${p}
-
-`),e.write(c.serializeSseEvent(b,p)),e.end()}return e.json(b)}catch(l){if(c.isNotification(i))return console.error("MCP notification processing error:",l),e.status(202).end();console.error("MCP RPC error:",l);const g=l?.code??-32603,b=l?.data,p=l.message||"MCP request failed",j=c.buildJsonRpcError(i.id??null,g,p,b);if(m){const q=c.primeEventId();return e.status(200),e.setHeader("Content-Type","text/event-stream"),e.setHeader("Cache-Control","no-cache"),e.setHeader("Connection","keep-alive"),e.write(`id: ${q}
-
-`),e.write(c.serializeSseEvent(j,q)),e.end()}return e.status(200).json(j)}}),d.get(`${r}`,async(t,e)=>{const o=c.primeEventId();e.status(200),e.setHeader("Content-Type","text/event-stream"),e.setHeader("Cache-Control","no-cache"),e.setHeader("Connection","keep-alive"),e.write(c.serializeSseEvent({jsonrpc:"2.0",method:"mcp/ready",params:{protocol:"streamable-http"}},o,15e3));const s=setInterval(()=>{e.write(`: keep-alive
-
-`)},15e3);t.on("close",()=>{clearInterval(s)})}),d.delete(`${r}`,(t,e)=>e.status(405).json({error:"Session termination not supported"}))}return d.use((r,t,e,o)=>{console.error("Express adapter error:",r),e.headersSent||e.status(500).json(n.formatErrorResponse(r,500))}),d}function A(C,v){const x=$(v);return C.use("/",x),C}function M(C){const v=w();return A(v,C)}exports.createCubeApp=M;exports.createCubeRouter=$;exports.mountCubeRoutes=A;
+`)},15e3);e.on(`close`,()=>{clearInterval(i)})}),g.delete(`${e}`,(e,t)=>t.status(405).json({error:`Session termination not supported`}))}return g.use((t,n,r,i)=>{console.error(`Express adapter error:`,t),r.headersSent||r.status(500).json(e.i(t,500))}),g}function a(e,t){let n=i(t);return e.use(`/`,n),e}function o(e){return a((0,n.default)(),e)}exports.createCubeApp=o,exports.createCubeRouter=i,exports.mountCubeRoutes=a;