drizzle-cube 0.8.0 → 0.8.1

package/dist/server/agent/chart-validation.d.ts

@@ -12,7 +12,7 @@ interface ValidationResult {
 /**
  * Validate chartConfig against the chart type's drop zone requirements.
  */
-export declare function validateChartConfig(chartType: string, chartConfig: Record<string, unknown> | undefined, _query: Record<string, unknown>): ValidationResult;
+export declare function validateChartConfig(chartType: string, chartConfig: Record<string, unknown> | undefined, query: Record<string, unknown>): ValidationResult;
 export declare function inferChartConfig(chartType: string, chartConfig: Record<string, unknown> | undefined, query: Record<string, unknown>): Record<string, unknown>;
 /**
  * Build per-chart-type requirements text for the agent tool description.

package/dist/server/ai/chart-schema.d.ts

@@ -0,0 +1,117 @@
+/**
+ * Chart Configuration Schema — shared by the MCP and Agent tools
+ *
+ * The chart-side sibling of `QUERY_PARAMS_SCHEMA`. Used by BOTH:
+ * - MCP `chart` tool (adapters/mcp-transport.ts)
+ * - Agent `add_portlet` / `save_as_dashboard` tools (server/agent/tools.ts)
+ *
+ * Most charts need nothing here — their axes are inferred from the query. The
+ * records table is the exception: it lays out `columns` rather than axes, and a
+ * column renders as plain text unless the caller says what it is. Formats are
+ * never guessed from the data, so a model that cannot see this shape produces a
+ * listing of unformatted strings.
+ */
+/** `chartConfig` fields specific to the records table. */
+export declare const RECORDS_TABLE_CHART_CONFIG_SCHEMA: {
+    readonly columns: {
+        readonly type: "array";
+        readonly items: {
+            readonly type: "string";
+        };
+        readonly description: "recordsTable only: fields to render as columns, in this order. Omit to show every field the query returns, in query order.";
+    };
+    readonly hiddenColumns: {
+        readonly type: "array";
+        readonly items: {
+            readonly type: "string";
+        };
+        readonly description: "recordsTable only: fields fetched for row context or rowLink tokens but never displayed (e.g. an id used in the URL template). Leave empty unless a field is genuinely only needed behind the scenes — anything listed here disappears from the table.";
+    };
+};
+/** `displayConfig` fields specific to the records table. */
+export declare const RECORDS_TABLE_DISPLAY_CONFIG_SCHEMA: {
+    readonly columnFormats: {
+        readonly type: "object";
+        readonly description: "recordsTable only: how each column renders, keyed by field name (e.g. \"Employees.salary\"). Formats are never inferred from the data — a column is plain text unless you say otherwise, so set this for any numeric, date, status or ratio column.";
+        readonly additionalProperties: {
+            readonly type: "object";
+            readonly required: readonly ["kind"];
+            readonly properties: {
+                readonly kind: {
+                    readonly type: "string";
+                    readonly enum: readonly ["text", "number", "date", "badge", "progress"];
+                    readonly description: "Required on every entry. text: as-is. number: formatted numeric. date: formatted date. badge: coloured pill for statuses/categories. progress: bar or ring for a bounded value. Use \"text\" when you only want to set a label.";
+                };
+                readonly numberFormat: {
+                    readonly type: "object";
+                    readonly description: "kind \"number\": { unit: \"currency\"|\"percent\"|\"number\"|\"custom\", decimals, abbreviate (K/M/B), currencyCode, customPrefix, customSuffix }.";
+                };
+                readonly dateGranularity: {
+                    readonly type: "string";
+                    readonly description: "kind \"date\": granularity to render at (e.g. \"day\", \"month\", \"year\").";
+                };
+                readonly badgeColors: {
+                    readonly type: "array";
+                    readonly description: "kind \"badge\": an ARRAY of { value, colorIndex } entries, not an object keyed by value. colorIndex is a number index into the dashboard palette (0, 1, 2, …), never a colour name like \"green\" — the palette is themed, so names would not follow it. A value with no entry renders neutral rather than being assigned a guessed colour.";
+                    readonly items: {
+                        readonly type: "object";
+                        readonly required: readonly ["value", "colorIndex"];
+                        readonly properties: {
+                            readonly value: {
+                                readonly type: "string";
+                                readonly description: "Cell value to colour, matched exactly";
+                            };
+                            readonly colorIndex: {
+                                readonly type: "number";
+                                readonly description: "Index into the dashboard colour palette";
+                            };
+                        };
+                    };
+                };
+                readonly progressMin: {
+                    readonly type: "number";
+                    readonly description: "kind \"progress\": lower bound (default 0). Values are clamped.";
+                };
+                readonly progressMax: {
+                    readonly type: "number";
+                    readonly description: "kind \"progress\": upper bound (default 100). Values are clamped.";
+                };
+                readonly progressStyle: {
+                    readonly type: "string";
+                    readonly enum: readonly ["bar", "circle"];
+                    readonly description: "kind \"progress\": full-width bar, or a compact ring for narrow columns.";
+                };
+                readonly label: {
+                    readonly type: "string";
+                    readonly description: "Header override; defaults to the field title from the cube metadata.";
+                };
+                readonly align: {
+                    readonly type: "string";
+                    readonly enum: readonly ["left", "right"];
+                    readonly description: "Cell alignment. Numbers usually read better right-aligned.";
+                };
+            };
+        };
+    };
+    readonly rowLink: {
+        readonly type: "object";
+        readonly description: "recordsTable only: makes each row a link.";
+        readonly required: readonly ["urlTemplate"];
+        readonly properties: {
+            readonly urlTemplate: {
+                readonly type: "string";
+                readonly description: "URL with {Cube.field} tokens substituted from the row, including hidden columns (e.g. \"/employees/{Employees.id}\"). Relative paths and http(s) URLs only.";
+            };
+            readonly target: {
+                readonly type: "string";
+                readonly enum: readonly ["self", "blank"];
+                readonly description: "Open in the same tab (default) or a new one.";
+            };
+        };
+    };
+    readonly pageSize: {
+        readonly type: "number";
+        readonly enum: readonly [25, 50, 100];
+        readonly description: "recordsTable only: rows per page (default 25).";
+    };
+};

package/dist/server/ai/discovery-helpers.d.ts

@@ -21,6 +21,20 @@ export declare function accumulateFieldScores(keyword: string, fields: Field[],
 }): number;
 /** Return the top-N field names from a score map, highest score first. */
 export declare function topScoredFields(scores: Map<string, number>, limit: number): string[];
