Spaces:
Sleeping
Sleeping
fix: rever
Browse files- test_model.py +0 -46
test_model.py
DELETED
@@ -1,46 +0,0 @@
|
|
1 |
-
import logging
|
2 |
-
import sys
|
3 |
-
from services.model_handler import ModelHandler
|
4 |
-
|
5 |
-
# Configure logging
|
6 |
-
logging.basicConfig(
|
7 |
-
level=logging.INFO,
|
8 |
-
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
9 |
-
)
|
10 |
-
|
11 |
-
def test_model(force_default=False):
|
12 |
-
"""
|
13 |
-
Testa o ModelHandler com uma consulta de exemplo.
|
14 |
-
|
15 |
-
Args:
|
16 |
-
force_default: Se deve forçar o uso da resposta padrão
|
17 |
-
"""
|
18 |
-
logging.info("Initializing model handler...")
|
19 |
-
model_handler = ModelHandler()
|
20 |
-
|
21 |
-
# Definir a consulta de teste
|
22 |
-
test_query = "O que é autismo?"
|
23 |
-
logging.info(f"Testing with query: {test_query}")
|
24 |
-
|
25 |
-
# Definir se deve forçar a resposta padrão
|
26 |
-
model_handler.force_default_response = force_default
|
27 |
-
|
28 |
-
# Gerar a resposta
|
29 |
-
answer = model_handler.generate_answer(test_query)
|
30 |
-
|
31 |
-
# Imprimir a resposta
|
32 |
-
print("\n==================================================\n")
|
33 |
-
if force_default:
|
34 |
-
print("RESPOSTA PADRÃO FORÇADA:")
|
35 |
-
else:
|
36 |
-
print("RESPOSTA DO MODELO:")
|
37 |
-
print(answer)
|
38 |
-
print("\n==================================================\n")
|
39 |
-
|
40 |
-
if __name__ == "__main__":
|
41 |
-
# Testar com a resposta do modelo
|
42 |
-
test_model(force_default=False)
|
43 |
-
|
44 |
-
# Testar com a resposta padrão forçada
|
45 |
-
logging.info("Testing with forced default response...")
|
46 |
-
test_model(force_default=True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|