drizzle-cube 0.7.1 → 0.8.1

package/dist/server/compiler.d.ts

@@ -1,9 +1,72 @@
 import { SemanticQuery, QueryResult, SecurityContext, DatabaseExecutor, CubeMetadata, Cube, QueryAnalysis, CacheConfig, ExplainOptions, ExplainResult, ExecutionOptions, RLSSetupFn } from './types/index.js';
+import { QueryValidationResult } from './query-validator.js';
 import { PlanOptimiser } from './logical-plan/index.js';
 export { validateQueryAgainstCubes } from './query-validator.js';
+export type { QueryValidationIssue, QueryValidationResult } from './query-validator.js';
+/**
+ * Identifier of the base cube set — the cubes registered with {@link
+ * SemanticLayerCompiler.registerCube}, shared by every tenant. The empty string
+ * is used so a set id can never collide with it (`registerCubeSet` rejects it).
+ */
+export declare const BASE_CUBE_SET_ID = "";
+/**
+ * Security context for deployments with no tenancy at all.
+ *
+ * Every cube-resolving method requires a `SecurityContext` so that omitting one
+ * is a compile error rather than a silent fall back to the wrong tenant's
+ * cubes. Single-tenant callers pass this constant to say "no tenancy here" out
+ * loud, instead of an anonymous `{}` that reads like an oversight.
+ */
+export declare const SINGLE_TENANT_CONTEXT: SecurityContext;
+/** Reported to `onCubeSetRegistered` after each {@link SemanticLayerCompiler.registerCubeSet}. */
+export interface CubeSetRegistrationInfo {
+    /** The set that was registered. */
+    setId: string;
+    /** Cubes in the set after merging over the base set. */
+    cubeCount: number;
+    /** Total dimensions across those cubes — the number that explains a slow boot. */
+    dimensionCount: number;
+    /**
+     * Monotonic across the whole compiler — incremented on every registration of
+     * any set, never reset. Part of the cache key, so results computed from
+     * superseded definitions can never be served.
+     */
+    generation: number;
+    /** Wall-clock cost of this registration. */
+    durationMs: number;
+}
+/** Aggregate registration cost, for a single summary line after the boot loop. */
+export interface CubeSetStats {
+    /** Number of registered sets, excluding the base set. */
+    setCount: number;
+    /** Total cubes across all sets (merged), excluding the base set. */
+    cubeCount: number;
+    /** Sum of every set's most recent registration time. */
+    totalRegistrationMs: number;
+    /** The most expensive set to register, if any sets are registered. */
+    slowestSet?: {
+        setId: string;
+        durationMs: number;
+    };
+}
 export declare class SemanticLayerCompiler {
-    private cubes;
-    private metadataCache?;
+    /** Cubes shared by every tenant. */
+    private baseCubes;
+    /** Per-tenant overlays, keyed by cube-set id. */
+    private cubeSets;
+    /** Generated metadata per cube set (`BASE_CUBE_SET_ID` for the base set). */
+    private metadataCache;
+    /** Bumped on every base-set mutation; part of every set's cache key. */
+    private baseGeneration;
+    /**
+     * Monotonic registration counter, never reset. Used as each set's generation
+     * so that unregistering a set and registering a different one under the same
+     * id cannot reproduce an earlier cache key and serve its stale results.
+     */
+    private registrationCounter;
+    private contextToCubeSetId?;
+    private missingCubeSet;
+    private onCubeSetRegistered?;
     private cacheConfig?;
     private rlsSetup?;
     private planOptimiser?;
@@ -29,6 +92,25 @@ export declare class SemanticLayerCompiler {
          * Defaults to a no-op IdentityOptimiser when omitted.
          */
         planOptimiser?: PlanOptimiser;
+        /**
+         * Maps a security context to the cube set that serves it — drizzle-cube's
+         * equivalent of Cube's `contextToAppId`. Return `undefined` (or omit this
+         * option entirely) to serve the base set, which is the single-tenant
+         * behaviour and the default.
+         */
+        contextToCubeSetId?: (securityContext: SecurityContext) => string | number | undefined;
+        /**
+         * What to do when `contextToCubeSetId` names a set that is not registered.
+         * `'base'` (default) serves the base set; `'throw'` fails the request, for
+         * deployments where every tenant is required to have its own set.
+         */
+        missingCubeSet?: 'base' | 'throw';
+        /**
+         * Called after each `registerCubeSet` with its cost. Registering a set per
+         * tenant is real startup work, so it is reported rather than hidden; emit
+         * it to your own logger or metrics.
+         */
+        onCubeSetRegistered?: (info: CubeSetRegistrationInfo) => void;
     });
     /**
      * Set or update the database connection
@@ -65,12 +147,75 @@ export declare class SemanticLayerCompiler {
      * Validates calculated measures during registration
      */
     registerCube(cube: Cube): void;
