tinybird 0.0.1.dev133__py3-none-any.whl → 0.0.1.dev134__py3-none-any.whl

tinybird/tb/__cli__.py

@@ -4,5 +4,5 @@ __description__ = 'Tinybird Command Line Tool'
 __url__ = 'https://www.tinybird.co/docs/cli/introduction.html'
 __author__ = 'Tinybird'
 __author_email__ = 'support@tinybird.co'
-__version__ = '0.0.1.dev133'
-__revision__ = 'aa46678'
+__version__ = '0.0.1.dev134'
+__revision__ = '921269a'

tinybird/tb/config.py

@@ -42,7 +42,7 @@ CLOUD_HOSTS = {
 }
 
 
-async def get_config(
+def get_config(
     host: str, token: Optional[str], semver: Optional[str] = None, config_file: Optional[str] = None
 ) -> Dict[str, Any]:
     if host:
@@ -50,8 +50,8 @@ async def get_config(
 
     config = {}
     try:
-        async with aiofiles.open(config_file or Path(getcwd()) / ".tinyb") as file:
-            res = await file.read()
+        with open(config_file or Path(getcwd()) / ".tinyb") as file:
+            res = file.read()
             config = json.loads(res)
     except OSError:
         pass

tinybird/tb/modules/build.py

@@ -1,7 +1,6 @@
 import asyncio
 import json
 import logging
-import sys
 import threading
 import time
 from copy import deepcopy
@@ -16,7 +15,7 @@ import requests
 import tinybird.context as context
 from tinybird.tb.client import TinyB
 from tinybird.tb.modules.cli import cli
-from tinybird.tb.modules.common import push_data
+from tinybird.tb.modules.common import push_data, sys_exit
 from tinybird.tb.modules.config import CLIConfig
 from tinybird.tb.modules.datafile.build import folder_build
 from tinybird.tb.modules.datafile.exceptions import ParseException
@@ -99,12 +98,10 @@ def build_project(project: Project, tb_client: TinyB, file_changed: Optional[str
         except Exception as e:
             logging.debug(e, exc_info=True)
             click.echo(FeedbackManager.error(message="Couldn't parse response from server"))
-            sys.exit(1)
+            sys_exit("build_error", str(e))
 
         logging.debug(json.dumps(result, indent=2))
 
-        output_pre_formatted = False
-
         build_result = result.get("result")
         if build_result == "success":
             build = result.get("build")
@@ -174,8 +171,7 @@ def build_project(project: Project, tb_client: TinyB, file_changed: Optional[str
             build_errors = result.get("errors")
             full_error_msg = ""
             for build_error in build_errors:
-                output_pre_formatted = True
-                filename_bit = FeedbackManager.build_result_error_title(message=f"{build_error.get('filename', '')}")
+                filename_bit = build_error.get("filename", "")
                 error_bit = build_error.get("error") or build_error.get("message") or ""
                 error_msg = ((filename_bit + "\n") if filename_bit else "") + error_bit
                 full_error_msg += error_msg + "\n\n"
@@ -187,11 +183,9 @@ def build_project(project: Project, tb_client: TinyB, file_changed: Optional[str
     finally:
         for fd in fds:
             fd.close()
-        if error:
-            if not output_pre_formatted:
-                raise click.ClickException(error)
-            click.echo(error)
-            raise click.ClickException("")  # Needed so the caller shows the right message
+
+    if error:
+        raise click.ClickException(error)
 
 
 def append_fixture(
@@ -270,6 +264,7 @@ def process(
 ) -> None:
     time_start = time.time()
     build_failed = False
+    build_error: Optional[str] = None
     if file_changed and (file_changed.endswith(FixtureExtension.NDJSON) or file_changed.endswith(FixtureExtension.CSV)):
         rebuild_fixture(project, tb_client, file_changed)
     elif file_changed and file_changed.endswith(".sql"):
@@ -278,7 +273,8 @@ def process(
         try:
             build_project(project, tb_client, file_changed, silent)
         except click.ClickException as e:
-            click.echo(FeedbackManager.error(message=e))
+            click.echo(FeedbackManager.info(message=str(e)))
+            build_error = str(e)
             build_failed = True
         try:
             if file_changed and not build_failed:
@@ -294,7 +290,7 @@ def process(
     if build_failed:
         click.echo(FeedbackManager.error(message=f"✗ {rebuild_str} failed"))
         if not watch:
-            sys.exit(1)
+            sys_exit("build_error", build_error or "Unknown error")
     else:
         if not silent:
             click.echo(FeedbackManager.success(message=f"\n✓ {rebuild_str} completed in {elapsed_time:.1f}s"))

tinybird/tb/modules/common.py

@@ -250,6 +250,7 @@ line experience. To opt-out, set TB_CLI_TELEMETRY_OPTOUT to '1' or 'true'."""
 
     def __call__(self, *args, **kwargs) -> None:
         error_msg: Optional[str] = None
+        silent_error_msg: Optional[str] = None
         error_event: str = "error"
 
         exit_code: int = 0
@@ -265,14 +266,19 @@ line experience. To opt-out, set TB_CLI_TELEMETRY_OPTOUT to '1' or 'true'."""
             error_event = "auth_error"
             exit_code = 1
         except SystemExit as ex:
-            exit_code = int(ex.code) if ex.code else 0
+            if isinstance(ex.code, str):
+                exit_code = 1
+                error_event, silent_error_msg = get_error_event(ex.code)
+            else:
+                exit_code = ex.code or 0
         except Exception as ex:
             error_msg = str(ex)
             exit_code = 1
 
-        if error_msg:
-            click.echo(error_msg)
-            add_telemetry_event(error_event, error=error_msg)
+        if error_msg or silent_error_msg:
+            if error_msg:
+                click.echo(error_msg)
+            add_telemetry_event(error_event, error=error_msg or silent_error_msg)
         flush_telemetry(wait=True)
 
         sys.exit(exit_code)
@@ -2232,3 +2238,20 @@ async def ask_for_organization(
             else:
                 return None, None
     return organization_id, organization_name
+
+
+event_error_separator = "__error__"
+
+
+def sys_exit(event: str, msg: str) -> None:
+    sys.exit(f"{event}{event_error_separator}{msg}")
+
+
+def get_error_event(error: str) -> Tuple[str, str]:
+    try:
+        error_event = error.split(event_error_separator)[0]
+        silent_error_msg = error.split(event_error_separator)[1]
+    except Exception:
+        error_event = "error"
+        silent_error_msg = "Unknown error"
+    return error_event, silent_error_msg

tinybird/tb/modules/create.py

@@ -12,7 +12,7 @@ from tinybird.tb.modules.cli import cli
 from tinybird.tb.modules.common import _generate_datafile, coro, generate_datafile
 from tinybird.tb.modules.config import CLIConfig
 from tinybird.tb.modules.datafile.fixture import persist_fixture
-from tinybird.tb.modules.exceptions import CLIException
+from tinybird.tb.modules.exceptions import CLICreateException
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 from tinybird.tb.modules.llm import LLM
 from tinybird.tb.modules.llm_utils import extract_xml, parse_xml
@@ -70,17 +70,9 @@ async def create(
         user_token: Optional[str] = None
         created_something = False
         if prompt:
-            try:
-                user_token = ctx_config.get("user_token")
-                if not user_token:
-                    raise CLIException("No user token found")
-            except Exception as e:
-                click.echo(
-                    FeedbackManager.error(
-                        message=f"This action requires authentication. Run 'tb login' first. Error: {e}"
-                    )
-                )
-                return
+            user_token = ctx_config.get("user_token")
+            if not user_token:
+                raise Exception("This action requires authentication. Run 'tb login' first.")
 
         if not validate_project_structure(folder):
             click.echo(FeedbackManager.highlight(message="\n» Creating new project structure..."))
@@ -164,7 +156,7 @@ async def create(
         if not created_something:
             click.echo(FeedbackManager.warning(message="△ No resources created\n"))
     except Exception as e:
-        click.echo(FeedbackManager.error(message=f"Error: {str(e)}"))
+        raise CLICreateException(FeedbackManager.error(message=str(e)))
 
 
 PROJECT_PATHS = ("datasources", "endpoints", "materializations", "copies", "pipes", "fixtures", "tests", "connections")
@@ -328,7 +320,7 @@ def init_git(folder: str):
 
         click.echo(FeedbackManager.info_file_created(file=".gitignore"))
     except Exception as e:
-        raise CLIException(f"Error initializing Git: {e}")
+        raise Exception(f"Error initializing Git: {e}")
 
 
 def generate_pipe_file(name: str, content: str, folder: str) -> Path:

tinybird/tb/modules/deployment.py

@@ -10,7 +10,11 @@ import click
 import requests
 
 from tinybird.tb.modules.cli import cli
-from tinybird.tb.modules.common import echo_safe_humanfriendly_tables_format_smart_table, get_display_cloud_host
+from tinybird.tb.modules.common import (
+    echo_safe_humanfriendly_tables_format_smart_table,
+    get_display_cloud_host,
+    sys_exit,
+)
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 from tinybird.tb.modules.project import Project
 
@@ -27,10 +31,12 @@ def api_fetch(url: str, headers: dict) -> dict:
         error = result.get("error")
         logging.debug(json.dumps(result, indent=2))
         click.echo(FeedbackManager.error(message=f"Error: {error}"))
-        sys.exit(1)
+        sys_exit("deployment_error", error)
     except Exception:
-        click.echo(FeedbackManager.error(message="Error parsing response from API"))
-        sys.exit(1)
+        message = "Error parsing response from API"
+        click.echo(FeedbackManager.error(message=message))
+        sys_exit("deployment_error", message)
+    return {}
 
 
 def api_post(
@@ -50,11 +56,13 @@ def api_post(
         error = result.get("error")
         if error:
             click.echo(FeedbackManager.error(message=f"Error: {error}"))
-            sys.exit(1)
+            sys_exit("deployment_error", error)
         return result
     except Exception:
-        click.echo(FeedbackManager.error(message="Error parsing response from API"))
-        sys.exit(1)
+        message = "Error parsing response from API"
+        click.echo(FeedbackManager.error(message=message))
+        sys_exit("deployment_error", message)
+    return {}
 
 
 # TODO(eclbg): This logic should be in the server, and there should be a dedicated endpoint for promoting a deployment
@@ -65,12 +73,16 @@ def promote_deployment(host: Optional[str], headers: dict, wait: bool) -> None:
 
     deployments = result.get("deployments")
     if not deployments:
-        click.echo(FeedbackManager.error(message="No deployments found"))
-        sys.exit(1)
+        message = "No deployments found"
+        click.echo(FeedbackManager.error(message=message))
+        sys_exit("deployment_error", message)
+        return
 
     if len(deployments) < 2:
-        click.echo(FeedbackManager.error(message="Only one deployment found"))
-        sys.exit(1)
+        message = "Only one deployment found"
+        click.echo(FeedbackManager.error(message=message))
+        sys_exit("deployment_error", message)
+        return
 
     last_deployment, candidate_deployment = deployments[0], deployments[1]
 
@@ -79,17 +91,16 @@ def promote_deployment(host: Optional[str], headers: dict, wait: bool) -> None:
         deploy_errors = candidate_deployment.get("errors", [])
         for deploy_error in deploy_errors:
             click.echo(FeedbackManager.error(message=f"* {deploy_error}"))
-        sys.exit(1)
+        sys_exit("deployment_error", "Current deployment is not ready: " + str(deploy_errors))
+        return
 
     if candidate_deployment.get("live"):
         click.echo(FeedbackManager.error(message="Candidate deployment is already live"))
     else:
-        click.echo(FeedbackManager.success(message="Setting candidate deployment as live"))
-
         TINYBIRD_API_URL = f"{host}/v1/deployments/{candidate_deployment.get('id')}/set-live"
         result = api_post(TINYBIRD_API_URL, headers=headers)
 
-    click.echo(FeedbackManager.success(message="Removing old deployment"))
+    click.echo(FeedbackManager.highlight(message="» Removing old deployment"))
 
     TINYBIRD_API_URL = f"{host}/v1/deployments/{last_deployment.get('id')}"
     r = requests.delete(TINYBIRD_API_URL, headers=headers)
@@ -97,9 +108,10 @@ def promote_deployment(host: Optional[str], headers: dict, wait: bool) -> None:
     logging.debug(json.dumps(result, indent=2))
     if result.get("error"):
         click.echo(FeedbackManager.error(message=result.get("error")))
-        sys.exit(1)
+        sys_exit("deployment_error", result.get("error", "Unknown error"))
+    click.echo(FeedbackManager.info(message="✓ Old deployment removed"))
 
-    click.echo(FeedbackManager.success(message="Deployment promotion successfully started"))
+    click.echo(FeedbackManager.highlight(message="» Waiting for deployment to be promoted..."))
 
     if wait:
         while True:
@@ -108,7 +120,7 @@ def promote_deployment(host: Optional[str], headers: dict, wait: bool) -> None:
 
             last_deployment = result.get("deployment")
             if last_deployment.get("status") == "deleted":
-                click.echo(FeedbackManager.success(message="Deployment promoted successfully"))
+                click.echo(FeedbackManager.success(message="✓ Deployment is live!"))
                 break
 
             time.sleep(5)
@@ -154,7 +166,7 @@ def discard_deployment(host: Optional[str], headers: dict, wait: bool) -> None:
     logging.debug(json.dumps(result, indent=2))
     if result.get("error"):
         click.echo(FeedbackManager.error(message=result.get("error")))
-        sys.exit(1)
+        sys_exit("deployment_error", result.get("error", "Unknown error"))
 
     click.echo(FeedbackManager.success(message="Discard process successfully started"))
 
@@ -404,7 +416,7 @@ def create_deployment(
                 sys.exit(0)
 
             click.echo(FeedbackManager.error(message="\n✗ Deployment is not valid"))
-            sys.exit(1)
+            sys_exit("deployment_error", "Deployment is not valid: " + str(deployment.get("errors")))
 
         status = result.get("result")
         if status == "success":
@@ -415,15 +427,15 @@ def create_deployment(
             )
 
             if wait:
-                click.echo(FeedbackManager.info(message="\n✓ Deployment submitted successfully"))
+                click.echo(FeedbackManager.info(message="\n* Deployment submitted"))
             else:
                 click.echo(FeedbackManager.success(message="\n✓ Deployment submitted successfully"))
         elif status == "no_changes":
-            click.echo(FeedbackManager.highlight(message="Not deploying. No changes."))
+            click.echo(FeedbackManager.warning(message="△ Not deploying. No changes."))
             sys.exit(0)
         elif status == "failed":
             click.echo(FeedbackManager.error(message="Deployment failed"))
-            sys.exit(1)
+            sys_exit("deployment_error", "Deployment failed" + str(deployment.get("errors")))
         else:
             click.echo(FeedbackManager.error(message=f"Unknown deployment result {status}"))
     except Exception as e:
@@ -433,16 +445,17 @@ def create_deployment(
             fd.close()
 
         if not deployment and not check:
-            sys.exit(1)
+            sys_exit("deployment_error", "Deployment failed")
 
     if deployment and wait and not check:
+        click.echo(FeedbackManager.highlight(message="» Waiting for deployment to be ready..."))
         while True:
             url = f"{client.host}/v1/deployments/{deployment.get('id')}"
             res = api_fetch(url, HEADERS)
             deployment = res.get("deployment")
             if not deployment:
                 click.echo(FeedbackManager.error(message="Error parsing deployment from response"))
-                sys.exit(1)
+                sys_exit("deployment_error", "Error parsing deployment from response")
             if deployment.get("status") == "failed":
                 click.echo(FeedbackManager.error(message="Deployment failed"))
                 deploy_errors = deployment.get("errors")
@@ -452,14 +465,14 @@ def create_deployment(
                 if auto:
                     click.echo(FeedbackManager.error(message="Rolling back deployment"))
                     discard_deployment(client.host, HEADERS, wait=wait)
-                sys.exit(1)
+                sys_exit("deployment_error", "Deployment failed: " + str(deployment.get("errors")))
 
             if deployment.get("status") == "data_ready":
                 break
 
             time.sleep(5)
 
-        click.echo(FeedbackManager.success(message="Deployment is ready"))
+        click.echo(FeedbackManager.info(message="✓ Deployment is ready"))
 
         if auto:
             promote_deployment(client.host, HEADERS, wait=wait)
@@ -513,12 +526,12 @@ def print_changes(result: dict, project: Project) -> None:
         tokens.append((change_type, token_name, "\n".join(added_perms), "\n".join(removed_perms)))
 
     if resources:
-        click.echo(FeedbackManager.highlight(message="\n» Changes to be deployed...\n"))
+        click.echo(FeedbackManager.info(message="\n* Changes to be deployed:"))
         echo_safe_humanfriendly_tables_format_smart_table(resources, column_names=resources_columns)
     else:
-        click.echo(FeedbackManager.highlight(message="\n» No changes to be deployed\n"))
+        click.echo(FeedbackManager.info(message="\n* No changes to be deployed"))
     if tokens:
-        click.echo(FeedbackManager.highlight(message="\n» Changes in tokens to be deployed...\n"))
+        click.echo(FeedbackManager.info(message="\n* Changes in tokens to be deployed:"))
         echo_safe_humanfriendly_tables_format_smart_table(tokens, column_names=tokens_columns)
     else:
-        click.echo(FeedbackManager.highlight(message="\n» No changes in tokens to be deployed\n"))
+        click.echo(FeedbackManager.info(message="* No changes in tokens to be deployed"))

tinybird/tb/modules/exceptions.py

@@ -15,7 +15,8 @@ class CLIException(click.exceptions.ClickException):
 
     def __init__(self, message: str, telemetry_event: Optional[str] = None, **kw_telemetry_event_data: Any) -> None:
         telemetry_event = telemetry_event or "error"
-        data: Dict[str, Any] = {"error": message}
+        message_without_color = message.replace("\033[91m", "").replace("\033[0m", "")
+        data: Dict[str, Any] = {"error": message_without_color}
         data.update(kw_telemetry_event_data)
         add_telemetry_event(telemetry_event, **data)
         super().__init__(message)
@@ -89,3 +90,52 @@ class CLITokenException(CLIException):
 
     def __init__(self, message: str, **kw_telemetry_event_data: Any) -> None:
         super().__init__(message, "token_error", **kw_telemetry_event_data)
+
+
+class CLIBuildException(CLIException):
+    """Exceptions generated by the build commands"""
+
+    def __init__(self, message: str, **kw_telemetry_event_data: Any) -> None:
+        super().__init__(message, "build_error", **kw_telemetry_event_data)
+
+
+class CLIDeploymentException(CLIException):
+    """Exceptions generated by the deployment commands"""
+
+    def __init__(self, message: str, **kw_telemetry_event_data: Any) -> None:
+        super().__init__(message, "deployment_error", **kw_telemetry_event_data)
+
+
+class CLITestException(CLIException):
+    """Exceptions generated by the test commands"""
+
+    def __init__(self, message: str, **kw_telemetry_event_data: Any) -> None:
+        super().__init__(message, "test_error", **kw_telemetry_event_data)
+
+
+class CLICreateException(CLIException):
+    """Exceptions generated by the test commands"""
+
+    def __init__(self, message: str, **kw_telemetry_event_data: Any) -> None:
+        super().__init__(message, "create_error", **kw_telemetry_event_data)
+
+
+class CLIMockException(CLIException):
+    """Exceptions generated by the mock commands"""
+
+    def __init__(self, message: str, **kw_telemetry_event_data: Any) -> None:
+        super().__init__(message, "mock_error", **kw_telemetry_event_data)
+
+
+class CLILoginException(CLIException):
+    """Exceptions generated by the login commands"""
+
+    def __init__(self, message: str, **kw_telemetry_event_data: Any) -> None:
+        super().__init__(message, "login_error", **kw_telemetry_event_data)
+
+
+class CLILocalException(CLIException):
+    """Exceptions generated by the local commands"""
+
+    def __init__(self, message: str, **kw_telemetry_event_data: Any) -> None:
+        super().__init__(message, "local_error", **kw_telemetry_event_data)

tinybird/tb/modules/local.py

@@ -11,7 +11,7 @@ from docker.client import DockerClient
 from docker.models.containers import Container
 from tinybird.tb.modules.cli import cli
 from tinybird.tb.modules.common import coro
-from tinybird.tb.modules.exceptions import CLIException
+from tinybird.tb.modules.exceptions import CLIException, CLILocalException
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 from tinybird.tb.modules.local_common import TB_CONTAINER_NAME, TB_IMAGE_NAME, TB_LOCAL_PORT
 
@@ -71,7 +71,11 @@ def start_tinybird_local(
         if health == "healthy":
             break
         if health == "unhealthy":
-            raise CLIException("Tinybird Local is unhealthy. Try running `tb local restart` in a few seconds.")
+            raise CLILocalException(
+                FeedbackManager.error(
+                    message="Tinybird Local is unhealthy. Try running `tb local restart` in a few seconds."
+                )
+            )
 
         time.sleep(5)
 
@@ -125,8 +129,10 @@ def get_docker_client() -> DockerClient:
         client.ping()
         return client
     except Exception:
-        raise CLIException(
-            "No container runtime is running. Make sure a Docker-compatible runtime is installed and running."
+        raise CLILocalException(
+            FeedbackManager.error(
+                message="No container runtime is running. Make sure a Docker-compatible runtime is installed and running."
+            )
         )
 
 

tinybird/tb/modules/local_common.py

@@ -7,7 +7,8 @@ import requests
 
 from tinybird.tb.client import AuthNoTokenException, TinyB
 from tinybird.tb.modules.config import CLIConfig
-from tinybird.tb.modules.exceptions import CLIException
+from tinybird.tb.modules.exceptions import CLILocalException
+from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 TB_IMAGE_NAME = "tinybirdco/tinybird-local:latest"
 TB_CONTAINER_NAME = "tinybird-local"
@@ -34,7 +35,9 @@ async def get_tinybird_local_config(config_obj: Dict[str, Any], build: bool = Fa
         # ruff: noqa: ASYNC210
         tokens = requests.get(f"{TB_LOCAL_HOST}/tokens").json()
     except Exception:
-        raise CLIException("Tinybird local is not running. Please run `tb local start` first.")
+        raise CLILocalException(
+            FeedbackManager.error(message="Tinybird local is not running. Please run `tb local start` first.")
+        )
 
     user_token = tokens["user_token"]
     admin_token = tokens["admin_token"]

tinybird/tb/modules/login.py

@@ -13,7 +13,7 @@ import requests
 
 from tinybird.tb.modules.cli import CLIConfig, cli
 from tinybird.tb.modules.common import coro
-from tinybird.tb.modules.exceptions import CLIAuthException
+from tinybird.tb.modules.exceptions import CLILoginException
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 
@@ -115,69 +115,74 @@ def start_server(auth_callback, auth_host):
 @coro
 async def login(host: str, auth_host: str, workspace: str):
     """Authenticate using the browser."""
-    host = host.rstrip("/")
-    auth_host = auth_host.rstrip("/")
-
-    auth_event = threading.Event()
-    auth_code: list[str] = []  # Using a list to store the code, as it's mutable
-
-    def auth_callback(code):
-        auth_code.append(code)
-        auth_event.set()
-
-    click.echo(FeedbackManager.highlight(message="» Opening browser for authentication..."))
-
-    # Start the local server in a separate thread
-    server_thread = threading.Thread(target=start_server, args=(auth_callback, auth_host))
-    server_thread.daemon = True
-    server_thread.start()
-
-    # Open the browser to the auth page
-    callback_url = f"http://localhost:{AUTH_SERVER_PORT}"
-    params = {
-        "redirect_uri": callback_url,
-        "apiHost": host,
-    }
-
-    if workspace:
-        params["workspace"] = workspace
-
-    auth_url = f"{auth_host}/api/cli-login?{urlencode(params)}"
-    webbrowser.open(auth_url)
-
-    # Wait for the authentication to complete or timeout
-    if auth_event.wait(timeout=60):  # Wait for up to 60 seconds
-        params = {}
-        params["code"] = auth_code[0]
-        response = requests.get(  # noqa: ASYNC210
-            f"{auth_host}/api/cli-login?{urlencode(params)}",
-        )
-
-        data = response.json()
-        cli_config = CLIConfig.get_project_config()
-        cli_config.set_token(data.get("workspace_token", ""))
-        cli_config.set_token_for_host(data.get("workspace_token", ""), host)
-        cli_config.set_user_token(data.get("user_token", ""))
-        host = data.get("api_host", host)
-        cli_config.set_host(host)
-        ws = await cli_config.get_client(token=data.get("workspace_token", ""), host=host).workspace_info(version="v1")
-        for k in ("id", "name", "user_email", "user_id", "scope"):
-            if k in ws:
-                cli_config[k] = ws[k]
-
-        path = os.path.join(os.getcwd(), ".tinyb")
-        cli_config.persist_to_file(override_with_path=path)
-
-        auth_info: Dict[str, Any] = await cli_config.get_user_client().check_auth_login()
-        if not auth_info.get("is_valid", False):
-            raise CLIAuthException(FeedbackManager.error_auth_login_not_valid(host=cli_config.get_host()))
-
-        if not auth_info.get("is_user", False):
-            raise CLIAuthException(FeedbackManager.error_auth_login_not_user(host=cli_config.get_host()))
-
-        click.echo(FeedbackManager.gray(message="\nWorkspace: ") + FeedbackManager.info(message=ws["name"]))
-        click.echo(FeedbackManager.gray(message="User: ") + FeedbackManager.info(message=ws["user_email"]))
-        click.echo(FeedbackManager.gray(message="Host: ") + FeedbackManager.info(message=host))
-        click.echo(FeedbackManager.success(message="\n✓ Authentication successful!"))
-    else:
-        click.echo(FeedbackManager.error(message="Authentication failed or timed out."))
+    try:
+        host = host.rstrip("/")
+        auth_host = auth_host.rstrip("/")
+
+        auth_event = threading.Event()
+        auth_code: list[str] = []  # Using a list to store the code, as it's mutable
+
+        def auth_callback(code):
+            auth_code.append(code)
+            auth_event.set()
+
+        click.echo(FeedbackManager.highlight(message="» Opening browser for authentication..."))
+
+        # Start the local server in a separate thread
+        server_thread = threading.Thread(target=start_server, args=(auth_callback, auth_host))
+        server_thread.daemon = True
+        server_thread.start()
+
+        # Open the browser to the auth page
+        callback_url = f"http://localhost:{AUTH_SERVER_PORT}"
+        params = {
+            "redirect_uri": callback_url,
+            "apiHost": host,
+        }
+
+        if workspace:
+            params["workspace"] = workspace
+
+        auth_url = f"{auth_host}/api/cli-login?{urlencode(params)}"
+        webbrowser.open(auth_url)
+
+        # Wait for the authentication to complete or timeout
+        if auth_event.wait(timeout=60):  # Wait for up to 60 seconds
+            params = {}
+            params["code"] = auth_code[0]
+            response = requests.get(  # noqa: ASYNC210
+                f"{auth_host}/api/cli-login?{urlencode(params)}",
+            )
+
+            data = response.json()
+            cli_config = CLIConfig.get_project_config()
+            cli_config.set_token(data.get("workspace_token", ""))
+            cli_config.set_token_for_host(data.get("workspace_token", ""), host)
+            cli_config.set_user_token(data.get("user_token", ""))
+            host = data.get("api_host", host)
+            cli_config.set_host(host)
+            ws = await cli_config.get_client(token=data.get("workspace_token", ""), host=host).workspace_info(
+                version="v1"
+            )
+            for k in ("id", "name", "user_email", "user_id", "scope"):
+                if k in ws:
+                    cli_config[k] = ws[k]
+
+            path = os.path.join(os.getcwd(), ".tinyb")
+            cli_config.persist_to_file(override_with_path=path)
+
+            auth_info: Dict[str, Any] = await cli_config.get_user_client().check_auth_login()
+            if not auth_info.get("is_valid", False):
+                raise Exception(FeedbackManager.error_auth_login_not_valid(host=cli_config.get_host()))
+
+            if not auth_info.get("is_user", False):
+                raise Exception(FeedbackManager.error_auth_login_not_user(host=cli_config.get_host()))
+
+            click.echo(FeedbackManager.gray(message="\nWorkspace: ") + FeedbackManager.info(message=ws["name"]))
+            click.echo(FeedbackManager.gray(message="User: ") + FeedbackManager.info(message=ws["user_email"]))
+            click.echo(FeedbackManager.gray(message="Host: ") + FeedbackManager.info(message=host))
+            click.echo(FeedbackManager.success(message="\n✓ Authentication successful!"))
+        else:
+            raise Exception("Authentication failed or timed out.")
+    except Exception as e:
+        raise CLILoginException(FeedbackManager.error(message=str(e)))

tinybird/tb/modules/mock.py

@@ -6,9 +6,10 @@ import click
 from tinybird.prompts import mock_prompt
 from tinybird.tb.client import TinyB
 from tinybird.tb.modules.cli import cli
-from tinybird.tb.modules.common import CLIException, coro, push_data
+from tinybird.tb.modules.common import coro, push_data
 from tinybird.tb.modules.config import CLIConfig
 from tinybird.tb.modules.datafile.fixture import persist_fixture, persist_fixture_sql
+from tinybird.tb.modules.exceptions import CLIMockException
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 from tinybird.tb.modules.llm import LLM
 from tinybird.tb.modules.llm_utils import extract_xml
@@ -60,19 +61,15 @@ async def mock(ctx: click.Context, datasource: str, rows: int, prompt: str, form
                 datasource_path = Path(datasource_from_glob[0])
 
         if not datasource_path.exists():
-            raise CLIException(f"Datasource '{datasource_path.stem}' not found")
+            raise Exception(f"Datasource '{datasource_path.stem}' not fosund")
 
         datasource_content = datasource_path.read_text()
         config = CLIConfig.get_project_config()
         user_client = config.get_client(token=ctx_config.get("token"), host=ctx_config.get("host"))
         user_token = ctx_config.get("user_token")
 
-        try:
-            if not user_token:
-                raise CLIException("No user token found")
-        except Exception:
-            click.echo(FeedbackManager.error(message="This action requires authentication. Run 'tb login' first."))
-            return
+        if not user_token:
+            raise Exception("This action requires authentication. Run 'tb login' first.")
 
         llm = LLM(user_token=user_token, host=user_client.host)
         prompt = f"<datasource_schema>{datasource_content}</datasource_schema>\n<user_input>{prompt}</user_input>"
@@ -92,7 +89,7 @@ async def mock(ctx: click.Context, datasource: str, rows: int, prompt: str, form
                     data = result.get("data", [])[:rows]
                     error_response = result.get("error", None)
                     if error_response:
-                        raise CLIException(error_response)
+                        raise Exception(error_response)
                 else:
                     data = result
                 break
@@ -112,7 +109,7 @@ async def mock(ctx: click.Context, datasource: str, rows: int, prompt: str, form
         click.echo(FeedbackManager.success(message=f"✓ Sample data for {datasource_name} created with {rows} rows"))
 
     except Exception as e:
-        click.echo(FeedbackManager.error_exception(error=f"Error: {e}"))
+        raise CLIMockException(FeedbackManager.error(message=str(e)))
 
 
 async def append_fixture(

tinybird/tb/modules/telemetry.py

@@ -14,7 +14,6 @@ from urllib.parse import urlencode
 
 import requests
 
-from tinybird.syncasync import async_to_sync
 from tinybird.tb.config import CURRENT_VERSION, get_config
 
 TELEMETRY_TIMEOUT: int = 1
@@ -120,7 +119,7 @@ class TelemetryHelper:
 
         event_data["fingerprint_id"] = hashlib.sha256(platform.node().encode()).hexdigest()
 
-        config = async_to_sync(get_config)(None, None)
+        config = get_config(host="", token=None)
         if config:
             event_data["workspace_id"] = config.get("workspace_id", "")
             event_data["workspace_name"] = config.get("workspace_name", "")

tinybird/tb/modules/test.py

@@ -6,7 +6,6 @@
 import asyncio
 import difflib
 import glob
-import os
 import sys
 import urllib.parse
 from pathlib import Path
@@ -21,7 +20,7 @@ from tinybird.tb.client import TinyB
 from tinybird.tb.modules.build import process
 from tinybird.tb.modules.cli import cli
 from tinybird.tb.modules.config import CLIConfig
-from tinybird.tb.modules.exceptions import CLIException
+from tinybird.tb.modules.exceptions import CLITestException
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 from tinybird.tb.modules.llm import LLM
 from tinybird.tb.modules.llm_utils import extract_xml, parse_xml
@@ -74,32 +73,17 @@ def test_create(ctx: click.Context, name_or_filename: str, prompt: str) -> None:
     """
     Create a test for an existing pipe
     """
-    project: Project = ctx.ensure_object(dict)["project"]
-    client: TinyB = ctx.ensure_object(dict)["client"]
-    click.echo(FeedbackManager.highlight(message="\n» Building project"))
-    process(project=project, tb_client=client, watch=False, silent=True)
-    click.echo(FeedbackManager.info(message="✓ Done!"))
-    config = CLIConfig.get_project_config()
-    root_path = project.path
-    folder = project.folder
-
     try:
-        if ".pipe" in name_or_filename:
-            pipe_path = Path(name_or_filename)
-            if not pipe_path.exists():
-                raise CLIException(FeedbackManager.error(message=f"Pipe {name_or_filename} not found"))
-        else:
-            pipe_folders = ("endpoints", "copies", "materializations", "sinks", "pipes")
-            pipe_path = next(
-                root_path / folder / f"{name_or_filename}.pipe"
-                for folder in pipe_folders
-                if os.path.exists(root_path / folder / f"{name_or_filename}.pipe")
-            )
-            if not pipe_path:
-                raise CLIException(FeedbackManager.error(message=f"Pipe {name_or_filename} not found"))
-
+        project: Project = ctx.ensure_object(dict)["project"]
+        client: TinyB = ctx.ensure_object(dict)["client"]
+        click.echo(FeedbackManager.highlight(message="\n» Building project"))
+        process(project=project, tb_client=client, watch=False, silent=True)
+        click.echo(FeedbackManager.info(message="✓ Done!\n"))
+        config = CLIConfig.get_project_config()
+        folder = project.folder
+        pipe_path = get_pipe_path(name_or_filename, folder)
         pipe_name = pipe_path.stem
-        click.echo(FeedbackManager.highlight(message=f"\n» Creating tests for {pipe_name} endpoint..."))
+        click.echo(FeedbackManager.highlight(message=f"» Creating tests for {pipe_name} endpoint..."))
         pipe_content = pipe_path.read_text()
         pipe = asyncio.run(client._req(f"/v0/pipes/{pipe_name}"))
         parameters = set([param["name"] for node in pipe["nodes"] for param in node["params"]])
@@ -111,7 +95,7 @@ def test_create(ctx: click.Context, name_or_filename: str, prompt: str) -> None:
         )
         user_token = config.get_user_token()
         if not user_token:
-            raise CLIException(FeedbackManager.error(message="No user token found"))
+            raise Exception("No user token found")
 
         llm = LLM(user_token=user_token, host=config.get_client().host)
         response_llm = llm.ask(system_prompt=system_prompt, prompt=prompt)
@@ -155,7 +139,7 @@ def test_create(ctx: click.Context, name_or_filename: str, prompt: str) -> None:
 
         click.echo(FeedbackManager.success(message="✓ Done!\n"))
     except Exception as e:
-        raise CLIException(FeedbackManager.error_exception(error=e))
+        raise CLITestException(FeedbackManager.error(message=str(e)))
 
 
 @test.command(
@@ -172,12 +156,12 @@ def test_update(ctx: click.Context, pipe: str) -> None:
         click.echo(FeedbackManager.highlight(message="\n» Building project"))
         process(project=project, tb_client=client, watch=False, silent=True)
         click.echo(FeedbackManager.info(message="✓ Done!"))
-        pipe_tests_path = Path(pipe)
-        pipe_name = pipe
+        pipe_tests_path = get_pipe_path(pipe, folder)
+        pipe_name = pipe_tests_path.stem
         if pipe_tests_path.suffix == ".yaml":
             pipe_name = pipe_tests_path.stem
         else:
-            pipe_tests_path = Path("tests", f"{pipe}.yaml")
+            pipe_tests_path = Path("tests", f"{pipe_name}.yaml")
 
         click.echo(FeedbackManager.highlight(message=f"\n» Updating tests expectations for {pipe_name} endpoint..."))
         pipe_tests_path = Path(project.folder) / pipe_tests_path
@@ -206,7 +190,7 @@ def test_update(ctx: click.Context, pipe: str) -> None:
 
         click.echo(FeedbackManager.success(message="✓ Done!\n"))
     except Exception as e:
-        click.echo(FeedbackManager.error(message=str(e)))
+        raise CLITestException(FeedbackManager.error(message=str(e)))
 
 
 @test.command(
@@ -216,70 +200,74 @@ def test_update(ctx: click.Context, pipe: str) -> None:
 @click.argument("name", nargs=-1)
 @click.pass_context
 def run_tests(ctx: click.Context, name: Tuple[str, ...]) -> None:
-    client: TinyB = ctx.ensure_object(dict)["client"]
-    project: Project = ctx.ensure_object(dict)["project"]
-    click.echo(FeedbackManager.highlight(message="\n» Building project"))
-    process(project=project, tb_client=client, watch=False, silent=True)
-    click.echo(FeedbackManager.info(message="✓ Done!"))
-
-    click.echo(FeedbackManager.highlight(message="\n» Running tests"))
-    paths = [Path(n) for n in name]
-    endpoints = [f"{project.path}/tests/{p.stem}.yaml" for p in paths]
-    test_files: List[str] = (
-        endpoints if len(endpoints) > 0 else glob.glob(f"{project.path}/tests/**/*.y*ml", recursive=True)
-    )
+    try:
+        client: TinyB = ctx.ensure_object(dict)["client"]
+        project: Project = ctx.ensure_object(dict)["project"]
+        click.echo(FeedbackManager.highlight(message="\n» Building project"))
+        process(project=project, tb_client=client, watch=False, silent=True)
+        click.echo(FeedbackManager.info(message="✓ Done!"))
 
-    def run_test(test_file):
-        test_file_path = Path(test_file)
-        click.echo(FeedbackManager.info(message=f"* {test_file_path.stem}{test_file_path.suffix}"))
-        test_file_content = yaml.safe_load(test_file_path.read_text())
+        click.echo(FeedbackManager.highlight(message="\n» Running tests"))
+        paths = [Path(n) for n in name]
+        endpoints = [f"{project.path}/tests/{p.stem}.yaml" for p in paths]
+        test_files: List[str] = (
+            endpoints if len(endpoints) > 0 else glob.glob(f"{project.path}/tests/**/*.y*ml", recursive=True)
+        )
 
-        for test in test_file_content:
-            try:
-                test_params = test["parameters"].split("?")[1] if "?" in test["parameters"] else test["parameters"]
-                response = None
-                try:
-                    response = asyncio.run(
-                        get_pipe_data(client, pipe_name=test_file_path.stem, test_params=test_params)
-                    )
-                except Exception:
-                    continue
+        def run_test(test_file):
+            test_file_path = Path(test_file)
+            click.echo(FeedbackManager.info(message=f"* {test_file_path.stem}{test_file_path.suffix}"))
+            test_file_content = yaml.safe_load(test_file_path.read_text())
 
-                expected_result = response.text
-                if response.status_code >= 400:
-                    expected_result = response.json()["error"]
-                    if "expected_http_status" not in test:
-                        raise Exception("Expected to not fail but got an error")
-                    if test["expected_http_status"] != response.status_code:
-                        raise Exception(f"Expected {test['expected_http_status']} but got {response.status_code}")
-
-                if test["expected_result"] != expected_result:
-                    diff = difflib.ndiff(
-                        test["expected_result"].splitlines(keepends=True), expected_result.splitlines(keepends=True)
-                    )
-                    printable_diff = "".join(diff)
-                    raise Exception(
-                        f"\nExpected: \n{test['expected_result']}\nGot: \n{expected_result}\nDiff: \n{printable_diff}"
-                    )
-                click.echo(FeedbackManager.info(message=f"✓ {test['name']} passed"))
-            except Exception as e:
-                click.echo(FeedbackManager.error(message=f"✗ {test['name']} failed"))
-                click.echo(FeedbackManager.error(message=f"\n** Output and expected output are different: \n{e}"))
-                return False
-        return True
-
-    failed_tests_count = 0
-    test_count = len(test_files)
-
-    for test_file in test_files:
-        if not run_test(test_file):
-            failed_tests_count += 1
-
-    if failed_tests_count:
-        click.echo(FeedbackManager.error(message=f"\n✗ {test_count - failed_tests_count}/{test_count} passed"))
-        sys.exit(1)
-    else:
-        click.echo(FeedbackManager.success(message=f"\n✓ {test_count}/{test_count} passed"))
+            for test in test_file_content:
+                try:
+                    test_params = test["parameters"].split("?")[1] if "?" in test["parameters"] else test["parameters"]
+                    response = None
+                    try:
+                        response = asyncio.run(
+                            get_pipe_data(client, pipe_name=test_file_path.stem, test_params=test_params)
+                        )
+                    except Exception:
+                        continue
+
+                    expected_result = response.text
+                    if response.status_code >= 400:
+                        expected_result = response.json()["error"]
+                        if "expected_http_status" not in test:
+                            raise Exception("Expected to not fail but got an error")
+                        if test["expected_http_status"] != response.status_code:
+                            raise Exception(f"Expected {test['expected_http_status']} but got {response.status_code}")
+
+                    if test["expected_result"] != expected_result:
+                        diff = difflib.ndiff(
+                            test["expected_result"].splitlines(keepends=True), expected_result.splitlines(keepends=True)
+                        )
+                        printable_diff = "".join(diff)
+                        raise Exception(
+                            f"\nExpected: \n{test['expected_result']}\nGot: \n{expected_result}\nDiff: \n{printable_diff}"
+                        )
+                    click.echo(FeedbackManager.info(message=f"✓ {test['name']} passed"))
+                except Exception as e:
+                    click.echo(FeedbackManager.error(message=f"✗ {test['name']} failed"))
+                    click.echo(FeedbackManager.error(message=f"\n** Output and expected output are different: \n{e}"))
+                    return False
+            return True
+
+        failed_tests_count = 0
+        test_count = len(test_files)
+
+        for test_file in test_files:
+            if not run_test(test_file):
+                failed_tests_count += 1
+
+        if failed_tests_count:
+            error = f"\n✗ {test_count - failed_tests_count}/{test_count} passed"
+            click.echo(FeedbackManager.error(message=error))
+            sys.exit(1)
+        else:
+            click.echo(FeedbackManager.success(message=f"\n✓ {test_count}/{test_count} passed"))
+    except Exception as e:
+        raise CLITestException(FeedbackManager.error(message=str(e)))
 
 
 async def get_pipe_data(client: TinyB, pipe_name: str, test_params: str) -> Response:
@@ -296,3 +284,20 @@ async def get_pipe_data(client: TinyB, pipe_name: str, test_params: str) -> Resp
         "pipeline": pipe_name,
     }
     return await client._req_raw(f"""/v0/sql?{urllib.parse.urlencode(params)}&{test_params}""")
+
+
+def get_pipe_path(name_or_filename: str, folder: str) -> Path:
+    pipe_path: Optional[Path] = None
+
+    if ".pipe" in name_or_filename:
+        pipe_path = Path(name_or_filename)
+        if not pipe_path.exists():
+            pipe_path = None
+    else:
+        pipes = glob.glob(f"{folder}/**/{name_or_filename}.pipe", recursive=True)
+        pipe_path = next((Path(p) for p in pipes if Path(p).exists()), None)
+
+    if not pipe_path:
+        raise Exception(f"Pipe {name_or_filename} not found")
+
+    return pipe_path

{tinybird-0.0.1.dev133.dist-info → tinybird-0.0.1.dev134.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev133
+Version: 0.0.1.dev134
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/cli/introduction.html
 Author: Tinybird

{tinybird-0.0.1.dev133.dist-info → tinybird-0.0.1.dev134.dist-info}/RECORD

@@ -12,34 +12,34 @@ tinybird/syncasync.py,sha256=IPnOx6lMbf9SNddN1eBtssg8vCLHMt76SuZ6YNYm-Yk,27761
 tinybird/tornado_template.py,sha256=jjNVDMnkYFWXflmT8KU_Ssbo5vR8KQq3EJMk5vYgXRw,41959
 tinybird/ch_utils/constants.py,sha256=aYvg2C_WxYWsnqPdZB1ZFoIr8ZY-XjUXYyHKE9Ansj0,3890
 tinybird/ch_utils/engine.py,sha256=BZuPM7MFS7vaEKK5tOMR2bwSAgJudPrJt27uVEwZmTY,40512
-tinybird/tb/__cli__.py,sha256=pUQYSMtGIN0vu5mnf42KNZzpAaEz0aTgpMckE_sV-c0,252
+tinybird/tb/__cli__.py,sha256=erSaiNuWIQUqfvRFXRdp6KocbeKh_UOlcrPrViJUkzI,252
 tinybird/tb/cli.py,sha256=Xw5EhwCQu6Xx55DzoJsZJODdi7tkA391mii-xgGnpEU,1066
 tinybird/tb/client.py,sha256=aaPKq5C77e72kR7IMv9WrvnvNki8mKMOTi9EsCp0eUc,55962
-tinybird/tb/config.py,sha256=iIia7F67UI9p51Ja4jALonyttIi4eH7PwOn3HvcvOb8,4008
+tinybird/tb/config.py,sha256=HLMHbJg6BjeGZ2KiJA-BCCVnk8w959xsSaDEEePakZg,3981
 tinybird/tb/modules/auth.py,sha256=_OeYnmTH83lnqCgQEdS6K0bx1KBUeRmZk2M7JnRmWpk,9037
-tinybird/tb/modules/build.py,sha256=KqXMJJW8JEuTlH3KyV1ylCLUob_oFhwDTM7bYLgrK9s,17359
+tinybird/tb/modules/build.py,sha256=ohHiYwygJtldWsina-2QpvIt-g2hIqOkm5BEnh4JhdA,17187
 tinybird/tb/modules/cicd.py,sha256=Ug7LijuHDIqKR6rm8OLZispWp75Zv6TyonAme8K9jaI,7114
 tinybird/tb/modules/cli.py,sha256=Y_5hu9xwyTIZw4bQoe0MYLnRIzmR7hUjql_oZBxd4Qg,13407
-tinybird/tb/modules/common.py,sha256=bdKNjHtQKcsZFyZIsLY8vEoTD0Xf-1UNBnh1pAyWPZU,83456
+tinybird/tb/modules/common.py,sha256=Uad-hW9VDYYQsqMYVTqysOTFoBMuVCwQqPhzVj99Nh0,84182
 tinybird/tb/modules/config.py,sha256=ziqW_t_mRVvWOd85VoB4vKyvgMkEfpXDf9H4v38p2xc,11422
 tinybird/tb/modules/connection.py,sha256=7qJGhTG0U__ZhY2HBWP8F1Z6tXrxwgAU-eOup6pXhhU,6905
 tinybird/tb/modules/copy.py,sha256=2Mm4FWKehOG7CoOhiF1m9UZJgJn0W1_cMolqju8ONYg,5805
-tinybird/tb/modules/create.py,sha256=PDAuzeBiqN9q7GY2p4zcMWDojdNjz6_aJgI5dxR-FVM,16701
+tinybird/tb/modules/create.py,sha256=hgfnV0zMOYDdXZ1nEK_eEQK66TBFM8XBHvHi_Y9R7z8,16439
 tinybird/tb/modules/datasource.py,sha256=XvY7iTRefG9iEBLQkqBSF47bK-CrLu2AzDnigGI2mSc,16925
-tinybird/tb/modules/deployment.py,sha256=4Zt7jPbqt18fB5kPx7DbO91Bh6xzBBTEUFY7O89shuU,19560
+tinybird/tb/modules/deployment.py,sha256=M3NPFQXIEsDh9-J-pzPLuHF9k4SIZVvgUnHp05KOf-E,20448
 tinybird/tb/modules/endpoint.py,sha256=XySDt3pk66vxOZ0egUfz4bY8bEk3BjOXkv-L0OIJ3sc,12083
-tinybird/tb/modules/exceptions.py,sha256=4A2sSjCEqKUMqpP3WI00zouCWW4uLaghXXLZBSw04mY,3363
+tinybird/tb/modules/exceptions.py,sha256=5jK91w1LPmtqIUfDpHe_Op5OxGz8-p1BPgtLREMIni0,5217
 tinybird/tb/modules/feedback_manager.py,sha256=8I6n5JFPS4xP-Pfb9iu7hqx5VX773uGDXH2E5QeANeg,74179
 tinybird/tb/modules/infra.py,sha256=eX0StUipaXaxIhMxO82cnAnFKTCa7EPOt-e8pWoSLX0,33206
 tinybird/tb/modules/job.py,sha256=n4dSSBgnA8NqD7srGahf2xRj6wxkmX9Vl0J-QJ_a2w0,2966
 tinybird/tb/modules/llm.py,sha256=KfsCYmKeW1VQz0iDZhGKCRkQv_Y3kTHh6JuxvofOguE,1076
 tinybird/tb/modules/llm_utils.py,sha256=nS9r4FAElJw8yXtmdYrx-rtI2zXR8qXfi1QqUDCfxvg,3469
-tinybird/tb/modules/local.py,sha256=702U0p9pPFwZHkVNEsp2Y5gJxBKtOvd2XoS4dn1p-m8,11111
-tinybird/tb/modules/local_common.py,sha256=75Mje9_vkq7VehSKHS1zJB8UyVsJpqDFmuJGWwQmrXQ,3193
-tinybird/tb/modules/login.py,sha256=rLYdV7MLCTT1F4VBAWGv6SO9efMiM9-977ayLecUzjI,6842
+tinybird/tb/modules/local.py,sha256=-oXIeVXjzdfxVPQaXw68YyQOb48EFQZeUys8_U0xePs,11300
+tinybird/tb/modules/local_common.py,sha256=lyJ8vIFsDcVGYbfoCEBNgkwROsF5YJovxME1CRykEj8,3321
+tinybird/tb/modules/login.py,sha256=8oybH3rwTqNrHx0gN2zG-9MGSYsKwLuD2hKZsKJtrQA,7152
 tinybird/tb/modules/logout.py,sha256=ULooy1cDBD02-r7voZmhV7udA0ML5tVuflJyShrh56Y,1022
 tinybird/tb/modules/materialization.py,sha256=GvnItYXzBKRZjcuOIuOEwtdM4LxYbTH4M5vGZR233Hc,5323
-tinybird/tb/modules/mock.py,sha256=KScG2WohfCjxakBS2Rfb4-7QKS4OSkoVgiMyUvkCWWE,4924
+tinybird/tb/modules/mock.py,sha256=tyhs6izqVGmDzMHdW5yHiprNPjeJDa2I-cLwrLRDC_U,4816
 tinybird/tb/modules/open.py,sha256=OuctINN77oexpSjth9uoIZPCelKO4Li-yyVxeSnk1io,1371
 tinybird/tb/modules/pipe.py,sha256=AQKEDagO6e3psPVjJkS_MDbn8aK-apAiLp26k7jgAV0,2432
 tinybird/tb/modules/project.py,sha256=RdVOzJiuFPqkCzPhzGetbgcFeiMEn9IMviFCG58XKgI,3142
@@ -48,8 +48,8 @@ tinybird/tb/modules/secret.py,sha256=WsqzxxLh9W_jkuHL2JofMXdIJy0lT5WEI-7bQSIDgAc
 tinybird/tb/modules/shell.py,sha256=cvT0EKpnSRHfuo3avykBRMsCwLBqmzSR7sNUdbQ6lXA,14116
 tinybird/tb/modules/table.py,sha256=4XrtjM-N0zfNtxVkbvLDQQazno1EPXnxTyo7llivfXk,11035
 tinybird/tb/modules/tag.py,sha256=anPmMUBc-TbFovlpFi8GPkKA18y7Y0GczMsMms5TZsU,3502
-tinybird/tb/modules/telemetry.py,sha256=FcFw_70-0SyC_2BdToF_sjVpXpMD7i00fCm2SO4cHx8,11397
-tinybird/tb/modules/test.py,sha256=9qNIM5bosYlaZYd5rorC5zcrTmjZeTUSuOIP-wkjXXI,12132
+tinybird/tb/modules/telemetry.py,sha256=IHHks8-jnrkyQIY_Y6ZmU8hdVHMGSS0pkSUOgCPcYY8,11346
+tinybird/tb/modules/test.py,sha256=WmPoZBSHBORiM66xp54KxfY3s0YwjFMN_ZTT8xObFO4,12377
 tinybird/tb/modules/token.py,sha256=2fmKwu10_M0pqs6YmJVeILR9ZQB0ejRAET86agASbKM,13488
 tinybird/tb/modules/watch.py,sha256=_92co0BjTikQuy7MbHr4TDu9A75QdHsrAO8v7DlYLg4,8898
 tinybird/tb/modules/workspace.py,sha256=SlK08psp0tu5t_URHyIczm696buW6KD6FPs9Lg1aNRE,6614
@@ -78,8 +78,8 @@ tinybird/tb_cli_modules/config.py,sha256=IsgdtFRnUrkY8-Zo32lmk6O7u3bHie1QCxLwgp4
 tinybird/tb_cli_modules/exceptions.py,sha256=pmucP4kTF4irIt7dXiG-FcnI-o3mvDusPmch1L8RCWk,3367
 tinybird/tb_cli_modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
 tinybird/tb_cli_modules/telemetry.py,sha256=Hh2Io8ZPROSunbOLuMvuIFU4TqwWPmQTqal4WS09K1A,10449
-tinybird-0.0.1.dev133.dist-info/METADATA,sha256=y0qf78t8pUbZYXxlRWXKybZCiiGD6UhpW7OombLD4Hs,1612
-tinybird-0.0.1.dev133.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev133.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev133.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev133.dist-info/RECORD,,
+tinybird-0.0.1.dev134.dist-info/METADATA,sha256=BVIpZF5O2MRJZ4HoxPRmqDbz-s0dG_-sQQKNZuC9hh0,1612
+tinybird-0.0.1.dev134.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev134.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev134.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev134.dist-info/RECORD,,