Create test_answer_manager.py
Browse files- 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())
|