Update report_generator.py

report_generator.py

@@ -49,21 +49,16 @@ def generate_ai_report(diagnostic_data: dict) -> str:
     if not failed_sessions_only:
         return "# Executive Health & Deployment Readiness\n\nAll multi-turn trajectories operated at optimal parameters. The agent achieved a **100.0% Success Rate (Health Rating: [A])**. Recommended for production deployment."
 
-    # Citim cheia direct la apel
     groq_api_key = os.getenv("GROQ_API_KEY")
 
-    if not HAS_GROQ:
-        print("[Groq Debug]: Pachetul groq nu este importat.")
-        return generate_fallback_markdown(diagnostic_data)
-
-    if not groq_api_key:
-        print("[Groq Debug]: GROQ_API_KEY nu a fost gasita in os.getenv(). Asigura-te ca e setata in Space Secrets!")
+    if not HAS_GROQ or not groq_api_key:
+        print("[Groq Debug]: Groq sau GROQ_API_KEY indisponibil.")
         return generate_fallback_markdown(diagnostic_data)
 
     try:
         groq_client = Groq(api_key=groq_api_key)
     except Exception as e:
-        print(f"[Groq Debug]: Eroare la initializarea clientului: {e}")
+        print(f"[Groq Debug]: Eroare la client Groq: {e}")
         return generate_fallback_markdown(diagnostic_data)
 
     filtered_report = {
@@ -92,14 +87,19 @@ def generate_ai_report(diagnostic_data: dict) -> str:
         f"Failure Diagnostic JSON:\n{json.dumps(filtered_report)}"
     )
 
-    models_to_try = [
-        "llama-3.3-70b-versatile",
-        "llama-3.1-8b-instant"
-    ]
+    # 1. Intrebam Groq ce modele sunt disponibile in contul tau
+    try:
+        models_list = groq_client.models.list()
+        available_models = [m.id for m in models_list.data if "whisper" not in m.id.lower() and "embed" not in m.id.lower()]
+        print(f"[Groq Debug]: Modele active detectate in contul tau: {available_models}")
+    except Exception as e:
+        print(f"[Groq Debug]: Nu am putut lista modelele: {e}")
+        available_models = ["llama-3.3-70b-versatile", "llama-3.1-8b-instant", "gemma2-9b-it"]
 
-    for model_name in models_to_try:
+    # 2. Rulam primul model disponibil
+    for model_name in available_models:
         try:
-            print(f"[Groq Debug]: Incercam apelul catre modelul {model_name}...")
+            print(f"[Groq Debug]: Incercam apelul catre modelul detectat: [{model_name}]...")
             completion = groq_client.chat.completions.create(
                 model=model_name,
                 messages=[
@@ -109,7 +109,7 @@ def generate_ai_report(diagnostic_data: dict) -> str:
                 temperature=0.1,
                 max_tokens=400
             )
-            print(f"[Groq Debug]: SUCCESS cu modelul {model_name}!")
+            print(f"[Groq Debug]: 🎉 SUCCESS cu modelul [{model_name}]!")
             return completion.choices[0].message.content
         except Exception as e:
             print(f"[Groq Debug]: Eroare la modelul {model_name}: {e}")