tinybird 0.0.1.dev2__py3-none-any.whl → 0.0.1.dev4__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.dev2'
-__revision__ = 'bdbc260'
+__version__ = '0.0.1.dev4'
+__revision__ = '388c81a'

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}")
@@ -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_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,17 @@ 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.config import CLIConfig
+from tinybird.tb_cli_modules.llm import LLM
 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 +42,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 +63,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 +101,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)
@@ -157,13 +154,20 @@ async def build(
         )
 
         for filename in filenames:
-            if filename.endswith(".datasource"):
+            if filename.endswith(".datasource") and not skip_datasources:
                 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)
+                llm_config = CLIConfig.get_llm_config()
+                llm = LLM(key=llm_config["api_key"])
+                has_json_path = "`json:" in datasource_content
+
+                if has_json_path:
+                    await llm.generate_sql_sample_data(
+                        tb_client=tb_client,
+                        datasource_name=ds_name,
+                        datasource_content=datasource_content,
+                    )
 
         if watch:
             filename = filenames[0]
@@ -176,7 +180,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 +195,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/common.py

@@ -153,7 +153,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 +1090,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 +1161,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

@@ -0,0 +1,72 @@
+import asyncio
+import json
+from typing import Any, Awaitable, Callable, Dict, List
+
+from openai import OpenAI
+from pydantic import BaseModel
+
+from tinybird.client import TinyB
+from tinybird.tb_cli_modules.prompts import create_project_prompt, sample_data_sql_prompt
+
+
+class DataFile(BaseModel):
+    name: str
+    content: str
+
+
+class DataProject(BaseModel):
+    datasources: List[DataFile]
+    pipes: List[DataFile]
+
+
+class LLM:
+    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:
+        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_name: str, datasource_content: str, row_count: int = 20
+    ) -> str:
+        async def action_fn():
+            response = self.client.chat.completions.create(
+                model="gpt-4o-mini",
+                messages=[
+                    {"role": "system", "content": sample_data_sql_prompt.format(row_count=row_count)},
+                    {"role": "user", "content": datasource_content},
+                ],
+            )
+
+            sql = response.choices[0].message.content or ""
+            result = await tb_client.query(f"{sql} FORMAT JSON")
+            return result.get("data", [])
+
+        async def checker_fn(sample_data: List[Dict[str, Any]]):
+            ndjson_data = "\n".join([json.dumps(row) for row in sample_data])
+            try:
+                result = await tb_client.datasource_events(datasource_name, ndjson_data)
+                return result.get("successful_rows", 0) > 0
+            except Exception:
+                return False
+
+        return await self._execute(action_fn, checker_fn)

tinybird/tb_cli_modules/local.py

