tinybird 0.0.1.dev241__py3-none-any.whl → 0.0.1.dev244__py3-none-any.whl

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.dev241'
-__revision__ = 'b44ab3b'
+__version__ = '0.0.1.dev244'
+__revision__ = '434c7ff'

tinybird/tb/modules/agent/agent.py

@@ -1,5 +1,8 @@
 import sys
+import uuid
 from datetime import datetime
+from functools import partial
+from typing import Any
 
 import click
 from prompt_toolkit import prompt
@@ -22,23 +25,33 @@ from tinybird.prompts import (
     sink_pipe_instructions,
 )
 from tinybird.tb.client import TinyB
+from tinybird.tb.modules.agent.animations import ThinkingAnimation
 from tinybird.tb.modules.agent.banner import display_banner
 from tinybird.tb.modules.agent.memory import clear_history, load_history
 from tinybird.tb.modules.agent.models import create_model
-from tinybird.tb.modules.agent.prompts import datafile_instructions, plan_instructions, sql_instructions
+from tinybird.tb.modules.agent.prompts import (
+    datafile_instructions,
+    plan_instructions,
+    resources_prompt,
+    sql_instructions,
+)
 from tinybird.tb.modules.agent.tools.create_datafile import create_datafile
 from tinybird.tb.modules.agent.tools.explore import explore_data
 from tinybird.tb.modules.agent.tools.plan import plan
 from tinybird.tb.modules.agent.tools.preview_datafile import preview_datafile
 from tinybird.tb.modules.agent.utils import TinybirdAgentContext
+from tinybird.tb.modules.build_common import process as build_process
+from tinybird.tb.modules.exceptions import CLIBuildException
 from tinybird.tb.modules.feedback_manager import FeedbackManager
+from tinybird.tb.modules.local_common import get_tinybird_local_client
+from tinybird.tb.modules.project import Project
 
 
 class TinybirdAgent:
-    def __init__(self, token: str, host: str, folder: str):
+    def __init__(self, token: str, host: str, project: Project):
         self.token = token
         self.host = host
