drizzle-cube 0.4.20 → 0.4.22

package/dist/adapters/anthropic-BTkjgFpT.cjs

@@ -0,0 +1 @@
+"use strict";var l=Object.create;var i=Object.defineProperty;var c=Object.getOwnPropertyDescriptor;var u=Object.getOwnPropertyNames;var p=Object.getPrototypeOf,d=Object.prototype.hasOwnProperty;var y=(o,e,r,s)=>{if(e&&typeof e=="object"||typeof e=="function")for(let t of u(e))!d.call(o,t)&&t!==r&&i(o,t,{get:()=>e[t],enumerable:!(s=c(e,t))||s.enumerable});return o};var m=(o,e,r)=>(r=o!=null?l(p(o)):{},y(e||!o||!o.__esModule?i(r,"default",{value:o,enumerable:!0}):r,o));Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});class _{client;constructor(e){this.apiKey=e}apiKey;initialized=!1;async ensureClient(){if(this.initialized)return;let e;try{const r=await import("@anthropic-ai/sdk");e=r.default||r.Anthropic||r}catch{throw new 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();const{messages:r,system:s}=this.formatMessages(e.messages,e.system);return this.client.messages.create({model:e.model,max_tokens:e.maxTokens,system:s,tools:this.formatTools(e.tools),messages:r,stream:!0})}async*parseStreamEvents(e){for await(const r of e){const s=r;switch(s.type){case"content_block_start":{const t=s.content_block;t.type==="tool_use"&&(yield{type:"tool_use_start",id:t.id,name:t.name});break}case"content_block_delta":{const t=s.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":{const t=s.message;t?.usage?.input_tokens!=null&&(yield{type:"message_meta",inputTokens:t.usage.input_tokens,stopReason:""});break}case"message_delta":{const t=s.usage,a=s.delta?.stop_reason||"";yield{type:"message_meta",outputTokens:t?.output_tokens,stopReason:a};break}}}}formatTools(e){return e.map(r=>({name:r.name,description:r.description,input_schema:r.parameters}))}formatMessages(e,r){return{messages:e.map(t=>t.role==="tool_result"?{role:"user",content:t.content}:{role:t.role,content:t.content}),system:r}}formatToolResults(e){return{role:"user",content:e.map(r=>({type:"tool_result",tool_use_id:r.toolUseId,content:r.content,...r.isError?{is_error:!0}:{}}))}}shouldContinue(e){return e==="tool_use"}formatError(e){if(!e||!(e instanceof Error))return"Something went wrong. Please try again.";const r=e.message||"",s={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."},t=e;if(t.status||t.type){const a=t.error?.type||t.type||"";if(s[a])return s[a]}if(r.startsWith("{")||r.startsWith("Error: {")){try{const a=JSON.parse(r.replace(/^Error:\s*/,"")),n=a.error?.type||a.type||"";if(s[n])return s[n]}catch{}return"The AI service encountered an error. Please try again."}return r}}exports.AnthropicProvider=_;

package/dist/adapters/anthropic-CTu9E801.js

@@ -0,0 +1,126 @@
+class i {
+  client;
+  constructor(e) {
+    this.apiKey = e;
+  }
+  apiKey;
+  initialized = !1;
+  async ensureClient() {
+    if (this.initialized) return;
+    let e;
+    try {
+      const r = await import(
+        /* webpackIgnore: true */
+        "@anthropic-ai/sdk"
+      );
+      e = r.default || r.Anthropic || r;
+    } catch {
+      throw new 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();
+    const { messages: r, system: s } = this.formatMessages(e.messages, e.system);
+    return this.client.messages.create({
+      model: e.model,
+      max_tokens: e.maxTokens,
+      system: s,
+      tools: this.formatTools(e.tools),
+      messages: r,
+      stream: !0
+    });
+  }
+  async *parseStreamEvents(e) {
+    for await (const r of e) {
+      const s = r;
+      switch (s.type) {
+        case "content_block_start": {
+          const t = s.content_block;
+          t.type === "tool_use" && (yield { type: "tool_use_start", id: t.id, name: t.name });
+          break;
+        }
+        case "content_block_delta": {
+          const t = s.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": {
+          const t = s.message;
+          t?.usage?.input_tokens != null && (yield {
+            type: "message_meta",
+            inputTokens: t.usage.input_tokens,
+            stopReason: ""
+          });
+          break;
+        }
+        case "message_delta": {
+          const t = s.usage, o = s.delta?.stop_reason || "";
+          yield {
+            type: "message_meta",
+            outputTokens: t?.output_tokens,
+            stopReason: o
+          };
+          break;
+        }
+      }
+    }
+  }
+  formatTools(e) {
+    return e.map((r) => ({
+      name: r.name,
+      description: r.description,
+      input_schema: r.parameters
+    }));
+  }
+  formatMessages(e, r) {
+    return { messages: e.map((t) => t.role === "tool_result" ? { role: "user", content: t.content } : { role: t.role, content: t.content }), system: r };
+  }
+  formatToolResults(e) {
+    return {
+      role: "user",
+      content: e.map((r) => ({
+        type: "tool_result",
+        tool_use_id: r.toolUseId,
+        content: r.content,
+        ...r.isError ? { is_error: !0 } : {}
+      }))
+    };
+  }
+  shouldContinue(e) {
+    return e === "tool_use";
+  }
+  formatError(e) {
+    if (!e || !(e instanceof Error))
+      return "Something went wrong. Please try again.";
+    const r = e.message || "", s = {
+      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."
+    }, t = e;
+    if (t.status || t.type) {
+      const o = t.error?.type || t.type || "";
+      if (s[o])
+        return s[o];
+    }
+    if (r.startsWith("{") || r.startsWith("Error: {")) {
+      try {
+        const o = JSON.parse(r.replace(/^Error:\s*/, "")), a = o.error?.type || o.type || "";
+        if (s[a])
+          return s[a];
+      } catch {
+      }
+      return "The AI service encountered an error. Please try again.";
+    }
+    return r;
+  }
+}
+export {
+  i as AnthropicProvider
+};

package/dist/adapters/express/index.cjs

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

package/dist/adapters/express/index.js

@@ -1,54 +1,54 @@
-import E, { Router as J } from "express";
+import R, { Router as J } from "express";
 import L from "cors";
-import { S as D, v as _, b as w, a as K, n as V, p as z, w as T, d as B, i as P, M as F, c as U, e as S, s as Q } from "../mcp-transport-m1X1GtwG.js";
-import { formatErrorResponse as d, formatCubeResponse as A, handleBatchRequest as X, formatMetaResponse as G, formatSqlResponse as H, handleDryRun as M } from "../utils.js";
-function W(h) {
+import { S as K, v as D, b as x, a as _, n as V, p as z, w as T, d as B, i as I, M as U, c as F, e as Q, s as E } from "../mcp-transport-m1X1GtwG.js";
+import { formatErrorResponse as u, formatCubeResponse as P, handleBatchRequest as X, formatMetaResponse as G, formatSqlResponse as H, handleDryRun as M } from "../utils.js";
+function W(v) {
   const {
-    cubes: m,
-    drizzle: x,
-    schema: N,
+    cubes: h,
+    drizzle: S,
+    schema: O,
     extractSecurityContext: y,
-    engineType: k,
-    cors: R,
+    engineType: N,
+    cors: $,
     basePath: p = "/cubejs-api/v1",
-    jsonLimit: $ = "10mb",
-    cache: O,
-    mcp: v = { enabled: !0 },
-    agent: g
-  } = h;
-  if (!m || m.length === 0)
+    jsonLimit: A = "10mb",
+    cache: k,
+    mcp: w = { enabled: !0 },
+    agent: m
+  } = v;
+  if (!h || h.length === 0)
     throw new Error("At least one cube must be provided in the cubes array");
-  const u = J();
-  R && u.use(L(R)), u.use(E.json({ limit: $ })), u.use(E.urlencoded({ extended: !0, limit: $ }));
-  const i = new D({
-    drizzle: x,
-    schema: N,
-    engineType: k,
-    cache: O
+  const c = J();
+  $ && c.use(L($)), c.use(R.json({ limit: A })), c.use(R.urlencoded({ extended: !0, limit: A }));
+  const i = new K({
+    drizzle: S,
+    schema: O,
+    engineType: N,
+    cache: k
   });
-  if (m.forEach((r) => {
+  if (h.forEach((r) => {
     i.registerCube(r);
-  }), u.post(`${p}/load`, async (r, t) => {
+  }), c.post(`${p}/load`, async (r, t) => {
     try {
       const e = r.body.query || r.body, o = await y(r, t), a = i.validateQuery(e);
       if (!a.isValid)
-        return t.status(400).json(d(
+        return t.status(400).json(u(
           `Query validation failed: ${a.errors.join(", ")}`,
           400
         ));
       const n = r.headers["x-cache-control"] === "no-cache", s = await i.executeMultiCubeQuery(e, o, { skipCache: n });
-      t.json(A(e, s, i));
+      t.json(P(e, s, i));
     } catch (e) {
-      console.error("Query execution error:", e), t.status(500).json(d(
+      console.error("Query execution error:", e), t.status(500).json(u(
         e instanceof Error ? e.message : "Query execution failed",
         500
       ));
     }
-  }), u.get(`${p}/load`, async (r, t) => {
+  }), c.get(`${p}/load`, async (r, t) => {
     try {
       const e = r.query.query;
       if (!e)
-        return t.status(400).json(d(
+        return t.status(400).json(u(
           "Query parameter is required",
           400
         ));
@@ -56,107 +56,107 @@ function W(h) {
       try {
         o = JSON.parse(e);
       } catch {
-        return t.status(400).json(d(
+        return t.status(400).json(u(
           "Invalid JSON in query parameter",
           400
         ));
       }
       const a = await y(r, t), n = i.validateQuery(o);
       if (!n.isValid)
-        return t.status(400).json(d(
+        return t.status(400).json(u(
           `Query validation failed: ${n.errors.join(", ")}`,
           400
         ));
-      const s = r.headers["x-cache-control"] === "no-cache", l = await i.executeMultiCubeQuery(o, a, { skipCache: s });
-      t.json(A(o, l, i));
+      const s = r.headers["x-cache-control"] === "no-cache", f = await i.executeMultiCubeQuery(o, a, { skipCache: s });
+      t.json(P(o, f, i));
     } catch (e) {
-      console.error("Query execution error:", e), t.status(500).json(d(
+      console.error("Query execution error:", e), t.status(500).json(u(
         e instanceof Error ? e.message : "Query execution failed",
         500
       ));
     }
-  }), u.post(`${p}/batch`, async (r, t) => {
+  }), c.post(`${p}/batch`, async (r, t) => {
     try {
       const { queries: e } = r.body;
       if (!e || !Array.isArray(e))
-        return t.status(400).json(d(
+        return t.status(400).json(u(
           'Request body must contain a "queries" array',
           400
         ));
       if (e.length === 0)
-        return t.status(400).json(d(
+        return t.status(400).json(u(
           "Queries array cannot be empty",
           400
         ));
       const o = await y(r, t), a = r.headers["x-cache-control"] === "no-cache", n = await X(e, o, i, { skipCache: a });
       t.json(n);
     } catch (e) {
-      console.error("Batch execution error:", e), t.status(500).json(d(
+      console.error("Batch execution error:", e), t.status(500).json(u(
         e instanceof Error ? e.message : "Batch execution failed",
         500
       ));
     }
-  }), u.get(`${p}/meta`, (r, t) => {
+  }), c.get(`${p}/meta`, (r, t) => {
     try {
       const e = i.getMetadata();
       t.json(G(e));
     } catch (e) {
-      console.error("Metadata error:", e), t.status(500).json(d(
+      console.error("Metadata error:", e), t.status(500).json(u(
         e instanceof Error ? e.message : "Failed to fetch metadata",
         500
       ));
     }
-  }), u.post(`${p}/sql`, async (r, t) => {
+  }), c.post(`${p}/sql`, async (r, t) => {
     try {
       const e = r.body, o = await y(r, t), a = i.validateQuery(e);
       if (!a.isValid)
-        return t.status(400).json(d(
+        return t.status(400).json(u(
           `Query validation failed: ${a.errors.join(", ")}`,
           400
         ));
       const n = e.measures?.[0] || e.dimensions?.[0];
       if (!n)
-        return t.status(400).json(d(
+        return t.status(400).json(u(
           "No measures or dimensions specified",
           400
         ));
-      const s = n.split(".")[0], l = await i.generateSQL(s, e, o);
-      t.json(H(e, l));
+      const s = n.split(".")[0], f = await i.generateSQL(s, e, o);
+      t.json(H(e, f));
     } catch (e) {
-      console.error("SQL generation error:", e), t.status(500).json(d(
+      console.error("SQL generation error:", e), t.status(500).json(u(
         e instanceof Error ? e.message : "SQL generation failed",
         500
       ));
     }
-  }), u.get(`${p}/sql`, async (r, t) => {
+  }), c.get(`${p}/sql`, async (r, t) => {
     try {
       const e = r.query.query;
       if (!e)
-        return t.status(400).json(d(
+        return t.status(400).json(u(
           "Query parameter is required",
           400
         ));
       const o = JSON.parse(e), a = await y(r, t), n = i.validateQuery(o);
       if (!n.isValid)
-        return t.status(400).json(d(
+        return t.status(400).json(u(
           `Query validation failed: ${n.errors.join(", ")}`,
           400
         ));
       const s = o.measures?.[0] || o.dimensions?.[0];
       if (!s)
-        return t.status(400).json(d(
+        return t.status(400).json(u(
           "No measures or dimensions specified",
           400
         ));
-      const l = s.split(".")[0], j = await i.generateSQL(l, o, a);
-      t.json(H(o, j));
+      const f = s.split(".")[0], b = await i.generateSQL(f, o, a);
+      t.json(H(o, b));
     } catch (e) {
-      console.error("SQL generation error:", e), t.status(500).json(d(
+      console.error("SQL generation error:", e), t.status(500).json(u(
         e instanceof Error ? e.message : "SQL generation failed",
         500
       ));
     }
-  }), u.post(`${p}/dry-run`, async (r, t) => {
+  }), c.post(`${p}/dry-run`, async (r, t) => {
     try {
       const e = r.body.query || r.body, o = await y(r, t), a = await M(e, o, i);
       t.json(a);
@@ -166,7 +166,7 @@ function W(h) {
         valid: !1
       });
     }
-  }), u.get(`${p}/dry-run`, async (r, t) => {
+  }), c.get(`${p}/dry-run`, async (r, t) => {
     try {
       const e = r.query.query;
       if (!e)
@@ -182,7 +182,7 @@ function W(h) {
         valid: !1
       });
     }
-  }), u.post(`${p}/explain`, async (r, t) => {
+  }), c.post(`${p}/explain`, async (r, t) => {
     try {
       const e = r.body.query || r.body, o = r.body.options || {}, a = await y(r, t), n = i.validateQuery(e);
       if (!n.isValid)
@@ -196,47 +196,50 @@ function W(h) {
         error: e instanceof Error ? e.message : "Explain query failed"
       });
     }
-  }), g && u.post(`${p}/agent/chat`, async (r, t) => {
+  }), m && c.post(`${p}/agent/chat`, async (r, t) => {
     try {
-      const { handleAgentChat: e } = await import("../handler-CbDMdSY5.js"), { message: o, sessionId: a, history: n } = r.body;
+      const { handleAgentChat: e } = await import("../handler-9Rdn7zM2.js"), { message: o, sessionId: a, history: n } = r.body;
       if (!o || typeof o != "string")
         return t.status(400).json({ error: "message is required and must be a string" });
-      let s = (g.apiKey || "").trim();
-      if (g.allowClientApiKey) {
-        const c = r.headers["x-agent-api-key"];
-        c && (s = c.trim());
+      let s = (m.apiKey || "").trim();
+      if (m.allowClientApiKey) {
+        const l = r.headers["x-agent-api-key"];
+        l && (s = l.trim());
       }
       if (!s)
         return t.status(401).json({
           error: "No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header."
         });
-      const l = await y(r, t), j = g.buildSystemContext?.(l);
+      const f = m.allowClientApiKey ? r.headers["x-agent-provider"] : void 0, b = m.allowClientApiKey ? r.headers["x-agent-model"] : void 0, d = m.allowClientApiKey ? r.headers["x-agent-provider-endpoint"] : void 0, g = await y(r, t), j = m.buildSystemContext?.(g);
       t.writeHead(200, {
         "Content-Type": "text/event-stream",
         "Cache-Control": "no-cache",
         Connection: "keep-alive"
       });
       try {
-        const c = e({
+        const l = e({
           message: o,
           sessionId: a,
           history: n,
           semanticLayer: i,
-          securityContext: l,
-          agentConfig: g,
+          securityContext: g,
+          agentConfig: m,
           apiKey: s,
-          systemContext: j
+          systemContext: j,
+          providerOverride: f,
+          modelOverride: b,
+          baseURLOverride: d
         });
-        for await (const f of c)
-          t.write(`data: ${JSON.stringify(f)}
+        for await (const C of l)
+          t.write(`data: ${JSON.stringify(C)}
 
 `);
-      } catch (c) {
-        const f = {
+      } catch (l) {
+        const C = {
           type: "error",
-          data: { message: c instanceof Error ? c.message : "Stream failed" }
+          data: { message: l instanceof Error ? l.message : "Stream failed" }
         };
-        t.write(`data: ${JSON.stringify(f)}
+        t.write(`data: ${JSON.stringify(C)}
 
 `);
       } finally {
@@ -247,18 +250,18 @@ function W(h) {
         error: e instanceof Error ? e.message : "Agent chat failed"
       });
     }
-  }), v.enabled !== !1) {
-    const r = v.basePath ?? "/mcp";
-    u.post(`${r}`, async (t, e) => {
-      const o = _(
+  }), w.enabled !== !1) {
+    const r = w.basePath ?? "/mcp";
+    c.post(`${r}`, async (t, e) => {
+      const o = D(
         t.headers.origin,
-        v.allowedOrigins ? { allowedOrigins: v.allowedOrigins } : {}
+        w.allowedOrigins ? { allowedOrigins: w.allowedOrigins } : {}
       );
       if (!o.valid)
-        return e.status(403).json(w(null, -32600, o.reason));
+        return e.status(403).json(x(null, -32600, o.reason));
       const a = t.headers.accept;
-      if (!K(a))
-        return e.status(400).json(w(null, -32600, "Accept header must include both application/json and text/event-stream"));
+      if (!_(a))
+        return e.status(400).json(x(null, -32600, "Accept header must include both application/json and text/event-stream"));
       const n = V(t.headers);
       if (!n.ok)
         return e.status(426).json({
@@ -267,10 +270,10 @@ function W(h) {
         });
       const s = z(t.body);
       if (!s)
-        return e.status(400).json(w(null, -32600, "Invalid JSON-RPC 2.0 request"));
-      const l = T(a), j = s.method === "initialize";
+        return e.status(400).json(x(null, -32600, "Invalid JSON-RPC 2.0 request"));
+      const f = T(a), b = s.method === "initialize";
       try {
-        const c = await B(
+        const d = await B(
           s.method,
           s.params,
           {
@@ -281,34 +284,34 @@ function W(h) {
             negotiatedProtocol: n.negotiated
           }
         );
-        if (P(s))
+        if (I(s))
           return e.status(202).end();
-        const f = j && c && typeof c == "object" && "sessionId" in c ? c.sessionId : void 0;
-        f && e.setHeader(F, f);
-        const C = U(s.id ?? null, c);
-        if (l) {
-          const b = S();
-          return e.status(200), e.setHeader("Content-Type", "text/event-stream"), e.setHeader("Cache-Control", "no-cache"), e.setHeader("Connection", "keep-alive"), e.write(`id: ${b}
+        const g = b && d && typeof d == "object" && "sessionId" in d ? d.sessionId : void 0;
+        g && e.setHeader(U, g);
+        const j = F(s.id ?? null, d);
+        if (f) {
+          const l = Q();
+          return e.status(200), e.setHeader("Content-Type", "text/event-stream"), e.setHeader("Cache-Control", "no-cache"), e.setHeader("Connection", "keep-alive"), e.write(`id: ${l}
 
-`), e.write(Q(C, b)), e.end();
+`), e.write(E(j, l)), e.end();
         }
-        return e.json(C);
-      } catch (c) {
-        if (P(s))
-          return console.error("MCP notification processing error:", c), e.status(202).end();
-        console.error("MCP RPC error:", c);
-        const f = c?.code ?? -32603, C = c?.data, b = c.message || "MCP request failed", q = w(s.id ?? null, f, b, C);
-        if (l) {
-          const I = S();
-          return e.status(200), e.setHeader("Content-Type", "text/event-stream"), e.setHeader("Cache-Control", "no-cache"), e.setHeader("Connection", "keep-alive"), e.write(`id: ${I}
+        return e.json(j);
+      } catch (d) {
+        if (I(s))
+          return console.error("MCP notification processing error:", d), e.status(202).end();
+        console.error("MCP RPC error:", d);
+        const g = d?.code ?? -32603, j = d?.data, l = d.message || "MCP request failed", C = x(s.id ?? null, g, l, j);
+        if (f) {
+          const q = Q();
+          return e.status(200), e.setHeader("Content-Type", "text/event-stream"), e.setHeader("Cache-Control", "no-cache"), e.setHeader("Connection", "keep-alive"), e.write(`id: ${q}
 
-`), e.write(Q(q, I)), e.end();
+`), e.write(E(C, q)), e.end();
         }
-        return e.status(200).json(q);
+        return e.status(200).json(C);
       }
-    }), u.get(`${r}`, async (t, e) => {
-      const o = S();
-      e.status(200), e.setHeader("Content-Type", "text/event-stream"), e.setHeader("Cache-Control", "no-cache"), e.setHeader("Connection", "keep-alive"), e.write(Q({
+    }), c.get(`${r}`, async (t, e) => {
+      const o = Q();
+      e.status(200), e.setHeader("Content-Type", "text/event-stream"), e.setHeader("Cache-Control", "no-cache"), e.setHeader("Connection", "keep-alive"), e.write(E({
         jsonrpc: "2.0",
         method: "mcp/ready",
         params: { protocol: "streamable-http" }
@@ -321,19 +324,19 @@ function W(h) {
       t.on("close", () => {
         clearInterval(a);
       });
-    }), u.delete(`${r}`, (t, e) => e.status(405).json({ error: "Session termination not supported" }));
+    }), c.delete(`${r}`, (t, e) => e.status(405).json({ error: "Session termination not supported" }));
   }
-  return u.use((r, t, e, o) => {
-    console.error("Express adapter error:", r), e.headersSent || e.status(500).json(d(r, 500));
-  }), u;
+  return c.use((r, t, e, o) => {
+    console.error("Express adapter error:", r), e.headersSent || e.status(500).json(u(r, 500));
+  }), c;
 }
-function Y(h, m) {
-  const x = W(m);
-  return h.use("/", x), h;
+function Y(v, h) {
+  const S = W(h);
+  return v.use("/", S), v;
 }
-function oe(h) {
-  const m = E();
-  return Y(m, h);
+function oe(v) {
+  const h = R();
+  return Y(h, v);
 }
 export {
   oe as createCubeApp,