drizzle-cube 0.4.19 → 0.4.21

package/dist/adapters/fastify/index.js

@@ -1,35 +1,35 @@
-import { S as D, v as K, b as x, a as T, n as V, p as z, w as _, d as B, i as j, M as F, c as U, e as E, s as P } from "../mcp-transport-m1X1GtwG.js";
-import { formatErrorResponse as u, formatCubeResponse as A, handleBatchRequest as X, formatMetaResponse as G, formatSqlResponse as I, handleDryRun as M } from "../utils.js";
-const N = function(i, k, R) {
+import { S as H, v as D, b as E, a as T, n as V, p as z, w as _, d as B, i as j, M as F, c as U, e as P, s as Q } from "../mcp-transport-m1X1GtwG.js";
+import { formatErrorResponse as d, formatCubeResponse as A, handleBatchRequest as X, formatMetaResponse as G, formatSqlResponse as I, handleDryRun as O } from "../utils.js";
+const M = function(i, N, $) {
   const {
-    cubes: S,
-    drizzle: O,
+    cubes: R,
+    drizzle: k,
     schema: J,
     extractSecurityContext: y,
     engineType: L,
     cors: q,
     basePath: p = "/cubejs-api/v1",
-    bodyLimit: m = 10485760,
+    bodyLimit: g = 10485760,
     // 10MB
-    cache: H,
-    mcp: f = { enabled: !0 },
-    agent: v
-  } = k;
-  if (!S || S.length === 0)
-    return R(new Error("At least one cube must be provided in the cubes array"));
+    cache: K,
+    mcp: S = { enabled: !0 },
+    agent: h
+  } = N;
+  if (!R || R.length === 0)
+    return $(new Error("At least one cube must be provided in the cubes array"));
   q && i.register(import("@fastify/cors"), q), i.addHook("onRequest", async (t, r) => {
     t.method === "POST" && (t.body = void 0);
   });
-  const c = new D({
-    drizzle: O,
+  const c = new H({
+    drizzle: k,
     schema: J,
     engineType: L,
-    cache: H
+    cache: K
   });
-  if (S.forEach((t) => {
+  if (R.forEach((t) => {
     c.registerCube(t);
   }), i.post(`${p}/load`, {
-    bodyLimit: m,
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
@@ -38,16 +38,16 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const e = t.body, a = e.query || e, n = await y(t), s = c.validateQuery(a);
-      if (!s.isValid)
-        return r.status(400).send(u(
-          `Query validation failed: ${s.errors.join(", ")}`,
+      const e = t.body, a = e.query || e, n = await y(t), o = c.validateQuery(a);
+      if (!o.isValid)
+        return r.status(400).send(d(
+          `Query validation failed: ${o.errors.join(", ")}`,
           400
         ));
-      const o = t.headers["x-cache-control"] === "no-cache", l = await c.executeMultiCubeQuery(a, n, { skipCache: o });
-      return A(a, l, c);
+      const s = t.headers["x-cache-control"] === "no-cache", u = await c.executeMultiCubeQuery(a, n, { skipCache: s });
+      return A(a, u, c);
     } catch (e) {
-      return t.log.error(e, "Query execution error"), r.status(500).send(u(
+      return t.log.error(e, "Query execution error"), r.status(500).send(d(
         e instanceof Error ? e.message : "Query execution failed",
         500
       ));
@@ -69,27 +69,27 @@ const N = function(i, k, R) {
       try {
         a = JSON.parse(e);
       } catch {
-        return r.status(400).send(u(
+        return r.status(400).send(d(
           "Invalid JSON in query parameter",
           400
         ));
       }
-      const n = await y(t), s = c.validateQuery(a);
-      if (!s.isValid)
-        return r.status(400).send(u(
-          `Query validation failed: ${s.errors.join(", ")}`,
+      const n = await y(t), o = c.validateQuery(a);
+      if (!o.isValid)
+        return r.status(400).send(d(
+          `Query validation failed: ${o.errors.join(", ")}`,
           400
         ));
-      const o = t.headers["x-cache-control"] === "no-cache", l = await c.executeMultiCubeQuery(a, n, { skipCache: o });
-      return A(a, l, c);
+      const s = t.headers["x-cache-control"] === "no-cache", u = await c.executeMultiCubeQuery(a, n, { skipCache: s });
+      return A(a, u, c);
     } catch (e) {
-      return t.log.error(e, "Query execution error"), r.status(500).send(u(
+      return t.log.error(e, "Query execution error"), r.status(500).send(d(
         e instanceof Error ? e.message : "Query execution failed",
         500
       ));
     }
   }), i.post(`${p}/batch`, {
-    bodyLimit: m,
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
@@ -106,19 +106,19 @@ const N = function(i, k, R) {
     try {
       const { queries: e } = t.body;
       if (!e || !Array.isArray(e))
-        return r.status(400).send(u(
+        return r.status(400).send(d(
           'Request body must contain a "queries" array',
           400
         ));
       if (e.length === 0)
-        return r.status(400).send(u(
+        return r.status(400).send(d(
           "Queries array cannot be empty",
           400
         ));
       const a = await y(t), n = t.headers["x-cache-control"] === "no-cache";
       return await X(e, a, c, { skipCache: n });
     } catch (e) {
-      return t.log.error(e, "Batch execution error"), r.status(500).send(u(
+      return t.log.error(e, "Batch execution error"), r.status(500).send(d(
         e instanceof Error ? e.message : "Batch execution failed",
         500
       ));
@@ -128,13 +128,13 @@ const N = function(i, k, R) {
       const e = c.getMetadata();
       return G(e);
     } catch (e) {
-      return t.log.error(e, "Metadata error"), r.status(500).send(u(
+      return t.log.error(e, "Metadata error"), r.status(500).send(d(
         e instanceof Error ? e.message : "Failed to fetch metadata",
         500
       ));
     }
   }), i.post(`${p}/sql`, {
-    bodyLimit: m,
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
@@ -145,20 +145,20 @@ const N = function(i, k, R) {
     try {
       const e = t.body, a = await y(t), n = c.validateQuery(e);
       if (!n.isValid)
-        return r.status(400).send(u(
+        return r.status(400).send(d(
           `Query validation failed: ${n.errors.join(", ")}`,
           400
         ));
-      const s = e.measures?.[0] || e.dimensions?.[0];
-      if (!s)
-        return r.status(400).send(u(
+      const o = e.measures?.[0] || e.dimensions?.[0];
+      if (!o)
+        return r.status(400).send(d(
           "No measures or dimensions specified",
           400
         ));
-      const o = s.split(".")[0], l = await c.generateSQL(o, e, a);
-      return I(e, l);
+      const s = o.split(".")[0], u = await c.generateSQL(s, e, a);
+      return I(e, u);
     } catch (e) {
-      return t.log.error(e, "SQL generation error"), r.status(500).send(u(
+      return t.log.error(e, "SQL generation error"), r.status(500).send(d(
         e instanceof Error ? e.message : "SQL generation failed",
         500
       ));
@@ -175,28 +175,28 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const { query: e } = t.query, a = JSON.parse(e), n = await y(t), s = c.validateQuery(a);
-      if (!s.isValid)
-        return r.status(400).send(u(
-          `Query validation failed: ${s.errors.join(", ")}`,
+      const { query: e } = t.query, a = JSON.parse(e), n = await y(t), o = c.validateQuery(a);
+      if (!o.isValid)
+        return r.status(400).send(d(
+          `Query validation failed: ${o.errors.join(", ")}`,
           400
         ));
-      const o = a.measures?.[0] || a.dimensions?.[0];
-      if (!o)
-        return r.status(400).send(u(
+      const s = a.measures?.[0] || a.dimensions?.[0];
+      if (!s)
+        return r.status(400).send(d(
           "No measures or dimensions specified",
           400
         ));
-      const l = o.split(".")[0], g = await c.generateSQL(l, a, n);
-      return I(a, g);
+      const u = s.split(".")[0], w = await c.generateSQL(u, a, n);
+      return I(a, w);
     } catch (e) {
-      return t.log.error(e, "SQL generation error"), r.status(500).send(u(
+      return t.log.error(e, "SQL generation error"), r.status(500).send(d(
         e instanceof Error ? e.message : "SQL generation failed",
         500
       ));
     }
   }), i.post(`${p}/dry-run`, {
-    bodyLimit: m,
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
@@ -206,7 +206,7 @@ const N = function(i, k, R) {
   }, async (t, r) => {
     try {
       const e = t.body, a = e.query || e, n = await y(t);
-      return await M(a, n, c);
+      return await O(a, n, c);
     } catch (e) {
       return t.log.error(e, "Dry-run error"), r.status(400).send({
         error: e instanceof Error ? e.message : "Dry-run validation failed",
@@ -226,7 +226,7 @@ const N = function(i, k, R) {
   }, async (t, r) => {
     try {
       const { query: e } = t.query, a = JSON.parse(e), n = await y(t);
-      return await M(a, n, c);
+      return await O(a, n, c);
     } catch (e) {
       return t.log.error(e, "Dry-run error"), r.status(400).send({
         error: e instanceof Error ? e.message : "Dry-run validation failed",
@@ -234,7 +234,7 @@ const N = function(i, k, R) {
       });
     }
   }), i.post(`${p}/explain`, {
-    bodyLimit: m,
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
@@ -243,17 +243,17 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const e = t.body, a = e.query || e, n = e.options || {}, s = await y(t), o = c.validateQuery(a);
-      return o.isValid ? await c.explainQuery(a, s, n) : r.status(400).send({
-        error: `Query validation failed: ${o.errors.join(", ")}`
+      const e = t.body, a = e.query || e, n = e.options || {}, o = await y(t), s = c.validateQuery(a);
+      return s.isValid ? await c.explainQuery(a, o, n) : r.status(400).send({
+        error: `Query validation failed: ${s.errors.join(", ")}`
       });
     } catch (e) {
       return t.log.error(e, "Explain error"), r.status(500).send({
         error: e instanceof Error ? e.message : "Explain query failed"
       });
     }
-  }), v && i.post(`${p}/agent/chat`, {
-    bodyLimit: m,
+  }), h && i.post(`${p}/agent/chat`, {
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
@@ -262,44 +262,48 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const { handleAgentChat: e } = await import("../handler-BV4JuWNW.js"), a = t.body, { message: n, sessionId: s, history: o } = a;
+      const { handleAgentChat: e } = await import("../handler-9Rdn7zM2.js"), a = t.body, { message: n, sessionId: o, history: s } = a;
       if (!n || typeof n != "string")
         return r.status(400).send({ error: "message is required and must be a string" });
-      let l = (v.apiKey || "").trim();
-      if (v.allowClientApiKey) {
-        const d = t.headers["x-agent-api-key"];
-        d && (l = d.trim());
+      let u = (h.apiKey || "").trim();
+      if (h.allowClientApiKey) {
+        const m = t.headers["x-agent-api-key"];
+        m && (u = m.trim());
       }
-      if (!l)
+      if (!u)
         return r.status(401).send({
           error: "No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header."
         });
-      const g = await y(t);
+      const w = h.allowClientApiKey ? t.headers["x-agent-provider"] : void 0, l = h.allowClientApiKey ? t.headers["x-agent-model"] : void 0, v = h.allowClientApiKey ? t.headers["x-agent-provider-endpoint"] : void 0, b = await y(t), f = h.buildSystemContext?.(b);
       r.raw.writeHead(200, {
         "Content-Type": "text/event-stream",
         "Cache-Control": "no-cache",
         Connection: "keep-alive"
       });
       try {
-        const d = e({
+        const m = e({
           message: n,
-          sessionId: s,
-          history: o,
+          sessionId: o,
+          history: s,
           semanticLayer: c,
-          securityContext: g,
-          agentConfig: v,
-          apiKey: l
+          securityContext: b,
+          agentConfig: h,
+          apiKey: u,
+          systemContext: f,
+          providerOverride: w,
+          modelOverride: l,
+          baseURLOverride: v
         });
-        for await (const h of d)
-          r.raw.write(`data: ${JSON.stringify(h)}
+        for await (const C of m)
+          r.raw.write(`data: ${JSON.stringify(C)}
 
 `);
-      } catch (d) {
-        const h = {
+      } catch (m) {
+        const C = {
           type: "error",
-          data: { message: d instanceof Error ? d.message : "Stream failed" }
+          data: { message: m instanceof Error ? m.message : "Stream failed" }
         };
-        r.raw.write(`data: ${JSON.stringify(h)}
+        r.raw.write(`data: ${JSON.stringify(C)}
 
 `);
       } finally {
@@ -311,10 +315,10 @@ const N = function(i, k, R) {
           error: e instanceof Error ? e.message : "Agent chat failed"
         });
     }
-  }), f.enabled !== !1) {
-    const t = f.basePath ?? "/mcp";
+  }), S.enabled !== !1) {
+    const t = S.basePath ?? "/mcp";
     i.post(`${t}`, {
-      bodyLimit: m,
+      bodyLimit: g,
       schema: {
         body: {
           type: "object",
@@ -322,71 +326,71 @@ const N = function(i, k, R) {
         }
       }
     }, async (r, e) => {
-      const a = K(
+      const a = D(
         r.headers.origin,
-        f.allowedOrigins ? { allowedOrigins: f.allowedOrigins } : {}
+        S.allowedOrigins ? { allowedOrigins: S.allowedOrigins } : {}
       );
       if (!a.valid)
-        return e.status(403).send(x(null, -32600, a.reason));
+        return e.status(403).send(E(null, -32600, a.reason));
       const n = r.headers.accept;
       if (!T(n))
-        return e.status(400).send(x(null, -32600, "Accept header must include both application/json and text/event-stream"));
-      const s = V(r.headers);
-      if (!s.ok)
+        return e.status(400).send(E(null, -32600, "Accept header must include both application/json and text/event-stream"));
+      const o = V(r.headers);
+      if (!o.ok)
         return e.status(426).send({
           error: "Unsupported MCP protocol version",
-          supported: s.supported
+          supported: o.supported
         });
-      const o = z(r.body);
-      if (!o)
-        return e.status(400).send(x(null, -32600, "Invalid JSON-RPC 2.0 request"));
-      const l = _(n), g = o.method === "initialize";
+      const s = z(r.body);
+      if (!s)
+        return e.status(400).send(E(null, -32600, "Invalid JSON-RPC 2.0 request"));
+      const u = _(n), w = s.method === "initialize";
       try {
-        const d = await B(
-          o.method,
-          o.params,
+        const l = await B(
+          s.method,
+          s.params,
           {
             semanticLayer: c,
             extractSecurityContext: y,
             rawRequest: r,
             rawResponse: e,
-            negotiatedProtocol: s.negotiated
+            negotiatedProtocol: o.negotiated
           }
         );
-        if (j(o))
+        if (j(s))
           return e.status(202).send();
-        const h = g && d && typeof d == "object" && "sessionId" in d ? d.sessionId : void 0;
-        h && e.header(F, h);
-        const C = U(o.id ?? null, d);
-        if (l) {
-          const w = E();
-          e.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${w}
+        const v = w && l && typeof l == "object" && "sessionId" in l ? l.sessionId : void 0;
+        v && e.header(F, v);
+        const b = U(s.id ?? null, l);
+        if (u) {
+          const f = P();
+          e.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${f}
 
-${P(C, w)}`);
+${Q(b, f)}`);
           return;
         }
-        return e.send(C);
-      } catch (d) {
-        if (j(o))
-          return r.log.error(d, "MCP notification processing error"), e.status(202).send();
-        r.log.error(d, "MCP RPC error");
-        const h = d?.code ?? -32603, C = d?.data, w = d.message || "MCP request failed", Q = x(o.id ?? null, h, w, C);
-        if (l) {
-          const $ = E();
-          e.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${$}
+        return e.send(b);
+      } catch (l) {
+        if (j(s))
+          return r.log.error(l, "MCP notification processing error"), e.status(202).send();
+        r.log.error(l, "MCP RPC error");
+        const v = l?.code ?? -32603, b = l?.data, f = l.message || "MCP request failed", m = E(s.id ?? null, v, f, b);
+        if (u) {
+          const C = P();
+          e.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${C}
 
-${P(Q, $)}`);
+${Q(m, C)}`);
           return;
         }
-        return e.send(Q);
+        return e.send(m);
       }
     }), i.get(`${t}`, async (r, e) => {
-      const a = E();
+      const a = P();
       e.raw.writeHead(200, {
         "Content-Type": "text/event-stream",
         "Cache-Control": "no-cache",
         Connection: "keep-alive"
-      }), e.raw.write(P({
+      }), e.raw.write(Q({
         jsonrpc: "2.0",
         method: "mcp/ready",
         params: { protocol: "streamable-http" }
@@ -404,20 +408,20 @@ ${P(Q, $)}`);
   i.setErrorHandler(async (t, r, e) => {
     r.log.error(t, "Fastify cube adapter error"), e.statusCode < 400 && e.status(500);
     const a = t instanceof Error ? t : String(t);
-    return u(a, e.statusCode);
-  }), R();
+    return d(a, e.statusCode);
+  }), $();
 };
-async function Z(b, i) {
-  await b.register(N, i);
+async function Z(x, i) {
+  await x.register(M, i);
 }
-function ee(b) {
+function ee(x) {
   const i = require("fastify")({
     logger: !0
   });
-  return i.register(N, b), i;
+  return i.register(M, x), i;
 }
 export {
   ee as createCubeApp,
-  N as cubePlugin,
+  M as cubePlugin,
   Z as registerCubeRoutes
 };

package/dist/adapters/google-BAK9pnQf.cjs

@@ -0,0 +1,2 @@
+"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});function l(u){const e={};for(const[t,s]of Object.entries(u))if(t==="type"&&typeof s=="string")e[t]=s.toUpperCase();else if(t==="properties"&&typeof s=="object"&&s!==null){const n={};for(const[a,o]of Object.entries(s))typeof o=="object"&&o!==null?n[a]=l(o):n[a]=o;e[t]=n}else t==="items"&&typeof s=="object"&&s!==null?e[t]=l(s):e[t]=s;return e}class p{apiKey;sdk;initialized=!1;constructor(e){this.apiKey=e}async ensureSDK(){if(!this.initialized){try{this.sdk=await Promise.resolve().then(()=>require("./index-BgCeQBuN.cjs"))}catch{throw new 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();const{GoogleGenerativeAI:t}=this.sdk,n=new t(this.apiKey).getGenerativeModel({model:e.model,systemInstruction:e.system,generationConfig:{maxOutputTokens:e.maxTokens}}),{messages:a}=this.formatMessages(e.messages,e.system),o=this.formatTools(e.tools);return(await n.generateContentStream({contents:a,tools:o.length>0?[{functionDeclarations:o}]:void 0})).stream}async*parseStreamEvents(e){let t="stop",s=!1;for await(const n of e){const a=n;a.usageMetadata&&(yield{type:"message_meta",inputTokens:a.usageMetadata.promptTokenCount,outputTokens:a.usageMetadata.candidatesTokenCount,stopReason:""});const o=a.candidates?.[0];if(!o)continue;o.finishReason&&(o.finishReason==="STOP"?t="stop":o.finishReason==="MAX_TOKENS"?t="max_tokens":t=o.finishReason);const i=o.content?.parts;if(i){for(const r of i)if(r.text&&!r.thought&&(yield{type:"text_delta",text:r.text}),r.functionCall){s=!0;const f=`gemini-tc-${Date.now()}-${Math.random().toString(36).slice(2,7)}`,c=r.thoughtSignature?{thoughtSignature:r.thoughtSignature}:void 0;yield{type:"tool_use_start",id:f,name:r.functionCall.name,...c?{metadata:c}:{}},yield{type:"tool_input_delta",json:JSON.stringify(r.functionCall.args||{})},yield{type:"tool_use_end"}}}}yield{type:"message_meta",stopReason:s?"tool_use":t}}formatTools(e){return e.map(t=>({name:t.name,description:t.description,parameters:l(t.parameters)}))}formatMessages(e,t){const s=[];for(const n of e)if(n.role==="tool_result"){const a=n._toolResults;if(a&&a.length>0)s.push({role:"user",parts:a.map(o=>({functionResponse:{name:o.toolName||o.toolUseId,response:{content:o.content,isError:o.isError||!1}}}))});else{const o=typeof n.content=="string"?n.content:JSON.stringify(n.content);o&&s.push({role:"user",parts:[{text:o}]})}}else if(n.role==="user"){const a=typeof n.content=="string"?n.content:JSON.stringify(n.content);s.push({role:"user",parts:[{text:a}]})}else if(n.role==="assistant")if(typeof n.content=="string")s.push({role:"model",parts:[{text:n.content}]});else{const a=n.content,o=[];for(const i of a)if(i.type==="text"&&i.text)o.push({text:i.text});else if(i.type==="tool_use"){const r={functionCall:{name:i.name,args:i.input||{}}};i.metadata?.thoughtSignature&&(r.thoughtSignature=i.metadata.thoughtSignature),o.push(r)}o.length>0&&s.push({role:"model",parts:o})}return{messages:s}}formatToolResults(e){return{role:"tool_result",content:e.map(t=>`${t.toolName||t.toolUseId}: ${t.content}`).join(`
+`),_toolResults:e}}shouldContinue(e){return e==="tool_use"}formatError(e){if(!e||!(e instanceof Error))return"Something went wrong. Please try again.";const t=e.message||"",s=e;return s.status===429?"Too many requests. Please wait a moment and try again.":s.status===403||s.status===401?"Authentication failed. Please check your API key configuration.":s.status===503||s.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=p;

package/dist/adapters/google-DficVAsJ.js

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