prediction-market-agent-tooling 0.64.6.dev622__tar.gz → 0.64.6.dev623__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.64.6.dev622
+Version: 0.64.6.dev623
 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.6.dev623}/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.6.dev623}/pyproject.toml

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