drizzle-cube 0.4.29 → 0.4.30

package/dist/server/google-BXwMolCu.js

@@ -0,0 +1,154 @@
+//#region src/server/agent/providers/google.ts
+function e(t) {
+	let n = {};
+	for (let [r, i] of Object.entries(t)) if (r === "type" && typeof i == "string") n[r] = i.toUpperCase();
+	else if (r === "properties" && typeof i == "object" && i) {
+		let t = {};
+		for (let [n, r] of Object.entries(i)) typeof r == "object" && r ? t[n] = e(r) : t[n] = r;
+		n[r] = t;
+	} else r === "items" && typeof i == "object" && i ? n[r] = e(i) : n[r] = i;
+	return n;
+}
+var t = class {
+	apiKey;
+	sdk;
+	initialized = !1;
+	constructor(e) {
+		this.apiKey = e;
+	}
+	async ensureSDK() {
+		if (!this.initialized) {
+			try {
+				this.sdk = await import(
+					/* webpackIgnore: true */
+					"./dist-CMWZC51B.js"
+);
+			} catch {
+				throw Error("@google/generative-ai is required for the Google provider. Install it with: npm install @google/generative-ai");
+			}
+			this.initialized = !0;
+		}
+	}
+	async createStream(e) {
+		await this.ensureSDK();
+		let { GoogleGenerativeAI: t } = this.sdk, n = new t(this.apiKey).getGenerativeModel({
+			model: e.model,
+			systemInstruction: e.system,
+			generationConfig: { maxOutputTokens: e.maxTokens }
+		}), { messages: r } = this.formatMessages(e.messages, e.system), i = this.formatTools(e.tools);
+		return (await n.generateContentStream({
+			contents: r,
+			tools: i.length > 0 ? [{ functionDeclarations: i }] : void 0
+		})).stream;
+	}
+	async *parseStreamEvents(e) {
+		let t = "stop", n = !1;
+		for await (let r of e) {
+			let e = r;
+			e.usageMetadata && (yield {
+				type: "message_meta",
+				inputTokens: e.usageMetadata.promptTokenCount,
+				outputTokens: e.usageMetadata.candidatesTokenCount,
+				stopReason: ""
+			});
+			let i = e.candidates?.[0];
+			if (!i) continue;
+			i.finishReason && (t = i.finishReason === "STOP" ? "stop" : i.finishReason === "MAX_TOKENS" ? "max_tokens" : i.finishReason);
+			let a = i.content?.parts;
+			if (a) {
+				for (let e of a) if (e.text && !e.thought && (yield {
+					type: "text_delta",
+					text: e.text
+				}), e.functionCall) {
+					n = !0;
+					let t = `gemini-tc-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`, r = e.thoughtSignature ? { thoughtSignature: e.thoughtSignature } : void 0;
+					yield {
+						type: "tool_use_start",
+						id: t,
+						name: e.functionCall.name,
+						...r ? { metadata: r } : {}
+					}, yield {
+						type: "tool_input_delta",
+						json: JSON.stringify(e.functionCall.args || {})
+					}, yield { type: "tool_use_end" };
+				}
+			}
+		}
+		yield {
+			type: "message_meta",
+			stopReason: n ? "tool_use" : t
+		};
+	}
+	formatTools(t) {
+		return t.map((t) => ({
+			name: t.name,
+			description: t.description,
+			parameters: e(t.parameters)
+		}));
+	}
+	formatMessages(e, t) {
+		let n = [];
+		for (let t of e) if (t.role === "tool_result") {
+			let e = t._toolResults;
+			if (e && e.length > 0) n.push({
+				role: "user",
+				parts: e.map((e) => ({ functionResponse: {
+					name: e.toolName || e.toolUseId,
+					response: {
+						content: e.content,
+						isError: e.isError || !1
+					}
+				} }))
+			});
+			else {
+				let e = typeof t.content == "string" ? t.content : JSON.stringify(t.content);
+				e && n.push({
+					role: "user",
+					parts: [{ text: e }]
+				});
+			}
+		} else if (t.role === "user") {
+			let e = typeof t.content == "string" ? t.content : JSON.stringify(t.content);
+			n.push({
+				role: "user",
+				parts: [{ text: e }]
+			});
+		} else if (t.role === "assistant") if (typeof t.content == "string") n.push({
+			role: "model",
+			parts: [{ text: t.content }]
+		});
+		else {
+			let e = t.content, r = [];
+			for (let t of e) if (t.type === "text" && t.text) r.push({ text: t.text });
+			else if (t.type === "tool_use") {
+				let e = { functionCall: {
+					name: t.name,
+					args: t.input || {}
+				} };
+				t.metadata?.thoughtSignature && (e.thoughtSignature = t.metadata.thoughtSignature), r.push(e);
+			}
+			r.length > 0 && n.push({
+				role: "model",
+				parts: r
+			});
+		}
+		return { messages: n };
+	}
+	formatToolResults(e) {
+		return {
+			role: "tool_result",
+			content: e.map((e) => `${e.toolName || e.toolUseId}: ${e.content}`).join("\n"),
+			_toolResults: e
+		};
+	}
+	shouldContinue(e) {
+		return e === "tool_use";
+	}
+	formatError(e) {
+		if (!e || !(e instanceof Error)) return "Something went wrong. Please try again.";
+		let t = e.message || "", n = e;
+		return n.status === 429 ? "Too many requests. Please wait a moment and try again." : n.status === 403 || n.status === 401 ? "Authentication failed. Please check your API key configuration." : n.status === 503 || n.status === 500 ? "The AI service is temporarily unavailable. Please try again in a moment." : t.includes("SAFETY") ? "The request was blocked by safety filters. Please rephrase your request." : t.includes("RECITATION") ? "The response was blocked due to recitation concerns. Please try a different query." : t.startsWith("{") || t.startsWith("[") ? "The AI service encountered an error. Please try again." : t;
+	}
+};
+//#endregion
+export { t as GoogleProvider };

