drizzle-cube 0.5.8 → 0.6.0

package/dist/server/google-BOnRIN1w.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}function t(e){return e===`STOP`?`stop`:e===`MAX_TOKENS`?`max_tokens`:e}function*n(e){if(e.text&&!e.thought&&(yield{type:`text_delta`,text:e.text}),e.functionCall){let t=`gemini-tc-${Date.now()}-${Math.random().toString(36).slice(2,7)}`,n=e.thoughtSignature?{thoughtSignature:e.thoughtSignature}:void 0;yield{type:`tool_use_start`,id:t,name:e.functionCall.name,...n?{metadata:n}:{}},yield{type:`tool_input_delta`,json:JSON.stringify(e.functionCall.args||{})},yield{type:`tool_use_end`}}}function r(e){let t=e._toolResults;if(t&&t.length>0)return{role:`user`,parts:t.map(e=>({functionResponse:{name:e.toolName||e.toolUseId,response:{content:e.content,isError:e.isError||!1}}}))};let n=typeof e.content==`string`?e.content:JSON.stringify(e.content);return n?{role:`user`,parts:[{text:n}]}:null}function i(e){let t=[];for(let n of e)if(n.type===`text`&&n.text)t.push({text:n.text});else if(n.type===`tool_use`){let e={functionCall:{name:n.name,args:n.input||{}}};n.metadata?.thoughtSignature&&(e.thoughtSignature=n.metadata.thoughtSignature),t.push(e)}return t}function a(e){if(e.role===`tool_result`)return r(e);if(e.role===`user`)return{role:`user`,parts:[{text:typeof e.content==`string`?e.content:JSON.stringify(e.content)}]};if(e.role===`assistant`){if(typeof e.content==`string`)return{role:`model`,parts:[{text:e.content}]};let t=i(e.content);return t.length>0?{role:`model`,parts:t}:null}return null}var o=class{apiKey;sdk;initialized=!1;constructor(e){this.apiKey=e}async ensureSDK(){if(!this.initialized){try{this.sdk=await Promise.resolve().then(()=>require("./dist-BOdf-GQO.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 r=`stop`,i=!1;for await(let a of e){let e=a;e.usageMetadata&&(yield{type:`message_meta`,inputTokens:e.usageMetadata.promptTokenCount,outputTokens:e.usageMetadata.candidatesTokenCount,stopReason:``});let o=e.candidates?.[0];if(!o)continue;o.finishReason&&(r=t(o.finishReason));let s=o.content?.parts;if(s)for(let e of s)for(let t of n(e))t.type===`tool_use_start`&&(i=!0),yield t}yield{type:`message_meta`,stopReason:i?`tool_use`:r}}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){let e=a(t);e&&n.push(e)}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=o;

package/dist/server/{google-CFYljAOF.js → google-CvKM9yVM.js}

@@ -9,7 +9,76 @@ function e(t) {
 	} else r === "items" && typeof i == "object" && i ? n[r] = e(i) : n[r] = i;
 	return n;
 }
-var t = class {
+function t(e) {
+	return e === "STOP" ? "stop" : e === "MAX_TOKENS" ? "max_tokens" : e;
+}
+function* n(e) {
+	if (e.text && !e.thought && (yield {
+		type: "text_delta",
+		text: e.text
+	}), e.functionCall) {
+		let t = `gemini-tc-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`, n = e.thoughtSignature ? { thoughtSignature: e.thoughtSignature } : void 0;
+		yield {
+			type: "tool_use_start",
+			id: t,
+			name: e.functionCall.name,
+			...n ? { metadata: n } : {}
+		}, yield {
+			type: "tool_input_delta",
+			json: JSON.stringify(e.functionCall.args || {})
+		}, yield { type: "tool_use_end" };
+	}
+}
+function r(e) {
+	let t = e._toolResults;
+	if (t && t.length > 0) return {
+		role: "user",
+		parts: t.map((e) => ({ functionResponse: {
+			name: e.toolName || e.toolUseId,
+			response: {
+				content: e.content,
+				isError: e.isError || !1
+			}
+		} }))
+	};
+	let n = typeof e.content == "string" ? e.content : JSON.stringify(e.content);
+	return n ? {
+		role: "user",
+		parts: [{ text: n }]
+	} : null;
+}
+function i(e) {
+	let t = [];
+	for (let n of e) if (n.type === "text" && n.text) t.push({ text: n.text });
+	else if (n.type === "tool_use") {
+		let e = { functionCall: {
+			name: n.name,
+			args: n.input || {}
+		} };
+		n.metadata?.thoughtSignature && (e.thoughtSignature = n.metadata.thoughtSignature), t.push(e);
+	}
+	return t;
+}
+function a(e) {
+	if (e.role === "tool_result") return r(e);
+	if (e.role === "user") return {
+		role: "user",
+		parts: [{ text: typeof e.content == "string" ? e.content : JSON.stringify(e.content) }]
+	};
+	if (e.role === "assistant") {
+		if (typeof e.content == "string") return {
+			role: "model",
+			parts: [{ text: e.content }]
+		};
+		let t = i(e.content);
+		return t.length > 0 ? {
+			role: "model",
+			parts: t
+		} : null;
+	}
+	return null;
+}
+var o = class {
 	apiKey;
 	sdk;
 	initialized = !1;
@@ -42,41 +111,24 @@ var t = class {
 		})).stream;
 	}
 	async *parseStreamEvents(e) {
-		let t = "stop", n = !1;
-		for await (let r of e) {
-			let e = r;
+		let r = "stop", i = !1;
+		for await (let a of e) {
+			let e = a;
 			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" };
-				}
-			}
+			let o = e.candidates?.[0];
+			if (!o) continue;
+			o.finishReason && (r = t(o.finishReason));
+			let s = o.content?.parts;
+			if (s) for (let e of s) for (let t of n(e)) t.type === "tool_use_start" && (i = !0), yield t;
 		}
 		yield {
 			type: "message_meta",
-			stopReason: n ? "tool_use" : t
+			stopReason: i ? "tool_use" : r
 		};
 	}
 	formatTools(t) {
@@ -88,49 +140,9 @@ var t = class {
 	}
 	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
-			});
+		for (let t of e) {
+			let e = a(t);
+			e && n.push(e);
 		}
 		return { messages: n };
 	}
@@ -151,4 +163,4 @@ var t = class {
 	}
 };
 //#endregion
-export { t as GoogleProvider };
+export { o as GoogleProvider };