+/**
+ * Titles for suggested fields whose name does not already say what they are.
+ *
+ * Most field names describe themselves (`Employees.salary`), so repeating the
+ * title would be noise. A user-defined (EAV) attribute is the case this exists
+ * for: it is addressed by a generated slot name like `attr_2`, and the title is
+ * the only thing that says it means "Completion %". Discovery scores against
+ * that title, so without this the caller is handed an opaque identifier and told
+ * it is relevant.
+ *
+ * Returned as a separate name-keyed map rather than annotated into the field
+ * lists, which callers are told to copy verbatim into queries.
+ */
+export declare function buildFieldTitles(fields: Field[], suggested: string[]): Record<string, string> | undefined;
 /**
  * Score a single field as a best-match candidate (used by findBestFieldMatch).
  * Returns the best score across name, title and synonyms (no description match).

package/dist/server/ai/discovery.d.ts

@@ -11,6 +11,12 @@ export interface CubeDiscoveryResult {
     matchedOn: ('name' | 'title' | 'description' | 'exampleQuestions' | 'measures' | 'dimensions')[];
     suggestedMeasures: string[];
     suggestedDimensions: string[];
+    /**
+     * Titles for suggested fields whose name does not say what they are — chiefly
+     * user-defined (EAV) attributes, addressed as `attr_<id>`. Keyed by field name
+     * so the names above stay copyable verbatim. Omitted when nothing needs it.
+     */
+    fieldTitles?: Record<string, string>;
     capabilities: {
         query: true;
         funnel: boolean;

package/dist/server/ai/query-schema.d.ts

@@ -97,7 +97,7 @@ export declare const QUERY_PARAMS_SCHEMA: {
     };
     readonly ungrouped: {
         readonly type: "boolean";
-        readonly description: "When true, returns raw row-level data without GROUP BY. Requires at least one dimension. Incompatible with count/countDistinct measures and analysis modes.";
+        readonly description: "When true, returns raw row-level data without GROUP BY. Requires at least one dimension. Incompatible with count/countDistinct measures and analysis modes. Cannot reference two cubes joined by a hasMany relationship in EITHER direction (e.g. Departments hasMany Employees) — an ungrouped query stays at one record grain, so keep it to a single cube plus its to-one joins. A field from a hasMany-related cube has to be pre-aggregated in a grouped query instead.";
     };
     readonly funnel: {
         readonly type: "object";

package/dist/server/attribute-dimensions.d.ts

@@ -81,14 +81,17 @@ export interface BuildAttributeDimensionsOptions {
  *   returned by a subquery" on Postgres.
  * - **Security inside the subquery**, not merely on the outer query.
  *
- * **Grouping limitation.** The subquery correlates on the record key, so a
- * *grouped* query must also group by that key — `dimensions: ['Employees.id',
- * 'Employees.attr_1']` is fine, `dimensions: ['Employees.attr_1']` with a
- * measure is not. Postgres rejects the latter with "subquery uses ungrouped
- * column … from outer query", and MySQL with `only_full_group_by`; SQLite
- * permits it. Record-grain listings (`ungrouped: true`) — the case this exists
- * for — are unaffected. If you need to aggregate freely by an attribute, pivot
- * it into a real column with a view (see the performance ladder below).
+ * **Grouping.** The subquery correlates on the record key, so a *grouped* query
+ * must also group by that key or Postgres rejects it with "subquery uses
+ * ungrouped column … from outer query" (MySQL likewise under
+ * `only_full_group_by`; SQLite permits it). The caller does not have to know
+ * that: each generated dimension declares its key via `Dimension.correlatesOn`,
+ * and the GROUP BY builder adds it. The grouping is therefore per record — the
+ * only grain at which a per-record attribute can be read — so an aggregate over
+ * an attribute counts records, not attribute values. If you need to aggregate
+ * freely *by* an attribute, pivot it into a real column with a view (see the
+ * performance ladder below). Record-grain listings (`ungrouped: true`) skip
+ * GROUP BY entirely and are unaffected.
  *
  * Performance: projection is cheap (a page of 25 rows is 25 indexed lookups per
  * attribute), but `ORDER BY (SELECT …)` and `WHERE (SELECT …) = …` are

package/dist/server/builders/group-by-builder.d.ts

@@ -24,6 +24,21 @@ export declare class GroupByBuilder {
      * Window functions do not require GROUP BY and operate on individual rows.
      */
     buildGroupByFields(cubes: Map<string, Cube> | Cube, query: SemanticQuery, context: QueryContext, queryPlan?: PhysicalQueryPlan): (SQL | AnyColumn)[];
+    /**
+     * Add the correlation key of any selected dimension that declares one.
+     *
+     * A dimension built from a correlated subquery (`buildAttributeDimensions`)
+     * references the record key inside its SQL. Grouping by the subquery alone is
+     * rejected by Postgres ("subquery uses ungrouped column … from outer query")
+     * and by MySQL under `only_full_group_by`, so the key it correlates on is
+     * grouped too.
+     *
+     * This only ever fires for queries that are invalid on those engines today, so
+     * nothing that currently works changes shape. The key is the record's primary
+     * key, so adding it makes the grouping per-record — which is the grain such a
+     * dimension can be read at in the first place.
+     */
+    private addCorrelationKeys;
     /**
      * Determine whether the query contains at least one aggregate (or calculated,
      * or post-aggregation window over an aggregate) measure — which forces GROUP BY.

package/dist/server/compiler.d.ts

@@ -1,6 +1,8 @@
 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
@@ -346,10 +348,7 @@ export declare class SemanticLayerCompiler {
      * Validate a query against registered cubes
      * Ensures all referenced cubes and fields exist
      */
-    validateQuery(query: SemanticQuery, securityContext: SecurityContext): {
-        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/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;