@razzium/piece-aimw-api 0.0.14 → 0.0.16

package/package.json

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

package/src/index.ts

@@ -57,16 +57,17 @@ export const openaiAuth = PieceAuth.SecretText({
         authentication: {
           type: AuthenticationType.BEARER_TOKEN,
           token: auth.auth as string,
-          //token: "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpZCI6ImZjY2Y4NzQyLTUyZWItNDdmZS05YWNjLTFjYzUyZjc2ZDdkYiJ9.NSWnWMO0ERGRU3QT4are2u-fo7R1FJxb2byyIwInTj4",
         },
       });
       return {
         valid: true,
       };
-    } catch (e) {
+    } catch (e: any) {
+      console.error('Auth validation error:', e?.message || e);
+      const errorMessage = e?.message || e?.response?.data?.message || 'Invalid API key';
       return {
         valid: false,
-        error: 'Invalid API key',
+        error: errorMessage,
       };
     }
   },

package/src/lib/actions/send-prompt.ts

@@ -219,16 +219,28 @@ export const askOpenAI = createAction({
     // });
 
     // Send prompt
-    const completion = await openai.chat.completions.create({
-      model: model,
-      messages: [...messageHistory],
-      // messages: [...roles, ...messageHistory],
-      // temperature: temperature,
-      // top_p: topP,
-      // frequency_penalty: frequencyPenalty,
-      // presence_penalty: presencePenalty,
-      // max_completion_tokens: maxTokens,
-    });
+    let completion;
+    try {
+      completion = await openai.chat.completions.create({
+        model: model,
+        messages: [...messageHistory],
+        // messages: [...roles, ...messageHistory],
+        // temperature: temperature,
+        // top_p: topP,
+        // frequency_penalty: frequencyPenalty,
+        // presence_penalty: presencePenalty,
+        // max_completion_tokens: maxTokens,
+      });
+    } catch (error: any) {
+      const errorMessage = error?.message || error?.error?.message || JSON.stringify(error);
+      const errorStatus = error?.status || error?.response?.status || 'unknown';
+      throw new Error(`AIMW API Error (${errorStatus}): ${errorMessage}`);
+    }
+
+    // Validate API response
+    if (!completion?.choices?.[0]?.message) {
+      throw new Error('Invalid API response: no completion message returned');
+    }
 
     // Add response to message history
     messageHistory = [...messageHistory, completion.choices[0].message];