tinybird 0.0.1.dev29__py3-none-any.whl → 0.0.1.dev31__py3-none-any.whl

tinybird/tb/modules/test.py

@@ -6,11 +6,13 @@
 import difflib
 import glob
 import os
+import urllib.parse
 from pathlib import Path
-from typing import Any, Dict, Iterable, List, Optional, Tuple
+from typing import Any, Dict, List, Optional, Tuple
 
 import click
 import yaml
+from requests import Response
 
 from tinybird.prompts import test_create_prompt
 from tinybird.tb.modules.cli import cli
@@ -66,9 +68,9 @@ def test(ctx: click.Context) -> None:
     type=click.Path(exists=True, file_okay=False),
     help="Folder where datafiles will be placed",
 )
-@click.option("--prompt", type=str, default=None, help="Prompt to be used to create the test")
+@click.option("--prompt", type=str, default="", help="Prompt to be used to create the test")
 @coro
-async def test_create(name_or_filename: str, prompt: Optional[str], folder: str) -> None:
+async def test_create(name_or_filename: str, prompt: str, folder: str) -> None:
     """
     Create a test for an existing pipe
     """
@@ -80,15 +82,13 @@ async def test_create(name_or_filename: str, prompt: Optional[str], folder: str)
                 raise CLIException(FeedbackManager.error(message=f"Pipe {name_or_filename} not found"))
         else:
             pipe_folders = ("endpoints", "copies", "materializations", "sinks", "pipes")
-            pipe_path = next(
-                (
+            try:
+                pipe_path = next(
                     root_path / folder / f"{name_or_filename}.pipe"
                     for folder in pipe_folders
                     if (root_path / folder / f"{name_or_filename}.pipe").exists()
-                ),
-                None,
-            )
-            if not pipe_path:
+                )
+            except Exception:
                 raise CLIException(FeedbackManager.error(message=f"Pipe {name_or_filename} not found"))
 
         pipe_name = pipe_path.stem
@@ -97,8 +97,8 @@ async def test_create(name_or_filename: str, prompt: Optional[str], folder: str)
         pipe_content = pipe_path.read_text()
 
         client = await get_tinybird_local_client(os.path.abspath(folder))
-        pipe_nodes = await client._req(f"/v0/pipes/{pipe_name}")
-        parameters = set([param["name"] for node in pipe_nodes["nodes"] for param in node["params"]])
+        pipe = await client._req(f"/v0/pipes/{pipe_name}")
+        parameters = set([param["name"] for node in pipe["nodes"] for param in node["params"]])
 
         system_prompt = test_create_prompt.format(
             name=pipe_name,
@@ -107,24 +107,26 @@ async def test_create(name_or_filename: str, prompt: Optional[str], folder: str)
         )
         config = CLIConfig.get_project_config(folder)
         user_token = config.get_user_token()
+        if not user_token:
+            raise CLIException(FeedbackManager.error(message="No user token found"))
         llm = LLM(user_token=user_token, client=config.get_client())
 
-        response = await llm.ask(prompt, system_prompt=system_prompt)
-        response = extract_xml(response, "response")
-        tests_content = parse_xml(response, "test")
+        response_llm = await llm.ask(system_prompt=system_prompt, prompt=prompt)
+        response_xml = extract_xml(response_llm, "response")
+        tests_content = parse_xml(response_xml, "test")
 
         tests: List[Dict[str, Any]] = []
         for test_content in tests_content:
-            test = {}
+            test: Dict[str, Any] = {}
             test["name"] = extract_xml(test_content, "name")
             test["description"] = extract_xml(test_content, "description")
-            parameters = extract_xml(test_content, "parameters")
-            test["parameters"] = parameters.split("?")[1] if "?" in parameters else parameters
+            parameters_api = extract_xml(test_content, "parameters")
+            test["parameters"] = parameters_api.split("?")[1] if "?" in parameters_api else parameters_api
             test["expected_result"] = ""
 
             response = None
             try:
-                response = await client._req_raw(f"/v0/pipes/{pipe_name}.ndjson?{parameters}")
+                response = await get_pipe_data(client, pipe_name=pipe_name, test_params=test["parameters"])
             except Exception:
                 pass
 
@@ -181,7 +183,7 @@ async def test_update(pipe: str, folder: str) -> None:
             test_params = test["parameters"].split("?")[1] if "?" in test["parameters"] else test["parameters"]
             response = None
             try:
