@attrove/mcp 0.1.3 → 0.1.4

package/esm/src/server.js

@@ -0,0 +1,241 @@
+/**
+ * Attrove MCP Server
+ *
+ * Model Context Protocol server that enables AI assistants like Claude
+ * to access user context through the Attrove API.
+ */
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
+import { Attrove, isAttroveError, ValidationError, ErrorCodes } from '@attrove/sdk';
+import { allToolDefinitions, executeQueryTool, executeSearchTool, executeIntegrationsTool, } from './tools';
+import { getVersion } from './version';
+/**
+ * Validate that a value is an array of strings.
+ * Returns undefined if value is null/undefined.
+ * Throws ValidationError if value is present but invalid.
+ */
+function validateStringArray(value, fieldName) {
+    if (value === undefined || value === null) {
+        return undefined;
+    }
+    if (!Array.isArray(value)) {
+        throw new ValidationError(`${fieldName} must be an array of strings, received ${typeof value}`, ErrorCodes.VALIDATION_INVALID_FORMAT, { field: fieldName, received: typeof value });
+    }
+    if (!value.every((item) => typeof item === 'string')) {
+        const types = value.map(item => typeof item).join(', ');
+        throw new ValidationError(`${fieldName} must contain only strings, received types: [${types}]`, ErrorCodes.VALIDATION_INVALID_FORMAT, { field: fieldName, receivedTypes: types });
+    }
+    return value;
+}
+/**
+ * Validate input for the query tool.
+ *
+ * @throws {ValidationError} If arguments are missing or invalid
+ */
+function validateQueryInput(args) {
+    if (!args || typeof args !== 'object') {
+        throw new ValidationError('Missing or invalid arguments. Expected an object with a "query" field.', ErrorCodes.VALIDATION_REQUIRED_FIELD, { expected: 'object with query field', received: typeof args });
+    }
+    const input = args;
+    if (typeof input.query !== 'string' || input.query.trim() === '') {
+        throw new ValidationError('Missing required parameter "query". Please provide a question to ask.', ErrorCodes.VALIDATION_REQUIRED_FIELD, { field: 'query' });
+    }
+    return {
+        query: input.query,
+        integration_ids: validateStringArray(input.integration_ids, 'integration_ids'),
+        include_sources: typeof input.include_sources === 'boolean' ? input.include_sources : undefined,
+    };
+}
+/**
+ * Validate input for the search tool.
+ *
+ * @throws {ValidationError} If arguments are missing or invalid
+ */
+function validateSearchInput(args) {
+    if (!args || typeof args !== 'object') {
+        throw new ValidationError('Missing or invalid arguments. Expected an object with a "query" field.', ErrorCodes.VALIDATION_REQUIRED_FIELD, { expected: 'object with query field', received: typeof args });
+    }
+    const input = args;
+    if (typeof input.query !== 'string' || input.query.trim() === '') {
+        throw new ValidationError('Missing required parameter "query". Please provide a search query.', ErrorCodes.VALIDATION_REQUIRED_FIELD, { field: 'query' });
+    }
+    return {
+        query: input.query,
+        after_date: typeof input.after_date === 'string' ? input.after_date : undefined,
+        before_date: typeof input.before_date === 'string' ? input.before_date : undefined,
+        sender_domains: validateStringArray(input.sender_domains, 'sender_domains'),
+        include_body_text: typeof input.include_body_text === 'boolean' ? input.include_body_text : undefined,
+    };
+}
+/**
+ * Format an error for MCP response.
+ * Preserves error type information, code, and details for debugging.
+ */
+function formatErrorResponse(error) {
+    let errorMessage;
+    let errorDetails;
+    if (isAttroveError(error)) {
+        errorMessage = `${error.name}: ${error.message}`;
+        const detailParts = [];
+        if (error.code) {
+            detailParts.push(`Code: ${error.code}`);
+        }
+        if (error.status) {
+            detailParts.push(`Status: ${error.status}`);
+        }
+        if (error.details) {
+            try {
+                detailParts.push(`Details: ${JSON.stringify(error.details)}`);
+            }
+            catch (serializeError) {
+                const reason = serializeError instanceof Error ? serializeError.message : 'Unknown reason';
+                detailParts.push(`Details: [Not serializable: ${reason}]`);
+            }
+        }
+        if (detailParts.length > 0) {
+            errorDetails = detailParts.join('\n');
+        }
+    }
+    else if (error instanceof Error) {
+        errorMessage = error.message;
+    }
+    else {
+        errorMessage = String(error);
+    }
+    const text = errorDetails
+        ? `Error: ${errorMessage}\n\n${errorDetails}`
+        : `Error: ${errorMessage}`;
+    return {
+        content: [{ type: 'text', text }],
+        isError: true,
+    };
+}
+/**
+ * Create and configure the Attrove MCP server.
+ */
+export function createServer(config) {
+    const server = new Server({
+        name: 'attrove',
+        version: getVersion(),
+    }, {
+        capabilities: {
+            tools: {},
+        },
+    });
+    // Create Attrove client
+    // Type assertion is safe because startServer validates the apiKey prefix
+    const client = new Attrove({
+        apiKey: config.apiKey,
+        userId: config.userId,
+        baseUrl: config.baseUrl,
+    });
+    // Register tool list handler
+    server.setRequestHandler(ListToolsRequestSchema, async () => {
+        return {
+            tools: allToolDefinitions.map((tool) => ({
+                name: tool.name,
+                description: tool.description,
+                inputSchema: tool.inputSchema,
+            })),
+        };
+    });
+    // Register tool call handler
+    server.setRequestHandler(CallToolRequestSchema, async (request) => {
+        const { name, arguments: args } = request.params;
+        // Validate input outside try-catch so validation errors propagate clearly
+        // and programming bugs in validation don't get caught as API errors
+        let validatedInput;
+        switch (name) {
+            case 'attrove_query':
+                validatedInput = validateQueryInput(args);
+                break;
+            case 'attrove_search':
+                validatedInput = validateSearchInput(args);
+                break;
+            case 'attrove_integrations':
+                // No input validation needed
+                break;
+            default:
+                return {
+                    content: [
+                        {
+                            type: 'text',
+                            text: `Unknown tool: ${name}. Available tools: attrove_query, attrove_search, attrove_integrations`,
+                        },
+                    ],
+                    isError: true,
+                };
+        }
+        // Execute tool with narrowed try-catch for API/network errors only
+        try {
+            let result;
+            switch (name) {
+                case 'attrove_query':
+                    result = await executeQueryTool(client, validatedInput);
+                    break;
+                case 'attrove_search':
+                    result = await executeSearchTool(client, validatedInput);
+                    break;
+                case 'attrove_integrations':
+                    result = await executeIntegrationsTool(client);
+                    break;
+                default:
+                    // Already handled above, but TypeScript needs this
+                    throw new Error(`Unexpected tool: ${name}`);
+            }
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: result,
+                    },
+                ],
+            };
+        }
+        catch (error) {
+            // Expected API/network errors - format directly for user
+            if (isAttroveError(error) || error instanceof ValidationError) {
+                return formatErrorResponse(error);
+            }
+            // Unexpected errors (programming bugs, system errors) - always log, then format for user.
+            // We catch all errors to keep the MCP server running, but logging ensures visibility.
+            // prettier-ignore
+            console.error('[AttroveMCP] Unexpected error in tool handler:', error);
+            return formatErrorResponse(error);
+        }
+    });
+    return server;
+}
+/**
+ * Start the MCP server with stdio transport.
+ */
+export async function startServer(config) {
+    const server = createServer(config);
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+}
+/**
+ * Get configuration from environment variables.
+ *
+ * @throws {Error} If required environment variables are missing
+ */
+export function getConfigFromEnv() {
+    const apiKey = process.env.ATTROVE_API_KEY;
+    const userId = process.env.ATTROVE_USER_ID;
+    const baseUrl = process.env.ATTROVE_BASE_URL;
+    if (!apiKey) {
+        throw new Error('ATTROVE_API_KEY environment variable is required. ' +
+            'Set it to your Attrove API key (sk_...).');
+    }
+    if (!userId) {
+        throw new Error('ATTROVE_USER_ID environment variable is required. ' +
+            'Set it to the user ID (UUID) for the user whose context you want to access.');
+    }
+    return {
+        apiKey,
+        userId,
+        baseUrl,
+    };
+}
+//# sourceMappingURL=server.js.map

