@magemetrics/core 0.11.6 → 0.12.0-rc1

package/dist/index.d.ts

@@ -428,6 +428,8 @@ declare interface components {
                 /** @enum {string} */
                 model: "claude-haiku-4-5";
             } | null;
+            /** @enum {string} */
+            execution_platform: "native" | "sandbox" | "sandbox_public";
             master_prompt_name: string | null;
             master_prompt_version: number | null;
             created_at: string;
@@ -484,27 +486,12 @@ declare const CreateCanvasSchema: z.ZodObject<{
     flow_id: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
 
-/**
- * A file attachment to include in the first message of a flow.
- *
- * Structurally compatible with the AI SDK's `FileUIPart`, so the output of
- * `useFileUpload().getFileParts()` can be passed directly — no mapping needed.
- */
-declare type CreateFlowFile = default_2.infer<typeof CreateFlowFileSchema>;
-
-declare const CreateFlowFileSchema: z.ZodObject<{
-    type: z.ZodLiteral<"file">;
-    mediaType: z.ZodString;
-    filename: z.ZodOptional<z.ZodString>;
-    url: z.ZodString;
-}, z.core.$strip>;
-
 /**
  * Parameters for starting a flow
  */
 export declare type CreateFlowParam = {
     query: string;
-    files?: CreateFlowFile[];
+    uploadedFileIds?: UploadedFileId[];
 } | {
     triggerId: string;
     variables: Record<string, string>;
@@ -621,6 +608,9 @@ declare const FrontendRecentFlowsSchema: z.ZodObject<{
     created_at: z.ZodString;
     application_id: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
     user_id: z.ZodNullable<z.ZodString>;
+    context_tokens: z.ZodOptional<z.ZodNumber>;
+    acc_input_tokens: z.ZodOptional<z.ZodNumber>;
+    acc_output_tokens: z.ZodOptional<z.ZodNumber>;
 }, z.core.$strip>;
 
 export declare const getPublicApiClient: (apiUrl: string, apiKey: string) => Client<PublicPaths>;
@@ -631,11 +621,6 @@ declare const inputSchema: z.ZodObject<{
     rawRequest: z.ZodString;
     userRequest: z.ZodString;
     isNewAnalysisGoal: z.ZodBoolean;
-    responseType: z.ZodEnum<{
-        text: "text";
-        visualization: "visualization";
-        table: "table";
-    }>;
     difficultyLevel: z.ZodEnum<{
         simple: "simple";
         moderate: "moderate";
@@ -842,6 +827,9 @@ export declare class MageMetricsClient {
             title: string | null;
             user_id: string | null;
             application_id?: number | null | undefined;
+            context_tokens?: number | undefined;
+            acc_input_tokens?: number | undefined;
+            acc_output_tokens?: number | undefined;
         }>;
         getRecentFlows: (params?: {
             limit?: number;
@@ -849,6 +837,13 @@ export declare class MageMetricsClient {
         createFlow: (request: CreateFlowParam) => Promise<{
             flowId: string;
         }>;
+        uploadFile: (file: File, flowId?: string) => Promise<{
+            uploadedFileId: string;
+            filename: string;
+            mediaType: string;
+            sizeBytes: number;
+        }>;
+        deleteUploadedFile: (uploadedFileId: string) => Promise<void>;
         canvas: {
             list: (query?: string) => Promise<{
                 id: string;
@@ -863,6 +858,9 @@ export declare class MageMetricsClient {
                     title: string | null;
                     user_id: string | null;
                     application_id?: number | null | undefined;
+                    context_tokens?: number | undefined;
+                    acc_input_tokens?: number | undefined;
+                    acc_output_tokens?: number | undefined;
                 }[];
             }[]>;
             get: (canvasId: string) => Promise<{
@@ -1369,6 +1367,9 @@ declare interface operations {
                         title: string | null;
                         user_id: string | null;
                         application_id?: number | null;
+                        context_tokens?: number;
+                        acc_input_tokens?: number;
+                        acc_output_tokens?: number;
                     };
                 };
             };
@@ -1422,6 +1423,9 @@ declare interface operations {
                         title: string | null;
                         user_id: string | null;
                         application_id?: number | null;
+                        context_tokens?: number;
+                        acc_input_tokens?: number;
+                        acc_output_tokens?: number;
                     }[];
                 };
             };
@@ -2173,6 +2177,7 @@ declare interface operations {
                         filename?: string;
                         url: string;
                     }[];
+                    uploadedFileIds?: string[];
                 };
             };
         };
@@ -2966,6 +2971,9 @@ declare interface operations {
                             title: string | null;
                             user_id: string | null;
                             application_id?: number | null;
+                            context_tokens?: number;
+                            acc_input_tokens?: number;
+                            acc_output_tokens?: number;
                         }[];
                     }[];
                 };
