drizzle-cube 0.8.2 → 0.9.0

package/dist/server/agent/agent-prompts.d.ts

@@ -0,0 +1,14 @@
+/**
+ * Notebook-agent prompt fragments.
+ *
+ * The agent used to inline `MCP_GUIDE_PROMPT` from `../ai/mcp-prompts.js`, which
+ * documents the MCP transport's `discover` / `validate` / `load` tools. Those
+ * are real tools — over MCP. The notebook agent has a different set entirely, so
+ * the guide was naming tools it cannot call.
+ *
+ * `MCP_GUIDE_PROMPT` stays as it is: it is a public export and is served through
+ * `ALL_PROMPTS`, so real MCP clients depend on it. This is the agent's own copy.
+ * `QUERY_LANGUAGE_PROMPT` and `DATE_FILTERING_PROMPT` remain shared — they name
+ * no tools and are correct for both.
+ */
+export declare const AGENT_WORKFLOW_GUIDE: string;

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

@@ -13,6 +13,20 @@ 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;
+/**
+ * Pick a chart type that can actually render the query.
+ *
+ * A `bar` over a measures-only query has no category axis: `resolveChartAxisFields`
+ * reports `axisInvalid` and the portlet renders a config-error card. Rejecting it
+ * instead costs the model a full query+portlet rebuild, so swap to a type that
+ * shows the same numbers and tell it what happened.
+ *
+ * The note is model-facing, not user-facing, so it is deliberately not translated.
+ */
+export declare function resolveChartTypeFallback(chartType: string, chartConfig: Record<string, unknown> | undefined, query: Record<string, unknown>): {
+    chartType: string;
+    note?: string;
+};
 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/agent/data-shape.d.ts

@@ -0,0 +1,41 @@
+/**
+ * Data-shape summary for the agent's `execute_query` results.
+ *
+ * The agent used to receive every row and no summary, so it chose a chart type
+ * blind and then spent its context — and its output-token budget — on raw data.
+ * A per-field summary is what the chart decision actually needs: how many
+ * distinct categories there are, and whether measures share a scale.
+ */
+/** How many rows of the result set are handed back to the model. */
+export declare const AGENT_RESULT_ROW_LIMIT = 25;
+/** Per-field summary of a query result. */
+export interface FieldShape {
+    field: string;
+    kind: 'measure' | 'dimension' | 'timeDimension';
+    type?: string;
+    /** Distinct non-null values seen. Capped at the rows inspected. */
+    distinctCount: number;
+    nullCount: number;
+    /** Numeric range, present only when every non-null value is a number. */
+    min?: number;
+    max?: number;
+}
+interface ResultAnnotation {
+    measures?: Record<string, {
+        type?: string;
+    }>;
+    dimensions?: Record<string, {
+        type?: string;
+    }>;
+    timeDimensions?: Record<string, {
+        type?: string;
+    }>;
+}
+/**
+ * Build a per-field summary of a query result.
+ *
+ * Computed from the rows handed to the model, so `distinctCount` is a floor
+ * rather than an exact count when the result was truncated.
+ */
+export declare function summariseDataShape(rows: Record<string, unknown>[], annotation: ResultAnnotation | undefined): FieldShape[];
+export {};

package/dist/server/agent/providers/anthropic.d.ts

@@ -20,5 +20,6 @@ export declare class AnthropicProvider implements LLMProvider {
     };
     formatToolResults(results: ToolResult[]): unknown;
     shouldContinue(stopReason: string): boolean;
+    isTruncated(stopReason: string): boolean;
     formatError(error: unknown): string;
 }

package/dist/server/agent/providers/google.d.ts

@@ -19,5 +19,6 @@ export declare class GoogleProvider implements LLMProvider {
     };
     formatToolResults(results: ToolResult[]): unknown;
     shouldContinue(stopReason: string): boolean;
+    isTruncated(stopReason: string): boolean;
     formatError(error: unknown): string;
 }

package/dist/server/agent/providers/openai.d.ts

@@ -22,5 +22,6 @@ export declare class OpenAIProvider implements LLMProvider {
     };
     formatToolResults(results: ToolResult[]): unknown[];
     shouldContinue(stopReason: string): boolean;
+    isTruncated(stopReason: string): boolean;
     formatError(error: unknown): string;
 }

package/dist/server/agent/providers/types.d.ts

