Update multi_inference.py

multi_inference.py

@@ -3,16 +3,16 @@
 import requests
 import os
 
-# Load DeepSeek keys from Hugging Face Secrets
+# DeepSeek API keys
 DEEPSEEK_KEYS = [
+    os.getenv("DEEPSEEK_KEY", "").strip(),
     os.getenv("DEEPSEEK_KEY_1", "").strip(),
     os.getenv("DEEPSEEK_KEY_2", "").strip(),
     os.getenv("DEEPSEEK_KEY_3", "").strip(),
-    os.getenv("DEEPSEEK_KEY_4", "").strip(),
-    os.getenv("DEEPSEEK_KEY_5", "").strip()
+    os.getenv("DEEPSEEK_KEY_4", "").strip()
 ]
 
-# Load OpenAI keys named as open1, open2, etc.
+# OpenAI keys (named open1, open2, etc.)
 OPENAI_KEYS = [
     os.getenv("open1", "").strip(),
     os.getenv("open2", "").strip(),
@@ -21,7 +21,6 @@ OPENAI_KEYS = [
     os.getenv("open5", "").strip()
 ]
 
-
 def try_deepseek(prompt, key):
     try:
         url = "https://api.deepseek.com/v1/chat/completions"
@@ -40,11 +39,10 @@ def try_deepseek(prompt, key):
         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] DeepSeek: {result.get('error', {}).get('message', 'Unknown error')}"
     except Exception as e:
         return f"[ERROR] DeepSeek Exception: {str(e)}"
 
-
 def try_openai(prompt, key):
     try:
         url = "https://api.openai.com/v1/chat/completions"
@@ -63,21 +61,21 @@ def try_openai(prompt, key):
         result = res.json()
         if "choices" in result:
             return result["choices"][0]["message"]["content"]
-        return f"[ERROR] OpenAI: {result.get('error', {}).get('message', 'Unknown Error')}"
+        return f"[ERROR] OpenAI: {result.get('error', {}).get('message', 'Unknown error')}"
     except Exception as e:
         return f"[ERROR] OpenAI Exception: {str(e)}"
 
-
-# Main smart fallback wrapper
 def multi_query(prompt):
     for key in DEEPSEEK_KEYS:
-        output = try_deepseek(prompt, key)
-        if not output.startswith("[ERROR]"):
-            return output
+        if key:
+            response = try_deepseek(prompt, key)
+            if not response.startswith("[ERROR]"):
+                return response
 
     for key in OPENAI_KEYS:
-        output = try_openai(prompt, key)
-        if not output.startswith("[ERROR]"):
-            return output
+        if key:
+            response = try_openai(prompt, key)
+            if not response.startswith("[ERROR]"):
+                return response
 
-    return "[ALL APIs FAILED: Out of quota or network error]"
+    return "[ALL APIs FAILED: Out of quota or invalid keys]"