package/esm/src/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../../../../../packages/mcp/src/server.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,qBAAqB,EAAE,sBAAsB,EAAE,MAAM,oCAAoC,CAAC;AACnG,OAAO,EAAE,OAAO,EAAgB,cAAc,EAAE,eAAe,EAAE,UAAU,EAAgB,MAAM,cAAc,CAAC;AAChH,OAAO,EACL,kBAAkB,EAClB,gBAAgB,EAChB,iBAAiB,EACjB,uBAAuB,GAGxB,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AAYvC;;;;GAIG;AACH,SAAS,mBAAmB,CAAC,KAAc,EAAE,SAAiB;IAC5D,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QAC1C,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1B,MAAM,IAAI,eAAe,CACvB,GAAG,SAAS,0CAA0C,OAAO,KAAK,EAAE,EACpE,UAAU,CAAC,yBAAyB,EACpC,EAAE,KAAK,EAAE,SAAS,EAAE,QAAQ,EAAE,OAAO,KAAK,EAAE,CAC7C,CAAC;IACJ,CAAC;IACD,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC,EAAE,CAAC;QACrE,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxD,MAAM,IAAI,eAAe,CACvB,GAAG,SAAS,gDAAgD,KAAK,GAAG,EACpE,UAAU,CAAC,yBAAyB,EACpC,EAAE,KAAK,EAAE,SAAS,EAAE,aAAa,EAAE,KAAK,EAAE,CAC3C,CAAC;IACJ,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAsBD;;;;GAIG;AACH,SAAS,kBAAkB,CAAC,IAAa;IACvC,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QACtC,MAAM,IAAI,eAAe,CACvB,wEAAwE,EACxE,UAAU,CAAC,yBAAyB,EACpC,EAAE,QAAQ,EAAE,yBAAyB,EAAE,QAAQ,EAAE,OAAO,IAAI,EAAE,CAC/D,CAAC;IACJ,CAAC;IAED,MAAM,KAAK,GAAG,IAA+B,CAAC;IAE9C,IAAI,OAAO,KAAK,CAAC,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;QACjE,MAAM,IAAI,eAAe,CACvB,uEAAuE,EACvE,UAAU,CAAC,yBAAyB,EACpC,EAAE,KAAK,EAAE,OAAO,EAAE,CACnB,CAAC;IACJ,CAAC;IAED,OAAO;QACL,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,eAAe,EAAE,mBAAmB,CAAC,KAAK,CAAC,eAAe,EAAE,iBAAiB,CAAC;QAC9E,eAAe,EAAE,OAAO,KAAK,CAAC,eAAe,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC,SAAS;KAChG,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAS,mBAAmB,CAAC,IAAa;IACxC,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QACtC,MAAM,IAAI,eAAe,CACvB,wEAAwE,EACxE,UAAU,CAAC,yBAAyB,EACpC,EAAE,QAAQ,EAAE,yBAAyB,EAAE,QAAQ,EAAE,OAAO,IAAI,EAAE,CAC/D,CAAC;IACJ,CAAC;IAED,MAAM,KAAK,GAAG,IAA+B,CAAC;IAE9C,IAAI,OAAO,KAAK,CAAC,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;QACjE,MAAM,IAAI,eAAe,CACvB,oEAAoE,EACpE,UAAU,CAAC,yBAAyB,EACpC,EAAE,KAAK,EAAE,OAAO,EAAE,CACnB,CAAC;IACJ,CAAC;IAED,OAAO;QACL,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,UAAU,EAAE,OAAO,KAAK,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS;QAC/E,WAAW,EAAE,OAAO,KAAK,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;QAClF,cAAc,EAAE,mBAAmB,CAAC,KAAK,CAAC,cAAc,EAAE,gBAAgB,CAAC;QAC3E,iBAAiB,EAAE,OAAO,KAAK,CAAC,iBAAiB,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC,SAAS;KACtG,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,SAAS,mBAAmB,CAAC,KAAc;IACzC,IAAI,YAAoB,CAAC;IACzB,IAAI,YAAgC,CAAC;IAErC,IAAI,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1B,YAAY,GAAG,GAAG,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC;QACjD,MAAM,WAAW,GAAa,EAAE,CAAC;QAEjC,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;YACf,WAAW,CAAC,IAAI,CAAC,SAAS,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QAC1C,CAAC;QACD,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACjB,WAAW,CAAC,IAAI,CAAC,WAAW,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9C,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC;gBACH,WAAW,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAChE,CAAC;YAAC,OAAO,cAAc,EAAE,CAAC;gBACxB,MAAM,MAAM,GAAG,cAAc,YAAY,KAAK,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,gBAAgB,CAAC;gBAC3F,WAAW,CAAC,IAAI,CAAC,+BAA+B,MAAM,GAAG,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;QAED,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,YAAY,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;SAAM,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAClC,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC;IAC/B,CAAC;SAAM,CAAC;QACN,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IAED,MAAM,IAAI,GAAG,YAAY;QACvB,CAAC,CAAC,UAAU,YAAY,OAAO,YAAY,EAAE;QAC7C,CAAC,CAAC,UAAU,YAAY,EAAE,CAAC;IAE7B,OAAO;QACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;QACjC,OAAO,EAAE,IAAI;KACd,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY,CAAC,MAAuB;IAClD,MAAM,MAAM,GAAG,IAAI,MAAM,CACvB;QACE,IAAI,EAAE,SAAS;QACf,OAAO,EAAE,UAAU,EAAE;KACtB,EACD;QACE,YAAY,EAAE;YACZ,KAAK,EAAE,EAAE;SACV;KACF,CACF,CAAC;IAEF,wBAAwB;IACxB,yEAAyE;IACzE,MAAM,MAAM,GAAG,IAAI,OAAO,CAAC;QACzB,MAAM,EAAE,MAAM,CAAC,MAAsB;QACrC,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,OAAO,EAAE,MAAM,CAAC,OAAO;KACxB,CAAC,CAAC;IAEH,6BAA6B;IAC7B,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QAC1D,OAAO;YACL,KAAK,EAAE,kBAAkB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACvC,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,WAAW,EAAE,IAAI,CAAC,WAAW;aAC9B,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,6BAA6B;IAC7B,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAwB,EAAE,EAAE;QACjF,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QAEjD,0EAA0E;QAC1E,oEAAoE;QACpE,IAAI,cAA4D,CAAC;QAEjE,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,eAAe;gBAClB,cAAc,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC;gBAC1C,MAAM;YACR,KAAK,gBAAgB;gBACnB,cAAc,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;gBAC3C,MAAM;YACR,KAAK,sBAAsB;gBACzB,6BAA6B;gBAC7B,MAAM;YACR;gBACE,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,iBAAiB,IAAI,wEAAwE;yBACpG;qBACF;oBACD,OAAO,EAAE,IAAI;iBACd,CAAC;QACN,CAAC;QAED,mEAAmE;QACnE,IAAI,CAAC;YACH,IAAI,MAAc,CAAC;YAEnB,QAAQ,IAAI,EAAE,CAAC;gBACb,KAAK,eAAe;oBAClB,MAAM,GAAG,MAAM,gBAAgB,CAAC,MAAM,EAAE,cAAgC,CAAC,CAAC;oBAC1E,MAAM;gBACR,KAAK,gBAAgB;oBACnB,MAAM,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE,cAAiC,CAAC,CAAC;oBAC5E,MAAM;gBACR,KAAK,sBAAsB;oBACzB,MAAM,GAAG,MAAM,uBAAuB,CAAC,MAAM,CAAC,CAAC;oBAC/C,MAAM;gBACR;oBACE,mDAAmD;oBACnD,MAAM,IAAI,KAAK,CAAC,oBAAoB,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YAED,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,MAAM;qBACb;iBACF;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,yDAAyD;YACzD,IAAI,cAAc,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,eAAe,EAAE,CAAC;gBAC9D,OAAO,mBAAmB,CAAC,KAAK,CAAC,CAAC;YACpC,CAAC;YACD,0FAA0F;YAC1F,sFAAsF;YACtF,kBAAkB;YAClB,OAAO,CAAC,KAAK,CAAC,gDAAgD,EAAE,KAAK,CAAC,CAAC;YACvE,OAAO,mBAAmB,CAAC,KAAK,CAAC,CAAC;QACpC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,MAAuB;IACvD,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;IACpC,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;AAClC,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,gBAAgB;IAC9B,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;IAC3C,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;IAC3C,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAE7C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CACb,oDAAoD;YACpD,0CAA0C,CAC3C,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CACb,oDAAoD;YACpD,6EAA6E,CAC9E,CAAC;IACJ,CAAC;IAED,OAAO;QACL,MAAM;QACN,MAAM;QACN,OAAO;KACR,CAAC;AACJ,CAAC"}