+    /**
+     * Register the cubes that serve one tenant, overlaying the base set.
+     *
+     * Cubes are matched to the base set by name, so a set can either add cubes or
+     * replace a base cube with a tenant-specific version (the usual case for
+     * generated per-tenant dimensions). Call once per tenant at application boot;
+     * calling again for the same id replaces that set and invalidates its cached
+     * metadata and query results.
+     */
+    registerCubeSet(setId: string, cubes: Cube[]): void;
+    /**
+     * Remove a tenant's cube set. That tenant then resolves to the base set (or
+     * throws, under `missingCubeSet: 'throw'`).
+     * Returns true if the set existed.
+     */
+    unregisterCubeSet(setId: string): boolean;
+    /** Whether a cube set is registered for this id. */
+    hasCubeSet(setId: string): boolean;
+    /** Ids of every registered cube set, excluding the base set. */
+    getCubeSetIds(): string[];
+    /**
+     * Aggregate registration cost across all sets — for the single summary line
+     * worth logging after a boot loop.
+     */
+    getCubeSetStats(): CubeSetStats;
+    /**
+     * Validate and enrich a cube ahead of registration, against the cubes that
+     * will be visible alongside it.
+     */
+    private prepareCube;
+    /** Base cubes with an overlay applied over them, by cube name. */
+    private mergeWithBase;
+    /** Recompute every set's merged view — after any base-set mutation. */
+    private rebuildAllMergedSets;
+    /**
+     * Resolve the cube set serving this security context.
+     *
+     * Returns {@link BASE_CUBE_SET_ID} when no mapping is configured or the
+     * mapping yields nothing. A configured id that has no registered set falls
+     * back to the base set, or throws under `missingCubeSet: 'throw'`.
+     */
+    private resolveSetId;
+    /**
+     * The cubes this security context may see.
+     *
+     * This is the only path by which cube *contents* are read for a query,
+     * metadata, validation or MCP response — so no such path can reach a cube
+     * list without a security context. The returned map must be treated as
+     * read-only; it is the live merged view, not a copy.
+     */
+    private resolveCubes;
+    /**
+     * Cache-key component identifying which cube definitions produced a result.
+     *
+     * Appended unconditionally to every query cache key: the security-context
+     * hash is not enough, because `includeSecurityContext: false` and a custom
+     * `securityContextSerializer` can both hash two tenants identically. The
+     * generation makes re-registering a set invalidate its cached results, so a
+     * retyped or renamed dimension cannot be served stale for the TTL.
+     */
+    private cubeSetCacheKey;
     /**
      * Validate that all string-based cube references in joins resolve to registered cubes.
      * Call after all cubes are registered for strict startup validation.
      * Throws an error listing all unresolved references.
      */
     validateCubeReferences(): void;
+    /** Join targets naming a cube that is absent from the given cube scope. */
+    private unresolvedJoinRefs;
     /**
      * Validate calculated measures in a cube
      * Checks template syntax, dependency existence, and circular dependencies
@@ -79,15 +224,17 @@ export declare class SemanticLayerCompiler {
     /**
      * Get a cube by name
      */
-    getCube(name: string): Cube | undefined;
+    getCube(name: string, securityContext: SecurityContext): Cube | undefined;
     /**
      * Get all registered cubes
      */
