Cut out tooling from the PMA repository

examples/cloud_deployment/gcp/agent.py

@@ -0,0 +1,23 @@
+from flask.wrappers import Request
+import functions_framework
+import random
+
+from prediction_market_agent_tooling.markets.data_models import AgentMarket
+from prediction_market_agent_tooling.deploy.agent import DeployableAgent
+from prediction_market_agent_tooling.markets.markets import MarketType
+
+
+class DeployableCoinFlipAgent(DeployableAgent):
+    def pick_markets(self, markets: list[AgentMarket]) -> list[AgentMarket]:
+        if len(markets) > 1:
+            return random.sample(markets, 1)
+        return markets
+
+    def answer_binary_market(self, market: AgentMarket) -> bool:
+        return random.choice([True, False])
+
+
+@functions_framework.http
+def main(request: Request) -> str:
+    DeployableCoinFlipAgent().run(market_type=MarketType.MANIFOLD)
+    return "Success"

examples/cloud_deployment/gcp/deploy.py

@@ -0,0 +1,37 @@
+import os
+
+from prediction_market_agent_tooling.deploy.gcp.deploy import (
+    deploy_to_gcp,
+    remove_deployed_gcp_function,
+    run_deployed_gcp_function,
+    schedule_deployed_gcp_function,
+)
+from prediction_market_agent_tooling.deploy.gcp.utils import gcp_function_is_active
+from prediction_market_agent_tooling.markets.markets import MarketType
+from prediction_market_agent_tooling.config import APIKeys
+
+if __name__ == "__main__":
+    current_dir = os.path.dirname(os.path.realpath(__file__))
+    fname = deploy_to_gcp(
+        requirements_file=f"{current_dir}/../../pyproject.toml",
+        extra_deps=[
+            "git+https://github.com/gnosis/prediction-market-agent.git@evan/deploy-agent"
+        ],
+        function_file=f"{current_dir}/agent.py",
+        market_type=MarketType.MANIFOLD,
+        api_keys={"MANIFOLD_API_KEY": APIKeys().manifold_api_key},
+        memory=512,
+    )
+
+    # Check that the function is deployed
+    assert gcp_function_is_active(fname)
+
+    # Run the function
+    response = run_deployed_gcp_function(fname)
+    assert response.ok
+
+    # Schedule the function
+    schedule_deployed_gcp_function(fname, cron_schedule="* * * * *")
+
+    # Delete the function
+    remove_deployed_gcp_function(fname)

mypy.ini

@@ -1,6 +1,6 @@
 [mypy]
 python_version = 3.9
-files = prediction_market_agent_tooling/
+files = prediction_market_agent_tooling/, tests/, examples/
 plugins = pydantic.mypy
 warn_redundant_casts = True
 warn_unused_ignores = True