truefoundry 0.3.0rc9__tar.gz → 0.3.0rc11__tar.gz

{truefoundry-0.3.0rc9 → truefoundry-0.3.0rc11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.3.0rc9
+Version: 0.3.0rc11
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com
@@ -24,7 +24,7 @@ Requires-Dist: flytekit (==1.12.2) ; extra == "workflow"
 Requires-Dist: gitignorefile (>=1.1.2,<1.2.0)
 Requires-Dist: importlib-metadata (>=6.0.1,<8.0.0)
 Requires-Dist: importlib-resources (>=5.2.0,<6.0.0)
-Requires-Dist: mlfoundry (==0.11.2) ; extra == "ml"
+Requires-Dist: mlfoundry (==0.11.4) ; extra == "ml"
 Requires-Dist: openai (>=1.16.2,<2.0.0)
 Requires-Dist: packaging (>=20.0,<25.0)
 Requires-Dist: pydantic (>=1.10.0,<3)

{truefoundry-0.3.0rc9 → truefoundry-0.3.0rc11}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truefoundry"
-version = "0.3.0rc9"
+version = "0.3.0rc11"
 description = "Truefoundry CLI"
 authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
 readme = "README.md"
@@ -17,7 +17,7 @@ GitPython = ">=3.1.43,<3.2.0"
 importlib-metadata = ">=6.0.1,<8.0.0"
 importlib-resources = "^5.2.0"
 Mako = "^1.1.6"
-mlfoundry = { version = "0.11.2", optional = true }
+mlfoundry = { version = "0.11.4", optional = true }
 openai = ">=1.16.2,<2.0.0"
 packaging = ">=20.0,<25.0"
 pydantic = ">=1.10.0,<3"

{truefoundry-0.3.0rc9 → truefoundry-0.3.0rc11}/truefoundry/deploy/__init__.py

@@ -16,6 +16,7 @@ from truefoundry.deploy.lib.dao.application import (
     list_job_runs,
     terminate_job_run,
     trigger_job,
+    trigger_workflow,
 )
 from truefoundry.deploy.lib.dao.version import (
     get_version as get_application_version,

{truefoundry-0.3.0rc9 → truefoundry-0.3.0rc11}/truefoundry/deploy/cli/cli.py

@@ -13,6 +13,7 @@ from truefoundry.deploy.cli.commands import (
     get_logout_command,
     get_patch_application_command,
     get_patch_command,
+    get_trigger_command,
 )
 from truefoundry.deploy.cli.config import CliConfig
 from truefoundry.deploy.cli.const import GROUP_CLS
@@ -36,6 +37,7 @@ def create_truefoundry_cli():
     cli.add_command(get_deploy_command())
     cli.add_command(get_patch_application_command())
     cli.add_command(get_delete_command())
+    cli.add_command(get_trigger_command())
 
     if not (sys.platform.startswith("win32") or sys.platform.startswith("cygwin")):
         cli.add_command(get_patch_command())

{truefoundry-0.3.0rc9 → truefoundry-0.3.0rc11}/truefoundry/deploy/cli/commands/trigger_command.py

@@ -82,6 +82,64 @@ def trigger_job(application_fqn: str, command: Optional[Sequence[str]], params):
     )
 
 
+@click.command(
+    name="workflow",
+    cls=COMMAND_CLS,
+    context_settings={"ignore_unknown_options": True, "allow_extra_args": True},
+)
+@click.option(
+    "--application-fqn",
+    "--application_fqn",
+    type=click.STRING,
+    required=True,
+    help="FQN of the workflow application",
+)
+@click.argument(
+    "inputs",
+    type=click.STRING,
+    nargs=-1,
+    required=False,
+)
+@handle_exception_wrapper
+def trigger_workflow(application_fqn: str, inputs):
+    """
+    Trigger a Workflow on TrueFoundry
+
+        [b]tfy trigger workflow --application-fqn "my-cluster:my-workspace:my-workflow"[/]
+
+    \n
+    Additionally, you can pass inputs (if defined in the workflow)\n\n
+
+    Passing inputs:
+
+        [b]tfy trigger workflow --application-fqn "my-cluster:my-workspace:my-workflow" -- --input1_name input1_value --input2_name input2_value ...[/]
+    """
+    if inputs:
+        inputs_dict = {}
+        if len(inputs) % 2 != 0:
+            raise ClickException(
+                f"Found odd number of argument pairs: {inputs}. "
+                "Perhaps you forgot to pass a value for one of the inputs? "
+                "inputs for workflow should be passed in the "
+                "format `--input1_name input1_value --input2_name input2_value ...`"
+            )
+        for i in range(0, len(inputs), 2):
+            key = inputs[i]
+            value = inputs[i + 1]
+            if not key.startswith("--"):
+                raise ClickException(
+                    f"Got ambiguous argument {key!r} in inputs: {inputs}. "
+                    f"input names should be prefixed with '--' i.e. "
+                    "inputs for workflow should be passed in the "
+                    "format `--input1_name input1_value --input2_name input2_value ...`"
+                )
+            key = key.lstrip("-")
+            inputs_dict[key] = value
+
+    application.trigger_workflow(application_fqn=application_fqn, inputs=inputs)
+
+
 def get_trigger_command():
     trigger_command.add_command(trigger_job)
+    trigger_command.add_command(trigger_workflow)
     return trigger_command

truefoundry-0.3.0rc11/truefoundry/deploy/lib/auth/auth_service_client.py

@@ -0,0 +1,181 @@
+import time
+from abc import ABC, abstractmethod
+
+import requests
+
+from truefoundry.deploy.lib.clients.utils import poll_for_function, request_handling
+from truefoundry.deploy.lib.const import VERSION_PREFIX
+from truefoundry.deploy.lib.exceptions import BadRequestException
+from truefoundry.deploy.lib.model.entity import DeviceCode, Token
+from truefoundry.logger import logger
+
+
+class AuthServiceClient(ABC):
+    def __init__(self, base_url):
+        from truefoundry.deploy.lib.clients.servicefoundry_client import (
+            ServiceFoundryServiceClient,
+        )
+
+        client = ServiceFoundryServiceClient(init_session=False, base_url=base_url)
+
+        self._api_server_url = client._api_server_url
+        self._auth_server_url = client.tenant_info.auth_server_url
+        self._tenant_name = client.tenant_info.tenant_name
+
+    @classmethod
+    def from_base_url(cls, base_url: str) -> "AuthServiceClient":
+        from truefoundry.deploy.lib.clients.servicefoundry_client import (
+            ServiceFoundryServiceClient,
+        )
+
+        client = ServiceFoundryServiceClient(init_session=False, base_url=base_url)
+        if client.python_sdk_config.use_sfy_server_auth_apis:
+            return ServiceFoundryServerAuthServiceClient(base_url)
+        return AuthServerServiceClient(base_url)
+
+    @abstractmethod
+    def refresh_token(self, token: Token, host: str = None) -> Token: ...
+
+    @abstractmethod
+    def get_device_code(self) -> DeviceCode: ...
+
+    @abstractmethod
+    def get_token_from_device_code(
+        self, device_code: str, timeout: float = 60, poll_interval_seconds: int = 1
+    ) -> Token: ...
+
+
+class ServiceFoundryServerAuthServiceClient(AuthServiceClient):
+    def __init__(self, base_url):
+        super().__init__(base_url)
+
+    def refresh_token(self, token: Token, host: str = None) -> Token:
+        host_arg_str = f"--host {host}" if host else "--host HOST"
+        if not token.refresh_token:
+            # TODO: Add a way to propagate error messages without traceback to the output interface side
+            raise Exception(
+                f"Unable to resume login session. Please log in again using `tfy login {host_arg_str} --relogin`"
+            )
+        url = f"{self._api_server_url}/{VERSION_PREFIX}/oauth2/token"
+        data = {
+            "tenantName": token.tenant_name,
+            "refreshToken": token.refresh_token,
+            "grantType": "refresh_token",
+            "returnJWT": True,
+        }
+        res = requests.post(url, json=data)
+        try:
+            res = request_handling(res)
+            return Token.parse_obj(res)
+        except BadRequestException as ex:
+            raise Exception(
+                f"Unable to resume login session. Please log in again using `tfy login {host_arg_str} --relogin`"
+            ) from ex
+
+    def get_device_code(self) -> DeviceCode:
+        url = f"{self._api_server_url}/{VERSION_PREFIX}/oauth2/device-authorize"
+        data = {"tenantName": self._tenant_name}
+        res = requests.post(url, json=data)
+        res = request_handling(res)
+        # TODO: temporary cleanup of incorrect attributes
+        res = {"userCode": res.get("userCode"), "deviceCode": res.get("deviceCode")}
+        return DeviceCode.parse_obj(res)
+
+    def get_token_from_device_code(
+        self, device_code: str, timeout: float = 60, poll_interval_seconds: int = 1
+    ) -> Token:
+        timeout = timeout or 60
+        poll_interval_seconds = poll_interval_seconds or 1
+        url = f"{self._api_server_url}/{VERSION_PREFIX}/oauth2/token"
+        data = {
+            "tenantName": self._tenant_name,
+            "deviceCode": device_code,
+            "grantType": "device_code",
+            "returnJWT": True,
+        }
+        response = requests.post(url=url, json=data)
+        start_time = time.monotonic()
+
+        for response in poll_for_function(
+            requests.post, poll_after_secs=poll_interval_seconds, url=url, json=data
+        ):
+            if response.status_code == 201:
+                response = response.json()
+                return Token.parse_obj(response)
+            elif response.status_code == 202:
+                logger.debug("User has not authorized yet. Checking again.")
+            else:
+                raise Exception(
+                    "Failed to get token using device code. "
+                    f"status_code {response.status_code},\n {response.text}"
+                )
+            time_elapsed = time.monotonic() - start_time
+            if time_elapsed > timeout:
+                logger.warning("Polled server for %s secs.", int(time_elapsed))
+                break
+
+        raise Exception(f"Did not get authorized within {timeout} seconds.")
+
+
+class AuthServerServiceClient(AuthServiceClient):
+    def __init__(self, base_url):
+        super().__init__(base_url)
+
+    def refresh_token(self, token: Token, host: str = None) -> Token:
+        host_arg_str = f"--host {host}" if host else "--host HOST"
+        if not token.refresh_token:
+            # TODO: Add a way to propagate error messages without traceback to the output interface side
+            raise Exception(
+                f"Unable to resume login session. Please log in again using `tfy login {host_arg_str} --relogin`"
+            )
+        url = f"{self._auth_server_url}/api/{VERSION_PREFIX}/oauth/token/refresh"
+        data = {
+            "tenantName": token.tenant_name,
+            "refreshToken": token.refresh_token,
+        }
+        res = requests.post(url, json=data)
+        try:
+            res = request_handling(res)
+            return Token.parse_obj(res)
+        except BadRequestException as ex:
+            raise Exception(
+                f"Unable to resume login session. Please log in again using `tfy login {host_arg_str} --relogin`"
+            ) from ex
+
+    def get_device_code(self) -> DeviceCode:
+        url = f"{self._auth_server_url}/api/{VERSION_PREFIX}/oauth/device"
+        data = {"tenantName": self._tenant_name}
+        res = requests.post(url, json=data)
+        res = request_handling(res)
+        return DeviceCode.parse_obj(res)
+
+    def get_token_from_device_code(
+        self, device_code: str, timeout: float = 60, poll_interval_seconds: int = 1
+    ) -> Token:
+        url = f"{self._auth_server_url}/api/{VERSION_PREFIX}/oauth/device/token"
+        data = {
+            "tenantName": self._tenant_name,
+            "deviceCode": device_code,
+        }
+        start_time = time.monotonic()
+        poll_interval_seconds = 1
+
+        for response in poll_for_function(
+            requests.post, poll_after_secs=poll_interval_seconds, url=url, json=data
+        ):
+            if response.status_code == 201:
+                response = response.json()
+                return Token.parse_obj(response)
+            elif response.status_code == 202:
+                logger.debug("User has not authorized yet. Checking again.")
+            else:
+                raise Exception(
+                    "Failed to get token using device code. "
+                    f"status_code {response.status_code},\n {response.text}"
+                )
+            time_elapsed = time.monotonic() - start_time
+            if time_elapsed > timeout:
+                logger.warning("Polled server for %s secs.", int(time_elapsed))
+                break
+
+        raise Exception(f"Did not get authorized within {timeout} seconds.")

{truefoundry-0.3.0rc9 → truefoundry-0.3.0rc11}/truefoundry/deploy/lib/auth/credential_provider.py

@@ -44,7 +44,7 @@ class EnvCredentialProvider(CredentialProvider):
         # TODO: Read host from cred file as well.
         base_url = resolve_base_url().strip("/")
         self._host = base_url
-        self._auth_service = AuthServiceClient(base_url=base_url)
+        self._auth_service = AuthServiceClient.from_base_url(base_url=base_url)
 
         servicefoundry_client = ServiceFoundryServiceClient(
             init_session=False, base_url=base_url
@@ -82,7 +82,7 @@ class FileCredentialProvider(CredentialProvider):
             self._token = self._last_cred_file_content.to_token()
             self._host = self._last_cred_file_content.host
 
-        self._auth_service = AuthServiceClient(base_url=self._host)
+        self._auth_service = AuthServiceClient.from_base_url(base_url=self._host)
 
     @staticmethod
     def can_provide() -> bool:

{truefoundry-0.3.0rc9 → truefoundry-0.3.0rc11}/truefoundry/deploy/lib/clients/servicefoundry_client.py

@@ -26,12 +26,14 @@ from truefoundry.deploy.lib.model.entity import (
     Deployment,
     DockerRegistryCredentials,
     JobRun,
+    PythonSDKConfig,
     TenantInfo,
     Token,
     TriggerJobResult,
     Workspace,
     WorkspaceResources,
 )
+from truefoundry.deploy.lib.util import timed_lru_cache
 from truefoundry.deploy.lib.win32 import allow_interrupt
 from truefoundry.deploy.v2.lib.models import (
     AppDeploymentStatusResponse,
@@ -93,6 +95,24 @@ def check_min_cli_version(fn):
     return inner
 
 
+@timed_lru_cache(seconds=30 * 60)
+def _cached_get_tenant_info(api_server_url: str) -> TenantInfo:
+    res = requests.get(
+        url=f"{api_server_url}/v1/tenant-id",
+        params={"hostName": urlparse(api_server_url).netloc},
+    )
+    res = request_handling(res)
+    return TenantInfo.parse_obj(res)
+
+
+@timed_lru_cache(seconds=30 * 60)
+def _cached_get_python_sdk_config(api_server_url: str) -> PythonSDKConfig:
+    url = f"{api_server_url}/v1/min-cli-version"
+    res = requests.get(url)
+    res = request_handling(res)
+    return PythonSDKConfig.parse_obj(res)
+
+
 class ServiceFoundryServiceClient:
     def __init__(self, init_session: bool = True, base_url: Optional[str] = None):
         self._session: Optional[ServiceFoundrySession] = None
@@ -111,20 +131,19 @@ class ServiceFoundryServiceClient:
     def base_url(self) -> str:
         return self._base_url
 
-    def get_tenant_info(self) -> TenantInfo:
-        res = requests.get(
-            url=f"{self._api_server_url}/v1/tenant-id",
-            params={"hostName": urlparse(self._api_server_url).netloc},
-        )
-        res = request_handling(res)
-        return TenantInfo.parse_obj(res)
+    @property
+    def tenant_info(self) -> TenantInfo:
+        return _cached_get_tenant_info(self._api_server_url)
+
+    @property
+    def python_sdk_config(self) -> PythonSDKConfig:
+        return _cached_get_python_sdk_config(self._api_server_url)
 
     @functools.cached_property
     def _min_cli_version_required(self) -> str:
-        url = f"{self._api_server_url}/v1/min-cli-version"
-        res = requests.get(url)
-        res = request_handling(res)
-        return res["truefoundryCliMinVersion"]
+        return _cached_get_python_sdk_config(
+            self._api_server_url
+        ).truefoundry_cli_min_version
 
     def _get_header(self):
         if not self._session:
@@ -616,6 +635,12 @@ class ServiceFoundryServiceClient:
         response = request_handling(res)
         return TriggerJobResult.parse_obj(response)
 
+    def trigger_workflow(self, application_id: str, inputs: Dict[str, Any]):
+        url = f"{self._api_server_url}/{VERSION_PREFIX}/workflow/{application_id}/executions"
+        res = requests.post(url, json=inputs, headers=self._get_header())
+        response = request_handling(res)
+        return response
+
     @check_min_cli_version
     def get_docker_registry_creds(
         self, docker_registry_fqn: str, cluster_id: str

{truefoundry-0.3.0rc9 → truefoundry-0.3.0rc11}/truefoundry/deploy/lib/dao/application.py

@@ -241,3 +241,23 @@ def terminate_job_run(
         job_run_name=job_run_name,
     )
     return response
+
+
+def trigger_workflow(application_fqn: str, inputs: Optional[Dict[str, Any]] = None):
+    inputs = inputs or {}
+    client = ServiceFoundryServiceClient()
+    _application_info = client.get_application_info_by_fqn(
+        application_fqn=application_fqn
+    )
+    application_info = client.get_application_info(
+        application_id=_application_info.applicationId
+    )
+    client.trigger_workflow(
+        application_id=application_info.id,
+        inputs=inputs,
+    )
+    logger.info(f"Started Execution for Workflow: {application_info.name}")
+    executions_page = (
+        f"{client.base_url.strip('/')}/deployments/{application_info.id}?tab=executions"
+    )
+    logger.info(f"You can check the executions at {executions_page}")

{truefoundry-0.3.0rc9 → truefoundry-0.3.0rc11}/truefoundry/deploy/lib/model/entity.py

@@ -297,6 +297,11 @@ class CredentialsFileContent(BaseModel):
 class DeviceCode(BaseModel):
     user_code: str = Field(alias="userCode")
     device_code: str = Field(alias="deviceCode")
+    verification_url: Optional[str] = Field(alias="verificationURI")
+    complete_verification_url: Optional[str] = Field(alias="verificationURIComplete")
+    expires_in_seconds: int = Field(alias="expiresInSeconds", default=60)
+    interval_in_seconds: int = Field(alias="intervalInSeconds", default=1)
+    message: Optional[str] = Field(alias="message")
 
     class Config:
         allow_population_by_field_name = True
@@ -306,6 +311,18 @@ class DeviceCode(BaseModel):
         return f"{auth_host}/authorize/device?userCode={self.user_code}"
 
 
+class PythonSDKConfig(BaseModel):
+    min_version: str = Field(alias="minVersion")
+    truefoundry_cli_min_version: str = Field(alias="truefoundryCliMinVersion")
+    use_sfy_server_auth_apis: Optional[bool] = Field(
+        alias="useSFYServerAuthAPIs", default=False
+    )
+
+    class Config:
+        allow_population_by_field_name = True
+        allow_mutation = False
+
+
 class JobRun(Base):
     name: str
     applicationName: str

{truefoundry-0.3.0rc9 → truefoundry-0.3.0rc11}/truefoundry/deploy/lib/session.py

@@ -92,7 +92,7 @@ def login(
                 api_key=api_key, servicefoundry_client=servicefoundry_client
             )
         else:
-            auth_service = AuthServiceClient(base_url=host)
+            auth_service = AuthServiceClient.from_base_url(base_url=host)
             # interactive login
             token = _login_with_device_code(base_url=host, auth_service=auth_service)
 
@@ -136,11 +136,23 @@ def _login_with_device_code(
 ) -> Token:
     logger.debug("Logging in with device code")
     device_code = auth_service.get_device_code()
-    url_to_go = device_code.get_user_clickable_url(auth_host=base_url)
-    output_hook.print_line(f"Opening:- {url_to_go}")
-    output_hook.print(
-        "Please click on the above link if it is not "
-        "automatically opened in a browser window."
+    auto_open_url = None
+    message = "Please click on the above link if it is not automatically opened in a browser window."
+    if device_code.complete_verification_url:
+        auto_open_url = device_code.complete_verification_url
+    elif device_code.verification_url:
+        if device_code.message:
+            message = device_code.message
+        else:
+            message = f"Please open the following URL in a browser and enter the code {device_code.user_code} when prompted: {device_code.verification_url}"
+    else:
+        auto_open_url = device_code.get_user_clickable_url(auth_host=base_url)
+    output_hook.print_line(message)
+    if auto_open_url:
+        output_hook.print_line(f"Opening:- {auto_open_url}")
+        click.launch(auto_open_url)
+    return auth_service.get_token_from_device_code(
+        device_code=device_code.device_code,
+        timeout=device_code.expires_in_seconds,
+        poll_interval_seconds=device_code.interval_in_seconds,
     )
-    click.launch(url_to_go)
-    return auth_service.get_token_from_device_code(device_code=device_code.device_code)

{truefoundry-0.3.0rc9 → truefoundry-0.3.0rc11}/truefoundry/deploy/lib/util.py

@@ -1,5 +1,7 @@
 import os
 import re
+from functools import lru_cache, wraps
+from time import monotonic_ns
 from typing import Union
 
 from truefoundry.deploy.lib.const import (
@@ -68,3 +70,21 @@ def find_list_paths(data, parent_key="", sep="."):
             new_key = f"{parent_key}[{i}]"
             list_paths.extend(find_list_paths(value, new_key, sep))
     return list_paths
+
+
+def timed_lru_cache(seconds: int = 300, maxsize: int = None):
+    def wrapper_cache(func):
+        func = lru_cache(maxsize=maxsize)(func)
+        func.delta = seconds * 10**9
+        func.expiration = monotonic_ns() + func.delta
+
+        @wraps(func)
+        def wrapped_func(*args, **kwargs):
+            if monotonic_ns() >= func.expiration:
+                func.cache_clear()
+                func.expiration = monotonic_ns() + func.delta
+            return func(*args, **kwargs)
+
+        return wrapped_func
+
+    return wrapper_cache

truefoundry-0.3.0rc9/truefoundry/deploy/lib/auth/auth_service_client.py

@@ -1,81 +0,0 @@
-import time
-
-import requests
-
-from truefoundry.deploy.lib.clients.utils import poll_for_function, request_handling
-from truefoundry.deploy.lib.const import VERSION_PREFIX
-from truefoundry.deploy.lib.exceptions import BadRequestException
-from truefoundry.deploy.lib.model.entity import DeviceCode, Token
-from truefoundry.logger import logger
-
-
-class AuthServiceClient:
-    def __init__(self, base_url):
-        from truefoundry.deploy.lib.clients.servicefoundry_client import (
-            ServiceFoundryServiceClient,
-        )
-
-        client = ServiceFoundryServiceClient(init_session=False, base_url=base_url)
-        tenant_info = client.get_tenant_info()
-
-        self._auth_server_url = tenant_info.auth_server_url
-        self._tenant_name = tenant_info.tenant_name
-
-    def refresh_token(self, token: Token, host: str = None) -> Token:
-        host_arg_str = f"--host {host}" if host else "--host HOST"
-        if not token.refresh_token:
-            # TODO: Add a way to propagate error messages without traceback to the output interface side
-            raise Exception(
-                f"Unable to resume login session. Please log in again using `tfy login {host_arg_str} --relogin`"
-            )
-        url = f"{self._auth_server_url}/api/{VERSION_PREFIX}/oauth/token/refresh"
-        data = {
-            "tenantName": token.tenant_name,
-            "refreshToken": token.refresh_token,
-        }
-        res = requests.post(url, data=data)
-        try:
-            res = request_handling(res)
-            return Token.parse_obj(res)
-        except BadRequestException as ex:
-            raise Exception(
-                f"Unable to resume login session. Please log in again using `tfy login {host_arg_str} --relogin`"
-            ) from ex
-
-    def get_device_code(self) -> DeviceCode:
-        url = f"{self._auth_server_url}/api/{VERSION_PREFIX}/oauth/device"
-        data = {"tenantName": self._tenant_name}
-        res = requests.post(url, data=data)
-        res = request_handling(res)
-        return DeviceCode.parse_obj(res)
-
-    def get_token_from_device_code(
-        self, device_code: str, timeout: float = 60
-    ) -> Token:
-        url = f"{self._auth_server_url}/api/{VERSION_PREFIX}/oauth/device/token"
-        data = {
-            "tenantName": self._tenant_name,
-            "deviceCode": device_code,
-        }
-        start_time = time.monotonic()
-        poll_interval_seconds = 1
-
-        for response in poll_for_function(
-            requests.post, poll_after_secs=poll_interval_seconds, url=url, data=data
-        ):
-            if response.status_code == 201:
-                response = response.json()
-                return Token.parse_obj(response)
-            elif response.status_code == 202:
-                logger.debug("User has not authorized yet. Checking again.")
-            else:
-                raise Exception(
-                    "Failed to get token using device code. "
-                    f"status_code {response.status_code},\n {response.text}"
-                )
-            time_elapsed = time.monotonic() - start_time
-            if time_elapsed > timeout:
-                logger.warning("Polled server for %s secs.", int(time_elapsed))
-                break
-
-        raise Exception(f"Did not get authorized within {timeout} seconds.")