Upload src/lib/aiService.ts with huggingface_hub

src/lib/aiService.ts

@@ -0,0 +1,45 @@
+import axios from 'axios';
+
+const OPENROUTER_API_URL = 'https://openrouter.ai/api/v1/chat/completions';
+
+const TARGET_MODELS = [
+  'mistralai/mistral-small',
+  'mistralai/mistral-medium',
+  'mistralai/mistral-nemo' // fallback free-tier compatible model
+];
+
+let currentModelIndex = 0;
+
+export async function getAiCompletion(prompt: string) {
+  let attempts = 0;
+  
+  while (attempts < TARGET_MODELS.length) {
+    const model = TARGET_MODELS[currentModelIndex];
+    try {
+      const response = await axios.post(
+        OPENROUTER_API_URL,
+        {
+          model: model,
+          messages: [{ role: 'user', content: prompt }],
+        },
+        {
+          headers: {
+            'Authorization': `Bearer ${process.env.OPENROUTER_API_KEY}`,
+            'HTTP-Referer': process.env.APP_URL || 'http://localhost:3000',
+            'X-Title': 'BRIDGEModulE Dashboard',
+          },
+        }
+      );
+      
+      // If successful, we keep this model as the active one for next time
+      return response.data.choices[0].message.content;
+    } catch (error: any) {
+      console.error(`Error with model ${model}:`, error?.response?.data || error.message);
+      attempts++;
+      // Rotate to the next model for the next attempt
+      currentModelIndex = (currentModelIndex + 1) % TARGET_MODELS.length;
+    }
+  }
+  
+  throw new Error('All OpenRouter Mistral models failed. Please check your API key and quotas.');
+}