wakeupmh commited on
Commit
b0e0ada
·
1 Parent(s): 2269229

fix: rever

Browse files
Files changed (1) hide show
  1. 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)