tinybird 0.0.1.dev247__tar.gz → 0.0.1.dev249__tar.gz

{tinybird-0.0.1.dev247 → tinybird-0.0.1.dev249}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev247
+Version: 0.0.1.dev249
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird

{tinybird-0.0.1.dev247 → tinybird-0.0.1.dev249}/tinybird/ch_utils/engine.py

@@ -134,8 +134,9 @@ class TableDetails:
         _version = self.details.get("version", None)
         return _version
 
-    def is_replicated(self):
-        return "Replicated" in self.details.get("engine", None)
+    def is_replicated(self) -> bool:
+        engine: Optional[str] = self.details.get("engine", None)
+        return engine is not None and "Replicated" in engine
 
     def is_mergetree_family(self) -> bool:
         return self.engine is not None and "mergetree" in self.engine.lower()

{tinybird-0.0.1.dev247 → tinybird-0.0.1.dev249}/tinybird/prompts.py

@@ -776,6 +776,8 @@ datasource_instructions = """
     - Use always json paths to define the schema. Example: `user_id` String `json:$.user_id`,
     - Array columns are supported with a special syntax. Example: `items` Array(String) `json:$.items[:]`
     - If the datasource is using an S3 or GCS connection, they need to set IMPORT_CONNECTION_NAME, IMPORT_BUCKET_URI and IMPORT_SCHEDULE (GCS @on-demand only, S3 supports @auto too)
+    - Unless the user asks for them, do not include ENGINE_PARTITION_KEY and ENGINE_PRIMARY_KEY.
+    - DateTime64 type without precision is not supported. Use DateTime64(3) instead.
 </datasource_file_instructions>
 """
 

{tinybird-0.0.1.dev247 → tinybird-0.0.1.dev249}/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__ = '0.0.1.dev247'
-__revision__ = '379a827'
+__version__ = '0.0.1.dev249'
+__revision__ = 'ef4e98c'

{tinybird-0.0.1.dev247 → tinybird-0.0.1.dev249}/tinybird/tb/modules/agent/agent.py

@@ -1,3 +1,4 @@
+import shlex
 import subprocess
 import sys
 from datetime import datetime
@@ -54,19 +55,29 @@ from tinybird.tb.modules.deployment_common import create_deployment
 from tinybird.tb.modules.exceptions import CLIBuildException, CLIMockException
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 from tinybird.tb.modules.local_common import get_tinybird_local_client
+from tinybird.tb.modules.login_common import login
 from tinybird.tb.modules.mock_common import append_mock_data, create_mock_data
 from tinybird.tb.modules.project import Project
 
 
 class TinybirdAgent:
-    def __init__(self, token: str, host: str, project: Project, dangerously_skip_permissions: bool):
+    def __init__(
+        self,
+        token: str,
+        user_token: str,
+        host: str,
+        workspace_id: str,
+        project: Project,
+        dangerously_skip_permissions: bool,
+    ):
         self.token = token