-        self.folder = folder
+        self.project = project
         self.messages: list[ModelMessage] = []
         self.agent = Agent(
             model=create_model(token, host),
@@ -82,9 +95,10 @@ You have access to the following tools:
 4. Without asking, use the `create_datafile` tool to create the datafile, because it will ask for confirmation before creating the file.
 5. Check the result of the `create_datafile` tool to see if the datafile was created successfully.
 6. If the datafile was created successfully, report the result to the user.
-7. If the datafile was not created successfully, finish the process and just wait for a new user prompt.
+7. If the datafile was not created, finish the process and just wait for a new user prompt.
+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, DO NOT continue with the plan. Stop the process and wait for the user before using any other tool.
+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.
 
 # When planning the creation or update of resources:
 {plan_instructions}
@@ -125,7 +139,7 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
                 Tool(explore_data, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
                 Tool(preview_datafile, docstring_format="google", require_parameter_descriptions=True, takes_ctx=False),
                 Tool(create_datafile, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
-                Tool(plan, docstring_format="google", require_parameter_descriptions=True, takes_ctx=False),
+                Tool(plan, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
             ],
         )
 
@@ -133,29 +147,40 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
         """Keep only the last 5 messages to manage token usage."""
         return self.messages[-5:] if len(self.messages) > 5 else self.messages
 
-    def run(self, user_prompt: str) -> None:
+    def run(self, user_prompt: str, project: Project) -> None:
+        user_prompt = f"{user_prompt}\n\n# Existing resources in the project:\n{resources_prompt(project)}"
         client = TinyB(token=self.token, host=self.host)
+        folder = self.project.folder
+        thinking_animation = ThinkingAnimation(message="Chirping", delay=0.15)
+        thinking_animation.start()
+
         result = self.agent.run_sync(
             user_prompt,
             deps=TinybirdAgentContext(
                 # context does not support the whole client, so we need to pass only the functions we need
                 explore_data=client.explore_data,
-                folder=self.folder,
+                build_project=partial(build_project, folder=folder),
+                get_project_files=project.get_project_files,
+                folder=folder,
+                thinking_animation=thinking_animation,
             ),
             message_history=self.messages,
         )
         new_messages = result.new_messages()
         self.messages.extend(new_messages)
-        click.echo("\n")
+        thinking_animation.stop()
+
         click.echo(result.output)
         click.echo("\n")
 
 
-def run_agent(token: str, host: str, folder: str):
+def run_agent(config: dict[str, Any], project: Project):
     display_banner()
 
     try:
-        agent = TinybirdAgent(token, host, folder)
+        token = config["token"]
+        host = config["host"]
+        agent = TinybirdAgent(token, host, project)
         click.echo()
         click.echo(FeedbackManager.success(message="Welcome to Tinybird Code"))
         click.echo(FeedbackManager.info(message="Describe what you want to create and I'll help you build it"))
@@ -200,7 +225,7 @@ def run_agent(token: str, host: str, folder: str):
                 elif user_input.strip() == "":
                     continue
                 else:
-                    agent.run(user_input)
+                    agent.run(user_input, project)
 
             except KeyboardInterrupt:
                 click.echo(FeedbackManager.info(message="Goodbye!"))
@@ -212,3 +237,12 @@ def run_agent(token: str, host: str, folder: str):
     except Exception as e:
         click.echo(FeedbackManager.error(message=f"Error: {e}"))
         sys.exit(1)
+
+
+def build_project(folder: str) -> None:
+    workspace_name = f"tmp_workspace_{uuid.uuid4()}"
+    project = Project(folder, workspace_name=workspace_name)
+    local_client = get_tinybird_local_client({"path": folder, "name": workspace_name}, test=True, silent=True)
+    build_error = build_process(project=project, tb_client=local_client, watch=False, silent=True, exit_on_error=False)
+    if build_error:
+        raise CLIBuildException(build_error)

tinybird/tb/modules/agent/animations.py

@@ -0,0 +1,86 @@
+import random
+import sys
+import threading
+from time import sleep
+from typing import Optional
+
+import click
+
+
+class ThinkingAnimation:
+    """Thinking animation that shows changing sparkles as a prefix."""
+
+    def __init__(
+        self,
+        message: str = "Thinking",
+        delay: float = 0.15,
+        colors: bool = True,
+        dots: bool = True,
+    ):
+        self.message = message
+        self.delay = delay
+        self.running = False
+        self.thread: Optional[threading.Thread] = None
+        self.colors = colors and sys.stdout.isatty()
+        self.dots = dots
+
+        # Default sparkle characters
+        self.sparkle_chars = ["✧", "✦", "⋆", "✳", "✺", "✹", "*", "·"]
+
+        # ANSI color codes
+        self.colors_list = [
+            "\033[33m",  # Yellow
+            "\033[36m",  # Cyan
+            "\033[35m",  # Magenta
+            "\033[93m",  # Bright Yellow
+            "\033[96m",  # Bright Cyan
+        ]
+        self.reset_color = "\033[0m"
+
+    def start(self):
+        """Start the animation in a separate thread."""
+        self.running = True
+        self.thread = threading.Thread(target=self._run_animation)
+        self.thread.daemon = True
+
+        click.echo("\n")
+        self.thread.start()
+
+    def stop(self):
+        """Stop the animation."""
+        self.running = False
+        if self.thread:
+            self.thread.join()
+        # Clear the line and reset cursor position
+        sys.stdout.write("\r" + " " * (len(self.message) + 10) + "\r")
+        sys.stdout.flush()
+
+    def _run_animation(self):
+        """Run the animation until stopped."""
+        frame_count = 0
+        dots_count = 0
+
+        while self.running:
+            # Choose a random sparkle for this frame
+            sparkle = random.choice(self.sparkle_chars)
+
+            if self.colors:
+                color = random.choice(self.colors_list)
+                colored_sparkle = f"{color}{sparkle}{self.reset_color}"
+            else:
+                colored_sparkle = sparkle
+
+            # Handle dots animation if enabled
+            if self.dots:
+                dots_count = (frame_count // 4) % 4  # Change dots every 4 frames
+                dots = "." * dots_count
+                display_message = f"{self.message}{dots}"
+            else:
+                display_message = self.message
+
+            # Print the message with the prefix sparkle
+            sys.stdout.write(f"\r{colored_sparkle} {display_message}")
+            sys.stdout.flush()
+
+            sleep(self.delay)
+            frame_count += 1

tinybird/tb/modules/agent/prompts.py

@@ -1,3 +1,7 @@
+from pathlib import Path
+
+from tinybird.tb.modules.project import Project
+
 plan_instructions = """
 When asked to create a plan, you MUST respond with this EXACT format and NOTHING ELSE:
 
@@ -81,3 +85,35 @@ datafile_instructions = """
 - Datasource files will be created under the `/datasources` folder.
 </datafile_instructions>
 """
+
+
+def resources_prompt(project: Project) -> str:
+    files = project.get_project_files()
+
+    if not files:
+        return "No resources found"
+
+    paths = [Path(file_path) for file_path in files]
+
+    def get_resource_type(path: Path) -> str:
+        if path.suffix.lower() == ".pipe":
+            return Project.get_pipe_type(str(path))
+        elif path.suffix.lower() == ".datasource":
+            return "datasource"
+        elif path.suffix.lower() == ".connection":
+            return "connection"
+        return "unknown"
+
+    return "\n".join(
+        [
+            f"""
+<resource>
+    <path>{file_path.relative_to(project.folder)}</path>
+    <type>{get_resource_type(file_path)}</type>
+    <name>{file_path.stem}</name>
+    <content>{file_path.read_text()}</content>
+</resource>
+"""
+            for file_path in paths
+        ]
+    )

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

@@ -4,13 +4,16 @@ import click
 from pydantic_ai import RunContext
 
 from tinybird.tb.modules.agent.utils import Datafile, TinybirdAgentContext, show_options
+from tinybird.tb.modules.exceptions import CLIBuildException
+from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 
-def get_resource_confirmation(resource: Datafile) -> bool:
+def get_resource_confirmation(resource: Datafile, exists: bool) -> bool:
     """Get user confirmation for creating a resource"""
     while True:
+        action = "create" if not exists else "update"
         result = show_options(
-            options=[f"Yes, create {resource.type} '{resource.name}'", "No, and tell Tinybird Code what to do"],
+            options=[f"Yes, {action} {resource.type} '{resource.name}'", "No, and tell Tinybird Code what to do"],
             title=f"What would you like to do with {resource.type} '{resource.name}'?",
         )
 
@@ -35,28 +38,29 @@ def create_datafile(ctx: RunContext[TinybirdAgentContext], resource: Datafile) -
         str: If the resource was created or not.
     """
     try:
-        click.echo()
+        ctx.deps.thinking_animation.stop()
         click.echo(resource.content)
-        confirmation = get_resource_confirmation(resource)
+        resource.pathname = resource.pathname.removeprefix("/")
+        path = Path(ctx.deps.folder) / resource.pathname
+        exists = str(path) in ctx.deps.get_project_files()
+        confirmation = get_resource_confirmation(resource, exists)
+        ctx.deps.thinking_animation.start()
 
         if not confirmation:
             return f"Resource {resource.pathname} was not created. User cancelled creation."
 
-        resource.pathname = resource.pathname.removeprefix("/")
-
-        path = Path(ctx.deps.folder) / resource.pathname
-
         folder_path = path.parent
-
-        if not folder_path.exists():
-            folder_path.mkdir()
-
-        if not path.exists():
-            path.touch()
+        folder_path.mkdir(parents=True, exist_ok=True)
+        path.touch(exist_ok=True)
 
         path.write_text(resource.content)
-
+        ctx.deps.build_project()
         return f"Created {resource.pathname}"
 
+    except CLIBuildException as e:
+        ctx.deps.thinking_animation.stop()
+        click.echo(FeedbackManager.error(message=e))
+        ctx.deps.thinking_animation.start()
+        return f"Error building project: {e}"
     except Exception as e:
         return f"Error creating {resource.pathname}: {e}"

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

@@ -1,6 +1,7 @@
 import click
+from pydantic_ai import RunContext
 
-from tinybird.tb.modules.agent.utils import show_options
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_options
 
 
 def get_plan_confirmation() -> bool:
@@ -22,7 +23,7 @@ def get_plan_confirmation() -> bool:
         return False
 
 
-def plan(plan: str) -> str:
+def plan(ctx: RunContext[TinybirdAgentContext], plan: str) -> str:
     """Given a plan, ask the user for confirmation to implement it
 
     Args:
@@ -32,9 +33,10 @@ def plan(plan: str) -> str:
         str: If the plan was implemented or not.
     """
     try:
-        click.echo()
+        ctx.deps.thinking_animation.stop()
         click.echo(plan)
         confirmation = get_plan_confirmation()
+        ctx.deps.thinking_animation.start()
 
         if not confirmation:
             return "Plan was not implemented. User cancelled implementation."

tinybird/tb/modules/agent/utils.py

@@ -20,6 +20,9 @@ from pydantic import BaseModel, Field
 class TinybirdAgentContext(BaseModel):
     explore_data: Callable[[str], str]
     folder: str
+    build_project: Callable[[], None]
+    thinking_animation: Any
+    get_project_files: Callable[[], List[str]]
 
 
 default_style = Style.from_dict(

tinybird/tb/modules/build.py

@@ -1,30 +1,26 @@
-import json
-import logging
 import threading
 import time
 from copy import deepcopy
 from functools import partial
 from pathlib import Path
-from typing import Callable, List, Optional
+from typing import Callable, List
 from urllib.parse import urlencode
 
 import click
-import requests
 
 import tinybird.context as context
 from tinybird.datafile.exceptions import ParseException
 from tinybird.datafile.parse_datasource import parse_datasource
 from tinybird.datafile.parse_pipe import parse_pipe
 from tinybird.tb.client import TinyB
+from tinybird.tb.modules.build_common import process
 from tinybird.tb.modules.cli import cli
-from tinybird.tb.modules.common import push_data, sys_exit
 from tinybird.tb.modules.config import CLIConfig
-from tinybird.tb.modules.datafile.fixture import FixtureExtension, get_fixture_dir, persist_fixture
 from tinybird.tb.modules.datafile.playground import folder_playground
 from tinybird.tb.modules.dev_server import BuildStatus, start_server
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 from tinybird.tb.modules.project import Project
-from tinybird.tb.modules.secret import load_secrets
+from tinybird.tb.modules.secret_common import load_secrets
 from tinybird.tb.modules.shell import Shell, print_table_formatted
 from tinybird.tb.modules.watch import watch_files, watch_project
 
@@ -87,275 +83,6 @@ def dev(ctx: click.Context, data_origin: str, ui: bool) -> None:
     )
 
 
-def build_project(project: Project, tb_client: TinyB, silent: bool = False) -> Optional[bool]:
-    MULTIPART_BOUNDARY_DATA_PROJECT = "data_project://"
-    DATAFILE_TYPE_TO_CONTENT_TYPE = {
-        ".datasource": "text/plain",
-        ".pipe": "text/plain",
-        ".connection": "text/plain",
-    }
-    TINYBIRD_API_URL = tb_client.host + "/v1/build"
-    logging.debug(TINYBIRD_API_URL)
-    TINYBIRD_API_KEY = tb_client.token
-    error: Optional[str] = None
-    try:
-        files = [
-            ("context://", ("cli-version", "1.0.0", "text/plain")),
-        ]
-        project_path = project.path
-        project_files = project.get_project_files()
-
-        if not project_files:
-            return False
-
-        for file_path in project_files:
-            relative_path = Path(file_path).relative_to(project_path).as_posix()
-            with open(file_path, "rb") as fd:
-                content_type = DATAFILE_TYPE_TO_CONTENT_TYPE.get(Path(file_path).suffix, "application/unknown")
-                files.append(
-                    (MULTIPART_BOUNDARY_DATA_PROJECT, (relative_path, fd.read().decode("utf-8"), content_type))
-                )
-        HEADERS = {"Authorization": f"Bearer {TINYBIRD_API_KEY}"}
-
-        r = requests.post(TINYBIRD_API_URL, files=files, headers=HEADERS)
-        try:
-            result = r.json()
-        except Exception as e:
-            logging.debug(e, exc_info=True)
-            click.echo(FeedbackManager.error(message="Couldn't parse response from server"))
-            sys_exit("build_error", str(e))
-
-        logging.debug(json.dumps(result, indent=2))
-
-        build_result = result.get("result")
-        if build_result == "success":
-            build = result.get("build")
-            new_datasources = build.get("new_datasource_names", [])
-            new_pipes = build.get("new_pipe_names", [])
-            new_connections = build.get("new_data_connector_names", [])
-            changed_datasources = build.get("changed_datasource_names", [])
-            changed_pipes = build.get("changed_pipe_names", [])
-            changed_connections = build.get("changed_data_connector_names", [])
-            deleted_datasources = build.get("deleted_datasource_names", [])
-            deleted_pipes = build.get("deleted_pipe_names", [])
-            deleted_connections = build.get("deleted_data_connector_names", [])
-
-            no_changes = (
-                not new_datasources
-                and not changed_datasources
-                and not new_pipes
-                and not changed_pipes
-                and not new_connections
-                and not changed_connections
-                and not deleted_datasources
-                and not deleted_pipes
-                and not deleted_connections
-            )
-            if no_changes:
-                return False
-            else:
-                if not silent:
-                    echo_changes(project, new_datasources, ".datasource", "created")
-                    echo_changes(project, changed_datasources, ".datasource", "changed")
-                    echo_changes(project, deleted_datasources, ".datasource", "deleted")
-                    echo_changes(project, new_pipes, ".pipe", "created")
-                    echo_changes(project, changed_pipes, ".pipe", "changed")
-                    echo_changes(project, deleted_pipes, ".pipe", "deleted")
-                    echo_changes(project, new_connections, ".connection", "created")
-                    echo_changes(project, changed_connections, ".connection", "changed")
-                    echo_changes(project, deleted_connections, ".connection", "deleted")
-            try:
-                for filename in project_files:
-                    if filename.endswith(".datasource"):
-                        ds_path = Path(filename)
-                        ds_name = ds_path.stem
-                        fixture_folder = get_fixture_dir(project.folder)
-                        fixture_extensions = [FixtureExtension.NDJSON, FixtureExtension.CSV]
-                        fixture_path = next(
-                            (
-                                fixture_folder / f"{ds_name}{ext}"
-                                for ext in fixture_extensions
-                                if (fixture_folder / f"{ds_name}{ext}").exists()
-                            ),
-                            None,
-                        )
-                        if not fixture_path:
-                            sql_path = fixture_folder / f"{ds_name}.sql"
-                            if sql_path.exists():
-                                fixture_path = rebuild_fixture_sql(project, tb_client, str(sql_path))
-
-                        if fixture_path:
-                            append_fixture(tb_client, ds_name, str(fixture_path))
-
-            except Exception as e:
-                click.echo(FeedbackManager.error_exception(error=f"Error appending fixtures for '{ds_name}': {e}"))
-
-            feedback = build.get("feedback", [])
-            for f in feedback:
-                click.echo(
-                    FeedbackManager.warning(message=f"△ {f.get('level')}: {f.get('resource')}: {f.get('message')}")
-                )
-        elif build_result == "failed":
-            build_errors = result.get("errors")
-            full_error_msg = ""
-            for build_error in build_errors:
-                filename_bit = build_error.get("filename", build_error.get("resource", ""))
-                error_bit = build_error.get("error") or build_error.get("message") or ""
-                error_msg = ((filename_bit + "\n") if filename_bit else "") + error_bit
-                full_error_msg += error_msg + "\n\n"
-            error = full_error_msg.strip("\n") or "Unknown build error"
-        else:
-            error = f"Unknown build result. Error: {result.get('error')}"
-    except Exception as e:
-        error = str(e)
-
-    if error:
-        raise click.ClickException(error)
-
-    return build_result
-
-
-def append_fixture(
-    tb_client: TinyB,
-    datasource_name: str,
-    url: str,
-):
-    # Append fixtures only if the datasource is empty
-    data = tb_client._req(f"/v0/datasources/{datasource_name}")
-    if data.get("statistics", {}).get("row_count", 0) > 0:
-        return
-
-    push_data(
-        tb_client,
-        datasource_name,
-        url,
-        mode="append",
-        concurrency=1,
-        silent=True,
-    )
-
-
-def rebuild_fixture(project: Project, tb_client: TinyB, fixture: str) -> None:
-    try:
-        fixture_path = Path(fixture)
-        datasources_path = Path(project.folder) / "datasources"
-        ds_name = fixture_path.stem
-
-        if ds_name not in project.datasources:
-            try:
-                ds_name = "_".join(fixture_path.stem.split("_")[:-1])
-            except Exception:
-                pass
-
-        ds_path = datasources_path / f"{ds_name}.datasource"
-
-        if ds_path.exists():
-            tb_client.datasource_truncate(ds_name)
-            append_fixture(tb_client, ds_name, str(fixture_path))
-    except Exception as e:
-        click.echo(FeedbackManager.error_exception(error=e))
-
-
-def show_data(tb_client: TinyB, filename: str, diff: Optional[str] = None):
-    table_name = diff
-    resource_path = Path(filename)
-    resource_name = resource_path.stem
-
-    pipeline = resource_name if filename.endswith(".pipe") else None
-
-    if not table_name:
-        table_name = resource_name
-
-    sql = f"SELECT * FROM {table_name} FORMAT JSON"
-
-    res = tb_client.query(sql, pipeline=pipeline)
-    print_table_formatted(res, table_name)
-    if Project.get_pipe_type(filename) == "endpoint":
-        example_params = {
-            "format": "json",
-            "pipe": resource_name,
-            "q": "",
-            "token": tb_client.token,
-        }
-        endpoint_url = tb_client._req(f"/examples/query.http?{urlencode(example_params)}")
-        if endpoint_url:
-            endpoint_url = endpoint_url.replace("http://localhost:8001", tb_client.host)
-            click.echo(FeedbackManager.gray(message="\nTest endpoint at ") + FeedbackManager.info(message=endpoint_url))
-
-
-def process(
-    project: Project,
-    tb_client: TinyB,
-    watch: bool,
-    file_changed: Optional[str] = None,
-    diff: Optional[str] = None,
-    silent: bool = False,
-    error: bool = False,
-    build_status: Optional[BuildStatus] = None,
-) -> Optional[str]:
-    time_start = time.time()
-    build_failed = False
-    build_error: Optional[str] = None
-    build_result: Optional[bool] = None
-    if build_status:
-        if build_status.building:
-            return build_status.error
-        else:
-            build_status.building = True
-    if file_changed and (file_changed.endswith(FixtureExtension.NDJSON) or file_changed.endswith(FixtureExtension.CSV)):
-        rebuild_fixture(project, tb_client, file_changed)
-        if build_status:
-            build_status.building = False
-            build_status.error = None
-    elif file_changed and file_changed.endswith(".sql"):
-        rebuild_fixture_sql(project, tb_client, file_changed)
-        if build_status:
-            build_status.building = False
-            build_status.error = None
-    elif file_changed and (file_changed.endswith(".env.local") or file_changed.endswith(".env")):
-        load_secrets(project, tb_client)
-        if build_status:
-            build_status.building = False
-            build_status.error = None
-    else:
-        try:
-            build_result = build_project(project, tb_client, silent)
-            if build_status:
-                build_status.building = False
-                build_status.error = None
-        except click.ClickException as e:
-            click.echo(FeedbackManager.info(message=str(e)))
-            build_error = str(e)
-            build_failed = True
-        try:
-            if file_changed and not build_failed and not build_status:
-                show_data(tb_client, file_changed, diff)
-        except Exception:
-            pass
-
-    time_end = time.time()
-    elapsed_time = time_end - time_start
-
-    rebuild_str = "Rebuild" if watch and file_changed else "Build"
-    if build_failed:
-        click.echo(FeedbackManager.error(message=f"✗ {rebuild_str} failed"))
-        if not watch:
-            sys_exit("build_error", build_error or "Unknown error")
-        build_error = build_error or "Unknown error"
-        if build_status:
-            build_status.error = build_error
-            build_status.building = False
-        return build_error
-    else:
-        if not silent:
-            if build_result == False:  # noqa: E712
-                click.echo(FeedbackManager.info(message="No changes. Build skipped."))
-            else:
-                click.echo(FeedbackManager.success(message=f"\n✓ {rebuild_str} completed in {elapsed_time:.1f}s"))
-
-    return None
-
-
 def run_watch(project: Project, tb_client: TinyB, process: Callable) -> None:
     shell = Shell(project=project, tb_client=tb_client, playground=False)
     click.echo(FeedbackManager.gray(message="\nWatching for changes..."))
@@ -368,27 +95,6 @@ def run_watch(project: Project, tb_client: TinyB, process: Callable) -> None:
     shell.run()
 
 
-def rebuild_fixture_sql(project: Project, tb_client: TinyB, sql_file: str) -> Path:
-    sql_path = Path(sql_file)
-    datasource_name = sql_path.stem
-    valid_extensions = [FixtureExtension.NDJSON, FixtureExtension.CSV]
-    fixtures_path = get_fixture_dir(project.folder)
-    current_fixture_path = next(
-        (
-            fixtures_path / f"{datasource_name}{extension}"
-            for extension in valid_extensions
-            if (fixtures_path / f"{datasource_name}{extension}").exists()
-        ),
-        None,
-    )
-    fixture_format = current_fixture_path.suffix.lstrip(".") if current_fixture_path else "ndjson"
-    sql = sql_path.read_text()
-    sql_format = "CSV" if fixture_format == "csv" else "JSON"
-    result = tb_client.query(f"{sql} FORMAT {sql_format}")
-    data = result.get("data", [])
-    return persist_fixture(datasource_name, data, project.folder, format=fixture_format)
-
-
 def is_vendor(f: Path) -> bool:
     return f.parts[0] == "vendor"
 
@@ -497,13 +203,3 @@ def build_and_print_resource(config: CLIConfig, tb_client: TinyB, filename: str)
     node_sql = last_node["sql"]
     res = tb_client.query(f"{node_sql} FORMAT JSON", playground=playground_id)
     print_table_formatted(res, name)
-
-
-def echo_changes(project: Project, changes: List[str], extension: str, status: str):
-    for resource in changes:
-        path_str = next(
-            (p for p in project.get_project_files() if p.endswith(resource + extension)), resource + extension
-        )
-        if path_str:
-            path_str = path_str.replace(f"{project.folder}/", "")
-            click.echo(FeedbackManager.info(message=f"✓ {path_str} {status}"))

tinybird/tb/modules/build_common.py

@@ -0,0 +1,321 @@
+import json
+import logging
+import time
+from pathlib import Path
+from typing import Optional
+from urllib.parse import urlencode
+
+import click
+import requests
+
+from tinybird.tb.client import TinyB
+from tinybird.tb.modules.common import push_data, sys_exit
+from tinybird.tb.modules.datafile.fixture import FixtureExtension, get_fixture_dir, persist_fixture
+from tinybird.tb.modules.dev_server import BuildStatus
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+from tinybird.tb.modules.project import Project
+from tinybird.tb.modules.secret_common import load_secrets
+from tinybird.tb.modules.shell import print_table_formatted
+
+
+def process(
+    project: Project,
+    tb_client: TinyB,
+    watch: bool,
+    file_changed: Optional[str] = None,
+    diff: Optional[str] = None,
+    silent: bool = False,
+    error: bool = False,
+    build_status: Optional[BuildStatus] = None,
+    exit_on_error: bool = True,
+) -> Optional[str]:
+    time_start = time.time()
+    build_failed = False
+    build_error: Optional[str] = None
+    build_result: Optional[bool] = None
+    if build_status:
+        if build_status.building:
+            return build_status.error
+        else:
+            build_status.building = True
+    if file_changed and (file_changed.endswith(FixtureExtension.NDJSON) or file_changed.endswith(FixtureExtension.CSV)):
+        rebuild_fixture(project, tb_client, file_changed)
+        if build_status:
+            build_status.building = False
+            build_status.error = None
+    elif file_changed and file_changed.endswith(".sql"):
+        rebuild_fixture_sql(project, tb_client, file_changed)
+        if build_status:
+            build_status.building = False
+            build_status.error = None
+    elif file_changed and (file_changed.endswith(".env.local") or file_changed.endswith(".env")):
+        load_secrets(project, tb_client)
+        if build_status:
+            build_status.building = False
+            build_status.error = None
+    else:
+        try:
+            build_result = build_project(project, tb_client, silent)
+            if build_status:
+                build_status.building = False
+                build_status.error = None
+        except click.ClickException as e:
+            if not silent:
+                click.echo(FeedbackManager.info(message=str(e)))
+            build_error = str(e)
+            build_failed = True
+        try:
+            if file_changed and not build_failed and not build_status:
+                show_data(tb_client, file_changed, diff)
+        except Exception:
+            pass
+
+    time_end = time.time()
+    elapsed_time = time_end - time_start
+
+    rebuild_str = "Rebuild" if watch and file_changed else "Build"
+    if build_failed:
+        if not silent:
+            click.echo(FeedbackManager.error(message=f"✗ {rebuild_str} failed"))
+            if not watch and exit_on_error:
+                sys_exit("build_error", build_error or "Unknown error")
+        build_error = build_error or "Unknown error"
+        if build_status:
+            build_status.error = build_error
+            build_status.building = False
+        return build_error
+    else:
+        if not silent:
+            if build_result == False:  # noqa: E712
+                click.echo(FeedbackManager.info(message="No changes. Build skipped."))
+            else:
+                click.echo(FeedbackManager.success(message=f"\n✓ {rebuild_str} completed in {elapsed_time:.1f}s"))
+
+    return None
+
+
+def rebuild_fixture(project: Project, tb_client: TinyB, fixture: str) -> None:
+    try:
+        fixture_path = Path(fixture)
+        datasources_path = Path(project.folder) / "datasources"
+        ds_name = fixture_path.stem
+
+        if ds_name not in project.datasources:
+            try:
+                ds_name = "_".join(fixture_path.stem.split("_")[:-1])
+            except Exception:
+                pass
+
+        ds_path = datasources_path / f"{ds_name}.datasource"
+
+        if ds_path.exists():
+            tb_client.datasource_truncate(ds_name)
+            append_fixture(tb_client, ds_name, str(fixture_path))
+    except Exception as e:
+        click.echo(FeedbackManager.error_exception(error=e))
+
+
+def rebuild_fixture_sql(project: Project, tb_client: TinyB, sql_file: str) -> Path:
+    sql_path = Path(sql_file)
+    datasource_name = sql_path.stem
+    valid_extensions = [FixtureExtension.NDJSON, FixtureExtension.CSV]
+    fixtures_path = get_fixture_dir(project.folder)
+    current_fixture_path = next(
+        (
+            fixtures_path / f"{datasource_name}{extension}"
+            for extension in valid_extensions
+            if (fixtures_path / f"{datasource_name}{extension}").exists()
+        ),
+        None,
+    )
+    fixture_format = current_fixture_path.suffix.lstrip(".") if current_fixture_path else "ndjson"
+    sql = sql_path.read_text()
+    sql_format = "CSV" if fixture_format == "csv" else "JSON"
+    result = tb_client.query(f"{sql} FORMAT {sql_format}")
+    data = result.get("data", [])
+    return persist_fixture(datasource_name, data, project.folder, format=fixture_format)
+
+
+def append_fixture(
+    tb_client: TinyB,
+    datasource_name: str,
+    url: str,
+):
+    # Append fixtures only if the datasource is empty
+    data = tb_client._req(f"/v0/datasources/{datasource_name}")
+    if data.get("statistics", {}).get("row_count", 0) > 0:
+        return
+
+    push_data(
+        tb_client,
+        datasource_name,
+        url,
+        mode="append",
+        concurrency=1,
+        silent=True,
+    )
+
+
+def show_data(tb_client: TinyB, filename: str, diff: Optional[str] = None):
+    table_name = diff
+    resource_path = Path(filename)
+    resource_name = resource_path.stem
+
+    pipeline = resource_name if filename.endswith(".pipe") else None
+
+    if not table_name:
+        table_name = resource_name
+
+    sql = f"SELECT * FROM {table_name} FORMAT JSON"
+
+    res = tb_client.query(sql, pipeline=pipeline)
+    print_table_formatted(res, table_name)
+    if Project.get_pipe_type(filename) == "endpoint":
+        example_params = {
+            "format": "json",
+            "pipe": resource_name,
+            "q": "",
+            "token": tb_client.token,
+        }
+        endpoint_url = tb_client._req(f"/examples/query.http?{urlencode(example_params)}")
+        if endpoint_url:
+            endpoint_url = endpoint_url.replace("http://localhost:8001", tb_client.host)
+            click.echo(FeedbackManager.gray(message="\nTest endpoint at ") + FeedbackManager.info(message=endpoint_url))
+
+
+def build_project(project: Project, tb_client: TinyB, silent: bool = False) -> Optional[bool]:
+    MULTIPART_BOUNDARY_DATA_PROJECT = "data_project://"
+    DATAFILE_TYPE_TO_CONTENT_TYPE = {
+        ".datasource": "text/plain",
+        ".pipe": "text/plain",
+        ".connection": "text/plain",
+    }
+    TINYBIRD_API_URL = tb_client.host + "/v1/build"
+    logging.debug(TINYBIRD_API_URL)
+    TINYBIRD_API_KEY = tb_client.token
+    error: Optional[str] = None
+    try:
+        files = [
+            ("context://", ("cli-version", "1.0.0", "text/plain")),
+        ]
+        project_path = project.path
+        project_files = project.get_project_files()
+
+        if not project_files:
+            return False
+
+        for file_path in project_files:
+            relative_path = Path(file_path).relative_to(project_path).as_posix()
+            with open(file_path, "rb") as fd:
+                content_type = DATAFILE_TYPE_TO_CONTENT_TYPE.get(Path(file_path).suffix, "application/unknown")
+                files.append(
+                    (MULTIPART_BOUNDARY_DATA_PROJECT, (relative_path, fd.read().decode("utf-8"), content_type))
+                )
+        HEADERS = {"Authorization": f"Bearer {TINYBIRD_API_KEY}"}
+
+        r = requests.post(TINYBIRD_API_URL, files=files, headers=HEADERS)
+        try:
+            result = r.json()
+        except Exception as e:
+            logging.debug(e, exc_info=True)
+            click.echo(FeedbackManager.error(message="Couldn't parse response from server"))
+            sys_exit("build_error", str(e))
+
+        logging.debug(json.dumps(result, indent=2))
+
+        build_result = result.get("result")
+        if build_result == "success":
+            build = result.get("build")
+            new_datasources = build.get("new_datasource_names", [])
+            new_pipes = build.get("new_pipe_names", [])
+            new_connections = build.get("new_data_connector_names", [])
+            changed_datasources = build.get("changed_datasource_names", [])
+            changed_pipes = build.get("changed_pipe_names", [])
+            changed_connections = build.get("changed_data_connector_names", [])
+            deleted_datasources = build.get("deleted_datasource_names", [])
+            deleted_pipes = build.get("deleted_pipe_names", [])
+            deleted_connections = build.get("deleted_data_connector_names", [])
+
+            no_changes = (
+                not new_datasources
+                and not changed_datasources
+                and not new_pipes
+                and not changed_pipes
+                and not new_connections
+                and not changed_connections
+                and not deleted_datasources
+                and not deleted_pipes
+                and not deleted_connections
+            )
+            if no_changes:
+                return False
+            else:
+                if not silent:
+                    echo_changes(project, new_datasources, ".datasource", "created")
+                    echo_changes(project, changed_datasources, ".datasource", "changed")
+                    echo_changes(project, deleted_datasources, ".datasource", "deleted")
+                    echo_changes(project, new_pipes, ".pipe", "created")
+                    echo_changes(project, changed_pipes, ".pipe", "changed")
+                    echo_changes(project, deleted_pipes, ".pipe", "deleted")
+                    echo_changes(project, new_connections, ".connection", "created")
+                    echo_changes(project, changed_connections, ".connection", "changed")
+                    echo_changes(project, deleted_connections, ".connection", "deleted")
+            try:
+                for filename in project_files:
+                    if filename.endswith(".datasource"):
+                        ds_path = Path(filename)
+                        ds_name = ds_path.stem
+                        fixture_folder = get_fixture_dir(project.folder)
+                        fixture_extensions = [FixtureExtension.NDJSON, FixtureExtension.CSV]
+                        fixture_path = next(
+                            (
+                                fixture_folder / f"{ds_name}{ext}"
+                                for ext in fixture_extensions
+                                if (fixture_folder / f"{ds_name}{ext}").exists()
+                            ),
+                            None,
+                        )
+                        if not fixture_path:
+                            sql_path = fixture_folder / f"{ds_name}.sql"
+                            if sql_path.exists():
+                                fixture_path = rebuild_fixture_sql(project, tb_client, str(sql_path))
+
+                        if fixture_path:
+                            append_fixture(tb_client, ds_name, str(fixture_path))
+
+            except Exception as e:
+                click.echo(FeedbackManager.error_exception(error=f"Error appending fixtures for '{ds_name}': {e}"))
+
+            feedback = build.get("feedback", [])
+            for f in feedback:
+                click.echo(
+                    FeedbackManager.warning(message=f"△ {f.get('level')}: {f.get('resource')}: {f.get('message')}")
+                )
+        elif build_result == "failed":
+            build_errors = result.get("errors")
+            full_error_msg = ""
+            for build_error in build_errors:
+                filename_bit = build_error.get("filename", build_error.get("resource", ""))
+                error_bit = build_error.get("error") or build_error.get("message") or ""
+                error_msg = ((filename_bit + "\n") if filename_bit else "") + error_bit
+                full_error_msg += error_msg + "\n\n"
+            error = full_error_msg.strip("\n") or "Unknown build error"
+        else:
+            error = f"Unknown build result. Error: {result.get('error')}"
+    except Exception as e:
+        error = str(e)
+
+    if error:
+        raise click.ClickException(error)
+
+    return build_result
+
+
+def echo_changes(project: Project, changes: list[str], extension: str, status: str):
+    for resource in changes:
+        path_str = next(
+            (p for p in project.get_project_files() if p.endswith(resource + extension)), resource + extension
+        )
+        if path_str:
+            path_str = path_str.replace(f"{project.folder}/", "")
+            click.echo(FeedbackManager.info(message=f"✓ {path_str} {status}"))

tinybird/tb/modules/cli.py

@@ -196,7 +196,7 @@ def cli(
     is_agent_mode = agent_mode_flag and ctx.invoked_subcommand is None
 
     if is_agent_mode:
-        run_agent(config.get("token", None), config["host"], str(project.path))
+        run_agent(config, project)
 
 
 @cli.command(hidden=True)

tinybird/tb/modules/secret.py

@@ -1,8 +1,8 @@
 import re
-from typing import Dict, Optional
+from typing import Optional
 
 import click
-from dotenv import dotenv_values, set_key
+from dotenv import set_key
 
 from tinybird.tb.client import TinyB
 from tinybird.tb.modules.cli import cli
@@ -97,54 +97,6 @@ def secret_rm(ctx: click.Context, name: str):
         click.echo(FeedbackManager.error(message=f"✗ Error: {e}"))
 
 
-def load_secrets(project: Project, client: TinyB):
-    try:
-        env_vars: Dict[str, str] = {}
-
-        # Load secrets from .env file
-        env_file = ".env"
-        env_path = project.path / env_file
-
-        if env_path.exists():
-            env_values = dotenv_values(env_path)
-            if env_values:
-                env_vars.update({k: v for k, v in env_values.items() if v is not None})
-
-        # Load secrets from .env.local file
-        env_file = ".env.local"
-        env_path = project.path / env_file
-
-        if env_path.exists():
-            env_values = dotenv_values(env_path)
-            if env_values:
-                env_vars.update({k: v for k, v in env_values.items() if v is not None})
-
-        if len(env_vars.keys()) == 0:
-            return
-
-        click.echo(FeedbackManager.highlight(message="\n» Loading secrets from .env files..."))
-
-        for name, value in env_vars.items():
-            if not value:
-                continue
-
-            try:
-                existing_secret = client.get_secret(name)
-            except Exception:
-                existing_secret = None
-            try:
-                if existing_secret:
-                    client.update_secret(name, value)
-                else:
-                    client.create_secret(name, value)
-            except Exception as e:
-                click.echo(FeedbackManager.error(message=f"✗ Error setting secret '{name}': {e}"))
-
-        click.echo(FeedbackManager.success(message="✓ Secrets loaded!"))
-    except Exception as e:
-        click.echo(FeedbackManager.error(message=f"✗ Error: {e}"))
-
-
 def save_secret_to_env_file(project: Project, name: str, value: str):
     env_path = project.path / ".env.local"
 

tinybird/tb/modules/secret_common.py

@@ -0,0 +1,56 @@
+from typing import Dict
+
+import click
+from dotenv import dotenv_values
+
+from tinybird.tb.client import TinyB
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+from tinybird.tb.modules.project import Project
+
+
+def load_secrets(project: Project, client: TinyB):
+    try:
+        env_vars: Dict[str, str] = {}
+
+        # Load secrets from .env file
+        env_file = ".env"
+        env_path = project.path / env_file
+
+        if env_path.exists():
+            env_values = dotenv_values(env_path)
+            if env_values:
+                env_vars.update({k: v for k, v in env_values.items() if v is not None})
+
+        # Load secrets from .env.local file
+        env_file = ".env.local"
+        env_path = project.path / env_file
+
+        if env_path.exists():
+            env_values = dotenv_values(env_path)
+            if env_values:
+                env_vars.update({k: v for k, v in env_values.items() if v is not None})
+
+        if len(env_vars.keys()) == 0:
+            return
+
+        click.echo(FeedbackManager.highlight(message="\n» Loading secrets from .env files..."))
+
+        for name, value in env_vars.items():
+            if not value:
+                continue
+
+            try:
+                existing_secret = client.get_secret(name)
+            except Exception:
+                existing_secret = None
+            try:
+                if existing_secret:
+                    client.update_secret(name, value)
+                else:
+                    client.create_secret(name, value)
+            except Exception as e:
+                click.echo(FeedbackManager.error(message=f"✗ Error setting secret '{name}': {e}"))
+
+        click.echo(FeedbackManager.success(message="✓ Secrets loaded!"))
+    except Exception as e:
+        click.echo(FeedbackManager.error(message=f"✗ Error: {e}"))

tinybird/tb/modules/test.py

@@ -26,7 +26,7 @@ from tinybird.tb.modules.llm import LLM
 from tinybird.tb.modules.llm_utils import extract_xml, parse_xml
 from tinybird.tb.modules.local_common import get_local_tokens, get_test_workspace_name
 from tinybird.tb.modules.project import Project
-from tinybird.tb.modules.secret import load_secrets
+from tinybird.tb.modules.secret_common import load_secrets
 
 yaml.SafeDumper.org_represent_str = yaml.SafeDumper.represent_str  # type: ignore[attr-defined]
 

{tinybird-0.0.1.dev241.dist-info → tinybird-0.0.1.dev244.dist-info}/METADATA

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

{tinybird-0.0.1.dev241.dist-info → tinybird-0.0.1.dev244.dist-info}/RECORD

@@ -17,14 +17,15 @@ 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=QWlsNI8ondzuZlYrE_y6bh1Cfz6_5AfBIndh4z9wk60,247
+tinybird/tb/__cli__.py,sha256=Ah7n0_bJGUxS5K4MX9UxDaK7Bk7ObK2JPR1kL-xkos0,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
 tinybird/tb/config.py,sha256=mhMTGnMB5KcxGoh3dewIr2Jjsa6pHE183gCPAQWyp6o,3973
-tinybird/tb/modules/build.py,sha256=MxGQY-KII4lYut-yOWGBIHFpg-m_cfHgpE7lwOzZCeQ,19871
+tinybird/tb/modules/build.py,sha256=2ijDOQDpb64v7zLRfl38UdJATkehoeunFfi2sozSjog,7609
+tinybird/tb/modules/build_common.py,sha256=3umtAVUir9Xdcu7auY5kpcA_RdQ6UoijwWXHoZEmpeg,12867
 tinybird/tb/modules/cicd.py,sha256=0KLKccha9IP749QvlXBmzdWv1On3mFwMY4DUcJlBxiE,7326
-tinybird/tb/modules/cli.py,sha256=_-kj4PSN_OPYk__WYgFfzLsDJfy4cQfveZg6tTOc220,16389
+tinybird/tb/modules/cli.py,sha256=c1r7Z7NZUIUt9O2c5IRbt4uBKiV8LGOXM4c8kOx_vyY,16344
 tinybird/tb/modules/common.py,sha256=jTTaDDHrZREt--032XhP6GkbfFwC79YJ5aH1Sl7bmbo,81925
 tinybird/tb/modules/config.py,sha256=gK7rgaWTDd4ZKCrNEg_Uemr26EQjqWt6TjyQKujxOws,11462
 tinybird/tb/modules/connection.py,sha256=-MY56NUAai6EMC4-wpi7bT0_nz_SA8QzTmHkV7HB1IQ,17810
@@ -52,27 +53,29 @@ tinybird/tb/modules/open.py,sha256=LYiuO8Z1I9O_v6pv58qpUCWFD6BT00BdeO21fRa4I4Y,1
 tinybird/tb/modules/pipe.py,sha256=xPKtezhnWZ6k_g82r4XpgKslofhuIxb_PvynH4gdUzI,2393
 tinybird/tb/modules/project.py,sha256=pOcvtgsR0ibPi0sNu-6GuAI4WS2DORRALezisjN3xY8,5662
 tinybird/tb/modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
-tinybird/tb/modules/secret.py,sha256=fSOwsG-swZC65IesYKCEuCmtRCqIvXGT3DscyYTxMms,5149
+tinybird/tb/modules/secret.py,sha256=9BIdh2PZDAbY2wRbf4ZDvkEltygztz1RMxgDmY1D0LI,3521
+tinybird/tb/modules/secret_common.py,sha256=HyCLAI9WniDLwfK6SAb7ZUWorWjtf8j_GghlaTaos_I,1829
 tinybird/tb/modules/shell.py,sha256=_9PaKkkh6ZjkixVtKNAtoCPqXMXMn1aQJM_Xzirn7ZM,13621
 tinybird/tb/modules/sink.py,sha256=dK2s__my0ePIUYrqBzhPSgdWN9rbpvP1G4dT7DJzz80,3865
 tinybird/tb/modules/table.py,sha256=4XrtjM-N0zfNtxVkbvLDQQazno1EPXnxTyo7llivfXk,11035
 tinybird/tb/modules/telemetry.py,sha256=T9gtsQffWqG_4hRBaUJPzOfMkPwz7mH-R6Bn1XRYViA,11482
-tinybird/tb/modules/test.py,sha256=DH3ZnKEJfa_7OTO803ejDNAOV98bJV1-K6-mWKfJ-sA,13138
+tinybird/tb/modules/test.py,sha256=ma6wLSf4PPr1vywSBafU80oR3nJLo3ZSK-_judTiiiE,13145
 tinybird/tb/modules/token.py,sha256=DkXW9FNCLGBisXewfk195jTJ6B1Iz7zq3cEEac48aAs,12731
 tinybird/tb/modules/watch.py,sha256=No0bK1M1_3CYuMaIgylxf7vYFJ72lTJe3brz6xQ-mJo,8819
 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=nUvSfgi7OdrWXRxLjLHzepsrICiduAZcW7iFE0x5Q5A,10323
+tinybird/tb/modules/agent/agent.py,sha256=h0va31JtVzwxaQ6kD1V8cwclNhpbzJeJbtKOps5MGIg,11864
+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=mf8dRCdof6uEFZWh5xQ_D_FStk7eDds7qWRNSbDklUM,589
-tinybird/tb/modules/agent/prompts.py,sha256=j7nkx1E2BnPP6Ra718FvSDGRWucyFgSX07V9QAiaiGg,4864
-tinybird/tb/modules/agent/utils.py,sha256=BWbS8m8HYrbbm1OnH7kLYLxL37-q1yVKa1ObWw7klfc,13041
+tinybird/tb/modules/agent/prompts.py,sha256=rAbcqkrw7BKVuV2sNRGpbLXxfo95suKWRLPwgNx7fdM,5784
+tinybird/tb/modules/agent/utils.py,sha256=SSfAmPTO-pv4UtRrsAbERHjjeqZ3Mkx6Y_eRD5QXr9M,13154
 tinybird/tb/modules/agent/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tinybird/tb/modules/agent/tools/create_datafile.py,sha256=06HK33h86T1d54zvnJXJz_1su3zIS7UJDPyBqhgo81c,1771
+tinybird/tb/modules/agent/tools/create_datafile.py,sha256=e7xMaPziVw5JPaQ8rmeEnNz6SCkR3MP1ZBvvXb6QOwE,2314
 tinybird/tb/modules/agent/tools/explore.py,sha256=ihALc_kBcsjrKT3hZyicqyIowB0g_K3AtNNi-5uz9-8,412
-tinybird/tb/modules/agent/tools/plan.py,sha256=acMbyjqI6cN7IwmMAC8-MPNSg0PEZVKGEk2NOY4gMi8,1162
+tinybird/tb/modules/agent/tools/plan.py,sha256=CMSGrqjdVyhsJ0U1M5B2eRFLZXE7HqJ4K8tl1Ile0f0,1324
 tinybird/tb/modules/agent/tools/preview_datafile.py,sha256=e9q5fR0afApcrntzFrnuHmd10ex7MG_GM6T0Pwc9bRI,850
 tinybird/tb/modules/datafile/build.py,sha256=NFKBrusFLU0WJNCXePAFWiEDuTaXpwc0lHlOQWEJ43s,51117
 tinybird/tb/modules/datafile/build_common.py,sha256=2yNdxe49IMA9wNvl25NemY2Iaz8L66snjOdT64dm1is,4511
@@ -94,8 +97,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.dev241.dist-info/METADATA,sha256=urGByLwa4s7Sstte1j0r7UZ6iYFUmnpX4ujKWztx9II,1733
-tinybird-0.0.1.dev241.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev241.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev241.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev241.dist-info/RECORD,,
+tinybird-0.0.1.dev244.dist-info/METADATA,sha256=6it3lKYPvmXa3ivNyaVevIU9DDt3vmAaDcHHwHNv6FI,1733
+tinybird-0.0.1.dev244.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev244.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev244.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev244.dist-info/RECORD,,