Merge pull request #62 from stevensf1998/feat/allow-reset-memory

Allow to reset memory for GAME production

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/hosted_game/agent.py

@@ -442,3 +442,6 @@ def add_share_template(
         ))
 
         return True
+    
+    def reset_memory(self):
+        return self.game_sdk.reset_memory()

src/game_sdk/hosted_game/sdk.py

@@ -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."