-                response = await client._req_raw(f"/v0/pipes/{pipe_name}.ndjson?{test_params}")
+                response = await get_pipe_data(client, pipe_name=pipe_name, test_params=test_params)
             except Exception:
                 continue
 
@@ -221,20 +223,21 @@ async def run_tests(name: Tuple[str, ...], folder: str) -> None:
     client = await get_tinybird_local_client(os.path.abspath(folder))
     paths = [Path(n) for n in name]
     endpoints = [f"./tests/{p.stem}.yaml" for p in paths]
-    test_files: Iterable[str] = endpoints if len(endpoints) > 0 else glob.glob("./tests/**/*.y*ml", recursive=True)
+    test_files: List[str] = endpoints if len(endpoints) > 0 else glob.glob("./tests/**/*.y*ml", recursive=True)
 
     async def run_test(test_file):
         test_file_path = Path(test_file)
         click.echo(FeedbackManager.info(message=f"\n* {test_file_path.stem}{test_file_path.suffix}"))
         test_file_content = yaml.safe_load(test_file_path.read_text())
+
         for test in test_file_content:
             try:
                 test_params = test["parameters"].split("?")[1] if "?" in test["parameters"] else test["parameters"]
                 response = None
                 try:
-                    response = await client._req_raw(f"/v0/pipes/{test_file_path.stem}.ndjson?{test_params}")
+                    response = await get_pipe_data(client, pipe_name=test_file_path.stem, test_params=test_params)
                 except Exception:
-                    raise Exception("Expected to not fail but got an error")
+                    continue
 
                 expected_result = response.text
                 if response.status_code >= 400:
@@ -271,3 +274,19 @@ async def run_tests(name: Tuple[str, ...], folder: str) -> None:
         exit(1)
     else:
         click.echo(FeedbackManager.success(message=f"\n✓ {test_count}/{test_count} passed"))
+
+
+async def get_pipe_data(client, pipe_name: str, test_params: str) -> Response:
+    pipe = await client._req(f"/v0/pipes/{pipe_name}")
+    output_node = next(
+        (node for node in pipe["nodes"] if node["node_type"] != "default" and node["node_type"] != "standard"),
+        {"name": "not_found"},
+    )
+    if output_node["node_type"] == "endpoint":
+        return await client._req_raw(f"/v0/pipes/{pipe_name}.ndjson?{test_params}")
+
+    params = {
+        "q": output_node["sql"],
+        "pipeline": pipe_name,
+    }
+    return await client._req_raw(f"""/v0/sql?{urllib.parse.urlencode(params)}&{test_params}""")

tinybird/tb/modules/update.py

