prediction-market-agent-tooling 0.64.2.dev603__py3-none-any.whl → 0.64.3.dev611__py3-none-any.whl

prediction_market_agent_tooling/loggers.py

@@ -4,6 +4,7 @@ import sys
 import typing as t
 import warnings
 from enum import Enum
+from pythonjsonlogger import jsonlogger
 
 from loguru import logger
 from pydantic_settings import BaseSettings, SettingsConfigDict
@@ -31,17 +32,35 @@ class LogConfig(BaseSettings):
     LOG_LEVEL: LogLevel = LogLevel.DEBUG
 
 
-class NoNewLineStreamHandler(logging.StreamHandler):  # type: ignore # StreamHandler is not typed in the standard library.
-    def format(self, record: logging.LogRecord) -> str:
-        return super().format(record).replace("\n", " ")
-
+class _CustomJsonFormatter(jsonlogger.JsonFormatter):
+    def add_fields(
+        self,
+        log_record: dict[str, t.Any],
+        record: logging.LogRecord,
+        message_dict: dict[str, t.Any],
+    ) -> None:
+        super().add_fields(log_record, record, message_dict)
+        # Include "level" and "severity" with the same value as "levelname" to be friendly with log aggregators.
+        if log_record.get("levelname"):
+            log_record["level"] = log_record["levelname"]
+            log_record["severity"] = log_record["levelname"]
+
+    @staticmethod
+    def get_handler() -> logging.StreamHandler:  # type: ignore # Seems correct, but mypy doesn't like it.
+        logHandler = logging.StreamHandler()
+        formatter = _CustomJsonFormatter("%(asctime)s %(levelname)s %(message)s")
+        logHandler.setFormatter(formatter)
+        return logHandler
+
+
+def _handle_exception(
+    exc_type: type[BaseException], exc_value: BaseException, exc_traceback: t.Any
+) -> None:
+    if issubclass(exc_type, KeyboardInterrupt):
+        sys.__excepthook__(exc_type, exc_value, exc_traceback)
+        return
 
-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"
+    logger.error("Uncaught exception", exc_info=(exc_type, exc_value, exc_traceback))
 
 
 def patch_logger() -> None:
@@ -57,49 +76,43 @@ 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
-        print_logging = print_using_loguru_info
-        handlers: list[logging.Handler] | None = [NoNewLineStreamHandler()]
+        handler = _CustomJsonFormatter.get_handler()
+        print_logging = print_using_logger_info
+        sys.excepthook = _handle_exception
 
     elif config.LOG_FORMAT == LogFormat.DEFAULT:
-        format_loguru, format_logging, datefmt_logging = None, None, None
+        handler = None
         print_logging = None
-        handlers = None
 
     else:
         raise ValueError(f"Unknown log format: {config.LOG_FORMAT}")
 
     # Change built-in logging.
-    if format_logging is not None:
+    if handler is not None:
         logging.basicConfig(
             level=config.LOG_LEVEL.value,
-            format=format_logging,
-            datefmt=datefmt_logging,
-            handlers=handlers,
+            handlers=[handler],
         )
-
         # Configure all existing loggers
         for logger_name in logging.root.manager.loggerDict:
             existing_logger = logging.getLogger(logger_name)
             existing_logger.setLevel(config.LOG_LEVEL.value)
-            if handlers is not None:
-                existing_logger.handlers = handlers
+            # Remove existing handlers
+            if existing_logger.hasHandlers():
+                existing_logger.handlers.clear()
+            # And add ours only
+            existing_logger.addHandler(handler)
             existing_logger.propagate = False
 
     # Change loguru.
-    if format_loguru is not None:
+    if handler is not None:
         logger.remove()
         logger.add(
-            sys.stdout,
-            format=format_loguru,
+            handler,
             level=config.LOG_LEVEL.value,
-            colorize=True,
+            colorize=False,
         )
 
-    # 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)
 
@@ -110,23 +123,13 @@ def patch_logger() -> None:
     logger.info(f"Patched logger for {config.LOG_FORMAT.value} format.")
 
 
-def print_using_loguru_info(
+def print_using_logger_info(
     *values: object,
     sep: str = " ",
     end: str = "\n",
     **kwargs: t.Any,
 ) -> None:
-    message = sep.join(map(str, values)) + end
-    message = message.strip().replace(
-        "\n", "\\n"
-    )  # Escape new lines, because otherwise logs will be broken.
-    logger.info(message)
-
-
-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}".strip().replace(
-        "\n", "\\n"
-    )  # Escape new lines, because otherwise logs will be broken.
+    logger.info(sep.join(map(str, values)) + end)
 
 
 patch_logger()

prediction_market_agent_tooling/tools/caches/db_cache.py

@@ -3,6 +3,7 @@ import inspect
 import json
 from datetime import timedelta
 from functools import wraps
+from types import UnionType
 from typing import (
     Any,
     Callable,
@@ -276,6 +277,17 @@ def convert_cached_output_to_pydantic(return_type: Any, data: Any) -> Any:
                 ): convert_cached_output_to_pydantic(value_type, v)
                 for k, v in data.items()
             }
+        # If the origin is a union and one of the unions is basemodel, convert it to it.
+        elif (
+            origin is UnionType
+            and (
+                base_model_from_args := next(
+                    (x for x in args if issubclass(x, BaseModel)), None
+                )
+            )
+            is not None
+        ):
+            return base_model_from_args.model_validate(data)
         else:
             # If the origin is not a dictionary, return the data as is
             return data

