iQuentin commited on
Commit
05cf778
·
verified ·
1 Parent(s): 1d9c94e

Create test_answer_manager.py

Browse files
Files changed (1) hide show
  1. test_answer_manager.py +53 -0
test_answer_manager.py ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from answer_data_manager import AnswerDataManager
2
+
3
+ def run_test_answer_manager()
4
+ # Initialize the manager
5
+ manager = AnswerDataManager("already_answered.json")
6
+ # Load existing data
7
+ manager.load_data()
8
+
9
+
10
+ # question 1
11
+ question1 = {
12
+ "task_id": "task_001",
13
+ "question": "What is it?",
14
+ "submitted_answer": "apple",
15
+ }
16
+
17
+ existing_answer = manager.get_answer_by_task_id(question1["task_id"])
18
+
19
+ if existing_answer:
20
+ # Do not invoke agent
21
+ print(f"retrouvé depuis l'archive : {existing_answer['submitted_answer']}")
22
+ else:
23
+ print("Pas de réponse, j'appelle l'agent !")
24
+ success = manager.add_answer(
25
+ task_id="hFg5v8",
26
+ question="how many",
27
+ submitted_answer="super nouvelle réponse"
28
+ )
29
+ if not success:
30
+ print("Erreur lors de la sauvegarde add_answer")
31
+
32
+ # manager.save_data()
33
+
34
+
35
+ # # Load existing data
36
+ # manager.load_data()
37
+
38
+ # # Add some sample data
39
+ # manager.add_answer("task_002", "What is it?", "apple")
40
+ # manager.add_answer("task_003", "question in words", "150 miles")
41
+
42
+ # # Retrieve data
43
+ # print("All answers:", manager.get_all_answers())
44
+
45
+ # # Get specific answer
46
+ # answer = manager.get_answer_by_task_id("task_002")
47
+ # print("Task 002 answer:", answer)
48
+
49
+ # # Update an answer
50
+ # manager.update_answer("task_002", submitted_answer="orange")
51
+
52
+ # # Show updated data
53
+ # print("Updated data:", manager.get_all_answers())