tinybird 0.0.1.dev247__py3-none-any.whl → 0.0.1.dev248__py3-none-any.whl

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/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.dev248'
+__revision__ = '8eed30e'

tinybird/tb/modules/agent/agent.py

@@ -54,6 +54,7 @@ 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
 
@@ -75,7 +76,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 +117,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 +219,39 @@ 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)
 
     try:
-        token = config["token"]
-        host = config["host"]
+        if not token or not host:
+            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()
+                host = cli_config.get_host()
+
+            if not token or not host:
+                click.echo(
+                    FeedbackManager.error(message="Tinybird Code requires authentication. Run 'tb login' first.")
+                )
+                return
+
+        display_banner()
         agent = TinybirdAgent(token, host, 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 +263,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(

tinybird/tb/modules/agent/tools/create_datafile.py

@@ -1,87 +1,13 @@
-import difflib
 from pathlib import Path
 
-try:
-    from colorama import Back, Fore, Style, init
-
-    init()
-except ImportError:  # fallback so that the imported classes always exist
-
-    class ColorFallback:
-        def __getattr__(self, name):
-            return ""
-
-    Fore = Back = Style = ColorFallback()
-
 import click
 from pydantic_ai import RunContext
 
-from tinybird.tb.modules.agent.utils import Datafile, TinybirdAgentContext, show_options
+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 create_line_numbered_diff(original_content: str, new_content: str, filename: str) -> str:
-    """Create a diff with line numbers similar to the example format"""
-    original_lines = original_content.splitlines()
-    new_lines = new_content.splitlines()
-
-    # Create a SequenceMatcher to find the differences
-    matcher = difflib.SequenceMatcher(None, original_lines, new_lines)
-
-    result = []
-    result.append(f"╭{'─' * 88}╮")
-    result.append(f"│ {filename:<86} │")
-    result.append(f"│{' ' * 88}│")
-
-    # Process the opcodes to build the diff
-    for tag, i1, i2, j1, j2 in matcher.get_opcodes():
-        if tag == "equal":
-            # Show context lines
-            for i, line in enumerate(original_lines[i1:i2]):
-                line_num = i1 + i + 1
-                result.append(f"│ {line_num:4}        {line:<74} │")
-        elif tag == "replace":
-            # Show removed lines
-            for i, line in enumerate(original_lines[i1:i2]):
-                line_num = i1 + i + 1
-                result.append(f"│ {Back.RED}{line_num:4} -      {line:<74}{Back.RESET} │")
-            # Show added lines
-            for i, line in enumerate(new_lines[j1:j2]):
-                line_num = i1 + i + 1
-                result.append(f"│ {Back.GREEN}{line_num:4} +      {line:<74}{Back.RESET} │")
-        elif tag == "delete":
-            # Show removed lines
-            for i, line in enumerate(original_lines[i1:i2]):
-                line_num = i1 + i + 1
-                result.append(f"│ {Back.RED}{line_num:4} -      {line:<74}{Back.RESET} │")
-        elif tag == "insert":
-            # Show added lines
-            for i, line in enumerate(new_lines[j1:j2]):
-                # Use the line number from the original position
-                line_num = i1 + i + 1
-                result.append(f"│ {Back.GREEN}{line_num:4} +      {line:<74}{Back.RESET} │")
-
-    result.append(f"╰{'─' * 88}╯")
-    return "\n".join(result)
-
-
-def create_line_numbered_content(content: str, filename: str) -> str:
-    """Create a formatted display of file content with line numbers"""
-    lines = content.splitlines()
-
-    result = []
-    result.append(f"╭{'─' * 88}╮")
-    result.append(f"│ {filename:<86} │")
-    result.append(f"│{' ' * 88}│")
-
-    for i, line in enumerate(lines, 1):
-        result.append(f"│ {i:4}        {line:<74} │")
-
-    result.append(f"╰{'─' * 88}╯")
-    return "\n".join(result)
-
-
 def get_resource_confirmation(resource: Datafile, exists: bool) -> bool:
     """Get user confirmation for creating a resource"""
     while True:
@@ -118,9 +44,9 @@ def create_datafile(ctx: RunContext[TinybirdAgentContext], resource: Datafile) -
         content = resource.content
         exists = str(path) in ctx.deps.get_project_files()
         if exists:
-            content = create_line_numbered_diff(path.read_text(), resource.content, resource.pathname)
+            content = create_terminal_box(path.read_text(), resource.content, title=resource.pathname)
         else:
-            content = create_line_numbered_content(resource.content, resource.pathname)
+            content = create_terminal_box(resource.content, title=resource.pathname)
         click.echo(content)
         confirmation = ctx.deps.dangerously_skip_permissions or get_resource_confirmation(resource, exists)
 

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/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/tb/modules/login.py

@@ -1,112 +1,12 @@
-import http.server
-import os
-import platform
-import random
-import shutil
-import socketserver
-import string
-import subprocess
-import sys
-import threading
-import time
-import urllib.parse
-import webbrowser
-from typing import Any, Dict, Optional
-from urllib.parse import urlencode
+from typing import Optional
 
 import click
-import requests
 
-from tinybird.tb.config import DEFAULT_API_HOST
-from tinybird.tb.modules.cli import CLIConfig, cli
-from tinybird.tb.modules.common import ask_for_region_interactively, get_regions
-from tinybird.tb.modules.exceptions import CLILoginException
-from tinybird.tb.modules.feedback_manager import FeedbackManager
+from tinybird.tb.modules.cli import cli
+from tinybird.tb.modules.login_common import login
 
-SERVER_MAX_WAIT_TIME = 180
 
-
-class AuthHandler(http.server.SimpleHTTPRequestHandler):
-    def do_GET(self):
-        # The access_token is in the URL fragment, which is not sent to the server
-        # We'll send a small HTML page that extracts the token and sends it back to the server
-        self.send_response(200)
-        self.send_header("Content-type", "text/html")
-        self.end_headers()
-        self.wfile.write(
-            """
-        <html>
-        <head>
-            <style>
-                body {{
-                    font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif;
-                    background: #f5f5f5;
-                    display: flex;
-                    align-items: center;
-                    justify-content: center;
-                    height: 100vh;
-                    margin: 0;
-                }}
-            </style>
-        </head>
-        <body>
-            <script>
-                const searchParams = new URLSearchParams(window.location.search);
-                const code = searchParams.get('code');
-                const workspace = searchParams.get('workspace');
-                const region = searchParams.get('region');
-                const provider = searchParams.get('provider');
-                const host = "{auth_host}";
-                fetch('/?code=' + code, {{method: 'POST'}})
-                    .then(() => {{
-                        window.location.href = host + "/" + provider + "/" + region + "/cli-login?workspace=" + workspace;
-                    }});
-            </script>
-        </body>
-        </html>
-        """.format(auth_host=self.server.auth_host).encode()  # type: ignore
-        )
-
-    def do_POST(self):
-        parsed_path = urllib.parse.urlparse(self.path)
-        query_params = urllib.parse.parse_qs(parsed_path.query)
-
-        if "code" in query_params:
-            code = query_params["code"][0]
-            self.server.auth_callback(code)  # type: ignore
-            self.send_response(200)
-            self.end_headers()
-        else:
-            self.send_error(400, "Missing 'code' parameter")
-
-        self.server.shutdown()
-
-    def log_message(self, format, *args):
-        # Suppress log messages
-        return
-
-
-AUTH_SERVER_PORT = 49160
-
-
-class AuthServer(socketserver.TCPServer):
-    allow_reuse_address = True
-
-    def __init__(self, server_address, RequestHandlerClass, auth_callback, auth_host):
-        self.auth_callback = auth_callback
-        self.auth_host = auth_host
-        super().__init__(server_address, RequestHandlerClass)
-
-
-def start_server(auth_callback, auth_host):
-    with AuthServer(("", AUTH_SERVER_PORT), AuthHandler, auth_callback, auth_host) as httpd:
-        httpd.timeout = 30
-        start_time = time.time()
-        while time.time() - start_time < SERVER_MAX_WAIT_TIME:  # Run for a maximum of 180 seconds
-            httpd.handle_request()
-
-
-@cli.command()
+@cli.command("login", help="Authenticate using the browser.")
 @click.option(
     "--host",
     type=str,
@@ -135,200 +35,5 @@ def start_server(auth_callback, auth_host):
     default="browser",
     help="Set the authentication method to use. Default: browser.",
 )
-def login(host: Optional[str], auth_host: str, workspace: str, interactive: bool, method: str):
-    """Authenticate using the browser."""
-    try:
-        cli_config = CLIConfig.get_project_config()
-        if not host and cli_config.get_token():
-            host = cli_config.get_host(use_defaults_if_needed=False)
-        if not host or interactive:
-            if interactive:
-                click.echo(FeedbackManager.highlight(message="» Select one region from the list below:"))
-            else:
-                click.echo(FeedbackManager.highlight(message="» No region detected, select one from the list below:"))
-
-            regions = get_regions(cli_config)
-            selected_region = ask_for_region_interactively(regions)
-
-            # If the user cancels the selection, we'll exit
-            if not selected_region:
-                sys.exit(1)
-            host = selected_region.get("api_host")
-
-        if not host:
-            host = DEFAULT_API_HOST
-
-        host = host.rstrip("/")
-        auth_host = auth_host.rstrip("/")
-
-        if method == "code":
-            display_code, one_time_code = create_one_time_code()
-            click.echo(FeedbackManager.info(message=f"First, copy your one-time code: {display_code}"))
-            click.echo(FeedbackManager.info(message="Press [Enter] to continue in the browser..."))
-            input()
-            click.echo(FeedbackManager.highlight(message="» Opening browser for authentication..."))
-            params = {
-                "apiHost": host,
-                "code": one_time_code,
-                "method": "code",
-            }
-            auth_url = f"{auth_host}/api/cli-login?{urlencode(params)}"
-            open_url(auth_url)
-            click.echo(
-                FeedbackManager.info(message="\nIf browser does not open, please open the following URL manually:")
-            )
-            click.echo(FeedbackManager.info(message=auth_url))
-
-            def poll_for_tokens():
-                while True:
-                    params = {
-                        "apiHost": host,
-                        "cliCode": one_time_code,
-                        "method": "code",
-                    }
-                    response = requests.get(f"{auth_host}/api/cli-login?{urlencode(params)}")
-
-                    try:
-                        if response.status_code == 200:
-                            data = response.json()
-                            user_token = data.get("user_token", "")
-                            workspace_token = data.get("workspace_token", "")
-                            if user_token and workspace_token:
-                                authenticate_with_tokens(data, host, cli_config)
-                                break
-                    except Exception:
-                        pass
-
-                    time.sleep(2)
-
-            poll_for_tokens()
-            return
-
-        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
-        params = {
-            "apiHost": host,
-        }
-
-        if workspace:
-            params["workspace"] = workspace
-
-        auth_url = f"{auth_host}/api/cli-login?{urlencode(params)}"
-        open_url(auth_url)
-
-        click.echo(FeedbackManager.info(message="\nIf browser does not open, please open the following URL manually:"))
-        click.echo(FeedbackManager.info(message=auth_url))
-
-        # Wait for the authentication to complete or timeout
-        if auth_event.wait(timeout=SERVER_MAX_WAIT_TIME):  # Wait for up to 180 seconds
-            params = {}
-            params["code"] = auth_code[0]
-            response = requests.get(
-                f"{auth_host}/api/cli-login?{urlencode(params)}",
-            )
-
-            data = response.json()
-            authenticate_with_tokens(data, host, cli_config)
-        else:
-            raise Exception("Authentication failed or timed out.")
-    except Exception as e:
-        raise CLILoginException(FeedbackManager.error(message=str(e)))
-
-
-def _running_in_wsl() -> bool:
-    """Return True when Python is executing inside a WSL distro."""
-    # Fast positive check (modern WSL always sets at least one of these):
-    if "WSL_DISTRO_NAME" in os.environ or "WSL_INTEROP" in os.environ:
-        return True
-
-    # Fall back to kernel /proc data
-    release = platform.uname().release.lower()
-    if "microsoft" in release:  # covers stock WSL kernels
-        return True
-    try:
-        if "microsoft" in open("/proc/version").read().lower():
-            return True
-    except FileNotFoundError:
-        pass
-    return False
-
-
-def open_url(url: str, *, new_tab: bool = False) -> bool:
-    # 1. Try the standard library first on CPython ≥ 3.11 this already
-    #    recognises WSL and fires up the Windows default browser for us.
-    try:
-        wb: Any = webbrowser.get()  # mypy: Any for Py < 3.10
-        if new_tab:
-            if wb.open_new_tab(url):
-                return True
-        else:
-            if wb.open(url):
-                return True
-    except webbrowser.Error:
-        pass  # keep going
-
-    # 2. Inside WSL, prefer `wslview` if the user has it (wslu package).
-    if _running_in_wsl() and shutil.which("wslview"):
-        subprocess.Popen(["wslview", url])
-        return True
-
-    # 3. Secondary WSL fallback use Windows **start** through cmd.exe.
-    #    Empty "" argument is required so long URLs are not treated as a window title.
-    if _running_in_wsl():
-        subprocess.Popen(["cmd.exe", "/c", "start", "", url])
-        return True
-
-    # 4. Unix last-ditch fallback xdg-open (most minimal container images have it)
-    if shutil.which("xdg-open"):
-        subprocess.Popen(["xdg-open", url])
-        return True
-
-    # 5. If everything failed, let the caller know.
-    return False
-
-
-def create_one_time_code():
-    """Create a random one-time code for the authentication process in the format of A2C4-D2G4 (only uppercase letters and digits)"""
-    seperator = "-"
-    full_code = "".join(random.choices(string.ascii_uppercase + string.digits, k=8))
-    parts = [full_code[:4], full_code[4:]]
-    return seperator.join(parts), full_code
-
-
-def authenticate_with_tokens(data: Dict[str, Any], host: Optional[str], cli_config: CLIConfig):
-    cli_config.set_token(data.get("workspace_token", ""))
-    host = host or data.get("api_host", "")
-    cli_config.set_token_for_host(data.get("workspace_token", ""), host)
-    cli_config.set_user_token(data.get("user_token", ""))
-    cli_config.set_host(host)
-    ws = 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] = 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!"))
+def login_cmd(host: Optional[str], auth_host: str, workspace: str, interactive: bool, method: str):
+    login(host, auth_host, workspace, interactive, method)

tinybird/tb/modules/login_common.py

@@ -0,0 +1,310 @@
+import http.server
+import os
+import platform
+import random
+import shutil
+import socketserver
+import string
+import subprocess
+import sys
+import threading
+import time
+import urllib.parse
+import webbrowser
+from typing import Any, Dict, Optional
+from urllib.parse import urlencode
+
+import click
+import requests
+
+from tinybird.tb.config import DEFAULT_API_HOST
+from tinybird.tb.modules.common import ask_for_region_interactively, get_regions
+from tinybird.tb.modules.config import CLIConfig
+from tinybird.tb.modules.exceptions import CLILoginException
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+
+SERVER_MAX_WAIT_TIME = 180
+
+
+class AuthHandler(http.server.SimpleHTTPRequestHandler):
+    def do_GET(self):
+        # The access_token is in the URL fragment, which is not sent to the server
+        # We'll send a small HTML page that extracts the token and sends it back to the server
+        self.send_response(200)
+        self.send_header("Content-type", "text/html")
+        self.end_headers()
+        self.wfile.write(
+            """
+        <html>
+        <head>
+            <style>
+                body {{
+                    font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif;
+                    background: #f5f5f5;
+                    display: flex;
+                    align-items: center;
+                    justify-content: center;
+                    height: 100vh;
+                    margin: 0;
+                }}
+            </style>
+        </head>
+        <body>
+            <script>
+                const searchParams = new URLSearchParams(window.location.search);
+                const code = searchParams.get('code');
+                const workspace = searchParams.get('workspace');
+                const region = searchParams.get('region');
+                const provider = searchParams.get('provider');
+                const host = "{auth_host}";
+                fetch('/?code=' + code, {{method: 'POST'}})
+                    .then(() => {{
+                        window.location.href = host + "/" + provider + "/" + region + "/cli-login?workspace=" + workspace;
+                    }});
+            </script>
+        </body>
+        </html>
+        """.format(auth_host=self.server.auth_host).encode()  # type: ignore
+        )
+
+    def do_POST(self):
+        parsed_path = urllib.parse.urlparse(self.path)
+        query_params = urllib.parse.parse_qs(parsed_path.query)
+
+        if "code" in query_params:
+            code = query_params["code"][0]
+            self.server.auth_callback(code)  # type: ignore
+            self.send_response(200)
+            self.end_headers()
+        else:
+            self.send_error(400, "Missing 'code' parameter")
+
+        self.server.shutdown()
+
+    def log_message(self, format, *args):
+        # Suppress log messages
+        return
+
+
+AUTH_SERVER_PORT = 49160
+
+
+class AuthServer(socketserver.TCPServer):
+    allow_reuse_address = True
+
+    def __init__(self, server_address, RequestHandlerClass, auth_callback, auth_host):
+        self.auth_callback = auth_callback
+        self.auth_host = auth_host
+        super().__init__(server_address, RequestHandlerClass)
+
+
+def start_server(auth_callback, auth_host):
+    with AuthServer(("", AUTH_SERVER_PORT), AuthHandler, auth_callback, auth_host) as httpd:
+        httpd.timeout = 30
+        start_time = time.time()
+        while time.time() - start_time < SERVER_MAX_WAIT_TIME:  # Run for a maximum of 180 seconds
+            httpd.handle_request()
+
+
+def login(
+    host: Optional[str],
+    auth_host: str = "https://cloud.tinybird.co",
+    workspace: Optional[str] = None,
+    interactive: bool = False,
+    method: str = "browser",
+):
+    try:
+        cli_config = CLIConfig.get_project_config()
+        if not host and cli_config.get_token():
+            host = cli_config.get_host(use_defaults_if_needed=False)
+        if not host or interactive:
+            if interactive:
+                click.echo(FeedbackManager.highlight(message="» Select one region from the list below:"))
+            else:
+                click.echo(FeedbackManager.highlight(message="» No region detected, select one from the list below:"))
+
+            regions = get_regions(cli_config)
+            selected_region = ask_for_region_interactively(regions)
+
+            # If the user cancels the selection, we'll exit
+            if not selected_region:
+                sys.exit(1)
+            host = selected_region.get("api_host")
+
+        if not host:
+            host = DEFAULT_API_HOST
+
+        host = host.rstrip("/")
+        auth_host = auth_host.rstrip("/")
+
+        if method == "code":
+            display_code, one_time_code = create_one_time_code()
+            click.echo(FeedbackManager.info(message=f"First, copy your one-time code: {display_code}"))
+            click.echo(FeedbackManager.info(message="Press [Enter] to continue in the browser..."))
+            input()
+            click.echo(FeedbackManager.highlight(message="» Opening browser for authentication..."))
+            params = {
+                "apiHost": host,
+                "code": one_time_code,
+                "method": "code",
+            }
+            auth_url = f"{auth_host}/api/cli-login?{urlencode(params)}"
+            open_url(auth_url)
+            click.echo(
+                FeedbackManager.info(message="\nIf browser does not open, please open the following URL manually:")
+            )
+            click.echo(FeedbackManager.info(message=auth_url))
+
+            def poll_for_tokens():
+                while True:
+                    params = {
+                        "apiHost": host,
+                        "cliCode": one_time_code,
+                        "method": "code",
+                    }
+                    response = requests.get(f"{auth_host}/api/cli-login?{urlencode(params)}")
+
+                    try:
+                        if response.status_code == 200:
+                            data = response.json()
+                            user_token = data.get("user_token", "")
+                            workspace_token = data.get("workspace_token", "")
+                            if user_token and workspace_token:
+                                authenticate_with_tokens(data, host, cli_config)
+                                break
+                    except Exception:
+                        pass
+
+                    time.sleep(2)
+
+            poll_for_tokens()
+            return
+
+        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
+        params = {
+            "apiHost": host,
+        }
+
+        if workspace:
+            params["workspace"] = workspace
+
+        auth_url = f"{auth_host}/api/cli-login?{urlencode(params)}"
+        open_url(auth_url)
+
+        click.echo(FeedbackManager.info(message="\nIf browser does not open, please open the following URL manually:"))
+        click.echo(FeedbackManager.info(message=auth_url))
+
+        # Wait for the authentication to complete or timeout
+        if auth_event.wait(timeout=SERVER_MAX_WAIT_TIME):  # Wait for up to 180 seconds
+            params = {}
+            params["code"] = auth_code[0]
+            response = requests.get(
+                f"{auth_host}/api/cli-login?{urlencode(params)}",
+            )
+
+            data = response.json()
+            authenticate_with_tokens(data, host, cli_config)
+        else:
+            raise Exception("Authentication failed or timed out.")
+    except Exception as e:
+        raise CLILoginException(FeedbackManager.error(message=str(e)))
+
+
+def _running_in_wsl() -> bool:
+    """Return True when Python is executing inside a WSL distro."""
+    # Fast positive check (modern WSL always sets at least one of these):
+    if "WSL_DISTRO_NAME" in os.environ or "WSL_INTEROP" in os.environ:
+        return True
+
+    # Fall back to kernel /proc data
+    release = platform.uname().release.lower()
+    if "microsoft" in release:  # covers stock WSL kernels
+        return True
+    try:
+        if "microsoft" in open("/proc/version").read().lower():
+            return True
+    except FileNotFoundError:
+        pass
+    return False
+
+
+def open_url(url: str, *, new_tab: bool = False) -> bool:
+    # 1. Try the standard library first on CPython ≥ 3.11 this already
+    #    recognises WSL and fires up the Windows default browser for us.
+    try:
+        wb: Any = webbrowser.get()  # mypy: Any for Py < 3.10
+        if new_tab:
+            if wb.open_new_tab(url):
+                return True
+        else:
+            if wb.open(url):
+                return True
+    except webbrowser.Error:
+        pass  # keep going
+
+    # 2. Inside WSL, prefer `wslview` if the user has it (wslu package).
+    if _running_in_wsl() and shutil.which("wslview"):
+        subprocess.Popen(["wslview", url])
+        return True
+
+    # 3. Secondary WSL fallback use Windows **start** through cmd.exe.
+    #    Empty "" argument is required so long URLs are not treated as a window title.
+    if _running_in_wsl():
+        subprocess.Popen(["cmd.exe", "/c", "start", "", url])
+        return True
+
+    # 4. Unix last-ditch fallback xdg-open (most minimal container images have it)
+    if shutil.which("xdg-open"):
+        subprocess.Popen(["xdg-open", url])
+        return True
+
+    # 5. If everything failed, let the caller know.
+    return False
+
+
+def create_one_time_code():
+    """Create a random one-time code for the authentication process in the format of A2C4-D2G4 (only uppercase letters and digits)"""
+    seperator = "-"
+    full_code = "".join(random.choices(string.ascii_uppercase + string.digits, k=8))
+    parts = [full_code[:4], full_code[4:]]
+    return seperator.join(parts), full_code
+
+
+def authenticate_with_tokens(data: Dict[str, Any], host: Optional[str], cli_config: CLIConfig):
+    cli_config.set_token(data.get("workspace_token", ""))
+    host = host or data.get("api_host", "")
+    cli_config.set_token_for_host(data.get("workspace_token", ""), host)
+    cli_config.set_user_token(data.get("user_token", ""))
+    cli_config.set_host(host)
+    ws = 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] = 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!"))