package/dist/server/google-DzQWXFwF.cjs

@@ -0,0 +1,2 @@
+function e(t){let n={};for(let[r,i]of Object.entries(t))if(r===`type`&&typeof i==`string`)n[r]=i.toUpperCase();else if(r===`properties`&&typeof i==`object`&&i){let t={};for(let[n,r]of Object.entries(i))typeof r==`object`&&r?t[n]=e(r):t[n]=r;n[r]=t}else r===`items`&&typeof i==`object`&&i?n[r]=e(i):n[r]=i;return n}var t=class{apiKey;sdk;initialized=!1;constructor(e){this.apiKey=e}async ensureSDK(){if(!this.initialized){try{this.sdk=await Promise.resolve().then(()=>require(`./dist-DxegvyZF.cjs`))}catch{throw Error(`@google/generative-ai is required for the Google provider. Install it with: npm install @google/generative-ai`)}this.initialized=!0}}async createStream(e){await this.ensureSDK();let{GoogleGenerativeAI:t}=this.sdk,n=new t(this.apiKey).getGenerativeModel({model:e.model,systemInstruction:e.system,generationConfig:{maxOutputTokens:e.maxTokens}}),{messages:r}=this.formatMessages(e.messages,e.system),i=this.formatTools(e.tools);return(await n.generateContentStream({contents:r,tools:i.length>0?[{functionDeclarations:i}]:void 0})).stream}async*parseStreamEvents(e){let t=`stop`,n=!1;for await(let r of e){let e=r;e.usageMetadata&&(yield{type:`message_meta`,inputTokens:e.usageMetadata.promptTokenCount,outputTokens:e.usageMetadata.candidatesTokenCount,stopReason:``});let i=e.candidates?.[0];if(!i)continue;i.finishReason&&(t=i.finishReason===`STOP`?`stop`:i.finishReason===`MAX_TOKENS`?`max_tokens`:i.finishReason);let a=i.content?.parts;if(a){for(let e of a)if(e.text&&!e.thought&&(yield{type:`text_delta`,text:e.text}),e.functionCall){n=!0;let t=`gemini-tc-${Date.now()}-${Math.random().toString(36).slice(2,7)}`,r=e.thoughtSignature?{thoughtSignature:e.thoughtSignature}:void 0;yield{type:`tool_use_start`,id:t,name:e.functionCall.name,...r?{metadata:r}:{}},yield{type:`tool_input_delta`,json:JSON.stringify(e.functionCall.args||{})},yield{type:`tool_use_end`}}}}yield{type:`message_meta`,stopReason:n?`tool_use`:t}}formatTools(t){return t.map(t=>({name:t.name,description:t.description,parameters:e(t.parameters)}))}formatMessages(e,t){let n=[];for(let t of e)if(t.role===`tool_result`){let e=t._toolResults;if(e&&e.length>0)n.push({role:`user`,parts:e.map(e=>({functionResponse:{name:e.toolName||e.toolUseId,response:{content:e.content,isError:e.isError||!1}}}))});else{let e=typeof t.content==`string`?t.content:JSON.stringify(t.content);e&&n.push({role:`user`,parts:[{text:e}]})}}else if(t.role===`user`){let e=typeof t.content==`string`?t.content:JSON.stringify(t.content);n.push({role:`user`,parts:[{text:e}]})}else if(t.role===`assistant`)if(typeof t.content==`string`)n.push({role:`model`,parts:[{text:t.content}]});else{let e=t.content,r=[];for(let t of e)if(t.type===`text`&&t.text)r.push({text:t.text});else if(t.type===`tool_use`){let e={functionCall:{name:t.name,args:t.input||{}}};t.metadata?.thoughtSignature&&(e.thoughtSignature=t.metadata.thoughtSignature),r.push(e)}r.length>0&&n.push({role:`model`,parts:r})}return{messages:n}}formatToolResults(e){return{role:`tool_result`,content:e.map(e=>`${e.toolName||e.toolUseId}: ${e.content}`).join(`
+`),_toolResults:e}}shouldContinue(e){return e===`tool_use`}formatError(e){if(!e||!(e instanceof Error))return`Something went wrong. Please try again.`;let t=e.message||``,n=e;return n.status===429?`Too many requests. Please wait a moment and try again.`:n.status===403||n.status===401?`Authentication failed. Please check your API key configuration.`:n.status===503||n.status===500?`The AI service is temporarily unavailable. Please try again in a moment.`:t.includes(`SAFETY`)?`The request was blocked by safety filters. Please rephrase your request.`:t.includes(`RECITATION`)?`The response was blocked due to recitation concerns. Please try a different query.`:t.startsWith(`{`)||t.startsWith(`[`)?`The AI service encountered an error. Please try again.`:t}};exports.GoogleProvider=t;