@@ -0,0 +1,182 @@
+import os
+import re
+from os import getcwd
+from pathlib import Path
+from typing import Optional
+
+import click
+
+from tinybird.client import TinyB
+from tinybird.prompts import mock_prompt, update_prompt
+from tinybird.tb.modules.cli import cli
+from tinybird.tb.modules.common import check_user_token_with_client, coro, generate_datafile
+from tinybird.tb.modules.config import CLIConfig
+from tinybird.tb.modules.datafile.fixture import build_fixture_name, persist_fixture
+from tinybird.tb.modules.exceptions import CLIException
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+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_tinybird_local_client
+
+
+@cli.command()
+@click.argument("prompt")
+@click.option(
+    "--folder",
+    default=".",
+    type=click.Path(exists=False, file_okay=False),
+    help="Folder where project files will be placed",
+)
+@coro
+async def update(
+    prompt: str,
+    folder: str,
+) -> None:
+    """Update resources in the project."""
+    folder = folder or getcwd()
+    folder_path = Path(folder)
+    if not folder_path.exists():
+        folder_path.mkdir()
+
+    try:
+        config = CLIConfig.get_project_config(folder)
+        tb_client = config.get_client()
+        user_token: Optional[str] = None
+        try:
+            user_token = config.get_user_token()
+            if not user_token:
+                raise CLIException("No user token found")
+            await check_user_token_with_client(tb_client, token=user_token)
+        except Exception as e:
+            click.echo(
+                FeedbackManager.error(message=f"This action requires authentication. Run 'tb login' first. Error: {e}")
+            )
+            return
+
+        local_client = await get_tinybird_local_client(folder)
+
+        click.echo(FeedbackManager.highlight(message="\n» Updating resources..."))
+        datasources_updated = await update_resources(tb_client, user_token, prompt, folder)
+        click.echo(FeedbackManager.success(message="✓ Done!\n"))
+
+        if datasources_updated and user_token:
+            click.echo(FeedbackManager.highlight(message="\n» Generating fixtures..."))
+
+            datasource_files = [f for f in os.listdir(Path(folder) / "datasources") if f.endswith(".datasource")]
+            for datasource_file in datasource_files:
+                datasource_path = Path(folder) / "datasources" / datasource_file
+                llm = LLM(user_token=user_token, client=tb_client)
+                datasource_name = datasource_path.stem
+                datasource_content = datasource_path.read_text()
+                has_json_path = "`json:" in datasource_content
+                if has_json_path:
+                    prompt = f"<datasource_schema>{datasource_content}</datasource_schema>\n<user_input>{prompt}</user_input>"
+                    response = await llm.ask(system_prompt=mock_prompt(rows=20), prompt=prompt)
+                    sql = extract_xml(response, "sql")
+                    sql = sql.split("FORMAT")[0]
+                    result = await local_client.query(f"{sql} FORMAT JSON")
+                    data = result.get("data", [])
+                    fixture_name = build_fixture_name(
+                        datasource_path.absolute().as_posix(), datasource_name, datasource_content
+                    )
+                    if data:
+                        persist_fixture(fixture_name, data, folder)
+                        click.echo(FeedbackManager.info(message=f"✓ /fixtures/{datasource_name}"))
+
+            click.echo(FeedbackManager.success(message="✓ Done!\n"))
+    except Exception as e:
+        click.echo(FeedbackManager.error(message=f"Error: {str(e)}"))
+
+
+async def update_resources(
+    tb_client: TinyB,
+    user_token: str,
+    prompt: str,
+    folder: str,
+):
+    datasource_paths = [
+        Path(folder) / "datasources" / f for f in os.listdir(Path(folder) / "datasources") if f.endswith(".datasource")
+    ]
+    pipes_paths = [
+        Path(folder) / "endpoints" / f for f in os.listdir(Path(folder) / "endpoints") if f.endswith(".pipe")
+    ]
+    resources_xml = "\n".join(
+        [
+            f"<resource><type>{resource_type}</type><name>{resource_name}</name><content>{resource_content}</content></resource>"
+            for resource_type, resource_name, resource_content in [
+                ("datasource", ds.stem, ds.read_text()) for ds in datasource_paths
+            ]
+            + [
+                (
+                    "pipe",
+                    pipe.stem,
+                    pipe.read_text(),
+                )
+                for pipe in pipes_paths
+            ]
+        ]
+    )
+    llm = LLM(user_token=user_token, client=tb_client)
+    result = await llm.ask(system_prompt=update_prompt(resources_xml), prompt=prompt)
+    result = extract_xml(result, "response")
+    resources = parse_xml(result, "resource")
+    datasources = []
+    pipes = []
+    for resource_xml in resources:
+        resource_type = extract_xml(resource_xml, "type")
+        name = extract_xml(resource_xml, "name")
+        content = extract_xml(resource_xml, "content")
+        resource = {
+            "name": name,
+            "content": content,
+        }
+        if resource_type.lower() == "datasource":
+            datasources.append(resource)
+        elif resource_type.lower() == "pipe":
+            pipes.append(resource)
+
+    for ds in datasources:
+        content = ds["content"].replace("```", "")
+        filename = f"{ds['name']}.datasource"
+        generate_datafile(
+            content,
+            filename=filename,
+            data=None,
+            _format="ndjson",
+            force=True,
+            folder=folder,
+        )
+
+    for pipe in pipes:
+        content = pipe["content"].replace("```", "")
+        generate_pipe_file(pipe["name"], content, folder)
+
+    return len(datasources) > 0
+
+
+def generate_pipe_file(name: str, content: str, folder: str):
+    def is_copy(content: str) -> bool:
+        return re.search(r"TYPE copy", content, re.IGNORECASE) is not None
+
+    def is_materialization(content: str) -> bool:
+        return re.search(r"TYPE materialized", content, re.IGNORECASE) is not None
+
+    def is_sink(content: str) -> bool:
+        return re.search(r"TYPE sink", content, re.IGNORECASE) is not None
+
+    if is_copy(content):
+        pathname = "copies"
+    elif is_materialization(content):
+        pathname = "materializations"
+    elif is_sink(content):
+        pathname = "sinks"
+    else:
+        pathname = "endpoints"
+
+    base = Path(folder) / pathname
+    if not base.exists():
+        base = Path()
+    f = base / (f"{name}.pipe")
+    with open(f"{f}", "w") as file:
+        file.write(content)
+    click.echo(FeedbackManager.info_file_created(file=f.relative_to(folder)))

