prediction-market-agent-tooling 0.64.5__tar.gz → 0.64.6__tar.gz

{prediction_market_agent_tooling-0.64.5 → prediction_market_agent_tooling-0.64.6}/PKG-INFO

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

{prediction_market_agent_tooling-0.64.5 → prediction_market_agent_tooling-0.64.6}/prediction_market_agent_tooling/tools/caches/db_cache.py

@@ -28,6 +28,8 @@ from prediction_market_agent_tooling.tools.datetime_utc import DatetimeUTC
 from prediction_market_agent_tooling.tools.db.db_manager import DBManager
 from prediction_market_agent_tooling.tools.utils import utcnow
 
+DB_CACHE_LOG_PREFIX = "[db-cache]"
+
 FunctionT = TypeVar("FunctionT", bound=Callable[..., Any])
 
 
@@ -173,7 +175,7 @@ def db_cache(
         if cached_result:
             logger.info(
                 # Keep the special [case-hit] identifier so we can easily track it in GCP.
-                f"[cache-hit] Cache hit for {full_function_name} with args {args_dict} and output {cached_result.result}"
+                f"{DB_CACHE_LOG_PREFIX} [cache-hit] Cache hit for {full_function_name} with args {args_dict} and output {cached_result.result}"
             )
             if is_pydantic_model:
                 # If the output contains any Pydantic models, we need to initialise them.
@@ -184,7 +186,7 @@ def db_cache(
                 except ValueError as e:
                     # In case of backward-incompatible pydantic model, just treat it as cache miss, to not error out.
                     logger.warning(
-                        f"Can not validate {cached_result=} into {return_type=} because {e=}, treating as cache miss."
+                        f"{DB_CACHE_LOG_PREFIX} [cache-miss] Can not validate {cached_result=} into {return_type=} because {e=}, treating as cache miss."
                     )
                     cached_result = None
             else:
@@ -194,7 +196,7 @@ def db_cache(
         computed_result = func(*args, **kwargs)
         # Keep the special [case-miss] identifier so we can easily track it in GCP.
         logger.info(
-            f"[cache-miss] Cache miss for {full_function_name} with args {args_dict}, computed the output {computed_result}"
+            f"{DB_CACHE_LOG_PREFIX} [cache-miss] Cache miss for {full_function_name} with args {args_dict}, computed the output {computed_result}"
         )
 
         # If postgres access was specified, save it.
@@ -212,16 +214,18 @@ def db_cache(
                 with DBManager(
                     api_keys.sqlalchemy_db_url.get_secret_value()
                 ).get_session() as session:
-                    logger.info(f"Saving {cache_entry} into database.")
+                    logger.info(
+                        f"{DB_CACHE_LOG_PREFIX} [cache-info] Saving {cache_entry} into database."
+                    )
                     session.add(cache_entry)
                     session.commit()
             except (DataError, psycopg2.errors.UntranslatableCharacter) as e:
                 (logger.error if log_error_on_unsavable_data else logger.warning)(
-                    f"Failed to save {cache_entry} into database, ignoring, because: {e}"
+                    f"{DB_CACHE_LOG_PREFIX} [cache-error] Failed to save {cache_entry} into database, ignoring, because: {e}"
                 )
             except Exception:
                 logger.exception(
-                    f"Failed to save {cache_entry} into database, ignoring."
+                    f"{DB_CACHE_LOG_PREFIX} [cache-error] Failed to save {cache_entry} into database, ignoring."
                 )
 
         return computed_result

{prediction_market_agent_tooling-0.64.5 → prediction_market_agent_tooling-0.64.6}/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.5 → prediction_market_agent_tooling-0.64.6}/pyproject.toml

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