@mastra/claude 1.0.1-alpha.11 → 1.0.1-alpha.14

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/claude",
-  "version": "1.0.1-alpha.11",
+  "version": "1.0.1-alpha.14",
   "description": "",
   "main": "dist/index.js",
   "module": "dist/claude.esm.js",
@@ -50,7 +50,7 @@
     "@hey-api/client-fetch": "^0.3.3",
     "zod": "^3.24.0",
     "ts-to-zod": "^3.13.0",
-    "@mastra/core": "^0.1.27-alpha.32"
+    "@mastra/core": "^0.1.27-alpha.35"
   },
   "scripts": {
     "analyze": "size-limit --why",
@@ -59,7 +59,7 @@
     "lint": "dts lint",
     "size": "size-limit",
     "start": "dts watch",
-    "test": "jest",
+    "test": "node --experimental-vm-modules node_modules/jest/bin/jest.js",
     "clean": "rm -rf dist && rm -rf node_modules",
     "gen:zod:schema": "pnpx ts-to-zod  src/client/types.gen.ts src/client/zodSchema.ts"
   }

package/src/index.ts

@@ -1,21 +1,15 @@
-import { Integration, ToolApi } from '@mastra/core';
-
+import { Integration } from '@mastra/core';
+import * as integrationClient from './client/services.gen';
 // @ts-ignore
 import ClaudeLogo from './assets/claude.png';
-import { comments } from './client/service-comments';
-import * as integrationClient from './client/services.gen';
-import * as zodSchema from './client/zodSchema';
+import { ClaudeConfig } from './types';
+import { ClaudeToolset } from './toolset';
 
-type ClaudeConfig = {
-  ANTHROPIC_API_KEY: string;
-  [key: string]: any;
-};
 
-export class ClaudeIntegration extends Integration {
+export class ClaudeIntegration extends Integration<void, typeof integrationClient> {
   readonly name = 'CLAUDE';
   readonly logoUrl = ClaudeLogo;
   config: ClaudeConfig;
-  readonly tools: Record<Exclude<keyof typeof integrationClient, 'client'>, ToolApi>;
   categories = ['ai', 'communications'];
   description = 'Claude is a next generation AI assistant built for work and trained to be safe, accurate, and secure.';
 
@@ -23,36 +17,15 @@ export class ClaudeIntegration extends Integration {
     super();
 
     this.config = config;
-    this.tools = this._generateIntegrationTools<typeof this.tools>();
-  }
 
-  protected get toolSchemas() {
-    return zodSchema;
   }
 
-  protected get toolDocumentations() {
-    return comments;
-  }
+  getStaticTools() {
+    const openapi = new ClaudeToolset({
+      config: this.config,
+    })
 
-  protected get baseClient() {
-    integrationClient.client.setConfig({
-      baseUrl: `https://api.anthropic.com/v1`,
-    });
-    return integrationClient;
+    return openapi.tools;
   }
 
-  getApiClient = async () => {
-    const value = {
-      'x-api-key': this.config?.['ANTHROPIC_API_KEY'],
-    } as Record<string, any>;
-
-    const baseClient = this.baseClient;
-
-    baseClient.client.interceptors.request.use((request, options) => {
-      request.headers.set('x-api-key', value?.['ANTHROPIC_API_KEY']);
-      return request;
-    });
-
-    return integrationClient;
-  };
 }

package/src/toolset.ts

@@ -0,0 +1,55 @@
+import { OpenAPIToolset, ToolAction } from '@mastra/core';
+
+// @ts-ignore
+import ClaudeLogo from './assets/claude.png';
+import { comments } from './client/service-comments';
+import * as integrationClient from './client/services.gen';
+import * as zodSchema from './client/zodSchema';
+import { ClaudeConfig } from './types';
+
+
+export class ClaudeToolset extends OpenAPIToolset {
+    readonly name = 'CLAUDE';
+    readonly logoUrl = ClaudeLogo;
+    config: ClaudeConfig;
+    readonly tools: Record<Exclude<keyof typeof integrationClient, 'client'>, ToolAction<any, any, any, any>>;
+    categories = ['ai', 'communications'];
+    description = 'Claude is a next generation AI assistant built for work and trained to be safe, accurate, and secure.';
+
+    constructor({ config }: { config: ClaudeConfig }) {
+        super();
+
+        this.config = config;
+        this.tools = this._generateIntegrationTools<typeof this.tools>();
+    }
+
+    protected get toolSchemas() {
+        return zodSchema;
+    }
+
+    protected get toolDocumentations() {
+        return comments;
+    }
+
+    protected get baseClient() {
+        integrationClient.client.setConfig({
+            baseUrl: `https://api.anthropic.com/v1`,
+        });
+        return integrationClient;
+    }
+
+    getApiClient = async () => {
+        const value = {
+            'x-api-key': this.config?.['ANTHROPIC_API_KEY'],
+        } as Record<string, any>;
+
+        const baseClient = this.baseClient;
+
+        baseClient.client.interceptors.request.use((request, options) => {
+            request.headers.set('x-api-key', value?.['ANTHROPIC_API_KEY']);
+            return request;
+        });
+
+        return integrationClient;
+    };
+}

package/src/types.ts

@@ -0,0 +1,4 @@
+export type ClaudeConfig = {
+    ANTHROPIC_API_KEY: string;
+    [key: string]: any;
+};