Fix formatting

Author: orangetin

src/together/api.py

@@ -1,8 +1,8 @@
 import os
-from typing import Any, Dict, List, Optional, cast
+import urllib.parse
+from typing import Any, List, Optional
 
 import requests
-import urllib.parse
 
 from together.files import Files
 from together.finetune import Finetune
@@ -40,7 +40,7 @@ def __init__(
         self.endpoint_url = endpoint_url
         self.supply_endpoint_url = supply_endpoint_url
 
-    def get_all_models(self) -> Dict[str, Any]:
+    def get_models(self) -> List[Any]:
         model_url = urllib.parse.urljoin(self.endpoint_url, "models/info?=")
         headers = {
             "Authorization": f"Bearer {self.together_api_key}",
@@ -55,23 +55,14 @@ def get_all_models(self) -> Dict[str, Any]:
             exit_1(self.logger)
 
         try:
-            response_json = list(response.json())
+            response_list = list(response.json())
         except Exception as e:
             self.logger.critical(
                 f"JSON Error raised: {e}\nResponse status code = {response.status_code}"
             )
             exit_1(self.logger)
 
-        return response_json
-
-    def get_available_models(self) -> List[str]:
-        res = self.get_supply()
-        names = res["result"].keys()
-        available_models = [
-            name[:-1] for name in names if res["result"][name]["num_asks"] > 0
-        ]
-
-        return available_models
+        return response_list
 
     def finetune(self) -> Finetune:
         return Finetune(

src/together/commands/api.py

@@ -35,12 +35,11 @@ def _add_list(
 
 def _run_list(args: argparse.Namespace) -> None:
     api = API(endpoint_url=args.endpoint, log_level=args.log)
-    response = api.get_all_models()
+    response = api.get_models()
     if args.raw:
         print(json.dumps(response, indent=4))
     else:
         models = []
         for i in response:
             models.append(i["name"])
         print(json.dumps(models, indent=4))
-