prediction-market-agent-tooling 0.60.6__py3-none-any.whl → 0.61.1__py3-none-any.whl

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)
 
@@ -283,6 +290,9 @@ class RPCConfig(BaseSettings):
     def chain_id(self) -> ChainID:
         return check_not_none(self.CHAIN_ID, "CHAIN_ID missing in the environment.")
 
+    def get_web3(self) -> Web3:
+        return Web3(Web3.HTTPProvider(self.gnosis_rpc_url))
+
 
 class CloudCredentials(BaseSettings):
     model_config = SettingsConfigDict(

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.60.6
+Version: 0.61.1
 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.dist-info → prediction_market_agent_tooling-0.61.1.dist-info}/RECORD

@@ -20,7 +20,7 @@ prediction_market_agent_tooling/benchmark/__init__.py,sha256=47DEQpj8HBSa-_TImW-
 prediction_market_agent_tooling/benchmark/agents.py,sha256=B1-uWdyeN4GGKMWGK_-CcAFJg1m9Y_XuaeIHPB29QR8,3971
 prediction_market_agent_tooling/benchmark/benchmark.py,sha256=MqTiaaJ3cYiOLUVR7OyImLWxcEya3Rl5JyFYW-K0lwM,17097
 prediction_market_agent_tooling/benchmark/utils.py,sha256=D0MfUkVZllmvcU0VOurk9tcKT7JTtwwOp-63zuCBVuc,2880
-prediction_market_agent_tooling/config.py,sha256=M1bbUiADK3cbOp-S19u_x5GciOM8h86ZNcZ0GSD04lQ,9923
+prediction_market_agent_tooling/config.py,sha256=So5l8KbgmzcCpxzzf13TNrEJPu_4iQnUDhzus6XRvSc,10151
 prediction_market_agent_tooling/deploy/agent.py,sha256=tMREXM2LwFsatbysCaNRvtCAyCNMAPMGgkkIEpCRj7g,25022
 prediction_market_agent_tooling/deploy/agent_example.py,sha256=dIIdZashExWk9tOdyDjw87AuUcGyM7jYxNChYrVK2dM,1001
 prediction_market_agent_tooling/deploy/betting_strategy.py,sha256=Y6Pb8OfSb6galRbfdNBvvNTgO-4dR2ybJ4o5GKJcMoM,12894
@@ -83,7 +83,7 @@ prediction_market_agent_tooling/tools/betting_strategies/utils.py,sha256=kpIb-ci
 prediction_market_agent_tooling/tools/caches/db_cache.py,sha256=dB8LNs2JvVRaFCeAKRmIQRwiirsMgtL31he8051wM-g,11431
 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=BCl3R-n5jpp5XPAho9yTuUxRbOoKbZSga1Qaa4RfAp8,20945
+prediction_market_agent_tooling/tools/contract.py,sha256=XM7v6Wmi5OXPtn0SS__27MhlaBHGJG3VEeQFSIBJo6U,20963
 prediction_market_agent_tooling/tools/costs.py,sha256=EaAJ7v9laD4VEV3d8B44M4u3_oEO_H16jRVCdoZ93Uw,954
 prediction_market_agent_tooling/tools/cow/cow_manager.py,sha256=WK6Uk722VotjLHtxDPHxvwBrWVb3rvTegg_3w58ehwU,3869
 prediction_market_agent_tooling/tools/cow/cow_order.py,sha256=M3zQohgAzy_LETnf9rKtS1L9rr7FP92CH6v0G2laZkM,4435
@@ -99,7 +99,7 @@ prediction_market_agent_tooling/tools/ipfs/ipfs_handler.py,sha256=CTTMfTvs_8PH4k
 prediction_market_agent_tooling/tools/is_invalid.py,sha256=TAHQXiiusAU45xJ11ZyEP7PnEfcjfzVG7qHRbsHiAd0,5335
 prediction_market_agent_tooling/tools/is_predictable.py,sha256=qVd6zqay2Dg2fyeAuZvAFqSHMg71TcPfCZULsVk2XvA,6797
 prediction_market_agent_tooling/tools/langfuse_.py,sha256=jI_4ROxqo41CCnWGS1vN_AeDVhRzLMaQLxH3kxDu3L8,1153
-prediction_market_agent_tooling/tools/langfuse_client_utils.py,sha256=83T31s-YbsnNBLyYCjmBI2BBKUEqJUuYFa0uCdkoqy8,5901
+prediction_market_agent_tooling/tools/langfuse_client_utils.py,sha256=IQboU9EPl4QEIo0poNylomevuVntpPpmkuNCzZl1Qdg,6058
 prediction_market_agent_tooling/tools/omen/reality_accuracy.py,sha256=M1SF7iSW1gVlQSTskdVFTn09uPLST23YeipVIWj54io,2236
 prediction_market_agent_tooling/tools/omen/sell_positions.py,sha256=hZCxXpcACO95DyiZ5oLFp982N0erZg4wccdSUKTgRlA,2307
 prediction_market_agent_tooling/tools/parallelism.py,sha256=6Gou0hbjtMZrYvxjTDFUDZuxmE2nqZVbb6hkg1hF82A,1022
@@ -117,8 +117,8 @@ prediction_market_agent_tooling/tools/tokens/main_token.py,sha256=7JPgVF4RbiFzLD
 prediction_market_agent_tooling/tools/transaction_cache.py,sha256=K5YKNL2_tR10Iw2TD9fuP-CTGpBbZtNdgbd0B_R7pjg,1814
 prediction_market_agent_tooling/tools/utils.py,sha256=jLG4nbEoIzzJiZ4RgMx4Q969Zdl0p0s63p8uET_0Fuw,6440
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=3wfqNxvMn44ivweFRoeKNVb9QRtFd7kFtp7VUY5juEE,12862
-prediction_market_agent_tooling-0.60.6.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.60.6.dist-info/METADATA,sha256=kuT3T3Lv5FgZC2BNXWTtV3BToC9c4pWfydW7fA86UJU,8629
-prediction_market_agent_tooling-0.60.6.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-prediction_market_agent_tooling-0.60.6.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.60.6.dist-info/RECORD,,
+prediction_market_agent_tooling-0.61.1.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.61.1.dist-info/METADATA,sha256=D-0HAHmdDruqGK77Sspw6a8Y9WFnqeqncwenPgUwhf8,8629
+prediction_market_agent_tooling-0.61.1.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+prediction_market_agent_tooling-0.61.1.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.61.1.dist-info/RECORD,,