entrypoint_function_name variable for GCP (#20)

prediction_market_agent_tooling/deploy/agent.py

@@ -75,13 +75,14 @@ def deploy_gcp(
     ) -> None:
         path_to_agent_file = os.path.relpath(inspect.getfile(self.__class__))
 
+        entrypoint_function_name = "main"
         entrypoint_template = f"""
 from {path_to_agent_file.replace("/", ".").replace(".py", "")} import *
 import functions_framework
 from prediction_market_agent_tooling.markets.markets import MarketType
 
 @functions_framework.http
-def main(request) -> str:
+def {entrypoint_function_name}(request) -> str:
     {self.__class__.__name__}().run(market_type={market_type.__class__.__name__}.{market_type.name})
     return "Success"
 """
@@ -101,6 +102,7 @@ def main(request) -> str:
                 env_vars=env_vars,
                 secrets=secrets,
                 memory=memory,
+                entrypoint_function_name=entrypoint_function_name,
             )
 
         # Check that the function is deployed

prediction_market_agent_tooling/deploy/gcp/deploy.py

@@ -28,6 +28,7 @@ def deploy_to_gcp(
     env_vars: dict[str, str] | None,
     secrets: dict[str, str] | None,
     memory: int,  # in MB
+    entrypoint_function_name: str,
 ) -> str:
     if requirements_file and not os.path.exists(requirements_file):
         raise ValueError(f"File {requirements_file} does not exist")
@@ -63,7 +64,7 @@ def deploy_to_gcp(
         cmd = gcloud_deploy_cmd(
             gcp_function_name=gcp_fname,
             source=tempdir,
-            entry_point="main",  # TODO check this function exists in main.py
+            entry_point=entrypoint_function_name,
             labels=labels,
             env_vars=env_vars,
             secrets=secrets,