update tests; update headers

.github/workflows/TagBot.yml

@@ -29,5 +29,4 @@ jobs:
         with:
           token: ${{ secrets.GITHUB_TOKEN }}
           # Edit the following line to reflect the actual name of the GitHub Secret containing your private key
-          #ssh: ${{ secrets.BOT_KEY }}
           # ssh: ${{ secrets.NAME_OF_MY_SSH_PRIVATE_KEY_SECRET }}

python/bytez/main.py

@@ -15,12 +15,12 @@ def __init__(self, api_key):
     Args:
         api_key (str): Your Bytez API key.
     """
-    self.auth = {"authorization": f"Key {api_key}"}
+    self.headers = {"authorization": f"Key {api_key}", "content-type": "application/json" }
 
   def _request(self, path="", body=None, stream=False):
     try:
       url = "https://api.bytez.com/" + path
-      headers = {**self.auth, "content-type": "application/json"} if body else self.auth
+      headers = self.headers
       method = "POST" if body else "GET"
       data = json.dumps(body) if body else None
 

python/test.py

@@ -13,7 +13,7 @@ def test_list_models(self):
 
     self.assertIsInstance(model_list, list, "should return array of models")
     self.assertNotEqual(len(model_list), 0, "array is not empty")
-    self.assertTrue(all('name' in model and 'requiredRamGB' in model for model in model_list),
+    self.assertTrue(all('modelId' in model and 'ramRequired' in model for model in model_list),
                     "all models should have name and RAM")
 
   def test_lists_running_instances(self):