tinybird 0.0.1.dev13__py3-none-any.whl → 0.0.1.dev15__py3-none-any.whl

tinybird/tb/modules/datafile/diff.py

@@ -121,12 +121,12 @@ async def diff_command(
     if filenames:
         if len(filenames) == 1:
             filenames = [filenames[0], *get_project_filenames(filenames[0])]
-        await folder_pull(client, target_dir, False, None, True, verbose=False)
+        await folder_pull(client, target_dir, True, verbose=False)
     else:
         filenames = get_project_filenames(".")
         if verbose:
             click.echo("Saving remote resources in .diff_tmp folder.\n")
-        await folder_pull(client, target_dir, False, None, True, verbose=verbose, progress_bar=progress_bar)
+        await folder_pull(client, target_dir, True, verbose=verbose, progress_bar=progress_bar)
 
     remote_datasources: List[Dict[str, Any]] = await client.datasources()
     remote_pipes: List[Dict[str, Any]] = await client.pipes()

tinybird/tb/modules/datafile/parse_datasource.py

@@ -36,6 +36,7 @@ def parse_datasource(
         ) from None
 
     if len(doc.nodes) > 1:
+        # TODO(eclbg): Turn this into a custom exception with a better message
         raise ValueError(f"{filename}: datasources can't have more than one node")
 
     return doc

tinybird/tb/modules/datafile/pull.py

@@ -1,7 +1,7 @@
 import re
 from asyncio import Semaphore, gather
 from pathlib import Path
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, Tuple
 
 import aiofiles
 import click
@@ -16,38 +16,39 @@ from tinybird.tb.modules.datafile.format_pipe import format_pipe
 async def folder_pull(
     client: TinyB,
     folder: str,
-    auto: bool,
-    match: Optional[str],
     force: bool,
     verbose: bool = True,
     progress_bar: bool = False,
     fmt: bool = False,
 ):
-    pattern = re.compile(match) if match else None
-
-    def _get_latest_versions(resources: List[str]):
+    def _get_latest_versions(resources: List[Tuple[str, str]]):
         versions: Dict[str, Any] = {}
 
-        for x in resources:
+        for x, resource_type in resources:
             t = get_name_version(x)
             t["original_name"] = x
             if t["version"] is None:
                 t["version"] = -1
             name = t["name"]
+            t["type"] = resource_type
 
             if name not in versions or name == x or versions[name]["version"] < t["version"]:
                 versions[name] = t
         return versions
 
-    def get_file_folder(extension: str):
-        if not auto:
-            return None
+    def get_file_folder(extension: str, resource_type: Optional[str]):
         if extension == "datasource":
             return "datasources"
+        if resource_type == "endpoint":
+            return "endpoints"
+        if resource_type == "sink":
+            return "sinks"
+        if resource_type == "copy":
+            return "copies"
+        if resource_type == "materialized":
+            return "materializations"
         if extension == "pipe":
             return "pipes"
-        if extension == "token":
-            return "tokens"
         return None
 
     async def write_files(
@@ -61,11 +62,6 @@ async def folder_pull(
         async def write_resource(k: Dict[str, Any]):
             name = f"{k['name']}.{extension}"
             try:
-                if pattern and not pattern.search(name):
-                    if verbose:
-                        click.echo(FeedbackManager.info_skipping_resource(resource=name))
-                    return
-
                 resource = await getattr(client, get_resource_function)(k["original_name"])
                 resource_to_write = resource
 
@@ -76,11 +72,11 @@ async def folder_pull(
                         resource_to_write = await format_pipe(name, content=resource)
 
                 dest_folder = folder
-                if "." in k["name"] and extension != "token":
+                if "." in k["name"]:
                     dest_folder = Path(folder) / "vendor" / k["name"].split(".", 1)[0]
                     name = f"{k['name'].split('.', 1)[1]}.{extension}"
 
-                file_folder = get_file_folder(extension)
+                file_folder = get_file_folder(extension, k["type"])
                 f = Path(dest_folder) / file_folder if file_folder is not None else Path(dest_folder)
 
                 if not f.exists():
@@ -125,11 +121,11 @@ async def folder_pull(
 
     try:
         datasources = await client.datasources()
-        remote_datasources = sorted([x["name"] for x in datasources])
+        remote_datasources = sorted([(x["name"], x.get("type", "csv")) for x in datasources], key=lambda x: x[0])
         datasources_versions = _get_latest_versions(remote_datasources)
 
         pipes = await client.pipes()
-        remote_pipes = sorted([x["name"] for x in pipes])
+        remote_pipes = sorted([(pipe["name"], pipe.get("type", "default")) for pipe in pipes], key=lambda x: x[0])
         pipes_versions = _get_latest_versions(remote_pipes)
 
         resources = list(datasources_versions.keys()) + list(pipes_versions.keys())

tinybird/tb/modules/llm.py

@@ -1,12 +1,12 @@
 import asyncio
-from datetime import datetime
+import urllib.parse
+from copy import deepcopy
 from typing import Awaitable, Callable, List
 
-from openai import OpenAI
 from pydantic import BaseModel
 
 from tinybird.client import TinyB
-from tinybird.tb.modules.prompts import create_project_prompt, sample_data_sql_prompt
+from tinybird.tb.modules.config import CLIConfig
 
 
 class DataFile(BaseModel):
@@ -20,8 +20,12 @@ class DataProject(BaseModel):
 
 
 class LLM:
-    def __init__(self, key: str):
-        self.client = OpenAI(api_key=key)
+    def __init__(self, client: TinyB):
+        self.client = client
+        user_token = CLIConfig.get_project_config().get_user_token()
+        user_client = deepcopy(client)
+        user_client.token = user_token
+        self.user_client = user_client
 
     async def _execute(self, action_fn: Callable[[], Awaitable[str]], checker_fn: Callable[[str], bool]):
         is_valid = False
@@ -38,36 +42,22 @@ class LLM:
         return result
 
     async def create_project(self, prompt: str) -> DataProject:
-        completion = self.client.beta.chat.completions.parse(
-            model="gpt-4o",
-            messages=[{"role": "system", "content": create_project_prompt}, {"role": "user", "content": prompt}],
-            response_format=DataProject,
-        )
-        return completion.choices[0].message.parsed or DataProject(datasources=[], pipes=[])
-
-    async def generate_sql_sample_data(
-        self, tb_client: TinyB, datasource_content: str, row_count: int = 20, context: str = ""
-    ) -> str:
-        async def action_fn():
-            response = self.client.chat.completions.create(
-                model="gpt-4o-mini",
-                messages=[
-                    {
-                        "role": "system",
-                        "content": sample_data_sql_prompt.format(
-                            current_datetime=datetime.now().isoformat(), row_count=row_count, context=context
-                        ),
-                    },
-                    {"role": "user", "content": datasource_content},
-                ],
+        try:
+            response = await self.user_client._req(
+                "/v0/llm/create",
+                method="POST",
+                data=f'{{"prompt": "{prompt}"}}',
+                headers={"Content-Type": "application/json"},
             )
-            return response.choices[0].message.content or ""
-
-        async def checker_fn(sql: str):
-            try:
-                result = await tb_client.query(f"DESCRIBE ({sql}) FORMAT JSON")
-                return len(result.get("data", [])) > 0
-            except Exception:
-                return False
-
-        return await self._execute(action_fn, checker_fn)
+            return DataProject.model_validate(response.get("result", {}))
+        except Exception:
+            return DataProject(datasources=[], pipes=[])
+
+    async def generate_sql_sample_data(self, schema: str, rows: int = 20, context: str = "") -> str:
+        response = await self.user_client._req(
+            "/v0/llm/mock",
+            method="POST",
+            data=f'{{"schema": "{urllib.parse.quote(schema)}", "rows": {rows}, "context": "{urllib.parse.quote(context)}"}}',
+            headers={"Content-Type": "application/json"},
+        )
+        return response.get("result", "")

tinybird/tb/modules/local.py

@@ -140,10 +140,9 @@ async def get_tinybird_local_client(path: Optional[str] = None):
 
         token = ws["token"]
 
-    config.set_token(token)
-    config.set_host(TB_LOCAL_HOST)
-    config.set_user_token(user_token)
-    config.persist_to_file()
+        config.set_token(token)
+        config.set_host(TB_LOCAL_HOST)
+        config.set_user_token(user_token)
     return config.get_client(host=TB_LOCAL_HOST, token=token)
 
 

tinybird/tb/modules/login.py

@@ -87,6 +87,9 @@ class AuthHandler(http.server.SimpleHTTPRequestHandler):
         return
 
 
+AUTH_SERVER_PORT = 49160
+
+
 class AuthServer(socketserver.TCPServer):
     allow_reuse_address = True
 
@@ -96,7 +99,7 @@ class AuthServer(socketserver.TCPServer):
 
 
 def start_server(auth_callback):
-    with AuthServer(("", 3000), AuthHandler, auth_callback) as httpd:
+    with AuthServer(("", AUTH_SERVER_PORT), AuthHandler, auth_callback) as httpd:
         httpd.timeout = 30
         start_time = time.time()
         while time.time() - start_time < 60:  # Run for a maximum of 60 seconds
@@ -132,7 +135,7 @@ def login(host: str, workspace: str):
 
     # Open the browser to the auth page
     client_id = "T6excMo8IKguvUw4vFNYfqlt9pe6msCU"
-    callback_url = "http://localhost:3000"
+    callback_url = f"http://localhost:{AUTH_SERVER_PORT}"
     params = {
         "client_id": client_id,
         "redirect_uri": callback_url,
@@ -159,7 +162,6 @@ def login(host: str, workspace: str):
         cli_config.set_token_for_host(workspace_token, host)
         cli_config.set_user_token(user_token)
         config.set_host(host)
-
         cli_config.persist_to_file()
         click.echo(FeedbackManager.success(message="✓ Authentication successful!"))
     else:

tinybird/tb/modules/mock.py

@@ -45,13 +45,14 @@ async def mock(datasource: str, rows: int, context: str, folder: str) -> None:
             click.echo(FeedbackManager.gray(message=f"Overriding context for {datasource_name}..."))
             context_path.write_text(context)
 
-
         click.echo(FeedbackManager.gray(message=f"Creating fixture for {datasource_name}..."))
         datasource_content = datasource_path.read_text()
-        llm_config = CLIConfig.get_llm_config()
-        llm = LLM(key=llm_config["api_key"])
+        config = CLIConfig.get_project_config()
+        user_client = config.get_client()
+        user_client.token = config.get_user_token()
+        llm = LLM(client=user_client)
         tb_client = await get_tinybird_local_client(os.path.abspath(folder))
-        sql = await llm.generate_sql_sample_data(tb_client, datasource_content, row_count=rows, context=context)
+        sql = await llm.generate_sql_sample_data(datasource_content, rows=rows, context=context)
         result = await tb_client.query(f"{sql} FORMAT JSON")
         data = result.get("data", [])[:rows]
         fixture_name = build_fixture_name(datasource_path.absolute(), datasource_name, datasource_content)

tinybird/tb/modules/prompts.py

@@ -76,6 +76,29 @@ SQL >
 </instructions>
 """
 
+create_test_calls_prompt = """
+You are a Tinybird expert. You will be given a pipe endpoint containing different nodes with SQL and Tinybird templating syntax. You will generate URLs to test it with different parameters combinations.
+
+<test>
+    <test_1>:
+        name: <test_name_1>
+        description: <description_1>
+        parameters: <url_encoded_parameters_1>
+    <test_2>:
+        name: <test_name_2>
+        description: <description_2>
+        parameters: <url_encoded_parameters_2>
+</test>
+<instructions>
+    - The test name must be unique.
+    - The test command must be a valid Tinybird command that can be run in the terminal.
+    - The test command can have as many parameters as are needed to test the pipe.
+    - The parameter within Tinybird templating syntax looks like this one {{String(my_param_name, default_value)}}.
+    - If there are no parameters in the , you can omit parametrs and generate a single test command.
+    - Extra context: {context}
+</instructions>
+"""
+
 sample_data_prompt = """Your role is to generate sample data for a given data source. A data source is a table in Tinybird. You will be given a schema for the data source, and you will need to generate a set of rows that fit that schema.
 
 The schema will be provided to you at the end of this prompt in the following format:
@@ -114,7 +137,6 @@ You will generate {row_count} rows of data in JSON format, newlines separated. Y
 
 For columns that represent timestamps, use values that are close to the current date and time. The current date and time is {current_datetime}"""
 
-
 sample_data_sql_prompt = """
 Given the schema for a Tinybird datasource, return a can you create a clickhouse sql query to generate some random data that matches that schema.
 

tinybird/tb/modules/test.py

@@ -4,23 +4,21 @@
 # - But please, **do not** interleave utility functions and command definitions.
 
 import glob
-from typing import Any, Dict, Iterable, List, Tuple
+import json
+import os
+from pathlib import Path
+from typing import Iterable, List, Optional, Tuple
 
 import click
+import yaml
 
-from tinybird.client import AuthNoTokenException
 from tinybird.feedback_manager import FeedbackManager
 from tinybird.tb.modules.cli import cli
-from tinybird.tb.modules.common import coro, create_tb_client, gather_with_concurrency
+from tinybird.tb.modules.common import coro
 from tinybird.tb.modules.config import CLIConfig
 from tinybird.tb.modules.exceptions import CLIException
-from tinybird.tb.modules.tinyunit.tinyunit import (
-    TestSummaryResults,
-    generate_file,
-    parse_file,
-    run_test_file,
-    test_run_summary,
-)
+from tinybird.tb.modules.llm import LLM, TestExpectation
+from tinybird.tb.modules.local import get_tinybird_local_client
 
 
 @cli.group()
@@ -30,78 +28,100 @@ def test(ctx: click.Context) -> None:
 
 
 @test.command(
-    name="run",
-    help="Run the test suite, a file, or a test. To skip test to run in branches and CI put them in a 'skip_in_branch' folder.",
+    name="create",
+    help="Create a test for an existing endpoint",
 )
-@click.argument("file", nargs=-1)
-@click.option("-v", "--verbose", is_flag=True, default=False, help="Enable verbose (show results)", type=bool)
-@click.option("--fail", "only_fail", is_flag=True, default=False, help="Showy onl failed/error tests", type=bool)
-@click.option("-c", "--concurrency", help="How many test to run concurrently", default=1, type=click.IntRange(1, 10))
+@click.argument("pipe", type=str)
+@click.option(
+    "--folder",
+    default=".",
+    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.pass_context
 @coro
-async def test_run(ctx: click.Context, file: Tuple[str, ...], verbose: bool, only_fail: bool, concurrency: int) -> None:
-    results: List[TestSummaryResults] = []
+async def test_create(ctx: click.Context, pipe: str, prompt: Optional[str], folder: Optional[str]) -> None:
+    """
+    Create a test for an existing endpoint
+    """
+
+    def generate_test_file(pipe_name: str, tests: List[TestExpectation]):
+        base = Path("tests")
+        if folder:
+            base = Path(folder) / base
+        base.mkdir(parents=True, exist_ok=True)
+
+        path = base / f"{pipe_name}.yaml"
+        with open(path, "w") as f:
+            yaml.dump(tests, f)
 
     try:
-        tb_client = create_tb_client(ctx)
-        config = CLIConfig.get_project_config()
-        if config.get("token") is None:
-            raise AuthNoTokenException
-        workspaces: List[Dict[str, Any]] = (await tb_client.user_workspaces_and_branches()).get("workspaces", [])
-        current_ws: Dict[str, Any] = next(
-            (workspace for workspace in workspaces if config and workspace.get("id", ".") == config.get("id", "..")), {}
-        )
+        pipe_path = Path(pipe)
+        pipe_name = pipe
+        if pipe_path.suffix == ".pipe":
+            pipe_name = pipe_path.stem
+        else:
+            pipe_path = Path("endpoints", f"{pipe}.pipe")
+            if not pipe_path.exists():
+                pipe_path = Path("pipes", f"{pipe}.pipe")
+
+        click.echo(FeedbackManager.gray(message=f"\nCreating tests for {pipe_name} endpoint..."))
+        pipe_path = Path(folder) / pipe_path
+        pipe_content = pipe_path.read_text()
+
+        llm_config = CLIConfig.get_llm_config()
+        llm = LLM(key=llm_config["api_key"])
+
+        client = await get_tinybird_local_client(os.path.abspath(folder))
+
+        test_expectations = await llm.create_test_commands(pipe_content=pipe_content, context=prompt)
+        valid_test_expectations = []
+        for test in test_expectations.tests:
+            try:
+                response = await client._req(f"/v0/pipes/{pipe_name}.ndjson?{test.parameters}")
+            except Exception:
+                continue
+
+            test.expected_result = json.dumps(response)
+            valid_test_expectations.append(test.model_dump())
+
+        if valid_test_expectations:
+            generate_test_file(pipe_name, valid_test_expectations)
+            click.echo(FeedbackManager.info(message=f"✓ /tests/{pipe_name}.yaml"))
+        click.echo(FeedbackManager.success(message="✓ Done!\n"))
     except Exception as e:
         raise CLIException(FeedbackManager.error_exception(error=e))
 
-    file_list: Iterable[str] = file if len(file) > 0 else glob.glob("./tests/**/*.y*ml", recursive=True)
-    click.echo(FeedbackManager.info_skipping_resource(resource="regression.yaml"))
-    file_list = [f for f in file_list if not f.endswith("regression.yaml")]
-    final_file_list = []
-    for f in file_list:
-        if "skip_in_branch" in f and current_ws and current_ws.get("is_branch"):
-            click.echo(FeedbackManager.info_skipping_resource(resource=f))
-        else:
-            final_file_list.append(f)
-    file_list = final_file_list
-
-    async def run_test(tb_client, test_file, results):
-        try:
-            test_result = await run_test_file(tb_client, test_file)
-            results.append(TestSummaryResults(filename=test_file, results=test_result, semver=tb_client.semver))
-        except Exception as e:
-            if verbose:
-                click.echo(FeedbackManager.error_exception(error=e))
-            raise CLIException(FeedbackManager.error_running_test(file=test_file))
-
-    test_tasks = [run_test(tb_client, test_file, results) for test_file in file_list]
-    await gather_with_concurrency(concurrency, *test_tasks)
-
-    if len(results) <= 0:
-        click.echo(FeedbackManager.warning_no_test_results())
-    else:
-        test_run_summary(results, only_fail=only_fail, verbose_level=int(verbose))
-
-
-@test.command(name="init", help="Initialize a file list with a simple test suite.")
-@click.argument("files", nargs=-1)
-@click.option("--force", is_flag=True, default=False, help="Override existing files")
-@click.pass_context
-@coro
-async def test_init(ctx: click.Context, files: Tuple[str, ...], force: bool) -> None:
-    if len(files) == 0:
-        files = ("tests/default.yaml",)
-
-    for file in files:
-        generate_file(file, overwrite=force)
-
 
-@test.command(name="parse", help="Read the contents of a test file list.")
-@click.argument("files", nargs=-1)
+@test.command(
+    name="run",
+    help="Run the test suite, a file, or a test. To skip test to run in branches and CI put them in a 'skip_in_branch' folder.",
+)
+@click.argument("file", nargs=-1)
+@click.option(
+    "--folder",
+    default=".",
+    type=click.Path(exists=True, file_okay=False),
+    help="Folder where tests will be placed",
+)
 @click.pass_context
 @coro
-async def test_parse(ctx: click.Context, files: Tuple[str, ...]) -> None:
-    for f in files:
-        click.echo(f"\nFile: {f}")
-        for test in parse_file(f):
-            click.echo(test)
+async def test_run(ctx: click.Context, file: Tuple[str, ...], folder: Optional[str]) -> None:
+    client = await get_tinybird_local_client(os.path.abspath(folder))
+    file_list: Iterable[str] = file if len(file) > 0 else glob.glob("./tests/**/*.y*ml", recursive=True)
+
+    async def run_test(test_file):
+        test_file_path = Path(test_file)
+        test_file_content = yaml.safe_load(test_file_path.read_text())
+        for test in test_file_content:
+            try:
+                response = await client._req(f"/v0/pipes/{test_file_path.stem}.ndjson?{test['parameters']}")
+                if test["expected_result"] != json.dumps(response):
+                    raise Exception("Test failed")
+                click.echo(FeedbackManager.success(message=f"✓ {test_file_path.name} - {test['name']}"))
+            except Exception:
+                click.echo(FeedbackManager.error(message=f"✗ {test_file_path.name} - {test['name']}"))
+
+    for test_file in file_list:
+        await run_test(test_file)

{tinybird-0.0.1.dev13.dist-info → tinybird-0.0.1.dev15.dist-info}/METADATA

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

{tinybird-0.0.1.dev13.dist-info → tinybird-0.0.1.dev15.dist-info}/RECORD

@@ -1,5 +1,4 @@
 tinybird/__cli__.py,sha256=pgYsVLcqL16wtSn6KtKweNZYoYJdEksTgSvQAW7hH64,250
-tinybird/check_pypi.py,sha256=RIhZcTg9dk5SFuvV3yVtFgBael_5_VYl4GzOZJojCfU,971
 tinybird/client.py,sha256=nd97gD2-8Ap8yDonBcVwk9eXDAL43hmIYdo-Pse43RE,50738
 tinybird/config.py,sha256=Z-BX9FrjgsLw1YwcCdF0IztLB97Zpc70VVPplO_pDSY,6089
 tinybird/connectors.py,sha256=lkpVSUmSuViEZBa4QjTK7YmPHUop0a5UFoTrSmlVq6k,15244
@@ -15,49 +14,49 @@ tinybird/syncasync.py,sha256=fAvq0qkRgqXqXMKwbY2iJNYqLT_r6mDsh1MRpGKrdRU,27763
 tinybird/tornado_template.py,sha256=o2HguxrL1Evnt8o3IvrsI8Zm6JtRQ3zhLJKf1XyR3SQ,41965
 tinybird/ch_utils/constants.py,sha256=aYvg2C_WxYWsnqPdZB1ZFoIr8ZY-XjUXYyHKE9Ansj0,3890
 tinybird/ch_utils/engine.py,sha256=OXkBhlzGjZotjD0vaT-rFIbSGV4tpiHxE8qO_ip0SyQ,40454
-tinybird/tb/cli.py,sha256=-ritrHugOxQCW7qtV1hSIW8xEHKWDILawEwlcP9AVyI,816
+tinybird/tb/cli.py,sha256=NipQJ0V-uqJDQSb4xkTw04ALzgqSfe1twmjkBkQJWT4,784
 tinybird/tb/modules/auth.py,sha256=hynZ-Temot8YBsySUWKSFzZlYadtFPxG3o6lCSu1n6E,9018
 tinybird/tb/modules/branch.py,sha256=R1tTUBGyI0p_dt2IAWbuyNOvemhjCIPwYxEmOxL3zOg,38468
-tinybird/tb/modules/build.py,sha256=_kiloHo4cGdyQyp7TgCmXGHtmPP0cCL1gq1m-YcwNHQ,11181
+tinybird/tb/modules/build.py,sha256=_EvBtX5FO7fnpHP74qlrBJqm7Uk1Bx9NarJ8P-M8vvg,12901
 tinybird/tb/modules/cicd.py,sha256=KCFfywFfvGRh24GZwqrhICiTK_arHelPs_X4EB-pXIw,7331
-tinybird/tb/modules/cli.py,sha256=c-XNRu-idb2Hz43IT9ejd-QjsZy-xPQ3rnrdVIz0wxM,56568
+tinybird/tb/modules/cli.py,sha256=pF7bxobeLJP-aN7RwJmLaYOKQZHlBnneZpYbiAbTmaM,55511
 tinybird/tb/modules/common.py,sha256=Vubc2AIR8BfEupnT5e1Y8OYGEyvNoIcjo8th-SaUflw,80111
 tinybird/tb/modules/config.py,sha256=ppWvACHrSLkb5hOoQLYNby2w8jR76-8Kx2NBCst7ntQ,11760
 tinybird/tb/modules/connection.py,sha256=ZSqBGoRiJedjHKEyB_fr1ybucOHtaad8d7uqGa2Q92M,28668
-tinybird/tb/modules/create.py,sha256=Ky5LOyDJLgaHyWDt8un100QxKgNiQpEal-QzIW0V85I,6590
+tinybird/tb/modules/create.py,sha256=WL2GzJBgH3z1m6oU58uZXyUQRzVB2rH4ig4BIdOJkLw,6307
 tinybird/tb/modules/datasource.py,sha256=tjcf5o-HYIdTkb_c1ErGUFIE-W6G992vsvCuDGcxb9Q,35818
 tinybird/tb/modules/exceptions.py,sha256=4A2sSjCEqKUMqpP3WI00zouCWW4uLaghXXLZBSw04mY,3363
 tinybird/tb/modules/fmt.py,sha256=UszEQO15fdzQ49QEj7Unhu68IKwSuKPsOrKhk2p2TAg,3547
 tinybird/tb/modules/job.py,sha256=eoBVyA24lYIPonU88Jn7FF9hBKz1kScy9_w_oWreuc4,2952
-tinybird/tb/modules/llm.py,sha256=TvJJ9BlKISAb1SVI-pnHp_PcHcxGfTyjxOE_qAz90Ck,2441
-tinybird/tb/modules/local.py,sha256=sImiZwUMsvJRGBVZovOGBqxXo0SBWYwpZ7b8zVG_QNc,6943
-tinybird/tb/modules/login.py,sha256=a96yvh9stVQXL-JGTBf8NKMrIkp3nu2oT8TymVeDz9o,5796
-tinybird/tb/modules/mock.py,sha256=RohmEhNfudVryn2pJrI4fASE74inovNxzN0ew85Y830,2747
+tinybird/tb/modules/llm.py,sha256=j8AZl-4BaCrtpM3_ozGt2Pr0BPfPFG5qYAaFgjJabGk,2008
+tinybird/tb/modules/local.py,sha256=ajOrQMttYLXNs-4uDW9njw7-eIAHENvlbrNbURhr6gI,6926
+tinybird/tb/modules/login.py,sha256=vIeysdttfGDBMkf_i3cqAVNR5s0X0D6exATcRsDdWiA,5849
+tinybird/tb/modules/mock.py,sha256=HorUdULzxCCCGsFlPvL4RJeMJvVNnKnhbxxSog2Cn1Y,2817
 tinybird/tb/modules/pipe.py,sha256=9wnfKbp2FkmLiJgVk3qbra76ktwsUTXghu6j9cCEahQ,31058
-tinybird/tb/modules/prompts.py,sha256=g0cBW2ePzuftib02wV82VIcAZd59buAAusnirAbzqVE,8662
+tinybird/tb/modules/prompts.py,sha256=KIDoAC0FvBSSsTft_mhpY6vxliNRBYOCsqikr_Hx0HA,9655
 tinybird/tb/modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
 tinybird/tb/modules/table.py,sha256=hG-PRDVuFp2uph41WpoLRV1yjp3RI2fi_iGGiI0rdxU,7695
 tinybird/tb/modules/tag.py,sha256=1qQWyk1p3Btv3LzM8VbJG-k7x2-pFuAlYCg3QL6QewI,3480
 tinybird/tb/modules/telemetry.py,sha256=iEGnMuCuNhvF6ln__j6X9MSTwL_0Hm-GgFHHHvhfknk,10466
-tinybird/tb/modules/test.py,sha256=psINFpSYT1eGgy32-_4q6CJ7LOcdwBpAfasMA0_tNOU,4330
+tinybird/tb/modules/test.py,sha256=UR7Fz8AHhIptZfxofidzNzTnfAgwhCSWaGyjKbD2TAQ,4605
 tinybird/tb/modules/token.py,sha256=r0oeG1RpOOzHtqbUaHBiOmhE55HfNIvReAAWyKl9fJg,12695
 tinybird/tb/modules/workspace.py,sha256=FVlh-kbiZp5Gvp6dGFxi0UD8ail77rMamXLhqdVwrZ0,10916
 tinybird/tb/modules/workspace_members.py,sha256=08W0onEYkKLEC5TkAI07cxN9XSquEm7HnL7OkHAVDjo,8715
-tinybird/tb/modules/datafile/build.py,sha256=bo5T-_9LWsw4dZoHDO2bgn4hpSNOK5u_RiiYlRGLToA,91948
+tinybird/tb/modules/datafile/build.py,sha256=rFdK_GerPDgPfyPfZ4EZ0-cQqWfHd6htS0ls-Yy7khk,92491
 tinybird/tb/modules/datafile/build_common.py,sha256=74547h5ja4C66DAwDMabj75FA_BUTJxTJv-24tSFmrs,4551
 tinybird/tb/modules/datafile/build_datasource.py,sha256=fquzEGwk9NL_0K5YYG86Xtvgn4J5YHtRUoKJxbQGO0s,17344
-tinybird/tb/modules/datafile/build_pipe.py,sha256=wxqvVY3vIlG2_IAX8__mevhxqGkOxQ4-YyoWE6v2OxE,27465
-tinybird/tb/modules/datafile/common.py,sha256=q0XPpNE-l011Um3TXh3BmkSkUlYP5Ydkn24jXLq1I9Y,34239
-tinybird/tb/modules/datafile/diff.py,sha256=ZaTPGjRFJWokhaad_rMSxfYT92PA96s4WhhvlZubgyA,6769
+tinybird/tb/modules/datafile/build_pipe.py,sha256=sSSl1rQMkR4uUbFCxK_aDXZi3JwKV64YZlBdBWgGKjo,27657
+tinybird/tb/modules/datafile/common.py,sha256=NkoCdj4p-Ak3n80DJB5a33Ucw2WTcSYa8iqw4KsRZGs,81082
+tinybird/tb/modules/datafile/diff.py,sha256=-iaP7GvAzZtZSa8jPgVpOFlTRutxgxRBLBcGL1_RFr4,6743
 tinybird/tb/modules/datafile/exceptions.py,sha256=8rw2umdZjtby85QbuRKFO5ETz_eRHwUY5l7eHsy1wnI,556
 tinybird/tb/modules/datafile/fixture.py,sha256=YHlL4tojmPwm343Y8KO6r7d5Bhsk7U3lKP-oLMeBMsY,1771
 tinybird/tb/modules/datafile/format_common.py,sha256=zNWDXvwSKC9_T5e9R92LLj9ekDflVWwsllhGQilZsnY,2184
 tinybird/tb/modules/datafile/format_datasource.py,sha256=tsnCjONISvhFuucKNbIHkT__UmlUbcswx5mwI9hiDQc,6216
 tinybird/tb/modules/datafile/format_pipe.py,sha256=R5tnlEccLn3KX6ehtC_H2sGQNrthuJUiVSN9z_-KGCY,7474
-tinybird/tb/modules/datafile/parse_datasource.py,sha256=9pp0fJ2Da-7pznaHq5OV7NvzswK0RuXOPHU4kXNRMnA,1111
+tinybird/tb/modules/datafile/parse_datasource.py,sha256=YKt4Sy830p2jqXW5jN1Bf60AM5o4gZ3QcZjxnyWIeBg,1190
 tinybird/tb/modules/datafile/parse_pipe.py,sha256=STgA12LOLUnnb_cvVvZeEE4ka-nfk0jsNzxJhWj94cY,2599
 tinybird/tb/modules/datafile/pipe_checker.py,sha256=cp80Bru41GlyMRvyERpdJNXns2MjmtIAWFnBLF4cPXs,24667
-tinybird/tb/modules/datafile/pull.py,sha256=MleQG9GTo9_ZoMKrneps-QzI4dr6n5IPosHWz2sTz20,5894
+tinybird/tb/modules/datafile/pull.py,sha256=wBXBAZIruIyCRQZvfYxMc7h1q35NlKF-hFIF-bUm4iY,5956
 tinybird/tb/modules/tinyunit/tinyunit.py,sha256=IkjRCvb8HnNEE84rtl0I1b9gQVpE_zCE8MvFFet51sg,11716
 tinybird/tb/modules/tinyunit/tinyunit_lib.py,sha256=hGh1ZaXC1af7rKnX7222urkj0QJMhMWclqMy59dOqwE,1922
 tinybird/tb_cli_modules/cicd.py,sha256=0lMkb6CVOFZl5HOwgY8mK4T4mgI7O8335UngLXtCc-c,13851
@@ -66,8 +65,8 @@ tinybird/tb_cli_modules/config.py,sha256=6NTgIdwf0X132A1j6G_YrdPep87ymZ9b5pABabK
 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.dev13.dist-info/METADATA,sha256=j-z3lZinpg9H9UK5iqH4nKMe9A0KjGt3etk9EPFq94M,2405
-tinybird-0.0.1.dev13.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tinybird-0.0.1.dev13.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev13.dist-info/top_level.txt,sha256=pgw6AzERHBcW3YTi2PW4arjxLkulk2msOz_SomfOEuc,45
-tinybird-0.0.1.dev13.dist-info/RECORD,,
+tinybird-0.0.1.dev15.dist-info/METADATA,sha256=D7NcYU4DyAQdBn5hevIYUm3gugIyzdVfXYT7L5DxTYQ,2405
+tinybird-0.0.1.dev15.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tinybird-0.0.1.dev15.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev15.dist-info/top_level.txt,sha256=pgw6AzERHBcW3YTi2PW4arjxLkulk2msOz_SomfOEuc,45
+tinybird-0.0.1.dev15.dist-info/RECORD,,

tinybird/check_pypi.py

@@ -1,25 +0,0 @@
-from typing import Optional
-
-import requests
-
-from tinybird.feedback_manager import FeedbackManager
-from tinybird.syncasync import sync_to_async
-from tinybird.tb_cli_modules.common import CLIException, getenv_bool
-
-PYPY_URL = "https://pypi.org/pypi/tinybird/json"
-requests_get = sync_to_async(requests.get, thread_sensitive=False)
-
-
-class CheckPypi:
-    async def get_latest_version(self) -> Optional[str]:
-        version: Optional[str] = None
-        try:
-            disable_ssl: bool = getenv_bool("TB_DISABLE_SSL_CHECKS", False)
-            response: requests.Response = await requests_get(PYPY_URL, verify=not disable_ssl)
-            if response.status_code != 200:
-                raise CLIException(FeedbackManager.error_exception(error=response.content.decode("utf-8")))
-            version = response.json()["info"]["version"]
-        except Exception as e:
-            raise CLIException(FeedbackManager.error_exception(error=str(e)))
-
-        return version