@@ -31,6 +31,12 @@ export interface ContentBlock {
     input?: Record<string, unknown>;
     /** Provider-specific metadata (e.g. Gemini thoughtSignature) */
     metadata?: Record<string, unknown>;
+    /**
+     * Set when the tool-call arguments could not be parsed as JSON. The handler
+     * returns an error tool_result instead of executing the tool with an empty
+     * input — running on `{}` silently produced empty notebook blocks.
+     */
+    inputParseError?: boolean;
 }
 export interface ToolResult {
     toolUseId: string;
@@ -58,6 +64,7 @@ export type NormalizedEvent = {
     type: 'tool_use_end';
     id?: string;
     input?: Record<string, unknown>;
+    parseError?: boolean;
 } | {
     type: 'message_meta';
     inputTokens?: number;
@@ -90,6 +97,12 @@ export interface LLMProvider {
     formatToolResults(results: ToolResult[]): unknown;
     /** Return true if the stop reason means the model wants to continue (tool use). */
     shouldContinue(stopReason: string): boolean;
+    /**
+     * Return true if the stop reason means the response was cut off by the output
+     * token limit rather than finished. Without this the loop breaks silently and
+     * the notebook just stops mid-analysis.
+     */
+    isTruncated(stopReason: string): boolean;
     /** Format a provider-specific error into a user-friendly message. */
     formatError(error: unknown): string;
 }

package/dist/server/agent/tools.d.ts

@@ -1,3 +1,4 @@
+import { BuiltInChartType } from '../../client/types.js';
 import { SemanticLayerCompiler } from '../compiler.js';
 import { SecurityContext } from '../types/index.js';
 import { AgentSSEEvent } from './types.js';
@@ -13,15 +14,28 @@ export interface ToolExecutionResult {
     /** Optional SSE event to emit as a side effect (add_portlet, add_markdown) */
     sideEffect?: AgentSSEEvent;
 }
+/**
+ * Chart types the notebook agent may create — an explicit list, not a derived
+ * one, so widening it stays a product decision. It now covers every
+ * `BuiltInChartType`: the config each one needs is expressible through the
+ * `add_portlet` chartConfig/displayConfig schema below, and a narrower list was
+ * making the agent reach for `bar` and `table` for everything.
+ *
+ * Two carry caveats the system prompt spells out rather than the schema:
+ * `gauge` needs an explicit `maxValue` (it otherwise scales to the data and
+ * reads as 100%), and `candlestick` needs its measures in OHLC order, which
+ * JSON Schema cannot express.
+ *
+ * Typed against `BuiltInChartType` (type-only import — no runtime dependency on
+ * the client graph) so a renamed or removed chart type fails `npm run typecheck`
+ * here instead of silently offering the model a type that no longer renders.
+ */
+export declare const AGENT_ALLOWED_CHART_TYPES: BuiltInChartType[];
 /**
  * Returns the array of tool definitions for the Anthropic Messages API.
  * Plain JSON schema — no Zod dependency.
  */
 export declare function getToolDefinitions(): ToolDefinition[];
-/**
- * Create a tool executor map that closes over the semantic layer and security context.
- * Returns a Map<toolName, handler> where each handler takes parsed input and returns a result.
- */
 export declare function createToolExecutor(options: {
     semanticLayer: SemanticLayerCompiler;
     securityContext: SecurityContext;

package/dist/server/agent/types.d.ts

@@ -100,7 +100,7 @@ export interface AgentConfig {
     model?: string;
     /** Maximum agentic turns per request (default: 25) */
     maxTurns?: number;
-    /** Maximum tokens per response (default: 4096) */
+    /** Maximum tokens per response (default: 8192) */
     maxTokens?: number;
     /** Allow X-Agent-Api-Key header to override server apiKey */
     allowClientApiKey?: boolean;
@@ -163,12 +163,16 @@ export type AgentSSEEvent = {
     data: {
         id: string;
         name: string;
+        input?: unknown;
         result?: unknown;
         isError?: boolean;
     };
 } | {
     type: 'add_portlet';
     data: PortletBlockData;
+} | {
+    type: 'update_portlet';
+    data: PortletBlockData;
 } | {
     type: 'add_markdown';
     data: MarkdownBlockData;

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

@@ -11,6 +11,43 @@
  * never guessed from the data, so a model that cannot see this shape produces a
  * listing of unformatted strings.
  */
+/**
+ * `chartConfig` fields for the charts whose mandatory drop zones are not the
+ * usual axes. Without these the model cannot state them at all: `heatmap` needs
+ * a `valueField` to colour its cells, and `activityGrid` lays itself out from a
+ * date field plus a value field. Both were only ever filled by inference.
+ */
+export declare const FIELD_ZONE_CHART_CONFIG_SCHEMA: {
+    readonly valueField: {
+        readonly type: "array";
+        readonly items: {
+            readonly type: "string";
+        };
+        readonly description: "heatmap and activityGrid: single measure driving cell colour intensity. Required for both — the chart renders empty without it.";
+    };
+    readonly dateField: {
+        readonly type: "array";
+        readonly items: {
+            readonly type: "string";
+        };
+        readonly description: "activityGrid only: single time dimension laying out the grid. Query it with day granularity over several months, or the grid has nothing to show.";
+    };
+};
+/** `displayConfig` fields for charts that need an explicit scale or template. */
+export declare const SCALE_DISPLAY_CONFIG_SCHEMA: {
+    readonly minValue: {
+        readonly type: "number";
+        readonly description: "gauge only: scale minimum (defaults to 0).";
+    };
+    readonly maxValue: {
+        readonly type: "number";
+        readonly description: "gauge only: scale maximum. ALWAYS set this — without it the gauge scales to the data it was given, so a single value always reads as 100%.";
+    };
+    readonly template: {
+        readonly type: "string";
+        readonly description: "kpiText only: sentence template around the value, e.g. \"${fieldLabel}: ${value}\". Defaults to that if omitted.";
+    };
+};
 /** `chartConfig` fields specific to the records table. */
 export declare const RECORDS_TABLE_CHART_CONFIG_SCHEMA: {
     readonly columns: {

package/dist/server/{anthropic-DG4r1LSm.cjs → anthropic-CqLkud62.cjs}

@@ -1 +1 @@
-var e=class{client;constructor(e){this.apiKey=e}apiKey;initialized=!1;async ensureClient(){if(this.initialized)return;let e;try{let t=await import(`@anthropic-ai/sdk`);e=t.default||t.Anthropic||t}catch{throw Error(`@anthropic-ai/sdk is required for the Anthropic provider. Install it with: npm install @anthropic-ai/sdk`)}this.client=new e({apiKey:this.apiKey}),this.initialized=!0}async createStream(e){await this.ensureClient();let{messages:t,system:n}=this.formatMessages(e.messages,e.system);return this.client.messages.create({model:e.model,max_tokens:e.maxTokens,system:n,tools:this.formatTools(e.tools),messages:t,stream:!0})}async*parseStreamEvents(e){for await(let t of e){let e=t;switch(e.type){case`content_block_start`:{let t=e.content_block;t.type===`tool_use`&&(yield{type:`tool_use_start`,id:t.id,name:t.name});break}case`content_block_delta`:{let t=e.delta;t.type===`text_delta`&&t.text?yield{type:`text_delta`,text:t.text}:t.type===`input_json_delta`&&t.partial_json&&(yield{type:`tool_input_delta`,json:t.partial_json});break}case`content_block_stop`:yield{type:`tool_use_end`};break;case`message_start`:{let t=e.message;t?.usage?.input_tokens!=null&&(yield{type:`message_meta`,inputTokens:t.usage.input_tokens,stopReason:``});break}case`message_delta`:{let t=e.usage,n=e.delta?.stop_reason||``;yield{type:`message_meta`,outputTokens:t?.output_tokens,stopReason:n};break}}}}formatTools(e){return e.map(e=>({name:e.name,description:e.description,input_schema:e.parameters}))}formatMessages(e,t){return{messages:e.map(e=>e.role===`tool_result`?{role:`user`,content:e.content}:{role:e.role,content:e.content}),system:t}}formatToolResults(e){return{role:`user`,content:e.map(e=>({type:`tool_result`,tool_use_id:e.toolUseId,content:e.content,...e.isError?{is_error:!0}:{}}))}}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={overloaded_error:`The AI service is temporarily overloaded. Please try again in a moment.`,rate_limit_error:`Too many requests. Please wait a moment and try again.`,api_error:`The AI service encountered an error. Please try again.`,authentication_error:`Authentication failed. Please check your API key configuration.`,invalid_request_error:`There was a problem with the request. Please try again.`},r=e;if(r.status||r.type){let e=r.error?.type||r.type||``;if(n[e])return n[e]}if(t.startsWith(`{`)||t.startsWith(`Error: {`)){try{let e=JSON.parse(t.replace(/^Error:\s*/,``)),r=e.error?.type||e.type||``;if(n[r])return n[r]}catch{}return`The AI service encountered an error. Please try again.`}return t}};exports.AnthropicProvider=e;
+var e=class{client;constructor(e){this.apiKey=e}apiKey;initialized=!1;async ensureClient(){if(this.initialized)return;let e;try{let t=await import(`@anthropic-ai/sdk`);e=t.default||t.Anthropic||t}catch{throw Error(`@anthropic-ai/sdk is required for the Anthropic provider. Install it with: npm install @anthropic-ai/sdk`)}this.client=new e({apiKey:this.apiKey}),this.initialized=!0}async createStream(e){await this.ensureClient();let{messages:t,system:n}=this.formatMessages(e.messages,e.system);return this.client.messages.create({model:e.model,max_tokens:e.maxTokens,system:n,tools:this.formatTools(e.tools),messages:t,stream:!0})}async*parseStreamEvents(e){for await(let t of e){let e=t;switch(e.type){case`content_block_start`:{let t=e.content_block;t.type===`tool_use`&&(yield{type:`tool_use_start`,id:t.id,name:t.name});break}case`content_block_delta`:{let t=e.delta;t.type===`text_delta`&&t.text?yield{type:`text_delta`,text:t.text}:t.type===`input_json_delta`&&t.partial_json&&(yield{type:`tool_input_delta`,json:t.partial_json});break}case`content_block_stop`:yield{type:`tool_use_end`};break;case`message_start`:{let t=e.message;t?.usage?.input_tokens!=null&&(yield{type:`message_meta`,inputTokens:t.usage.input_tokens,stopReason:``});break}case`message_delta`:{let t=e.usage,n=e.delta?.stop_reason||``;yield{type:`message_meta`,outputTokens:t?.output_tokens,stopReason:n};break}}}}formatTools(e){return e.map(e=>({name:e.name,description:e.description,input_schema:e.parameters}))}formatMessages(e,t){return{messages:e.map(e=>e.role===`tool_result`?{role:`user`,content:e.content}:{role:e.role,content:e.content}),system:t}}formatToolResults(e){return{role:`user`,content:e.map(e=>({type:`tool_result`,tool_use_id:e.toolUseId,content:e.content,...e.isError?{is_error:!0}:{}}))}}shouldContinue(e){return e===`tool_use`}isTruncated(e){return e===`max_tokens`}formatError(e){if(!e||!(e instanceof Error))return`Something went wrong. Please try again.`;let t=e.message||``,n={overloaded_error:`The AI service is temporarily overloaded. Please try again in a moment.`,rate_limit_error:`Too many requests. Please wait a moment and try again.`,api_error:`The AI service encountered an error. Please try again.`,authentication_error:`Authentication failed. Please check your API key configuration.`,invalid_request_error:`There was a problem with the request. Please try again.`},r=e;if(r.status||r.type){let e=r.error?.type||r.type||``;if(n[e])return n[e]}if(t.startsWith(`{`)||t.startsWith(`Error: {`)){try{let e=JSON.parse(t.replace(/^Error:\s*/,``)),r=e.error?.type||e.type||``;if(n[r])return n[r]}catch{}return`The AI service encountered an error. Please try again.`}return t}};exports.AnthropicProvider=e;

package/dist/server/{anthropic-CZh4hB-m.js → anthropic-MQlafOZv.js}

@@ -113,6 +113,9 @@ var e = class {
 	shouldContinue(e) {
 		return e === "tool_use";
 	}
+	isTruncated(e) {
+		return e === "max_tokens";
+	}
 	formatError(e) {
 		if (!e || !(e instanceof Error)) return "Something went wrong. Please try again.";
 		let t = e.message || "", n = {

package/dist/{adapters/google-PsZzwedr.cjs → server/google-6pofxIfm.cjs}

@@ -1,2 +1,2 @@
 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;
+`),_toolResults:e}}shouldContinue(e){return e===`tool_use`}isTruncated(e){return e===`max_tokens`}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;

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

@@ -174,6 +174,9 @@ var c = class {
 	shouldContinue(e) {
 		return e === "tool_use";
 	}
+	isTruncated(e) {
+		return e === "max_tokens";
+	}
 	formatError(e) {
 		if (!e || !(e instanceof Error)) return "Something went wrong. Please try again.";
 		let t = e.message || "", n = e;