{tinybird-0.0.1.dev247.dist-info → tinybird-0.0.1.dev248.dist-info}/METADATA

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

{tinybird-0.0.1.dev247.dist-info → tinybird-0.0.1.dev248.dist-info}/RECORD

@@ -3,7 +3,7 @@ tinybird/context.py,sha256=FfqYfrGX_I7PKGTQo93utaKPDNVYWelg4Hsp3evX5wM,1291
 tinybird/datatypes.py,sha256=r4WCvspmrXTJHiPjjyOTiZyZl31FO3Ynkwq4LQsYm6E,11059
 tinybird/feedback_manager.py,sha256=1INQFfRfuMCb9lfB8KNf4r6qC2khW568hoHjtk-wshI,69305
 tinybird/git_settings.py,sha256=Sw_8rGmribEFJ4Z_6idrVytxpFYk7ez8ei0qHULzs3E,3934
-tinybird/prompts.py,sha256=dg8Q8Q29X69j-PPP8KGooMz7-M3FNXEYqC0Fj0LcUbw,45316
+tinybird/prompts.py,sha256=Dx9pD5kyXdbzkGOcWdULWUg_aN5F1CUPAISm5PyMzOM,45498
 tinybird/sql.py,sha256=BufnOgclQokDyihtuXesOwHBsebN6wRXIxO5wKRkOwE,48299
 tinybird/sql_template.py,sha256=LChDztXUUrNO4Qukv2RMsdjQ-vhmepWiHVoX6yr140E,99983
 tinybird/sql_template_fmt.py,sha256=KUHdj5rYCYm_rKKdXYSJAE9vIyXUQLB0YSZnUXHeBlY,10196
