tinybird 0.0.1.dev261__py3-none-any.whl → 0.0.1.dev263__py3-none-any.whl

tinybird/tb/modules/agent/command_agent.py

@@ -0,0 +1,59 @@
+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 tests_files_prompt
+from tinybird.tb.modules.agent.tools.run_command import run_command
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext
+from tinybird.tb.modules.project import Project
+
+
+class CommandAgent:
+    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 running commands on the user's machine.
+You will be given a task to perform and you will use `run_command` tool to complete it.
+If you do not find a command that can solve the task, just say that there is no command that can solve the task.
+You can run `-h` in every level of the command to get help. E.g. `tb -h`, `tb datasource -h`, `tb datasource ls -h`.
+When you need to access Tinybird Cloud, add the `--cloud` flag. E.g. `tb --cloud datasource ls`.
+Token and host are not required to add to the commands.
+Always run first help commands to be sure that the commands you are running is not interactive.
+""",
+            ],
+            tools=[
+                Tool(run_command, 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)
+        self.messages.extend(result.new_messages())
+        return result

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.py

@@ -188,14 +188,27 @@ def deployment_group() -> None:
     default=None,
     help="URL of the template to use for the deployment. Example: https://github.com/tinybirdco/web-analytics-starter-kit/tree/main/tinybird",
 )
+@click.option(
+    "-v",
+    "--verbose",
+    is_flag=True,
+    default=False,
+    help="Show verbose output. Disabled by default.",
+)
 @click.pass_context
 def deployment_create(
-    ctx: click.Context, wait: bool, auto: bool, check: bool, allow_destructive_operations: bool, template: Optional[str]
+    ctx: click.Context,
+    wait: bool,
+    auto: bool,
+    check: bool,
+    allow_destructive_operations: bool,
+    template: Optional[str],
+    verbose: bool,
 ) -> None:
     """
     Validate and deploy the project server side.
     """
-    create_deployment_cmd(ctx, wait, auto, check, allow_destructive_operations, template)
+    create_deployment_cmd(ctx, wait, auto, check, allow_destructive_operations, template, verbose)
 
 
 @deployment_group.command(name="ls")
@@ -337,14 +350,27 @@ def deployment_discard(ctx: click.Context, wait: bool) -> None:
     default=None,
     help="URL of the template to use for the deployment. Example: https://github.com/tinybirdco/web-analytics-starter-kit/tree/main/tinybird",
 )
+@click.option(
+    "-v",
+    "--verbose",
+    is_flag=True,
+    default=False,
+    help="Show verbose output. Disabled by default.",
+)
 @click.pass_context
 def deploy(
-    ctx: click.Context, wait: bool, auto: bool, check: bool, allow_destructive_operations: bool, template: Optional[str]
+    ctx: click.Context,
+    wait: bool,
+    auto: bool,
+    check: bool,
+    allow_destructive_operations: bool,
+    template: Optional[str],
+    verbose: bool,
 ) -> None:
     """
     Deploy the project.
     """
-    create_deployment_cmd(ctx, wait, auto, check, allow_destructive_operations, template)
+    create_deployment_cmd(ctx, wait, auto, check, allow_destructive_operations, template, verbose)
 
 
 def create_deployment_cmd(
@@ -354,6 +380,7 @@ def create_deployment_cmd(
     check: Optional[bool] = None,
     allow_destructive_operations: Optional[bool] = None,
     template: Optional[str] = None,
+    verbose: bool = False,
 ) -> None:
     project: Project = ctx.ensure_object(dict)["project"]
     if template:
@@ -378,4 +405,4 @@ def create_deployment_cmd(
         click.echo(FeedbackManager.success(message="Template downloaded successfully"))
     client = ctx.ensure_object(dict)["client"]
     config: Dict[str, Any] = ctx.ensure_object(dict)["config"]
-    create_deployment(project, client, config, wait, auto, check, allow_destructive_operations)
+    create_deployment(project, client, config, wait, auto, verbose, check, allow_destructive_operations)

tinybird/tb/modules/deployment_common.py

@@ -194,6 +194,7 @@ def create_deployment(
     config: Dict[str, Any],
     wait: bool,
     auto: bool,
+    verbose: bool = False,
     check: Optional[bool] = None,
     allow_destructive_operations: Optional[bool] = None,
 ) -> None:
@@ -247,9 +248,12 @@ def create_deployment(
             if f.get("level", "").upper() == "ERROR":
                 feedback_func = FeedbackManager.error
                 feedback_icon = ""
-            else:
+            elif f.get("level", "").upper() == "WARNING":
                 feedback_func = FeedbackManager.warning
                 feedback_icon = "△ "
+            elif verbose and f.get("level", "").upper() == "INFO":
+                feedback_func = FeedbackManager.info
+                feedback_icon = ""
             resource = f.get("resource")
             resource_bit = f"{resource}: " if resource else ""
             click.echo(feedback_func(message=f"{feedback_icon}{f.get('level')}: {resource_bit}{f.get('message')}"))

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.dev261.dist-info → tinybird-0.0.1.dev263.dist-info}/METADATA

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