Refactor logging in Agent class to improve verbosity control

- Changed the log level for total input tokens from INFO to DEBUG to reduce verbosity during normal operation.
- Removed an INFO log statement for telemetry logging to streamline logging output and focus on error handling.
This commit is contained in:
Magnus Müller
2025-05-22 12:02:37 +02:00
parent 2d71f4cea3
commit e39da12f8d

View File

@@ -992,7 +992,6 @@ class Agent(Generic[Context]):
if not self._force_exit_telemetry_logged: # MODIFIED: Check the flag
try:
self._log_agent_event(max_steps=max_steps, agent_run_error=agent_run_error)
logger.info('Agent run telemetry logged.')
except Exception as log_e: # Catch potential errors during logging itself
logger.error(f'Failed to log telemetry event: {log_e}', exc_info=True)
else:
@@ -1150,7 +1149,7 @@ class Agent(Generic[Context]):
logger.info('❌ Unfinished')
total_tokens = self.state.history.total_input_tokens()
logger.info(f'📝 Total input tokens used (approximate): {total_tokens}')
logger.debug(f'📝 Total input tokens used (approximate): {total_tokens}')
if self.register_done_callback:
if inspect.iscoroutinefunction(self.register_done_callback):