@mastra/claude 1.0.0 → 1.0.2

package/src/client/schemas.gen.ts

@@ -0,0 +1,288 @@
+// This file is auto-generated by @hey-api/openapi-ts
+
+export const ErrorSchema = {
+    type: 'object',
+    properties: {
+        type: {
+            type: 'string',
+            nullable: false
+        },
+        message: {
+            type: 'string',
+            nullable: false
+        }
+    },
+    required: ['type', 'message']
+} as const;
+
+export const ErrorResponseSchema = {
+    type: 'object',
+    properties: {
+        error: {
+            '$ref': '#/components/schemas/Error'
+        }
+    },
+    required: ['error']
+} as const;
+
+export const CreateCompletionRequestSchema = {
+    type: 'object',
+    properties: {
+        model: {
+            description: `The model that will complete your prompt.
+As we improve Claude, we develop new versions of it that you can query.
+This parameter controls which version of Claude answers your request.
+Right now we are offering two model families: Claude, and Claude Instant.
+You can use them by setting model to "claude-2" or "claude-instant-1", respectively.
+See models for additional details.
+`,
+            oneOf: [
+                {
+                    type: 'string'
+                },
+                {
+                    type: 'string',
+                    enum: ['claude-2', 'claude-2.0', 'claude-instant-1', 'claude-instant-1.1']
+                }
+            ],
+            'x-oaiTypeLabel': 'string'
+        },
+        prompt: {
+            description: `The prompt that you want Claude to complete.
+
+For proper response generation you will need to format your prompt as follows:
+\n\nHuman: \${userQuestion}\n\nAssistant:
+See our comments on prompts for more context.
+`,
+            default: '<|endoftext|>',
+            nullable: true,
+            oneOf: [
+                {
+                    type: 'string',
+                    default: '',
+                    example: 'This is a test.'
+                },
+                {
+                    type: 'array',
+                    items: {
+                        type: 'string',
+                        default: '',
+                        example: 'This is a test.'
+                    }
+                },
+                {
+                    type: 'array',
+                    minItems: 1,
+                    items: {
+                        type: 'integer'
+                    },
+                    example: '[1212, 318, 257, 1332, 13]'
+                },
+                {
+                    type: 'array',
+                    minItems: 1,
+                    items: {
+                        type: 'array',
+                        minItems: 1,
+                        items: {
+                            type: 'integer'
+                        }
+                    },
+                    example: '[[1212, 318, 257, 1332, 13]]'
+                }
+            ]
+        },
+        max_tokens_to_sample: {
+            type: 'integer',
+            minimum: 1,
+            default: 256,
+            example: 256,
+            nullable: true,
+            description: `The maximum number of tokens to generate before stopping.
+
+Note that our models may stop before reaching this maximum. This parameter only specifies the absolute maximum number of tokens to generate.
+`
+        },
+        temperature: {
+            type: 'number',
+            minimum: 0,
+            maximum: 1,
+            default: 1,
+            example: 1,
+            nullable: true,
+            description: `Amount of randomness injected into the response.
+
+Defaults to 1. Ranges from 0 to 1. Use temp closer to 0 for analytical / multiple choice, and closer to 1 for creative and generative tasks.
+`
+        },
+        top_p: {
+            type: 'number',
+            minimum: 0,
+            maximum: 1,
+            default: 1,
+            example: 1,
+            nullable: true,
+            description: `Use nucleus sampling.
+
+In nucleus sampling, we compute the cumulative distribution over all the options 
+for each subsequent token in decreasing probability order and cut it off once 
+it reaches a particular probability specified by top_p. You should either alter temperature or top_p, but not both.
+`
+        },
+        top_k: {
+            type: 'number',
+            minimum: 0,
+            default: 5,
+            example: 5,
+            nullable: true,
+            description: `Only sample from the top K options for each subsequent token.
+
+Used to remove "long tail" low probability responses. Learn more technical details here.
+`
+        },
+        stream: {
+            description: `Whether to incrementally stream the response using server-sent events.
+See this guide to SSE events for details.type: boolean
+`,
+            nullable: true,
+            default: false
+        },
+        stop_sequences: {
+            description: `Sequences that will cause the model to stop generating completion text.
+Our models stop on "\n\nHuman:", and may include additional built-in stop sequences in the future. By providing the stop_sequences parameter, you may include additional strings that will cause the model to stop generating.
+`,
+            default: null,
+            nullable: true,
+            oneOf: [
+                {
+                    type: 'string',
+                    default: '<|endoftext|>',
+                    example: `
+`,
+                    nullable: true
+                },
+                {
+                    type: 'array',
+                    minItems: 1,
+                    maxItems: 4,
+                    items: {
+                        type: 'string',
+                        example: '["\\n"]'
+                    }
+                }
+            ]
+        },
+        metadata: {
+            type: 'object',
+            properties: {
+                user_id: {
+                    type: 'string',
+                    example: '13803d75-b4b5-4c3e-b2a2-6f21399b021b',
+                    description: `An external identifier for the user who is associated with the request.
+
+This should be a uuid, hash value, or other opaque identifier. Anthropic may use this id to help detect abuse. 
+Do not include any identifying information such as name, email address, or phone number.
+`
+                }
+            },
+            description: `An object describing metadata about the request.
+`
+        }
+    },
+    required: ['model', 'prompt', 'max_tokens_to_sample']
+} as const;
+
+export const CreateCompletionResponseSchema = {
+    type: 'object',
+    properties: {
+        stop_reason: {
+            type: 'string',
+            enum: ['stop_sequence', 'max_tokens'],
+            description: `The reason that we stopped sampling.
+
+This may be one the following values:
+
+"stop_sequence": we reached a stop sequence — either provided by you via the stop_sequences parameter, or a stop sequence built into the model
+"max_tokens": we exceeded max_tokens_to_sample or the model's maximum
+`
+        },
+        model: {
+            type: 'string',
+            description: `The model that performed the completion.
+`
+        },
+        completion: {
+            type: 'string',
+            description: `The resulting completion up to and excluding the stop sequences.
+`
+        }
+    },
+    required: ['completion', 'stop_reason', 'model']
+} as const;
+
+export const CreateCompletionStreamResponseSchema = {
+    type: 'object',
+    properties: {
+        stop_reason: {
+            type: 'string',
+            enum: ['stop_sequence', 'max_tokens'],
+            description: `The reason that we stopped sampling.
+
+This may be one the following values:
+
+"stop_sequence": we reached a stop sequence — either provided by you via the stop_sequences parameter, or a stop sequence built into the model
+"max_tokens": we exceeded max_tokens_to_sample or the model's maximum
+`
+        },
+        model: {
+            type: 'string',
+            description: `The model that performed the completion.
+`
+        },
+        completion: {
+            type: 'string',
+            description: `The resulting completion up to and excluding the stop sequences.
+`
+        },
+        choices: {
+            type: 'array',
+            items: {
+                type: 'object',
+                properties: {
+                    delta: {
+                        '$ref': '#/components/schemas/CompletionStreamResponseDelta'
+                    }
+                }
+            }
+        }
+    },
+    required: ['completion', 'stop_reason', 'model']
+} as const;
+
+export const CompletionStreamResponseDeltaSchema = {
+    type: 'object',
+    properties: {
+        completion: {
+            type: 'string',
+            description: 'The contents of the chunk message.',
+            nullable: true
+        },
+        stop_reason: {
+            type: 'string',
+            enum: ['stop_sequence', 'max_tokens'],
+            description: `The reason that we stopped sampling.
+
+This may be one the following values:
+
+"stop_sequence": we reached a stop sequence — either provided by you via the stop_sequences parameter, or a stop sequence built into the model
+"max_tokens": we exceeded max_tokens_to_sample or the model's maximum
+`,
+            nullable: true
+        },
+        model: {
+            type: 'string',
+            description: `The model that performed the completion.
+`
+        }
+    }
+} as const;

