@mithung/vunet-mcp-server 2.1.1 → 2.2.0

package/index.js

@@ -222,6 +222,42 @@ class VunetMCPServer {
   /**
    * Query a metric/data model
    */
+  /**
+   * Fetch datamodels from the VuNet tenant with pagination and optional search/filter.
+   * Automatically re-authenticates on 401.
+   */
+  async fetchDataModels({ limit = 10, offset = 0, search = "", signal_type = "", ordering = "" } = {}) {
+    await this.ensureAuthenticated();
+
+    const params = new URLSearchParams({ ordering, limit, offset, search });
+    if (signal_type) params.set("signal_type", signal_type);
+
+    const url = `${this.normalizeTenantUrl(this.tenantUrl)}/api/vuaccel/datamodel/?${params.toString()}`;
+
+    try {
+      const response = await axios.get(url, {
+        headers: { Authorization: `Bearer ${this.sessionToken}` },
+        httpsAgent: this.httpsAgent,
+      });
+      return response.data;
+    } catch (error) {
+      if (error.response?.status === 401) {
+        // Token expired — re-authenticate and retry once
+        this.isAuthenticated = false;
+        await this.ensureAuthenticated();
+        const retryResponse = await axios.get(url, {
+          headers: { Authorization: `Bearer ${this.sessionToken}` },
+          httpsAgent: this.httpsAgent,
+        });
+        return retryResponse.data;
+      }
+      const errorDetails = error.response?.data
+        ? JSON.stringify(error.response.data)
+        : error.message;
+      throw new Error(`Failed to fetch datamodels: ${errorDetails}`);
+    }
+  }
+
   async queryMetric(metricName, queryParams) {
     await this.ensureAuthenticated();
 
@@ -355,6 +391,36 @@ class VunetMCPServer {
             },
           },
         },
+        {
+          name: "vunet_fetch_datamodels",
+          description: "Fetch the actual live DataModels (signals) configured in this VuNet tenant by calling /api/vuaccel/datamodel/. Returns real signal names, types (Metric/Event/Log), data sources, and column details. Use this to discover what data is available before querying with vunet_query_metric.",
+          inputSchema: {
+            type: "object",
+            properties: {
+              limit: {
+                type: "number",
+                description: "Number of results to return per page (default: 10, max: 100)",
+              },
+              offset: {
+                type: "number",
+                description: "Pagination offset — number of records to skip (default: 0)",
+              },
+              search: {
+                type: "string",
+                description: "Search keyword to filter data models by name (e.g., 'UPI', 'CPU', 'transaction')",
+              },
+              signal_type: {
+                type: "string",
+                description: "Filter by signal type: 'Metric', 'Event', or 'Log'",
+                enum: ["Metric", "Event", "Log"],
+              },
+              ordering: {
+                type: "string",
+                description: "Field to sort results by (e.g., 'signal_name', '-creation_time')",
+              },
+            },
+          },
+        },
       ],
     }));
 
@@ -373,6 +439,9 @@ class VunetMCPServer {
           case "vunet_list_data_models":
             return await this.handleListDataModels(args);
 
+          case "vunet_fetch_datamodels":
+            return await this.handleFetchDataModels(args);
+
           default:
             throw new Error(`Unknown tool: ${name}`);
         }
@@ -430,8 +499,7 @@ class VunetMCPServer {
     };
   }
 
-  async handleListDataModels(args = {}) {
-    const category = args.category || "all";
+  async handleListDataModels(args = {}) {    const category = args.category || "all";
 
     const dataModels = {
       apm: [
@@ -544,6 +612,44 @@ class VunetMCPServer {
     };
   }
 
+  async handleFetchDataModels(args = {}) {
+    const { limit = 10, offset = 0, search = "", signal_type = "", ordering = "" } = args;
+
+    const data = await this.fetchDataModels({ limit, offset, search, signal_type, ordering });
+
+    // Shape a clean, LLM-friendly summary
+    const models = (data.results || []).map((m) => ({
+      signal_id: m.signal_id,
+      signal_name: m.signal_name,
+      signal_type: m.signal_type,
+      description: m.description || "",
+      sources: m.source_names || [],
+      columns: (m.metric_column_details || []).map((c) => c.column_name || c.name || c),
+      created_by: m.created_by,
+      last_modified: m.last_modified_time,
+    }));
+
+    const result = {
+      tenant: this.tenantUrl,
+      total_count: data.count,
+      returned: models.length,
+      offset,
+      has_more: offset + models.length < data.count,
+      next_offset: offset + models.length < data.count ? offset + models.length : null,
+      tip: "Use signal_name values with the vunet_query_metric tool to fetch actual data.",
+      datamodels: models,
+    };
+
+    return {
+      content: [
+        {
+          type: "text",
+          text: JSON.stringify(result, null, 2),
+        },
+      ],
+    };
+  }
+
   async run() {
     console.error("Vunet MCP Server starting...");
     

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mithung/vunet-mcp-server",
-  "version": "2.1.1",
+  "version": "2.2.0",
   "description": "Model Context Protocol (MCP) Server for Vunet vuSmartMaps - Multi-tenant observability platform integration",
   "main": "index.js",
   "type": "module",