@mastra/openai 1.0.0

package/package.json

@@ -0,0 +1,63 @@
+{
+  "name": "@mastra/openai",
+  "version": "1.0.0",
+  "description": "",
+  "main": "dist/index.js",
+  "module": "dist/mylib.esm.js",
+  "typings": "dist/index.d.ts",
+  "files": [
+    "dist",
+    "src"
+  ],
+  "husky": {
+    "hooks": {
+      "pre-commit": "dts lint"
+    }
+  },
+  "engines": {
+    "node": ">=20 <22"
+  },
+  "size-limit": [
+    {
+      "path": "dist/mylib.cjs.production.min.js",
+      "limit": "10 KB"
+    },
+    {
+      "path": "dist/mylib.esm.js",
+      "limit": "10 KB"
+    }
+  ],
+  "devDependencies": {
+    "@jest/globals": "^29.7.0",
+    "@rollup/plugin-image": "^3.0.3",
+    "@size-limit/preset-small-lib": "^11.1.4",
+    "@tsconfig/recommended": "^1.0.7",
+    "@types/jest": "^29.5.12",
+    "@types/lodash": "^4.17.7",
+    "@types/node": "^22.1.0",
+    "dts-cli": "^2.0.5",
+    "husky": "^9.1.4",
+    "jest": "^29.7.0",
+    "size-limit": "^11.1.4",
+    "ts-jest": "^29.2.4",
+    "tslib": "^2.6.3",
+    "typescript": "^5.5.4"
+  },
+  "keywords": [],
+  "author": "",
+  "license": "ISC",
+  "dependencies": {
+    "fets": "*",
+    "zod": "^3.23.8",
+    "@mastra/core": "0.1.0"
+  },
+  "scripts": {
+    "analyze": "size-limit --why",
+    "build": "dts build",
+    "build:dev": "dts watch",
+    "lint": "dts lint",
+    "size": "size-limit",
+    "start": "dts watch",
+    "test": "jest"
+  }
+}

package/src/Openai.test.ts

@@ -0,0 +1,59 @@
+import {
+  describe,
+  it,
+  beforeAll,
+  afterAll, //expect
+} from '@jest/globals';
+import { Framework } from '@mastra/core';
+
+import { OpenaiIntegration } from '.';
+
+const API_KEY = process.env.API_KEY!;
+const dbUri = process.env.DB_URL!;
+const connectionId = process.env.CONNECTION_ID!;
+
+const integrationName = 'OPENAI';
+
+const integrationFramework = Framework.init({
+  name: 'TestFramework',
+  integrations: [new OpenaiIntegration()],
+  systemApis: [],
+  systemEvents: {},
+  db: {
+    provider: 'postgres',
+    uri: dbUri,
+  },
+  systemHostURL: 'http://localhost:3000',
+  routeRegistrationPath: '/api/mastra',
+  blueprintDirPath: '',
+});
+
+//const integration = integrationFramework.getIntegration(integrationName) as OpenaiIntegration
+
+describe('openai', () => {
+  beforeAll(async () => {
+    await integrationFramework.connectIntegrationByCredential({
+      name: integrationName,
+      connectionId,
+      credential: {
+        value: {
+          API_KEY,
+        },
+        type: 'API_KEY',
+      },
+    });
+  });
+
+  it('should 200 on some apis', async () => {
+    //const client = await integration.getApiClient({ connectionId });
+    //const response = await client['/2010-04-01/Accounts.json'].get();
+    //expect(response.status).toBe(200);
+  });
+
+  afterAll(async () => {
+    await integrationFramework.disconnectIntegration({
+      name: integrationName,
+      connectionId,
+    });
+  });
+});

package/src/assets/openai.svg

@@ -0,0 +1,37 @@
+
+<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>
+    

package/src/index.ts

@@ -0,0 +1,67 @@
+import { Integration, OpenAPI, IntegrationCredentialType, IntegrationAuth } from '@mastra/core';
+import { createClient, type OASClient, type NormalizeOAS } from 'fets';
+import { z } from 'zod';
+
+// @ts-ignore
+import OpenaiLogo from './assets/openai.png';
+import { openapi } from './openapi';
+import { components } from './openapi-components';
+import { paths } from './openapi-paths';
+
+export class OpenaiIntegration extends Integration {
+  constructor() {
+    super({
+      authType: IntegrationCredentialType.API_KEY,
+      name: 'OPENAI',
+      logoUrl: OpenaiLogo,
+      authConnectionOptions: z.object({
+        API_KEY: z.string(),
+      }),
+    });
+  }
+
+  getOpenApiSpec() {
+    return { paths, components } as unknown as OpenAPI;
+  }
+
+  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>;
+
+    const client = createClient<NormalizeOAS<openapi>>({
+      endpoint: `https://api.openai.com/v1`,
+      globalParams: {
+        headers: {
+          Authorization: `Bearer ${value?.['API_KEY']}`,
+        },
+      },
+    });
+
+    return client as any;
+  };
+
+  registerEvents() {
+    this.events = {};
+    return this.events;
+  }
+
+  getAuthenticator() {
+    return new IntegrationAuth({
+      dataAccess: this.dataLayer!,
+      // @ts-ignore
+      onConnectionCreated: () => {
+        // TODO
+      },
+      config: {
+        INTEGRATION_NAME: this.name,
+        AUTH_TYPE: this.config.authType,
+      },
+    });
+  }
+}