Update report_generator.py

report_generator.py

@@ -10,6 +10,15 @@ except ImportError:
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 groq_client = Groq(api_key=GROQ_API_KEY) if (HAS_GROQ and GROQ_API_KEY) else None
 
+# Lista de modele cu fallback automat
+GROQ_MODELS = [
+    "llama-3.3-70b-versatile",
+    "llama-3.1-8b-instant",
+    "llama3-8b-8192",
+    "gemma2-9b-it",
+    "mixtral-8x7b-32768"
+]
+
 def generate_fallback_markdown(diagnostic_data: dict) -> str:
     summary = diagnostic_data.get("executive_summary", {})
     results = diagnostic_data.get("results", [])
@@ -80,17 +89,21 @@ def generate_ai_report(diagnostic_data: dict) -> str:
         f"Failure Diagnostic JSON:\n{json.dumps(filtered_report)}"
     )
 
-    try:
-        completion = groq_client.chat.completions.create(
-            model="llama-3.1-8b-instant",
-            messages=[
-                {"role": "system", "content": "You are a precise LLMops assistant. Write concise diagnostic reports with git diff prompt patches."},
-                {"role": "user", "content": prompt}
-            ],
-            temperature=0.1,
-            max_tokens=400
-        )
-        return completion.choices[0].message.content
-    except Exception as e:
-        print(f"[Groq AI Report Error]: {e}")
-        return generate_fallback_markdown(diagnostic_data)
+    # Incercam pe rand modelele din lista de fallback
+    for model_name in GROQ_MODELS:
+        try:
+            completion = groq_client.chat.completions.create(
+                model=model_name,
+                messages=[
+                    {"role": "system", "content": "You are a precise LLMops assistant. Write concise diagnostic reports with git diff prompt patches."},
+                    {"role": "user", "content": prompt}
+                ],
+                temperature=0.1,
+                max_tokens=400
+            )
+            return completion.choices[0].message.content
+        except Exception:
+            continue
+
+    # Daca toate modelele Groq sunt indisponibile, apelam fallback-ul local
+    return generate_fallback_markdown(diagnostic_data)