Update abstra-cli

abstra_cli/apis/files.py

@@ -23,3 +23,7 @@ def list_workspace_files():
 
 def delete_file(filepath):
     return api_main.hf_api_runner("DELETE", "file", {"filepath": filepath})
+
+
+def delete_folder(path):
+    return api_main.hf_api_runner("DELETE", "folder", {"path": path})

abstra_cli/apis/main.py

@@ -3,7 +3,6 @@
 import urllib.request
 import urllib.response
 
-from .public import get_info_from_token
 import abstra_cli.credentials as credentials
 
 
@@ -14,9 +13,19 @@
 
 def get_auth_info():
     api_token = credentials.get_credentials()
-    if not api_token:
-        return None, None, None
-    workspace_id, author_id = get_info_from_token(api_token)
+    headers = credentials.get_auth_headers(api_token)
+    response = requests.get(f"https://auth.abstra.cloud/abstra-cloud", headers=headers)
+    response_json = response.json()
+    if response_json is None:
+        return api_token, None, None
+
+    author_id = response_json.get("author_id")
+    workspaces = response_json.get("workspaces", [{}])
+
+    workspace_id = None
+    if len(workspaces) != 0:
+        workspace_id = workspaces[0].get("id")
+
     return api_token, workspace_id, author_id