prediction-market-agent-tooling 0.22.0__tar.gz → 0.23.0__tar.gz

{prediction_market_agent_tooling-0.22.0 → prediction_market_agent_tooling-0.23.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.22.0
+Version: 0.23.0
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12

prediction_market_agent_tooling-0.23.0/prediction_market_agent_tooling/loggers.py

@@ -0,0 +1,75 @@
+import logging
+import sys
+import warnings
+from enum import Enum
+
+from loguru import logger
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+
+class LogFormat(str, Enum):
+    DEFAULT = "default"
+    GCP = "gcp"
+
+
+class LogConfig(BaseSettings):
+    model_config = SettingsConfigDict(
+        env_file=".env", env_file_encoding="utf-8", extra="ignore"
+    )
+
+    LOG_FORMAT: LogFormat = LogFormat.DEFAULT
+
+
+GCP_LOG_LOGURU_FORMAT = (
+    "{level:<.1}{time:MMDD HH:mm:ss} {process} {name}:{line}] {message}"
+)
+GCP_LOG_LOGGING_FORMAT, GCP_LOG_FORMAT_LOGGING_DATEFMT = (
+    "%(levelname).1s%(asctime)s %(process)d %(name)s:%(lineno)d] %(message)s"
+), "%m%d %H:%M:%S"
+
+
+def patch_logger() -> None:
+    config = LogConfig()
+
+    if config.LOG_FORMAT == LogFormat.GCP:
+        format_loguru = GCP_LOG_LOGURU_FORMAT
+        format_logging = GCP_LOG_LOGGING_FORMAT
+        datefmt_logging = GCP_LOG_FORMAT_LOGGING_DATEFMT
+
+    elif config.LOG_FORMAT == LogFormat.DEFAULT:
+        format_loguru, format_logging, datefmt_logging = None, None, None
+
+    else:
+        raise ValueError(f"Unknown log format: {config.LOG_FORMAT}")
+
+    # Change built-in logging.
+    if format_logging is not None:
+        logging.basicConfig(
+            level=logging.DEBUG, format=format_logging, datefmt=datefmt_logging
+        )
+
+    # Change loguru.
+    if format_loguru is not None:
+        logger.remove()
+        logger.add(
+            sys.stdout,
+            format=format_loguru,
+            level="DEBUG",  # Can be the lowest level, higher ones will use by default this one.
+            colorize=True,
+        )
+
+    # Change warning formatting to a simpler one (no source code in a new line).
+    warnings.formatwarning = simple_warning_format
+    # Use logging module for warnings.
+    logging.captureWarnings(True)
+
+    logger.info(f"Patched logger for {config.LOG_FORMAT.value} format.")
+
+
+def simple_warning_format(message, category, filename, lineno, line=None):  # type: ignore[no-untyped-def] # Not typed in the standard library neither.
+    return f"{category.__name__}: {message}"
+
+
+if not getattr(logger, "_patched", False):
+    patch_logger()
+    logger._patched = True  # type: ignore[attr-defined] # Hacky way to store a flag on the logger object, to not patch it multiple times.

{prediction_market_agent_tooling-0.22.0 → prediction_market_agent_tooling-0.23.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prediction-market-agent-tooling"
-version = "0.22.0"
+version = "0.23.0"
 description = "Tools to benchmark, deploy and monitor prediction market agents."
 authors = ["Gnosis"]
 readme = "README.md"

prediction_market_agent_tooling-0.22.0/prediction_market_agent_tooling/loggers.py

@@ -1,54 +0,0 @@
-import sys
-import typing as t
-from enum import Enum
-
-from loguru import logger
-from pydantic_settings import BaseSettings, SettingsConfigDict
-
-if t.TYPE_CHECKING:
-    from loguru import Logger
-
-
-class LogFormat(str, Enum):
-    DEFAULT = "default"
-    GCP = "gcp"
-
-
-class LogConfig(BaseSettings):
-    model_config = SettingsConfigDict(
-        env_file=".env", env_file_encoding="utf-8", extra="ignore"
-    )
-
-    LOG_FORMAT: LogFormat = LogFormat.DEFAULT
-
-
-GCP_LOG_FORMAT = "{level:<.1}{time:MMDD HH:mm:ss.SSSSSS} {process} {name}:{line}] {message} | {extra}"
-
-
-def patch_logger(logger: "Logger") -> None:
-    config = LogConfig()
-
-    if config.LOG_FORMAT == LogFormat.GCP:
-        format_ = GCP_LOG_FORMAT
-
-    elif config.LOG_FORMAT == LogFormat.DEFAULT:
-        format_ = None
-
-    else:
-        raise ValueError(f"Unknown log format: {config.LOG_FORMAT}")
-
-    if format_ is not None:
-        logger.remove()
-        logger.add(
-            sys.stdout,
-            format=format_,
-            level="DEBUG",  # Can be the lowest level, higher ones will use by default this one.
-            colorize=True,
-        )
-
-    logger.info(f"Patched logger for {config.LOG_FORMAT.value} format.")
-
-
-if not getattr(logger, "_patched", False):
-    patch_logger(logger)
-    logger._patched = True  # type: ignore[attr-defined] # Hacky way to store a flag on the logger object, to not patch it multiple times.