@razzium/piece-aimw-api 0.0.21 → 0.0.22

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@razzium/piece-aimw-api",
-  "version": "0.0.21",
+  "version": "0.0.22",
   "dependencies": {
     "tiktoken": "1.0.11"
   }

package/src/index.ts

@@ -4,7 +4,7 @@ import {
   createCustomApiCallAction,
   httpClient,
 } from '@activepieces/pieces-common';
-import { PieceAuth, createPiece } from '@activepieces/pieces-framework';
+import { PieceAuth, createPiece, Property } from '@activepieces/pieces-framework';
 import { PieceCategory } from '@activepieces/shared';
 import { askAssistant } from './lib/actions/ask-assistant';
 import { generateImage } from './lib/actions/generate-image';
@@ -43,31 +43,19 @@ It is strongly recommended that you add your credit card information to your Ope
 `,
 };
 
-export const openaiAuth = PieceAuth.SecretText({
+export const openaiAuth = PieceAuth.CustomAuth({
   description: AI_PROVIDERS_MAKRDOWN.aimw,
-  displayName: 'API Key',
   required: true,
-  validate: async (auth) => {
-    try {
-      await httpClient.sendRequest<{
-        data: { id: string }[];
-      }>({
-        url: `${baseUrl}/models`,
-        method: HttpMethod.GET,
-        authentication: {
-          type: AuthenticationType.BEARER_TOKEN,
-          token: auth.auth as string,
-        },
-      });
-      return {
-        valid: true,
-      };
-    } catch (e) {
-      return {
-        valid: false,
-        error: 'Invalid API key',
-      };
-    }
+  props: {
+    baseApiUrl: Property.ShortText({
+      displayName: 'Base URL',
+      required: true,
+      defaultValue: 'https://cloud.activepieces.com/api/v1',
+    }),
+    apiKey: PieceAuth.SecretText({
+      displayName: 'API Key',
+      required: true,
+    }),
   },
 });
 
@@ -80,6 +68,15 @@ export const openai = createPiece({
   auth: openaiAuth,
   actions: [
     askOpenAI,
+    createCustomApiCallAction({
+      baseUrl: (auth) => {
+        return `${(auth as { baseApiUrl: string }).baseApiUrl}`;
+      },
+      auth: openaiAuth,
+      authMapping: async (auth) => ({
+        Authorization: `Bearer ${(auth as { apiKey: string }).apiKey}`,
+      }),
+    }),
     // askAssistant,
     // generateImage,
     // visionPrompt,