drizzle-cube 0.4.20 → 0.4.21

package/dist/adapters/fastify/index.cjs

@@ -1,11 +1,11 @@
-"use strict";var O=Object.create;var j=Object.defineProperty;var k=Object.getOwnPropertyDescriptor;var J=Object.getOwnPropertyNames;var L=Object.getPrototypeOf,D=Object.prototype.hasOwnProperty;var H=(l,n,v,R)=>{if(n&&typeof n=="object"||typeof n=="function")for(let f of J(n))!D.call(l,f)&&f!==v&&j(l,f,{get:()=>n[f],enumerable:!(R=k(n,f))||R.enumerable});return l};var T=(l,n,v)=>(v=l!=null?O(L(l)):{},H(n||!l||!l.__esModule?j(v,"default",{value:l,enumerable:!0}):v,l));Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const u=require("../mcp-transport-8u9G5oNa.cjs"),a=require("../utils.cjs"),q=function(n,v,R){const{cubes:f,drizzle:I,schema:A,extractSecurityContext:g,engineType:M,cors:P,basePath:h="/cubejs-api/v1",bodyLimit:C=10485760,cache:N,mcp:x={enabled:!0},agent:w}=v;if(!f||f.length===0)return R(new Error("At least one cube must be provided in the cubes array"));P&&n.register(import("@fastify/cors"),P),n.addHook("onRequest",async(r,t)=>{r.method==="POST"&&(r.body=void 0)});const d=new u.SemanticLayerCompiler({drizzle:I,schema:A,engineType:M,cache:N});if(f.forEach(r=>{d.registerCube(r)}),n.post(`${h}/load`,{bodyLimit:C,schema:{body:{type:"object",additionalProperties:!0}}},async(r,t)=>{try{const e=r.body,o=e.query||e,s=await g(r),i=d.validateQuery(o);if(!i.isValid)return t.status(400).send(a.formatErrorResponse(`Query validation failed: ${i.errors.join(", ")}`,400));const c=r.headers["x-cache-control"]==="no-cache",p=await d.executeMultiCubeQuery(o,s,{skipCache:c});return a.formatCubeResponse(o,p,d)}catch(e){return r.log.error(e,"Query execution error"),t.status(500).send(a.formatErrorResponse(e instanceof Error?e.message:"Query execution failed",500))}}),n.get(`${h}/load`,{schema:{querystring:{type:"object",properties:{query:{type:"string"}},required:["query"]}}},async(r,t)=>{try{const{query:e}=r.query;let o;try{o=JSON.parse(e)}catch{return t.status(400).send(a.formatErrorResponse("Invalid JSON in query parameter",400))}const s=await g(r),i=d.validateQuery(o);if(!i.isValid)return t.status(400).send(a.formatErrorResponse(`Query validation failed: ${i.errors.join(", ")}`,400));const c=r.headers["x-cache-control"]==="no-cache",p=await d.executeMultiCubeQuery(o,s,{skipCache:c});return a.formatCubeResponse(o,p,d)}catch(e){return r.log.error(e,"Query execution error"),t.status(500).send(a.formatErrorResponse(e instanceof Error?e.message:"Query execution failed",500))}}),n.post(`${h}/batch`,{bodyLimit:C,schema:{body:{type:"object",required:["queries"],properties:{queries:{type:"array",items:{type:"object"}}}}}},async(r,t)=>{try{const{queries:e}=r.body;if(!e||!Array.isArray(e))return t.status(400).send(a.formatErrorResponse('Request body must contain a "queries" array',400));if(e.length===0)return t.status(400).send(a.formatErrorResponse("Queries array cannot be empty",400));const o=await g(r),s=r.headers["x-cache-control"]==="no-cache";return await a.handleBatchRequest(e,o,d,{skipCache:s})}catch(e){return r.log.error(e,"Batch execution error"),t.status(500).send(a.formatErrorResponse(e instanceof Error?e.message:"Batch execution failed",500))}}),n.get(`${h}/meta`,async(r,t)=>{try{const e=d.getMetadata();return a.formatMetaResponse(e)}catch(e){return r.log.error(e,"Metadata error"),t.status(500).send(a.formatErrorResponse(e instanceof Error?e.message:"Failed to fetch metadata",500))}}),n.post(`${h}/sql`,{bodyLimit:C,schema:{body:{type:"object",additionalProperties:!0}}},async(r,t)=>{try{const e=r.body,o=await g(r),s=d.validateQuery(e);if(!s.isValid)return t.status(400).send(a.formatErrorResponse(`Query validation failed: ${s.errors.join(", ")}`,400));const i=e.measures?.[0]||e.dimensions?.[0];if(!i)return t.status(400).send(a.formatErrorResponse("No measures or dimensions specified",400));const c=i.split(".")[0],p=await d.generateSQL(c,e,o);return a.formatSqlResponse(e,p)}catch(e){return r.log.error(e,"SQL generation error"),t.status(500).send(a.formatErrorResponse(e instanceof Error?e.message:"SQL generation failed",500))}}),n.get(`${h}/sql`,{schema:{querystring:{type:"object",properties:{query:{type:"string"}},required:["query"]}}},async(r,t)=>{try{const{query:e}=r.query,o=JSON.parse(e),s=await g(r),i=d.validateQuery(o);if(!i.isValid)return t.status(400).send(a.formatErrorResponse(`Query validation failed: ${i.errors.join(", ")}`,400));const c=o.measures?.[0]||o.dimensions?.[0];if(!c)return t.status(400).send(a.formatErrorResponse("No measures or dimensions specified",400));const p=c.split(".")[0],E=await d.generateSQL(p,o,s);return a.formatSqlResponse(o,E)}catch(e){return r.log.error(e,"SQL generation error"),t.status(500).send(a.formatErrorResponse(e instanceof Error?e.message:"SQL generation failed",500))}}),n.post(`${h}/dry-run`,{bodyLimit:C,schema:{body:{type:"object",additionalProperties:!0}}},async(r,t)=>{try{const e=r.body,o=e.query||e,s=await g(r);return await a.handleDryRun(o,s,d)}catch(e){return r.log.error(e,"Dry-run error"),t.status(400).send({error:e instanceof Error?e.message:"Dry-run validation failed",valid:!1})}}),n.get(`${h}/dry-run`,{schema:{querystring:{type:"object",properties:{query:{type:"string"}},required:["query"]}}},async(r,t)=>{try{const{query:e}=r.query,o=JSON.parse(e),s=await g(r);return await a.handleDryRun(o,s,d)}catch(e){return r.log.error(e,"Dry-run error"),t.status(400).send({error:e instanceof Error?e.message:"Dry-run validation failed",valid:!1})}}),n.post(`${h}/explain`,{bodyLimit:C,schema:{body:{type:"object",additionalProperties:!0}}},async(r,t)=>{try{const e=r.body,o=e.query||e,s=e.options||{},i=await g(r),c=d.validateQuery(o);return c.isValid?await d.explainQuery(o,i,s):t.status(400).send({error:`Query validation failed: ${c.errors.join(", ")}`})}catch(e){return r.log.error(e,"Explain error"),t.status(500).send({error:e instanceof Error?e.message:"Explain query failed"})}}),w&&n.post(`${h}/agent/chat`,{bodyLimit:C,schema:{body:{type:"object",additionalProperties:!0}}},async(r,t)=>{try{const{handleAgentChat:e}=await Promise.resolve().then(()=>require("../handler-DtdjM1Vx.cjs")),o=r.body,{message:s,sessionId:i,history:c}=o;if(!s||typeof s!="string")return t.status(400).send({error:"message is required and must be a string"});let p=(w.apiKey||"").trim();if(w.allowClientApiKey){const m=r.headers["x-agent-api-key"];m&&(p=m.trim())}if(!p)return t.status(401).send({error:"No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header."});const E=await g(r),y=w.buildSystemContext?.(E);t.raw.writeHead(200,{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive"});try{const m=e({message:s,sessionId:i,history:c,semanticLayer:d,securityContext:E,agentConfig:w,apiKey:p,systemContext:y});for await(const b of m)t.raw.write(`data: ${JSON.stringify(b)}
+"use strict";var N=Object.create;var j=Object.defineProperty;var k=Object.getOwnPropertyDescriptor;var J=Object.getOwnPropertyNames;var L=Object.getPrototypeOf,K=Object.prototype.hasOwnProperty;var D=(l,n,v,S)=>{if(n&&typeof n=="object"||typeof n=="function")for(let f of J(n))!K.call(l,f)&&f!==v&&j(l,f,{get:()=>n[f],enumerable:!(S=k(n,f))||S.enumerable});return l};var H=(l,n,v)=>(v=l!=null?N(L(l)):{},D(n||!l||!l.__esModule?j(v,"default",{value:l,enumerable:!0}):v,l));Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const d=require("../mcp-transport-8u9G5oNa.cjs"),a=require("../utils.cjs"),Q=function(n,v,S){const{cubes:f,drizzle:A,schema:I,extractSecurityContext:m,engineType:O,cors:$,basePath:g="/cubejs-api/v1",bodyLimit:C=10485760,cache:M,mcp:q={enabled:!0},agent:b}=v;if(!f||f.length===0)return S(new Error("At least one cube must be provided in the cubes array"));$&&n.register(import("@fastify/cors"),$),n.addHook("onRequest",async(r,t)=>{r.method==="POST"&&(r.body=void 0)});const u=new d.SemanticLayerCompiler({drizzle:A,schema:I,engineType:O,cache:M});if(f.forEach(r=>{u.registerCube(r)}),n.post(`${g}/load`,{bodyLimit:C,schema:{body:{type:"object",additionalProperties:!0}}},async(r,t)=>{try{const e=r.body,o=e.query||e,s=await m(r),i=u.validateQuery(o);if(!i.isValid)return t.status(400).send(a.formatErrorResponse(`Query validation failed: ${i.errors.join(", ")}`,400));const c=r.headers["x-cache-control"]==="no-cache",p=await u.executeMultiCubeQuery(o,s,{skipCache:c});return a.formatCubeResponse(o,p,u)}catch(e){return r.log.error(e,"Query execution error"),t.status(500).send(a.formatErrorResponse(e instanceof Error?e.message:"Query execution failed",500))}}),n.get(`${g}/load`,{schema:{querystring:{type:"object",properties:{query:{type:"string"}},required:["query"]}}},async(r,t)=>{try{const{query:e}=r.query;let o;try{o=JSON.parse(e)}catch{return t.status(400).send(a.formatErrorResponse("Invalid JSON in query parameter",400))}const s=await m(r),i=u.validateQuery(o);if(!i.isValid)return t.status(400).send(a.formatErrorResponse(`Query validation failed: ${i.errors.join(", ")}`,400));const c=r.headers["x-cache-control"]==="no-cache",p=await u.executeMultiCubeQuery(o,s,{skipCache:c});return a.formatCubeResponse(o,p,u)}catch(e){return r.log.error(e,"Query execution error"),t.status(500).send(a.formatErrorResponse(e instanceof Error?e.message:"Query execution failed",500))}}),n.post(`${g}/batch`,{bodyLimit:C,schema:{body:{type:"object",required:["queries"],properties:{queries:{type:"array",items:{type:"object"}}}}}},async(r,t)=>{try{const{queries:e}=r.body;if(!e||!Array.isArray(e))return t.status(400).send(a.formatErrorResponse('Request body must contain a "queries" array',400));if(e.length===0)return t.status(400).send(a.formatErrorResponse("Queries array cannot be empty",400));const o=await m(r),s=r.headers["x-cache-control"]==="no-cache";return await a.handleBatchRequest(e,o,u,{skipCache:s})}catch(e){return r.log.error(e,"Batch execution error"),t.status(500).send(a.formatErrorResponse(e instanceof Error?e.message:"Batch execution failed",500))}}),n.get(`${g}/meta`,async(r,t)=>{try{const e=u.getMetadata();return a.formatMetaResponse(e)}catch(e){return r.log.error(e,"Metadata error"),t.status(500).send(a.formatErrorResponse(e instanceof Error?e.message:"Failed to fetch metadata",500))}}),n.post(`${g}/sql`,{bodyLimit:C,schema:{body:{type:"object",additionalProperties:!0}}},async(r,t)=>{try{const e=r.body,o=await m(r),s=u.validateQuery(e);if(!s.isValid)return t.status(400).send(a.formatErrorResponse(`Query validation failed: ${s.errors.join(", ")}`,400));const i=e.measures?.[0]||e.dimensions?.[0];if(!i)return t.status(400).send(a.formatErrorResponse("No measures or dimensions specified",400));const c=i.split(".")[0],p=await u.generateSQL(c,e,o);return a.formatSqlResponse(e,p)}catch(e){return r.log.error(e,"SQL generation error"),t.status(500).send(a.formatErrorResponse(e instanceof Error?e.message:"SQL generation failed",500))}}),n.get(`${g}/sql`,{schema:{querystring:{type:"object",properties:{query:{type:"string"}},required:["query"]}}},async(r,t)=>{try{const{query:e}=r.query,o=JSON.parse(e),s=await m(r),i=u.validateQuery(o);if(!i.isValid)return t.status(400).send(a.formatErrorResponse(`Query validation failed: ${i.errors.join(", ")}`,400));const c=o.measures?.[0]||o.dimensions?.[0];if(!c)return t.status(400).send(a.formatErrorResponse("No measures or dimensions specified",400));const p=c.split(".")[0],P=await u.generateSQL(p,o,s);return a.formatSqlResponse(o,P)}catch(e){return r.log.error(e,"SQL generation error"),t.status(500).send(a.formatErrorResponse(e instanceof Error?e.message:"SQL generation failed",500))}}),n.post(`${g}/dry-run`,{bodyLimit:C,schema:{body:{type:"object",additionalProperties:!0}}},async(r,t)=>{try{const e=r.body,o=e.query||e,s=await m(r);return await a.handleDryRun(o,s,u)}catch(e){return r.log.error(e,"Dry-run error"),t.status(400).send({error:e instanceof Error?e.message:"Dry-run validation failed",valid:!1})}}),n.get(`${g}/dry-run`,{schema:{querystring:{type:"object",properties:{query:{type:"string"}},required:["query"]}}},async(r,t)=>{try{const{query:e}=r.query,o=JSON.parse(e),s=await m(r);return await a.handleDryRun(o,s,u)}catch(e){return r.log.error(e,"Dry-run error"),t.status(400).send({error:e instanceof Error?e.message:"Dry-run validation failed",valid:!1})}}),n.post(`${g}/explain`,{bodyLimit:C,schema:{body:{type:"object",additionalProperties:!0}}},async(r,t)=>{try{const e=r.body,o=e.query||e,s=e.options||{},i=await m(r),c=u.validateQuery(o);return c.isValid?await u.explainQuery(o,i,s):t.status(400).send({error:`Query validation failed: ${c.errors.join(", ")}`})}catch(e){return r.log.error(e,"Explain error"),t.status(500).send({error:e instanceof Error?e.message:"Explain query failed"})}}),b&&n.post(`${g}/agent/chat`,{bodyLimit:C,schema:{body:{type:"object",additionalProperties:!0}}},async(r,t)=>{try{const{handleAgentChat:e}=await Promise.resolve().then(()=>require("../handler-B-tEntiU.cjs")),o=r.body,{message:s,sessionId:i,history:c}=o;if(!s||typeof s!="string")return t.status(400).send({error:"message is required and must be a string"});let p=(b.apiKey||"").trim();if(b.allowClientApiKey){const h=r.headers["x-agent-api-key"];h&&(p=h.trim())}if(!p)return t.status(401).send({error:"No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header."});const P=b.allowClientApiKey?r.headers["x-agent-provider"]:void 0,y=b.allowClientApiKey?r.headers["x-agent-model"]:void 0,E=b.allowClientApiKey?r.headers["x-agent-provider-endpoint"]:void 0,w=await m(r),R=b.buildSystemContext?.(w);t.raw.writeHead(200,{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive"});try{const h=e({message:s,sessionId:i,history:c,semanticLayer:u,securityContext:w,agentConfig:b,apiKey:p,systemContext:R,providerOverride:P,modelOverride:y,baseURLOverride:E});for await(const x of h)t.raw.write(`data: ${JSON.stringify(x)}
 
-`)}catch(m){const b={type:"error",data:{message:m instanceof Error?m.message:"Stream failed"}};t.raw.write(`data: ${JSON.stringify(b)}
+`)}catch(h){const x={type:"error",data:{message:h instanceof Error?h.message:"Stream failed"}};t.raw.write(`data: ${JSON.stringify(x)}
 
-`)}finally{t.raw.end()}}catch(e){if(r.log.error(e,"Agent chat error"),!t.raw.headersSent)return t.status(500).send({error:e instanceof Error?e.message:"Agent chat failed"})}}),x.enabled!==!1){const r=x.basePath??"/mcp";n.post(`${r}`,{bodyLimit:C,schema:{body:{type:"object",additionalProperties:!0}}},async(t,e)=>{const o=u.validateOriginHeader(t.headers.origin,x.allowedOrigins?{allowedOrigins:x.allowedOrigins}:{});if(!o.valid)return e.status(403).send(u.buildJsonRpcError(null,-32600,o.reason));const s=t.headers.accept;if(!u.validateAcceptHeader(s))return e.status(400).send(u.buildJsonRpcError(null,-32600,"Accept header must include both application/json and text/event-stream"));const i=u.negotiateProtocol(t.headers);if(!i.ok)return e.status(426).send({error:"Unsupported MCP protocol version",supported:i.supported});const c=u.parseJsonRpc(t.body);if(!c)return e.status(400).send(u.buildJsonRpcError(null,-32600,"Invalid JSON-RPC 2.0 request"));const p=u.wantsEventStream(s),E=c.method==="initialize";try{const y=await u.dispatchMcpMethod(c.method,c.params,{semanticLayer:d,extractSecurityContext:g,rawRequest:t,rawResponse:e,negotiatedProtocol:i.negotiated});if(u.isNotification(c))return e.status(202).send();const m=E&&y&&typeof y=="object"&&"sessionId"in y?y.sessionId:void 0;m&&e.header(u.MCP_SESSION_ID_HEADER,m);const b=u.buildJsonRpcResult(c.id??null,y);if(p){const S=u.primeEventId();e.header("Content-Type","text/event-stream").header("Cache-Control","no-cache").header("Connection","keep-alive").send(`id: ${S}
+`)}finally{t.raw.end()}}catch(e){if(r.log.error(e,"Agent chat error"),!t.raw.headersSent)return t.status(500).send({error:e instanceof Error?e.message:"Agent chat failed"})}}),q.enabled!==!1){const r=q.basePath??"/mcp";n.post(`${r}`,{bodyLimit:C,schema:{body:{type:"object",additionalProperties:!0}}},async(t,e)=>{const o=d.validateOriginHeader(t.headers.origin,q.allowedOrigins?{allowedOrigins:q.allowedOrigins}:{});if(!o.valid)return e.status(403).send(d.buildJsonRpcError(null,-32600,o.reason));const s=t.headers.accept;if(!d.validateAcceptHeader(s))return e.status(400).send(d.buildJsonRpcError(null,-32600,"Accept header must include both application/json and text/event-stream"));const i=d.negotiateProtocol(t.headers);if(!i.ok)return e.status(426).send({error:"Unsupported MCP protocol version",supported:i.supported});const c=d.parseJsonRpc(t.body);if(!c)return e.status(400).send(d.buildJsonRpcError(null,-32600,"Invalid JSON-RPC 2.0 request"));const p=d.wantsEventStream(s),P=c.method==="initialize";try{const y=await d.dispatchMcpMethod(c.method,c.params,{semanticLayer:u,extractSecurityContext:m,rawRequest:t,rawResponse:e,negotiatedProtocol:i.negotiated});if(d.isNotification(c))return e.status(202).send();const E=P&&y&&typeof y=="object"&&"sessionId"in y?y.sessionId:void 0;E&&e.header(d.MCP_SESSION_ID_HEADER,E);const w=d.buildJsonRpcResult(c.id??null,y);if(p){const R=d.primeEventId();e.header("Content-Type","text/event-stream").header("Cache-Control","no-cache").header("Connection","keep-alive").send(`id: ${R}
 
-${u.serializeSseEvent(b,S)}`);return}return e.send(b)}catch(y){if(u.isNotification(c))return t.log.error(y,"MCP notification processing error"),e.status(202).send();t.log.error(y,"MCP RPC error");const m=y?.code??-32603,b=y?.data,S=y.message||"MCP request failed",Q=u.buildJsonRpcError(c.id??null,m,S,b);if(p){const $=u.primeEventId();e.header("Content-Type","text/event-stream").header("Cache-Control","no-cache").header("Connection","keep-alive").send(`id: ${$}
+${d.serializeSseEvent(w,R)}`);return}return e.send(w)}catch(y){if(d.isNotification(c))return t.log.error(y,"MCP notification processing error"),e.status(202).send();t.log.error(y,"MCP RPC error");const E=y?.code??-32603,w=y?.data,R=y.message||"MCP request failed",h=d.buildJsonRpcError(c.id??null,E,R,w);if(p){const x=d.primeEventId();e.header("Content-Type","text/event-stream").header("Cache-Control","no-cache").header("Connection","keep-alive").send(`id: ${x}
 
-${u.serializeSseEvent(Q,$)}`);return}return e.send(Q)}}),n.get(`${r}`,async(t,e)=>{const o=u.primeEventId();e.raw.writeHead(200,{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive"}),e.raw.write(u.serializeSseEvent({jsonrpc:"2.0",method:"mcp/ready",params:{protocol:"streamable-http"}},o,15e3));const s=setInterval(()=>{e.raw.write(`: keep-alive
+${d.serializeSseEvent(h,x)}`);return}return e.send(h)}}),n.get(`${r}`,async(t,e)=>{const o=d.primeEventId();e.raw.writeHead(200,{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive"}),e.raw.write(d.serializeSseEvent({jsonrpc:"2.0",method:"mcp/ready",params:{protocol:"streamable-http"}},o,15e3));const s=setInterval(()=>{e.raw.write(`: keep-alive
 
-`)},15e3);t.raw.on("close",()=>{clearInterval(s)})}),n.delete(`${r}`,async(t,e)=>e.status(405).send({error:"Session termination not supported"}))}n.setErrorHandler(async(r,t,e)=>{t.log.error(r,"Fastify cube adapter error"),e.statusCode<400&&e.status(500);const o=r instanceof Error?r:String(r);return a.formatErrorResponse(o,e.statusCode)}),R()};async function z(l,n){await l.register(q,n)}function K(l){const n=require("fastify")({logger:!0});return n.register(q,l),n}exports.createCubeApp=K;exports.cubePlugin=q;exports.registerCubeRoutes=z;
+`)},15e3);t.raw.on("close",()=>{clearInterval(s)})}),n.delete(`${r}`,async(t,e)=>e.status(405).send({error:"Session termination not supported"}))}n.setErrorHandler(async(r,t,e)=>{t.log.error(r,"Fastify cube adapter error"),e.statusCode<400&&e.status(500);const o=r instanceof Error?r:String(r);return a.formatErrorResponse(o,e.statusCode)}),S()};async function T(l,n){await l.register(Q,n)}function z(l){const n=require("fastify")({logger:!0});return n.register(Q,l),n}exports.createCubeApp=z;exports.cubePlugin=Q;exports.registerCubeRoutes=T;

package/dist/adapters/fastify/index.js

@@ -1,34 +1,34 @@
-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 d, 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: p,
+    extractSecurityContext: y,
     engineType: L,
     cors: q,
-    basePath: m = "/cubejs-api/v1",
+    basePath: p = "/cubejs-api/v1",
     bodyLimit: g = 10485760,
     // 10MB
-    cache: H,
-    mcp: C = { enabled: !0 },
-    agent: f
-  } = 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(`${m}/load`, {
+  }), i.post(`${p}/load`, {
     bodyLimit: g,
     schema: {
       body: {
@@ -38,13 +38,13 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const e = t.body, a = e.query || e, n = await p(t), s = c.validateQuery(a);
-      if (!s.isValid)
+      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: ${s.errors.join(", ")}`,
+          `Query validation failed: ${o.errors.join(", ")}`,
           400
         ));
-      const o = t.headers["x-cache-control"] === "no-cache", u = await c.executeMultiCubeQuery(a, n, { skipCache: o });
+      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(d(
@@ -52,7 +52,7 @@ const N = function(i, k, R) {
         500
       ));
     }
-  }), i.get(`${m}/load`, {
+  }), i.get(`${p}/load`, {
     schema: {
       querystring: {
         type: "object",
@@ -74,13 +74,13 @@ const N = function(i, k, R) {
           400
         ));
       }
-      const n = await p(t), s = c.validateQuery(a);
-      if (!s.isValid)
+      const n = await y(t), o = c.validateQuery(a);
+      if (!o.isValid)
         return r.status(400).send(d(
-          `Query validation failed: ${s.errors.join(", ")}`,
+          `Query validation failed: ${o.errors.join(", ")}`,
           400
         ));
-      const o = t.headers["x-cache-control"] === "no-cache", u = await c.executeMultiCubeQuery(a, n, { skipCache: o });
+      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(d(
@@ -88,7 +88,7 @@ const N = function(i, k, R) {
         500
       ));
     }
-  }), i.post(`${m}/batch`, {
+  }), i.post(`${p}/batch`, {
     bodyLimit: g,
     schema: {
       body: {
@@ -115,7 +115,7 @@ const N = function(i, k, R) {
           "Queries array cannot be empty",
           400
         ));
-      const a = await p(t), n = t.headers["x-cache-control"] === "no-cache";
+      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(d(
@@ -123,7 +123,7 @@ const N = function(i, k, R) {
         500
       ));
     }
-  }), i.get(`${m}/meta`, async (t, r) => {
+  }), i.get(`${p}/meta`, async (t, r) => {
     try {
       const e = c.getMetadata();
       return G(e);
@@ -133,7 +133,7 @@ const N = function(i, k, R) {
         500
       ));
     }
-  }), i.post(`${m}/sql`, {
+  }), i.post(`${p}/sql`, {
     bodyLimit: g,
     schema: {
       body: {
@@ -143,19 +143,19 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const e = t.body, a = await p(t), n = c.validateQuery(e);
+      const e = t.body, a = await y(t), n = c.validateQuery(e);
       if (!n.isValid)
         return r.status(400).send(d(
           `Query validation failed: ${n.errors.join(", ")}`,
           400
         ));
-      const s = e.measures?.[0] || e.dimensions?.[0];
-      if (!s)
+      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], u = await c.generateSQL(o, e, a);
+      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(d(
@@ -163,7 +163,7 @@ const N = function(i, k, R) {
         500
       ));
     }
-  }), i.get(`${m}/sql`, {
+  }), i.get(`${p}/sql`, {
     schema: {
       querystring: {
         type: "object",
@@ -175,27 +175,27 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const { query: e } = t.query, a = JSON.parse(e), n = await p(t), s = c.validateQuery(a);
-      if (!s.isValid)
+      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: ${s.errors.join(", ")}`,
+          `Query validation failed: ${o.errors.join(", ")}`,
           400
         ));
-      const o = a.measures?.[0] || a.dimensions?.[0];
-      if (!o)
+      const s = a.measures?.[0] || a.dimensions?.[0];
+      if (!s)
         return r.status(400).send(d(
           "No measures or dimensions specified",
           400
         ));
-      const u = o.split(".")[0], b = await c.generateSQL(u, a, n);
-      return I(a, b);
+      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(d(
         e instanceof Error ? e.message : "SQL generation failed",
         500
       ));
     }
-  }), i.post(`${m}/dry-run`, {
+  }), i.post(`${p}/dry-run`, {
     bodyLimit: g,
     schema: {
       body: {
@@ -205,15 +205,15 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const e = t.body, a = e.query || e, n = await p(t);
-      return await M(a, n, c);
+      const e = t.body, a = e.query || e, n = await y(t);
+      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",
         valid: !1
       });
     }
-  }), i.get(`${m}/dry-run`, {
+  }), i.get(`${p}/dry-run`, {
     schema: {
       querystring: {
         type: "object",
@@ -225,15 +225,15 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const { query: e } = t.query, a = JSON.parse(e), n = await p(t);
-      return await M(a, n, c);
+      const { query: e } = t.query, a = JSON.parse(e), n = await y(t);
+      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",
         valid: !1
       });
     }
-  }), i.post(`${m}/explain`, {
+  }), i.post(`${p}/explain`, {
     bodyLimit: g,
     schema: {
       body: {
@@ -243,16 +243,16 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const e = t.body, a = e.query || e, n = e.options || {}, s = await p(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"
       });
     }
-  }), f && i.post(`${m}/agent/chat`, {
+  }), h && i.post(`${p}/agent/chat`, {
     bodyLimit: g,
     schema: {
       body: {
@@ -262,45 +262,48 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const { handleAgentChat: e } = await import("../handler-CbDMdSY5.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 u = (f.apiKey || "").trim();
-      if (f.allowClientApiKey) {
-        const y = t.headers["x-agent-api-key"];
-        y && (u = y.trim());
+      let u = (h.apiKey || "").trim();
+      if (h.allowClientApiKey) {
+        const m = t.headers["x-agent-api-key"];
+        m && (u = m.trim());
       }
       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 b = await p(t), l = f.buildSystemContext?.(b);
+      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 y = e({
+        const m = e({
           message: n,
-          sessionId: s,
-          history: o,
+          sessionId: o,
+          history: s,
           semanticLayer: c,
           securityContext: b,
-          agentConfig: f,
+          agentConfig: h,
           apiKey: u,
-          systemContext: l
+          systemContext: f,
+          providerOverride: w,
+          modelOverride: l,
+          baseURLOverride: v
         });
-        for await (const h of y)
-          r.raw.write(`data: ${JSON.stringify(h)}
+        for await (const C of m)
+          r.raw.write(`data: ${JSON.stringify(C)}
 
 `);
-      } catch (y) {
-        const h = {
+      } catch (m) {
+        const C = {
           type: "error",
-          data: { message: y instanceof Error ? y.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 {
@@ -312,8 +315,8 @@ const N = function(i, k, R) {
           error: e instanceof Error ? e.message : "Agent chat failed"
         });
     }
-  }), C.enabled !== !1) {
-    const t = C.basePath ?? "/mcp";
+  }), S.enabled !== !1) {
+    const t = S.basePath ?? "/mcp";
     i.post(`${t}`, {
       bodyLimit: g,
       schema: {
@@ -323,71 +326,71 @@ const N = function(i, k, R) {
         }
       }
     }, async (r, e) => {
-      const a = K(
+      const a = D(
         r.headers.origin,
-        C.allowedOrigins ? { allowedOrigins: C.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 u = _(n), b = 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 l = await B(
-          o.method,
-          o.params,
+          s.method,
+          s.params,
           {
             semanticLayer: c,
-            extractSecurityContext: p,
+            extractSecurityContext: y,
             rawRequest: r,
             rawResponse: e,
-            negotiatedProtocol: s.negotiated
+            negotiatedProtocol: o.negotiated
           }
         );
-        if (j(o))
+        if (j(s))
           return e.status(202).send();
-        const y = b && l && typeof l == "object" && "sessionId" in l ? l.sessionId : void 0;
-        y && e.header(F, y);
-        const h = U(o.id ?? null, l);
+        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 w = E();
-          e.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${w}
+          const f = P();
+          e.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${f}
 
-${P(h, w)}`);
+${Q(b, f)}`);
           return;
         }
-        return e.send(h);
+        return e.send(b);
       } catch (l) {
-        if (j(o))
+        if (j(s))
           return r.log.error(l, "MCP notification processing error"), e.status(202).send();
         r.log.error(l, "MCP RPC error");
-        const y = l?.code ?? -32603, h = l?.data, w = l.message || "MCP request failed", Q = x(o.id ?? null, y, w, h);
+        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 $ = E();
-          e.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${$}
+          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" }
@@ -406,19 +409,19 @@ ${P(Q, $)}`);
     r.log.error(t, "Fastify cube adapter error"), e.statusCode < 400 && e.status(500);
     const a = t instanceof Error ? t : String(t);
     return d(a, e.statusCode);
-  }), R();
+  }), $();
 };
-async function Z(v, i) {
-  await v.register(N, i);
+async function Z(x, i) {
+  await x.register(M, i);
 }
-function ee(v) {
+function ee(x) {
   const i = require("fastify")({
     logger: !0
   });
-  return i.register(N, v), 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;