package/src/client/service-comments.ts

@@ -0,0 +1,6 @@
+export const comments = {
+  "complete": {
+    "comment": "Creates a completion for the provided prompt and parameters.",
+    "doc": "Creates a completion for the provided prompt and parameters."
+  }
+}

package/src/client/services.gen.ts

@@ -0,0 +1,14 @@
+// This file is auto-generated by @hey-api/openapi-ts
+
+import { createClient, createConfig, type Options } from '@hey-api/client-fetch';
+import type { CompleteData, CompleteError, CompleteResponse } from './types.gen';
+
+export const client = createClient(createConfig());
+
+/**
+ * Creates a completion for the provided prompt and parameters.
+ */
+export const complete = <ThrowOnError extends boolean = false>(options: Options<CompleteData, ThrowOnError>) => { return (options?.client ?? client).post<CompleteResponse, CompleteError, ThrowOnError>({
+    ...options,
+    url: '/complete'
+}); };

package/src/client/types.gen.ts

@@ -0,0 +1,178 @@
+// This file is auto-generated by @hey-api/openapi-ts
+
+export type Error = {
+    type: string;
+    message: string;
+};
+
+export type ErrorResponse = {
+    error: Error;
+};
+
+export type CreateCompletionRequest = {
+    /**
+     * The model that will complete your prompt.
+     * As we improve Claude, we develop new versions of it that you can query.
+     * This parameter controls which version of Claude answers your request.
+     * Right now we are offering two model families: Claude, and Claude Instant.
+     * You can use them by setting model to "claude-2" or "claude-instant-1", respectively.
+     * See models for additional details.
+     *
+     */
+    model: (string | 'claude-2' | 'claude-2.0' | 'claude-instant-1' | 'claude-instant-1.1');
+    /**
+     * The prompt that you want Claude to complete.
+     *
+     * For proper response generation you will need to format your prompt as follows:
+     * \n\nHuman: ${userQuestion}\n\nAssistant:
+     * See our comments on prompts for more context.
+     *
+     */
+    prompt: ((string | Array<(string)> | Array<(number)> | Array<Array<(number)>>) | null);
+    /**
+     * The maximum number of tokens to generate before stopping.
+     *
+     * Note that our models may stop before reaching this maximum. This parameter only specifies the absolute maximum number of tokens to generate.
+     *
+     */
+    max_tokens_to_sample: (number) | null;
+    /**
+     * Amount of randomness injected into the response.
+     *
+     * Defaults to 1. Ranges from 0 to 1. Use temp closer to 0 for analytical / multiple choice, and closer to 1 for creative and generative tasks.
+     *
+     */
+    temperature?: (number) | null;
+    /**
+     * Use nucleus sampling.
+     *
+     * In nucleus sampling, we compute the cumulative distribution over all the options
+     * for each subsequent token in decreasing probability order and cut it off once
+     * it reaches a particular probability specified by top_p. You should either alter temperature or top_p, but not both.
+     *
+     */
+    top_p?: (number) | null;
+    /**
+     * Only sample from the top K options for each subsequent token.
+     *
+     * Used to remove "long tail" low probability responses. Learn more technical details here.
+     *
+     */
+    top_k?: (number) | null;
+    /**
+     * Whether to incrementally stream the response using server-sent events.
+     * See this guide to SSE events for details.type: boolean
+     *
+     */
+    stream?: unknown;
+    /**
+     * Sequences that will cause the model to stop generating completion text.
+     * Our models stop on "\n\nHuman:", and may include additional built-in stop sequences in the future. By providing the stop_sequences parameter, you may include additional strings that will cause the model to stop generating.
+     *
+     */
+    stop_sequences?: ((string) | null | Array<(string)>);
+    /**
+     * An object describing metadata about the request.
+     *
+     */
+    metadata?: {
+        /**
+         * An external identifier for the user who is associated with the request.
+         *
+         * This should be a uuid, hash value, or other opaque identifier. Anthropic may use this id to help detect abuse.
+         * Do not include any identifying information such as name, email address, or phone number.
+         *
+         */
+        user_id?: string;
+    };
+};
+
+export type CreateCompletionResponse = {
+    /**
+     * The reason that we stopped sampling.
+     *
+     * This may be one the following values:
+     *
+     * "stop_sequence": we reached a stop sequence — either provided by you via the stop_sequences parameter, or a stop sequence built into the model
+     * "max_tokens": we exceeded max_tokens_to_sample or the model's maximum
+     *
+     */
+    stop_reason: 'stop_sequence' | 'max_tokens';
+    /**
+     * The model that performed the completion.
+     *
+     */
+    model: string;
+    /**
+     * The resulting completion up to and excluding the stop sequences.
+     *
+     */
+    completion: string;
+};
+
+/**
+ * The reason that we stopped sampling.
+ *
+ * This may be one the following values:
+ *
+ * "stop_sequence": we reached a stop sequence — either provided by you via the stop_sequences parameter, or a stop sequence built into the model
+ * "max_tokens": we exceeded max_tokens_to_sample or the model's maximum
+ *
+ */
+export type stop_reason = 'stop_sequence' | 'max_tokens';
+
+export type CreateCompletionStreamResponse = {
+    /**
+     * The reason that we stopped sampling.
+     *
+     * This may be one the following values:
+     *
+     * "stop_sequence": we reached a stop sequence — either provided by you via the stop_sequences parameter, or a stop sequence built into the model
+     * "max_tokens": we exceeded max_tokens_to_sample or the model's maximum
+     *
+     */
+    stop_reason: 'stop_sequence' | 'max_tokens';
+    /**
+     * The model that performed the completion.
+     *
+     */
+    model: string;
+    /**
+     * The resulting completion up to and excluding the stop sequences.
+     *
+     */
+    completion: string;
+    choices?: Array<{
+        delta?: CompletionStreamResponseDelta;
+    }>;
+};
+
+export type CompletionStreamResponseDelta = {
+    /**
+     * The contents of the chunk message.
+     */
+    completion?: (string) | null;
+    /**
+     * The reason that we stopped sampling.
+     *
+     * This may be one the following values:
+     *
+     * "stop_sequence": we reached a stop sequence — either provided by you via the stop_sequences parameter, or a stop sequence built into the model
+     * "max_tokens": we exceeded max_tokens_to_sample or the model's maximum
+     *
+     */
+    stop_reason?: ('stop_sequence' | 'max_tokens') | null;
+    /**
+     * The model that performed the completion.
+     *
+     */
+    model?: string;
+};
+
+export type CompleteData = {
+    body: CreateCompletionRequest;
+};
+
+export type CompleteResponse = (CreateCompletionResponse);
+
+export type CompleteError = unknown;