+        self.user_token = user_token
         self.host = host
         self.dangerously_skip_permissions = dangerously_skip_permissions
         self.project = project
         self.messages: list[ModelMessage] = []
         self.agent = Agent(
-            model=create_model(token, host),
+            model=create_model(user_token, host, workspace_id),
             deps_type=TinybirdAgentContext,
             system_prompt=f"""
 You are a Tinybird Code, an agentic CLI that can help users to work with Tinybird.
@@ -75,7 +86,7 @@ You are an interactive CLI tool that helps users with data engineering tasks. Us
 
 # Tone and style
 You should be concise, direct, and to the point. 
-Remember that your output will be displayed on a command line interface. Your responses can use Github-flavored markdown for formatting.
+Remember that your output will be displayed on a command line interface. Your responses can use Github-flavored markdown for formatting. Do not use emojis.
 Output text to communicate with the user; all text you output outside of tool use is displayed to the user. Only use tools to complete tasks. Never use tools like Bash or code comments as means to communicate with the user during the session.
 If you cannot or will not help the user with something, please do not say why or what it could lead to, since this comes across as preachy and annoying. Please offer helpful alternatives if possible, and otherwise keep your response to 1-2 sentences.
 IMPORTANT: You should minimize output tokens as much as possible while maintaining helpfulness, quality, and accuracy. Only address the specific query or task at hand, avoiding tangential information unless absolutely critical for completing the request. If you can answer in 1-3 sentences or a short paragraph, please do.
@@ -116,6 +127,7 @@ You have access to the following tools:
 8. If the datafile was created successfully, but the built failed, try to fix the error and repeat the process.
 
 IMPORTANT: If the user cancels some of the steps or there is an error in file creation, DO NOT continue with the plan. Stop the process and wait for the user before using any other tool.
+IMPORTANT: Every time you finish a plan and start a new resource creation or update process, create a new plan before starting with the changes.
 
 # When planning the creation or update of resources:
 {plan_instructions}
@@ -217,18 +229,41 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
 
 
 def run_agent(config: dict[str, Any], project: Project, dangerously_skip_permissions: bool):
-    display_banner()
-
+    token = config.get("token", None)
+    host = config.get("host", None)
+    user_token = config.get("user_token", None)
+    workspace_id = config.get("id", None)
     try:
-        token = config["token"]
-        host = config["host"]
-        agent = TinybirdAgent(token, host, project, dangerously_skip_permissions)
+        if not token or not host or not workspace_id or not user_token:
+            yes = click.confirm(
+                FeedbackManager.warning(
+                    message="Tinybird Code requires authentication. Do you want to authenticate now? [Y/n]"
+                ),
+                prompt_suffix="",
+                show_default=False,
+                default=True,
+            )
+            if yes:
+                click.echo()
+                login(host, auth_host="https://cloud.tinybird.co", workspace=None, interactive=False, method="browser")
+                click.echo()
+                cli_config = CLIConfig.get_project_config()
+                token = cli_config.get_token()
+                user_token = cli_config.get_user_token()
+                host = cli_config.get_host()
+
+            if not token or not host or not user_token:
+                click.echo(
+                    FeedbackManager.error(message="Tinybird Code requires authentication. Run 'tb login' first.")
+                )
+                return
+
+        display_banner()
+        agent = TinybirdAgent(token, user_token, host, workspace_id, project, dangerously_skip_permissions)
         click.echo()
-        if config.get("token"):
-            click.echo(FeedbackManager.info(message="Describe what you want to create and I'll help you build it"))
-            click.echo(FeedbackManager.info(message="Run /help for more commands"))
-        else:
-            click.echo(FeedbackManager.info(message="Run /login to authenticate"))
+        click.echo(FeedbackManager.info(message="Describe what you want to create and I'll help you build it"))
+        click.echo(FeedbackManager.info(message="Run /help for more commands"))
+
         click.echo()
 
     except Exception as e:
@@ -240,7 +275,7 @@ def run_agent(config: dict[str, Any], project: Project, dangerously_skip_permiss
         while True:
             try:
                 user_input = prompt(
-                    [("class:prompt", "tb » ")],
+                    [("class:prompt", f"tb ({project.workspace_name}) » ")],
                     history=load_history(),
                     cursor=CursorShape.BLOCK,
                     style=Style.from_dict(
@@ -250,7 +285,10 @@ def run_agent(config: dict[str, Any], project: Project, dangerously_skip_permiss
                         }
                     ),
                 )
-
+                if user_input.startswith("tb "):
+                    cmd_parts = shlex.split(user_input)
+                    subprocess.run(cmd_parts, check=True)
+                    continue
                 if user_input.lower() in ["/exit", "/quit"]:
                     click.echo(FeedbackManager.info(message="Goodbye!"))
                     break

{tinybird-0.0.1.dev247 → tinybird-0.0.1.dev249}/tinybird/tb/modules/agent/models.py

@@ -7,11 +7,12 @@ from pydantic_ai.providers.anthropic import AnthropicProvider
 def create_model(
     token: str,
     base_url: str,
+    workspace_id: str,
     model: AnthropicModelName = "claude-4-sonnet-20250514",
 ):
     client = AsyncAnthropic(
         base_url=base_url,
-        http_client=AsyncClient(params={"token": token}),
+        http_client=AsyncClient(params={"token": token, "workspace_id": workspace_id}),
         auth_token=token,
     )
     return AnthropicModel(

tinybird-0.0.1.dev249/tinybird/tb/modules/agent/tools/create_datafile.py

@@ -0,0 +1,71 @@
+from pathlib import Path
+
+import click
+from pydantic_ai import RunContext
+
+from tinybird.tb.modules.agent.utils import Datafile, TinybirdAgentContext, create_terminal_box, show_options
+from tinybird.tb.modules.exceptions import CLIBuildException
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+
+
+def get_resource_confirmation(resource: Datafile, exists: bool) -> bool:
+    """Get user confirmation for creating a resource"""
+    while True:
+        action = "create" if not exists else "update"
+        result = show_options(
+            options=[f"Yes, {action} {resource.type} '{resource.name}'", "No, and tell Tinybird Code what to do"],
+            title=f"What would you like to do with {resource.type} '{resource.name}'?",
+        )
+
+        if result is None:  # Cancelled
+            return False
+
+        if result.startswith("Yes"):
+            return True
+        elif result.startswith("No"):
+            return False
+
+        return False
+
+
+def create_datafile(ctx: RunContext[TinybirdAgentContext], resource: Datafile) -> str:
+    """Given a resource representation, create a file in the project folder
+
+    Args:
+        resource (Datafile): The resource to create. Required.
+
+    Returns:
+        str: If the resource was created or not.
+    """
+    try:
+        ctx.deps.thinking_animation.stop()
+        resource.pathname = resource.pathname.removeprefix("/")
+        path = Path(ctx.deps.folder) / resource.pathname
+        content = resource.content
+        exists = str(path) in ctx.deps.get_project_files()
+        if exists:
+            content = create_terminal_box(path.read_text(), resource.content, title=resource.pathname)
+        else:
+            content = create_terminal_box(resource.content, title=resource.pathname)
+        click.echo(content)
+        confirmation = ctx.deps.dangerously_skip_permissions or get_resource_confirmation(resource, exists)
+
+        if not confirmation:
+            ctx.deps.thinking_animation.start()
+            return f"Resource {resource.pathname} was not created. User cancelled creation."
+
+        folder_path = path.parent
+        folder_path.mkdir(parents=True, exist_ok=True)
+        path.touch(exist_ok=True)
+        path.write_text(resource.content)
+        ctx.deps.build_project(test=True, silent=True)
+        ctx.deps.thinking_animation.start()
+        return f"Created {resource.pathname}"
+
+    except CLIBuildException as e:
+        ctx.deps.thinking_animation.stop()
+        click.echo(FeedbackManager.error(message=e))
+        ctx.deps.thinking_animation.start()
+        return f"Error building project: {e}"
+    except Exception as e:
+        return f"Error creating {resource.pathname}: {e}"

{tinybird-0.0.1.dev247 → tinybird-0.0.1.dev249}/tinybird/tb/modules/agent/tools/read_fixture_data.py

@@ -23,6 +23,6 @@ def read_fixture_data(ctx: RunContext[TinybirdAgentContext], fixture_pathname: s
     response = ctx.deps.analyze_fixture(fixture_path=str(fixture_path))
     # limit content to first 10 rows
     data = response["preview"]["data"][:10]
-    schema = response["analysis"]["schema"]
+    columns = response["analysis"]["columns"]
 
-    return f"#Result of analysis of {fixture_pathname}:\n##Data sample:\n{json.dumps(data)}\n##Schema:\n{schema}"
+    return f"#Result of analysis of {fixture_pathname}:\n##Columns:\n{json.dumps(columns)}\n##Data sample:\n{json.dumps(data)}"

{tinybird-0.0.1.dev247 → tinybird-0.0.1.dev249}/tinybird/tb/modules/agent/utils.py

@@ -1,6 +1,7 @@
+import difflib
 import os
 from contextlib import contextmanager
-from typing import Any, Callable, List, Optional
+from typing import Any, Callable, List, Optional, Tuple
 
 import click
 from prompt_toolkit.application import Application, get_app
@@ -13,9 +14,17 @@ from prompt_toolkit.layout.dimension import LayoutDimension as D
 from prompt_toolkit.mouse_events import MouseEventType
 from prompt_toolkit.patch_stdout import patch_stdout as pt_patch_stdout
 from prompt_toolkit.shortcuts import PromptSession
-from prompt_toolkit.styles import Style
+from prompt_toolkit.styles import Style as PromptStyle
 from pydantic import BaseModel, Field
 
+try:
+    from colorama import Back, Fore, Style, init
+
+    init(autoreset=True)
+    COLORAMA_AVAILABLE = True
+except ImportError:
+    COLORAMA_AVAILABLE = False
+
 
 class TinybirdAgentContext(BaseModel):
     folder: str
@@ -32,7 +41,7 @@ class TinybirdAgentContext(BaseModel):
     dangerously_skip_permissions: bool
 
 
-default_style = Style.from_dict(
+default_style = PromptStyle.from_dict(
     {
         "separator": "#6C6C6C",
         "questionmark": "#FF9D00 bold",
@@ -388,3 +397,283 @@ class Datafile(BaseModel):
     description: str
     pathname: str
     dependencies: List[str] = Field(default_factory=list)
+
+
+def create_terminal_box(content: str, new_content: Optional[str] = None, title: Optional[str] = None) -> str:
+    """
+    Create a formatted box with automatic line numbers that fills the terminal width.
+    Optionally shows a diff between content and new_content.
+
+    Args:
+        content: The original text content to display in the box (without line numbers)
+        new_content: Optional new content to show as a diff against the original
+        title: Optional title to display as header, if not provided will use first line of content
+
+    Returns:
+        A string containing the formatted box with line numbers added
+    """
+    # Get terminal width, default to 80 if can't determine
+    try:
+        terminal_width = os.get_terminal_size().columns
+    except:
+        terminal_width = 80
+
+    # Box characters
+    top_left = "╭"
+    top_right = "╮"
+    bottom_left = "╰"
+    bottom_right = "╯"
+    horizontal = "─"
+    vertical = "│"
+
+    # Calculate available width for content (terminal_width - 2 borders - 2 spaces padding)
+    available_width = terminal_width - 4
+
+    # Split content into lines
+    lines = content.strip().split("\n")
+    new_lines = new_content.strip().split("\n") if new_content else []
+
+    # Check if we have a title parameter or should use first line as header
+    header = title
+    content_lines = lines
+    new_content_lines = new_lines
+
+    if header is None and lines:
+        # Use first line as header if no title provided
+        header = lines[0]
+        content_lines = lines[1:] if len(lines) > 1 else []
+        if new_lines:
+            # Skip header in new content too
+            new_content_lines = new_lines[1:] if len(new_lines) > 1 else []
+    elif header is not None:
+        # Title provided, use all content lines as-is
+        content_lines = lines
+        new_content_lines = new_lines
+
+    # Process content lines
+    processed_lines = []
+
+    if new_content is None:
+        # No diff, just add line numbers
+        line_number = 1
+        for line in content_lines:
+            processed_lines.extend(_process_line(line, line_number, available_width, None))
+            line_number += 1
+    else:
+        # Create diff and process it properly
+        diff = list(
+            difflib.unified_diff(
+                content_lines,
+                new_content_lines,
+                lineterm="",
+                n=3,  # Add some context lines
+            )
+        )
+
+        # Process the unified diff output
+        old_line_num = 1
+        new_line_num = 1
+        old_index = 0
+        new_index = 0
+
+        # Parse the diff output
+        i = 0
+        while i < len(diff):
+            line = diff[i]
+            if line.startswith("@@"):
+                # Parse hunk header to get line numbers
+                import re
+
+                match = re.match(r"@@ -(\d+)(?:,\d+)? \+(\d+)(?:,\d+)? @@", line)
+                if match:
+                    old_line_num = int(match.group(1))
+                    new_line_num = int(match.group(2))
+                    old_index = old_line_num - 1
+                    new_index = new_line_num - 1
+            elif line.startswith("---") or line.startswith("+++"):
+                # Skip file headers
+                pass
+            elif line.startswith(" "):
+                # Context line (unchanged)
+                content = line[1:]  # Remove the leading space
+                if old_index < len(content_lines) and content_lines[old_index] == content:
+                    processed_lines.extend(_process_line(content, old_line_num, available_width, None))
+                    old_line_num += 1
+                    new_line_num += 1
+                    old_index += 1
+                    new_index += 1
+            elif line.startswith("-"):
+                # Removed line
+                content = line[1:]  # Remove the leading minus
+                processed_lines.extend(_process_line(content, old_line_num, available_width, "-"))
+                old_line_num += 1
+                old_index += 1
+            elif line.startswith("+"):
+                # Added line
+                content = line[1:]  # Remove the leading plus
+                processed_lines.extend(_process_line(content, new_line_num, available_width, "+"))
+                new_line_num += 1
+                new_index += 1
+            i += 1
+
+        # Add any remaining unchanged lines that weren't in the diff
+        while old_index < len(content_lines) and new_index < len(new_content_lines):
+            if content_lines[old_index] == new_content_lines[new_index]:
+                processed_lines.extend(_process_line(content_lines[old_index], old_line_num, available_width, None))
+                old_line_num += 1
+                new_line_num += 1
+                old_index += 1
+                new_index += 1
+            else:
+                break
+
+    # Build the box
+    result = []
+
+    # Top border
+    result.append(top_left + horizontal * (terminal_width - 2) + top_right)
+
+    # Add header if exists
+    if header:
+        # Center the header
+        header_padding = (available_width - len(header)) // 2
+        header_line = (
+            vertical
+            + " "
+            + " " * header_padding
+            + header
+            + " " * (available_width - len(header) - header_padding)
+            + " "
+            + vertical
+        )
+        result.append(header_line)
+        # Empty line after header
+        result.append(vertical + " " * (terminal_width - 2) + vertical)
+
+    # Content lines
+    for line_num, content, diff_marker in processed_lines:
+        if line_num is not None:
+            # Line with number
+            if COLORAMA_AVAILABLE:
+                line_num_str = f"{Fore.LIGHTBLACK_EX}{line_num:>4}{Style.RESET_ALL}"
+            else:
+                line_num_str = f"{line_num:>4}"
+
+            if diff_marker:
+                if COLORAMA_AVAILABLE:
+                    if diff_marker == "-":
+                        # Fill the entire content area with red background
+                        content_with_bg = f"{Back.RED}{diff_marker}   {content}{Style.RESET_ALL}"
+                        # Calculate padding needed for the content area
+                        content_area_width = available_width - 9  # 9 is reduced prefix length
+                        content_padding = content_area_width - len(f"{diff_marker}   {content}")
+                        if content_padding > 0:
+                            content_with_bg = (
+                                f"{Back.RED}{diff_marker}   {content}{' ' * content_padding}{Style.RESET_ALL}"
+                            )
+                        line = f"{vertical} {line_num_str} {content_with_bg}"
+                    elif diff_marker == "+":
+                        # Fill the entire content area with green background
+                        content_with_bg = f"{Back.GREEN}{diff_marker}   {content}{Style.RESET_ALL}"
+                        # Calculate padding needed for the content area
+                        content_area_width = available_width - 9  # 9 is reduced prefix length
+                        content_padding = content_area_width - len(f"{diff_marker}   {content}")
+                        if content_padding > 0:
+                            content_with_bg = (
+                                f"{Back.GREEN}{diff_marker}   {content}{' ' * content_padding}{Style.RESET_ALL}"
+                            )
+                        line = f"{vertical} {line_num_str} {content_with_bg}"
+                else:
+                    line = f"{vertical} {line_num:>4} {diff_marker}   {content}"
+            else:
+                line = f"{vertical} {line_num_str}     {content}"
+        else:
+            # Continuation line without number - fill background starting from where symbol would be
+            if diff_marker and COLORAMA_AVAILABLE:
+                if diff_marker == "-":
+                    # Calculate how much space we need to fill with background
+                    content_area_width = available_width - 9  # 9 is reduced prefix length
+                    content_padding = content_area_width - len(
+                        content
+                    )  # Don't subtract spaces, they're in the background
+                    if content_padding > 0:
+                        line = f"{vertical}      {Back.RED}    {content}{' ' * content_padding}{Style.RESET_ALL}"
+                    else:
+                        line = f"{vertical}      {Back.RED}    {content}{Style.RESET_ALL}"
+                elif diff_marker == "+":
+                    # Calculate how much space we need to fill with background
+                    content_area_width = available_width - 9  # 9 is reduced prefix length
+                    content_padding = content_area_width - len(
+                        content
+                    )  # Don't subtract spaces, they're in the background
+                    if content_padding > 0:
+                        line = f"{vertical}      {Back.GREEN}    {content}{' ' * content_padding}{Style.RESET_ALL}"
+                    else:
+                        line = f"{vertical}      {Back.GREEN}    {content}{Style.RESET_ALL}"
+            else:
+                line = f"{vertical}          {content}"
+
+        # Pad to terminal width
+        # Need to account for ANSI escape sequences not taking visual space
+        if COLORAMA_AVAILABLE:
+            # Calculate visible length (excluding ANSI codes)
+            visible_line = line
+            # Remove all ANSI escape sequences for length calculation
+            import re
+
+            ansi_escape = re.compile(r"\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])")
+            visible_line = ansi_escape.sub("", visible_line)
+            padding_needed = terminal_width - len(visible_line) - 1
+        else:
+            padding_needed = terminal_width - len(line) - 1
+
+        line += " " * padding_needed + vertical
+        result.append(line)
+
+    # Empty line before bottom (only if we have content)
+    if processed_lines:
+        result.append(vertical + " " * (terminal_width - 2) + vertical)
+
+    # Bottom border
+    result.append(bottom_left + horizontal * (terminal_width - 2) + bottom_right)
+
+    return "\n".join(result)
+
+
+def _process_line(
+    line: str, line_number: int, available_width: int, diff_marker: Optional[str]
+) -> List[Tuple[Optional[int], str, Optional[str]]]:
+    """
+    Process a single line, handling wrapping if necessary.
+
+    Returns a list of tuples (line_number, content, diff_marker)
+    """
+    # Calculate space needed for line number and spacing
+    # " 9999     " for normal lines or " 9999 +   " for diff lines
+    prefix_length = 9  # Reduced from 13 to 9
+
+    # Available width for actual content
+    content_width = available_width - prefix_length
+
+    processed: List[Tuple[Optional[int], str, Optional[str]]] = []
+
+    if len(line) <= content_width:
+        # Line fits, add it as is
+        processed.append((line_number, line, diff_marker))
+    else:
+        # Line needs wrapping
+        # First line with line number
+        first_part = line[:content_width]
+        processed.append((line_number, first_part, diff_marker))
+
+        # Remaining wrapped lines without line numbers
+        remaining = line[content_width:]
+        while remaining:
+            if len(remaining) <= content_width:
+                processed.append((None, remaining, diff_marker))
+                break
+            else:
+                processed.append((None, remaining[:content_width], diff_marker))
+                remaining = remaining[content_width:]
+
+    return processed

{tinybird-0.0.1.dev247 → tinybird-0.0.1.dev249}/tinybird/tb/modules/deployment.py

@@ -211,6 +211,7 @@ def deployment_ls(ctx: click.Context, include_deleted: bool) -> None:
     List all the deployments you have in the project.
     """
     client = ctx.ensure_object(dict)["client"]
+    output = ctx.ensure_object(dict)["output"]
 
     TINYBIRD_API_KEY = client.token
     HEADERS = {"Authorization": f"Bearer {TINYBIRD_API_KEY}"}
@@ -243,7 +244,27 @@ def deployment_ls(ctx: click.Context, include_deleted: bool) -> None:
         )
 
     table.reverse()