@@ -17,7 +17,7 @@ tinybird/datafile/exceptions.py,sha256=8rw2umdZjtby85QbuRKFO5ETz_eRHwUY5l7eHsy1w
 tinybird/datafile/parse_connection.py,sha256=tRyn2Rpr1TeWet5BXmMoQgaotbGdYep1qiTak_OqC5E,1825
 tinybird/datafile/parse_datasource.py,sha256=ssW8QeFSgglVFi3sDZj_HgkJiTJ2069v2JgqnH3CkDE,1825
 tinybird/datafile/parse_pipe.py,sha256=xf4m0Tw44QWJzHzAm7Z7FwUoUUtr7noMYjU1NiWnX0k,3880
-tinybird/tb/__cli__.py,sha256=vsGGI9kP2i8TgwhV7ZNt0lpfebdoTNrbxYl00oYXMjE,247
+tinybird/tb/__cli__.py,sha256=XU7dQJoJUS04mQcPTcpDweJQbUllPEGhxSfkfVitr2Y,247
 tinybird/tb/check_pypi.py,sha256=Gp0HkHHDFMSDL6nxKlOY51z7z1Uv-2LRexNTZSHHGmM,552
 tinybird/tb/cli.py,sha256=FdDFEIayjmsZEVsVSSvRiVYn_FHOVg_zWQzchnzfWho,1008
 tinybird/tb/client.py,sha256=pJbdkWMXGAqKseNAvdsRRnl_c7I-DCMB0dWCQnG82nU,54146
