prediction-market-agent-tooling 0.64.3.dev611__tar.gz → 0.64.4__tar.gz

{prediction_market_agent_tooling-0.64.3.dev611 → prediction_market_agent_tooling-0.64.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.64.3.dev611
+Version: 0.64.4
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.13

{prediction_market_agent_tooling-0.64.3.dev611 → prediction_market_agent_tooling-0.64.4}/prediction_market_agent_tooling/loggers.py

@@ -2,12 +2,12 @@ import builtins
 import logging
 import sys
 import typing as t
-import warnings
 from enum import Enum
-from pythonjsonlogger import jsonlogger
 
+import typer.main
 from loguru import logger
 from pydantic_settings import BaseSettings, SettingsConfigDict
+from pythonjsonlogger import jsonlogger
 
 
 class LogFormat(str, Enum):
@@ -63,12 +63,12 @@ def _handle_exception(
     logger.error("Uncaught exception", exc_info=(exc_type, exc_value, exc_traceback))
 
 
-def patch_logger() -> None:
+def patch_logger(force_patch: bool = False) -> None:
     """
     Function to patch loggers according to the deployed environment.
     Patches Loguru's logger, Python's default logger, warnings library and also monkey-patch print function as many libraries just use it.
     """
-    if not getattr(logger, "_patched", False):
+    if force_patch or not getattr(logger, "_patched", False):
         logger._patched = True  # type: ignore[attr-defined] # Hacky way to store a flag on the logger object, to not patch it multiple times.
     else:
         return
@@ -79,6 +79,7 @@ def patch_logger() -> None:
         handler = _CustomJsonFormatter.get_handler()
         print_logging = print_using_logger_info
         sys.excepthook = _handle_exception
+        typer.main.except_hook = _handle_exception  # type: ignore # Monkey patching, it's messy but it works.
 
     elif config.LOG_FORMAT == LogFormat.DEFAULT:
         handler = None

{prediction_market_agent_tooling-0.64.3.dev611 → prediction_market_agent_tooling-0.64.4}/pyproject.toml

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