tinybird 0.0.1.dev262__py3-none-any.whl → 0.0.1.dev264__py3-none-any.whl

tinybird/tb/modules/agent/prompts.py

@@ -16,7 +16,6 @@ from tinybird.prompts import (
     pipe_instructions,
     s3_connection_example,
     sink_pipe_instructions,
-    test_instructions,
 )
 from tinybird.tb.modules.project import Project
 
@@ -89,10 +88,7 @@ sql_instructions = """
     - Use node names as table names only when nodes are present in the same file.
     - Do not reference the current node name in the SQL.
     - SQL queries only accept SELECT statements with conditions, aggregations, joins, etc.
-    - Do NOT use CREATE TABLE, INSERT INTO, CREATE DATABASE, SHOW TABLES, etc.
-    - Use ONLY SELECT statements in the SQL section.
-    - INSERT INTO is not supported in SQL section.
-    - Do NOT query system.<table_name> tables.
+    - ONLY SELECT statements are allowed in any sql query.
     - When using functions try always ClickHouse functions first, then SQL functions.
     - Parameters are never quoted in any case.
     - Use the following syntax in the SQL section for the iceberg table function: iceberg('s3://bucket/path/to/table', {{tb_secret('aws_access_key_id')}}, {{tb_secret('aws_secret_access_key')}})
@@ -115,7 +111,6 @@ datafile_instructions = """
 def resources_prompt(project: Project) -> str:
     files = project.get_project_files()
     fixture_files = project.get_fixture_files()
-    test_files = project.get_test_files()
 
     resources_content = "# Existing resources in the project:\n"
     if files:
@@ -148,6 +143,29 @@ def resources_prompt(project: Project) -> str:
     else:
         fixture_content += "No fixture files found"
 
+    return resources_content + "\n" + fixture_content
+
+
+def tests_files_prompt(project: Project) -> str:
+    files = project.get_project_files()
+    test_files = project.get_test_files()
+
+    resources_content = "# Existing resources in the project:\n"
+    if files:
+        resources: list[dict[str, Any]] = []
+        for filename in files:
+            file_path = Path(filename)
+            resource = {
+                "path": str(file_path.relative_to(project.folder)),
+                "type": get_resource_type(file_path),
+                "name": file_path.stem,
+                "content": file_path.read_text(),
+            }
+            resources.append(resource)
+        resources_content = format_as_xml(resources, root_tag="resources", item_tag="resource")
+    else:
+        resources_content += "No resources found"
+
     test_content = "# Test files in the project:\n"
     if test_files:
         tests: list[dict[str, Any]] = []
@@ -163,7 +181,7 @@ def resources_prompt(project: Project) -> str:
     else:
         test_content += "No test files found"
 
-    return resources_content + "\n" + fixture_content + "\n" + test_content
+    return resources_content + "\n" + test_content
 
 
 def get_resource_type(path: Path) -> str:
@@ -522,14 +540,62 @@ FROM small_table
 JOIN (SELECT id, col FROM huge_table WHERE id IN (SELECT id FROM small_table)) filtered 
 ON small_table.id = filtered.id
 ```
-"""
 
+<dev_notes>
+IMPORTANT: DO NOT USE THE FOLLOWING WHEN QUERYING:
+- CREATE TABLE, INSERT INTO, CREATE DATABASE, SHOW TABLES, TRUNCATE TABLE, DELETE FROM, SHOW DATASOURCES, etc. are not allowed.
+- ONLY use SELECT statements.
+- currentDatabase is not allowed.
+- system tables are not allowed: system.tables, system.datasources, information_schema.tables...
+</dev_notes>
+"""
 
-agent_system_prompt = f"""
-You are a Tinybird Code, an agentic CLI that can help users to work with Tinybird.
 
-You are an interactive CLI tool that helps users with data engineering tasks. Use the instructions below and the tools available to you to assist the user.
+test_instructions = """
+# Working with test files:
+- The test file name must match the name of the pipe it is testing.
+- Every scenario name must be unique inside the test file.
+- When looking for the parameters available, you will find them in the pipe file in the following format: {{{{String(my_param_name, default_value)}}}}.
+- If the resource has no parameters, generate a single test with empty parameters.
+- The format of the parameters is the following: param1=value1&param2=value2&param3=value3
+- If some parameters are provided by the user and you need to use them, preserve in the same format as they were provided, like case sensitive
+- Test as many scenarios as possible.
+- Create tests only when the user explicitly asks for it with prompts like "Create tests for this endpoint" or "Create tests for this pipe".
+- If the user asks for "testing an endpoint" or "call an endpoint", just request to the endpoint.
+- The data that the tests are using is the data provided in the fixtures folder, so do not use `execute_query` or `request_endpoint` tools to analyze the data.
+- MANDATORY: Before creating the test, analyze the fixture files that the tables of the endpoint are using so you can create relevant tests.
+- IMPORTANT: expected_result field should always be an empty string, because it will be filled by the `create_test` tool.
+- If the endpoint does not have parameters, you can omit parameters and generate a single test.
+- The format of the test file is the following:
+<test_file_format>
+- name: kpis_single_day
+  description: Test hourly granularity for a single day
+  parameters: date_from=2024-01-01&date_to=2024-01-01
+  expected_result: ''
+
+- name: kpis_date_range
+  description: Test daily granularity for a date range
+  parameters: date_from=2024-01-01&date_to=2024-01-31
+  expected_result: ''
+
+- name: kpis_default_range
+  description: Test default behavior without date parameters (last 7 days)
+  parameters: ''
+  expected_result: ''
+
+- name: kpis_fixed_time
+  description: Test with fixed timestamp for consistent testing
+  parameters: fixed_time=2024-01-15T12:00:00
+  expected_result: ''
+
+- name: kpis_single_day
+  description: Test single day with hourly granularity
+  parameters: date_from=2024-01-01&date_to=2024-01-01
+  expected_result: ''
+</test_file_format>
+"""
 
+tone_and_style_instructions = """
 # Tone and style
 You should be concise, direct, and to the point. Maintain a professional tone. Do not use emojis.
 Remember that your output will be displayed on a command line interface. Your responses can use Github-flavored markdown for formatting. 
@@ -548,6 +614,15 @@ Do not add additional code explanation summary unless requested by the user. Aft
 
 # Code style
 IMPORTANT: DO NOT ADD ANY COMMENTS unless asked by the user.
+"""
+
+
+agent_system_prompt = f"""
+You are a Tinybird Code, an agentic CLI that can help users to work with Tinybird.
+
+You are an interactive CLI tool that helps users with data engineering tasks. Use the instructions below and the tools available to you to assist the user.
+
+{tone_and_style_instructions}
 
 # Tools
 You have access to the following tools:
@@ -662,15 +737,11 @@ GCS: {gcs_connection_example}
 - `DateTime` parameters accept values in format `YYYY-MM-DD HH:MM:SS`
 - `Date` parameters accept values in format `YYYYMMDD`
 
-# Working with tests:
-{test_instructions}
-<dev_notes>
-- Create tests only when the user explicitly asks for it with prompts like "Create tests for this endpoint" or "Create tests for this pipe".
-- If the user asks for "testing an endpoint" or "call an endpoint", just request to the endpoint.
-- The data that the tests are using is the data provided in the fixtures folder.
-- Querying data or requesting endpoints won't return the data that the tests are using.
-- MANDATORY: Before creating the test, analyze the fixture files that the tables of the endpoint are using so you can create relevant tests.
-</dev_notes>
+# Working with test files:
+- Use `manage_tests` tool to create, update or run tests.
+
+# Working with commands:
+- If you dont have a tool that can solve the task, use `run_command` tool to check if the task can be solved with a normal tinybird cli command.
 
 # Info
 Today is {datetime.now().strftime("%Y-%m-%d")}

tinybird/tb/modules/agent/testing_agent.py

@@ -0,0 +1,62 @@
+from pydantic_ai import Agent, RunContext, Tool
+from pydantic_ai.messages import ModelMessage
+from pydantic_ai.usage import Usage
+
+from tinybird.tb.modules.agent.animations import ThinkingAnimation
+from tinybird.tb.modules.agent.models import create_model
+from tinybird.tb.modules.agent.prompts import test_instructions, tests_files_prompt, tone_and_style_instructions
+from tinybird.tb.modules.agent.tools.test import create_tests as create_tests_tool
+from tinybird.tb.modules.agent.tools.test import run_tests as run_tests_tool
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext
+from tinybird.tb.modules.project import Project
+
+
+class TestingAgent:
+    def __init__(
+        self,
+        token: str,
+        user_token: str,
+        host: str,
+        workspace_id: str,
+        project: Project,
+        dangerously_skip_permissions: bool,
+        prompt_mode: bool,
+        thinking_animation: ThinkingAnimation,
+    ):
+        self.token = token
+        self.user_token = user_token
+        self.host = host
+        self.dangerously_skip_permissions = dangerously_skip_permissions or prompt_mode
+        self.project = project
+        self.thinking_animation = thinking_animation
+        self.messages: list[ModelMessage] = []
+        self.agent = Agent(
+            model=create_model(user_token, host, workspace_id),
+            deps_type=TinybirdAgentContext,
+            instructions=[
+                """
+You are part of Tinybird Code, an agentic CLI that can help users to work with Tinybird.                 
+You are a sub-agent of the main Tinybird Code agent. You are responsible for managing test files.
+You can do the following:
+- Create new test files.
+- Update existing test files.
+- Run tests.
+""",
+                tone_and_style_instructions,
+                test_instructions,
+            ],
+            tools=[
+                Tool(create_tests_tool, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
+                Tool(run_tests_tool, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
+            ],
+        )
+
+        @self.agent.instructions
+        def get_tests_files(ctx: RunContext[TinybirdAgentContext]) -> str:
+            return tests_files_prompt(self.project)
+
+    def run(self, task: str, deps: TinybirdAgentContext, usage: Usage):
+        result = self.agent.run_sync(task, deps=deps, usage=usage, message_history=self.messages)
+        new_messages = result.new_messages()
+        self.messages.extend(new_messages)
+        return result

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

@@ -56,7 +56,7 @@ def create_datafile(ctx: RunContext[TinybirdAgentContext], resource: Datafile) -
         action_text = "created" if not exists else "updated"
         click.echo(FeedbackManager.success(message=f"✓ {resource.pathname} {action_text}"))
         ctx.deps.thinking_animation.start()
-        return f"{action_text} {resource.pathname}"
+        return f"{action_text} {resource.pathname}. Project built successfully."
     except AgentRunCancelled as e:
         raise e
     except CLIBuildException as e:

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

@@ -6,6 +6,20 @@ from tinybird.tb.modules.agent.utils import TinybirdAgentContext
 from tinybird.tb.modules.common import echo_safe_humanfriendly_tables_format_pretty_table
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 
+forbidden_commands = [
+    "currentDatabase()",
+    "create table",
+    "insert into",
+    "create database",
+    "show tables",
+    "show datasources",
+    "truncate table",
+    "delete from",
+    "system.tables",
+    "system.datasources",
+    "information_schema.tables",
+]
+
 
 def execute_query(ctx: RunContext[TinybirdAgentContext], query: str, task: str, cloud: bool = False):
     """Execute a query:
@@ -19,6 +33,10 @@ def execute_query(ctx: RunContext[TinybirdAgentContext], query: str, task: str,
         str: The result of the query.
     """
     try:
+        for forbidden_command in forbidden_commands:
+            if forbidden_command in query.lower():
+                return f"Error executing query: {forbidden_command} is not allowed."
+
         cloud_or_local = "cloud" if cloud else "local"
         ctx.deps.thinking_animation.stop()
 

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

@@ -0,0 +1,38 @@
+import subprocess
+
+import click
+from pydantic_ai import RunContext
+
+from tinybird.tb.modules.agent.utils import AgentRunCancelled, TinybirdAgentContext, show_confirmation, show_input
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+
+
+def run_command(ctx: RunContext[TinybirdAgentContext], command: str):
+    """Run a tinybird CLI command with the given arguments
+
+    Args:
+        command (str): The command to run. Required. Examples: `tb --local sql "select 1"`, `tb --cloud datasource ls`, `tb --help`
+    """
+    try:
+        ctx.deps.thinking_animation.stop()
+        confirmation = show_confirmation(
+            title=f"Run command: {command}?", skip_confirmation=ctx.deps.dangerously_skip_permissions
+        )
+
+        if confirmation == "review":
+            feedback = show_input(ctx.deps.workspace_name)
+            ctx.deps.thinking_animation.start()
+            return f"User did not confirm the command and gave the following feedback: {feedback}"
+
+        click.echo(FeedbackManager.highlight(message=f"» Running command: {command}"))
+        command = command.replace("tb", "tb --no-version-warning")
+        result = subprocess.run(command, shell=True, capture_output=True, text=True)
+        click.echo(result.stdout)
+        ctx.deps.thinking_animation.start()
+        return result.stdout
+    except AgentRunCancelled as e:
+        raise e
+    except Exception as e:
+        click.echo(FeedbackManager.error(message=f"Error running command: {e}"))
+        ctx.deps.thinking_animation.start()
+        return f"Error running command: {e}"

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

@@ -11,7 +11,9 @@ from tinybird.tb.modules.agent.utils import (
     show_confirmation,
     show_input,
 )
+from tinybird.tb.modules.exceptions import CLIBuildException
 from tinybird.tb.modules.feedback_manager import FeedbackManager
+from tinybird.tb.modules.test_common import dump_tests, parse_tests
 
 
 def create_tests(ctx: RunContext[TinybirdAgentContext], pipe_name: str, test_content: str) -> str:
@@ -27,11 +29,32 @@ def create_tests(ctx: RunContext[TinybirdAgentContext], pipe_name: str, test_con
     running_tests = False
     try:
         ctx.deps.thinking_animation.stop()
+        ctx.deps.build_project_test(silent=True)
         path = Path(ctx.deps.folder) / "tests" / f"{pipe_name}.yaml"
         current_test_content: Optional[str] = None
         if path.exists():
             current_test_content = path.read_text()
 
+        pipe_tests_content = parse_tests(test_content)
+        for test in pipe_tests_content:
+            test_params = test["parameters"].split("?")[1] if "?" in test["parameters"] else test["parameters"]
+            response = None
+            try:
+                response = ctx.deps.get_pipe_data_test(pipe_name=pipe_name, test_params=test_params)
+            except Exception:
+                continue
+
+            if response.status_code >= 400:
+                test["expected_http_status"] = response.status_code
+                test["expected_result"] = response.json()["error"]
+            else:
+                if "expected_http_status" in test:
+                    del test["expected_http_status"]
+
+                test["expected_result"] = response.text or ""
+
+        test_content = dump_tests(pipe_tests_content)
+
         if current_test_content:
             content = create_terminal_box(current_test_content, new_content=test_content, title=path.name)
         else:
@@ -107,6 +130,11 @@ def run_tests(ctx: RunContext[TinybirdAgentContext], pipe_name: Optional[str] =
             return f"All tests in the project ran successfully\n{test_output}"
     except AgentRunCancelled as e:
         raise e
+    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:
         error_message = str(e)
         test_exit_code = "test_error__error__"

tinybird/tb/modules/agent/utils.py

@@ -20,6 +20,7 @@ from prompt_toolkit.shortcuts import PromptSession
 from prompt_toolkit.styles import Style as PromptStyle
 from pydantic import BaseModel, Field
 from pydantic_ai import RunContext
+from requests import Response
 
 from tinybird.tb.modules.agent.memory import load_history
 from tinybird.tb.modules.feedback_manager import FeedbackManager
@@ -40,6 +41,7 @@ class TinybirdAgentContext(BaseModel):
     get_project_files: Callable[[], List[str]]
     explore_data: Callable[[str], str]
     build_project: Callable[..., None]
+    build_project_test: Callable[..., None]
     deploy_project: Callable[[], None]
     deploy_check_project: Callable[[], None]
     mock_data: Callable[..., list[dict[str, Any]]]
@@ -50,6 +52,7 @@ class TinybirdAgentContext(BaseModel):
     execute_query_local: Callable[..., dict[str, Any]]
     request_endpoint_cloud: Callable[..., dict[str, Any]]
     request_endpoint_local: Callable[..., dict[str, Any]]
+    get_pipe_data_test: Callable[..., Response]
     get_datasource_datafile_cloud: Callable[..., str]
     get_datasource_datafile_local: Callable[..., str]
     get_pipe_datafile_cloud: Callable[..., str]
@@ -714,13 +717,15 @@ class AgentRunCancelled(Exception):
     pass
 
 
-def show_confirmation(title: str, skip_confirmation: bool = False) -> ConfirmationResult:
+def show_confirmation(title: str, skip_confirmation: bool = False, show_review: bool = True) -> ConfirmationResult:
     if skip_confirmation:
         return "yes"
 
     while True:
         result = show_options(
-            options=["Yes, continue", "No, tell Tinybird Code what to do", "Cancel"],
+            options=["Yes, continue", "No, tell Tinybird Code what to do", "Cancel"]
+            if show_review
+            else ["Yes, continue", "Cancel"],
             title=title,
         )
 

tinybird/tb/modules/cli.py

@@ -399,21 +399,23 @@ def create_ctx_client(ctx: Context, config: Dict[str, Any], cloud: bool, staging
     command_always_test = ["test"]
 
     if (
-        show_warnings
-        and (cloud or command in commands_always_cloud)
+        (cloud or command in commands_always_cloud)
         and command not in commands_always_local
         and command not in command_always_test
     ):
-        click.echo(
-            FeedbackManager.gray(message=f"Running against Tinybird Cloud: Workspace {config.get('name', 'default')}")
-        )
+        if show_warnings:
+            click.echo(
+                FeedbackManager.gray(
+                    message=f"Running against Tinybird Cloud: Workspace {config.get('name', 'default')}"
+                )
+            )
 
         method = None
         if ctx.params.get("token"):
             method = "token via --token option"
         elif os.environ.get("TB_TOKEN"):
             method = "token from TB_TOKEN environment variable"
-        if method:
+        if method and show_warnings:
             click.echo(FeedbackManager.gray(message=f"Authentication method: {method}"))
 
         return _get_tb_client(config.get("token", ""), config["host"], staging=staging)

tinybird/tb/modules/datasource.py

@@ -95,7 +95,7 @@ def datasource_ls(ctx: Context, match: Optional[str], format_: str):
                 shared_from,
                 name,
                 humanfriendly.format_number(stats.get("row_count")) if stats.get("row_count", None) else "-",
-                humanfriendly.format_size(int(stats.get("bytes"))) if stats.get("bytes", None) else "-",
+                humanfriendly.format_size(int(stats.get("bytes", 0))) if stats.get("bytes", None) else "-",
                 t["created_at"][:-7],
                 t["updated_at"][:-7],
                 t.get("service", ""),
@@ -455,6 +455,8 @@ def datasource_truncate(ctx, datasource_name, yes, cascade):
                     except Exception as e:
                         raise CLIDatasourceException(FeedbackManager.error_exception(error=e))
                     click.echo(FeedbackManager.success_truncate_datasource(datasource=cascade_ds))
+    else:
+        click.echo(FeedbackManager.info(message="Operation cancelled by user"))
 
 
 @datasource.command(name="delete")

tinybird/tb/modules/deployment_common.py

@@ -349,12 +349,31 @@ def create_deployment(
             promote_deployment(client.host, HEADERS, wait=wait)
 
 
+def _build_data_movement_message(kind: str, source_mv_name: Optional[str]) -> str:
+    if kind == "backfill_with_mv_queries":
+        return f"Using Materialized Pipe {source_mv_name or ''}"
+    elif kind == "backfill_with_forward_query":
+        return "From live deployment using Forward Query"
+    else:
+        return ""
+
+
 def print_changes(result: dict, project: Project) -> None:
     deployment = result.get("deployment", {})
     resources_columns = ["status", "name", "type", "path"]
     resources: list[list[Union[str, None]]] = []
     tokens_columns = ["Change", "Token name", "Added permissions", "Removed permissions"]
     tokens: list[Tuple[str, str, str, str]] = []
+    data_movements_columns = ["Datasource", "Backfill type"]
+    data_movements = deployment.get("data_movements")
+    if data_movements is not None:
+        data_movements = [
+            (
+                dm.get("datasource_name"),
+                _build_data_movement_message(dm.get("kind"), dm.get("source_mv_name")),
+            )
+            for dm in data_movements
+        ]
 
     for ds in deployment.get("new_datasource_names", []):
         resources.append(["new", ds, "datasource", project.get_resource_path(ds, "datasource")])
@@ -415,3 +434,9 @@ def print_changes(result: dict, project: Project) -> None:
         echo_safe_humanfriendly_tables_format_smart_table(tokens, column_names=tokens_columns)
     else:
         click.echo(FeedbackManager.gray(message="* No changes in tokens to be deployed"))
+    if data_movements is not None:
+        if data_movements:
+            click.echo(FeedbackManager.info(message="\n Data that will be copied with this deployment:"))
+            echo_safe_humanfriendly_tables_format_smart_table(data_movements, column_names=data_movements_columns)
+        else:
+            click.echo(FeedbackManager.gray(message="* No data will be copied with this deployment"))

tinybird/tb/modules/test_common.py

@@ -133,6 +133,15 @@ def create_test(
     return tests
 
 
+def parse_tests(tests_content: str) -> List[Dict[str, Any]]:
+    return yaml.safe_load(tests_content)
+
+
+def dump_tests(tests: List[Dict[str, Any]]) -> str:
+    yaml_str = yaml.safe_dump(tests, sort_keys=False)
+    return yaml_str.replace("- name:", "\n- name:")
+
+
 def update_test(pipe: str, project: Project, client: TinyB) -> None:
     try:
         folder = project.folder
@@ -149,7 +158,7 @@ def update_test(pipe: str, project: Project, client: TinyB) -> None:
 
         click.echo(FeedbackManager.highlight(message=f"\n» Updating tests expectations for {pipe_name} endpoint..."))
         pipe_tests_path = Path(project.folder) / pipe_tests_path
-        pipe_tests_content = yaml.safe_load(pipe_tests_path.read_text())
+        pipe_tests_content = parse_tests(pipe_tests_path.read_text())
         for test in pipe_tests_content:
             test_params = test["parameters"].split("?")[1] if "?" in test["parameters"] else test["parameters"]
             response = None
@@ -197,7 +206,7 @@ def run_tests(name: Tuple[str, ...], project: Project, client: TinyB) -> None:
         def run_test(test_file) -> Optional[str]:
             test_file_path = Path(test_file)
             click.echo(FeedbackManager.info(message=f"* {test_file_path.stem}{test_file_path.suffix}"))
-            test_file_content = yaml.safe_load(test_file_path.read_text())
+            test_file_content = parse_tests(test_file_path.read_text())
             test_file_errors = ""
             for test in test_file_content:
                 try:

{tinybird-0.0.1.dev262.dist-info → tinybird-0.0.1.dev264.dist-info}/METADATA

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

{tinybird-0.0.1.dev262.dist-info → tinybird-0.0.1.dev264.dist-info}/RECORD

@@ -12,12 +12,12 @@ tinybird/syncasync.py,sha256=IPnOx6lMbf9SNddN1eBtssg8vCLHMt76SuZ6YNYm-Yk,27761
 tinybird/tornado_template.py,sha256=jjNVDMnkYFWXflmT8KU_Ssbo5vR8KQq3EJMk5vYgXRw,41959
 tinybird/ch_utils/constants.py,sha256=fPgZtwbr1ymxaW7uqVWHKmAbt7uGj3SxCCS3xsEMJqA,4151
 tinybird/ch_utils/engine.py,sha256=4X1B-iuhdW_mxKnX_m3iCsxgP9RPVgR75g7yH1vsJ6A,40851
-tinybird/datafile/common.py,sha256=Xd_abjiTAK1oGxg2T59rLePNEdGMwSqe8PnoWEBJLyk,99161
+tinybird/datafile/common.py,sha256=lN-fNCaxtFWQAd0WN5Q-mgljM7F8sYOBQcU1JJogw1g,105324
 tinybird/datafile/exceptions.py,sha256=8rw2umdZjtby85QbuRKFO5ETz_eRHwUY5l7eHsy1wnI,556
 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=0prVU9s5VBR_0ep0okT2MF0GMOH7JG72BdEplvxz_Mg,247
+tinybird/tb/__cli__.py,sha256=1Q1Mk8PoldzfuJEf_f-VHn7VSe0wfdgeKCPK_RcPvag,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
@@ -25,15 +25,15 @@ tinybird/tb/config.py,sha256=mhMTGnMB5KcxGoh3dewIr2Jjsa6pHE183gCPAQWyp6o,3973
 tinybird/tb/modules/build.py,sha256=efD-vamK1NPaDo9R86Hn8be2DYoW0Hh5bZiH7knK5dk,7790
 tinybird/tb/modules/build_common.py,sha256=dthlaDn_CuwZnedQcUi7iIdDoHWfSbbbGQwiDgNcmC0,13062
 tinybird/tb/modules/cicd.py,sha256=0KLKccha9IP749QvlXBmzdWv1On3mFwMY4DUcJlBxiE,7326
-tinybird/tb/modules/cli.py,sha256=baeNPOu72L4aGI6ELsY8e2_eCq8ILOhTfd4qysz3KXk,16910
+tinybird/tb/modules/cli.py,sha256=qZIp8Ez4sunQ3_lIj1QGc3yOp8-SYfNHASHpE_7Ji6Y,16978
 tinybird/tb/modules/common.py,sha256=tj6DR2yOqMMQ0PILwFGXmMogxdrbQCgj36HdSM611rs,82657
 tinybird/tb/modules/config.py,sha256=gK7rgaWTDd4ZKCrNEg_Uemr26EQjqWt6TjyQKujxOws,11462
 tinybird/tb/modules/connection.py,sha256=-MY56NUAai6EMC4-wpi7bT0_nz_SA8QzTmHkV7HB1IQ,17810
 tinybird/tb/modules/copy.py,sha256=dPZkcIDvxjJrlQUIvToO0vsEEEs4EYumbNV77-BzNoU,4404
 tinybird/tb/modules/create.py,sha256=pJxHXG69c9Z_21s-7VuJ3RZOF_nJU51LEwiAkvI3dZY,23251
-tinybird/tb/modules/datasource.py,sha256=cxq0VVjjidxq-v_JSIIAH7L90XNRctgNKsHRoQ_42OI,41632
+tinybird/tb/modules/datasource.py,sha256=pae-ENeHYIF1HHYRSOziFC-2FPLUFa0KS60YpdlKCS8,41725
 tinybird/tb/modules/deployment.py,sha256=EDEVOqFk5fp0fvvs2jV0mT5POFd-i_8uZIUREwzAbEk,13199
-tinybird/tb/modules/deployment_common.py,sha256=CswUMfp5dW0NgiuP_ZGORqvqooHFYU2ZJaxOcpRFDT0,17866
+tinybird/tb/modules/deployment_common.py,sha256=g-2wnR63tlxmTObs6xRjZRgOOaX0fFn3Jq4x21qI6aw,18975
 tinybird/tb/modules/deprecations.py,sha256=rrszC1f_JJeJ8mUxGoCxckQTJFBCR8wREf4XXXN-PRc,4507
 tinybird/tb/modules/dev_server.py,sha256=57FCKuWpErwYUYgHspYDkLWEm9F4pbvVOtMrFXX1fVU,10129
 tinybird/tb/modules/endpoint.py,sha256=ksRj6mfDb9Xv63PhTkV_uKSosgysHElqagg3RTt21Do,11958
@@ -63,28 +63,30 @@ tinybird/tb/modules/sink.py,sha256=dK2s__my0ePIUYrqBzhPSgdWN9rbpvP1G4dT7DJzz80,3
 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=O2-mS4uMU6nPi7yWPpWzshAgOlYKiGS-tkM12pXQGMI,1906
-tinybird/tb/modules/test_common.py,sha256=LRgR2m2AgO5BYfdhboNL6ywIrRCffXsjzSwQE7A1W2Y,12459
+tinybird/tb/modules/test_common.py,sha256=YZwAdSfYVXdvArfTc9tH-2QBOhb_XbnJ3eKvyXTJuEM,12717
 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=Ye2v5sFDDT1tL54XTcpOid9V8s9EG8hpuTCLNQyxCHk,24502
+tinybird/tb/modules/agent/agent.py,sha256=cHpPZiUeW525i9PxspGoilTc-1HPAHCCzxiYdcdJQ7c,28214
 tinybird/tb/modules/agent/animations.py,sha256=4WOC5_2BracttmMCrV0H91tXfWcUzQHBUaIJc5FA7tE,3490
-tinybird/tb/modules/agent/banner.py,sha256=7f97PeCPW-oW9mReQn3D0by8mnDhoc0VbfebEPRPI7c,3070
+tinybird/tb/modules/agent/banner.py,sha256=2UpuuIqWxS0Ltab6i_FE4dkNxlJCdgKGCMtbwQGiLA8,7185
+tinybird/tb/modules/agent/command_agent.py,sha256=Q4M5qV9j3aETrXswoWZ02ZWQZhJm42HGHO1ymJrfDnw,2665
 tinybird/tb/modules/agent/memory.py,sha256=O6Kumn9AyKxcTkhI45yjAUZ3ZIAibLOcNWoiEuLYeqY,3245
 tinybird/tb/modules/agent/models.py,sha256=LW1D27gjcd_jwFmghEzteCgToDfodX2B6B5S8BYbysw,735
-tinybird/tb/modules/agent/prompts.py,sha256=dFpbcKFb8u8O4elwV__RcKmjvTIG84mB4fzGnwVKtzg,28124
-tinybird/tb/modules/agent/utils.py,sha256=1X1cjP607StB0NFuT2CsPO7fBU05eI7LGeib-mkFVms,29390
+tinybird/tb/modules/agent/prompts.py,sha256=NEIk3OK0xXm3QS1Iox4T1MMo19hLhfaICMEwzqVGW7E,31069
+tinybird/tb/modules/agent/testing_agent.py,sha256=mH7ccuWeTnkpENCmwym9ubDFncMXQoSo0jq2pRxJVDY,2553
+tinybird/tb/modules/agent/utils.py,sha256=VYuoodb05Ucq0b3qgIAf2KhOTTRD85_Pfab0xKwbg40,29610
 tinybird/tb/modules/agent/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tinybird/tb/modules/agent/tools/analyze.py,sha256=CR5LXg4fou-zYEksqnjpJ0icvxJVoKnTctoI1NRvqCM,3873
 tinybird/tb/modules/agent/tools/append.py,sha256=XA8ZeqxZcRL_0ZCd5FyggpWeH53mwTMby4lHV8wQa7c,6039
 tinybird/tb/modules/agent/tools/build.py,sha256=Hm-xDAP9ckMiKquT-DmDg5H0yxZefLOaWKANyoVSaEQ,846
-tinybird/tb/modules/agent/tools/create_datafile.py,sha256=-Luv7GrZ3tEdEFaTFiN2_b2WEkTVl2Tp437qsvNEwZU,6902
+tinybird/tb/modules/agent/tools/create_datafile.py,sha256=M2DKTSDnAMs22oSVfBagjXQOTJejf5Kp-RHd15LDKqQ,6931
 tinybird/tb/modules/agent/tools/deploy.py,sha256=2hKj6LMYiDea6ventsBjE6ArECGIryTdo3X-LYo5oZI,1248
 tinybird/tb/modules/agent/tools/deploy_check.py,sha256=2Wr9hQfKPlhqhumOv5TNl_xFctvdq_DHZ2dI2h_LggY,1048
 tinybird/tb/modules/agent/tools/diff_resource.py,sha256=_9xHcDzCTKk_E1wKQbuktVqV6U9sA0kqYaBxWvtliX0,2613
-tinybird/tb/modules/agent/tools/execute_query.py,sha256=h8be9_lhyMy7UZoCWrExigP-cbq3lLnlLHD9SzDGBCM,2703
+tinybird/tb/modules/agent/tools/execute_query.py,sha256=egUhKqRDF9C9458-rtu-yr-OeyvS0UrEUQwUgGd0Gts,3170
 tinybird/tb/modules/agent/tools/explore.py,sha256=ihALc_kBcsjrKT3hZyicqyIowB0g_K3AtNNi-5uz9-8,412
 tinybird/tb/modules/agent/tools/get_endpoint_stats.py,sha256=LiEK6ToyPDW2aI8ijclzuwdYAcFmwH-TyjqdFEzQWAc,1689
 tinybird/tb/modules/agent/tools/get_openapi_definition.py,sha256=mjIVVXtgvTs5LzOR8Bp4jB1XhLVMysHrHXawkErFdt8,2282
@@ -92,7 +94,8 @@ tinybird/tb/modules/agent/tools/mock.py,sha256=4JJ_45uWkbDhTGeKpwDv8L07ewsbF8u7q
 tinybird/tb/modules/agent/tools/plan.py,sha256=2KHLNkr2f1RfkbAR4mCVsv94LGosXd8-ky7v6BB1OtQ,985
 tinybird/tb/modules/agent/tools/preview_datafile.py,sha256=Gbao_FxhXstnUnngVQxztpizjugyfx1rOXTkw7Yabls,858
 tinybird/tb/modules/agent/tools/request_endpoint.py,sha256=xNYIEFI9sfq3SjevYqJaHH1EP6uEX5HHrbKS6qo3Imo,2852
-tinybird/tb/modules/agent/tools/test.py,sha256=0gXTfx6Yj1ZPDNDkKJj0RRhA6RQr6UtTM43KsHrdR_Y,4813
+tinybird/tb/modules/agent/tools/run_command.py,sha256=321w5dYOnapMf-TFIkgizErllwB5Poc0RYOjBAQeMPc,1576
+tinybird/tb/modules/agent/tools/test.py,sha256=CbGak_coopCTtqHoPWy-BwgLMIyEyeO34NTNkv18au4,6041
 tinybird/tb/modules/datafile/build.py,sha256=NFKBrusFLU0WJNCXePAFWiEDuTaXpwc0lHlOQWEJ43s,51117
 tinybird/tb/modules/datafile/build_common.py,sha256=2yNdxe49IMA9wNvl25NemY2Iaz8L66snjOdT64dm1is,4511
 tinybird/tb/modules/datafile/build_datasource.py,sha256=Ra8pVQBDafbFRUKlhpgohhTsRyp_ADKZJVG8Gd69idY,17227
@@ -113,8 +116,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.dev262.dist-info/METADATA,sha256=1iT43WWQIeJ-m391cjSMdLIL5aHzrSHDXlIs1us6bfM,1733
-tinybird-0.0.1.dev262.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev262.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev262.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev262.dist-info/RECORD,,
+tinybird-0.0.1.dev264.dist-info/METADATA,sha256=PCl2AWV403ysmvj35_eYngYE_1wfUNqlUlBN5mxep5M,1733
+tinybird-0.0.1.dev264.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev264.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev264.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev264.dist-info/RECORD,,