@@ -46,7 +46,8 @@ tinybird/tb/modules/llm.py,sha256=CpTq2YAk88E8ENpQA94-mas3UDN1aqa--9Al8GdwQtc,15
 tinybird/tb/modules/llm_utils.py,sha256=nS9r4FAElJw8yXtmdYrx-rtI2zXR8qXfi1QqUDCfxvg,3469
 tinybird/tb/modules/local.py,sha256=tpiw_F_qOIp42h3kTBwTm5GQDyuVLF0QNF1jmB0zR94,6845
 tinybird/tb/modules/local_common.py,sha256=_WODjW3oPshgsZ1jDFFx2nr0zrLi3Gxz5dlahWPobM8,17464
-tinybird/tb/modules/login.py,sha256=glqj5RWH26AseEoBl8XfrSDEjQTdko17i_pVWOIMoGc,12497
+tinybird/tb/modules/login.py,sha256=zerXZqIv15pbFk5XRt746xGcVnp01YmL_403byBf4jQ,1245
+tinybird/tb/modules/login_common.py,sha256=IfthYbHmC7EtsCXCB1iF4TngPOwfaHJ6Dfi_t7oBXnI,11640
 tinybird/tb/modules/logout.py,sha256=sniI4JNxpTrVeRCp0oGJuQ3yRerG4hH5uz6oBmjv724,1009
 tinybird/tb/modules/materialization.py,sha256=0O2JUCxLzz-DrXTUewVHlIyC6-Kyymw0hGXXDicMSHE,5403
 tinybird/tb/modules/mock.py,sha256=ET8sRpmXnQsd2sSJXH_KCdREU1_XQgkORru6T357Akc,3260