package/esm/src/tools/index.js

@@ -0,0 +1,19 @@
+/**
+ * MCP Tools
+ *
+ * Tool definitions and executors for the Attrove MCP server.
+ */
+import { queryToolDefinition, executeQueryTool } from './query';
+import { searchToolDefinition, executeSearchTool } from './search';
+import { integrationsToolDefinition, executeIntegrationsTool } from './integrations';
+// Re-export tool definitions
+export { queryToolDefinition, executeQueryTool, searchToolDefinition, executeSearchTool, integrationsToolDefinition, executeIntegrationsTool, };
+/**
+ * All tool definitions for registration with MCP.
+ */
+export const allToolDefinitions = [
+    queryToolDefinition,
+    searchToolDefinition,
+    integrationsToolDefinition,
+];
+//# sourceMappingURL=index.js.map

package/esm/src/tools/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../../packages/mcp/src/tools/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAkB,MAAM,SAAS,CAAC;AAChF,OAAO,EAAE,oBAAoB,EAAE,iBAAiB,EAAmB,MAAM,UAAU,CAAC;AACpF,OAAO,EAAE,0BAA0B,EAAE,uBAAuB,EAAE,MAAM,gBAAgB,CAAC;AAErF,6BAA6B;AAC7B,OAAO,EACL,mBAAmB,EACnB,gBAAgB,EAChB,oBAAoB,EACpB,iBAAiB,EACjB,0BAA0B,EAC1B,uBAAuB,GACxB,CAAC;AAKF;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG;IAChC,mBAAmB;IACnB,oBAAoB;IACpB,0BAA0B;CAClB,CAAC"}

