Spaces:
Runtime error
Runtime error
File size: 4,015 Bytes
6a42990 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 | import logging
from pathlib import Path
import configparser
################################################################################
# Config and startup utilities
################################################################################
_config = None
def read_config_file(use_cache=True, verbose=True) -> configparser.ConfigParser:
global _config
if use_cache and _config is not None:
# if we have a cached config and accept that, return it
return _config
else:
config = configparser.ConfigParser()
# Read the default values in the module directory.
config_file_path = Path(__file__).parent.absolute() / '../config.ini'
print(f"Looking for default config on: {config_file_path}") if verbose else None
if config_file_path.exists():
config.read(config_file_path)
_config = config
else:
raise ValueError(f"Failed to find default config on: {config_file_path}")
# Now, let's override any specific default value, if there's a custom .ini config.
# Try the directory of the current main program
config_file_path = Path.cwd() / "config.ini"
if config_file_path.exists():
print(f"Found custom config on: {config_file_path}") if verbose else None
config.read(config_file_path) # this only overrides the values that are present in the custom config
_config = config
return config
else:
if verbose:
print(f"Failed to find custom config on: {config_file_path}") if verbose else None
print("Will use only default values. IF THINGS FAIL, TRY CUSTOMIZING MODEL, API TYPE, etc.") if verbose else None
return config
def pretty_print_config(config):
print()
print("=================================")
print("Current TinyTroupe configuration ")
print("=================================")
for section in config.sections():
print(f"[{section}]")
for key, value in config.items(section):
print(f"{key} = {value}")
print()
def pretty_print_datetime():
from datetime import datetime
from datetime import timezone
now = datetime.now()
now_utc = now.astimezone(timezone.utc)
print(f"Current date and time (local): {now.strftime('%Y-%m-%d %H:%M:%S')}")
print(f"Current date and time (UTC): {now_utc.strftime('%Y-%m-%d %H:%M:%S')}")
def pretty_print_tinytroupe_version():
try:
import importlib.metadata
version = importlib.metadata.version("tinytroupe")
except Exception:
version = "unknown"
print(f"TinyTroupe version: {version}")
def start_logger(config: configparser.ConfigParser):
# create logger
logger = logging.getLogger("tinytroupe")
log_level = config['Logging'].get('LOGLEVEL', 'INFO').upper()
logger.setLevel(level=log_level)
# Clear any existing handlers to prevent duplicates
# This is especially important in Jupyter notebooks where modules get reloaded
for handler in logger.handlers[:]:
logger.removeHandler(handler)
# Prevent propagation to avoid duplicate messages from parent loggers
logger.propagate = False
# create console handler and set level to debug
ch = logging.StreamHandler()
ch.setLevel(log_level)
# create formatter
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
# add formatter to ch
ch.setFormatter(formatter)
# add ch to logger
logger.addHandler(ch)
def set_loglevel(log_level):
"""
Sets the log level for the TinyTroupe logger.
Args:
log_level (str): The log level to set (e.g., 'DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL').
"""
logger = logging.getLogger("tinytroupe")
logger.setLevel(level=log_level)
# Also update all handlers to the new log level
for handler in logger.handlers:
handler.setLevel(log_level) |