@@ -3279,6 +3287,11 @@ declare interface operations {
                         /** @enum {string} */
                         model: "claude-haiku-4-5";
                     };
+                    /**
+                     * @default native
+                     * @enum {string}
+                     */
+                    execution_platform?: "native" | "sandbox" | "sandbox_public";
                     master_prompt_name?: string;
                     master_prompt_version?: number;
                 };
@@ -3433,6 +3446,8 @@ declare interface operations {
                         /** @enum {string} */
                         model: "claude-haiku-4-5";
                     } | null;
+                    /** @enum {string} */
+                    execution_platform?: "native" | "sandbox" | "sandbox_public";
                     master_prompt_name?: string | null;
                     master_prompt_version?: number | null;
                 };
@@ -5138,6 +5153,119 @@ declare interface operations {
             };
         };
     };
+    generateDataReport: {
+        parameters: {
+            query?: never;
+            header?: {
+                "sp-access-token"?: string;
+            };
+            path?: never;
+            cookie?: never;
+        };
+        requestBody: {
+            content: {
+                "application/json": {
+                    /** @description The natural language data request */
+                    request: string;
+                };
+            };
+        };
+        responses: {
+            /** @description Data report generated successfully */
+            200: {
+                headers: {
+                    [name: string]: unknown;
+                };
+                content: {
+                    "application/json": {
+                        /** @enum {string} */
+                        status: "success";
+                        generation_id: string;
+                        sql: string;
+                        title: string | null;
+                        goal: string | null;
+                        view_name: string | null;
+                        view_schema: string | null;
+                        data_sample: {
+                            [key: string]: unknown;
+                        }[] | null;
+                        data_summary: {
+                            [key: string]: unknown;
+                        } | null;
+                        is_sample: boolean;
+                    };
+                };
+            };
+            /** @description Invalid request */
+            400: {
+                headers: {
+                    [name: string]: unknown;
+                };
+                content: {
+                    "application/json": {
+                        error: string;
+                    };
+                };
+            };
+            /** @description Service account authentication required */
+            403: {
+                headers: {
+                    [name: string]: unknown;
+                };
+                content: {
+                    "application/json": {
+                        error: string;
+                    };
+                };
+            };
+            /** @description SQL generation failed */
+            422: {
+                headers: {
+                    [name: string]: unknown;
+                };
+                content: {
+                    "application/json": {
+                        /** @enum {string} */
+                        status: "error";
+                        error: string;
+                        error_info?: {
+                            error_message: string;
+                            /** @enum {string} */
+                            category: "timeout" | "connection_error" | "auth_error" | "resource_exhausted" | "table_not_found" | "column_not_found" | "schema_not_found" | "view_not_found" | "function_not_found" | "syntax_error" | "ambiguous_column" | "type_mismatch" | "invalid_cast" | "invalid_argument" | "division_by_zero" | "constraint_violation" | "permission_denied" | "unknown";
+                            details?: string | null;
+                            hint?: string | null;
+                            line_num?: number | null;
+                            col_num?: number | null;
+                        } | null;
+                        failed_sql?: string | null;
+                    };
+                };
+            };
+            /** @description Rate limit exceeded for this API key */
+            429: {
+                headers: {
+                    [name: string]: unknown;
+                };
+                content: {
+                    "application/json": {
+                        error: string;
+                        retry_after_ms: number;
+                    };
+                };
+            };
+            /** @description Internal server error */
+            500: {
+                headers: {
+                    [name: string]: unknown;
+                };
+                content: {
+                    "application/json": {
+                        error: string;
+                    };
+                };
+            };
+        };
+    };
 }
 
 declare type PatchParam<T, Config extends RemoveParamsConfig, P extends keyof Config> = Simplify<T extends {
@@ -5902,6 +6030,22 @@ declare interface paths {
         patch: operations["updateTableColumn"];
         trace?: never;
     };
+    "/api/v1/data-reports/generate": {
+        parameters: {
+            query?: never;
+            header?: never;
+            path?: never;
+            cookie?: never;
+        };
+        get?: never;
+        put?: never;
+        post: operations["generateDataReport"];
+        delete?: never;
+        options?: never;
+        head?: never;
+        patch?: never;
+        trace?: never;
+    };
 }
 
 declare type PublicApiClient = Simplify<Client<PublicPaths, `${string}/${string}`>>;
@@ -6151,4 +6295,9 @@ declare const UpdateCanvasSchema: z.ZodObject<{
     is_public: z.ZodOptional<z.ZodBoolean>;
 }, z.core.$strip>;
 
