prediction-market-agent-tooling 0.57.19.dev351__tar.gz → 0.58.0__tar.gz

{prediction_market_agent_tooling-0.57.19.dev351 → prediction_market_agent_tooling-0.58.0}/PKG-INFO

@@ -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 → prediction_market_agent_tooling-0.58.0}/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-0.57.19.dev351 → prediction_market_agent_tooling-0.58.0}/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-0.57.19.dev351 → prediction_market_agent_tooling-0.58.0}/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 → prediction_market_agent_tooling-0.58.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prediction-market-agent-tooling"
-version = "0.57.19.dev351"
+version = "0.58.0"
 description = "Tools to benchmark, deploy and monitor prediction market agents."
 authors = ["Gnosis"]
 readme = "README.md"
@@ -57,6 +57,8 @@ optuna = { version = "^4.1.0", optional = true}
 httpx = ">=0.25.2,<1.0.0"
 cowdao-cowpy = "^1.0.0rc1"
 eth-keys = "^0.6.1"
+proto-plus = "^1.0.0"
+protobuf = "^4.0.0"
 
 [tool.poetry.extras]
 openai = ["openai"]