tinybird/tb/modules/watch.py

@@ -1,10 +1,18 @@
 import asyncio
 import os
 import time
+from pathlib import Path
 from typing import Any, Callable, List, Optional, Union
 
 import click
-from watchdog.events import DirMovedEvent, FileMovedEvent, FileSystemEventHandler
+from watchdog.events import (
+    DirDeletedEvent,
+    DirMovedEvent,
+    FileDeletedEvent,
+    FileMovedEvent,
+    FileSystemEventHandler,
+    PatternMatchingEventHandler,
+)
 from watchdog.observers import Observer
 
 from tinybird.tb.modules.feedback_manager import FeedbackManager
@@ -110,3 +118,70 @@ def watch_files(
         observer.stop()
 
     observer.join()
+
+
+class WatchProjectHandler(PatternMatchingEventHandler):
+    def __init__(self, shell: Shell, folder: str, process: Callable[[List[str]], None]):
+        self.shell = shell
+        self.process = process
+        super().__init__(
+            patterns=[f"{folder}/**/*.datasource", f"{folder}/**/*.pipe"],
+            ignore_patterns=[f"{folder}/vendor/"],
+        )
+
+    def should_process(self, event: Any) -> Optional[str]:
+        if event.is_directory:
+            return None
+
+        if os.path.exists(event.src_path):
+            return event.src_path
+
+        if os.path.exists(event.dest_path):
+            return event.dest_path
+
+        return event.src_path
+
+    def _process(self) -> None:
+        click.echo(FeedbackManager.highlight(message="» Rebuilding..."))
+        time_start = time.time()
+        self.process()
+        time_end = time.time()
+        elapsed_time = time_end - time_start
+        click.echo(
+            FeedbackManager.success(message="\n✓ ")
+            + FeedbackManager.gray(message=f"Rebuild completed in {elapsed_time:.1f}s")
+        )
+        self.shell.reprint_prompt()
+
+    def on_modified(self, event: Any) -> None:
+        if path := self.should_process(event):
+            filename = Path(path).name
+            click.echo(FeedbackManager.highlight(message=f"\n\n⟲ Changes detected in {filename}\n"))
+            self._process()
+
+    def on_deleted(self, event: Union[DirDeletedEvent, FileDeletedEvent]) -> None:
+        filename = Path(event.src_path).name
+        if event.is_directory:
+            click.echo(FeedbackManager.highlight(message=f"\n\n⟲ Deleted directory: {filename}\n"))
+        else:
+            click.echo(FeedbackManager.highlight(message=f"\n\n⟲ Deleted file: {filename}\n"))
+        self._process()
+
+
+def watch_project(
+    shell: Shell,
+    process: Callable,
+    folder: str,
+) -> None:
+    event_handler = WatchProjectHandler(shell=shell, folder=folder, process=process)
+    observer = Observer()
+    observer.schedule(event_handler, path=folder, recursive=True)
+    observer.start()
+
+    try:
+        while True:
+            time.sleep(1)
+    except KeyboardInterrupt:
+        observer.stop()
+
+    observer.join()

{tinybird-0.0.1.dev29.dist-info → tinybird-0.0.1.dev31.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tinybird
-Version: 0.0.1.dev29
+Version: 0.0.1.dev31
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/cli/introduction.html
 Author: Tinybird
@@ -8,6 +8,7 @@ Author-email: support@tinybird.co
 Requires-Python: >=3.9, <3.13
 Description-Content-Type: text/x-rst
 Requires-Dist: aiofiles (==24.1.0)
+Requires-Dist: anthropic (==0.42.0)
 Requires-Dist: click (<8.2,>=8.1.6)
 Requires-Dist: clickhouse-toolset (==0.33.dev0)
 Requires-Dist: colorama (==0.4.6)

{tinybird-0.0.1.dev29.dist-info → tinybird-0.0.1.dev31.dist-info}/RECORD

@@ -1,58 +1,59 @@
-tinybird/__cli__.py,sha256=pgYsVLcqL16wtSn6KtKweNZYoYJdEksTgSvQAW7hH64,250
-tinybird/client.py,sha256=xM9Czi9JRjt3VpTAoe2di4m36VYIT3ngpn-DgkWFv-Y,51505
-tinybird/config.py,sha256=RRNb3Z1zkykTpusBTSyszuIySHGMsfL5aVxxt1kdr6w,6147
+tinybird/__cli__.py,sha256=esPl5QDTzuQgHe5FuxWLm-fURFigGGwjnYLh9GuWUw4,232
+tinybird/client.py,sha256=P-bc7s7SbrnzkZkSsb_TO58q4Jxy-bBD5S5qpYiXiZg,51517
+tinybird/config.py,sha256=ENRNyEMXHj_P882o31iFz0hTveziLabVRrxiWE5RRBE,6233
 tinybird/connectors.py,sha256=lkpVSUmSuViEZBa4QjTK7YmPHUop0a5UFoTrSmlVq6k,15244
-tinybird/context.py,sha256=kutUQ0kCwparowI74_YLXx6wtTzGLRouJ6oGHVBPzBo,1291
+tinybird/context.py,sha256=eeFneyn_oTUoTIfst59CnSkskchvnWuRvyI7UKRFq14,1407
 tinybird/datatypes.py,sha256=XNypumfqNjsvLJ5iNXnbVHRvAJe0aQwI3lS6Cxox-e0,10979
 tinybird/feedback_manager.py,sha256=cNUbt0Jxim02UiIdlyP12DJfXfFFzxDCfJK9XRWZ9A0,67488
 tinybird/git_settings.py,sha256=Sw_8rGmribEFJ4Z_6idrVytxpFYk7ez8ei0qHULzs3E,3934
-tinybird/prompts.py,sha256=ZZ13gs6FzRr-oQ3W0kkZG5PAYd3HI7iJQ3WMqxIaMv8,28580
+tinybird/prompts.py,sha256=xrZMZykphO9DRfW5yvCJNZMhPq0v1zab3d9jKy9plhU,24064
 tinybird/sql.py,sha256=eulpRe05ZFrKFrxYawgxDxxrktFE8uL6hSL1gHIWKyg,46166
-tinybird/sql_template.py,sha256=IqYRfUxDYBCoOYjqqvn--_8QXLv9FSRnJ0bInx7q1Xs,93051
+tinybird/sql_template.py,sha256=GmMLAI10MTqjQo9qztuQHLRWs67teozsWDxUBdvkAn4,93668
 tinybird/sql_template_fmt.py,sha256=1z-PuqSZXtzso8Z_mPqUc-NxIxUrNUcVIPezNieZk-M,10196
 tinybird/sql_toolset.py,sha256=HAo6xtCjlbhikdpK6xfMC_RvEG6uzRropwhKSHEkIIc,14343
 tinybird/syncasync.py,sha256=IPnOx6lMbf9SNddN1eBtssg8vCLHMt76SuZ6YNYm-Yk,27761
 tinybird/tornado_template.py,sha256=oflXyoL2LSCegvl6bAzqw2JIqRaN5WPjhYYDtQcfuOE,41869
 tinybird/ch_utils/constants.py,sha256=aYvg2C_WxYWsnqPdZB1ZFoIr8ZY-XjUXYyHKE9Ansj0,3890
 tinybird/ch_utils/engine.py,sha256=OXkBhlzGjZotjD0vaT-rFIbSGV4tpiHxE8qO_ip0SyQ,40454
-tinybird/tb/__cli__.py,sha256=7oc08x3qXwnjl6DKj-E5hjnKQyZZeUOMUecnt_qghXc,251
-tinybird/tb/cli.py,sha256=D5Xs9RPbBY6U6lXTtzRSWNs2pG6uSzax-cfoKMpO7Jg,889
+tinybird/tb/__cli__.py,sha256=SgyAWbfPDLAaYyoq9rFOPpcXyLPQqCPh2m8Bt_6lxaI,251
+tinybird/tb/cli.py,sha256=dmrf8O9cqBuOYUmVBbNDSNQpHUhBg6IkBPaISAj4kHo,923
 tinybird/tb/modules/auth.py,sha256=EzRWFmwRkXNhUmRaruEVFLdkbUg8xMSix0cAWl5D4Jg,9029
-tinybird/tb/modules/build.py,sha256=8wzHG_8xrA5D3rjnACTuSeJS4AqFL5rJFzpJnYWCBnQ,8150
-tinybird/tb/modules/build_server.py,sha256=6LbJYCSIY6TEEH9DDjZ19pzaAH08roYcWpiP0QNuO6c,2674
+tinybird/tb/modules/build.py,sha256=J_YCaJIkvoM1CMpsYQ4gM1LinaqJmJmvfcmhxEmWq_g,3725
+tinybird/tb/modules/build_client.py,sha256=SoGaGPBYT_0BoRp2lxNyBBvQQ1GFAMpx_o62jb91DUI,8046
 tinybird/tb/modules/cicd.py,sha256=SjCyvvy0WUnsjFs2biwwXvcf0Ddpmghhd8-SnMyfsRM,5355
-tinybird/tb/modules/cli.py,sha256=J3s9RHw1Xwx0q4qUYw0QQGAtjpxhbmKNEFpRiPf4NIU,20844
-tinybird/tb/modules/common.py,sha256=W4MyzUAMYl1xRF9bG374NzVjZrz9JCo90MGP5HDBf8o,71529
-tinybird/tb/modules/config.py,sha256=fUO-rVPImPnk8DNt0_AUYDlG77eC9Cs2jiVlB9LXCmg,11255
+tinybird/tb/modules/cli.py,sha256=i5KQF1kJqd3lflYl0e3tnKrMZBIAOlMF1T5pxa6e7tM,22124
+tinybird/tb/modules/common.py,sha256=e4U7AT0dUBG6O-7Iq2CVN1UHPd6-ZCFucyW0L5gBi4g,70592
+tinybird/tb/modules/config.py,sha256=mie3oMVTf5YOUFEiLs88P16U4LkJafJjSpjwyAkFHog,10979
 tinybird/tb/modules/connection.py,sha256=FhDM-OAnLN2epbO2YonpjJQhHqBjyuanBsZmKlDXrqg,28679
-tinybird/tb/modules/create.py,sha256=kVLEwTfT-SxcF_IWaSh3vVQ5qVPbKQhwVHNO1kzoNyw,10250
-tinybird/tb/modules/datasource.py,sha256=Xz2syobLp7SSDFFjCNT90uI8iH-MZ4KC8p7g1SY499Y,32897
+tinybird/tb/modules/create.py,sha256=HclKekBKTaobVQeYq9ooSmuBwlrebi-YsFOFJJMGTZs,10860
+tinybird/tb/modules/datasource.py,sha256=PmIoDSLqBNHQvguX9t1T3z9lsG3HB67cmwhQO5_8Lc4,32918
 tinybird/tb/modules/deploy.py,sha256=j1MuiGsWplG1BQs8eG6-STcYyzvHqVESPCOo6P-dcF8,9242
 tinybird/tb/modules/exceptions.py,sha256=4A2sSjCEqKUMqpP3WI00zouCWW4uLaghXXLZBSw04mY,3363
 tinybird/tb/modules/feedback_manager.py,sha256=e8tqehRR0Buhs8O0n8N2Sg2vnnBVb1NLtnZqkPrYD_A,68379
 tinybird/tb/modules/fmt.py,sha256=poh6_cwVGSf-sBu6LKWuO2TANL_J8Sgm25sPpwxa3Aw,3558
 tinybird/tb/modules/job.py,sha256=956Pj8BEEsiD2GZsV9RKKVM3I_CveOLgS82lykO5ukk,2963
-tinybird/tb/modules/llm.py,sha256=yBa_XwWd-iuMMYNg7BZWXxZcwDyW9sfLrcdK81hX6Dk,3215
+tinybird/tb/modules/llm.py,sha256=mguBJaTJ74a5mQXcBQwbCjKugfvcFm_Wzg2e_hSkooQ,953
 tinybird/tb/modules/llm_utils.py,sha256=hIfBU7vMUHUt25pljim3WdZdJTNr9hDo3mHezqI5B7w,766
-tinybird/tb/modules/local.py,sha256=lHUji6FMhRv4J9sLrHDqyMT-8OwzFSVzDfapnUKS-ZA,5242
-tinybird/tb/modules/local_common.py,sha256=fRHJrEix19lAOE4EpzRhWBJH8FUk7hCCvynP0Icf7Ww,2298
-tinybird/tb/modules/login.py,sha256=NwpwPQYMVb0mW5-uPR5zb5hl06XqH7xAJfA1GntZzPk,6315
-tinybird/tb/modules/mock.py,sha256=b0MqSVvJsaXdUKvagXfPGrh-XB8cYmuyXJ2D4P7uyB0,3658
+tinybird/tb/modules/local.py,sha256=bSBBk2A0yj4fQHxsdzbZAH1Q_xrqT7vFOIPoz7TNabo,5260
+tinybird/tb/modules/local_common.py,sha256=Na4zHQP2H9oE840KYUn69_f0PcqExQtaKFikaiNf43s,2296
+tinybird/tb/modules/login.py,sha256=0cS-f3MsQFHc6xjw8FRWJm4EJBH9C7Ri68EcO_tiwes,6508
+tinybird/tb/modules/mock.py,sha256=eSfD-m4w5NLGxjfGRlYplQhC0FKpanysIweG_8GsbNE,3665
 tinybird/tb/modules/pipe.py,sha256=eYmMBiSj1Ur_hXUs74YZ9mCSAyiICDmkuKuTemlxPUY,27018
 tinybird/tb/modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
-tinybird/tb/modules/shell.py,sha256=OfFQ4lx3v_XSTr5cm_mTDG6CLc3zFdikM2pOamifG84,13469
+tinybird/tb/modules/shell.py,sha256=84WiRppf4bLZSSEscSqFov-7RpDEQ8u0DFovqlnXauk,13093
 tinybird/tb/modules/table.py,sha256=4XrtjM-N0zfNtxVkbvLDQQazno1EPXnxTyo7llivfXk,11035
 tinybird/tb/modules/tag.py,sha256=anPmMUBc-TbFovlpFi8GPkKA18y7Y0GczMsMms5TZsU,3502
 tinybird/tb/modules/telemetry.py,sha256=iEGnMuCuNhvF6ln__j6X9MSTwL_0Hm-GgFHHHvhfknk,10466
-tinybird/tb/modules/test.py,sha256=8rsj8HC9mrldMRx0ViF0XtkF8NHWOsFvQtMTQnJqhpM,10834
+tinybird/tb/modules/test.py,sha256=2UhRiMRPP2vEsa2ayM_ujLc1cB8gHkKHCLJ50GbkQxc,11577
 tinybird/tb/modules/token.py,sha256=AePr-QMv_vtWwZDWQ92Zp0kPrCjze61i4npiPhoLMZg,12717
-tinybird/tb/modules/watch.py,sha256=9D6NTo9cllyJfvObtOzZEbv3SJN42EgfVdYQPN1liZU,3964
+tinybird/tb/modules/update.py,sha256=wXzLhBzKD1kWEN1HejOVkKpFQqomZGfP1AiCvE-K04A,6827
+tinybird/tb/modules/watch.py,sha256=na1H0lVCyMTCgzm2h8FHyKWhWN5poAehWgJeXT2rLXg,6281
 tinybird/tb/modules/workspace.py,sha256=6icAgnTvfL3d1kx4L1Z1cGXCD_2Yx0fNRjbZHNxRbYc,10927
 tinybird/tb/modules/workspace_members.py,sha256=Ai6iCOzXX1zQ8q9iXIFSFHsBJlT-8Q28DaG5Ie-UweY,8726
-tinybird/tb/modules/datafile/build.py,sha256=AAyMOkrliBKF0ZPVQBuSx9Qc1D-Y-08sXw90YAg_LiM,57559
+tinybird/tb/modules/datafile/build.py,sha256=pwgsIuvHwb2cdsl3IWOAPyj6S9vB3jn_BXGRcKT7I2Y,57577
 tinybird/tb/modules/datafile/build_common.py,sha256=IXl-Z51zUi1dypV7meNenX0iu2UmowNeqgG6WHyMHlk,4562
 tinybird/tb/modules/datafile/build_datasource.py,sha256=4aP8_DYCRGghXntZSeWDNJxjps1QRVa7WHoYCzQwQts,17355
-tinybird/tb/modules/datafile/build_pipe.py,sha256=n6jjfBaDl014-jxM5gI9XjUb0XCmgEVoAoAuy_Br8nw,27842
+tinybird/tb/modules/datafile/build_pipe.py,sha256=Jgv3YKIvMfjPiSIdw1k2mpaoDdAWMiMRaSHwRgyI97E,28258
 tinybird/tb/modules/datafile/common.py,sha256=TSJozXsXzxvkAC_N7dV-Iv_cy2AIGHUNtCYgwlQAbtU,75762
 tinybird/tb/modules/datafile/diff.py,sha256=-0J7PsBO64T7LOZSkZ4ZFHHCPvT7cKItnJkbz2PkndU,6754
 tinybird/tb/modules/datafile/exceptions.py,sha256=8rw2umdZjtby85QbuRKFO5ETz_eRHwUY5l7eHsy1wnI,556
@@ -72,8 +73,8 @@ tinybird/tb_cli_modules/config.py,sha256=6u6B5QCdiQLbJkCkwtnKGs9H3nP-KXXhC75mF7B
 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=iEGnMuCuNhvF6ln__j6X9MSTwL_0Hm-GgFHHHvhfknk,10466
-tinybird-0.0.1.dev29.dist-info/METADATA,sha256=Wi5oqXBScwv47ev72oU9M0Pa1m3nZPi3VaL1i5iL64c,2446
-tinybird-0.0.1.dev29.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tinybird-0.0.1.dev29.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev29.dist-info/top_level.txt,sha256=pgw6AzERHBcW3YTi2PW4arjxLkulk2msOz_SomfOEuc,45
-tinybird-0.0.1.dev29.dist-info/RECORD,,
+tinybird-0.0.1.dev31.dist-info/METADATA,sha256=nl0L_9CisXRnrHeEukw_5Z733F7vlRvHMVgHPdVrcL0,2482
+tinybird-0.0.1.dev31.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tinybird-0.0.1.dev31.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev31.dist-info/top_level.txt,sha256=pgw6AzERHBcW3YTi2PW4arjxLkulk2msOz_SomfOEuc,45
+tinybird-0.0.1.dev31.dist-info/RECORD,,

tinybird/tb/modules/build_server.py

@@ -1,75 +0,0 @@
-import asyncio
-import glob
-import json
-import logging
-from pathlib import Path
-from typing import List
-
-import click
-import requests
-
-from tinybird.tb.modules.cli import cli
-from tinybird.tb.modules.feedback_manager import FeedbackManager
-from tinybird.tb.modules.local_common import get_tinybird_local_client
-
-
-def project_files(project_path: Path) -> List[str]:
-    project_file_extensions = ("datasource", "pipe")
-    project_files = []
-    for extension in project_file_extensions:
-        for project_file in glob.glob(f"{project_path}/**/*.{extension}", recursive=True):
-            logging.debug(f"Found project file: {project_file}")
-            project_files.append(project_file)
-    return project_files
-
-
-@cli.command()
-@click.argument("project_path", type=click.Path(exists=True), default=Path.cwd())
-def build_server(project_path: Path) -> None:
-    """
-    Validate and build the project server side.
-    """
-
-    MULTIPART_BOUNDARY_DATA_PROJECT = "data_project://"
-    DATAFILE_TYPE_TO_CONTENT_TYPE = {
-        ".datasource": "text/plain",
-        ".pipe": "text/plain",
-    }
-
-    tb_client = asyncio.run(get_tinybird_local_client(str(project_path)))
-    TINYBIRD_API_URL = tb_client.host + "/v1/build"
-    TINYBIRD_API_KEY = tb_client.token
-
-    files = [
-        ("context://", ("cli-version", "1.0.0", "text/plain")),
-    ]
-    fds = []
-    for file_path in project_files(project_path):
-        relative_path = str(Path(file_path).relative_to(project_path))
-        fd = open(file_path, "rb")
-        fds.append(fd)
-        content_type = DATAFILE_TYPE_TO_CONTENT_TYPE.get(Path(file_path).suffix, "application/unknown")
-        files.append((MULTIPART_BOUNDARY_DATA_PROJECT, (relative_path, fd, content_type)))
-
-    try:
-        HEADERS = {"Authorization": f"Bearer {TINYBIRD_API_KEY}"}
-
-        r = requests.post(TINYBIRD_API_URL, files=files, headers=HEADERS)
-        result = r.json()
-        logging.debug(json.dumps(result, indent=2))
-
-        build_result = result.get("result")
-        if build_result == "success":
-            click.echo(FeedbackManager.success(message="Build completed successfully"))
-        elif build_result == "failed":
-            click.echo(FeedbackManager.error(message="Build failed"))
-            build_errors = result.get("errors")
-            for build_error in build_errors:
-                click.echo(
-                    FeedbackManager.error(message=f"{build_error.get('filename')}\n\n{build_error.get('error')}")
-                )
-        else:
-            click.echo(FeedbackManager.error(message=f"Unknown build result. Error: {result.get('error')}"))
-    finally:
-        for fd in fds:
-            fd.close()