Spaces:
Sleeping
Sleeping
| import logging | |
| import datetime | |
| import contextvars | |
| current_trace_run = contextvars.ContextVar('current_trace_run', default=None) | |
| class LangSmithHandler(logging.Handler): | |
| def emit(self, record): | |
| run = current_trace_run.get() | |
| if run: | |
| try: | |
| run.add_event({ | |
| "name": "log", | |
| "time": datetime.datetime.now(datetime.timezone.utc), | |
| "kwargs": { | |
| "level": record.levelname, | |
| "message": self.format(record), | |
| "logger_name": record.name, | |
| "file": record.filename, | |
| "line": record.lineno | |
| } | |
| }) | |
| except Exception: | |
| pass | |
| def setup_logger(name: str) -> logging.Logger: | |
| logger = logging.getLogger(name) | |
| if not logger.handlers: | |
| logger.setLevel(logging.INFO) | |
| formatter = logging.Formatter('%(asctime)s - %(levelname)s - [%(name)s] %(message)s') | |
| ch = logging.StreamHandler() | |
| ch.setFormatter(formatter) | |
| logger.addHandler(ch) | |
| ls_handler = LangSmithHandler() | |
| ls_handler.setFormatter(formatter) | |
| logger.addHandler(ls_handler) | |
| return logger | |