Allow heartbeat input

Author: stevensf1998

src/virtuals_sdk/game.py

@@ -161,13 +161,17 @@ def __init__(
         goal: str = "",
         description: str = "",
         world_info: str = "",
+        main_heartbeat: int = 15,
+        reaction_heartbeat: int = 5
     ):
         self.game_sdk = sdk.GameSDK(api_key)
         self.goal = goal
         self.description = description
         self.world_info = world_info
         self.enabled_functions: List[str] = []
         self.custom_functions: List[Function] = []
+        self.main_heartbeat = main_heartbeat
+        self.reaction_heartbeat = reaction_heartbeat
 
     def set_goal(self, goal: str):
         self.goal = goal
@@ -180,6 +184,14 @@ def set_description(self, description: str):
     def set_world_info(self, world_info: str):
         self.world_info = world_info
         return True
+    
+    def set_main_heartbeat(self, main_heartbeat: int):
+        self.main_heartbeat = main_heartbeat
+        return True
+    
+    def set_reaction_heartbeat(self, reaction_heartbeat: int):
+        self.reaction_heartbeat = reaction_heartbeat
+        return True
 
     def get_goal(self) -> str:
         return self.goal
@@ -254,7 +266,9 @@ def deploy_twitter(self):
             self.description,
             self.world_info,
             self.enabled_functions,
-            self.custom_functions
+            self.custom_functions,
+            self.main_heartbeat,
+            self.reaction_heartbeat
         )
 
     def export(self) -> str:

src/virtuals_sdk/sdk.py

@@ -90,7 +90,7 @@ def react(self, session_id: str, platform: str, goal: str,
 
         return response.json()["data"]
 
-    def deploy(self, goal: str, description: str, world_info: str, functions: list, custom_functions: list):
+    def deploy(self, goal: str, description: str, world_info: str, functions: list, custom_functions: list, main_heartbeat: int, reaction_heartbeat: int):
         """
         Simulate the agent configuration
         """
@@ -102,7 +102,11 @@ def deploy(self, goal: str, description: str, world_info: str, functions: list,
                     "description": description,
                     "worldInfo": world_info,
                     "functions": functions,
-                    "customFunctions": custom_functions
+                    "customFunctions": custom_functions,
+                    "gameState" : {
+                        "mainHeartbeat" : main_heartbeat,
+                        "reactionHeartbeat" : reaction_heartbeat,
+                    }
                 }
             },
             headers={"x-api-key": self.api_key}