@draftt-io/draftt-mcp 1.0.2 → 1.1.0

package/dist/api/drafttClient.js

@@ -4,14 +4,16 @@ exports.DrafttClient = void 0;
 const client_1 = require("./client");
 const client_2 = require("./client");
 class DrafttClient extends client_1.ApiClient {
-    constructor(logger, accessKey) {
+    constructor(logger, accessKey, timeout) {
         const apiKey = accessKey || process.env.DRAFTT_ACCESS_KEY;
         if (!apiKey) {
             throw new Error("Access key is required for authentication");
         }
+        const baseUrl = process.env.DRAFTT_API_BASE_URL || "https://api.draftt.io/v1";
         super({
-            baseUrl: "https://api.draftt.io/v1",
-            logger
+            baseUrl,
+            logger,
+            timeout: timeout || 30000,
         });
         this.accessKey = apiKey;
     }
@@ -19,13 +21,9 @@ class DrafttClient extends client_1.ApiClient {
         return {
             Authorization: `Bearer ${this.accessKey}`,
             "Content-Type": "application/json",
+            "Request-Source": "draftt-mcp",
         };
     }
-    async componentSearch(args) {
-        this.logger.debug("Component search with args:", args);
-        const requestBody = this.buildSearchBody(args);
-        return this.post("/component/search", requestBody);
-    }
     async componentGet(args) {
         this.logger.debug("Component get with args:", args);
         if (!args.componentId) {
@@ -33,10 +31,10 @@ class DrafttClient extends client_1.ApiClient {
         }
         return this.get(`/component/${args.componentId}`);
     }
-    async policySearch(args) {
-        this.logger.debug("Policy search with args:", args);
+    async componentSearch(args) {
+        this.logger.debug("Component search with args:", args);
         const requestBody = this.buildSearchBody(args);
-        return this.post("/policy/search", requestBody);
+        return this.post("/component/search", requestBody);
     }
     async policyGet(args) {
         this.logger.debug("Policy get with args:", args);
@@ -45,13 +43,18 @@ class DrafttClient extends client_1.ApiClient {
         }
         return this.get(`/policy/${args.policyId}`);
     }
-    async policyComponentSearch(args) {
-        this.logger.debug("Policy component search with args:", args);
+    async policySearch(args) {
+        this.logger.debug("Policy search with args:", args);
+        const requestBody = this.buildSearchBody(args);
+        return this.post("/policy/search", requestBody);
+    }
+    async policyComponentQuery(args) {
+        this.logger.debug("Policy component query with args:", args);
         if (!args.policyId) {
             throw new client_2.ApiError("MISSING_PARAM", "Missing required parameter: policyId");
         }
-        const requestBody = this.buildSearchBody(args);
-        return this.post(`/policy/${args.policyId}/component/search`, requestBody);
+        const requestBody = this.buildQueryBody(args);
+        return this.post(`/policy/${args.policyId}/component/query`, requestBody);
     }
     async policyComponentGet(args) {
         this.logger.debug("Policy component get with args:", args);
@@ -67,16 +70,36 @@ class DrafttClient extends client_1.ApiClient {
         this.logger.debug("Draftt get with args:", args);
         return this.post(`/draftt/search`, {});
     }
+    async drafttQuery(args) {
+        this.logger.debug("Draftt query with args:", args);
+        if (!args.componentId) {
+            throw new client_2.ApiError("MISSING_PARAM", "Missing required parameter: componentId");
+        }
+        const requestBody = this.buildQueryBody(args);
+        return this.post(`/component/${args.componentId}/draftt/query`, requestBody);
+    }
     buildSearchBody(args) {
         const searchBody = {};
         // Skip undefined values to keep the request payload clean
         Object.entries(args).forEach(([key, value]) => {
-            if (value !== undefined && key !== 'policyId') { // Skip policyId as it's used in the URL for some endpoints
+            if (value !== undefined && key !== "policyId") {
+                // Skip policyId as it's used in the URL for some endpoints
                 searchBody[key] = value;
             }
         });
         return searchBody;
     }
+    buildQueryBody(args) {
+        const queryBody = {};
+        // Skip undefined values and URL parameters
+        Object.entries(args).forEach(([key, value]) => {
+            if (value !== undefined && key !== "policyId" && key !== "componentId") {
+                // Skip policyId and componentId as they're used in the URL
+                queryBody[key] = value;
+            }
+        });
+        return queryBody;
+    }
 }
 exports.DrafttClient = DrafttClient;
 //# sourceMappingURL=drafttClient.js.map

package/dist/api/drafttClient.js.map

@@ -1 +1 @@
-{"version":3,"file":"drafttClient.js","sourceRoot":"","sources":["../../src/api/drafttClient.ts"],"names":[],"mappings":";;;AAAA,qCAAuD;AAIvD,qCAAoC;AAEpC,MAAa,YAAa,SAAQ,kBAAS;IAGvC,YAAY,MAAW,EAAE,SAAkB;QACvC,MAAM,MAAM,GAAG,SAAS,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QAE1D,IAAI,CAAC,MAAM,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;QACjE,CAAC;QAED,KAAK,CAAC;YACF,OAAO,EAAE,0BAA0B;YACnC,MAAM;SACT,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC;IAC5B,CAAC;IAES,UAAU;QAChB,OAAO;YACH,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE;YACzC,cAAc,EAAE,kBAAkB;SACrC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,IAAyB;QAC3C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE,IAAI,CAAC,CAAC;QACvD,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,WAAW,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,IAAsB;QACrC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,IAAI,CAAC,CAAC;QACpD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACpB,MAAM,IAAI,iBAAQ,CAAC,eAAe,EAAE,yCAAyC,CAAC,CAAC;QACnF,CAAC;QACD,OAAO,IAAI,CAAC,GAAG,CAAC,cAAc,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,IAAsB;QACrC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,IAAI,CAAC,CAAC;QACpD,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,WAAW,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,IAAmB;QAC/B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,IAAI,CAAC,CAAC;QACjD,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACjB,MAAM,IAAI,iBAAQ,CAAC,eAAe,EAAE,sCAAsC,CAAC,CAAC;QAChF,CAAC;QACD,OAAO,IAAI,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,IAA+B;QACvD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE,IAAI,CAAC,CAAC;QAC9D,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACjB,MAAM,IAAI,iBAAQ,CAAC,eAAe,EAAE,sCAAsC,CAAC,CAAC;QAChF,CAAC;QACD,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,QAAQ,mBAAmB,EAAE,WAAW,CAAC,CAAC;IAC/E,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,IAA4B;QACjD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE,IAAI,CAAC,CAAC;QAC3D,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACjB,MAAM,IAAI,iBAAQ,CAAC,eAAe,EAAE,sCAAsC,CAAC,CAAC;QAChF,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACpB,MAAM,IAAI,iBAAQ,CAAC,eAAe,EAAE,yCAAyC,CAAC,CAAC;QACnF,CAAC;QACD,OAAO,IAAI,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,QAAQ,cAAc,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IAC9E,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,IAAmB;QAC/B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,IAAI,CAAC,CAAC;QACjD,OAAO,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAClC,CAAC,CAAC;IACP,CAAC;IAEO,eAAe,CAAC,IAAS;QAC7B,MAAM,UAAU,GAAwB,EAAE,CAAC;QAE3C,0DAA0D;QAC1D,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;YAC1C,IAAI,KAAK,KAAK,SAAS,IAAI,GAAG,KAAK,UAAU,EAAE,CAAC,CAAC,2DAA2D;gBACxG,UAAU,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YAC5B,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,OAAO,UAAU,CAAC;IACtB,CAAC;CACJ;AA3FD,oCA2FC"}
+{"version":3,"file":"drafttClient.js","sourceRoot":"","sources":["../../src/api/drafttClient.ts"],"names":[],"mappings":";;;AAAA,qCAA6C;AAS7C,qCAAoC;AAEpC,MAAa,YAAa,SAAQ,kBAAS;IAGvC,YAAY,MAAc,EAAE,SAAkB,EAAE,OAAgB;QAC5D,MAAM,MAAM,GAAG,SAAS,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QAE1D,IAAI,CAAC,MAAM,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;QACjE,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,0BAA0B,CAAC;QAE9E,KAAK,CAAC;YACF,OAAO;YACP,MAAM;YACN,OAAO,EAAE,OAAO,IAAI,KAAK;SAC5B,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC;IAC5B,CAAC;IAES,UAAU;QAChB,OAAO;YACH,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE;YACzC,cAAc,EAAE,kBAAkB;YAClC,gBAAgB,EAAE,YAAY;SACjC,CAAC;IACN,CAAC;IAGD,KAAK,CAAC,YAAY,CAAC,IAAsB;QACrC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,IAAI,CAAC,CAAC;QACpD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACpB,MAAM,IAAI,iBAAQ,CAAC,eAAe,EAAE,yCAAyC,CAAC,CAAC;QACnF,CAAC;QACD,OAAO,IAAI,CAAC,GAAG,CAAC,cAAc,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,IAAyB;QAC3C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE,IAAI,CAAC,CAAC;QACvD,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,WAAW,CAAC,CAAC;IACvD,CAAC;IAGD,KAAK,CAAC,SAAS,CAAC,IAAmB;QAC/B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,IAAI,CAAC,CAAC;QACjD,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACjB,MAAM,IAAI,iBAAQ,CAAC,eAAe,EAAE,sCAAsC,CAAC,CAAC;QAChF,CAAC;QACD,OAAO,IAAI,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,IAAsB;QACrC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,IAAI,CAAC,CAAC;QACpD,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,WAAW,CAAC,CAAC;IACpD,CAAC;IAGD,KAAK,CAAC,oBAAoB,CAAC,IAA8B;QACrD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mCAAmC,EAAE,IAAI,CAAC,CAAC;QAC7D,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACjB,MAAM,IAAI,iBAAQ,CAAC,eAAe,EAAE,sCAAsC,CAAC,CAAC;QAChF,CAAC;QACD,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QAC9C,OAAO,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,QAAQ,kBAAkB,EAAE,WAAW,CAAC,CAAC;IAC9E,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,IAA4B;QACjD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE,IAAI,CAAC,CAAC;QAC3D,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACjB,MAAM,IAAI,iBAAQ,CAAC,eAAe,EAAE,sCAAsC,CAAC,CAAC;QAChF,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACpB,MAAM,IAAI,iBAAQ,CAAC,eAAe,EAAE,yCAAyC,CAAC,CAAC;QACnF,CAAC;QACD,OAAO,IAAI,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,QAAQ,cAAc,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IAC9E,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,IAAmB;QAC/B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,IAAI,CAAC,CAAC;QACjD,OAAO,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,IAAqB;QACnC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,IAAI,CAAC,CAAC;QACnD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACpB,MAAM,IAAI,iBAAQ,CAAC,eAAe,EAAE,yCAAyC,CAAC,CAAC;QACnF,CAAC;QACD,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QAC9C,OAAO,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,CAAC,WAAW,eAAe,EAAE,WAAW,CAAC,CAAC;IACjF,CAAC;IAGO,eAAe,CAAC,IAA4C;QAChE,MAAM,UAAU,GAA4B,EAAE,CAAC;QAE/C,0DAA0D;QAC1D,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;YAC1C,IAAI,KAAK,KAAK,SAAS,IAAI,GAAG,KAAK,UAAU,EAAE,CAAC;gBAC5C,2DAA2D;gBAC3D,UAAU,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YAC5B,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,OAAO,UAAU,CAAC;IACtB,CAAC;IAEO,cAAc,CAAC,IAAgD;QACnE,MAAM,SAAS,GAA4B,EAAE,CAAC;QAE9C,2CAA2C;QAC3C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;YAC1C,IAAI,KAAK,KAAK,SAAS,IAAI,GAAG,KAAK,UAAU,IAAI,GAAG,KAAK,aAAa,EAAE,CAAC;gBACrE,2DAA2D;gBAC3D,SAAS,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YAC3B,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,OAAO,SAAS,CAAC;IACrB,CAAC;CACJ;AA1HD,oCA0HC"}

package/dist/api/responseFormatter.d.ts

@@ -1,9 +1,9 @@
-export declare function formatSuccessResponse(data: any): {
-    content: {
-        type: string;
-        text: string;
-    }[];
-};
+export declare function formatComponentSummary(component: any): string;
+export declare function formatPolicyComponentSummary(policyComponent: any): string;
+export declare function formatSuccessResponse(data: any, options?: {
+    useSummary?: boolean;
+    toolName?: string;
+}): any;
 export declare function formatErrorResponse(code: string, message: string, details?: any): {
     content: {
         type: string;

package/dist/api/responseFormatter.js

@@ -1,21 +1,174 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.formatComponentSummary = formatComponentSummary;
+exports.formatPolicyComponentSummary = formatPolicyComponentSummary;
 exports.formatSuccessResponse = formatSuccessResponse;
 exports.formatErrorResponse = formatErrorResponse;
-function formatSuccessResponse(data) {
+function formatComponentSummary(component) {
+    if (!component || typeof component !== "object") {
+        return JSON.stringify(component, null, 2);
+    }
+    const summary = {
+        id: component.id || component.componentId,
+        name: component.name,
+        type: component.type,
+        technology: component.technology,
+        currentVersion: component.version || component.currentVersion,
+        region: component.region,
+        status: component.status,
+        tags: component.tags,
+    };
+    let formattedSummary = `## Component Summary\n`;
+    if (summary.name)
+        formattedSummary += `**Name:** ${summary.name}\n`;
+    if (summary.type)
+        formattedSummary += `**Type:** ${summary.type}\n`;
+    if (summary.technology)
+        formattedSummary += `**Technology:** ${summary.technology}\n`;
+    if (summary.currentVersion)
+        formattedSummary += `**Current Version:** ${summary.currentVersion}\n`;
+    if (summary.region)
+        formattedSummary += `**Region:** ${summary.region}\n`;
+    if (summary.status)
+        formattedSummary += `**Status:** ${summary.status}\n`;
+    if (summary.tags && Object.keys(summary.tags).length > 0) {
+        formattedSummary += `**Tags:** ${Object.entries(summary.tags)
+            .map(([k, v]) => `${k}=${v}`)
+            .join(", ")}\n`;
+    }
+    return formattedSummary;
+}
+function formatPolicyComponentSummary(policyComponent) {
+    if (!policyComponent || typeof policyComponent !== "object") {
+        return JSON.stringify(policyComponent, null, 2);
+    }
+    const summary = {
+        id: policyComponent.id || policyComponent.componentId,
+        name: policyComponent.name,
+        type: policyComponent.type,
+        technology: policyComponent.technology,
+        currentVersion: policyComponent.version || policyComponent.currentVersion,
+        region: policyComponent.region,
+        isCompliant: policyComponent.isCompliant,
+        recommendedVersion: policyComponent.recommendedVersion,
+        requiredVersion: policyComponent.requiredVersion,
+        desiredVersion: policyComponent.desiredVersion,
+        dueDate: policyComponent.dueDate,
+        urgency: policyComponent.urgency,
+        priority: policyComponent.priority,
+        tags: policyComponent.tags,
+    };
+    let formattedSummary = `## Policy Component Summary\n`;
+    if (summary.name)
+        formattedSummary += `**Component:** ${summary.name}\n`;
+    if (summary.type)
+        formattedSummary += `**Type:** ${summary.type}\n`;
+    if (summary.technology)
+        formattedSummary += `**Technology:** ${summary.technology}\n`;
+    if (summary.currentVersion)
+        formattedSummary += `**Current Version:** ${summary.currentVersion}\n`;
+    if (summary.isCompliant !== undefined) {
+        formattedSummary += `**Compliance Status:** ${summary.isCompliant ? "✅ Compliant" : "❌ Non-Compliant"}\n`;
+    }
+    if (summary.recommendedVersion)
+        formattedSummary += `**Recommended Version:** ${summary.recommendedVersion}\n`;
+    if (summary.requiredVersion)
+        formattedSummary += `**Required Version:** ${summary.requiredVersion}\n`;
+    if (summary.desiredVersion)
+        formattedSummary += `**Desired Version:** ${summary.desiredVersion}\n`;
+    if (summary.dueDate)
+        formattedSummary += `**Due Date:** ${summary.dueDate}\n`;
+    if (summary.urgency !== undefined)
+        formattedSummary += `**Urgency:** ${summary.urgency}\n`;
+    if (summary.priority !== undefined)
+        formattedSummary += `**Priority:** ${summary.priority}\n`;
+    if (summary.region)
+        formattedSummary += `**Region:** ${summary.region}\n`;
+    return formattedSummary;
+}
+function detectLargeJsonBlob(data) {
+    const jsonString = JSON.stringify(data);
+    return jsonString.length > 1000 || (Array.isArray(data) && data.length > 5);
+}
+function shouldSummarize(data, toolName) {
+    if (!toolName)
+        return false;
+    return (toolName.includes("component") &&
+        (detectLargeJsonBlob(data) || (typeof data === "object" && data !== null)));
+}
+function formatSuccessResponse(data, options) {
+    const { useSummary = true, toolName } = options || {};
+    if (!useSummary || !shouldSummarize(data, toolName)) {
+        return {
+            content: [{ type: "text", text: JSON.stringify(data, null, 2) }],
+        };
+    }
+    if (Array.isArray(data)) {
+        if (toolName?.includes("component_search") ||
+            toolName?.includes("policy_component_search")) {
+            const summaries = data
+                .map((item, index) => {
+                if (toolName.includes("policy_component")) {
+                    return `### Component ${index + 1}\n${formatPolicyComponentSummary(item)}`;
+                }
+                else {
+                    return `### Component ${index + 1}\n${formatComponentSummary(item)}`;
+                }
+            })
+                .join("\n---\n");
+            const totalCount = data.length;
+            const header = `# Search Results (${totalCount} found)\n\n`;
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: header +
+                            summaries +
+                            `\n\n*Note: Use raw JSON format if you need complete details.*`,
+                    },
+                ],
+            };
+        }
+    }
+    else if (typeof data === "object" && data !== null) {
+        if (toolName?.includes("component_get")) {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: formatComponentSummary(data) +
+                            `\n\n*Use raw JSON format if you need complete details.*`,
+                    },
+                ],
+            };
+        }
+        else if (toolName?.includes("policy_component_get")) {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: formatPolicyComponentSummary(data) +
+                            `\n\n*Use raw JSON format if you need complete details.*`,
+                    },
+                ],
+            };
+        }
+    }
     return {
-        content: [{ type: "text", text: JSON.stringify(data) }]
+        content: [{ type: "text", text: JSON.stringify(data, null, 2) }],
     };
 }
 function formatErrorResponse(code, message, details) {
     return {
-        content: [{
+        content: [
+            {
                 type: "text",
                 text: JSON.stringify({
-                    error: { code, message, details }
-                })
-            }],
-        isError: true
+                    error: { code, message, details },
+                }, null, 2),
+            },
+        ],
+        isError: true,
     };
 }
 //# sourceMappingURL=responseFormatter.js.map

