tinybird 0.0.1.dev249__tar.gz → 0.0.1.dev250__tar.gz

{tinybird-0.0.1.dev249 → tinybird-0.0.1.dev250}/PKG-INFO

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

{tinybird-0.0.1.dev249 → tinybird-0.0.1.dev250}/tinybird/config.py

@@ -33,6 +33,7 @@ LEGACY_HOSTS = {
     "https://api.europe-west2.gcp.tinybird.co": "https://cloud.tinybird.co/gcp/europe-west2",
     "https://api.ap-east.aws.tinybird.co": "https://app.tinybird.co/aws/ap-east",
     "https://api.northamerica-northeast2.gcp.tinybird.co": "https://app.tinybird.co/gcp/northamerica-northeast2",
+    "https://api.us-west1.gcp.tinybird.co": "https://app.tinybird.co/gcp/us-west1",
     "https://api.wadus1.gcp.tinybird.co": "https://app.wadus.tinybird.co/gcp/wadus1",
     "https://api.wadus2.gcp.tinybird.co": "https://app.wadus.tinybird.co/gcp/wadus2",
     "https://api.wadus3.gcp.tinybird.co": "https://app.wadus.tinybird.co/gcp/wadus3",
@@ -55,6 +56,7 @@ LEGACY_HOSTS = {
     "https://ui.split.tinybird.co": "https://app.tinybird.co/aws/split-us-east",
     "https://ui.split.us-west-2.aws.tinybird.co": "https://app.tinybird.co/aws/split-us-west-2",
     "https://ui.northamerica-northeast2.gcp.tinybird.co": "https://app.tinybird.co/gcp/northamerica-northeast2",
+    "https://ui.us-west1.gcp.tinybird.co": "https://app.tinybird.co/gcp/us-west1",
     "https://api.split.tinybird.co": "https://app.tinybird.co/aws/split-us-east",
     "https://api.split.us-west-2.aws.tinybird.co": "https://app.tinybird.co/aws/split-us-west-2",
     "https://ui.wadus1.gcp.tinybird.co": "https://app.wadus.tinybird.co/gcp/wadus1",
@@ -81,6 +83,7 @@ CLOUD_HOSTS = {
     "https://api.eu-west-1.aws.tinybird.co": "https://cloud.tinybird.co/aws/eu-west-1",
     "https://api.europe-west2.gcp.tinybird.co": "https://cloud.tinybird.co/gcp/europe-west2",
     "https://api.northamerica-northeast2.gcp.tinybird.co": "https://cloud.tinybird.co/gcp/northamerica-northeast2",
+    "https://api.us-west1.gcp.tinybird.co": "https://cloud.tinybird.co/gcp/us-west1",
     "https://api.ap-east.aws.tinybird.co": "https://cloud.tinybird.co/aws/ap-east",
     "https://ui.tinybird.co": "https://cloud.tinybird.co/gcp/europe-west3",
     "https://ui.us-east.tinybird.co": "https://cloud.tinybird.co/gcp/us-east4",
@@ -89,6 +92,7 @@ CLOUD_HOSTS = {
     "https://ui.eu-central-1.aws.tinybird.co": "https://cloud.tinybird.co/aws/eu-central-1",
     "https://ui.europe-west2.gcp.tinybird.co": "https://cloud.tinybird.co/gcp/europe-west2",
     "https://ui.northamerica-northeast2.gcp.tinybird.co": "https://cloud.tinybird.co/gcp/northamerica-northeast2",
+    "https://ui.us-west1.gcp.tinybird.co": "https://cloud.tinybird.co/gcp/us-west1",
 }
 
 

{tinybird-0.0.1.dev249 → tinybird-0.0.1.dev250}/tinybird/prompts.py

@@ -815,7 +815,7 @@ pipe_instructions = """
 <pipe_file_instructions>
     - The pipe names must be unique.
     - Nodes do NOT use the same name as the Pipe they belong to. So if the pipe name is "my_pipe", the nodes must be named different like "my_pipe_node_1", "my_pipe_node_2", etc.
-    - Nodes can't have the same exact name as the Pipe they belong to.
+    - Node names MUST be different from the resource names in the project.
     - Avoid more than one node per pipe unless it is really necessary or requested by the user.
     - No indentation is allowed for property names: DESCRIPTION, NODE, SQL, TYPE, etc.
     - Allowed TYPE values are: endpoint, copy, materialized, sink.

{tinybird-0.0.1.dev249 → tinybird-0.0.1.dev250}/tinybird/sql_template.py

@@ -1388,13 +1388,14 @@ _namespace = {
     "close": None,
     "print": None,
     "input": None,
-    "id": None,
 }
 
 
 reserved_vars = set(["_tt_tmp", "_tt_append", "isinstance", "str", "error", "custom_error", *list(vars(builtins))])
 for p in DEFAULT_PARAM_NAMES:  # we handle these in an specific manner
     reserved_vars.discard(p)  # `format` is part of builtins
+# Allow 'id' to be used as a template parameter - https://gitlab.com/tinybird/analytics/-/issues/19119
+reserved_vars.discard("id")
 error_vars = ["error", "custom_error"]
 
 
@@ -1463,6 +1464,9 @@ def generate(self, **kwargs) -> Tuple[str, TemplateExecutionResults]:
         return Expression(f"-- disable {feature}\n")
 
     namespace.update(_namespace)
+    # This is to fix the issue https://gitlab.com/tinybird/analytics/-/issues/19119
+    # We need to do the override here because if we modify the _namespace, we would filter out parameters from the users
+    namespace.update({"id": None})
     namespace.update(kwargs)
     namespace.update(
         {

{tinybird-0.0.1.dev249 → tinybird-0.0.1.dev250}/tinybird/tb/__cli__.py

@@ -4,5 +4,5 @@ __description__ = 'Tinybird Command Line Tool'
 __url__ = 'https://www.tinybird.co/docs/forward/commands'
 __author__ = 'Tinybird'
 __author_email__ = 'support@tinybird.co'
-__version__ = '0.0.1.dev249'
-__revision__ = 'ef4e98c'
+__version__ = '0.0.1.dev250'
+__revision__ = '1c47b89'

{tinybird-0.0.1.dev249 → tinybird-0.0.1.dev250}/tinybird/tb/modules/agent/agent.py

@@ -4,12 +4,9 @@ import sys
 from datetime import datetime
 from functools import partial
 from pathlib import Path
-from typing import Any
+from typing import Any, Optional
 
 import click
-from prompt_toolkit import prompt
-from prompt_toolkit.cursor_shapes import CursorShape
-from prompt_toolkit.styles import Style
 from pydantic_ai import Agent, Tool
 from pydantic_ai.messages import ModelMessage
 
@@ -29,7 +26,7 @@ from tinybird.prompts import (
 from tinybird.tb.client import TinyB
 from tinybird.tb.modules.agent.animations import ThinkingAnimation
 from tinybird.tb.modules.agent.banner import display_banner
-from tinybird.tb.modules.agent.memory import clear_history, load_history
+from tinybird.tb.modules.agent.memory import clear_history
 from tinybird.tb.modules.agent.models import create_model, model_costs
 from tinybird.tb.modules.agent.prompts import (
     datafile_instructions,
@@ -43,11 +40,12 @@ from tinybird.tb.modules.agent.tools.create_datafile import create_datafile
 from tinybird.tb.modules.agent.tools.deploy import deploy
 from tinybird.tb.modules.agent.tools.deploy_check import deploy_check
 from tinybird.tb.modules.agent.tools.explore import explore_data
+from tinybird.tb.modules.agent.tools.get_endpoint_stats import get_endpoint_stats
 from tinybird.tb.modules.agent.tools.mock import mock
 from tinybird.tb.modules.agent.tools.plan import plan
 from tinybird.tb.modules.agent.tools.preview_datafile import preview_datafile
 from tinybird.tb.modules.agent.tools.read_fixture_data import read_fixture_data
-from tinybird.tb.modules.agent.utils import TinybirdAgentContext
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_input
 from tinybird.tb.modules.build_common import process as build_process
 from tinybird.tb.modules.common import _analyze, _get_tb_client
 from tinybird.tb.modules.config import CLIConfig
@@ -115,6 +113,7 @@ You have access to the following tools:
 8. `mock` - Create mock data for a landing datasource.
 9. `read_fixture_data` - Read a fixture data file present in the project folder.
 10. `append` - Append existing fixture to a datasource.
+11. `get_endpoint_stats` - Get metrics of the requests to an endpoint.
 
 # When creating or updating datafiles:
 1. Use `plan` tool to plan the creation or update of resources.
@@ -126,6 +125,24 @@ You have access to the following tools:
 7. If the datafile was not created, finish the process and just wait for a new user prompt.
 8. If the datafile was created successfully, but the built failed, try to fix the error and repeat the process.
 
+# When creating a landing datasource given a .ndjson file:
+- If the user does not specify anything about the desired schema, create a schema like this:
+SCHEMA >
+    `data` String `json:$`
+- Use always json paths with .ndjson files.
+
+# When user wants to optimize an endpoint:
+First check if the query is optimized. E.g is filtering by a column present in the sorting key.
+Avoid when possible to not update the landing datasource.
+Check endpoint stats to analyze how the endpoint is performing. Use `get_endpoint_stats` tool to get the stats.
+When your data is in Tinybird, you can create intermediate data sources to preprocess data and make the endpoints faster. This can be done by using materialized views or copy pipes.
+- Copy pipes capture the result of a pipe at a specific point in time and write it to a target data source. They can run on a schedule or run on demand, making them ideal for event-sourced snapshots, data experimentation, and deduplication with snapshots.
+- Materialized views continuously re-evaluate a query as new events are inserted, maintaining an always up-to-date derived dataset. Unlike copy pipes which create point-in-time snapshots, materialized views provide real-time transformations of your data.
+Each approach has its own strengths and use cases:
+- Use copy pipes when you need scheduled or on-demand snapshots of your data.
+- Use materialized views when you need continuous, real-time transformations.
+Finally, update the existing endpoint itself, do not add a new one.
+
 IMPORTANT: If the user cancels some of the steps or there is an error in file creation, DO NOT continue with the plan. Stop the process and wait for the user before using any other tool.
 IMPORTANT: Every time you finish a plan and start a new resource creation or update process, create a new plan before starting with the changes.
 
@@ -175,6 +192,9 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
                 Tool(mock, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
                 Tool(read_fixture_data, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
                 Tool(append, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
+                Tool(
+                    get_endpoint_stats, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True
+                ),
             ],
         )
 
@@ -186,7 +206,7 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
         user_prompt = f"{user_prompt}\n\n{resources_prompt(project)}"
         client = TinyB(token=self.token, host=self.host)
         folder = self.project.folder
-
+        click.echo()
         thinking_animation = ThinkingAnimation(message="Chirping", delay=0.15)
         thinking_animation.start()
         result = self.agent.run_sync(
@@ -200,6 +220,8 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
                 mock_data=partial(mock_data, project=project, config=config),
                 append_data=partial(append_data, config=config),
                 analyze_fixture=partial(analyze_fixture, config=config),
+                execute_cloud_query=partial(execute_cloud_query, config=config),
+                execute_local_query=partial(execute_local_query, config=config),
                 get_project_files=project.get_project_files,
                 folder=folder,
                 thinking_animation=thinking_animation,
@@ -219,13 +241,13 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
             request_tokens * model_costs["input_cost_per_token"]
             + response_tokens * model_costs["output_cost_per_token"]
         )
-        click.echo()
         click.echo(result.output)
         click.echo("\n")
         click.echo(f"Input tokens: {request_tokens}")
         click.echo(f"Output tokens: {response_tokens}")
         click.echo(f"Total tokens: {total_tokens}")
         click.echo(f"Cost: ${cost:.6f}")
+        click.echo("\n")
 
 
 def run_agent(config: dict[str, Any], project: Project, dangerously_skip_permissions: bool):
@@ -233,6 +255,7 @@ def run_agent(config: dict[str, Any], project: Project, dangerously_skip_permiss
     host = config.get("host", None)
     user_token = config.get("user_token", None)
     workspace_id = config.get("id", None)
+    workspace_name = config.get("name", None)
     try:
         if not token or not host or not workspace_id or not user_token:
             yes = click.confirm(
@@ -274,20 +297,11 @@ def run_agent(config: dict[str, Any], project: Project, dangerously_skip_permiss
     try:
         while True:
             try:
-                user_input = prompt(
-                    [("class:prompt", f"tb ({project.workspace_name}) » ")],
-                    history=load_history(),
-                    cursor=CursorShape.BLOCK,
-                    style=Style.from_dict(
-                        {
-                            "prompt": "#40a8a8 bold",
-                            "": "",  # Normal color for user input
-                        }
-                    ),
-                )
+                user_input = show_input(workspace_name)
                 if user_input.startswith("tb "):
                     cmd_parts = shlex.split(user_input)
-                    subprocess.run(cmd_parts, check=True)
+                    subprocess.run(cmd_parts)
+                    click.echo()
                     continue
                 if user_input.lower() in ["/exit", "/quit"]:
                     click.echo(FeedbackManager.info(message="Goodbye!"))
@@ -393,3 +407,13 @@ def analyze_fixture(config: dict[str, Any], fixture_path: str) -> dict[str, Any]
     local_client = get_tinybird_local_client(config, test=False, silent=True)
     meta, _data = _analyze(fixture_path, local_client, Path(fixture_path).suffix.lstrip("."))
     return meta
+
+
+def execute_cloud_query(config: dict[str, Any], query: str, pipe_name: Optional[str] = None) -> str:
+    client = _get_tb_client(config["token"], config["host"])
+    return client.query(sql=query, pipeline=pipe_name)
+
+
+def execute_local_query(config: dict[str, Any], query: str, pipe_name: Optional[str] = None) -> str:
+    local_client = get_tinybird_local_client(config, test=False, silent=True)
+    return local_client.query(sql=query, pipeline=pipe_name)

{tinybird-0.0.1.dev249 → tinybird-0.0.1.dev250}/tinybird/tb/modules/agent/animations.py

@@ -4,15 +4,13 @@ import threading
 from time import sleep
 from typing import Optional
 
-import click
-
 
 class ThinkingAnimation:
     """Thinking animation that shows changing sparkles as a prefix."""
 
     def __init__(
         self,
-        message: str = "Thinking",
+        message: str = "Chirping",
         delay: float = 0.15,
         colors: bool = True,
         dots: bool = True,
@@ -37,17 +35,35 @@ class ThinkingAnimation:
         ]
         self.reset_color = "\033[0m"
 
-    def start(self):
+    def start(self, message: Optional[str] = None):
         """Start the animation in a separate thread."""
+        # Stop any existing animation first
+        if self.running:
+            self._stop_without_reset()
+
+        if message:
+            self.message = message
         self.running = True
         self.thread = threading.Thread(target=self._run_animation)
         self.thread.daemon = True
 
-        click.echo("\n")
+        # Clear the current line before starting new animation
+        sys.stdout.write("\r" + " " * (len(self.message) + 10) + "\r")
+        sys.stdout.flush()
         self.thread.start()
 
     def stop(self):
         """Stop the animation."""
+        self.message = "Chirping"
+        self.running = False
+        if self.thread:
+            self.thread.join()
+        # Clear the line and reset cursor position
+        sys.stdout.write("\r" + " " * (len(self.message) + 10) + "\r")
+        sys.stdout.flush()
+
+    def _stop_without_reset(self):
+        """Stop the animation without resetting the message."""
         self.running = False
         if self.thread:
             self.thread.join()
@@ -78,8 +94,11 @@ class ThinkingAnimation:
             else:
                 display_message = self.message
 
-            # Print the message with the prefix sparkle
-            sys.stdout.write(f"\r{colored_sparkle} {display_message}")
+            # Print the message with the prefix sparkle, padding to clear any leftover characters
+            line_content = f"{colored_sparkle} {display_message}"
+            # Pad with spaces to clear any leftover characters from longer previous messages
+            padded_line = line_content.ljust(len(self.message) + 10)
+            sys.stdout.write(f"\r{padded_line}")
             sys.stdout.flush()
 
             sleep(self.delay)

{tinybird-0.0.1.dev249 → tinybird-0.0.1.dev250}/tinybird/tb/modules/agent/prompts.py

@@ -22,8 +22,10 @@ Steps:
 <dev_notes>
 You can skip steps where resources will not be created or updated.
 Always add 'Build project' step after generating resources.
-Always add 'Generate mock data' step after building project if a landing datasource was created.
-Always add 'Append existing fixture' step after building project if a fixture file was provided at the beginning of the plan.
+Always add 'Generate mock data' step after building project if a landing datasource was created without providing a fixture file.
+Always add 'Append existing fixture' step after building project if a landing datasource was created after providing a fixture file.
+Solve the specific user request, do not add extra steps that are not related to the user request.
+Reuse the existing resources if possible.
 </dev_notes>
 
 Resource dependencies:

{tinybird-0.0.1.dev249 → tinybird-0.0.1.dev250}/tinybird/tb/modules/agent/tools/append.py

@@ -1,29 +1,10 @@
 import click
 from pydantic_ai import RunContext
 
-from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_options
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_confirmation, show_input
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 
-def get_append_confirmation(datasource_name: str) -> bool:
-    """Get user confirmation for appending existing fixture"""
-    while True:
-        result = show_options(
-            options=["Yes, append existing fixture", "No, and tell Tinybird Code what to do"],
-            title=f"Do you want to append existing fixture for datasource {datasource_name}?",
-        )
-
-        if result is None:  # Cancelled
-            return False
-
-        if result.startswith("Yes"):
-            return True
-        elif result.startswith("No"):
-            return False
-
-        return False
-
-
 def append(ctx: RunContext[TinybirdAgentContext], datasource_name: str, fixture_pathname: str) -> str:
     """Append existing fixture to a datasource
 
@@ -36,10 +17,20 @@ def append(ctx: RunContext[TinybirdAgentContext], datasource_name: str, fixture_
     """
     try:
         ctx.deps.thinking_animation.stop()
-        confirmation = ctx.deps.dangerously_skip_permissions or get_append_confirmation(datasource_name)
+        confirmation = show_confirmation(
+            title=f"Append existing fixture for datasource {datasource_name}?",
+            skip_confirmation=ctx.deps.dangerously_skip_permissions,
+        )
+
         ctx.deps.thinking_animation.start()
 
-        if not confirmation:
+        if confirmation == "review":
+            click.echo()
+            feedback = show_input(ctx.deps.workspace_name)
+            ctx.deps.thinking_animation.start()
+            return f"User did not confirm the proposed plan and gave the following feedback: {feedback}"
+
+        if confirmation == "cancel":
             return "User rejected appending existing fixture. Skipping..."
 
         ctx.deps.thinking_animation.stop()

{tinybird-0.0.1.dev249 → tinybird-0.0.1.dev250}/tinybird/tb/modules/agent/tools/create_datafile.py

@@ -3,31 +3,17 @@ from pathlib import Path
 import click
 from pydantic_ai import RunContext
 
-from tinybird.tb.modules.agent.utils import Datafile, TinybirdAgentContext, create_terminal_box, show_options
+from tinybird.tb.modules.agent.utils import (
+    Datafile,
+    TinybirdAgentContext,
+    create_terminal_box,
+    show_confirmation,
+    show_input,
+)
 from tinybird.tb.modules.exceptions import CLIBuildException
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 
-def get_resource_confirmation(resource: Datafile, exists: bool) -> bool:
-    """Get user confirmation for creating a resource"""
-    while True:
-        action = "create" if not exists else "update"
-        result = show_options(
-            options=[f"Yes, {action} {resource.type} '{resource.name}'", "No, and tell Tinybird Code what to do"],
-            title=f"What would you like to do with {resource.type} '{resource.name}'?",
-        )
-
-        if result is None:  # Cancelled
-            return False
-
-        if result.startswith("Yes"):
-            return True
-        elif result.startswith("No"):
-            return False
-
-        return False
-
-
 def create_datafile(ctx: RunContext[TinybirdAgentContext], resource: Datafile) -> str:
     """Given a resource representation, create a file in the project folder
 
@@ -48,19 +34,34 @@ def create_datafile(ctx: RunContext[TinybirdAgentContext], resource: Datafile) -
         else:
             content = create_terminal_box(resource.content, title=resource.pathname)
         click.echo(content)
-        confirmation = ctx.deps.dangerously_skip_permissions or get_resource_confirmation(resource, exists)
 
-        if not confirmation:
+        action = "Create" if not exists else "Update"
+        confirmation = show_confirmation(
+            title=f"{action} '{resource.pathname}'?",
+            skip_confirmation=ctx.deps.dangerously_skip_permissions,
+        )
+
+        if confirmation == "review":
+            click.echo()
+            feedback = show_input(ctx.deps.workspace_name)
+            ctx.deps.thinking_animation.start()
+            return f"User did not confirm the proposed changes and gave the following feedback: {feedback}"
+
+        if confirmation == "cancel":
             ctx.deps.thinking_animation.start()
-            return f"Resource {resource.pathname} was not created. User cancelled creation."
+            return f"User cancelled {action} of {resource.pathname}. Stop resource creation."
 
+        action_text = "Creating" if not exists else "Updating"
+        click.echo(FeedbackManager.highlight(message=f"\n» {action_text} {resource.pathname}..."))
         folder_path = path.parent
         folder_path.mkdir(parents=True, exist_ok=True)
         path.touch(exist_ok=True)
         path.write_text(resource.content)
         ctx.deps.build_project(test=True, silent=True)
+        action_text = "created" if not exists else "updated"
+        click.echo(FeedbackManager.success(message=f"✓ {resource.pathname} {action_text} successfully"))
         ctx.deps.thinking_animation.start()
-        return f"Created {resource.pathname}"
+        return f"{action_text} {resource.pathname}"
 
     except CLIBuildException as e:
         ctx.deps.thinking_animation.stop()
@@ -68,4 +69,7 @@ def create_datafile(ctx: RunContext[TinybirdAgentContext], resource: Datafile) -
         ctx.deps.thinking_animation.start()
         return f"Error building project: {e}"
     except Exception as e:
+        ctx.deps.thinking_animation.stop()
+        click.echo(FeedbackManager.error(message=e))
+        ctx.deps.thinking_animation.start()
         return f"Error creating {resource.pathname}: {e}"

tinybird-0.0.1.dev250/tinybird/tb/modules/agent/tools/deploy.py

@@ -0,0 +1,36 @@
+import click
+from pydantic_ai import RunContext
+
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_confirmation, show_input
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+
+
+def deploy(ctx: RunContext[TinybirdAgentContext]) -> str:
+    """Deploy the project"""
+    try:
+        ctx.deps.thinking_animation.stop()
+        confirmation = show_confirmation(
+            title="Deploy the project?",
+            skip_confirmation=ctx.deps.dangerously_skip_permissions,
+        )
+
+        if confirmation == "review":
+            click.echo()
+            feedback = show_input(ctx.deps.workspace_name)
+            ctx.deps.thinking_animation.start()
+            return f"User did not confirm deployment and gave the following feedback: {feedback}"
+
+        if confirmation == "cancel":
+            ctx.deps.thinking_animation.start()
+            return "User cancelled deployment. Stop deployment."
+
+        click.echo(FeedbackManager.highlight(message="\n» Deploying project..."))
+        ctx.deps.deploy_project()
+        click.echo(FeedbackManager.success(message="✓ Project deployed successfully"))
+        ctx.deps.thinking_animation.start()
+        return "Project deployed successfully"
+    except Exception as e:
+        ctx.deps.thinking_animation.stop()
+        click.echo(FeedbackManager.error(message=e))
+        ctx.deps.thinking_animation.start()
+        return f"Error depoying project: {e}"

tinybird-0.0.1.dev250/tinybird/tb/modules/agent/tools/get_endpoint_stats.py

@@ -0,0 +1,51 @@
+import click
+from pydantic_ai import RunContext
+
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+
+
+def get_endpoint_stats(
+    ctx: RunContext[TinybirdAgentContext], endpoint_name: str, interval_days: int = 1, cloud: bool = True
+):
+    """Get stats for an endpoint:
+
+    Args:
+        endpoint_name (str): The name of the endpoint to get stats for. Required.
+        interval_days (int): The number of days to get stats for. Optional.
+        cloud (bool): Whether to get stats from cloud or local. Optional.
+
+    Returns:
+        str: The result of the stats.
+    """
+    if interval_days == 1:
+        pipe_stats = "tinybird.pipe_stats_rt"
+        date_column = "start_datetime"
+    else:
+        pipe_stats = "tinybird.pipe_stats"
+        date_column = "date"
+
+    days = "day" if interval_days == 1 else "days"
+    cloud_or_local = "cloud" if cloud else "local"
+    ctx.deps.thinking_animation.stop()
+
+    click.echo(
+        FeedbackManager.highlight(
+            message=f"» Analyzing {cloud_or_local} requests in the last {interval_days} {days} for '{endpoint_name}' endpoint"
+        )
+    )
+
+    query = f"""SELECT * FROM {pipe_stats} 
+    WHERE {date_column} > NOW() - INTERVAL {interval_days} DAY
+    AND pipe_name = '{endpoint_name}'
+    LIMIT 100
+    FORMAT JSON
+    """
+
+    execute_query = ctx.deps.execute_cloud_query if cloud else ctx.deps.execute_local_query
+
+    result = execute_query(query=query)
+    click.echo(FeedbackManager.success(message="✓ Done!"))
+    click.echo()
+    ctx.deps.thinking_animation.start()
+    return f"Result for {endpoint_name} in the last {interval_days} {days}: {result}"

{tinybird-0.0.1.dev249 → tinybird-0.0.1.dev250}/tinybird/tb/modules/agent/tools/mock.py

@@ -1,30 +1,11 @@
 import click
 from pydantic_ai import RunContext
 
-from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_options
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_confirmation, show_input
 from tinybird.tb.modules.datafile.fixture import persist_fixture
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 
-def get_mock_confirmation(datasource_name: str) -> bool:
-    """Get user confirmation for creating mock data"""
-    while True:
-        result = show_options(
-            options=["Yes, create mock data", "No, and tell Tinybird Code what to do"],
-            title=f"Do you want to generate mock data for datasource {datasource_name}?",
-        )
-
-        if result is None:  # Cancelled
-            return False
-
-        if result.startswith("Yes"):
-            return True
-        elif result.startswith("No"):
-            return False
-
-        return False
-
-
 def mock(ctx: RunContext[TinybirdAgentContext], datasource_name: str, data_format: str, rows: int) -> str:
     """Create mock data for a datasource
 
@@ -38,13 +19,21 @@ def mock(ctx: RunContext[TinybirdAgentContext], datasource_name: str, data_forma
     """
     try:
         ctx.deps.thinking_animation.stop()
-        confirmation = ctx.deps.dangerously_skip_permissions or get_mock_confirmation(datasource_name)
-        ctx.deps.thinking_animation.start()
+        confirmation = show_confirmation(
+            title=f"Generate mock data for datasource '{datasource_name}'?",
+            skip_confirmation=ctx.deps.dangerously_skip_permissions,
+        )
 
-        if not confirmation:
-            return "User rejected mock data generation. Skipping..."
+        if confirmation == "review":
+            click.echo()
+            feedback = show_input(ctx.deps.workspace_name)
+            ctx.deps.thinking_animation.start()
+            return f"User did not confirm mock data generation and gave the following feedback: {feedback}"
+
+        if confirmation == "cancel":
+            ctx.deps.thinking_animation.start()
+            return "User cancelled mock data generation. Stop mock data generation."
 
-        ctx.deps.thinking_animation.stop()
         click.echo(FeedbackManager.highlight(message=f"\n» Generating mock data for {datasource_name}..."))
         data = ctx.deps.mock_data(datasource_name=datasource_name, data_format=data_format, rows=rows)
         fixture_path = persist_fixture(datasource_name, data, ctx.deps.folder, format=data_format)

tinybird-0.0.1.dev250/tinybird/tb/modules/agent/tools/plan.py

@@ -0,0 +1,37 @@
+import click
+from pydantic_ai import RunContext
+
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_confirmation, show_input
+
+
+def plan(ctx: RunContext[TinybirdAgentContext], plan: str) -> str:
+    """Given a plan, ask the user for confirmation to implement it
+
+    Args:
+        plan (str): The plan to implement. Required.
+
+    Returns:
+        str: If the plan was implemented or not.
+    """
+    try:
+        ctx.deps.thinking_animation.stop()
+        click.echo(plan)
+        confirmation = show_confirmation(
+            title="Do you want to implement the plan?", skip_confirmation=ctx.deps.dangerously_skip_permissions
+        )
+
+        if confirmation == "review":
+            click.echo()
+            feedback = show_input(ctx.deps.workspace_name)
+            ctx.deps.thinking_animation.start()
+            return f"User did not confirm the proposed plan and gave the following feedback: {feedback}"
+
+        ctx.deps.thinking_animation.start()
+
+        if confirmation == "cancel":
+            return "User cancelled the plan. Stop the process."
+
+        return "User confirmed the plan. Implementing..."
+
+    except Exception as e:
+        return f"Error implementing the plan: {e}"