prediction-market-agent-tooling 0.64.5__py3-none-any.whl → 0.64.6__py3-none-any.whl

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/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.dist-info → prediction_market_agent_tooling-0.64.6.dist-info}/METADATA

@@ -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.dist-info → prediction_market_agent_tooling-0.64.6.dist-info}/RECORD

@@ -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=e7CLFPTlxYFx0pM3x5sWyn_g9pIigPSvrP0mMmiOhUg,11859
+prediction_market_agent_tooling/tools/caches/db_cache.py,sha256=rZIGhgijquwwPtp_qncSAPR1SDF2XxIVZL1ir0fgzWw,12127
 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
@@ -98,7 +98,7 @@ prediction_market_agent_tooling/tools/db/db_manager.py,sha256=GtzHH1NLl8HwqC8Z7s
 prediction_market_agent_tooling/tools/google_utils.py,sha256=D-6FB2HRtmxaKZJ_Za-qj6VZCp5XU18rF4wLMMrqEUg,2557
 prediction_market_agent_tooling/tools/hexbytes_custom.py,sha256=ytwr4N8t-9UF1F4wXsln3IV1HBKgaYirsErc3NsKL-M,2119
 prediction_market_agent_tooling/tools/httpx_cached_client.py,sha256=RxD-hwtZCMctnMwfzy8t51W9Z9gxFGtDYxBIMChazpc,406
-prediction_market_agent_tooling/tools/image_gen/image_gen.py,sha256=HzRwBx62hOXBOmrtpkXaP9Qq1Ku03uUGdREocyjLQ_k,1266
+prediction_market_agent_tooling/tools/image_gen/image_gen.py,sha256=i7ole0NMprELk-I3qJQLqFNZjiqM1w6s11IqM4Ctaqw,1349
 prediction_market_agent_tooling/tools/image_gen/market_thumbnail_gen.py,sha256=jgOow4zw6g8dIJrog6Tp-3NQs4wjUJ3VgVKyMQv-0QM,1286
 prediction_market_agent_tooling/tools/ipfs/ipfs_handler.py,sha256=CTTMfTvs_8PH4kAtlQby2aeEKwgpmxtuGbd4oYIdJ2A,1201
 prediction_market_agent_tooling/tools/is_invalid.py,sha256=TAHQXiiusAU45xJ11ZyEP7PnEfcjfzVG7qHRbsHiAd0,5335
@@ -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.5.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.64.5.dist-info/METADATA,sha256=EPfSPw78Gf7zr3qLdKFha7YltbMwA-EvW4PkI8lykMU,8741
-prediction_market_agent_tooling-0.64.5.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-prediction_market_agent_tooling-0.64.5.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.64.5.dist-info/RECORD,,
+prediction_market_agent_tooling-0.64.6.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.64.6.dist-info/METADATA,sha256=2r_sU6cc-W140QZ1PCjAsOpNpua3ocBXIFDwXED2zq4,8741
+prediction_market_agent_tooling-0.64.6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+prediction_market_agent_tooling-0.64.6.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.64.6.dist-info/RECORD,,