package/esm/src/tools/integrations.js

@@ -0,0 +1,46 @@
+/**
+ * Integrations Tool
+ *
+ * MCP tool for listing and managing connected integrations.
+ */
+/**
+ * Tool definition for attrove_integrations.
+ */
+export const integrationsToolDefinition = {
+    name: 'attrove_integrations',
+    description: `List the user's connected integrations (Gmail, Slack, etc.).
+
+This tool shows which services the user has connected and their current status. Use this to:
+- Check what data sources are available for queries
+- Verify integration status before searching
+- Help users understand their connected services
+
+The response includes:
+- Integration ID (for filtering queries)
+- Provider name (gmail, slack, google_calendar, etc.)
+- Connection status (connected, disconnected, expired, error, pending)`,
+    inputSchema: {
+        type: 'object',
+        properties: {},
+        required: [],
+    },
+};
+/**
+ * Execute the integrations tool.
+ */
+export async function executeIntegrationsTool(client) {
+    const integrations = await client.integrations.list();
+    if (integrations.length === 0) {
+        return 'No integrations connected. The user needs to connect their Gmail, Slack, or other services first.';
+    }
+    let result = `Connected integrations (${integrations.length}):\n`;
+    for (const integration of integrations) {
+        const status = integration.is_active ? '[Active]' : '[Inactive]';
+        const authStatus = integration.auth_status === 'connected' ? '' : ` (${integration.auth_status})`;
+        result += `\n- **${integration.name}** (${integration.provider})\n`;
+        result += `  - ID: ${integration.id}\n`;
+        result += `  - Status: ${status}${authStatus}\n`;
+    }
+    return result;
+}
+//# sourceMappingURL=integrations.js.map