@@ -0,0 +1,167 @@
+import time
+
+import click
+import requests
+
+import docker
+from tinybird.feedback_manager import FeedbackManager
+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_HOST = f"http://localhost:{TB_LOCAL_PORT}"
+
+
+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 = 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:
+        if container:
+            container.remove(force=True)
+
+        container = docker_client.containers.run(
+            TB_IMAGE_NAME,
+            name=TB_CONTAINER_NAME,
+            detach=True,
+            ports={"80/tcp": TB_LOCAL_PORT},
+            remove=False,
+            platform="linux/amd64",
+        )
+
+    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
+            # dont parse the json as docker sometimes returns warning messages
+            # todo: rafa, make this rigth
+            if b'"healthcheck": 1' in run:
+                break
+            raise RuntimeError("Unexpected response from Tinybird")
+        except Exception:
+            if attempt == 9:  # Last attempt
+                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."""
+    try:
+        client = docker.from_env()
+        client.ping()
+        return client
+    except Exception:
+        raise CLIException("Docker is not running or installed. Please ensure Docker is installed and running.")
+
+
+def stop_tinybird_local(docker_client):
+    """Stop the Tinybird container."""
+    try:
+        container = docker_client.containers.get(TB_CONTAINER_NAME)
+        container.stop()
+    except Exception:
+        pass
+
+
+def remove_tinybird_local(docker_client):
+    """Remove the Tinybird container."""
+    try:
+        container = docker_client.containers.get(TB_CONTAINER_NAME)
+        container.remove(force=True)
+    except Exception:
+        pass
+
+
+def get_tinybird_local_client():
+    """Get a Tinybird client connected to the local environment."""
+    config = CLIConfig.get_project_config()
+    tokens = requests.get(f"{TB_LOCAL_HOST}/tokens").json()
+    token = tokens["workspace_admin_token"]
+    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-0.0.1.dev2.dist-info → tinybird-0.0.1.dev4.dist-info}/METADATA

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

{tinybird-0.0.1.dev2.dist-info → tinybird-0.0.1.dev4.dist-info}/RECORD

@@ -1,12 +1,12 @@
-tinybird/__cli__.py,sha256=PHs2gdn0u2epRONYAzC2TOYLhI2SR6FPDDS5NDrS0TY,250
+tinybird/__cli__.py,sha256=IVqhagi6E4quIGa9lfJelt_mgVbNil0_gOfvItSxOE0,250
 tinybird/check_pypi.py,sha256=_4NkharLyR_ELrAdit-ftqIWvOf7jZNPt3i76frlo9g,975
 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=Eey5Ih84DBKspjfzIzWpZbpJkkaIqRrpPSljbm7hfNo,67822
 tinybird/git_settings.py,sha256=XUL9ZUj59-ZVQJDYmMEq4UpnuuOuQOHGlNcX3JgQHjQ,3954
 tinybird/sql.py,sha256=gfRKjdqEygcE1WOTeQ1QV2Jal8Jzl4RSX8fftu1KSEs,45825
 tinybird/sql_template.py,sha256=IqYRfUxDYBCoOYjqqvn--_8QXLv9FSRnJ0bInx7q1Xs,93051
@@ -19,17 +19,19 @@ tinybird/ch_utils/constants.py,sha256=aYvg2C_WxYWsnqPdZB1ZFoIr8ZY-XjUXYyHKE9Ansj
 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=FVo3531YlGj1lHm1aI4aYepgn0ffbPfzSUM1lloY1a4,8589
 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/common.py,sha256=VxzCcCaVoUnwvO-AN6JqeEHK9AzichzSgN_QVgqLHHM,78413
+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=epJqg4peovBS2urlyGUccmvuyy3dp57SnYkatTGcD_I,2483
+tinybird/tb_cli_modules/local.py,sha256=52XgRCUqK8_07iXLLLnmP5KiAWVoXNbuCxlnobPcTQQ,5718
 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/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
@@ -41,8 +43,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.dev2.dist-info/METADATA,sha256=C0w-b-Ur4VfPVJya8uxCZvv2cBxlgaDePZnV2m2-Y9k,2404
-tinybird-0.0.1.dev2.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tinybird-0.0.1.dev2.dist-info/entry_points.txt,sha256=PKPKuPmA4IfJYnCFHHUiw-aAWZuBomFvwCklv1OyCjE,43
-tinybird-0.0.1.dev2.dist-info/top_level.txt,sha256=pgw6AzERHBcW3YTi2PW4arjxLkulk2msOz_SomfOEuc,45
-tinybird-0.0.1.dev2.dist-info/RECORD,,
+tinybird-0.0.1.dev4.dist-info/METADATA,sha256=LGWZlh2D8xM9K8TG6sURXP2cB5r861y8Zw6EJm1-W2s,2404
+tinybird-0.0.1.dev4.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tinybird-0.0.1.dev4.dist-info/entry_points.txt,sha256=PKPKuPmA4IfJYnCFHHUiw-aAWZuBomFvwCklv1OyCjE,43
+tinybird-0.0.1.dev4.dist-info/top_level.txt,sha256=pgw6AzERHBcW3YTi2PW4arjxLkulk2msOz_SomfOEuc,45
+tinybird-0.0.1.dev4.dist-info/RECORD,,