Merge pull request game-by-virtuals#66 from game-by-virtuals/fix_response_handling

Correcting response handling

Author: celesteanglm

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"]