@514labs/moose-lib 0.6.437 → 0.6.438

package/dist/index.mjs

@@ -4434,8 +4434,6 @@ function defineQueryModel(config) {
     dimensions,
     metrics,
     columns: columnDefs,
-    dimensionNames,
-    metricNames,
     columnNames,
     query: async (request, client) => {
       const result = await client.execute(toSql(request));
@@ -4523,6 +4521,14 @@ function camelToSnake(s) {
 function titleFromName(name) {
   return name.replace(/^query_/, "Query ").replace(/^list_/, "List ").replace(/_/g, " ").replace(/\b\w/g, (c) => c.toUpperCase());
 }
+function buildEnumDescription(metadata) {
+  const entries = Object.entries(metadata);
+  if (entries.length === 0) return void 0;
+  const lines = entries.map(([name, def]) => {
+    return def.description ? `- ${name}: ${def.description}` : `- ${name}`;
+  });
+  return lines.join("\n");
+}
 function zodBaseType(inputType) {
   if (inputType === "number") return z.number();
   return z.string();
@@ -4560,13 +4566,19 @@ function createModelTool(model, options = {}) {
   const requiredSet = new Set(requiredFilters);
   const schema = {};
   const filterParamMap = {};
-  if (model.dimensionNames.length > 0) {
-    const names = model.dimensionNames;
-    schema.dimensions = z.array(z.enum(names)).optional();
-  }
-  if (model.metricNames.length > 0) {
-    const names = model.metricNames;
-    schema.metrics = z.array(z.enum(names)).optional();
+  const dimensionNames = Object.keys(model.dimensions ?? {});
+  if (dimensionNames.length > 0) {
+    const names = dimensionNames;
+    const desc = buildEnumDescription(model.dimensions);
+    const dimSchema = z.array(z.enum(names)).optional();
+    schema.dimensions = desc ? dimSchema.describe(desc) : dimSchema;
+  }
+  const metricNames = Object.keys(model.metrics ?? {});
+  if (metricNames.length > 0) {
+    const names = metricNames;
+    const desc = buildEnumDescription(model.metrics);
+    const metSchema = z.array(z.enum(names)).optional();
+    schema.metrics = desc ? metSchema.describe(desc) : metSchema;
   }
   if (model.columnNames.length > 0) {
     const names = model.columnNames;
@@ -4590,9 +4602,10 @@ function createModelTool(model, options = {}) {
         paramType = baseType;
       }
       if (requiredSet.has(filterName) && op === "eq") {
-        schema[paramName] = paramType;
+        schema[paramName] = filterDef.description ? paramType.describe(filterDef.description) : paramType;
       } else {
-        schema[paramName] = paramType.optional();
+        const opt = paramType.optional();
+        schema[paramName] = filterDef.description ? opt.describe(filterDef.description) : opt;
       }
       filterParamMap[paramName] = { filterName, op };
     }
@@ -4600,10 +4613,10 @@ function createModelTool(model, options = {}) {
   schema.limit = z.number().min(1).max(maxLimit).default(defaultLimit).optional();
   function buildRequest(params) {
     const request = {};
-    if (model.dimensionNames.length > 0) {
+    if (dimensionNames.length > 0) {
       request.dimensions = params.dimensions ?? mergedDefaults.dimensions;
     }
-    if (model.metricNames.length > 0) {
+    if (metricNames.length > 0) {
       request.metrics = params.metrics ?? mergedDefaults.metrics;
     }
     if (model.columnNames.length > 0) {