+/** A previously uploaded file identifier to attach to the first message of a flow. */
+declare type UploadedFileId = default_2.infer<typeof UploadedFileIdSchema>;
+
+declare const UploadedFileIdSchema: z.ZodUUID;
+
 export { }

package/dist/index.js

@@ -1,5 +1,5 @@
 import { sha256 } from '@noble/hashes/sha2.js';
-import z7, { z } from 'zod';
+import z8, { z } from 'zod';
 import { GoTrueClient } from '@supabase/auth-js';
 import createApiClient2 from 'openapi-fetch';
 import dayjs from 'dayjs';
@@ -58,7 +58,7 @@ var addApiKeyHeader = (apiKey) => {
 
 // package.json
 var package_default = {
-  version: "0.11.6"};
+  version: "0.12.0-rc1"};
 
 // src/core/MageMetricsEventEmitter.ts
 var MageMetricsEventEmitter = class {
@@ -152,7 +152,7 @@ var hashString = (value) => {
   return Array.from(hash).map((b) => b.toString(16).padStart(2, "0")).join("");
 };
 
-// ../../node_modules/.pnpm/@asteasolutions+zod-to-openapi@8.4.1_zod@4.3.6/node_modules/@asteasolutions/zod-to-openapi/dist/index.mjs
+// ../../node_modules/.pnpm/@asteasolutions+zod-to-openapi@8.5.0_zod@4.3.6/node_modules/@asteasolutions/zod-to-openapi/dist/index.mjs
 function __rest(s, e) {
   var t = {};
   for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
@@ -313,7 +313,7 @@ var Metadata = class {
   }
   static getOpenApiMetadata(zodSchema) {
     const metadata = this.collectMetadata(zodSchema);
-    const _a = metadata !== null && metadata !== void 0 ? metadata : {}, rest = __rest(_a, ["_internal"]);
+    const _a = metadata !== null && metadata !== void 0 ? metadata : {}, { _internal } = _a, rest = __rest(_a, ["_internal"]);
     return rest;
   }
   static getInternalMetadata(zodSchema) {
@@ -917,7 +917,11 @@ var DatabaseFlowSchema = z.object({
   title: z.string().nullable(),
   user_id: z.string().nullable(),
   application_id: z.number().nullable().optional(),
-  flow_steps: z.array(FlowStepSchema)
+  flow_steps: z.array(FlowStepSchema),
+  // Token usage tracking (provider-reported)
+  context_tokens: z.number().optional(),
+  acc_input_tokens: z.number().optional(),
+  acc_output_tokens: z.number().optional()
 });
 z.object({
   table: z.string(),
@@ -1196,6 +1200,18 @@ var columnsQueryParams = z.string().transform((str) => {
 var CHAT_MAX_FILE_COUNT = 10;
 var CHAT_ALLOWED_FILE_MEDIA_TYPES = /* @__PURE__ */ new Set(["application/pdf"]);
 [...CHAT_ALLOWED_FILE_MEDIA_TYPES].join(",");
+var UploadedFileIdSchema = z.uuid();
+var UploadedFileSchema = z.object({
+  uploadedFileId: UploadedFileIdSchema,
+  filename: z.string(),
+  mediaType: z.string(),
+  sizeBytes: z.number().int().nonnegative()
+});
+z.object({
+  uploadedFileId: UploadedFileIdSchema
+});
+var CreateUploadedFilePath = "/api/v1/uploaded-files";
+var DeleteUploadedFilePath = "/api/v1/uploaded-files/{uploadedFileId}";
 
 // ../shared/dist/src/endpoints/companion/flows.routes.js
 var FlowIdParam = z.object({ flowId: z.uuid() }).openapi("FlowId", { type: "string" });
@@ -1882,7 +1898,8 @@ var CreateFlow = createRoute({
           schema: z.object({
             userQuery: z.string(),
             applicationName: z.string().optional(),
-            files: z.array(CreateFlowFileSchema).max(CHAT_MAX_FILE_COUNT).optional()
+            files: z.array(CreateFlowFileSchema).max(CHAT_MAX_FILE_COUNT).optional(),
+            uploadedFileIds: z.array(UploadedFileIdSchema).max(CHAT_MAX_FILE_COUNT).optional()
           })
         }
       }
@@ -1998,20 +2015,20 @@ var CreateSpeechToken = createRoute({
     }
   }
 });
-z7.discriminatedUnion("status", [
-  z7.object({
-    status: z7.literal("error"),
-    error: z7.string()
+z8.discriminatedUnion("status", [
+  z8.object({
+    status: z8.literal("error"),
+    error: z8.string()
   }),
-  z7.object({
-    status: z7.literal("success"),
-    flowId: z7.string()
+  z8.object({
+    status: z8.literal("success"),
+    flowId: z8.string()
   })
 ]);
-var TriggerFlowBody = z7.object({
-  triggerId: z7.string(),
-  variables: z7.record(z7.string(), z7.string()),
-  applicationName: z7.string().optional()
+var TriggerFlowBody = z8.object({
+  triggerId: z8.string(),
+  variables: z8.record(z8.string(), z8.string()),
+  applicationName: z8.string().optional()
 });
 var TriggerFlow = createRoute({
   method: "post",
@@ -2035,9 +2052,9 @@ var TriggerFlow = createRoute({
       description: "Flow id",
       content: {
         "application/json": {
-          schema: z7.object({
-            status: z7.literal("success"),
-            flowId: z7.string()
+          schema: z8.object({
+            status: z8.literal("success"),
+            flowId: z8.string()
           })
         }
       }
@@ -2045,8 +2062,8 @@ var TriggerFlow = createRoute({
     400: {
       content: {
         "application/json": {
-          schema: z7.object({
-            error: z7.string()
+          schema: z8.object({
+            error: z8.string()
           })
         }
       },
@@ -2055,7 +2072,7 @@ var TriggerFlow = createRoute({
     404: {
       content: {
         "application/json": {
-          schema: z7.object({ error: z7.string() })
+          schema: z8.object({ error: z8.string() })
         }
       },
       description: "Unable to retrieve trigger with this id"
@@ -2064,8 +2081,8 @@ var TriggerFlow = createRoute({
       description: "Something wrong happened",
       content: {
         "application/json": {
-          schema: z7.object({
-            error: z7.string()
+          schema: z8.object({
+            error: z8.string()
           })
         }
       }
@@ -2095,9 +2112,9 @@ createRoute({
       description: "Flow id",
       content: {
         "application/json": {
-          schema: z7.object({
-            status: z7.literal("success"),
-            flowId: z7.string()
+          schema: z8.object({
+            status: z8.literal("success"),
+            flowId: z8.string()
           })
         }
       }
@@ -2105,8 +2122,8 @@ createRoute({
     400: {
       content: {
         "application/json": {
-          schema: z7.object({
-            error: z7.string()
+          schema: z8.object({
+            error: z8.string()
           })
         }
       },
@@ -2115,7 +2132,7 @@ createRoute({
     404: {
       content: {
         "application/json": {
-          schema: z7.object({ error: z7.string() })
+          schema: z8.object({ error: z8.string() })
         }
       },
       description: "Unable to retrieve trigger with this id"
@@ -2124,8 +2141,8 @@ createRoute({
       description: "Something wrong happened",
       content: {
         "application/json": {
-          schema: z7.object({
-            error: z7.string()
+          schema: z8.object({
+            error: z8.string()
           })
         }
       }
@@ -3234,7 +3251,7 @@ var MageMetricsClient = class {
             body: {
               userQuery: request.query,
               applicationName: this.config.applicationName,
-              files: request.files
+              uploadedFileIds: request.uploadedFileIds
             }
           }
         );
@@ -3270,6 +3287,53 @@ var MageMetricsClient = class {
         return { flowId: data.flowId };
       }
     },
+    uploadFile: async (file, flowId) => {
+      await this.waitForAuth();
+      const formData = new FormData();
+      formData.append("file", file);
+      const url = new URL(`${this.config.apiUrl}${CreateUploadedFilePath}`);
+      if (flowId) {
+        url.searchParams.set("flowId", flowId);
+      }
+      const response = await fetch(url, {
+        method: "POST",
+        headers: await this.getHeaders(),
+        body: formData
+      });
+      const data = await response.json();
+      if (!response.ok) {
+        const error = typeof data === "object" && data !== null && "error" in data && typeof data.error === "string" ? data.error : "File upload failed";
+        throw new ApiError(error, response, {
+          status: response.status,
+          statusText: response.statusText,
+          url: response.url,
+          method: "POST"
+        });
+      }
+      return UploadedFileSchema.parse(data);
+    },
+    deleteUploadedFile: async (uploadedFileId) => {
+      await this.waitForAuth();
+      const path = DeleteUploadedFilePath.replace(
+        "{uploadedFileId}",
+        uploadedFileId
+      );
+      const response = await fetch(`${this.config.apiUrl}${path}`, {
+        method: "DELETE",
+        headers: await this.getHeaders()
+      });
+      if (response.ok) {
+        return;
+      }
+      const data = await response.json().catch(() => null);
+      const error = typeof data === "object" && data !== null && "error" in data && typeof data.error === "string" ? data.error : "Failed to delete uploaded file";
+      throw new ApiError(error, response, {
+        status: response.status,
+        statusText: response.statusText,
+        url: response.url,
+        method: "DELETE"
+      });
+    },
     canvas: {
       list: async (query) => {
         await this.waitForAuth();