@mastra/openai 1.0.0 → 1.0.1-alpha.1

package/src/index.ts

@@ -1,67 +1,59 @@
-import { Integration, OpenAPI, IntegrationCredentialType, IntegrationAuth } from '@mastra/core';
-import { createClient, type OASClient, type NormalizeOAS } from 'fets';
-import { z } from 'zod';
+import { Integration, ToolApi } from '@mastra/core';
 
 // @ts-ignore
 import OpenaiLogo from './assets/openai.png';
-import { openapi } from './openapi';
-import { components } from './openapi-components';
-import { paths } from './openapi-paths';
+import { comments } from './client/service-comments';
+import * as integrationClient from './client/services.gen';
+import * as zodSchema from './client/zodSchema';
+
+type OpenaiConfig = {
+  API_KEY: string;
+  [key: string]: any;
+};
 
 export class OpenaiIntegration extends Integration {
-  constructor() {
-    super({
-      authType: IntegrationCredentialType.API_KEY,
-      name: 'OPENAI',
-      logoUrl: OpenaiLogo,
-      authConnectionOptions: z.object({
-        API_KEY: z.string(),
-      }),
-    });
+  readonly name = 'OPENAI';
+  readonly logoUrl = OpenaiLogo;
+  config: OpenaiConfig;
+  readonly tools: Record<Exclude<keyof typeof integrationClient, 'client'>, ToolApi>;
+  categories = ['ai'];
+  description =
+    'OpenAI is an artificial intelligence platform that provides a set of tools and APIs for building AI-powered applications.';
+
+  constructor({ config }: { config: OpenaiConfig }) {
+    super();
+
+    this.config = config;
+    this.tools = this._generateIntegrationTools<typeof this.tools>();
   }
 
-  getOpenApiSpec() {
-    return { paths, components } as unknown as OpenAPI;
+  protected get toolSchemas() {
+    return zodSchema;
   }
 
-  getApiClient = async ({ connectionId }: { connectionId: string }): Promise<OASClient<NormalizeOAS<openapi>>> => {
-    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>;
+  protected get toolDocumentations() {
+    return comments;
+  }
 
-    const client = createClient<NormalizeOAS<openapi>>({
-      endpoint: `https://api.openai.com/v1`,
-      globalParams: {
-        headers: {
-          Authorization: `Bearer ${value?.['API_KEY']}`,
-        },
-      },
+  protected get baseClient() {
+    integrationClient.client.setConfig({
+      baseUrl: `https://api.openai.com/v1`,
     });
+    return integrationClient;
+  }
 
-    return client as any;
-  };
+  getApiClient = async () => {
+    const value = {
+      API_KEY: this.config?.['API_KEY'],
+    } as Record<string, any>;
 
-  registerEvents() {
-    this.events = {};
-    return this.events;
-  }
+    const baseClient = this.baseClient;
 
-  getAuthenticator() {
-    return new IntegrationAuth({
-      dataAccess: this.dataLayer!,
-      // @ts-ignore
-      onConnectionCreated: () => {
-        // TODO
-      },
-      config: {
-        INTEGRATION_NAME: this.name,
-        AUTH_TYPE: this.config.authType,
-      },
+    baseClient.client.interceptors.request.use((request, options) => {
+      request.headers.set('Authorization', `Bearer ${value?.['API_KEY']}`);
+      return request;
     });
-  }
+
+    return integrationClient;
+  };
 }

package/src/assets/openai.svg

@@ -1,37 +0,0 @@
-
-<svg version="1.1" id="Capa_1" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink"
-	 viewBox="0 0 297.5 297.5" xml:space="preserve">
-<g id="XMLID_40_">
-	<g>
-		<path style="fill:#ACBFC7;" d="M277.71,158.52v85.7H19.79v-85.7h6.53v40.54c0,16.98,13.81,30.78,30.78,30.78
-			s30.78-13.8,30.78-30.78v-40.54h30.09v40.54c0,16.98,13.81,30.78,30.78,30.78c16.98,0,30.78-13.8,30.78-30.78v-40.54h30.1v40.54
-			c0,16.98,13.8,30.78,30.78,30.78c16.97,0,30.78-13.8,30.78-30.78v-40.54H277.71z"/>
-		<rect x="218.66" y="53.28" style="fill:#CDD9DD;" width="43.49" height="10.53"/>
-		<rect x="229.17" y="83.35" style="fill:#CDD9DD;" width="22.48" height="23.92"/>
-		<rect x="137.51" y="83.35" style="fill:#CDD9DD;" width="22.48" height="23.92"/>
-		<rect x="127.01" y="53.28" style="fill:#CDD9DD;" width="43.49" height="10.53"/>
-		<rect x="35.35" y="53.28" style="fill:#CDD9DD;" width="43.49" height="10.53"/>
-		<rect x="45.86" y="83.35" style="fill:#CDD9DD;" width="22.48" height="23.92"/>
-		<path style="fill:#FF4855;" d="M251.65,126.81v72.25c0,6.2-5.05,11.24-11.24,11.24c-6.2,0-11.24-5.04-11.24-11.24v-72.25H251.65z"
-			/>
-		<path style="fill:#D61616;" d="M68.34,126.81v72.25c0,6.2-5.04,11.24-11.24,11.24s-11.24-5.04-11.24-11.24v-72.25H68.34z"/>
-		<path style="fill:#FFD63F;" d="M159.99,126.81v72.25c0,6.2-5.04,11.24-11.24,11.24s-11.24-5.04-11.24-11.24v-72.25H159.99z"/>
-		<path d="M297.25,148.75v105.24c0,5.4-4.37,9.77-9.77,9.77H10.02c-5.39,0-9.77-4.37-9.77-9.77V148.75c0-5.4,4.38-9.77,9.77-9.77
-			h16.3V83.35h-0.74c-5.39,0-9.77-4.38-9.77-9.77V43.51c0-5.4,4.38-9.77,9.77-9.77h63.03c5.4,0,9.77,4.37,9.77,9.77v30.07
-			c0,5.39-4.37,9.77-9.77,9.77h-0.73v55.63h30.09V83.35h-0.73c-5.4,0-9.77-4.38-9.77-9.77V43.51c0-5.4,4.37-9.77,9.77-9.77h63.03
-			c5.39,0,9.77,4.37,9.77,9.77v30.07c0,5.39-4.38,9.77-9.77,9.77h-0.74v55.63h30.1V83.35h-0.74c-5.39,0-9.77-4.38-9.77-9.77V43.51
-			c0-5.4,4.38-9.77,9.77-9.77h63.03c5.4,0,9.77,4.37,9.77,9.77v30.07c0,5.39-4.37,9.77-9.77,9.77h-0.73v55.63h16.29
-			C292.88,138.98,297.25,143.35,297.25,148.75z M277.71,244.22v-85.7h-6.52v40.54c0,16.98-13.81,30.78-30.78,30.78
-			c-16.98,0-30.78-13.8-30.78-30.78v-40.54h-30.1v40.54c0,16.98-13.8,30.78-30.78,30.78c-16.97,0-30.78-13.8-30.78-30.78v-40.54
-			H87.88v40.54c0,16.98-13.81,30.78-30.78,30.78s-30.78-13.8-30.78-30.78v-40.54h-6.53v85.7H277.71z M262.15,63.81V53.28h-43.49
-			v10.53H262.15z M251.65,199.06v-72.25h-22.48v72.25c0,6.2,5.04,11.24,11.24,11.24C246.6,210.3,251.65,205.26,251.65,199.06z
-			 M251.65,107.27V83.35h-22.48v23.92H251.65z M170.5,63.81V53.28h-43.49v10.53H170.5z M159.99,199.06v-72.25h-22.48v72.25
-			c0,6.2,5.04,11.24,11.24,11.24S159.99,205.26,159.99,199.06z M159.99,107.27V83.35h-22.48v23.92H159.99z M78.84,63.81V53.28H35.35
-			v10.53H78.84z M68.34,199.06v-72.25H45.86v72.25c0,6.2,5.04,11.24,11.24,11.24S68.34,205.26,68.34,199.06z M68.34,107.27V83.35
-			H45.86v23.92H68.34z"/>
-	</g>
-	<g>
-	</g>
-</g>
-</svg>
-