-    getAllCubes(): Cube[];
+    getAllCubes(securityContext: SecurityContext): Cube[];
     /**
-     * Get all cubes as a Map for multi-cube queries
+     * Get all cubes as a Map for multi-cube queries.
+     *
+     * Returns the live merged view for this context — treat it as read-only.
      */
-    getAllCubesMap(): Map<string, Cube>;
+    getAllCubesMap(securityContext: SecurityContext): Map<string, Cube>;
     /**
      * Unified query execution method that handles both single and multi-cube queries
      * @param options.skipCache - Skip cache lookup (but still cache the fresh result)
@@ -107,7 +254,7 @@ export declare class SemanticLayerCompiler {
      * Uses caching to improve performance for repeated requests
      * Cache is invalidated when cubes are modified (registerCube, removeCube, clearCubes)
      */
-    getMetadata(): CubeMetadata[];
+    getMetadata(securityContext: SecurityContext): CubeMetadata[];
     /**
      * Extract column name from Drizzle column reference
      * Handles different column types and extracts the actual column name
@@ -172,7 +319,7 @@ export declare class SemanticLayerCompiler {
     /**
      * Check if a cube exists
      */
-    hasCube(name: string): boolean;
+    hasCube(name: string, securityContext: SecurityContext): boolean;
     /**
      * Unregister a cube by name.
      * Returns true if the cube existed and was removed, false if not found.
@@ -191,18 +338,17 @@ export declare class SemanticLayerCompiler {
      * Called whenever cubes are modified
      */
     private invalidateMetadataCache;
+    /** Cubes for an already-resolved set id. */
+    private cubeSetId2Cubes;
     /**
      * Get cube names
      */
-    getCubeNames(): string[];
+    getCubeNames(securityContext: SecurityContext): string[];
     /**
      * Validate a query against registered cubes
      * Ensures all referenced cubes and fields exist
      */
-    validateQuery(query: SemanticQuery): {
-        isValid: boolean;
-        errors: string[];
-    };
+    validateQuery(query: SemanticQuery, securityContext: SecurityContext): QueryValidationResult;
     /**
      * Analyze query planning decisions for debugging and transparency
      * Returns detailed metadata about how the query would be planned

package/dist/server/execution/query-result-cache.d.ts

@@ -6,7 +6,7 @@ export declare class QueryResultCache {
      * Returns the cache key when caching is enabled and a provider is configured,
      * otherwise undefined (which disables lookup and store).
      */
-    generateKey(query: SemanticQuery, securityContext: SecurityContext): string | undefined;
+    generateKey(query: SemanticQuery, securityContext: SecurityContext, cubeSetKey?: string): string | undefined;
     /**
      * Look up a cached result. Returns the cached value (with cache metadata
      * attached) on a hit, or undefined on miss / bypass / no-cache. Emits

package/dist/server/executor.d.ts

@@ -95,6 +95,21 @@ export declare class QueryExecutor {
      *   (e.g. comparison period sub-queries cache at the comparison level).
      */
     private executeStandardQuery;
+    /**
+     * Count the rows the query would return with no limit or offset — Cube's
+     * `total`.
+     *
+     * Rebuilds the *same* physical plan with pagination and ordering stripped and
+     * counts over it as a subquery. The wrapper is what makes a grouped query
+     * count groups rather than base rows, and rebuilding from `planQuery` rather
+     * than reading the paginated query's effective values matters because
+     * `applyLimitAndOffset` injects a default limit when an offset arrives
+     * without one.
+     *
+     * Runs inside the caller's RLS context, so the count sees exactly the rows
+     * the page did.
+     */
+    private executeTotalCount;
     /**
      * Create a query context with optional filter cache.
      */

package/dist/{adapters/google-eKAc20al.js → server/google-DRY7jnOh.js}

@@ -1,18 +1,36 @@
 //#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)) t[n] = typeof r == "object" && r ? e(r) : r;