package/dist/api/responseFormatter.js.map

@@ -1 +1 @@
-{"version":3,"file":"responseFormatter.js","sourceRoot":"","sources":["../../src/api/responseFormatter.ts"],"names":[],"mappings":";;AAAA,sDAIC;AAED,kDAUC;AAhBD,SAAgB,qBAAqB,CAAC,IAAS;IAC3C,OAAO;QACH,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;KAC1D,CAAC;AACN,CAAC;AAED,SAAgB,mBAAmB,CAAC,IAAY,EAAE,OAAe,EAAE,OAAa;IAC5E,OAAO;QACH,OAAO,EAAE,CAAC;gBACN,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACjB,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;iBACpC,CAAC;aACL,CAAC;QACF,OAAO,EAAE,IAAI;KAChB,CAAC;AACN,CAAC"}
+{"version":3,"file":"responseFormatter.js","sourceRoot":"","sources":["../../src/api/responseFormatter.ts"],"names":[],"mappings":";;AAqBA,wDAiCC;AAED,oEA+CC;AAgBD,sDA0EC;AAED,kDAgBC;AA9LD,SAAgB,sBAAsB,CAAC,SAAc;IACjD,IAAI,CAAC,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;QAC9C,OAAO,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IAC9C,CAAC;IAED,MAAM,OAAO,GAAqB;QAC9B,EAAE,EAAE,SAAS,CAAC,EAAE,IAAI,SAAS,CAAC,WAAW;QACzC,IAAI,EAAE,SAAS,CAAC,IAAI;QACpB,IAAI,EAAE,SAAS,CAAC,IAAI;QACpB,UAAU,EAAE,SAAS,CAAC,UAAU;QAChC,cAAc,EAAE,SAAS,CAAC,OAAO,IAAI,SAAS,CAAC,cAAc;QAC7D,MAAM,EAAE,SAAS,CAAC,MAAM;QACxB,MAAM,EAAE,SAAS,CAAC,MAAM;QACxB,IAAI,EAAE,SAAS,CAAC,IAAI;KACvB,CAAC;IAEF,IAAI,gBAAgB,GAAG,wBAAwB,CAAC;IAEhD,IAAI,OAAO,CAAC,IAAI;QAAE,gBAAgB,IAAI,aAAa,OAAO,CAAC,IAAI,IAAI,CAAC;IACpE,IAAI,OAAO,CAAC,IAAI;QAAE,gBAAgB,IAAI,aAAa,OAAO,CAAC,IAAI,IAAI,CAAC;IACpE,IAAI,OAAO,CAAC,UAAU;QAAE,gBAAgB,IAAI,mBAAmB,OAAO,CAAC,UAAU,IAAI,CAAC;IACtF,IAAI,OAAO,CAAC,cAAc;QACtB,gBAAgB,IAAI,wBAAwB,OAAO,CAAC,cAAc,IAAI,CAAC;IAC3E,IAAI,OAAO,CAAC,MAAM;QAAE,gBAAgB,IAAI,eAAe,OAAO,CAAC,MAAM,IAAI,CAAC;IAC1E,IAAI,OAAO,CAAC,MAAM;QAAE,gBAAgB,IAAI,eAAe,OAAO,CAAC,MAAM,IAAI,CAAC;IAE1E,IAAI,OAAO,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvD,gBAAgB,IAAI,aAAa,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC;aACxD,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;aAC5B,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;IACxB,CAAC;IAED,OAAO,gBAAgB,CAAC;AAC5B,CAAC;AAED,SAAgB,4BAA4B,CAAC,eAAoB;IAC7D,IAAI,CAAC,eAAe,IAAI,OAAO,eAAe,KAAK,QAAQ,EAAE,CAAC;QAC1D,OAAO,IAAI,CAAC,SAAS,CAAC,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACpD,CAAC;IAED,MAAM,OAAO,GAA2B;QACpC,EAAE,EAAE,eAAe,CAAC,EAAE,IAAI,eAAe,CAAC,WAAW;QACrD,IAAI,EAAE,eAAe,CAAC,IAAI;QAC1B,IAAI,EAAE,eAAe,CAAC,IAAI;QAC1B,UAAU,EAAE,eAAe,CAAC,UAAU;QACtC,cAAc,EAAE,eAAe,CAAC,OAAO,IAAI,eAAe,CAAC,cAAc;QACzE,MAAM,EAAE,eAAe,CAAC,MAAM;QAC9B,WAAW,EAAE,eAAe,CAAC,WAAW;QACxC,kBAAkB,EAAE,eAAe,CAAC,kBAAkB;QACtD,eAAe,EAAE,eAAe,CAAC,eAAe;QAChD,cAAc,EAAE,eAAe,CAAC,cAAc;QAC9C,OAAO,EAAE,eAAe,CAAC,OAAO;QAChC,OAAO,EAAE,eAAe,CAAC,OAAO;QAChC,QAAQ,EAAE,eAAe,CAAC,QAAQ;QAClC,IAAI,EAAE,eAAe,CAAC,IAAI;KAC7B,CAAC;IAEF,IAAI,gBAAgB,GAAG,+BAA+B,CAAC;IAEvD,IAAI,OAAO,CAAC,IAAI;QAAE,gBAAgB,IAAI,kBAAkB,OAAO,CAAC,IAAI,IAAI,CAAC;IACzE,IAAI,OAAO,CAAC,IAAI;QAAE,gBAAgB,IAAI,aAAa,OAAO,CAAC,IAAI,IAAI,CAAC;IACpE,IAAI,OAAO,CAAC,UAAU;QAAE,gBAAgB,IAAI,mBAAmB,OAAO,CAAC,UAAU,IAAI,CAAC;IACtF,IAAI,OAAO,CAAC,cAAc;QACtB,gBAAgB,IAAI,wBAAwB,OAAO,CAAC,cAAc,IAAI,CAAC;IAE3E,IAAI,OAAO,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;QACpC,gBAAgB,IAAI,0BAA0B,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,iBAAiB,IAAI,CAAC;IAC9G,CAAC;IAED,IAAI,OAAO,CAAC,kBAAkB;QAC1B,gBAAgB,IAAI,4BAA4B,OAAO,CAAC,kBAAkB,IAAI,CAAC;IACnF,IAAI,OAAO,CAAC,eAAe;QACvB,gBAAgB,IAAI,yBAAyB,OAAO,CAAC,eAAe,IAAI,CAAC;IAC7E,IAAI,OAAO,CAAC,cAAc;QACtB,gBAAgB,IAAI,wBAAwB,OAAO,CAAC,cAAc,IAAI,CAAC;IAC3E,IAAI,OAAO,CAAC,OAAO;QAAE,gBAAgB,IAAI,iBAAiB,OAAO,CAAC,OAAO,IAAI,CAAC;IAE9E,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS;QAAE,gBAAgB,IAAI,gBAAgB,OAAO,CAAC,OAAO,IAAI,CAAC;IAC3F,IAAI,OAAO,CAAC,QAAQ,KAAK,SAAS;QAAE,gBAAgB,IAAI,iBAAiB,OAAO,CAAC,QAAQ,IAAI,CAAC;IAC9F,IAAI,OAAO,CAAC,MAAM;QAAE,gBAAgB,IAAI,eAAe,OAAO,CAAC,MAAM,IAAI,CAAC;IAE1E,OAAO,gBAAgB,CAAC;AAC5B,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAS;IAClC,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACxC,OAAO,UAAU,CAAC,MAAM,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AAChF,CAAC;AAED,SAAS,eAAe,CAAC,IAAS,EAAE,QAAiB;IACjD,IAAI,CAAC,QAAQ;QAAE,OAAO,KAAK,CAAC;IAE5B,OAAO,CACH,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC;QAC9B,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,CAAC,CAAC,CAC7E,CAAC;AACN,CAAC;AAED,SAAgB,qBAAqB,CACjC,IAAS,EACT,OAGC;IAED,MAAM,EAAE,UAAU,GAAG,IAAI,EAAE,QAAQ,EAAE,GAAG,OAAO,IAAI,EAAE,CAAC;IAEtD,IAAI,CAAC,UAAU,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAE,CAAC;QAClD,OAAO;YACH,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;SACnE,CAAC;IACN,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QACtB,IACI,QAAQ,EAAE,QAAQ,CAAC,kBAAkB,CAAC;YACtC,QAAQ,EAAE,QAAQ,CAAC,yBAAyB,CAAC,EAC/C,CAAC;YACC,MAAM,SAAS,GAAG,IAAI;iBACjB,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;gBACjB,IAAI,QAAQ,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC;oBACxC,OAAO,iBAAiB,KAAK,GAAG,CAAC,KAAK,4BAA4B,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/E,CAAC;qBAAM,CAAC;oBACJ,OAAO,iBAAiB,KAAK,GAAG,CAAC,KAAK,sBAAsB,CAAC,IAAI,CAAC,EAAE,CAAC;gBACzE,CAAC;YACL,CAAC,CAAC;iBACD,IAAI,CAAC,SAAS,CAAC,CAAC;YAErB,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC;YAC/B,MAAM,MAAM,GAAG,qBAAqB,UAAU,aAAa,CAAC;YAE5D,OAAO;gBACH,OAAO,EAAE;oBACL;wBACI,IAAI,EAAE,MAAM;wBACZ,IAAI,EACA,MAAM;4BACN,SAAS;4BACT,+DAA+D;qBACtE;iBACJ;aACJ,CAAC;QACN,CAAC;IACL,CAAC;SAAM,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;QACnD,IAAI,QAAQ,EAAE,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YACtC,OAAO;gBACH,OAAO,EAAE;oBACL;wBACI,IAAI,EAAE,MAAM;wBACZ,IAAI,EACA,sBAAsB,CAAC,IAAI,CAAC;4BAC5B,yDAAyD;qBAChE;iBACJ;aACJ,CAAC;QACN,CAAC;aAAM,IAAI,QAAQ,EAAE,QAAQ,CAAC,sBAAsB,CAAC,EAAE,CAAC;YACpD,OAAO;gBACH,OAAO,EAAE;oBACL;wBACI,IAAI,EAAE,MAAM;wBACZ,IAAI,EACA,4BAA4B,CAAC,IAAI,CAAC;4BAClC,yDAAyD;qBAChE;iBACJ;aACJ,CAAC;QACN,CAAC;IACL,CAAC;IAED,OAAO;QACH,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;KACnE,CAAC;AACN,CAAC;AAED,SAAgB,mBAAmB,CAAC,IAAY,EAAE,OAAe,EAAE,OAAa;IAC5E,OAAO;QACH,OAAO,EAAE;YACL;gBACI,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAChB;oBACI,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;iBACpC,EACD,IAAI,EACJ,CAAC,CACJ;aACJ;SACJ;QACD,OAAO,EAAE,IAAI;KAChB,CAAC;AACN,CAAC"}

