test_model.py

import logging
import sys
from services.model_handler import ModelHandler

# Configure logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)

def test_model(force_default=False):
    """
    Testa o ModelHandler com uma consulta de exemplo.
    
    Args:
        force_default: Se deve forçar o uso da resposta padrão
    """
    logging.info("Initializing model handler...")
    model_handler = ModelHandler()
    
    # Definir a consulta de teste
    test_query = "O que é autismo?"
    logging.info(f"Testing with query: {test_query}")
    
    # Definir se deve forçar a resposta padrão
    model_handler.force_default_response = force_default
    
    # Gerar a resposta
    answer = model_handler.generate_answer(test_query)
    
    # Imprimir a resposta
    print("\n==================================================\n")
    if force_default:
        print("RESPOSTA PADRÃO FORÇADA:")
    else:
        print("RESPOSTA DO MODELO:")
    print(answer)
    print("\n==================================================\n")

if __name__ == "__main__":
    # Testar com a resposta do modelo
    test_model(force_default=False)
    
    # Testar com a resposta padrão forçada
    logging.info("Testing with forced default response...")
    test_model(force_default=True)