-		n[r] = t;
-	} else n[r] = r === "items" && typeof i == "object" && i ? e(i) : i;
-	return n;
+function e(n, r = 0) {
+	if (!n || typeof n != "object") return "value";
+	let i = n, a = i.enum;
+	if (Array.isArray(a) && a.length > 0) return a.map((e) => typeof e == "string" ? `"${e}"` : String(e)).join("|");
+	let o = typeof i.type == "string" ? i.type.toLowerCase() : "value";
+	if (o === "array") return r >= t ? "array" : `array of ${e(i.items, r + 1)}`;
+	if (o === "object") {
+		let n = i.properties;
+		if (!n || r >= t) return "object";
+		let a = new Set(Array.isArray(i.required) ? i.required : []);
+		return `{ ${Object.entries(n).map(([t, n]) => {
+			let i = a.has(t) ? " (required)" : "";
+			return `${t}: ${e(n, r + 1)}${i}`;
+		}).join(", ")} }`;
+	}
+	return o;
+}
+var t = 3;
+function n(t) {
+	let r = {}, i = [];
+	for (let [a, o] of Object.entries(t)) if (a === "type" && typeof o == "string") r[a] = o.toUpperCase();
+	else if (a === "properties" && typeof o == "object" && o) {
+		let e = {};
+		for (let [t, r] of Object.entries(o)) e[t] = typeof r == "object" && r ? n(r) : r;
+		r[a] = e;
+	} else a === "items" && typeof o == "object" && o ? r[a] = n(o) : (a === "anyOf" || a === "oneOf" || a === "allOf") && Array.isArray(o) ? r[a] = o.map((e) => e && typeof e == "object" ? n(e) : e) : a === "additionalProperties" ? o && typeof o == "object" && i.push(`Each value is: ${e(o)}.`) : a === "enum" && Array.isArray(o) && o.some((e) => typeof e != "string") ? i.push(`Allowed values: ${o.join(", ")}.`) : r[a] = o;
+	return i.length > 0 && (r.description = [typeof r.description == "string" ? r.description : "", ...i].filter(Boolean).join(" ")), r;
 }
