Only one instance on GCP deployment and small tweaks

examples/cloud_deployment/gcp/deploy.py

@@ -1,13 +1,27 @@
 import getpass
 
-from prediction_market_agent_tooling.deploy.agent_example import DeployableCoinFlipAgent
+import typer
+
+from prediction_market_agent_tooling.deploy.agent_example import (
+    DeployableAgent,
+    DeployableAlwaysRaiseAgent,
+    DeployableCoinFlipAgent,
+)
 from prediction_market_agent_tooling.markets.markets import MarketType
 
-if __name__ == "__main__":
-    agent = DeployableCoinFlipAgent()
+
+def main(
+    agent_name: str,
+    cron_schedule: str = "0 */2 * * *",
+    branch: str = "main",
+    custom_gcp_fname: str | None = None,
+) -> None:
+    agent: DeployableAgent = {
+        "coin_flip": DeployableCoinFlipAgent,
+        "always_raise": DeployableAlwaysRaiseAgent,
+    }[agent_name]()
     agent.deploy_gcp(
-        # TODO: Switch to main.
-        repository="git+https://github.com/gnosis/prediction-market-agent-tooling.git@peter/refactor-deployment",
+        repository=f"git+https://github.com/gnosis/prediction-market-agent-tooling.git@{branch}",
         market_type=MarketType.MANIFOLD,
         labels={
             "owner": getpass.getuser()
@@ -18,5 +32,10 @@
             "MANIFOLD_API_KEY": f"JUNG_PERSONAL_GMAIL_MANIFOLD_API_KEY:latest"
         },  # Must be in the format "env_var_in_container => secret_name:version", you can create secrets using `gcloud secrets create --labels owner=<your-name> <secret-name>` command.
         memory=256,
-        cron_schedule="0 */2 * * *",
+        cron_schedule=cron_schedule,
+        gcp_fname=custom_gcp_fname,
     )
+
+
+if __name__ == "__main__":
+    typer.run(main)

prediction_market_agent_tooling/deploy/agent.py

@@ -71,6 +71,7 @@ def deploy_gcp(
         env_vars: dict[str, str] | None = None,
         secrets: dict[str, str] | None = None,
         cron_schedule: str | None = None,
+        gcp_fname: str | None = None,
     ) -> None:
         path_to_agent_file = os.path.relpath(inspect.getfile(self.__class__))
 
@@ -85,7 +86,7 @@ def main(request) -> str:
     return "Success"
 """
 
-        gcp_fname = self.get_gcloud_fname(market_type)
+        gcp_fname = gcp_fname or self.get_gcloud_fname(market_type)
 
         with tempfile.NamedTemporaryFile(mode="w", suffix=".py") as f:
             f.write(entrypoint_template)

prediction_market_agent_tooling/deploy/agent_example.py

@@ -6,9 +6,12 @@
 
 class DeployableCoinFlipAgent(DeployableAgent):
     def pick_markets(self, markets: list[AgentMarket]) -> list[AgentMarket]:
-        if len(markets) > 1:
-            return random.sample(markets, 1)
-        return markets
+        return random.sample(markets, 1)
 
     def answer_binary_market(self, market: AgentMarket) -> bool:
         return random.choice([True, False])
+
+
+class DeployableAlwaysRaiseAgent(DeployableAgent):
+    def answer_binary_market(self, market: AgentMarket) -> bool:
+        raise RuntimeError("I always raise!")

prediction_market_agent_tooling/deploy/gcp/utils.py

@@ -15,6 +15,8 @@ def gcloud_deploy_cmd(
     env_vars: dict[str, str] | None,
     secrets: dict[str, str] | None,
     memory: int,  # in MB
+    timeout: int = 180,
+    retry_on_failure: bool = False,
 ) -> str:
     cmd = (
         f"gcloud functions deploy {gcp_function_name} "
@@ -26,7 +28,14 @@ def gcloud_deploy_cmd(
         f"--entry-point {entry_point} "
         f"--memory {memory}MB "
         f"--no-allow-unauthenticated "
+        f"--timeout {timeout}s "
+        # Explicitly set no concurrency, min instances to 0 (agent is executed only once in a while) and max instances to 1 (parallel agents aren't allowed).
+        "--concurrency 1 "
+        "--min-instances 0 "
+        "--max-instances 1 "
     )
+    if retry_on_failure:
+        cmd += "--retry "
     if labels:
         for k, v in labels.items():
             cmd += f"--update-labels {k}={v} "

prediction_market_agent_tooling/markets/data_models.py

@@ -156,7 +156,7 @@ class ManifoldMarket(BaseModel):
     isResolved: bool
     resolution: t.Optional[str] = None
     resolutionTime: t.Optional[datetime] = None
-    lastBetTime: datetime
+    lastBetTime: t.Optional[datetime] = None
     lastCommentTime: t.Optional[datetime] = None
     lastUpdatedTime: datetime
     mechanism: str