{prediction_market_agent_tooling-0.64.2.dev603.dist-info → prediction_market_agent_tooling-0.64.3.dev611.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.64.2.dev603
+Version: 0.64.3.dev611
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.13
@@ -46,6 +46,7 @@ Requires-Dist: pydantic-settings (>=2.4.0,<3.0.0)
 Requires-Dist: pymongo (>=4.8.0,<5.0.0)
 Requires-Dist: pytest-postgresql (>=6.1.1,<7.0.0)
 Requires-Dist: python-dateutil (>=2.9.0.post0,<3.0.0)
+Requires-Dist: python-json-logger (>=3.3.0,<4.0.0)
 Requires-Dist: safe-cli (>=1.0.0,<2.0.0)
 Requires-Dist: safe-eth-py (>=6.0.0b41,<7.0.0)
 Requires-Dist: scikit-learn (>=1.3.1,<2.0.0)

{prediction_market_agent_tooling-0.64.2.dev603.dist-info → prediction_market_agent_tooling-0.64.3.dev611.dist-info}/RECORD

@@ -35,7 +35,7 @@ prediction_market_agent_tooling/gtypes.py,sha256=bUIZfZIGvIi3aiZNu5rVE9kRevw8sfM
 prediction_market_agent_tooling/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/jobs/jobs_models.py,sha256=8vYafsK1cqMWQtjBoq9rruroF84xAVD00vBTMWH6QMg,2166
 prediction_market_agent_tooling/jobs/omen/omen_jobs.py,sha256=Pf6QxPXGyie-2l_wZUjaGPTjZTlpv50_JhP40mULBaU,5048
-prediction_market_agent_tooling/loggers.py,sha256=MvCkQSJL2_0yErNatqr81sJlc4aOgPzDp9VNrIhKUcc,4140
+prediction_market_agent_tooling/loggers.py,sha256=ae6v2xUIARZ1CYO_z2W5tvtonEtqxozdFQFz5D3oDrE,4104
 prediction_market_agent_tooling/markets/agent_market.py,sha256=1NomilM0GCXcRq_1N_cr2AbSK5ONTowFeRbrhc7V5zE,14929
 prediction_market_agent_tooling/markets/base_subgraph_handler.py,sha256=7RaYO_4qAmQ6ZGM8oPK2-CkiJfKmV9MxM-rJlduaecU,1971
 prediction_market_agent_tooling/markets/blockchain_utils.py,sha256=qm21scopQ6dfewkoqQF6lWLDGg2BblsKUdC9aG93Hmc,2249
@@ -86,7 +86,7 @@ prediction_market_agent_tooling/tools/betting_strategies/market_moving.py,sha256
 prediction_market_agent_tooling/tools/betting_strategies/minimum_bet_to_win.py,sha256=-FUSuQQgjcWSSnoFxnlAyTeilY6raJABJVM2QKkFqAY,438
 prediction_market_agent_tooling/tools/betting_strategies/stretch_bet_between.py,sha256=THMXwFlskvzbjnX_OiYtDSzI8XVFyULWfP2525_9UGc,429
 prediction_market_agent_tooling/tools/betting_strategies/utils.py,sha256=68zFWUj43GUaSpOPowVrbI-t6qbCE29RsVHNzCVuJ9U,175
-prediction_market_agent_tooling/tools/caches/db_cache.py,sha256=dB8LNs2JvVRaFCeAKRmIQRwiirsMgtL31he8051wM-g,11431
+prediction_market_agent_tooling/tools/caches/db_cache.py,sha256=e7CLFPTlxYFx0pM3x5sWyn_g9pIigPSvrP0mMmiOhUg,11859
 prediction_market_agent_tooling/tools/caches/inmemory_cache.py,sha256=ZW5iI5rmjqeAebu5T7ftRnlkxiL02IC-MxCfDB80x7w,1506
 prediction_market_agent_tooling/tools/caches/serializers.py,sha256=vFDx4fsPxclXp2q0sv27j4al_M_Tj9aR2JJP-xNHQXA,2151
 prediction_market_agent_tooling/tools/contract.py,sha256=Yex8MVYvdBTMZLESLqKpwEyT8EGAfkJRri5kCaPqrBM,21235
@@ -124,8 +124,8 @@ prediction_market_agent_tooling/tools/tokens/usd.py,sha256=yuW8iPPtcpP4eLH2nORMD
 prediction_market_agent_tooling/tools/transaction_cache.py,sha256=K5YKNL2_tR10Iw2TD9fuP-CTGpBbZtNdgbd0B_R7pjg,1814
 prediction_market_agent_tooling/tools/utils.py,sha256=AC2a68jwASMWuQi-w8twl8b_M52YwrEJ81abmuEaqMY,6661
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=zRq-eeBGWt8uUGN9G_WfjmJ0eVvO8aWE9S0Pz_Y6AOA,12342
-prediction_market_agent_tooling-0.64.2.dev603.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.64.2.dev603.dist-info/METADATA,sha256=SPD7ax9Sdd7xI70-SypOdJzFqGvgNaLfvs7HT7biO6w,8697
-prediction_market_agent_tooling-0.64.2.dev603.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-prediction_market_agent_tooling-0.64.2.dev603.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.64.2.dev603.dist-info/RECORD,,
+prediction_market_agent_tooling-0.64.3.dev611.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.64.3.dev611.dist-info/METADATA,sha256=CNc9oHj3mAyUZQN9ElRLGAzvRhHaxkJc9gs-nr3VGbM,8748
+prediction_market_agent_tooling-0.64.3.dev611.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+prediction_market_agent_tooling-0.64.3.dev611.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.64.3.dev611.dist-info/RECORD,,

{prediction_market_agent_tooling-0.64.2.dev603.dist-info → prediction_market_agent_tooling-0.64.3.dev611.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.2
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any