-    echo_safe_humanfriendly_tables_format_smart_table(table, column_names=columns)
+
+    # Handle different output formats
+    if output == "json":
+        # Create JSON structure
+        deployments_json = []
+        for row in table:
+            deployments_json.append({"id": row[0], "status": row[1], "created_at": row[2]})
+        from tinybird.tb.modules.common import echo_json
+
+        echo_json({"deployments": deployments_json})
+    elif output == "csv":
+        # Create CSV output
+        csv_output = f"{columns[0]},{columns[1]},{columns[2]}\n"
+        for row in table:
+            csv_output += f"{row[0]},{row[1]},{row[2]}\n"
+        from tinybird.tb.modules.common import force_echo
+
+        force_echo(csv_output)
+    else:
+        # Default human-readable output
+        echo_safe_humanfriendly_tables_format_smart_table(table, column_names=columns)
 
 
 @deployment_group.command(name="promote")

tinybird-0.0.1.dev249/tinybird/tb/modules/login.py

@@ -0,0 +1,39 @@
+from typing import Optional
+
+import click
+
+from tinybird.tb.modules.cli import cli
+from tinybird.tb.modules.login_common import login
+
+
+@cli.command("login", help="Authenticate using the browser.")
+@click.option(
+    "--host",
+    type=str,
+    default=None,
+    help="Set custom host if it's different than https://api.europe-west2.gcp.tinybird.co. See https://www.tinybird.co/docs/api-reference/overview#regions-and-endpoints for the available list of regions.",
+)
+@click.option(
+    "--auth-host",
+    default="https://cloud.tinybird.co",
+    help="Set the host to authenticate to. If unset, the default host will be used.",
+)
+@click.option(
+    "--workspace",
+    help="Set the workspace to authenticate to. If unset, the default workspace will be used.",
+)
+@click.option(
+    "-i",
+    "--interactive",
+    is_flag=True,
+    default=False,
+    help="Show available regions and select where to authenticate to",
+)
+@click.option(
+    "--method",
+    type=click.Choice(["browser", "code"]),
+    default="browser",
+    help="Set the authentication method to use. Default: browser.",
+)
+def login_cmd(host: Optional[str], auth_host: str, workspace: str, interactive: bool, method: str):
+    login(host, auth_host, workspace, interactive, method)