Merge branch 'main' into feat/update-readme

Author: celesteanglm

examples/hosted_agent/example-twitter.py

@@ -109,3 +109,6 @@
 
 # running simulation module only for platform twitter
 agent.simulate_twitter(session_id="session-twitter")
+
+# reset production memory
+#agent.reset_memory()

src/game_sdk/game/api.py

@@ -16,10 +16,10 @@ def _get_access_token(self) -> str:
             headers={"x-api-key": self.api_key}
         )
 
-        response_json = response.json()
         if response.status_code != 200:
-            raise ValueError(f"Failed to get token: {response_json}")
+            raise ValueError(f"Failed to get token (status {response.status_code}). Response: {response.text}")
 
+        response_json = response.json()
         return response_json["data"]["accessToken"]
 
     def _post(self, endpoint: str, data: dict) -> dict:
@@ -43,10 +43,10 @@ def _post(self, endpoint: str, data: dict) -> dict:
             headers={"Authorization": f"Bearer {access_token}"},
         )
 
-        response_json = response.json()
         if response.status_code != 200:
-            raise ValueError(f"Failed to post data: {response_json}")
+            raise ValueError(f"Failed to post data (status {response.status_code}). Response: {response.text}")
 
+        response_json = response.json()
         return response_json["data"]
 
     def create_agent(self, name: str, description: str, goal: str) -> str:

src/game_sdk/game/api_v2.py

@@ -28,9 +28,10 @@ def create_agent(self, name: str, description: str, goal: str) -> str:
             json=payload
         )
 
-        response_json = response.json()
         if response.status_code != 200:
-            raise ValueError(f"Failed to post data: {response_json}")
+            raise ValueError(f"Failed to create agent (status {response.status_code}). Response: {response.text}")
+
+        response_json = response.json()
 
         return response_json["data"]["id"]
 
@@ -53,9 +54,10 @@ def create_workers(self, workers: List) -> str:
             json=payload
         )
 
-        response_json = response.json()
         if response.status_code != 200:
-            raise ValueError(f"Failed to post data: {response_json}")
+            raise ValueError(f"Failed to get token (status {response.status_code}). Response: {response.text}")
+
+        response_json = response.json()
 
         return response_json["data"]["id"]
 
@@ -75,9 +77,10 @@ def set_worker_task(self, agent_id: str, task: str) -> Dict:
             json=payload
         )
 
-        response_json = response.json()
         if response.status_code != 200:
-            raise ValueError(f"Failed to post data: {response_json}")
+            raise ValueError(f"Failed to set worker task (status {response.status_code}). Response: {response.text}")
+
+        response_json = response.json()
 
         return response_json["data"]
 
@@ -93,9 +96,10 @@ def get_worker_action(self, agent_id: str, submission_id: str, data: dict) -> Di
             }
         )
 
-        response_json = response.json()
         if response.status_code != 200:
-            raise ValueError(f"Failed to post data: {response_json}")
+            raise ValueError(f"Failed to get worker action (status {response.status_code}). Response: {response.text}")
+
+        response_json = response.json()
 
         return response_json["data"]
 
@@ -111,8 +115,9 @@ def get_agent_action(self, agent_id: str, data: dict) -> Dict:
             }
         )
 
-        response_json = response.json()
         if response.status_code != 200:
-            raise ValueError(f"Failed to post data: {response_json}")
+            raise ValueError(f"Failed to get agent action (status {response.status_code}). Response: {response.text}")
+
+        response_json = response.json()
 
         return response_json["data"]

src/game_sdk/hosted_game/README.md

@@ -68,6 +68,10 @@ agent.get_description()
 agent.get_world_info()
 agent.get_task_description()
 
+# set game engine model
+# Available models: llama_3_1_405b, deepseek_r1, llama_3_3_70b_instruct, qwen2p5_72b_instruct, deepseek_v3
+agent.set_game_engine_model("llama_3_1_405b")
+
 # Set heartbeat
 agent.set_main_heartbeat(15)
 agent.set_reaction_heartbeat(5)

src/game_sdk/hosted_game/agent.py

