prediction-market-agent-tooling 0.64.6.dev622__tar.gz → 0.64.7__tar.gz

{prediction_market_agent_tooling-0.64.6.dev622 → prediction_market_agent_tooling-0.64.7}/PKG-INFO

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

{prediction_market_agent_tooling-0.64.6.dev622 → prediction_market_agent_tooling-0.64.7}/prediction_market_agent_tooling/loggers.py

@@ -8,6 +8,7 @@ import typer.main
 from loguru import logger
 from pydantic_settings import BaseSettings, SettingsConfigDict
 from pythonjsonlogger import jsonlogger
+from tenacity import RetryError
 
 
 class LogFormat(str, Enum):
@@ -68,7 +69,15 @@ def _handle_exception(
         sys.__excepthook__(exc_type, exc_value, exc_traceback)
         return
 
-    logger.error("Uncaught exception", exc_info=(exc_type, exc_value, exc_traceback))
+    info: BaseException | None = exc_value
+    if isinstance(exc_value, RetryError):
+        # In case of RetryError from tenacity, add last attempt's exp to the log, otherwise we won't see the exception's message in the logs.
+        info = exc_value.last_attempt.exception()
+
+    logger.error(
+        f"Uncaught exception: {info}",
+        exc_info=(exc_type, exc_value, exc_traceback),
+    )
 
 
 def patch_logger(force_patch: bool = False) -> None:

{prediction_market_agent_tooling-0.64.6.dev622 → prediction_market_agent_tooling-0.64.7}/prediction_market_agent_tooling/tools/image_gen/image_gen.py

@@ -23,25 +23,27 @@ def generate_image(
         raise ImportError(
             "openai not installed, please install extras `openai` to use this function."
         )
-    response = (
-        OpenAI(
-            api_key=APIKeys().openai_api_key.get_secret_value(),
-        )
-        .images.generate(
-            model=model,
-            prompt=prompt,
-            size=size,
-            quality=quality,
-            response_format="b64_json",
-            n=1,
-        )
-        .data[0]
+    response = OpenAI(
+        api_key=APIKeys().openai_api_key.get_secret_value(),
+    ).images.generate(
+        model=model,
+        prompt=prompt,
+        size=size,
+        quality=quality,
+        response_format="b64_json",
+        n=1,
     )
+
+    if response.data is None or len(response.data) == 0:
+        raise ValueError("No image data returned from the API.")
+
+    image_data = response.data[0]
+
     image = Image.open(
         io.BytesIO(
             base64.b64decode(
                 check_not_none(
-                    response.b64_json, "Can't be none if response_format is b64_json."
+                    image_data.b64_json, "Can't be none if response_format is b64_json."
                 )
             )
         )

{prediction_market_agent_tooling-0.64.6.dev622 → prediction_market_agent_tooling-0.64.7}/pyproject.toml

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