@@ -67,24 +68,24 @@ tinybird/tb/modules/watch.py,sha256=No0bK1M1_3CYuMaIgylxf7vYFJ72lTJe3brz6xQ-mJo,
 tinybird/tb/modules/workspace.py,sha256=Q_8HcxMsNg8QG9aBlwcWS2umrDP5IkTIHqqz3sfmGuc,11341
 tinybird/tb/modules/workspace_members.py,sha256=5JdkJgfuEwbq-t6vxkBhYwgsiTDxF790wsa6Xfif9nk,8608
 tinybird/tb/modules/agent/__init__.py,sha256=i3oe3vDIWWPaicdCM0zs7D7BJ1W0k7th93ooskHAV00,54
-tinybird/tb/modules/agent/agent.py,sha256=3mP1HwjEGWuyuUByyA7xe2FrW2l-ZiChRKDxMs7vJIg,16887
+tinybird/tb/modules/agent/agent.py,sha256=8NBSwQuu3AagieLfbNOXSWeud3WGyThBuyPuzZTtxJQ,17932
 tinybird/tb/modules/agent/animations.py,sha256=z0MNLf8TnUO8qAjgYvth_wc9a9283pNVz1Z4jl15Ggs,2558
 tinybird/tb/modules/agent/banner.py,sha256=KX_e467uiy1gWOZ4ofTZt0GCFGQqHQ_8Ob27XLQqda0,3053
 tinybird/tb/modules/agent/memory.py,sha256=H6SJK--2L5C87B7AJd_jMqsq3sCvFvZwZXmajuT0GBE,1171
 tinybird/tb/modules/agent/models.py,sha256=Of74wcU8oX05ricTqmhMHVHfeYo_pQbnbCI_q3mlx5E,682
 tinybird/tb/modules/agent/prompts.py,sha256=fZMTbTbq8SHWob8-wA5fQFnZ9lJa7Y_66_9JvJT3xuc,6818
-tinybird/tb/modules/agent/utils.py,sha256=K0mszvO27wqsmxjWZOjrshUIVb3qxpYr1hM_enlbrxw,13441
+tinybird/tb/modules/agent/utils.py,sha256=7Y8bq_rZlqre8_OvLVjIvE8ZLOBpuKmXamyaNs02zzc,25231
 tinybird/tb/modules/agent/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tinybird/tb/modules/agent/tools/append.py,sha256=2q5y32jeNHJgbMsVmABn7y-KCoKwkqbMFJAm8OZ4zQc,2161
 tinybird/tb/modules/agent/tools/build.py,sha256=LhzJMx6tbxC7gogIrxhfKJc-SDgoSR-FC6IunfaCdn8,758
-tinybird/tb/modules/agent/tools/create_datafile.py,sha256=IeLB81HNQ5k8fy1x3ZDkKbi0iPqSMZz-xNi2Kga0I1M,5431
+tinybird/tb/modules/agent/tools/create_datafile.py,sha256=ZVLj5VT8udfYDqfbYu9U3f2IG2wSB3POwC-zB_JvfsA,2692
 tinybird/tb/modules/agent/tools/deploy.py,sha256=Vv1SHalxZsl5QttaON0jBwJenj1cVOQiQ-cMiK2ULZg,1443
 tinybird/tb/modules/agent/tools/deploy_check.py,sha256=VqMYC7l3_cihmmM_pi8w1t8rJ3P0xDc7pHs_st9k-9Q,684
 tinybird/tb/modules/agent/tools/explore.py,sha256=ihALc_kBcsjrKT3hZyicqyIowB0g_K3AtNNi-5uz9-8,412
 tinybird/tb/modules/agent/tools/mock.py,sha256=c4fY8_D92tOUBr0DoqoA5lEE3FgvUQHP6JE75mfTBko,2491
 tinybird/tb/modules/agent/tools/plan.py,sha256=wQY4gNtFTOEy2yZUGf8VqefPUbbz5DgMZdrzGRk-wiE,1365
 tinybird/tb/modules/agent/tools/preview_datafile.py,sha256=e9q5fR0afApcrntzFrnuHmd10ex7MG_GM6T0Pwc9bRI,850
-tinybird/tb/modules/agent/tools/read_fixture_data.py,sha256=QCmdXccVVRqkDYmUdPst94qr29dmlE20KSMyBHpzDjs,961
+tinybird/tb/modules/agent/tools/read_fixture_data.py,sha256=XgeDld6YTOjWNcJ7cr8bHD2phG6W-h5UuC2amGSBnQw,977
 tinybird/tb/modules/datafile/build.py,sha256=NFKBrusFLU0WJNCXePAFWiEDuTaXpwc0lHlOQWEJ43s,51117
 tinybird/tb/modules/datafile/build_common.py,sha256=2yNdxe49IMA9wNvl25NemY2Iaz8L66snjOdT64dm1is,4511
 tinybird/tb/modules/datafile/build_datasource.py,sha256=Ra8pVQBDafbFRUKlhpgohhTsRyp_ADKZJVG8Gd69idY,17227
@@ -105,8 +106,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.dev247.dist-info/METADATA,sha256=IBLQijubQLYO67wrCCvUpdLPbEqv1JBnf7RCCEOiKG8,1733
-tinybird-0.0.1.dev247.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev247.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev247.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev247.dist-info/RECORD,,
+tinybird-0.0.1.dev248.dist-info/METADATA,sha256=Q2bYzjGQkmxcw_IutVCHU3qK6Au414Yse04ys2JugeA,1733
+tinybird-0.0.1.dev248.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev248.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev248.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev248.dist-info/RECORD,,