@@ -164,7 +164,6 @@ class ContentLLMTemplate:
     top_p: float = 0.7
     repetition_penalty: float = 1.0
     type: str = None
-    taskDescription: str = None
     isSandbox: bool = False
 
     def _validate_fields(self):
@@ -190,11 +189,7 @@ def _validate_fields(self):
             self.type = self.template_type
             self.isSandbox = False
             self.userPrompt = self.user_prompt or ""
-            
-            # Additional settings for REPLY only
-            if self.template_type == "REPLY":
-                self.taskDescription = self.taskDescription or "Process incoming tweet. Ignore if it is boring or unimportant. Ignore if the conversation has gone too long."
-                
+                            
         elif self.template_type in ["TWITTER_START_SYSTEM_PROMPT", "TWITTER_END_SYSTEM_PROMPT", "SHARED"]:
             if not self.system_prompt:
                 raise ValueError("system_prompt is required")
@@ -259,7 +254,8 @@ def __init__(
         world_info: str = "",
         main_heartbeat: int = 15,
         reaction_heartbeat: int = 5,
-        task_description: str = ""
+        task_description: str = "",
+        game_engine_model: str = "llama_3_1_405b"
     ):
         self.game_sdk = sdk.GameSDK(api_key)
         self.goal = goal
@@ -272,7 +268,7 @@ def __init__(
         self.templates: List[ContentLLMTemplate] = []
         self.tweet_usernames: List[str] = []
         self.task_description: str = task_description
-        
+        self.game_engine_model: str = game_engine_model
 
     def set_goal(self, goal: str):
         self.goal = goal
@@ -298,8 +294,16 @@ def set_task_description(self, task_description: str):
         self.task_description = task_description
         return True
 
+    def set_game_engine_model(self, game_engine_model: str):
+        # Available models: llama_3_1_405b, deepseek_r1, llama_3_3_70b_instruct, qwen2p5_72b_instruct, deepseek_v3
+        self.game_engine_model = game_engine_model
+        return True
+    
     def get_task_description(self) -> str:
         return self.task_description
+    
+    def get_game_engine_model(self) -> str:
+        return self.game_engine_model
 
     def get_goal(self) -> str:
         return self.goal
@@ -378,7 +382,10 @@ def deploy_twitter(self):
             self.enabled_functions,
             self.custom_functions,
             self.main_heartbeat,
-            self.reaction_heartbeat
+            self.reaction_heartbeat,
+            self.tweet_usernames,
+            self.templates,
+            self.game_engine_model
         )
 
     def export(self) -> str:
@@ -442,3 +449,6 @@ def add_share_template(
         ))
 
         return True
+    
+    def reset_memory(self):
+        return self.game_sdk.reset_memory()

src/game_sdk/hosted_game/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, main_heartbeat: int, reaction_heartbeat: int, tweet_usernames: list = None, templates: list = None, task_description: str = None):
+    def deploy(self, goal: str, description: str, world_info: str, functions: list, custom_functions: list, main_heartbeat: int, reaction_heartbeat: int, tweet_usernames: list = None, templates: list = None, game_engine_model: str = "llama_3_1_405b"):
         """
         Simulate the agent configuration
         """
@@ -104,9 +104,9 @@ def deploy(self, goal: str, description: str, world_info: str, functions: list,
                 "mainHeartbeat": main_heartbeat,
                 "reactionHeartbeat": reaction_heartbeat,
             },
-            "taskDescription": task_description
+            "gameEngineModel": game_engine_model
         }
-        
+            
         if tweet_usernames is not None:
             payload["tweetUsernames"] = tweet_usernames
 
@@ -126,3 +126,12 @@ def deploy(self, goal: str, description: str, world_info: str, functions: list,
             raise Exception(response.json())
 
         return response.json()["data"]
+    
+    def reset_memory(self):
+        response = requests.get(
+            f"{self.api_url}/reset-session", headers={"x-api-key": self.api_key})
+
+        if (response.status_code != 200):
+            raise Exception("Failed to reset memory.")
+
+        return "Memory reset successfully."