@magemetrics/core 0.6.0 → 0.7.0

package/dist/index.d.ts

@@ -448,6 +448,7 @@ export declare const getPublicApiClient: (apiUrl: string, apiKey: string) => Cli
 declare const HEADER_SP_TOKEN: "sp-access-token";
 
 declare const inputSchema: z.ZodObject<{
+    rawRequest: z.ZodString;
     userRequest: z.ZodString;
     isNewAnalysisGoal: z.ZodBoolean;
     responseType: z.ZodEnum<{
@@ -490,6 +491,18 @@ declare const inputSchema_7: z.ZodObject<{
     query: z.ZodString;
 }, z.core.$strip>;
 
+declare const inputSchema_8: z.ZodObject<{
+    questions: z.ZodArray<z.ZodObject<{
+        question: z.ZodString;
+        header: z.ZodString;
+        multiSelect: z.ZodBoolean;
+        options: z.ZodArray<z.ZodObject<{
+            label: z.ZodString;
+            description: z.ZodString;
+        }, z.core.$strip>>;
+    }, z.core.$strip>>;
+}, z.core.$strip>;
+
 declare type InternalApiClient = Simplify<Client<NoAuthPaths, `${string}/${string}`>>;
 
 export declare class MageMetricsChatTransport extends DefaultChatTransport<MMChatUIMessage> {
@@ -787,6 +800,7 @@ declare type MMUiTools = {
     getAvailableTables: UiTool_5;
     getAvailableColumnsForTable: UiTool_6;
     webSearch: UiTool_7;
+    askUserQuestion: UiTool_8;
 };
 
 declare type NoAuthPaths = Simplify<{
@@ -1569,6 +1583,8 @@ declare interface operations {
                         friendliness_score?: number;
                         friendliness_score_reason?: string;
                         flow_data_id: number;
+                        output_dataset?: string;
+                        sql?: string;
                     };
                 };
             };
@@ -4452,6 +4468,17 @@ declare const QuickActionsSchema: z.ZodArray<z.ZodObject<{
     explanation: z.ZodString;
 }, z.core.$strip>>;
 
+declare const RedactedAskUserQuestionResponse: z.ZodObject<{
+    tool: z.ZodDefault<z.ZodLiteral<"askUserQuestion">>;
+    result: z.ZodDiscriminatedUnion<[z.ZodObject<{
+        status: z.ZodLiteral<"success">;
+        answers: z.ZodRecord<z.ZodString, z.ZodString>;
+    }, z.core.$strip>, z.ZodObject<{
+        status: z.ZodLiteral<"error">;
+        message: z.ZodString;
+    }, z.core.$strip>], "status">;
+}, z.core.$strip>;
+
 declare const redactedOutputSchema: z.ZodObject<{
     tool: z.ZodLiteral<"generateDataReport">;
     result: z.ZodOptional<z.ZodDiscriminatedUnion<[z.ZodObject<{
@@ -4609,6 +4636,8 @@ declare type UiTool_6 = InferUITool<Tool<z.infer<typeof inputSchema_6>, z.infer<
 
 declare type UiTool_7 = InferUITool<Tool<z.infer<typeof inputSchema_7>, z.infer<typeof redactedOutputSchema_6>>>;
 
+declare type UiTool_8 = InferUITool<Tool<z.infer<typeof inputSchema_8>, z.infer<typeof RedactedAskUserQuestionResponse>>>;
+
 declare const UpdateCanvasSchema: z.ZodObject<{
     id: z.ZodUUID;
     title: z.ZodOptional<z.ZodNullable<z.ZodString>>;

package/dist/index.js

@@ -57,7 +57,7 @@ var addApiKeyHeader = (apiKey) => {
 
 // package.json
 var package_default = {
-  version: "0.6.0"};
+  version: "0.7.0"};
 
 // src/core/MageMetricsEventEmitter.ts
 var MageMetricsEventEmitter = class {
@@ -151,7 +151,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.1.0_zod@4.1.13/node_modules/@asteasolutions/zod-to-openapi/dist/index.mjs
+// ../../node_modules/.pnpm/@asteasolutions+zod-to-openapi@8.1.0_zod@4.2.1/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)
@@ -1144,6 +1144,9 @@ var ReportExplainabilitySchema = z.object({
 var FrontendReportExplainabilitySchema = ReportExplainabilitySchema.omit({
   id: true,
   created_at: true
+}).extend({
+  output_dataset: z.string().optional(),
+  sql: z.string().optional()
 });
 
 // ../shared/dist/src/endpoints/companion/flows.routes.js
@@ -3127,13 +3130,28 @@ var toSearchParams = ({ cursor, filters, sorting, limit }) => {
   }
   return params;
 };
+var isToolResultMessage = (message) => {
+  if (message.role !== "assistant") {
+    return false;
+  }
+  return message.parts.some(
+    (part) => part.type.startsWith("tool-") && "state" in part && part.state === "output-available"
+  );
+};
 var MageMetricsChatTransport = class extends DefaultChatTransport {
   constructor(apiUrl, flowId, options) {
     super({
       ...options,
       api: `${apiUrl}/api/v1/chat/${flowId}`,
       prepareSendMessagesRequest({ messages, id: chatId }) {
-        return { body: { message: messages.at(-1), id: chatId } };
+        const lastMessage = messages.at(-1);
+        if (lastMessage && isToolResultMessage(lastMessage)) {
+          lastMessage.metadata = {
+            ...typeof lastMessage.metadata === "object" ? lastMessage.metadata : void 0,
+            isFrontendToolSubmission: true
+          };
+        }
+        return { body: { message: lastMessage, id: chatId } };
       }
     });
   }