prediction-market-agent-tooling 0.57.19.dev351__py3-none-any.whl → 0.58.0__py3-none-any.whl

prediction_market_agent_tooling/config.py

@@ -1,7 +1,9 @@
+import json
 import typing as t
+from copy import deepcopy
 
 from eth_account.signers.local import LocalAccount
-from pydantic import Field
+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
@@ -9,6 +11,7 @@ from safe_eth.eth import EthereumClient
 from safe_eth.safe.safe import SafeV141
 from web3 import Account
 
+from prediction_market_agent_tooling.deploy.gcp.utils import gcp_get_secret_value
 from prediction_market_agent_tooling.gtypes import (
     ChainID,
     ChecksumAddress,
@@ -60,6 +63,34 @@ class APIKeys(BaseSettings):
     ENABLE_CACHE: bool = False
     CACHE_DIR: str = "./.cache"
 
+    @model_validator(mode="before")
+    @classmethod
+    def _model_validator(cls, data: t.Any) -> t.Any:
+        data = deepcopy(data)
+        data = cls._replace_gcp_secrets(data)
+        return data
+
+    @staticmethod
+    def _replace_gcp_secrets(data: t.Any) -> t.Any:
+        if isinstance(data, dict):
+            for k, v in data.items():
+                # Check if the value is meant to be fetched from GCP Secret Manager, if so, replace it with it.
+                if isinstance(v, (str, SecretStr)):
+                    secret_value = (
+                        v.get_secret_value() if isinstance(v, SecretStr) else v
+                    )
+                    if secret_value.startswith("gcps:"):
+                        # We assume that secrets are dictionaries and the value is a key in the dictionary,
+                        # example usage: `BET_FROM_PRIVATE_KEY=gcps:my-agent:private_key`
+                        _, secret_name, key_name = secret_value.split(":")
+                        secret_data = json.loads(gcp_get_secret_value(secret_name))[
+                            key_name
+                        ]
+                        data[k] = secret_data
+        else:
+            raise ValueError("Data must be a dictionary.")
+        return data
+
     @property
     def manifold_user_id(self) -> str:
         return get_authenticated_user(

prediction_market_agent_tooling/deploy/agent.py

@@ -25,10 +25,7 @@ from prediction_market_agent_tooling.deploy.gcp.deploy import (
     run_deployed_gcp_function,
     schedule_deployed_gcp_function,
 )
-from prediction_market_agent_tooling.deploy.gcp.utils import (
-    gcp_function_is_active,
-    gcp_resolve_api_keys_secrets,
-)
+from prediction_market_agent_tooling.deploy.gcp.utils import gcp_function_is_active
 from prediction_market_agent_tooling.deploy.trade_interval import (
     FixedInterval,
     TradeInterval,
@@ -228,7 +225,7 @@ def {entrypoint_function_name}(request) -> str:
         monitor_agent = MARKET_TYPE_TO_DEPLOYED_AGENT[market_type].from_api_keys(
             name=gcp_fname,
             start_time=start_time or utcnow(),
-            api_keys=gcp_resolve_api_keys_secrets(api_keys),
+            api_keys=api_keys,
         )
         env_vars |= monitor_agent.model_dump_prefixed()
 

prediction_market_agent_tooling/deploy/gcp/utils.py

@@ -8,7 +8,6 @@ from google.cloud.functions_v2.services.function_service.client import (
 from google.cloud.functions_v2.types.functions import Function
 from google.cloud.secretmanager import SecretManagerServiceClient
 
-from prediction_market_agent_tooling.config import APIKeys
 from prediction_market_agent_tooling.deploy.gcp.kubernetes_models import (
     KubernetesCronJobsModel,
 )
@@ -203,16 +202,3 @@ def gcp_get_secret_value(name: str, version: str = "latest") -> str:
     return client.access_secret_version(
         name=f"projects/{get_gcloud_project_id()}/secrets/{name}/versions/{version}"
     ).payload.data.decode("utf-8")
-
-
-def gcp_resolve_api_keys_secrets(api_keys: APIKeys) -> APIKeys:
-    return APIKeys.model_validate(
-        api_keys.model_dump_public()
-        | {
-            k: gcp_get_secret_value(
-                name=v.rsplit(":", 1)[0],
-                version=v.rsplit(":", 1)[1],
-            )
-            for k, v in api_keys.model_dump_secrets().items()
-        }
-    )

{prediction_market_agent_tooling-0.57.19.dev351.dist-info → prediction_market_agent_tooling-0.58.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.57.19.dev351
+Version: 0.58.0
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.13
@@ -38,6 +38,8 @@ Requires-Dist: openai (>=1.0.0,<2.0.0) ; extra == "openai"
 Requires-Dist: optuna (>=4.1.0,<5.0.0) ; extra == "optuna"
 Requires-Dist: pinatapy-vourhey (>=0.2.0,<0.3.0)
 Requires-Dist: prompt-toolkit (>=3.0.43,<4.0.0)
+Requires-Dist: proto-plus (>=1.0.0,<2.0.0)
+Requires-Dist: protobuf (>=4.0.0,<5.0.0)
 Requires-Dist: psycopg2-binary (>=2.9.9,<3.0.0)
 Requires-Dist: pydantic (>=2.6.1,<3.0.0)
 Requires-Dist: pydantic-settings (>=2.4.0,<3.0.0)

{prediction_market_agent_tooling-0.57.19.dev351.dist-info → prediction_market_agent_tooling-0.58.0.dist-info}/RECORD

@@ -22,14 +22,14 @@ 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=QBrbqG9wN1JFBUt_kCl2ybPamobKP6UhfiaRZK_VjMo,7805
-prediction_market_agent_tooling/deploy/agent.py,sha256=58Ms6wFcRM3yhpdDVT-ohryZhHJNinn1Z4MdrmCMXvM,23627
+prediction_market_agent_tooling/config.py,sha256=owJ3goDbH1aeX8PzeJCeGK5pitYJqqk7yFOkaTbDarY,9209
+prediction_market_agent_tooling/deploy/agent.py,sha256=3VzmTIekbICWrpoo0VjIbR7s_AKjhYaAaUP5RITDGek,23554
 prediction_market_agent_tooling/deploy/agent_example.py,sha256=dIIdZashExWk9tOdyDjw87AuUcGyM7jYxNChYrVK2dM,1001
 prediction_market_agent_tooling/deploy/betting_strategy.py,sha256=kMrIE3wMv_IB6nJd_1DmDXDkEZhsXFOgyTd7JZ0gqHI,13068
 prediction_market_agent_tooling/deploy/constants.py,sha256=M5ty8URipYMGe_G-RzxRydK3AFL6CyvmqCraJUrLBnE,82
 prediction_market_agent_tooling/deploy/gcp/deploy.py,sha256=CYUgnfy-9XVk04kkxA_5yp0GE9Mw5caYqlFUZQ2j3ks,3739
 prediction_market_agent_tooling/deploy/gcp/kubernetes_models.py,sha256=OsPboCFGiZKsvGyntGZHwdqPlLTthITkNF5rJFvGgU8,2582
-prediction_market_agent_tooling/deploy/gcp/utils.py,sha256=oyW0jgrUT2Tr49c7GlpcMsYNQjoCSOcWis3q-MmVAhU,6089
+prediction_market_agent_tooling/deploy/gcp/utils.py,sha256=dzAvSQWRc2pndHFyKhVgrTtDm91K-csgusEaFRyfSow,5664
 prediction_market_agent_tooling/deploy/trade_interval.py,sha256=Xk9j45alQ_vrasGvsNyuW70XHIQ7wfvjoxNR3F6HYCw,1155
 prediction_market_agent_tooling/gtypes.py,sha256=G9KOKqYcxoKLv5Tfto4g5zq46FeIKxGl4RTArLIJn3I,2563
 prediction_market_agent_tooling/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -117,8 +117,8 @@ prediction_market_agent_tooling/tools/tokens/main_token.py,sha256=5iHO7-iehSlXuu
 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=wqUDCed3iNrn1Wao1iwGN6tzIrhpzrTRj319wlveJEo,12275
-prediction_market_agent_tooling-0.57.19.dev351.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.57.19.dev351.dist-info/METADATA,sha256=oN2mVZGHltIEUDrKImmPS009l2SxPLdITAVL1k40BuI,8553
-prediction_market_agent_tooling-0.57.19.dev351.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-prediction_market_agent_tooling-0.57.19.dev351.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.57.19.dev351.dist-info/RECORD,,
+prediction_market_agent_tooling-0.58.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.58.0.dist-info/METADATA,sha256=E_PM7SHleOyz3e52aCzYfAdmKRrfRZqR05PzbQLZ6kk,8629
+prediction_market_agent_tooling-0.58.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+prediction_market_agent_tooling-0.58.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.58.0.dist-info/RECORD,,

{prediction_market_agent_tooling-0.57.19.dev351.dist-info → prediction_market_agent_tooling-0.58.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.0.1
+Generator: poetry-core 2.1.1
 Root-Is-Purelib: true
 Tag: py3-none-any