Update ai_logic.py

ai_logic.py

@@ -3,7 +3,7 @@ import os
 import json
 import time
 import pickle
-from typing import Dict, List, Optional
+from typing import Dict, List, Optional, Tuple
 from bs4 import BeautifulSoup
 from urllib.parse import urljoin, urlparse
 from langchain.text_splitter import RecursiveCharacterTextSplitter
@@ -241,22 +241,24 @@ class HuggingFaceAPIClient:
             "temperature": 0.7
         }
         try:
-            response = requests.post(url, headers=self.headers, json=payload, timeout=60)
+            response = requests.post(url, headers=self.headers, json=payload, timeout=30)
             response.raise_for_status()
             return response.json()["choices"][0]["message"]["content"].strip()
-        except Exception as e:
-            return f"Erro na API: {str(e)}"
+        except requests.exceptions.HTTPError as http_err:
+            return f"Erro HTTP: {http_err.response.status_code} - {http_err.response.text}"
+        except requests.exceptions.RequestException as e:
+            return f"Erro na requisição: {str(e)}"
 
-    def test_model_availability(self, model_name: str) -> bool:
+    def test_model_availability(self, model_name: str) -> Tuple[bool, str]:
         """Testa se um modelo está disponível na API do Hugging Face."""
-        test_messages = [
-            {"role": "user", "content": "Teste de disponibilidade."}
-        ]
+        test_messages = [{"role": "user", "content": "Teste de disponibilidade."}]
         try:
             response = self.query_model(model_name, test_messages, max_tokens=10)
-            return not response.startswith("Erro na API")
-        except Exception:
-            return False
+            if response.startswith("Erro"):
+                return False, response
+            return True, "Modelo disponível."
+        except Exception as e:
+            return False, f"Erro ao testar modelo: {str(e)}"
 
 api_client = HuggingFaceAPIClient(HF_TOKEN)
 
@@ -264,13 +266,28 @@ api_client = HuggingFaceAPIClient(HF_TOKEN)
 def test_and_update_models():
     """Testa a disponibilidade dos novos modelos e atualiza a lista MODELS."""
     print("Testando disponibilidade dos novos modelos...")
+    available_models = []
+    unavailable_models = []
+
     for model_label, model_name in NEW_MODELS_TO_TEST:
-        if api_client.test_model_availability(model_name):
+        is_available, message = api_client.test_model_availability(model_name)
+        if is_available:
             MODELS[model_label] = model_name
-            print(f"Modelo {model_label} ({model_name}) adicionado à lista.")
+            available_models.append((model_label, model_name, message))
         else:
-            print(f"Modelo {model_label} ({model_name}) não disponível.")
-    print(f"Total de modelos disponíveis: {len(MODELS)}")
+            unavailable_models.append((model_label, model_name, message))
+
+    # Exibir resultados
+    print("\nResultados da validação:")
+    if available_models:
+        print("\nModelos disponíveis e adicionados:")
+        for label, name, msg in available_models:
+            print(f"- {label} ({name}): {msg}")
+    if unavailable_models:
+        print("\nModelos não disponíveis:")
+        for label, name, msg in unavailable_models:
+            print(f"- {label} ({name}): {msg}")
+    print(f"\nTotal de modelos disponíveis em MODELS: {len(MODELS)}")
 
 # --- Chat Principal ---
 def responder_como_aldo(session_id: str, pergunta: str, modelo: str = DEFAULT_MODEL) -> str: