prediction-market-agent-tooling 0.50.0__tar.gz → 0.50.1__tar.gz

{prediction_market_agent_tooling-0.50.0 → prediction_market_agent_tooling-0.50.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.50.0
+Version: 0.50.1
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12

{prediction_market_agent_tooling-0.50.0 → prediction_market_agent_tooling-0.50.1}/prediction_market_agent_tooling/config.py

@@ -45,6 +45,7 @@ class APIKeys(BaseSettings):
     LANGFUSE_HOST: t.Optional[str] = None
     LANGFUSE_DEPLOYMENT_VERSION: t.Optional[str] = None
 
+    ENABLE_IPFS_UPLOAD: bool = False
     PINATA_API_KEY: t.Optional[SecretStr] = None
     PINATA_API_SECRET: t.Optional[SecretStr] = None
 
@@ -151,6 +152,12 @@ class APIKeys(BaseSettings):
             and self.LANGFUSE_HOST is not None
         )
 
+    @property
+    def enable_ipfs_upload(self) -> bool:
+        return check_not_none(
+            self.ENABLE_IPFS_UPLOAD, "ENABLE_IPFS_UPLOAD missing in the environment."
+        )
+
     @property
     def pinata_api_key(self) -> SecretStr:
         return check_not_none(

{prediction_market_agent_tooling-0.50.0 → prediction_market_agent_tooling-0.50.1}/prediction_market_agent_tooling/deploy/agent.py

@@ -11,6 +11,7 @@ from functools import cached_property
 from pydantic import BaseModel, BeforeValidator, computed_field
 from typing_extensions import Annotated
 from web3 import Web3
+from web3.constants import HASH_ZERO
 
 from prediction_market_agent_tooling.config import APIKeys
 from prediction_market_agent_tooling.deploy.betting_strategy import (
@@ -302,7 +303,6 @@ class DeployableTraderAgent(DeployableAgent):
     ) -> None:
         super().__init__(enable_langfuse=enable_langfuse)
         self.place_bet = place_bet
-        self.ipfs_handler = IPFSHandler(APIKeys())
 
     def get_betting_strategy(self, market: AgentMarket) -> BettingStrategy:
         user_id = market.get_user_id(api_keys=APIKeys())
@@ -522,16 +522,21 @@ class DeployableTraderAgent(DeployableAgent):
             if processed_market.answer.reasoning
             else ""
         )
-        ipfs_hash = self.ipfs_handler.store_agent_result(
-            IPFSAgentResult(reasoning=reasoning)
-        )
+
+        ipfs_hash_decoded = HexBytes(HASH_ZERO)
+        if keys.enable_ipfs_upload:
+            logger.info("Storing prediction on IPFS.")
+            ipfs_hash = IPFSHandler(keys).store_agent_result(
+                IPFSAgentResult(reasoning=reasoning)
+            )
+            ipfs_hash_decoded = ipfscidv0_to_byte32(ipfs_hash)
 
         tx_hashes = [
             HexBytes(HexStr(i.id)) for i in processed_market.trades if i.id is not None
         ]
         prediction = ContractPrediction(
             publisher=keys.public_key,
-            ipfs_hash=ipfscidv0_to_byte32(ipfs_hash),
+            ipfs_hash=ipfs_hash_decoded,
             tx_hashes=tx_hashes,
             estimated_probability_bps=int(processed_market.answer.p_yes * 10000),
         )

{prediction_market_agent_tooling-0.50.0 → prediction_market_agent_tooling-0.50.1}/pyproject.toml

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