tinybird 3.3.2.dev0__tar.gz → 3.5.1__tar.gz

{tinybird-3.3.2.dev0 → tinybird-3.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 3.3.2.dev0
+Version: 3.5.1
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird
@@ -57,6 +57,26 @@ The Tinybird command-line tool allows you to use all the Tinybird functionality
 Changelog
 ----------
 
+3.5.2
+*******
+
+- `Changed` `tb deploy --auto/--no-auto` keeps the same CLI behavior (default `--auto`) while using server-side deployment `auto_promote` under the hood when enabled.
+
+3.5.1
+*******
+
+- `Fixed` Support special characters in workspace names in the login flow.
+
+3.5.0
+*******
+
+- `Changed` AI-assisted CLI commands now show a deprecation warning indicating they will be removed in a future major release and include agent skills guidance (`npx skills add @tinybirdco/tinybird-agent-skills`).
+
+3.4.0
+*******
+
+- `Changed` Prompt-based AI flows now print a deprecation warning (`tb --prompt`, `tb create --prompt`, `tb datasource create --prompt`, `tb test create`, `tb mock`) ahead of their removal in a future release.
+
 3.3.1
 *******
 

{tinybird-3.3.2.dev0 → tinybird-3.5.1}/tinybird/sql_toolset.py

@@ -3,6 +3,7 @@ import logging
 import re
 import threading
 from collections import defaultdict
+from dataclasses import dataclass
 from datetime import datetime
 from functools import lru_cache
 from typing import FrozenSet, List, Optional, Set, Tuple, Union
@@ -74,6 +75,24 @@ def explain_plan(sql: str) -> str:
     return chquery.explain_ast(sql)
 
 
+@dataclass(frozen=True)
+class ColumnInfo:
+    name: str
+    type: str
+    nullable: bool
+    default_specifier: str = ""
+    default_expression: str | None = None
+    codec: str | None = None
+    comment: str | None = None
+    ttl: str | None = None
+    is_primary_key: bool = False
+
+
+def get_columns_from_create_query(sql_schema: str) -> list[ColumnInfo]:
+    columns = chquery.get_columns_from_create_query(sql_schema)
+    return [ColumnInfo(**col) for col in columns]
+
+
 def has_join(sql: str) -> bool:
     return any(line.rstrip().startswith("TableJoin") for line in explain_plan(sql).split())
 

{tinybird-3.3.2.dev0 → tinybird-3.5.1}/tinybird/tb/__cli__.py

@@ -4,5 +4,5 @@ __description__ = 'Tinybird Command Line Tool'
 __url__ = 'https://www.tinybird.co/docs/forward/commands'
 __author__ = 'Tinybird'
 __author_email__ = 'support@tinybird.co'
-__version__ = '3.3.2.dev0'
-__revision__ = 'f22e1ea'
+__version__ = '3.5.1'
+__revision__ = '296fce5'

{tinybird-3.3.2.dev0 → tinybird-3.5.1}/tinybird/tb/modules/cli.py

@@ -36,6 +36,7 @@ from tinybird.tb.modules.common import (
     force_echo,
     getenv_bool,
     try_update_config_with_remote,
+    warn_prompt_ai_deprecation,
 )
 from tinybird.tb.modules.config import CURRENT_VERSION, CLIConfig
 from tinybird.tb.modules.datafile.build import build_graph
@@ -54,6 +55,52 @@ DEFAULT_PATTERNS: List[Tuple[str, Union[str, Callable[[str], str]]]] = [
 VERSION = f"{__cli__.__version__} (rev {__cli__.__revision__})"
 
 
+def _get_raw_cli_args(ctx: Context) -> List[str]:
+    root_ctx = ctx.find_root()
+    root_obj = root_ctx.obj if isinstance(root_ctx.obj, dict) else {}
+    raw_args = root_obj.get("_tb_raw_args")
+    if isinstance(raw_args, list):
+        return raw_args
+    return sys.argv[1:]
+
+
+def _argv_has_option(args: List[str], flag: str) -> bool:
+    return any(arg == flag or arg.startswith(f"{flag}=") for arg in args)
+
+
+def _argv_has_sequence(args: List[str], command_parts: List[str]) -> bool:
+    sequence_len = len(command_parts)
+    return any(args[i : i + sequence_len] == command_parts for i in range(len(args) - sequence_len + 1))
+
+
+def _should_warn_prompt_ai_deprecation(
+    ctx: Context,
+    is_agent_mode: bool,
+    is_prompt_mode: bool,
+) -> bool:
+    raw_args = _get_raw_cli_args(ctx)
+
+    if is_agent_mode or is_prompt_mode:
+        return True
+
+    if ctx.invoked_subcommand == "mock":
+        return True
+
+    if ctx.invoked_subcommand == "test" and _argv_has_sequence(raw_args, ["test", "create"]):
+        return True
+
+    if ctx.invoked_subcommand == "create" and (
+        _argv_has_option(raw_args, "--prompt") or _argv_has_option(raw_args, "-p")
+    ):
+        return True
+
+    return (
+        ctx.invoked_subcommand == "datasource"
+        and _argv_has_sequence(raw_args, ["datasource", "create"])
+        and _argv_has_option(raw_args, "--prompt")
+    )
+
+
 @click.group(
     cls=CatchAuthExceptions,
     context_settings={
@@ -129,6 +176,7 @@ def cli(
         click.echo(FeedbackManager.warning_disabled_ssl_checks())
 
     is_agent_mode = ctx.invoked_subcommand is None
+    is_prompt_mode = prompt is not None
     if not environ.get("PYTEST", None) and version_warning and not token and not is_agent_mode:
         latest_version = CheckPypi().get_latest_version()
         if latest_version:
@@ -215,6 +263,10 @@ def cli(
         return
 
     ctx.ensure_object(dict)["project"] = project
+
+    if _should_warn_prompt_ai_deprecation(ctx, is_agent_mode, is_prompt_mode):
+        warn_prompt_ai_deprecation()
+
     client = create_ctx_client(
         ctx,
         config,
@@ -235,8 +287,6 @@ def cli(
     # Check if current folder is tracked from previous sessions
     check_current_folder_in_sessions(ctx)
 
-    is_prompt_mode = prompt is not None
-
     if is_agent_mode or is_prompt_mode:
         if any(arg in sys.argv for arg in ["--cloud", "--local", "--branch"]):
             raise CLIException(

{tinybird-3.3.2.dev0 → tinybird-3.5.1}/tinybird/tb/modules/common.py

@@ -74,6 +74,10 @@ _PATTERN_AWS_ARN = re.compile(r"arn:aws:iam::(\d+):root")
 SUPPORTED_FORMATS = ["csv", "ndjson", "json", "parquet"]
 OLDEST_ROLLBACK = "oldest_rollback"
 MAIN_BRANCH = "main"
+PROMPT_AI_DEPRECATION_WARNING = (
+    "Tinybird Code and `--prompt` commands are now deprecated and will be removed in the next major release.\n"
+    "You can instead bring the best of Tinybird Code to your preferred coding agent with skills. Run: npx skills add @tinybirdco/tinybird-agent-skills\n"
+)
 
 
 def obfuscate_token(value: Optional[str]) -> Optional[str]:
@@ -82,6 +86,17 @@ def obfuscate_token(value: Optional[str]) -> Optional[str]:
     return f"{value[:4]}...{value[-8:]}"
 
 
+def warn_prompt_ai_deprecation() -> None:
+    ctx = click.get_current_context(silent=True)
+    if ctx:
+        root_ctx = ctx.find_root()
+        root_obj = root_ctx.ensure_object(dict)
+        if root_obj.get("_prompt_ai_deprecation_warned"):
+            return
+        root_obj["_prompt_ai_deprecation_warned"] = True
+    click.echo(FeedbackManager.warning(message=PROMPT_AI_DEPRECATION_WARNING))
+
+
 def gather_with_concurrency(n, *tasks):
     results = []
     for task in tasks:
@@ -237,6 +252,12 @@ class CatchAuthExceptions(AliasedGroup):
         add_telemetry_sysinfo_event()
         super().__init__(*args, **kwargs)
 
+    def parse_args(self, ctx: click.Context, args: List[str]) -> List[str]:
+        # Keep original invocation args so root-level handlers can inspect
+        # subcommand flags without relying on sys.argv (e.g. in CliRunner tests).
+        ctx.ensure_object(dict)["_tb_raw_args"] = list(args)
+        return super().parse_args(ctx, args)
+
     def format_epilog(self, ctx: Context, formatter: click.formatting.HelpFormatter) -> None:
         super().format_epilog(ctx, formatter)
 

{tinybird-3.3.2.dev0 → tinybird-3.5.1}/tinybird/tb/modules/datasource.py

@@ -38,6 +38,7 @@ from tinybird.tb.modules.common import (
     get_format_from_filename_or_url,
     normalize_datasource_name,
     push_data,
+    warn_prompt_ai_deprecation,
 )
 from tinybird.tb.modules.config import CLIConfig
 from tinybird.tb.modules.connection_kafka import (
@@ -975,6 +976,7 @@ ENGINE "MergeTree"
             return
 
         if datasource_type == "prompt":
+            warn_prompt_ai_deprecation()
             if not config.get("user_token"):
                 raise Exception("This action requires authentication. Run 'tb login' first.")
 

{tinybird-3.3.2.dev0 → tinybird-3.5.1}/tinybird/tb/modules/deployment_common.py

@@ -87,6 +87,63 @@ def api_post(
     return {}
 
 
+def _is_first_deployment_with_seed_live(host: Optional[str], headers: dict) -> bool:
+    """Best-effort check for first real deployment when seed deployment (id=0) is still live."""
+    try:
+        response = requests.get(f"{host}/v1/deployments", headers=headers)
+        if response.status_code >= 300:
+            return False
+
+        result = response.json()
+        logging.debug(json.dumps(result, indent=2))
+        deployments = result.get("deployments") or []
+        if not deployments:
+            return False
+
+        live_deployment = next((deployment for deployment in deployments if deployment.get("live")), deployments[0])
+        return str(live_deployment.get("id")) == "0"
+    except Exception:
+        # Hint calculation must never break deployment flow.
+        return False
+
+
+def _get_deployment_job(client: TinyB, deployment_id: Optional[Union[str, int]]) -> Optional[Dict[str, Any]]:
+    if deployment_id is None:
+        return None
+
+    try:
+        deployment_id = str(deployment_id)
+        jobs = client.jobs()
+        return next(
+            (job for job in jobs if job.get("kind") == "deployment" and str(job.get("deployment_id")) == deployment_id),
+            None,
+        )
+    except Exception:
+        return None
+
+
+def _cancel_deployment_job(client: TinyB, deployment_job: Optional[Dict[str, Any]]) -> None:
+    if not deployment_job:
+        return
+
+    job_id = str(deployment_job.get("id") or deployment_job.get("job_id") or "")
+    if not job_id:
+        return
+
+    try:
+        result = client.job_cancel(job_id)
+    except Exception:
+        return
+
+    current_job_status = result.get("status")
+    if current_job_status == "cancelling":
+        click.echo(FeedbackManager.success_job_cancellation_cancelling(job_id=job_id))
+    elif current_job_status == "cancelled":
+        click.echo(FeedbackManager.success_job_cancellation_cancelled(job_id=job_id))
+    else:
+        click.echo(FeedbackManager.error_job_cancelled_but_status_unknown(job_id=job_id))
+
+
 # TODO(eclbg): This logic should be in the server, and there should be a dedicated endpoint for promoting a deployment
 # potato
 def promote_deployment(host: Optional[str], headers: dict, wait: bool, ingest_hint: Optional[bool] = True) -> None:
@@ -272,12 +329,15 @@ def create_deployment(
             )
 
     deployment = None
+    deployment_job: Optional[Dict[str, Any]] = None
     try:
         HEADERS = {"Authorization": f"Bearer {TINYBIRD_API_KEY}"}
         params = {}
         if check:
             click.echo(FeedbackManager.highlight(message="\n» Validating deployment...\n"))
             params["check"] = "true"
+        elif auto:
+            params["auto_promote"] = "true"
         if allow_destructive_operations:
             params["allow_destructive_operations"] = "true"
 
@@ -338,15 +398,7 @@ def create_deployment(
                 FeedbackManager.gray(message="Deployment URL: ")
                 + f"{bcolors.UNDERLINE}{host}/{config.get('name')}/deployments/{deployment.get('id')}{bcolors.ENDC}"
             )
-            jobs = client.jobs()
-            deployment_job = next(
-                (
-                    job
-                    for job in jobs
-                    if job.get("kind") == "deployment" and job.get("deployment_id") == deployment.get("id")
-                ),
-                None,
-            )
+            deployment_job = _get_deployment_job(client, deployment.get("id"))
             if deployment_job:
                 echo_job_url(
                     token=client.token,
@@ -379,52 +431,76 @@ def create_deployment(
 
     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("deployment_error", "Error parsing deployment from response")
-                return
-
-            status = deployment.get("status")
-            errors = deployment.get("errors")
-            feedback = deployment.get("feedback")
-
-            if status == "failed":
-                # Just wait until we poll again and see deleting or deleted to report errors
-                pass
-
-            if status == "data_ready":
-                break
+        waiting_auto_promote = False
+        is_first_deployment = str(deployment.get("id")) == "1"
+        try:
+            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("deployment_error", "Error parsing deployment from response")
+                    return
 
-            if status in ["deleting", "deleted"]:
+                status = deployment.get("status")
                 errors = deployment.get("errors")
-                if errors:
-                    verb = "is being" if status == "deleting" else "was"
-                    click.echo(
-                        FeedbackManager.error(
-                            message=f"Deployment failed and {verb} deleted automatically. Deployment errors:"
+                feedback = deployment.get("feedback")
+
+                if status == "failed":
+                    # Just wait until we poll again and see deleting or deleted to report errors
+                    pass
+
+                if status == "data_ready":
+                    if auto and not deployment.get("live"):
+                        if not waiting_auto_promote:
+                            click.echo(FeedbackManager.info(message="✓ Deployment is ready"))
+                            click.echo(FeedbackManager.highlight(message="» Waiting for deployment to be promoted..."))
+                            if not is_first_deployment:
+                                is_first_deployment = _is_first_deployment_with_seed_live(client.host, HEADERS)
+                            waiting_auto_promote = True
+                        time.sleep(5)
+                        continue
+                    break
+
+                if status in ["deleting", "deleted"]:
+                    errors = deployment.get("errors")
+                    if errors:
+                        verb = "is being" if status == "deleting" else "was"
+                        click.echo(
+                            FeedbackManager.error(
+                                message=f"Deployment failed and {verb} deleted automatically. Deployment errors:"
+                            )
                         )
+                        for error in errors:
+                            click.echo(FeedbackManager.error(message=f"* {error}"))
+                    sys_exit(
+                        "deployment_error",
+                        f"Deployment deleted after failure. Errors: {str(errors + (feedback or []))}",
                     )
-                    for error in errors:
-                        click.echo(FeedbackManager.error(message=f"* {error}"))
-                sys_exit(
-                    "deployment_error", f"Deployment deleted after failure. Errors: {str(errors + (feedback or []))}"
-                )
-
-            time.sleep(5)
 
-        click.echo(FeedbackManager.info(message="✓ Deployment is ready"))
+                time.sleep(5)
+        except KeyboardInterrupt:
+            if auto:
+                if not deployment_job:
+                    deployment_job = _get_deployment_job(client, deployment.get("id") if deployment else None)
+                _cancel_deployment_job(client, deployment_job)
+            raise click.Abort()
 
         if auto:
-            promote_deployment(client.host, HEADERS, wait=wait, ingest_hint=ingest_hint)
-            # Fetch the final deployment state after promotion for JSON output
-            if output == "json":
-                url = f"{client.host}/v1/deployments/{deployment.get('id')}"
-                res = api_fetch(url, HEADERS)
-                deployment = res.get("deployment")
+            if not waiting_auto_promote:
+                click.echo(FeedbackManager.info(message="✓ Deployment is ready"))
+                click.echo(FeedbackManager.highlight(message="» Waiting for deployment to be promoted..."))
+            click.echo(FeedbackManager.info(message="✓ Deployment promoted"))
+            click.echo(FeedbackManager.success(message=f"✓ Deployment #{deployment.get('id')} is live!"))
+            if ingest_hint and len(deployment.get("new_data_connector_ids", [])) == 0 and is_first_deployment:
+                click.echo(
+                    FeedbackManager.info(
+                        message="Need help ingesting your data? Learn how at https://www.tinybird.co/docs/forward/get-data-in"
+                    )
+                )
+        else:
+            click.echo(FeedbackManager.info(message="✓ Deployment is ready"))
 
     # Output JSON at the appropriate time based on the execution path
     if output == "json" and deployment:

{tinybird-3.3.2.dev0 → tinybird-3.5.1}/tinybird/tb/modules/login_common.py

@@ -63,7 +63,7 @@ class AuthHandler(http.server.SimpleHTTPRequestHandler):
                 const host = "{auth_host}";
                 fetch('/?code=' + code, {{method: 'POST'}})
                     .then(() => {{
-                        window.location.href = host + "/" + provider + "/" + region + "/cli-login?workspace=" + workspace;
+                        window.location.href = host + "/" + provider + "/" + region + "/cli-login?workspace=" + encodeURIComponent(workspace);
                     }});
             </script>
         </body>

{tinybird-3.3.2.dev0 → tinybird-3.5.1}/tinybird.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 3.3.2.dev0
+Version: 3.5.1
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird
@@ -57,6 +57,26 @@ The Tinybird command-line tool allows you to use all the Tinybird functionality
 Changelog
 ----------
 
+3.5.2
+*******
+
+- `Changed` `tb deploy --auto/--no-auto` keeps the same CLI behavior (default `--auto`) while using server-side deployment `auto_promote` under the hood when enabled.
+
+3.5.1
+*******
+
+- `Fixed` Support special characters in workspace names in the login flow.
+
+3.5.0
+*******
+
+- `Changed` AI-assisted CLI commands now show a deprecation warning indicating they will be removed in a future major release and include agent skills guidance (`npx skills add @tinybirdco/tinybird-agent-skills`).
+
+3.4.0
+*******
+
+- `Changed` Prompt-based AI flows now print a deprecation warning (`tb --prompt`, `tb create --prompt`, `tb datasource create --prompt`, `tb test create`, `tb mock`) ahead of their removal in a future release.
+
 3.3.1
 *******