package/src/client/zodSchema.ts

@@ -0,0 +1,74 @@
+// Generated by ts-to-zod
+import { z } from 'zod';
+
+export const errorSchema = z.object({
+  type: z.string(),
+  message: z.string(),
+});
+
+export const errorResponseSchema = z.object({
+  error: errorSchema,
+});
+
+export const createCompletionRequestSchema = z.object({
+  model: z.union([
+    z.string(),
+    z.literal('claude-2'),
+    z.literal('claude-2.0'),
+    z.literal('claude-instant-1'),
+    z.literal('claude-instant-1.1'),
+  ]),
+  prompt: z.union([z.string(), z.array(z.string()), z.array(z.number()), z.array(z.array(z.number()))]).nullable(),
+  max_tokens_to_sample: z.number().nullable(),
+  temperature: z.number().optional().nullable(),
+  top_p: z.number().optional().nullable(),
+  top_k: z.number().optional().nullable(),
+  stream: z.unknown().optional(),
+  stop_sequences: z
+    .union([z.string(), z.array(z.string())])
+    .optional()
+    .nullable(),
+  metadata: z
+    .object({
+      user_id: z.string().optional(),
+    })
+    .optional(),
+});
+
+export const createCompletionResponseSchema = z.object({
+  stop_reason: z.union([z.literal('stop_sequence'), z.literal('max_tokens')]),
+  model: z.string(),
+  completion: z.string(),
+});
+
+export const stopReasonSchema = z.union([z.literal('stop_sequence'), z.literal('max_tokens')]);
+
+export const completionStreamResponseDeltaSchema = z.object({
+  completion: z.string().optional().nullable(),
+  stop_reason: z
+    .union([z.literal('stop_sequence'), z.literal('max_tokens')])
+    .optional()
+    .nullable(),
+  model: z.string().optional(),
+});
+
+export const completeDataSchema = z.object({
+  body: createCompletionRequestSchema,
+});
+
+export const completeResponseSchema = createCompletionResponseSchema;
+
+export const completeErrorSchema = z.unknown();
+
+export const createCompletionStreamResponseSchema = z.object({
+  stop_reason: z.union([z.literal('stop_sequence'), z.literal('max_tokens')]),
+  model: z.string(),
+  completion: z.string(),
+  choices: z
+    .array(
+      z.object({
+        delta: completionStreamResponseDeltaSchema.optional(),
+      }),
+    )
+    .optional(),
+});

