DrishtiSharma commited on
Commit
dd8bef5
·
verified ·
1 Parent(s): 9b0a917

Create logger.py

Browse files
Files changed (1) hide show
  1. core/logger.py +25 -0
core/logger.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import logging
2
+ from datetime import datetime
3
+ from typing import Any, Dict
4
+
5
+ class Logger:
6
+ def __init__(self):
7
+ logging.basicConfig(
8
+ level=logging.INFO,
9
+ format='%(asctime)s - %(levelname)s - %(message)s',
10
+ handlers=[
11
+ logging.FileHandler('medical_ai_agents.log'),
12
+ logging.StreamHandler()
13
+ ]
14
+ )
15
+ self.logger = logging.getLogger(__name__)
16
+
17
+ def log_input(self, task_type: str, input_data: Any):
18
+ self.logger.info(f"Task: {task_type} - Input received at {datetime.now()}")
19
+
20
+ def log_output(self, task_type: str, result: Dict[str, Any], validation: Dict[str, bool]):
21
+ self.logger.info(f"Task: {task_type} - Output generated at {datetime.now()}")
22
+ self.logger.info(f"Validation result: {validation['is_valid']}")
23
+
24
+ def log_error(self, task_type: str, error_message: str):
25
+ self.logger.error(f"Task: {task_type} - Error: {error_message}")