feat: remove verification of instance in the decorator and use delete_token response instead

Author: atadj

src/sumo/wrapper/_decorators.py

@@ -1,8 +1,6 @@
 # For sphinx:
 from functools import wraps
 
-from sumo.wrapper._auth_provider import AuthProviderSumoToken
-
 
 def raise_for_status(func):
     @wraps(func)
@@ -11,9 +9,7 @@ def wrapper(self, *args, **kwargs):
         # so we could simply write
         # return func(*args, **kwargs).raise_for_status()
         response = func(self, *args, **kwargs)
-        if response.status_code == 401 and isinstance(
-            self.auth, AuthProviderSumoToken
-        ):
+        if response.status_code == 401:
             self._handle_invalid_shared_key()
         response.raise_for_status()
         return response
@@ -28,9 +24,7 @@ async def wrapper(self, *args, **kwargs):
         # so we could simply write
         # return func(*args, **kwargs).raise_for_status()
         response = await func(self, *args, **kwargs)
-        if response.status_code == 401 and isinstance(
-            self.auth, AuthProviderSumoToken
-        ):
+        if response.status_code == 401:
             self._handle_invalid_shared_key()
         response.raise_for_status()
         return response

src/sumo/wrapper/sumo_client.py

@@ -210,13 +210,10 @@ def blob_client(self) -> BlobClient:
 
     def _handle_invalid_shared_key(self):
         """Handle the invalid shared key by deleting it."""
-        self.auth.delete_token()
-        logger.error(
-            "Invalid shared key detected and deleted, run again to reset automatically"
-        )
-        print(
-            "Invalid shared key detected and deleted, run again to reset automatically"
-        )
+        if self.auth.delete_token():
+            print(
+                "Invalid shared key detected and deleted, run again to reset automatically"
+            )
 
     @raise_for_status
     def get(self, path: str, params: Optional[Dict] = None) -> httpx.Response: