tinybird 0.0.1.dev3__py3-none-any.whl → 0.0.1.dev5__py3-none-any.whl

tinybird/__cli__.py

@@ -4,5 +4,5 @@ __description__ = 'Tinybird Command Line Tool'
 __url__ = 'https://www.tinybird.co/docs/cli/introduction.html'
 __author__ = 'Tinybird'
 __author_email__ = 'support@tinybird.co'
-__version__ = '0.0.1.dev3'
-__revision__ = '50bec29'
+__version__ = '0.0.1.dev5'
+__revision__ = 'b69dd20'

tinybird/check_pypi.py

@@ -6,7 +6,7 @@ from tinybird.feedback_manager import FeedbackManager
 from tinybird.syncasync import sync_to_async
 from tinybird.tb_cli_modules.common import CLIException, getenv_bool
 
-PYPY_URL = "https://pypi.org/pypi/tinybird-cli/json"
+PYPY_URL = "https://pypi.org/pypi/tinybird/json"
 requests_get = sync_to_async(requests.get, thread_sensitive=False)
 
 

tinybird/datafile.py

@@ -3125,7 +3125,7 @@ async def new_pipe(
                 except Exception as e:
                     raise click.ClickException(FeedbackManager.error_creating_pipe(error=e))
 
-        if data.get("type") == "endpoint":
+        if data.get("type") == "endpoint" and t:
             click.echo(FeedbackManager.success_test_endpoint(host=host, pipe=p["name"], token=t["token"]))
 
 

tinybird/feedback_manager.py

@@ -51,6 +51,10 @@ def prompt_message(message: str) -> Callable[..., str]:
     return print_message(message, bcolors.HEADER)
 
 
+def gray_message(message: str) -> Callable[..., str]:
+    return print_message(message, bcolors.CGREY)
+
+
 class FeedbackManager:
     error_exception = error_message("{error}")
     simple_error_exception = simple_error_message("{error}")
@@ -561,9 +565,9 @@ Ready? """
     )
     warning_fixture_not_found = warning_message("** Warning: No fixture found for the datasource {datasource_name}")
     warning_update_version = warning_message(
-        '** UPDATE AVAILABLE: please run "pip install tinybird-cli=={latest_version}" to update or `export TB_VERSION_WARNING=0` to skip the check.'
+        '** UPDATE AVAILABLE: please run "pip install tinybird=={latest_version}" to update or `export TB_VERSION_WARNING=0` to skip the check.'
     )
-    warning_current_version = warning_message("** current: tinybird-cli {current_version}\n")
+    warning_current_version = warning_message("** current: tinybird {current_version}\n")
     warning_confirm_truncate_datasource = prompt_message(
         "Do you want to truncate {datasource}? Once truncated, your data can't be recovered"
     )
@@ -712,9 +716,9 @@ Ready? """
     info_removing_pipe = info_message("** Removing pipe {pipe}")
     info_removing_pipe_not_found = info_message("** {pipe} not found")
     info_dry_removing_pipe = info_message("** [DRY RUN] Removing pipe {pipe}")
-    info_path_created = info_message("** - /{path} created")
-    info_file_created = info_message("** - {file} created")
-    info_path_already_exists = info_message("** - /{path} already exists, skipping")
+    info_path_created = info_message("✓ /{path}")
+    info_file_created = info_message("✓ /{file}")
+    info_path_already_exists = info_message("✓ /{path} already exists, skipping")
     info_dottinyb_already_ignored = info_message("** - '.tinyb' already in .gitignore, skipping")
     info_dotdifftemp_already_ignored = info_message("** - '.diff_tmp' not found or already in .gitignore, skipping")
     info_dottinyenv_already_exists = info_message("** - '.tinyenv' already exists, skipping")
@@ -935,7 +939,7 @@ Ready? """
     success_create = success_message("** '{name}' created")
     success_delete = success_message("** '{name}' deleted")
     success_dynamodb_initial_load = success_message("** Initial load of DynamoDB table started: {job_url}")
-    success_progress_blocks = success_message("** \N{FRONT-FACING BABY CHICK} done")
+    success_progress_blocks = success_message("✓ Done!")
     success_now_using_config = success_message("** Now using {name} ({id})")
     success_connector_config = success_message(
         "** {connector} configuration written to {file_name} file, consider adding it to .gitignore"
@@ -1026,3 +1030,4 @@ Ready? """
     info = info_message("{message}")
     highlight = info_highlight_message("{message}")
     error = error_message("{message}")
+    gray = gray_message("{message}")

tinybird/tb_cli.py

@@ -14,6 +14,7 @@ import tinybird.tb_cli_modules.create
 import tinybird.tb_cli_modules.datasource
 import tinybird.tb_cli_modules.fmt
 import tinybird.tb_cli_modules.job
+import tinybird.tb_cli_modules.mock
 import tinybird.tb_cli_modules.pipe
 import tinybird.tb_cli_modules.tag
 import tinybird.tb_cli_modules.test

tinybird/tb_cli_modules/build.py

@@ -1,11 +1,12 @@
 import asyncio
-import json
 import os
+import random
 import time
 from pathlib import Path
 from typing import Any, Awaitable, Callable, Dict, List, Union
 
 import click
+import humanfriendly
 from watchdog.events import FileSystemEventHandler
 from watchdog.observers import Observer
 
@@ -20,20 +21,15 @@ from tinybird.datafile import (
     parse_datasource,
     parse_pipe,
 )
-from tinybird.feedback_manager import FeedbackManager, info_highlight_message, success_message
+from tinybird.feedback_manager import FeedbackManager, bcolors
 from tinybird.tb_cli_modules.cli import cli
 from tinybird.tb_cli_modules.common import (
     coro,
-    echo_safe_humanfriendly_tables_format_smart_table,
 )
-from tinybird.tb_cli_modules.create import generate_sample_data_from_columns
 from tinybird.tb_cli_modules.local import (
-    get_docker_client,
     get_tinybird_local_client,
-    remove_tinybird_local,
-    start_tinybird_local,
-    stop_tinybird_local,
 )
+from tinybird.tb_cli_modules.table import format_table
 
 
 class FileChangeHandler(FileSystemEventHandler):
@@ -44,7 +40,7 @@ class FileChangeHandler(FileSystemEventHandler):
     def on_modified(self, event: Any) -> None:
         if not event.is_directory and any(event.src_path.endswith(ext) for ext in [".datasource", ".pipe"]):
             filename = event.src_path.split("/")[-1]
-            click.echo(info_highlight_message(f"\n⟲ Changes detected in {filename}\n")())
+            click.echo(FeedbackManager.highlight(message=f"\n⟲ Changes detected in {filename}\n"))
             try:
                 self.process([event.src_path])
             except Exception as e:
@@ -65,7 +61,10 @@ def watch_files(
             process(files, watch=True)
         time_end = time.time()
         elapsed_time = time_end - time_start
-        click.echo(success_message(f"\n✓ Rebuild completed in {elapsed_time:.1f}s")())
+        click.echo(
+            FeedbackManager.success(message="\n✓ ")
+            + FeedbackManager.gray(message=f"Rebuild completed in {elapsed_time:.1f}s")
+        )
 
     event_handler = FileChangeHandler(filenames, lambda f: asyncio.run(process_wrapper(f)))
     observer = Observer()
@@ -100,23 +99,19 @@ def watch_files(
     help="Watch for changes in the files and re-check them.",
 )
 @click.option(
-    "--restart",
+    "--skip-datasources",
     is_flag=True,
-    help="Restart the Tinybird development environment before building the first time.",
+    help="Skip rebuilding datasources.",
 )
 @coro
 async def build(
     folder: str,
     watch: bool,
-    restart: bool,
+    skip_datasources: bool,
 ) -> None:
     """
     Watch for changes in the files and re-check them.
     """
-    docker_client = get_docker_client()
-    if restart:
-        remove_tinybird_local(docker_client)
-        start_tinybird_local(docker_client)
     ignore_sql_errors = FeatureFlags.ignore_sql_errors()
     context.disable_template_security_validation.set(True)
     is_internal = has_internal_datafiles(folder)
@@ -156,15 +151,6 @@ async def build(
             only_pipes=only_pipes,
         )
 
-        for filename in filenames:
-            if filename.endswith(".datasource"):
-                ds_path = Path(filename)
-                ds_name = ds_path.stem
-                datasource_content = ds_path.read_text()
-                sample_data = await generate_sample_data_from_columns(tb_client, datasource_content)
-                ndjson_data = "\n".join([json.dumps(row) for row in sample_data])
-                await tb_client.datasource_events(ds_name, ndjson_data)
-
         if watch:
             filename = filenames[0]
             if filename.endswith(".pipe"):
@@ -176,7 +162,7 @@ async def build(
         try:
             click.echo("⚡ Building project...")
             time_start = time.time()
-            await process(filenames=filenames, watch=False)
+            await process(filenames=filenames, watch=False, only_pipes=skip_datasources)
             time_end = time.time()
             elapsed_time = time_end - time_start
             click.echo(FeedbackManager.success(message=f"\n✓ Build completed in {elapsed_time:.1f}s\n"))
@@ -191,31 +177,44 @@ async def build(
 
 
 async def build_and_print_pipe(tb_client: TinyB, filename: str):
-    pipe_name = os.path.basename(filename.split(".")[0])
-    res = await tb_client.query(f"SELECT * FROM {pipe_name} LIMIT 5 FORMAT JSON", pipeline=pipe_name)
+    rebuild_colors = [bcolors.FAIL, bcolors.OKBLUE, bcolors.WARNING, bcolors.OKGREEN, bcolors.HEADER]
+    rebuild_index = random.randint(0, len(rebuild_colors) - 1)
+    rebuild_color = rebuild_colors[rebuild_index % len(rebuild_colors)]
+    pipe_name = Path(filename).stem
+    res = await tb_client.query(f"SELECT * FROM {pipe_name} FORMAT JSON", pipeline=pipe_name)
     data = []
-    for d in res["data"]:
+    limit = 5
+    for d in res["data"][:5]:
         data.append(d.values())
     meta = res["meta"]
-    column_names = [col["name"] for col in meta]
-    echo_safe_humanfriendly_tables_format_smart_table(data, column_names=column_names)
-
-
-@cli.command()
-@coro
-async def stop() -> None:
-    """Stop Tinybird development environment"""
-    click.echo(FeedbackManager.info(message="Shutting down Tinybird development environment..."))
-    docker_client = get_docker_client()
-    stop_tinybird_local(docker_client)
-    click.echo(FeedbackManager.success(message="Tinybird development environment stopped"))
-
+    row_count = res.get("rows", 0)
+    stats = res.get("statistics", {})
+    elapsed = stats.get("elapsed", 0)
+    node_name = "endpoint"
+    cols = len(meta)
+    try:
 
-@cli.command()
-@coro
-async def start() -> None:
-    """Start Tinybird development environment"""
-    click.echo(FeedbackManager.info(message="Starting Tinybird development environment..."))
-    docker_client = get_docker_client()
-    start_tinybird_local(docker_client)
-    click.echo(FeedbackManager.success(message="Tinybird development environment started"))
+        def print_message(message: str, color=bcolors.CGREY):
+            return f"{color}{message}{bcolors.ENDC}"
+
+        table = format_table(data, meta)
+        colored_char = print_message("│", rebuild_color)
+        table_with_marker = "\n".join(f"{colored_char} {line}" for line in table.split("\n"))
+        click.echo(f"\n{colored_char} {print_message('⚡', rebuild_color)} Running {pipe_name} → {node_name}")
+        click.echo(colored_char)
+        click.echo(table_with_marker)
+        click.echo(colored_char)
+        rows_read = humanfriendly.format_number(stats.get("rows_read", 0))
+        bytes_read = humanfriendly.format_size(stats.get("bytes_read", 0))
+        elapsed = humanfriendly.format_timespan(elapsed) if elapsed >= 1 else f"{elapsed * 1000:.2f}ms"
+        stats_message = f"» {bytes_read} ({rows_read} rows x {cols} cols) in {elapsed}"
+        rows_message = f"» Showing {limit} first rows" if row_count > limit else "» Showing all rows"
+        click.echo(f"{colored_char} {print_message(stats_message, bcolors.OKGREEN)}")
+        click.echo(f"{colored_char} {print_message(rows_message, bcolors.CGREY)}")
+    except ValueError as exc:
+        if str(exc) == "max() arg is an empty sequence":
+            click.echo("------------")
+            click.echo("Empty")
+            click.echo("------------")
+        else:
+            raise exc

tinybird/tb_cli_modules/cli.py

@@ -10,14 +10,12 @@ import pprint
 import re
 import shutil
 import sys
-from datetime import datetime
 from os import environ, getcwd
 from pathlib import Path
 from typing import Any, Callable, Dict, Iterable, List, Optional, Tuple, Union
 
 import click
 import humanfriendly
-import requests
 from click import Context
 from packaging import version
 
@@ -58,7 +56,7 @@ from tinybird.tb_cli_modules.common import (
     _get_tb_client,
     coro,
     create_tb_client,
-    echo_safe_humanfriendly_tables_format_smart_table,
+    echo_safe_format_table,
     folder_init,
     get_current_main_workspace,
     getenv_bool,
@@ -69,7 +67,6 @@ from tinybird.tb_cli_modules.common import (
     try_update_config_with_remote,
 )
 from tinybird.tb_cli_modules.config import CLIConfig
-from tinybird.tb_cli_modules.prompts import sample_data_prompt
 from tinybird.tb_cli_modules.telemetry import add_telemetry_event
 
 __old_click_echo = click.echo
@@ -927,7 +924,7 @@ async def sql(
             dd = []
             for d in res["data"]:
                 dd.append(d.values())
-            echo_safe_humanfriendly_tables_format_smart_table(dd, column_names=res["data"][0].keys())
+            echo_safe_format_table(dd, columns=res["meta"])
     else:
         click.echo(FeedbackManager.info_no_rows())
 
@@ -1572,67 +1569,3 @@ async def deploy(
         raise
     except Exception as e:
         raise CLIException(str(e))
-
-
-@cli.command()
-@click.argument("datasource_file", type=click.Path(exists=True))
-@click.option("--count", type=int, default=10, help="Number of events to send")
-@click.option("--model", type=str, default=None, help="Model to use for data generation")
-@click.option("--print-data", is_flag=True, default=False, help="Print the data being sent")
-@click.pass_context
-def load_sample_data(ctx: Context, datasource_file: str, count: int, model: Optional[str], print_data: bool) -> None:
-    """Load sample data into a datasource.
-
-    Args:
-        ctx: Click context object
-        datasource_file: Path to the datasource file to load sample data into
-    """
-    import llm
-
-    try:
-        # TODO(eclbg): allow passing a datasource name instead of a file
-        datasource_path = Path(datasource_file)
-        if datasource_path.suffix != ".datasource":
-            raise CLIException(FeedbackManager.error_file_extension(filename=datasource_file))
-
-        datasource_name = datasource_path.stem
-
-        response = requests.get("http://localhost:80/tokens")
-        token = response.json()["workspace_admin_token"]
-
-        with open(datasource_file) as f:
-            content = f.read()
-            schema_start = next(i for i, line in enumerate(content.splitlines()) if line.strip().startswith("SCHEMA >"))
-            schema_end = next(
-                i
-                for i, line in enumerate(content.splitlines()[schema_start + 1 :], schema_start + 1)
-                if not line.strip()
-            )
-            schema = "\n".join(content.splitlines()[schema_start:schema_end])
-        llm_model = llm.get_model(model)
-        click.echo(f"Using model: {model}")
-        prompt = sample_data_prompt.format(current_datetime=datetime.now().isoformat(), row_count=count)
-        # prompt = sample_data_with_errors_prompt.format(current_datetime=datetime.now().isoformat())  # This prompt will generate data with errors
-        full_prompt = prompt + "\n\n" + schema
-        sent_events = 0
-        while sent_events < count:
-            click.echo(f"Generating data for '{datasource_name}'")
-            data = llm_model.prompt(full_prompt)
-
-            click.echo(f"Sending data to '{datasource_name}'")
-            headers = {"Authorization": f"Bearer {token}"}
-            if print_data:
-                click.echo(f"Data: {data}")
-            response = requests.post(
-                f"http://localhost:80/v0/events?name={datasource_name}",
-                data=data,
-                headers=headers,
-            )
-            if response.status_code not in (200, 202):
-                raise CLIException(f"Failed to send data: {response.text}")
-            click.echo(f"Response: {response.text}")
-            sent_events += 10
-            click.echo(f"Sent 10 events to datasource '{datasource_name}'")
-
-    except Exception as e:
-        raise CLIException(FeedbackManager.error_exception(error=str(e)))

tinybird/tb_cli_modules/common.py

@@ -53,6 +53,7 @@ from tinybird.config import (
     get_display_host,
     write_config,
 )
+from tinybird.tb_cli_modules.table import format_table
 
 if TYPE_CHECKING:
     from tinybird.connectors import Connector
@@ -134,6 +135,23 @@ def echo_safe_humanfriendly_tables_format_smart_table(data: Iterable[Any], colum
             raise exc
 
 
+def echo_safe_format_table(data: Iterable[Any], columns) -> None:
+    """
+    There is a bug in the humanfriendly library: it breaks to render the small table for small terminals
+    (`format_robust_table`) if we call format_smart_table with an empty dataset. This catches the error and prints
+    what we would call an empty "robust_table".
+    """
+    try:
+        click.echo(format_table(data, columns))
+    except ValueError as exc:
+        if str(exc) == "max() arg is an empty sequence":
+            click.echo("------------")
+            click.echo("Empty")
+            click.echo("------------")
+        else:
+            raise exc
+
+
 def normalize_datasource_name(s: str) -> str:
     s = re.sub(r"[^0-9a-zA-Z_]", "_", s)
     if s[0] in "0123456789":
@@ -153,7 +171,7 @@ def generate_datafile(
     if not f.exists() or force:
         with open(f"{f}", "w") as ds_file:
             ds_file.write(datafile)
-        click.echo(FeedbackManager.success(message=f"** Generated {f}"))
+        click.echo(FeedbackManager.info_file_created(file=f))
 
         if data and (base / "fixtures").exists():
             # Generating a fixture for Parquet files is not so trivial, since Parquet format
@@ -1090,7 +1108,7 @@ async def push_data(
     cb.First = True  # type: ignore[attr-defined]
     cb.prev_done = 0  # type: ignore[attr-defined]
 
-    click.echo(FeedbackManager.info_starting_import_process())
+    click.echo(FeedbackManager.gray(message=f"\nImporting data to {datasource_name} Data Source..."))
 
     if isinstance(url, list):
         urls = url
@@ -1161,17 +1179,16 @@ async def push_data(
     except Exception as e:
         raise CLIException(FeedbackManager.error_exception(error=e))
     else:
-        click.echo(FeedbackManager.success_progress_blocks())
         if mode == "append" and parser and parser != "clickhouse":
             click.echo(FeedbackManager.success_appended_rows(appended_rows=appended_rows))
 
-        click.echo(FeedbackManager.success_total_rows(datasource=datasource_name, total_rows=total_rows))
-
         if mode == "replace":
             click.echo(FeedbackManager.success_replaced_datasource(datasource=datasource_name))
         else:
-            click.echo(FeedbackManager.success_appended_datasource(datasource=datasource_name))
-        click.echo(FeedbackManager.info_data_pushed(datasource=datasource_name))
+            click.echo(FeedbackManager.highlight(message="» 2.57m rows x 9 cols in 852.04ms"))
+
+        click.echo(FeedbackManager.success_progress_blocks())
+
     finally:
         try:
             for url in urls:

tinybird/tb_cli_modules/config.py

@@ -338,6 +338,14 @@ class CLIConfig:
             CLIConfig._projects[working_dir] = result
         return result
 
+    @staticmethod
+    def get_llm_config(working_dir: Optional[str] = None) -> Dict[str, Any]:
+        return (
+            CLIConfig.get_project_config(working_dir)
+            .get("llms", {})
+            .get("openai", {"model": "gpt-4o-mini", "api_key": None})
+        )
+
     @staticmethod
     def reset() -> None:
         CLIConfig._global = None

tinybird/tb_cli_modules/create.py

@@ -1,4 +1,3 @@
-import json
 import os
 from os import getcwd
 from pathlib import Path
@@ -6,7 +5,6 @@ from typing import Any, Dict, List, Optional
 
 import click
 from click import Context
-from openai import OpenAI
 
 from tinybird.client import TinyB
 from tinybird.datafile import folder_build
@@ -16,8 +14,9 @@ from tinybird.tb_cli_modules.common import _generate_datafile, coro, generate_da
 from tinybird.tb_cli_modules.config import CLIConfig
 from tinybird.tb_cli_modules.exceptions import CLIDatasourceException
 from tinybird.tb_cli_modules.llm import LLM
-from tinybird.tb_cli_modules.local import get_docker_client, set_up_tinybird_local
-from tinybird.tb_cli_modules.prompts import sample_data_sql_prompt
+from tinybird.tb_cli_modules.local import (
+    get_tinybird_local_client,
+)
 
 
 @cli.command()
@@ -48,12 +47,13 @@ async def create(
     folder: Optional[str],
 ) -> None:
     """Initialize a new project."""
-    click.echo(FeedbackManager.highlight(message="Setting up Tinybird development environment..."))
+    click.echo(FeedbackManager.gray(message="Setting up Tinybird Local"))
     folder = folder or getcwd()
     try:
-        docker_client = get_docker_client()
-        tb_client = set_up_tinybird_local(docker_client)
+        tb_client = get_tinybird_local_client()
+        click.echo(FeedbackManager.gray(message="Creating new project structure..."))
         await project_create(tb_client, data, prompt, folder)
+        click.echo(FeedbackManager.success(message="✓ Scaffolding completed!\n"))
         workspaces: List[Dict[str, Any]] = (await tb_client.user_workspaces()).get("workspaces", [])
         datasources = await tb_client.datasources()
         pipes = await tb_client.pipes(dependencies=True)
@@ -69,11 +69,15 @@ async def create(
         elif prompt:
             datasource_files = [f for f in os.listdir(Path(folder) / "datasources") if f.endswith(".datasource")]
             for datasource_file in datasource_files:
-                datasource_content = Path(folder) / "datasources" / datasource_file
-                sample_data = await generate_sample_data_from_columns(tb_client, datasource_content)
-                ndjson_data = "\n".join([json.dumps(row) for row in sample_data])
-                await tb_client.datasource_events(datasource_file, ndjson_data)
-        click.echo(FeedbackManager.success(message="\n✔ Tinybird development environment is ready"))
+                datasource_path = Path(folder) / "datasources" / datasource_file
+                llm_config = CLIConfig.get_llm_config()
+                llm = LLM(key=llm_config["api_key"])
+                datasource_name = datasource_path.stem
+                datasource_content = datasource_path.read_text()
+                has_json_path = "`json:" in datasource_content
+                if has_json_path:
+                    await llm.generate_sql_sample_data(tb_client, datasource_name, datasource_content)
+        click.echo(FeedbackManager.success(message="\n✓ Tinybird Local is ready!"))
     except Exception as e:
         click.echo(FeedbackManager.error(message=f"Error: {str(e)}"))
 
@@ -84,7 +88,7 @@ async def project_create(
     prompt: Optional[str],
     folder: str,
 ):
-    project_paths = ["datasources", "endpoints", "copies", "sinks", "playgrounds", "materializations"]
+    project_paths = ["datasources", "endpoints", "materializations", "copies", "sinks"]
     force = True
     for x in project_paths:
         try:
@@ -92,7 +96,7 @@ async def project_create(
             f.mkdir()
             click.echo(FeedbackManager.info_path_created(path=x))
         except FileExistsError:
-            pass
+            click.echo(FeedbackManager.info_path_created(path=x))
 
     def generate_pipe_file(name: str, content: str):
         base = Path("endpoints")
@@ -101,7 +105,7 @@ async def project_create(
         f = base / (f"{name}.pipe")
         with open(f"{f}", "w") as file:
             file.write(content)
-        click.echo(FeedbackManager.success(message=f"** Generated {f}"))
+        click.echo(FeedbackManager.info_file_created(file=f))
 
     if data:
         path = Path(folder) / data
@@ -119,10 +123,8 @@ TYPE ENDPOINT
         )
     elif prompt:
         try:
-            config = CLIConfig.get_project_config()
-            model = config.get("llms", {}).get("openai", {}).get("model", "gpt-4o-mini")
-            api_key = config.get("llms", {}).get("openai", {}).get("api_key", None)
-            llm = LLM(model=model, key=api_key)
+            llm_config = CLIConfig.get_llm_config()
+            llm = LLM(key=llm_config["api_key"])
             result = await llm.create_project(prompt)
             for ds in result.datasources:
                 content = ds.content.replace("```", "")
@@ -198,29 +200,3 @@ async def append_datasource(
         ignore_empty=ignore_empty,
         concurrency=concurrency,
     )
-
-
-def generate_sql_sample_data(datasource_content: str, row_count: int, model: str, api_key: str) -> str:
-    client = OpenAI(api_key=api_key)
-
-    response = client.chat.completions.create(
-        model=model,
-        messages=[
-            {"role": "system", "content": sample_data_sql_prompt.format(row_count=row_count)},
-            {"role": "user", "content": datasource_content},
-        ],
-    )
-
-    return response.choices[0].message.content or ""
-
-
-async def generate_sample_data_from_columns(
-    tb_client: TinyB, datasource_content: str, row_count: int = 20
-) -> List[Dict[str, Any]]:
-    config = CLIConfig.get_project_config()
-    model = config.get("llms", {}).get("openai", {}).get("model", "gpt-4o-mini")
-    api_key = config.get("llms", {}).get("openai", {}).get("api_key", None)
-    sql = generate_sql_sample_data(datasource_content, row_count, model, api_key)
-    result = await tb_client.query(f"{sql} FORMAT JSON")
-    data = result.get("data", [])
-    return data

tinybird/tb_cli_modules/llm.py

@@ -1,9 +1,12 @@
-from typing import List, Optional
+import asyncio
+from datetime import datetime
+from typing import Awaitable, Callable, List
 
 from openai import OpenAI
 from pydantic import BaseModel
 
-from tinybird.tb_cli_modules.prompts import create_project_prompt
+from tinybird.client import TinyB
+from tinybird.tb_cli_modules.prompts import create_project_prompt, sample_data_sql_prompt
 
 
 class DataFile(BaseModel):
@@ -17,15 +20,54 @@ class DataProject(BaseModel):
 
 
 class LLM:
-    def __init__(self, model: Optional[str] = None, key: Optional[str] = None):
-        self.model = model
-        self.key = key
+    def __init__(self, key: str):
+        self.client = OpenAI(api_key=key)
+
+    async def _execute(self, action_fn: Callable[[], Awaitable[str]], checker_fn: Callable[[str], bool]):
+        is_valid = False
+        times = 0
+
+        while not is_valid and times < 5:
+            result = await action_fn()
+            if asyncio.iscoroutinefunction(checker_fn):
+                is_valid = await checker_fn(result)
+            else:
+                is_valid = checker_fn(result)
+            times += 1
+
+        return result
 
     async def create_project(self, prompt: str) -> DataProject:
-        client = OpenAI(api_key=self.key)
-        completion = client.beta.chat.completions.parse(
-            model=self.model,
+        completion = self.client.beta.chat.completions.parse(
+            model="gpt-4o-mini",
             messages=[{"role": "system", "content": create_project_prompt}, {"role": "user", "content": prompt}],
             response_format=DataProject,
         )
         return completion.choices[0].message.parsed or DataProject(datasources=[], pipes=[])
+
+    async def generate_sql_sample_data(
+        self, tb_client: TinyB, datasource_content: str, row_count: int = 20, context: str = ""
+    ) -> str:
+        async def action_fn():
+            response = self.client.chat.completions.create(
+                model="gpt-4o-mini",
+                messages=[
+                    {
+                        "role": "system",
+                        "content": sample_data_sql_prompt.format(
+                            current_datetime=datetime.now().isoformat(), row_count=row_count, context=context
+                        ),
+                    },
+                    {"role": "user", "content": datasource_content},
+                ],
+            )
+            return response.choices[0].message.content or ""
+
+        async def checker_fn(sql: str):
+            try:
+                result = await tb_client.query(f"DESCRIBE ({sql}) FORMAT JSON")
+                return len(result.get("data", [])) > 0
+            except Exception:
+                return False
+
+        return await self._execute(action_fn, checker_fn)

tinybird/tb_cli_modules/local.py

@@ -1,3 +1,4 @@
+import os
 import time
 
 import click
@@ -5,13 +6,17 @@ import requests
 
 import docker
 from tinybird.feedback_manager import FeedbackManager
-from tinybird.tb_cli_modules.common import CLIException
+from tinybird.tb_cli_modules.cli import cli
+from tinybird.tb_cli_modules.common import (
+    coro,
+)
 from tinybird.tb_cli_modules.config import CLIConfig
+from tinybird.tb_cli_modules.exceptions import CLIException
 
 # TODO: Use the official Tinybird image once it's available 'tinybirdco/tinybird-local:latest'
 TB_IMAGE_NAME = "registry.gitlab.com/tinybird/analytics/tinybird-local-jammy-3.11:latest"
 TB_CONTAINER_NAME = "tinybird-local"
-TB_LOCAL_PORT = 80
+TB_LOCAL_PORT = int(os.getenv("TB_LOCAL_PORT", 80))
 TB_LOCAL_HOST = f"http://localhost:{TB_LOCAL_PORT}"
 
 
@@ -19,26 +24,38 @@ def start_tinybird_local(
     docker_client,
 ):
     """Start the Tinybird container."""
+    pull_show_prompt = False
+    pull_required = False
+    try:
+        local_image = docker_client.images.get(TB_IMAGE_NAME)
+        local_image_id = local_image.attrs["RepoDigests"][0].split("@")[1]
+        remote_image = docker_client.images.get_registry_data(TB_IMAGE_NAME)
+        pull_show_prompt = local_image_id != remote_image.id
+    except Exception:
+        pull_show_prompt = False
+        pull_required = True
+
+    if (
+        pull_show_prompt
+        and click.prompt(FeedbackManager.info(message="** New version detected, download? [y/N]")).lower() == "y"
+    ):
+        click.echo(FeedbackManager.info(message="** Downloading latest version of Tinybird Local..."))
+        pull_required = True
+
+    if pull_required:
+        docker_client.images.pull(TB_IMAGE_NAME, platform="linux/amd64")
+
+    container = None
     containers = docker_client.containers.list(all=True, filters={"name": TB_CONTAINER_NAME})
     if containers:
-        # Container `start` is idempotent. It's safe to call it even if the container is already running.
         container = containers[0]
+
+    if container and not pull_required:
+        # Container `start` is idempotent. It's safe to call it even if the container is already running.
         container.start()
     else:
-        pull_required = False
-        try:
-            local_image = docker_client.images.get(TB_IMAGE_NAME)
-            local_image_id = local_image.attrs["RepoDigests"][0].split("@")[1]
-            remote_image = docker_client.images.get_registry_data(TB_IMAGE_NAME)
-            pull_required = local_image_id != remote_image.id
-        except Exception:
-            pull_required = True
-
-        if pull_required:
-            click.echo(
-                FeedbackManager.info(message="** Downloading latest version of Tinybird development environment...")
-            )
-            docker_client.images.pull(TB_IMAGE_NAME, platform="linux/amd64")
+        if container:
+            container.remove(force=True)
 
         container = docker_client.containers.run(
             TB_IMAGE_NAME,
@@ -49,7 +66,7 @@ def start_tinybird_local(
             platform="linux/amd64",
         )
 
-    click.echo(FeedbackManager.info(message="** Waiting for Tinybird development environment to be ready..."))
+    click.echo(FeedbackManager.info(message="** Waiting for Tinybird Local to be ready..."))
     for attempt in range(10):
         try:
             run = container.exec_run("tb --no-version-warning sql 'SELECT 1 AS healthcheck' --format json").output
@@ -60,9 +77,11 @@ def start_tinybird_local(
             raise RuntimeError("Unexpected response from Tinybird")
         except Exception:
             if attempt == 9:  # Last attempt
-                raise CLIException("Tinybird local environment not ready yet. Please try again in a few seconds.")
+                raise CLIException("Tinybird local not ready yet. Please try again in a few seconds.")
             time.sleep(5)  # Wait 5 seconds before retrying
 
+    click.echo(FeedbackManager.success(message="✓ All set!\n"))
+
 
 def get_docker_client():
     """Check if Docker is installed and running."""
@@ -92,12 +111,6 @@ def remove_tinybird_local(docker_client):
         pass
 
 
-def set_up_tinybird_local(docker_client):
-    """Set up the Tinybird local environment."""
-    start_tinybird_local(docker_client)
-    return get_tinybird_local_client()
-
-
 def get_tinybird_local_client():
     """Get a Tinybird client connected to the local environment."""
     config = CLIConfig.get_project_config()
@@ -106,3 +119,50 @@ def get_tinybird_local_client():
     config.set_token(token)
     config.set_host(TB_LOCAL_HOST)
     return config.get_client(host=TB_LOCAL_HOST, token=token)
+
+
+@cli.group()
+@click.pass_context
+def local(ctx):
+    """Local commands"""
+
+
+@local.command()
+@coro
+async def stop() -> None:
+    """Stop Tinybird development environment"""
+    click.echo(FeedbackManager.info(message="Shutting down Tinybird development environment..."))
+    docker_client = get_docker_client()
+    stop_tinybird_local(docker_client)
+    click.echo(FeedbackManager.success(message="Tinybird development environment stopped"))
+
+
+@local.command()
+@coro
+async def remove() -> None:
+    """Remove Tinybird development environment"""
+    click.echo(FeedbackManager.info(message="Removing Tinybird development environment..."))
+    docker_client = get_docker_client()
+    remove_tinybird_local(docker_client)
+    click.echo(FeedbackManager.success(message="Tinybird development environment removed"))
+
+
+@local.command()
+@coro
+async def start() -> None:
+    """Start Tinybird development environment"""
+    click.echo(FeedbackManager.info(message="Starting Tinybird development environment..."))
+    docker_client = get_docker_client()
+    start_tinybird_local(docker_client)
+    click.echo(FeedbackManager.success(message="Tinybird development environment started"))
+
+
+@local.command()
+@coro
+async def restart() -> None:
+    """Restart Tinybird development environment"""
+    click.echo(FeedbackManager.info(message="Restarting Tinybird development environment..."))
+    docker_client = get_docker_client()
+    remove_tinybird_local(docker_client)
+    start_tinybird_local(docker_client)
+    click.echo(FeedbackManager.success(message="Tinybird development environment restarted"))

tinybird/tb_cli_modules/mock.py

@@ -0,0 +1,53 @@
+import json
+from pathlib import Path
+
+import click
+
+from tinybird.feedback_manager import FeedbackManager
+from tinybird.tb_cli_modules.cli import cli
+from tinybird.tb_cli_modules.common import CLIException, coro
+from tinybird.tb_cli_modules.config import CLIConfig
+from tinybird.tb_cli_modules.llm import LLM
+from tinybird.tb_cli_modules.local import get_tinybird_local_client
+
+
+@cli.command()
+@click.argument("datasource", type=str)
+@click.option("--rows", type=int, default=10, help="Number of events to send")
+@click.option("--context", type=str, default="", help="Extra context to use for data generation")
+@coro
+async def mock(datasource: str, rows: int, context: str) -> None:
+    """Load sample data into a datasource.
+
+    Args:
+        ctx: Click context object
+        datasource_file: Path to the datasource file to load sample data into
+    """
+    import llm
+
+    try:
+        datasource_path = Path(datasource)
+        datasource_name = datasource
+        if datasource_path.suffix == ".datasource":
+            datasource_name = datasource_path.stem
+        else:
+            datasource_path = Path("datasources", f"{datasource}.datasource")
+
+        datasource_content = datasource_path.read_text()
+        llm_config = CLIConfig.get_llm_config()
+        llm = LLM(key=llm_config["api_key"])
+        tb_client = get_tinybird_local_client()
+        sql = await llm.generate_sql_sample_data(tb_client, datasource_content, rows, context)
+        result = await tb_client.query(f"{sql} FORMAT JSON")
+        data = result.get("data", [])
+        max_rows_per_request = 100
+        sent_rows = 0
+        for i in range(0, len(data), max_rows_per_request):
+            batch = data[i : i + max_rows_per_request]
+            ndjson_data = "\n".join([json.dumps(row) for row in batch])
+            await tb_client.datasource_events(datasource_name, ndjson_data)
+            sent_rows += len(batch)
+        click.echo(f"Sent {sent_rows} events to datasource '{datasource_name}'")
+
+    except Exception as e:
+        raise CLIException(FeedbackManager.error_exception(error=str(e)))

tinybird/tb_cli_modules/prompts.py

@@ -130,4 +130,7 @@ FROM numbers({row_count})
 - Do NOT add a semicolon at the end of the query
 - Do NOT add any FORMAT at the end of the query, because it will be added later by Tinybird.
 
+# Extra context:
+{context}
+
 """

tinybird/tb_cli_modules/table.py

@@ -0,0 +1,185 @@
+# Standard library modules.
+import collections
+import re
+
+# Modules included in our package.
+from humanfriendly.compat import coerce_string
+from humanfriendly.tables import format_robust_table
+from humanfriendly.terminal import (
+    ansi_strip,
+    ansi_width,
+    ansi_wrap,
+    find_terminal_size,
+    terminal_supports_colors,
+)
+
+from tinybird.feedback_manager import bcolors
+
+NUMERIC_DATA_PATTERN = re.compile(r"^\d+(\.\d+)?$")
+
+
+def format_table(data, columns):
+    """
+    Render tabular data using the most appropriate representation.
+
+    :param data: An iterable (e.g. a :func:`tuple` or :class:`list`)
+                 containing the rows of the table, where each row is an
+                 iterable containing the columns of the table (strings).
+    :param column_names: An iterable of column names (strings).
+    :returns: The rendered table (a string).
+
+    If you want an easy way to render tabular data on a terminal in a human
+    friendly format then this function is for you! It works as follows:
+
+    - If the input data doesn't contain any line breaks the function
+      :func:`format_pretty_table()` is used to render a pretty table. If the
+      resulting table fits in the terminal without wrapping the rendered pretty
+      table is returned.
+
+    - If the input data does contain line breaks or if a pretty table would
+      wrap (given the width of the terminal) then the function
+      :func:`format_robust_table()` is used to render a more robust table that
+      can deal with data containing line breaks and long text.
+    """
+    # Normalize the input in case we fall back from a pretty table to a robust
+    # table (in which case we'll definitely iterate the input more than once).
+    data = [normalize_columns(r) for r in data]
+
+    column_names = []
+    column_types = []
+
+    for c in columns:
+        column_names.append(c["name"])
+        column_types.append(c["type"])
+
+    column_names = normalize_columns(column_names)
+    # Make sure the input data doesn't contain any line breaks (because pretty
+    # tables break horribly when a column's text contains a line break :-).
+    if not any(any("\n" in c for c in r) for r in data):
+        # Render a pretty table.
+        pretty_table = format_pretty_table(data, column_names, column_types)
+        # Check if the pretty table fits in the terminal.
+        table_width = max(map(ansi_width, pretty_table.splitlines()))
+        num_rows, num_columns = find_terminal_size()
+        if table_width <= num_columns:
+            # The pretty table fits in the terminal without wrapping!
+            return pretty_table
+    # Fall back to a robust table when a pretty table won't work.
+    return format_robust_table(data, column_names)
+
+
+def format_pretty_table(data, column_names=None, column_types=None, horizontal_bar="─", vertical_bar=" "):
+    """
+    Render a table using characters like dashes and vertical bars to emulate borders.
+
+    :param data: An iterable (e.g. a :func:`tuple` or :class:`list`)
+                 containing the rows of the table, where each row is an
+                 iterable containing the columns of the table (strings).
+    :param column_names: An iterable of column names (strings).
+    :param column_types: An iterable of column types (strings).
+    :param horizontal_bar: The character used to represent a horizontal bar (a
+                           string).
+    :param vertical_bar: The character used to represent a vertical bar (a
+                         string).
+    :returns: The rendered table (a string).
+
+    Here's an example:
+
+    >>> from humanfriendly.tables import format_pretty_table
+    >>> column_names = ['Version', 'Uploaded on', 'Downloads']
+    >>> humanfriendly_releases = [
+    ... ['1.23', '2015-05-25', '218'],
+    ... ['1.23.1', '2015-05-26', '1354'],
+    ... ['1.24', '2015-05-26', '223'],
+    ... ['1.25', '2015-05-26', '4319'],
+    ... ['1.25.1', '2015-06-02', '197'],
+    ... ]
+    >>> print(format_pretty_table(humanfriendly_releases, column_names))
+    -------------------------------------
+    | Version | Uploaded on | Downloads |
+    -------------------------------------
+    | 1.23    | 2015-05-25  |       218 |
+    | 1.23.1  | 2015-05-26  |      1354 |
+    | 1.24    | 2015-05-26  |       223 |
+    | 1.25    | 2015-05-26  |      4319 |
+    | 1.25.1  | 2015-06-02  |       197 |
+    -------------------------------------
+
+    Notes about the resulting table:
+
+    - If a column contains numeric data (integer and/or floating point
+      numbers) in all rows (ignoring column names of course) then the content
+      of that column is right-aligned, as can be seen in the example above. The
+      idea here is to make it easier to compare the numbers in different
+      columns to each other.
+
+    - The column names are highlighted in color so they stand out a bit more
+      (see also :data:`.HIGHLIGHT_COLOR`). The following screen shot shows what
+      that looks like (my terminals are always set to white text on a black
+      background):
+
+      .. image:: images/pretty-table.png
+    """
+    # Normalize the input because we'll have to iterate it more than once.
+    data = [normalize_columns(r, expandtabs=True) for r in data]
+    if column_names is not None:
+        column_names = normalize_columns(column_names)
+        if column_names:
+            if terminal_supports_colors():
+                column_names = [highlight_column_name(n) for n in column_names]
+            data.insert(0, column_names)
+            if column_types is not None:
+                column_types = normalize_columns(column_types)
+                column_types = [highlight_column_type(t) for t in column_types]
+                data.insert(1, column_types)
+    # Calculate the maximum width of each column.
+    widths = collections.defaultdict(int)
+    numeric_data = collections.defaultdict(list)
+    for row_index, row in enumerate(data):
+        for column_index, column in enumerate(row):
+            widths[column_index] = max(widths[column_index], ansi_width(column))
+            if not (column_names and row_index <= 1):  # Skip both header rows for numeric check
+                numeric_data[column_index].append(bool(NUMERIC_DATA_PATTERN.match(ansi_strip(column))))
+
+    horizontal_bar = highlight_horizontal_bar(horizontal_bar)
+    # Create a horizontal bar of dashes as a delimiter.
+    line_delimiter = horizontal_bar * (sum(widths.values()) + len(widths) * 3 + 1)
+    # Start the table with a vertical bar.
+    lines = []
+    # Format the rows and columns.
+    for row_index, row in enumerate(data):
+        line = [vertical_bar]
+        for column_index, column in enumerate(row):
+            padding = " " * (widths[column_index] - ansi_width(column))
+            if all(numeric_data[column_index]):
+                line.append(" " + padding + column + " ")
+            else:
+                line.append(" " + column + padding + " ")
+            line.append(vertical_bar)
+        lines.append("".join(line))
+        if column_names and column_types and row_index > 0 and row_index < len(data) - 1:
+            lines.append(line_delimiter)
+    # Join the lines, returning a single string.
+    return "\n".join(lines)
+
+
+def normalize_columns(row, expandtabs=False):
+    results = []
+    for value in row:
+        text = coerce_string(value)
+        if expandtabs:
+            text = text.expandtabs()
+        results.append(text)
+    return results
+
+
+def highlight_column_name(name):
+    return ansi_wrap(name)
+
+
+def highlight_column_type(type):
+    return f"{bcolors.CGREY}\033[3m{type}\033[23m{bcolors.ENDC}"
+
+
+def highlight_horizontal_bar(horizontal_bar):
+    return f"{bcolors.CGREY}\033[3m{horizontal_bar}\033[23m{bcolors.ENDC}"

{tinybird-0.0.1.dev3.dist-info → tinybird-0.0.1.dev5.dist-info}/METADATA

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

{tinybird-0.0.1.dev3.dist-info → tinybird-0.0.1.dev5.dist-info}/RECORD

@@ -1,40 +1,42 @@
-tinybird/__cli__.py,sha256=bgGjpK7NkAkSj1v-IXZIG9nlQnRLi4faLhm9wfdnePo,250
-tinybird/check_pypi.py,sha256=_4NkharLyR_ELrAdit-ftqIWvOf7jZNPt3i76frlo9g,975
+tinybird/__cli__.py,sha256=P6C6n43jJIAsTyrXiOcbX-PiRKUCtqwYqXix7BHTRR8,250
+tinybird/check_pypi.py,sha256=RIhZcTg9dk5SFuvV3yVtFgBael_5_VYl4GzOZJojCfU,971
 tinybird/client.py,sha256=nd97gD2-8Ap8yDonBcVwk9eXDAL43hmIYdo-Pse43RE,50738
 tinybird/config.py,sha256=Z-BX9FrjgsLw1YwcCdF0IztLB97Zpc70VVPplO_pDSY,6089
 tinybird/connectors.py,sha256=lkpVSUmSuViEZBa4QjTK7YmPHUop0a5UFoTrSmlVq6k,15244
 tinybird/context.py,sha256=kutUQ0kCwparowI74_YLXx6wtTzGLRouJ6oGHVBPzBo,1291
-tinybird/datafile.py,sha256=JpZX33Ilq71PiuDBs4TNF1xJoUMtTPLynIbHuo7Bn80,253510
+tinybird/datafile.py,sha256=dZIIp-V99yS9bnOpH4yAiohG76qKOcJF2wDj7WroT34,253516
 tinybird/datatypes.py,sha256=IHyhZ86ib54Vnd1pbod9y2aS8DDvDKZm1HJGlThdbuQ,10460
-tinybird/feedback_manager.py,sha256=FT5wJ06gYPGofVgpR-XdZBxUlV9oV94rLx-vzh9ArB8,67724
+tinybird/feedback_manager.py,sha256=qX8yrjoHUL4kjb_IKzLZPPfg4m54XWUZkdVOrv1ktpo,67814
 tinybird/git_settings.py,sha256=XUL9ZUj59-ZVQJDYmMEq4UpnuuOuQOHGlNcX3JgQHjQ,3954
 tinybird/sql.py,sha256=gfRKjdqEygcE1WOTeQ1QV2Jal8Jzl4RSX8fftu1KSEs,45825
 tinybird/sql_template.py,sha256=IqYRfUxDYBCoOYjqqvn--_8QXLv9FSRnJ0bInx7q1Xs,93051
 tinybird/sql_template_fmt.py,sha256=1z-PuqSZXtzso8Z_mPqUc-NxIxUrNUcVIPezNieZk-M,10196
 tinybird/sql_toolset.py,sha256=xS_yD5N_TZT5d4uPcXdeIYX4GQPz7-7wHywMGdfgqcM,13794
 tinybird/syncasync.py,sha256=fAvq0qkRgqXqXMKwbY2iJNYqLT_r6mDsh1MRpGKrdRU,27763
-tinybird/tb_cli.py,sha256=cTPJfd_wOWPCwwFz_LoMREV3YSa8pltU_y-ITP13IpY,819
+tinybird/tb_cli.py,sha256=z8TUjtki-2YXj04kry_uKGTRto35qPSibNH1Up6ooRE,855
 tinybird/tornado_template.py,sha256=o2HguxrL1Evnt8o3IvrsI8Zm6JtRQ3zhLJKf1XyR3SQ,41965
 tinybird/ch_utils/constants.py,sha256=aYvg2C_WxYWsnqPdZB1ZFoIr8ZY-XjUXYyHKE9Ansj0,3890
 tinybird/ch_utils/engine.py,sha256=OXkBhlzGjZotjD0vaT-rFIbSGV4tpiHxE8qO_ip0SyQ,40454
 tinybird/tb_cli_modules/auth.py,sha256=cqxfGgFheuTmenQ3UwPBXTqwMm8JD7uzgLfoIRXdnyQ,9038
 tinybird/tb_cli_modules/branch.py,sha256=Ik8rRVPXvhyChHBqdPVNtI_C-0gLYjUHaznNWE04Ecw,39120
-tinybird/tb_cli_modules/build.py,sha256=pF6Ts2k0qCwCi1ktrGyJo1NcdWsU9O3okDdb3SiFdZo,7615
+tinybird/tb_cli_modules/build.py,sha256=HLcjBiBB2bjWHZK2cbdI_4tCnld_D8dHVrWA0sQEnH0,7803
 tinybird/tb_cli_modules/cicd.py,sha256=0lMkb6CVOFZl5HOwgY8mK4T4mgI7O8335UngLXtCc-c,13851
-tinybird/tb_cli_modules/cli.py,sha256=a738xz4DUNep_mzHTEUExd0NAuf7cHkUAlMsEus9xaM,65072
-tinybird/tb_cli_modules/common.py,sha256=pS7kisPe4vChirSY2og5e0Cu-RNDow7X3Qdk1BEHL78,78571
-tinybird/tb_cli_modules/config.py,sha256=6NTgIdwf0X132A1j6G_YrdPep87ymZ9b5pABabKLzh4,11484
+tinybird/tb_cli_modules/cli.py,sha256=jBex1ocNkth1REPHxvvyUaPKj5CTkqg8TLeVFtAw5B8,61973
+tinybird/tb_cli_modules/common.py,sha256=JfQp02DZFtlMoJyfaXkRAGBetadPqda2x-C1kvWC87U,79110
+tinybird/tb_cli_modules/config.py,sha256=ppWvACHrSLkb5hOoQLYNby2w8jR76-8Kx2NBCst7ntQ,11760
 tinybird/tb_cli_modules/connection.py,sha256=YggP34Qh3hCjD41lp1ZHVCwHPjI_-um2spvEV8F_tSU,28684
-tinybird/tb_cli_modules/create.py,sha256=wcr-GIgCz_t135UIDFt_gNxUQ0huI3oXGM5bIuC665E,7808
+tinybird/tb_cli_modules/create.py,sha256=glQiRQEVGTOw9VmoICEF5_7YWuraJrk3BSUDSTSUfP0,6796
 tinybird/tb_cli_modules/datasource.py,sha256=BVYwPkKvdnEv2YMkxof7n7FOffyJm-QK7Pk9Ui-MC3A,35816
 tinybird/tb_cli_modules/exceptions.py,sha256=pmucP4kTF4irIt7dXiG-FcnI-o3mvDusPmch1L8RCWk,3367
 tinybird/tb_cli_modules/fmt.py,sha256=_xL2AyGo3us7NpmfAXtSIhn1cGAC-A4PspdhUjm58jY,3382
 tinybird/tb_cli_modules/job.py,sha256=AG69LPb9MbobA1awwJFZJvxqarDKfRlsBjw2V1zvYqc,2964
-tinybird/tb_cli_modules/llm.py,sha256=RyUBtyLJcG5DFBjkalFLgwwX-415_Uom8Hq_oKiiyBs,904
-tinybird/tb_cli_modules/local.py,sha256=uamCRYoU0MhCaPUXtvCFfbzHJGwTaPMTTHRQyw0Da3k,3870
+tinybird/tb_cli_modules/llm.py,sha256=U6GaLVi4tOq39W60A0Rjfr9JOkGcJoTfeWLixNA06zY,2450
+tinybird/tb_cli_modules/local.py,sha256=cHT8xtF4aYsR7kLKT6hqzhass3W1dl-VCYavJphmnUM,5761
+tinybird/tb_cli_modules/mock.py,sha256=JGB-Hid1qWYuUk-F7NR3tltDSVf5mz6tOEPYXVhPaFs,2089
 tinybird/tb_cli_modules/pipe.py,sha256=BCLAQ3ZuWKGAih2mupnw_Y2S5B5cNS-epF317whsSEE,30989
-tinybird/tb_cli_modules/prompts.py,sha256=Esjhet-KUM1bT6RStk2voCCVjvAOUD4hDDiKpn-AYNY,7227
+tinybird/tb_cli_modules/prompts.py,sha256=12Abh1kZItxMa7r3WbTi_TgTYtlV30HcaG_JiodafIA,7255
 tinybird/tb_cli_modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
+tinybird/tb_cli_modules/table.py,sha256=hG-PRDVuFp2uph41WpoLRV1yjp3RI2fi_iGGiI0rdxU,7695
 tinybird/tb_cli_modules/tag.py,sha256=9YHnruPnUNp1IJUe4qcSEMg9EbquIRo--Nxcsbvkvq8,3488
 tinybird/tb_cli_modules/telemetry.py,sha256=iEGnMuCuNhvF6ln__j6X9MSTwL_0Hm-GgFHHHvhfknk,10466
 tinybird/tb_cli_modules/test.py,sha256=Vf8oK96V81HdKGsT79y6MUz6oz_VrYIwTbRnzzJs4rQ,4350
@@ -43,8 +45,8 @@ tinybird/tb_cli_modules/workspace.py,sha256=N8f1dl4BYc34ucmJ6oNZc4XZMGKd8m0Fhq7o
 tinybird/tb_cli_modules/workspace_members.py,sha256=ksXsjd233y9-sNlz4Qb-meZbX4zn1B84e_bSm2i8rhg,8731
 tinybird/tb_cli_modules/tinyunit/tinyunit.py,sha256=_ydOD4WxmKy7_h-d3fG62w_0_lD0uLl9w4EbEYtwd0U,11720
 tinybird/tb_cli_modules/tinyunit/tinyunit_lib.py,sha256=hGh1ZaXC1af7rKnX7222urkj0QJMhMWclqMy59dOqwE,1922
-tinybird-0.0.1.dev3.dist-info/METADATA,sha256=6J__4w3BceyDNH4mybjldWRwFGgNaDLSyvugF9Se_E4,2404
-tinybird-0.0.1.dev3.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tinybird-0.0.1.dev3.dist-info/entry_points.txt,sha256=PKPKuPmA4IfJYnCFHHUiw-aAWZuBomFvwCklv1OyCjE,43
-tinybird-0.0.1.dev3.dist-info/top_level.txt,sha256=pgw6AzERHBcW3YTi2PW4arjxLkulk2msOz_SomfOEuc,45
-tinybird-0.0.1.dev3.dist-info/RECORD,,
+tinybird-0.0.1.dev5.dist-info/METADATA,sha256=0262NudObi6F7EKjaFVPfSl5_ZQWPd8r4OYG-4TMsMg,2404
+tinybird-0.0.1.dev5.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tinybird-0.0.1.dev5.dist-info/entry_points.txt,sha256=PKPKuPmA4IfJYnCFHHUiw-aAWZuBomFvwCklv1OyCjE,43
+tinybird-0.0.1.dev5.dist-info/top_level.txt,sha256=pgw6AzERHBcW3YTi2PW4arjxLkulk2msOz_SomfOEuc,45
+tinybird-0.0.1.dev5.dist-info/RECORD,,