Yadav122 commited on
Commit
4bc9267
·
1 Parent(s): f86f7fa
Files changed (1) hide show
  1. app/core/logger.py +11 -8
app/core/logger.py CHANGED
@@ -26,9 +26,13 @@ class Logger:
26
  def __init__(self, json_logs: bool = False, log_level: str = "INFO"):
27
  self.json_logs = json_logs
28
  self.log_level = log_level.upper()
 
29
 
30
- self.environment = os.getenv("ENVIRONMENT", "PROD").upper() # Default to PROD
31
- self.log_file_path = os.getenv("LOG_FILE_PATH", self._get_default_log_file_path())
 
 
 
32
 
33
  def _get_default_log_file_path(self) -> str:
34
  """Get the default log file path."""
@@ -132,12 +136,11 @@ class Logger:
132
  stream_handler.setFormatter(formatter)
133
  root_logger.addHandler(stream_handler)
134
 
135
- # Only add file logging if LOG_FILE_PATH is explicitly set and not in Hugging Face environment
136
- log_file_path = os.getenv("LOG_FILE_PATH")
137
- if log_file_path and os.getenv("ENVIRONMENT") != "HUGGINGFACE":
138
  try:
139
  file_handler = TimedRotatingFileHandler(
140
- filename=log_file_path,
141
  when="midnight",
142
  interval=1,
143
  backupCount=7,
@@ -146,10 +149,10 @@ class Logger:
146
  file_handler.setFormatter(formatter)
147
  root_logger.addHandler(file_handler)
148
  except PermissionError:
149
- # If file logging fails, just log to console
150
  pass
151
 
152
- root_logger.setLevel(self.log_level.upper())
153
  return root_logger
154
 
155
  def _configure(self):
 
26
  def __init__(self, json_logs: bool = False, log_level: str = "INFO"):
27
  self.json_logs = json_logs
28
  self.log_level = log_level.upper()
29
+ self.environment = os.getenv("ENVIRONMENT", "PROD").upper()
30
 
31
+ # Skip file logging in Hugging Face environment
32
+ if self.environment == "HUGGINGFACE":
33
+ self.log_file_path = None
34
+ else:
35
+ self.log_file_path = os.getenv("LOG_FILE_PATH", self._get_default_log_file_path())
36
 
37
  def _get_default_log_file_path(self) -> str:
38
  """Get the default log file path."""
 
136
  stream_handler.setFormatter(formatter)
137
  root_logger.addHandler(stream_handler)
138
 
139
+ # Only add file logging if not in Hugging Face environment and log_file_path is set
140
+ if self.environment != "HUGGINGFACE" and self.log_file_path:
 
141
  try:
142
  file_handler = TimedRotatingFileHandler(
143
+ filename=self.log_file_path,
144
  when="midnight",
145
  interval=1,
146
  backupCount=7,
 
149
  file_handler.setFormatter(formatter)
150
  root_logger.addHandler(file_handler)
151
  except PermissionError:
152
+ # If file logging fails, just continue with console logging
153
  pass
154
 
155
+ root_logger.setLevel(self.log_level)
156
  return root_logger
157
 
158
  def _configure(self):