package/src/index.ts

@@ -1,14 +1,16 @@
-import { Integration, OpenAPI, IntegrationCredentialType, IntegrationAuth } from '@mastra/core';
-import { createClient, type OASClient, type NormalizeOAS } from 'fets';
+import { Integration, IntegrationCredentialType, IntegrationAuth } from '@mastra/core';
 import { z } from 'zod';
 
 // @ts-ignore
-import ClaudeLogo from './assets/claude.svg';
-import { openapi } from './openapi';
-import { components } from './openapi-components';
-import { paths } from './openapi-paths';
+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';
 
 export class ClaudeIntegration extends Integration {
+  categories = ['ai', 'communications'];
+  description = 'Claude is a next generation AI assistant built for work and trained to be safe, accurate, and secure.';
+
   constructor() {
     super({
       authType: IntegrationCredentialType.API_KEY,
@@ -20,30 +22,38 @@ export class ClaudeIntegration extends Integration {
     });
   }
 
-  getOpenApiSpec() {
-    return { paths, components } as unknown as OpenAPI;
+  getClientZodSchema() {
+    return zodSchema;
+  }
+
+  getCommentsForClientApis() {
+    return comments;
   }
 
-  getApiClient = async ({ connectionId }: { connectionId: string }): Promise<OASClient<NormalizeOAS<openapi>>> => {
+  getBaseClient() {
+    integrationClient.client.setConfig({
+      baseUrl: `https://api.anthropic.com/v1`,
+    });
+    return integrationClient;
+  }
+
+  getApiClient = async ({ connectionId }: { connectionId: string }) => {
     const connection = await this.dataLayer?.getConnection({ name: this.name, connectionId });
 
     if (!connection) {
       throw new Error(`Connection not found for connectionId: ${connectionId}`);
     }
-
     const credential = await this.dataLayer?.getCredentialsByConnection(connection.id);
-    const value = credential?.value as Record<string, string>;
+    const value = credential?.value as Record<string, any>;
 
-    const client = createClient<NormalizeOAS<openapi>>({
-      endpoint: `https://api.anthropic.com/v1`,
-      globalParams: {
-        headers: {
-          'x-api-key': value?.['ANTHROPIC_API_KEY'],
-        },
-      },
+    const baseClient = this.getBaseClient();
+
+    baseClient.client.interceptors.request.use((request, options) => {
+      request.headers.set('x-api-key', value?.['ANTHROPIC_API_KEY']);
+      return request;
     });
 
-    return client;
+    return integrationClient;
   };
 
   registerEvents() {