package/dist/handlers/componentHandler.d.ts

@@ -4,14 +4,16 @@ export declare class ComponentHandler {
     private client;
     constructor(client: DrafttClient);
     search(args: ComponentSearchArgs): Promise<{
+        isError?: boolean | undefined;
         content: {
-            type: string;
+            type: "text";
             text: string;
         }[];
     }>;
     get(args: ComponentGetArgs): Promise<{
+        isError?: boolean | undefined;
         content: {
-            type: string;
+            type: "text";
             text: string;
         }[];
     }>;

package/dist/handlers/componentHandler.js

@@ -2,8 +2,8 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ComponentHandler = void 0;
 const client_1 = require("../api/client");
-const responseFormatter_1 = require("../api/responseFormatter");
 const logger_1 = require("../utils/logger");
+const json_1 = require("../utils/json");
 class ComponentHandler {
     constructor(client) {
         this.client = client;
@@ -13,14 +13,15 @@ class ComponentHandler {
         try {
             logger.debug("Processing component search with args:", args);
             const result = await this.client.componentSearch(args);
-            return (0, responseFormatter_1.formatSuccessResponse)(result);
+            return (0, json_1.createSafeResponse)(result);
         }
         catch (error) {
             logger.error("Error in component search:", error);
             if (error instanceof client_1.ApiError) {
-                return (0, responseFormatter_1.formatErrorResponse)(error.code, error.message, error.details);
+                return (0, json_1.createSafeResponse)({ error: { code: error.code, message: error.message, details: error.details } }, true);
             }
-            return (0, responseFormatter_1.formatErrorResponse)("INTERNAL_ERROR", `Unexpected error during component search: ${error instanceof Error ? error.message : String(error)}`);
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            return (0, json_1.createSafeResponse)({ error: { code: "INTERNAL_ERROR", message: `Unexpected error during component search: ${errorMessage}` } }, true);
         }
     }
     async get(args) {
@@ -28,17 +29,18 @@ class ComponentHandler {
         try {
             logger.debug("Processing component get with args:", args);
             if (!args.componentId) {
-                return (0, responseFormatter_1.formatErrorResponse)("MISSING_PARAM", "Missing required parameter: componentId");
+                return (0, json_1.createSafeResponse)({ error: { code: "MISSING_PARAM", message: "Missing required parameter: componentId" } }, true);
             }
             const result = await this.client.componentGet(args);
-            return (0, responseFormatter_1.formatSuccessResponse)(result);
+            return (0, json_1.createSafeResponse)(result);
         }
         catch (error) {
             logger.error("Error in component get:", error);
             if (error instanceof client_1.ApiError) {
-                return (0, responseFormatter_1.formatErrorResponse)(error.code, error.message, error.details);
+                return (0, json_1.createSafeResponse)({ error: { code: error.code, message: error.message, details: error.details } }, true);
             }
-            return (0, responseFormatter_1.formatErrorResponse)("INTERNAL_ERROR", `Unexpected error during component get: ${error instanceof Error ? error.message : String(error)}`);
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            return (0, json_1.createSafeResponse)({ error: { code: "INTERNAL_ERROR", message: `Unexpected error during component get: ${errorMessage}` } }, true);
         }
     }
 }

package/dist/handlers/componentHandler.js.map

@@ -1 +1 @@
-{"version":3,"file":"componentHandler.js","sourceRoot":"","sources":["../../src/handlers/componentHandler.ts"],"names":[],"mappings":";;;AACA,0CAAyC;AACzC,gEAAsF;AAEtF,4CAA4C;AAE5C,MAAa,gBAAgB;IACzB,YAAoB,MAAoB;QAApB,WAAM,GAAN,MAAM,CAAc;IAAG,CAAC;IAE5C,KAAK,CAAC,MAAM,CAAC,IAAyB;QAClC,MAAM,MAAM,GAAG,IAAA,kBAAS,GAAE,CAAC;QAE3B,IAAI,CAAC;YACD,MAAM,CAAC,KAAK,CAAC,wCAAwC,EAAE,IAAI,CAAC,CAAC;YAC7D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YACvD,OAAO,IAAA,yCAAqB,EAAC,MAAM,CAAC,CAAC;QACzC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,4BAA4B,EAAE,KAAK,CAAC,CAAC;YAClD,IAAI,KAAK,YAAY,iBAAQ,EAAE,CAAC;gBAC5B,OAAO,IAAA,uCAAmB,EAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YACzE,CAAC;YACD,OAAO,IAAA,uCAAmB,EACtB,gBAAgB,EAChB,6CAA6C,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CACxG,CAAC;QACN,CAAC;IACL,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,IAAsB;QAC5B,MAAM,MAAM,GAAG,IAAA,kBAAS,GAAE,CAAC;QAE3B,IAAI,CAAC;YACD,MAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE,IAAI,CAAC,CAAC;YAE1D,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;gBACpB,OAAO,IAAA,uCAAmB,EAAC,eAAe,EAAE,yCAAyC,CAAC,CAAC;YAC3F,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YACpD,OAAO,IAAA,yCAAqB,EAAC,MAAM,CAAC,CAAC;QACzC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;YAC/C,IAAI,KAAK,YAAY,iBAAQ,EAAE,CAAC;gBAC5B,OAAO,IAAA,uCAAmB,EAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YACzE,CAAC;YACD,OAAO,IAAA,uCAAmB,EACtB,gBAAgB,EAChB,0CAA0C,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CACrG,CAAC;QACN,CAAC;IACL,CAAC;CACJ;AA7CD,4CA6CC"}
+{"version":3,"file":"componentHandler.js","sourceRoot":"","sources":["../../src/handlers/componentHandler.ts"],"names":[],"mappings":";;;AACA,0CAAyC;AAEzC,4CAA4C;AAC5C,wCAAmD;AAEnD,MAAa,gBAAgB;IACzB,YAAoB,MAAoB;QAApB,WAAM,GAAN,MAAM,CAAc;IAAG,CAAC;IAE5C,KAAK,CAAC,MAAM,CAAC,IAAyB;QAClC,MAAM,MAAM,GAAG,IAAA,kBAAS,GAAE,CAAC;QAE3B,IAAI,CAAC;YACD,MAAM,CAAC,KAAK,CAAC,wCAAwC,EAAE,IAAI,CAAC,CAAC;YAC7D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YACvD,OAAO,IAAA,yBAAkB,EAAC,MAAM,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,4BAA4B,EAAE,KAAK,CAAC,CAAC;YAClD,IAAI,KAAK,YAAY,iBAAQ,EAAE,CAAC;gBAC5B,OAAO,IAAA,yBAAkB,EACrB,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,EAAE,EAC/E,IAAI,CACP,CAAC;YACN,CAAC;YACD,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,OAAO,IAAA,yBAAkB,EACrB,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,6CAA6C,YAAY,EAAE,EAAE,EAAE,EAC3G,IAAI,CACP,CAAC;QACN,CAAC;IACL,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,IAAsB;QAC5B,MAAM,MAAM,GAAG,IAAA,kBAAS,GAAE,CAAC;QAE3B,IAAI,CAAC;YACD,MAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE,IAAI,CAAC,CAAC;YAE1D,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;gBACpB,OAAO,IAAA,yBAAkB,EACrB,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,yCAAyC,EAAE,EAAE,EACxF,IAAI,CACP,CAAC;YACN,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YACpD,OAAO,IAAA,yBAAkB,EAAC,MAAM,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;YAC/C,IAAI,KAAK,YAAY,iBAAQ,EAAE,CAAC;gBAC5B,OAAO,IAAA,yBAAkB,EACrB,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,EAAE,EAC/E,IAAI,CACP,CAAC;YACN,CAAC;YACD,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,OAAO,IAAA,yBAAkB,EACrB,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,0CAA0C,YAAY,EAAE,EAAE,EAAE,EACxG,IAAI,CACP,CAAC;QACN,CAAC;IACL,CAAC;CAEJ;AAzDD,4CAyDC"}

package/dist/handlers/drafttHandler.d.ts

@@ -1,11 +1,19 @@
 import { DrafttClient } from "../api/drafttClient";
-import { DrafttGetArgs } from "../models/draftt";
+import { DrafttGetArgs, DrafttQueryArgs } from "../models/draftt";
 export declare class DrafttHandler {
     private client;
     constructor(client: DrafttClient);
     get(args: DrafttGetArgs): Promise<{
+        isError?: boolean | undefined;
         content: {
-            type: string;
+            type: "text";
+            text: string;
+        }[];
+    }>;
+    query(args: DrafttQueryArgs): Promise<{
+        isError?: boolean | undefined;
+        content: {
+            type: "text";
             text: string;
         }[];
     }>;

package/dist/handlers/drafttHandler.js

@@ -2,8 +2,8 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.DrafttHandler = void 0;
 const client_1 = require("../api/client");
-const responseFormatter_1 = require("../api/responseFormatter");
 const logger_1 = require("../utils/logger");
+const json_1 = require("../utils/json");
 class DrafttHandler {
     constructor(client) {
         this.client = client;
@@ -13,14 +13,34 @@ class DrafttHandler {
         try {
             logger.debug("Processing draftt get with args:", args);
             const result = await this.client.drafttGet(args);
-            return (0, responseFormatter_1.formatSuccessResponse)(result);
+            return (0, json_1.createSafeResponse)(result);
         }
         catch (error) {
             logger.error("Error in draftt get:", error);
             if (error instanceof client_1.ApiError) {
-                return (0, responseFormatter_1.formatErrorResponse)(error.code, error.message, error.details);
+                return (0, json_1.createSafeResponse)({ error: { code: error.code, message: error.message, details: error.details } }, true);
             }
-            return (0, responseFormatter_1.formatErrorResponse)("INTERNAL_ERROR", `Unexpected error during draftt get: ${error instanceof Error ? error.message : String(error)}`);
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            return (0, json_1.createSafeResponse)({ error: { code: "INTERNAL_ERROR", message: `Unexpected error during draftt get: ${errorMessage}` } }, true);
+        }
+    }
+    async query(args) {
+        const logger = (0, logger_1.getLogger)();
+        try {
+            logger.debug("Processing draftt query with args:", args);
+            if (!args.componentId) {
+                return (0, json_1.createSafeResponse)({ error: { code: "MISSING_PARAM", message: "Missing required parameter: componentId" } }, true);
+            }
+            const result = await this.client.drafttQuery(args);
+            return (0, json_1.createSafeResponse)(result);
+        }
+        catch (error) {
+            logger.error("Error in draftt query:", error);
+            if (error instanceof client_1.ApiError) {
+                return (0, json_1.createSafeResponse)({ error: { code: error.code, message: error.message, details: error.details } }, true);
+            }
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            return (0, json_1.createSafeResponse)({ error: { code: "INTERNAL_ERROR", message: `Unexpected error during draftt query: ${errorMessage}` } }, true);
         }
     }
 }

package/dist/handlers/drafttHandler.js.map

@@ -1 +1 @@
-{"version":3,"file":"drafttHandler.js","sourceRoot":"","sources":["../../src/handlers/drafttHandler.ts"],"names":[],"mappings":";;;AACA,0CAAyC;AACzC,gEAAsF;AAEtF,4CAA4C;AAE5C,MAAa,aAAa;IACtB,YAAoB,MAAoB;QAApB,WAAM,GAAN,MAAM,CAAc;IAAG,CAAC;IAE5C,KAAK,CAAC,GAAG,CAAC,IAAmB;QACzB,MAAM,MAAM,GAAG,IAAA,kBAAS,GAAE,CAAC;QAE3B,IAAI,CAAC;YACD,MAAM,CAAC,KAAK,CAAC,kCAAkC,EAAE,IAAI,CAAC,CAAC;YAGvD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACjD,OAAO,IAAA,yCAAqB,EAAC,MAAM,CAAC,CAAC;QACzC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;YAC5C,IAAI,KAAK,YAAY,iBAAQ,EAAE,CAAC;gBAC5B,OAAO,IAAA,uCAAmB,EAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YACzE,CAAC;YACD,OAAO,IAAA,uCAAmB,EACtB,gBAAgB,EAChB,uCAAuC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAClG,CAAC;QACN,CAAC;IACL,CAAC;CACJ;AAvBD,sCAuBC"}
+{"version":3,"file":"drafttHandler.js","sourceRoot":"","sources":["../../src/handlers/drafttHandler.ts"],"names":[],"mappings":";;;AACA,0CAAyC;AAEzC,4CAA4C;AAC5C,wCAAmD;AAEnD,MAAa,aAAa;IACtB,YAAoB,MAAoB;QAApB,WAAM,GAAN,MAAM,CAAc;IAAG,CAAC;IAE5C,KAAK,CAAC,GAAG,CAAC,IAAmB;QACzB,MAAM,MAAM,GAAG,IAAA,kBAAS,GAAE,CAAC;QAE3B,IAAI,CAAC;YACD,MAAM,CAAC,KAAK,CAAC,kCAAkC,EAAE,IAAI,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACjD,OAAO,IAAA,yBAAkB,EAAC,MAAM,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;YAC5C,IAAI,KAAK,YAAY,iBAAQ,EAAE,CAAC;gBAC5B,OAAO,IAAA,yBAAkB,EACrB,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,EAAE,EAC/E,IAAI,CACP,CAAC;YACN,CAAC;YACD,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,OAAO,IAAA,yBAAkB,EACrB,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,uCAAuC,YAAY,EAAE,EAAE,EAAE,EACrG,IAAI,CACP,CAAC;QACN,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAqB;QAC7B,MAAM,MAAM,GAAG,IAAA,kBAAS,GAAE,CAAC;QAE3B,IAAI,CAAC;YACD,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE,IAAI,CAAC,CAAC;YAEzD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;gBACpB,OAAO,IAAA,yBAAkB,EACrB,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,yCAAyC,EAAE,EAAE,EACxF,IAAI,CACP,CAAC;YACN,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACnD,OAAO,IAAA,yBAAkB,EAAC,MAAM,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;YAC9C,IAAI,KAAK,YAAY,iBAAQ,EAAE,CAAC;gBAC5B,OAAO,IAAA,yBAAkB,EACrB,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,EAAE,EAC/E,IAAI,CACP,CAAC;YACN,CAAC;YACD,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,OAAO,IAAA,yBAAkB,EACrB,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,yCAAyC,YAAY,EAAE,EAAE,EAAE,EACvG,IAAI,CACP,CAAC;QACN,CAAC;IACL,CAAC;CACJ;AAzDD,sCAyDC"}

package/dist/handlers/policyHandler.d.ts

@@ -1,30 +1,34 @@
 import { DrafttClient } from "../api/drafttClient";
-import { PolicyComponentGetArgs, PolicyComponentSearchArgs } from "../models/component";
+import { PolicyComponentGetArgs, PolicyComponentQueryArgs } from "../models/component";
 import { PolicyGetArgs, PolicySearchArgs } from "../models/policy";
 export declare class PolicyHandler {
     private client;
     constructor(client: DrafttClient);
     search(args: PolicySearchArgs): Promise<{
+        isError?: boolean | undefined;
         content: {
-            type: string;
+            type: "text";
             text: string;
         }[];
     }>;
     get(args: PolicyGetArgs): Promise<{
+        isError?: boolean | undefined;
         content: {
-            type: string;
+            type: "text";
             text: string;
         }[];
     }>;
-    componentSearch(args: PolicyComponentSearchArgs): Promise<{
+    componentQuery(args: PolicyComponentQueryArgs): Promise<{
+        isError?: boolean | undefined;
         content: {
-            type: string;
+            type: "text";
             text: string;
         }[];
     }>;
     componentGet(args: PolicyComponentGetArgs): Promise<{
+        isError?: boolean | undefined;
         content: {
-            type: string;
+            type: "text";
             text: string;
         }[];
     }>;