package/esm/src/tools/integrations.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"integrations.js","sourceRoot":"","sources":["../../../../../../packages/mcp/src/tools/integrations.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH;;GAEG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG;IACxC,IAAI,EAAE,sBAAsB;IAC5B,WAAW,EAAE;;;;;;;;;;uEAUwD;IACrE,WAAW,EAAE;QACX,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE,EAAE;QACd,QAAQ,EAAE,EAAE;KACb;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,MAAe;IAEf,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC;IAEtD,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,OAAO,mGAAmG,CAAC;IAC7G,CAAC;IAED,IAAI,MAAM,GAAG,2BAA2B,YAAY,CAAC,MAAM,MAAM,CAAC;IAElE,KAAK,MAAM,WAAW,IAAI,YAAY,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,YAAY,CAAC;QACjE,MAAM,UAAU,GAAG,WAAW,CAAC,WAAW,KAAK,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,WAAW,CAAC,WAAW,GAAG,CAAC;QAElG,MAAM,IAAI,SAAS,WAAW,CAAC,IAAI,OAAO,WAAW,CAAC,QAAQ,KAAK,CAAC;QACpE,MAAM,IAAI,WAAW,WAAW,CAAC,EAAE,IAAI,CAAC;QACxC,MAAM,IAAI,eAAe,MAAM,GAAG,UAAU,IAAI,CAAC;IACnD,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/esm/src/tools/query.js

@@ -0,0 +1,66 @@
+/**
+ * Query Tool
+ *
+ * MCP tool for RAG queries against user context.
+ */
+/**
+ * Tool definition for attrove_query.
+ */
+export const queryToolDefinition = {
+    name: 'attrove_query',
+    description: `Ask a question about the user's communications and get an AI-generated answer.
+
+This tool searches across all connected integrations (Gmail, Slack, Google Meet, etc.) and returns a comprehensive answer based on the relevant context.
+
+Use this tool when the user asks questions like:
+- "What did Sarah say about the Q4 budget?"
+- "Summarize my meeting with the engineering team"
+- "What are the action items from yesterday's standup?"
+- "When is my next meeting with John?"`,
+    inputSchema: {
+        type: 'object',
+        properties: {
+            query: {
+                type: 'string',
+                description: 'The question to ask about the user\'s context',
+            },
+            integration_ids: {
+                type: 'array',
+                items: { type: 'string' },
+                description: 'Optional: Filter to specific integration IDs (UUIDs)',
+            },
+            include_sources: {
+                type: 'boolean',
+                description: 'Optional: Include source snippets in the response',
+                default: false,
+            },
+        },
+        required: ['query'],
+    },
+};
+/**
+ * Execute the query tool.
+ */
+export async function executeQueryTool(client, input) {
+    const options = {};
+    if (input.integration_ids?.length) {
+        options.integrationIds = input.integration_ids;
+    }
+    if (input.include_sources) {
+        options.includeSources = true;
+    }
+    const response = await client.query(input.query, options);
+    // Format the response for MCP
+    let result = response.answer;
+    if (response.sources?.length) {
+        result += '\n\n**Sources:**';
+        for (const source of response.sources) {
+            result += `\n- ${source.title}: "${source.snippet}"`;
+        }
+    }
+    if (response.used_message_ids.length > 0) {
+        result += `\n\n_Based on ${response.used_message_ids.length} messages._`;
+    }
+    return result;
+}
+//# sourceMappingURL=query.js.map

package/esm/src/tools/query.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"query.js","sourceRoot":"","sources":["../../../../../../packages/mcp/src/tools/query.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG;IACjC,IAAI,EAAE,eAAe;IACrB,WAAW,EAAE;;;;;;;;uCAQwB;IACrC,WAAW,EAAE;QACX,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE;YACV,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,+CAA+C;aAC7D;YACD,eAAe,EAAE;gBACf,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACzB,WAAW,EAAE,sDAAsD;aACpE;YACD,eAAe,EAAE;gBACf,IAAI,EAAE,SAAS;gBACf,WAAW,EAAE,mDAAmD;gBAChE,OAAO,EAAE,KAAK;aACf;SACF;QACD,QAAQ,EAAE,CAAC,OAAO,CAAC;KACpB;CACF,CAAC;AAWF;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,MAAe,EACf,KAAqB;IAErB,MAAM,OAAO,GAAiB,EAAE,CAAC;IAEjC,IAAI,KAAK,CAAC,eAAe,EAAE,MAAM,EAAE,CAAC;QAClC,OAAO,CAAC,cAAc,GAAG,KAAK,CAAC,eAAe,CAAC;IACjD,CAAC;IACD,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;QAC1B,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC;IAChC,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAE1D,8BAA8B;IAC9B,IAAI,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;IAE7B,IAAI,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;QAC7B,MAAM,IAAI,kBAAkB,CAAC;QAC7B,KAAK,MAAM,MAAM,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACtC,MAAM,IAAI,OAAO,MAAM,CAAC,KAAK,MAAM,MAAM,CAAC,OAAO,GAAG,CAAC;QACvD,CAAC;IACH,CAAC;IAED,IAAI,QAAQ,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzC,MAAM,IAAI,iBAAiB,QAAQ,CAAC,gBAAgB,CAAC,MAAM,aAAa,CAAC;IAC3E,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/esm/src/tools/search.js

@@ -0,0 +1,143 @@
+/**
+ * Search Tool
+ *
+ * MCP tool for semantic search across user context.
+ */
+import { MAX_BODY_PREVIEW_LENGTH } from '../constants';
+/**
+ * Tool definition for attrove_search.
+ */
+export const searchToolDefinition = {
+    name: 'attrove_search',
+    description: `Search for specific messages or conversations in the user's communications.
+
+This tool performs semantic search across connected integrations and returns matching messages grouped by conversation. Use this when you need:
+- Raw message data without AI summarization
+- To find specific conversations or threads
+- To explore what information is available about a topic
+
+Use this tool when the user asks things like:
+- "Find all emails about the product launch"
+- "Show me conversations with the marketing team"
+- "Search for messages mentioning the deadline"`,
+    inputSchema: {
+        type: 'object',
+        properties: {
+            query: {
+                type: 'string',
+                description: 'The search query (semantic, not keyword-based)',
+            },
+            after_date: {
+                type: 'string',
+                description: 'Optional: Only include messages after this date (ISO 8601)',
+            },
+            before_date: {
+                type: 'string',
+                description: 'Optional: Only include messages before this date (ISO 8601)',
+            },
+            sender_domains: {
+                type: 'array',
+                items: { type: 'string' },
+                description: 'Optional: Filter by sender email domains (e.g., ["acme.com"])',
+            },
+            include_body_text: {
+                type: 'boolean',
+                description: 'Optional: Include full message body text',
+                default: true,
+            },
+        },
+        required: ['query'],
+    },
+};
+/**
+ * Execute the search tool.
+ */
+export async function executeSearchTool(client, input) {
+    const options = {
+        includeBodyText: input.include_body_text !== false,
+    };
+    if (input.after_date) {
+        options.afterDate = input.after_date;
+    }
+    if (input.before_date) {
+        options.beforeDate = input.before_date;
+    }
+    if (input.sender_domains?.length) {
+        options.senderDomains = input.sender_domains;
+    }
+    const response = await client.search(input.query, options);
+    // Format the response for MCP
+    const messageCount = response.key_messages?.length ?? 0;
+    if (messageCount === 0) {
+        return `No messages found matching "${input.query}".`;
+    }
+    // Defensive check for conversations structure
+    if (!response.conversations || typeof response.conversations !== 'object') {
+        return `Found ${messageCount} messages, but conversation data is unavailable.`;
+    }
+    const conversationCount = Object.keys(response.conversations).length;
+    let result = `Found ${messageCount} relevant messages across ${conversationCount} conversations:\n`;
+    // Track skipped data for warnings
+    let skippedConversations = 0;
+    let skippedThreads = 0;
+    let skippedMessages = 0;
+    for (const [convId, conv] of Object.entries(response.conversations)) {
+        // Skip malformed conversation entries
+        if (!conv || !conv.threads) {
+            skippedConversations++;
+            continue;
+        }
+        const convName = conv.conversation_name || `Conversation ${convId}`;
+        result += `\n## ${convName}\n`;
+        for (const [threadId, messages] of Object.entries(conv.threads)) {
+            // Skip malformed thread entries
+            if (!Array.isArray(messages)) {
+                skippedThreads++;
+                continue;
+            }
+            if (messages.length > 1) {
+                result += `\n**Thread (${messages.length} messages):**\n`;
+            }
+            for (const msg of messages) {
+                if (!msg) {
+                    skippedMessages++;
+                    continue;
+                }
+                // Defensive date formatting to handle invalid dates gracefully
+                let dateStr = 'Unknown date';
+                if (msg.received_at) {
+                    const parsedDate = new Date(msg.received_at);
+                    dateStr = Number.isNaN(parsedDate.getTime()) ? 'Invalid date' : parsedDate.toLocaleDateString();
+                }
+                result += `- **${msg.sender_name || 'Unknown'}** (${dateStr}): `;
+                if (msg.body_text) {
+                    const preview = msg.body_text.length > MAX_BODY_PREVIEW_LENGTH
+                        ? msg.body_text.substring(0, MAX_BODY_PREVIEW_LENGTH) + '...'
+                        : msg.body_text;
+                    result += preview.replace(/\n/g, ' ');
+                }
+                else {
+                    result += '[Message content not included]';
+                }
+                result += '\n';
+            }
+        }
+    }
+    // Add warning if data was skipped due to malformed entries
+    const totalSkipped = skippedConversations + skippedThreads + skippedMessages;
+    if (totalSkipped > 0) {
+        const warnings = [];
+        if (skippedConversations > 0) {
+            warnings.push(`${skippedConversations} conversation(s)`);
+        }
+        if (skippedThreads > 0) {
+            warnings.push(`${skippedThreads} thread(s)`);
+        }
+        if (skippedMessages > 0) {
+            warnings.push(`${skippedMessages} message(s)`);
+        }
+        result += `\n---\n_Note: ${warnings.join(', ')} skipped due to incomplete data._\n`;
+    }
+    return result;
+}
+//# sourceMappingURL=search.js.map

package/esm/src/tools/search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"search.js","sourceRoot":"","sources":["../../../../../../packages/mcp/src/tools/search.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,uBAAuB,EAAE,MAAM,cAAc,CAAC;AAEvD;;GAEG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG;IAClC,IAAI,EAAE,gBAAgB;IACtB,WAAW,EAAE;;;;;;;;;;gDAUiC;IAC9C,WAAW,EAAE;QACX,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE;YACV,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,gDAAgD;aAC9D;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,4DAA4D;aAC1E;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,6DAA6D;aAC3E;YACD,cAAc,EAAE;gBACd,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACzB,WAAW,EAAE,+DAA+D;aAC7E;YACD,iBAAiB,EAAE;gBACjB,IAAI,EAAE,SAAS;gBACf,WAAW,EAAE,0CAA0C;gBACvD,OAAO,EAAE,IAAI;aACd;SACF;QACD,QAAQ,EAAE,CAAC,OAAO,CAAC;KACpB;CACF,CAAC;AAaF;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,MAAe,EACf,KAAsB;IAEtB,MAAM,OAAO,GAAkB;QAC7B,eAAe,EAAE,KAAK,CAAC,iBAAiB,KAAK,KAAK;KACnD,CAAC;IAEF,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;QACrB,OAAO,CAAC,SAAS,GAAG,KAAK,CAAC,UAAU,CAAC;IACvC,CAAC;IACD,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;QACtB,OAAO,CAAC,UAAU,GAAG,KAAK,CAAC,WAAW,CAAC;IACzC,CAAC;IACD,IAAI,KAAK,CAAC,cAAc,EAAE,MAAM,EAAE,CAAC;QACjC,OAAO,CAAC,aAAa,GAAG,KAAK,CAAC,cAAc,CAAC;IAC/C,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAE3D,8BAA8B;IAC9B,MAAM,YAAY,GAAG,QAAQ,CAAC,YAAY,EAAE,MAAM,IAAI,CAAC,CAAC;IAExD,IAAI,YAAY,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO,+BAA+B,KAAK,CAAC,KAAK,IAAI,CAAC;IACxD,CAAC;IAED,8CAA8C;IAC9C,IAAI,CAAC,QAAQ,CAAC,aAAa,IAAI,OAAO,QAAQ,CAAC,aAAa,KAAK,QAAQ,EAAE,CAAC;QAC1E,OAAO,SAAS,YAAY,kDAAkD,CAAC;IACjF,CAAC;IAED,MAAM,iBAAiB,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC;IACrE,IAAI,MAAM,GAAG,SAAS,YAAY,6BAA6B,iBAAiB,mBAAmB,CAAC;IAEpG,kCAAkC;IAClC,IAAI,oBAAoB,GAAG,CAAC,CAAC;IAC7B,IAAI,cAAc,GAAG,CAAC,CAAC;IACvB,IAAI,eAAe,GAAG,CAAC,CAAC;IAExB,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAmC,EAAE,CAAC;QACtG,sCAAsC;QACtC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAC3B,oBAAoB,EAAE,CAAC;YACvB,SAAS;QACX,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,IAAI,gBAAgB,MAAM,EAAE,CAAC;QACpE,MAAM,IAAI,QAAQ,QAAQ,IAAI,CAAC;QAE/B,KAAK,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAsC,EAAE,CAAC;YACrG,gCAAgC;YAChC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC7B,cAAc,EAAE,CAAC;gBACjB,SAAS;YACX,CAAC;YAED,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,MAAM,IAAI,eAAe,QAAQ,CAAC,MAAM,iBAAiB,CAAC;YAC5D,CAAC;YAED,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;gBAC3B,IAAI,CAAC,GAAG,EAAE,CAAC;oBACT,eAAe,EAAE,CAAC;oBAClB,SAAS;gBACX,CAAC;gBAED,+DAA+D;gBAC/D,IAAI,OAAO,GAAG,cAAc,CAAC;gBAC7B,IAAI,GAAG,CAAC,WAAW,EAAE,CAAC;oBACpB,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;oBAC7C,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,UAAU,CAAC,kBAAkB,EAAE,CAAC;gBAClG,CAAC;gBACD,MAAM,IAAI,OAAO,GAAG,CAAC,WAAW,IAAI,SAAS,OAAO,OAAO,KAAK,CAAC;gBAEjE,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;oBAClB,MAAM,OAAO,GAAG,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,uBAAuB;wBAC5D,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,EAAE,uBAAuB,CAAC,GAAG,KAAK;wBAC7D,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC;oBAClB,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;gBACxC,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,gCAAgC,CAAC;gBAC7C,CAAC;gBACD,MAAM,IAAI,IAAI,CAAC;YACjB,CAAC;QACH,CAAC;IACH,CAAC;IAED,2DAA2D;IAC3D,MAAM,YAAY,GAAG,oBAAoB,GAAG,cAAc,GAAG,eAAe,CAAC;IAC7E,IAAI,YAAY,GAAG,CAAC,EAAE,CAAC;QACrB,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,IAAI,oBAAoB,GAAG,CAAC,EAAE,CAAC;YAC7B,QAAQ,CAAC,IAAI,CAAC,GAAG,oBAAoB,kBAAkB,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,cAAc,GAAG,CAAC,EAAE,CAAC;YACvB,QAAQ,CAAC,IAAI,CAAC,GAAG,cAAc,YAAY,CAAC,CAAC;QAC/C,CAAC;QACD,IAAI,eAAe,GAAG,CAAC,EAAE,CAAC;YACxB,QAAQ,CAAC,IAAI,CAAC,GAAG,eAAe,aAAa,CAAC,CAAC;QACjD,CAAC;QACD,MAAM,IAAI,iBAAiB,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,qCAAqC,CAAC;IACtF,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}