prediction-market-agent-tooling 0.60.6__tar.gz → 0.61.0__tar.gz

{prediction_market_agent_tooling-0.60.6 → prediction_market_agent_tooling-0.61.0}/PKG-INFO

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

{prediction_market_agent_tooling-0.60.6 → prediction_market_agent_tooling-0.61.0}/prediction_market_agent_tooling/config.py

@@ -3,7 +3,7 @@ import typing as t
 from copy import deepcopy
 
 from eth_account.signers.local import LocalAccount
-from pydantic import BeforeValidator, Field, model_validator
+from pydantic import Field, model_validator
 from pydantic.types import SecretStr
 from pydantic.v1.types import SecretStr as SecretStrV1
 from pydantic_settings import BaseSettings, SettingsConfigDict
@@ -39,10 +39,7 @@ class APIKeys(BaseSettings):
     METACULUS_API_KEY: t.Optional[SecretStr] = None
     METACULUS_USER_ID: t.Optional[int] = None
     BET_FROM_PRIVATE_KEY: t.Optional[PrivateKey] = None
-    SAFE_ADDRESS: t.Annotated[
-        ChecksumAddress | None,
-        BeforeValidator(lambda x: x if x is None else Web3.to_checksum_address(x)),
-    ] = None
+    SAFE_ADDRESS: str | None = None
     OPENAI_API_KEY: t.Optional[SecretStr] = None
     GRAPH_API_KEY: t.Optional[SecretStr] = None
     TENDERLY_FORK_RPC: t.Optional[str] = None
@@ -104,6 +101,12 @@ class APIKeys(BaseSettings):
         data.SAFE_ADDRESS = None
         return data
 
+    @property
+    def safe_address_checksum(self) -> ChecksumAddress | None:
+        return (
+            Web3.to_checksum_address(self.SAFE_ADDRESS) if self.SAFE_ADDRESS else None
+        )
+
     @property
     def serper_api_key(self) -> SecretStr:
         return check_not_none(
@@ -148,7 +151,11 @@ class APIKeys(BaseSettings):
     @property
     def bet_from_address(self) -> ChecksumAddress:
         """If the SAFE is available, we always route transactions via SAFE. Otherwise we use the EOA."""
-        return self.SAFE_ADDRESS if self.SAFE_ADDRESS else self.public_key
+        return (
+            self.safe_address_checksum
+            if self.safe_address_checksum
+            else self.public_key
+        )
 
     @property
     def openai_api_key(self) -> SecretStr:
@@ -257,10 +264,10 @@ class APIKeys(BaseSettings):
         }
 
     def check_if_is_safe_owner(self, ethereum_client: EthereumClient) -> bool:
-        if not self.SAFE_ADDRESS:
+        if not self.safe_address_checksum:
             raise ValueError("Cannot check ownership if safe_address is not defined.")
 
-        s = SafeV141(self.SAFE_ADDRESS, ethereum_client)
+        s = SafeV141(self.safe_address_checksum, ethereum_client)
         public_key_from_signer = private_key_to_public_key(self.bet_from_private_key)
         return s.retrieve_is_owner(public_key_from_signer)
 

{prediction_market_agent_tooling-0.60.6 → prediction_market_agent_tooling-0.61.0}/prediction_market_agent_tooling/tools/contract.py

@@ -113,13 +113,13 @@ class ContractBaseClass(BaseModel):
         Used for changing a state (writing) to the contract.
         """
 
-        if api_keys.SAFE_ADDRESS:
+        if api_keys.safe_address_checksum:
             return send_function_on_contract_tx_using_safe(
                 web3=web3 or self.get_web3(),
                 contract_address=self.address,
                 contract_abi=self.abi,
                 from_private_key=api_keys.bet_from_private_key,
-                safe_address=api_keys.SAFE_ADDRESS,
+                safe_address=api_keys.safe_address_checksum,
                 function_name=function_name,
                 function_params=function_params,
                 tx_params=tx_params,

{prediction_market_agent_tooling-0.60.6 → prediction_market_agent_tooling-0.61.0}/prediction_market_agent_tooling/tools/langfuse_client_utils.py

@@ -64,24 +64,28 @@ def get_traces_for_agent(
     has_output: bool,
     client: Langfuse,
     to_timestamp: DatetimeUTC | None = None,
+    tags: str | list[str] | None = None,
 ) -> list[TraceWithDetails]:
     """
     Fetch agent traces using pagination
     """
+    total_pages = -1
     page = 1  # index starts from 1
     all_agent_traces = []
     while True:
-        logger.debug(f"fetching page {page}")
+        logger.debug(f"Fetching Langfuse page {page} / {total_pages}.")
         traces = client.fetch_traces(
             name=trace_name,
             limit=100,
             page=page,
             from_timestamp=from_timestamp,
             to_timestamp=to_timestamp,
+            tags=tags,
         )
         if not traces.data:
             break
         page += 1
+        total_pages = traces.meta.total_pages
 
         agent_traces = [
             t

{prediction_market_agent_tooling-0.60.6 → prediction_market_agent_tooling-0.61.0}/pyproject.toml

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