-function t(e) {
+function r(e) {
 	return e === "STOP" ? "stop" : e === "MAX_TOKENS" ? "max_tokens" : e;
 }
-function* n(e) {
+function* i(e) {
 	if (e.text && !e.thought && (yield {
 		type: "text_delta",
 		text: e.text
@@ -29,7 +47,7 @@ function* n(e) {
 		}, yield { type: "tool_use_end" };
 	}
 }
-function r(e) {
+function a(e) {
 	let t = e._toolResults;
 	if (t && t.length > 0) return {
 		role: "user",
@@ -47,7 +65,7 @@ function r(e) {
 		parts: [{ text: n }]
 	} : null;
 }
-function i(e) {
+function o(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") {
@@ -59,8 +77,8 @@ function i(e) {
 	}
 	return t;
 }
-function a(e) {
-	if (e.role === "tool_result") return r(e);
+function s(e) {
+	if (e.role === "tool_result") return a(e);
 	if (e.role === "user") return {
 		role: "user",
 		parts: [{ text: typeof e.content == "string" ? e.content : JSON.stringify(e.content) }]
@@ -70,7 +88,7 @@ function a(e) {
 			role: "model",
 			parts: [{ text: e.content }]
 		};
-		let t = i(e.content);
+		let t = o(e.content);
 		return t.length > 0 ? {
 			role: "model",
 			parts: t
@@ -78,7 +96,7 @@ function a(e) {
 	}
 	return null;
 }
-var o = class {
+var c = class {
 	apiKey;
 	sdk;
 	initialized = !1;
@@ -111,7 +129,7 @@ var o = class {
 		})).stream;
 	}
 	async *parseStreamEvents(e) {
-		let r = "stop", i = !1;
+		let t = "stop", n = !1;
 		for await (let a of e) {
 			let e = a;
 			e.usageMetadata && (yield {
@@ -122,26 +140,26 @@ var o = class {
 			});
 			let o = e.candidates?.[0];
 			if (!o) continue;
-			o.finishReason && (r = t(o.finishReason));
+			o.finishReason && (t = r(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;
+			if (s) for (let e of s) for (let t of i(e)) t.type === "tool_use_start" && (n = !0), yield t;
 		}
 		yield {
 			type: "message_meta",
-			stopReason: i ? "tool_use" : r
+			stopReason: n ? "tool_use" : t
 		};
 	}
-	formatTools(t) {
-		return t.map((t) => ({
-			name: t.name,
-			description: t.description,
-			parameters: e(t.parameters)
+	formatTools(e) {
+		return e.map((e) => ({
+			name: e.name,
+			description: e.description,
+			parameters: n(e.parameters)
 		}));
 	}
 	formatMessages(e, t) {
 		let n = [];
 		for (let t of e) {
-			let e = a(t);
+			let e = s(t);
 			e && n.push(e);
 		}
 		return { messages: n };
@@ -163,4 +181,4 @@ var o = class {
 	}
 };
 //#endregion
-export { o as GoogleProvider };
+export { c as GoogleProvider };

package/dist/server/{google-BhvizbFR.cjs → google-PsZzwedr.cjs}

@@ -1,2 +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))t[n]=typeof r==`object`&&r?e(r):r;n[r]=t}else n[r]=r===`items`&&typeof i==`object`&&i?e(i):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-CohrBmKe.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;
+function e(n,r=0){if(!n||typeof n!=`object`)return`value`;let i=n,a=i.enum;if(Array.isArray(a)&&a.length>0)return a.map(e=>typeof e==`string`?`"${e}"`:String(e)).join(`|`);let o=typeof i.type==`string`?i.type.toLowerCase():`value`;if(o===`array`)return r>=t?`array`:`array of ${e(i.items,r+1)}`;if(o===`object`){let n=i.properties;if(!n||r>=t)return`object`;let a=new Set(Array.isArray(i.required)?i.required:[]);return`{ ${Object.entries(n).map(([t,n])=>{let i=a.has(t)?` (required)`:``;return`${t}: ${e(n,r+1)}${i}`}).join(`, `)} }`}return o}var t=3;function n(t){let r={},i=[];for(let[a,o]of Object.entries(t))if(a===`type`&&typeof o==`string`)r[a]=o.toUpperCase();else if(a===`properties`&&typeof o==`object`&&o){let e={};for(let[t,r]of Object.entries(o))e[t]=typeof r==`object`&&r?n(r):r;r[a]=e}else a===`items`&&typeof o==`object`&&o?r[a]=n(o):(a===`anyOf`||a===`oneOf`||a===`allOf`)&&Array.isArray(o)?r[a]=o.map(e=>e&&typeof e==`object`?n(e):e):a===`additionalProperties`?o&&typeof o==`object`&&i.push(`Each value is: ${e(o)}.`):a===`enum`&&Array.isArray(o)&&o.some(e=>typeof e!=`string`)?i.push(`Allowed values: ${o.join(`, `)}.`):r[a]=o;return i.length>0&&(r.description=[typeof r.description==`string`?r.description:``,...i].filter(Boolean).join(` `)),r}function r(e){return e===`STOP`?`stop`:e===`MAX_TOKENS`?`max_tokens`:e}function*i(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 a(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 o(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 s(e){if(e.role===`tool_result`)return a(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=o(e.content);return t.length>0?{role:`model`,parts:t}:null}return null}var c=class{apiKey;sdk;initialized=!1;constructor(e){this.apiKey=e}async ensureSDK(){if(!this.initialized){try{this.sdk=await Promise.resolve().then(()=>require("./dist-CohrBmKe.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 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&&(t=r(o.finishReason));let s=o.content?.parts;if(s)for(let e of s)for(let t of i(e))t.type===`tool_use_start`&&(n=!0),yield t}yield{type:`message_meta`,stopReason:n?`tool_use`:t}}formatTools(e){return e.map(e=>({name:e.name,description:e.description,parameters:n(e.parameters)}))}formatMessages(e,t){let n=[];for(let t of e){let e=s(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=c;