Update multi_inference.py

multi_inference.py

@@ -1,80 +1,34 @@
-# multi_inference.py (unsafe version — hardcoded API keys)
+# multi_inference.py (Groq only)
 
 import requests
 
-# === HARD CODED API KEYS ===
+# === HARDCODED GROQ KEY ===
+GROQ_API_KEY = "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"  # ✅ Active key
 
-OPENAI_KEYS = [
-    "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",
-    "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",
-    "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",
-    "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",
-    "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"
-]
-
-DEEPSEEK_KEYS = [
-    "sk-6de286df19b04cac8bf6295a2114b0aa",
-    "sk-5b4ecbbd66864a88925525fab14430d1",
-    "sk-9c216dd03c9945218f12475a0fa6c8c4",
-    "sk-79f0d8a98b88447880c01d012c572831",
-    "sk-2be33a5e561b4788b9446e4e4e99bc2e"
-]
-
-def try_openai(prompt, key):
+def try_groq(prompt):
     try:
-        url = "https://api.openai.com/v1/chat/completions"
+        url = "https://api.groq.com/openai/v1/chat/completions"
         headers = {
-            "Authorization": f"Bearer {key}",
+            "Authorization": f"Bearer {GROQ_API_KEY}",
             "Content-Type": "application/json"
         }
         data = {
-            "model": "gpt-3.5-turbo",
+            "model": "llama3-70b-8192",
             "messages": [
                 {"role": "system", "content": "You are a helpful assistant."},
                 {"role": "user", "content": prompt}
-            ]
+            ],
+            "temperature": 0.7
         }
-        res = requests.post(url, headers=headers, json=data)
-        result = res.json()
-        if "choices" in result:
-            return result["choices"][0]["message"]["content"]
-        return f"[ERROR] OpenAI: {result.get('error', {}).get('message', 'Unknown error')}"
-    except Exception as e:
-        return f"[ERROR] OpenAI Exception: {str(e)}"
 
-def try_deepseek(prompt, key):
-    try:
-        url = "https://api.deepseek.com/v1/chat/completions"
-        headers = {
-            "Authorization": f"Bearer {key}",
-            "Content-Type": "application/json"
-        }
-        data = {
-            "model": "deepseek-chat",
-            "messages": [
-                {"role": "system", "content": "You are a helpful assistant."},
-                {"role": "user", "content": prompt}
-            ]
-        }
         res = requests.post(url, headers=headers, json=data)
         result = res.json()
+
         if "choices" in result:
             return result["choices"][0]["message"]["content"]
-        return f"[ERROR] DeepSeek: {result.get('error', {}).get('message', 'Unknown error')}"
+        return f"[ERROR] Groq: {result.get('error', {}).get('message', 'Unknown error')}"
     except Exception as e:
-        return f"[ERROR] DeepSeek Exception: {str(e)}"
+        return f"[ERROR] Groq Exception: {str(e)}"
 
 def multi_query(prompt):
-    for key in OPENAI_KEYS:
-        if key:
-            response = try_openai(prompt, key)
-            if not response.startswith("[ERROR]"):
-                return response
-
-    for key in DEEPSEEK_KEYS:
-        if key:
-            response = try_deepseek(prompt, key)
-            if not response.startswith("[ERROR]"):
-                return response
-
-    return "[ALL APIs FAILED: Out of quota or invalid keys]"
+    return try_groq(prompt)