tinybird 0.0.1.dev280__tar.gz → 0.0.1.dev282__tar.gz

{tinybird-0.0.1.dev280 → tinybird-0.0.1.dev282}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev280
+Version: 0.0.1.dev282
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird
@@ -22,7 +22,8 @@ Requires-Dist: humanfriendly~=8.2
 Requires-Dist: plotext==5.3.2
 Requires-Dist: prompt_toolkit==3.0.48
 Requires-Dist: pydantic~=2.11.7
-Requires-Dist: pydantic-ai-slim[anthropic]~=0.4.2
+Requires-Dist: pydantic-ai-slim[anthropic]~=0.5.0
+Requires-Dist: pydantic-ai-slim[retries]~=0.5.0
 Requires-Dist: pyperclip==1.8.2
 Requires-Dist: pyyaml<6.1,>=6.0
 Requires-Dist: requests<3,>=2.28.1

{tinybird-0.0.1.dev280 → tinybird-0.0.1.dev282}/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.dev280'
-__revision__ = 'e419ff7'
+__version__ = '0.0.1.dev282'
+__revision__ = '77eb85f'

{tinybird-0.0.1.dev280 → tinybird-0.0.1.dev282}/tinybird/tb/modules/agent/agent.py

@@ -16,7 +16,7 @@ from requests import Response
 
 from tinybird.tb.check_pypi import CheckPypi
 from tinybird.tb.client import TinyB
-from tinybird.tb.config import CURRENT_VERSION
+from tinybird.tb.config import CURRENT_VERSION, get_display_cloud_host
 from tinybird.tb.modules.agent.animations import ThinkingAnimation
 from tinybird.tb.modules.agent.banner import display_banner
 from tinybird.tb.modules.agent.command_agent import CommandAgent
@@ -212,7 +212,12 @@ class TinybirdAgent:
 
         @self.agent.instructions
         def get_local_host(ctx: RunContext[TinybirdAgentContext]) -> str:
-            return f"Tinybird Local host: {ctx.deps.local_host}"
+            return f"""
+# Tinybird Local info:
+- API Host: {ctx.deps.local_host}
+- Token: {ctx.deps.local_token}
+- UI Dashboard URL: {get_display_cloud_host(ctx.deps.local_host)}/{ctx.deps.workspace_name}
+            """
 
         @self.agent.instructions
         def get_cloud_host(ctx: RunContext[TinybirdAgentContext]) -> str:
@@ -230,12 +235,14 @@ class TinybirdAgent:
 
             region_provider = region["provider"]
             region_name = region["name"]
-            return f"""Tinybird Cloud info (region details):
+            return f"""
+# Tinybird Cloud info (region details):
 - API Host: {ctx.deps.host}
 - Workspace ID: {ctx.deps.workspace_id}
 - Workspace Name: {project.workspace_name} (in Tinybird Local the workspace name is the same because it is synced with Cloud)
 - Region provider: {region_provider}
 - Region name: {region_name}
+- UI Dashboard URL: {get_display_cloud_host(ctx.deps.host)}/{ctx.deps.workspace_name}
 """
 
         @self.agent.instructions

tinybird-0.0.1.dev282/tinybird/tb/modules/agent/models.py

@@ -0,0 +1,61 @@
+from typing import Optional
+
+from anthropic import AsyncAnthropic
+from httpx import AsyncClient, HTTPStatusError
+from pydantic_ai.models.anthropic import AnthropicModel, AnthropicModelName
+from pydantic_ai.providers.anthropic import AnthropicProvider
+from pydantic_ai.retries import AsyncTenacityTransport, wait_retry_after
+from tenacity import AsyncRetrying, retry_if_exception_type, stop_after_attempt, wait_exponential
+
+
+def create_retrying_client(token: str, workspace_id: str):
+    """Create a client with smart retry handling for multiple error types."""
+
+    def should_retry_status(response):
+        """Raise exceptions for retryable HTTP status codes."""
+        if response.status_code in (400, 429, 502, 503, 504):
+            response.raise_for_status()  # This will raise HTTPStatusError
+
+    transport = AsyncTenacityTransport(
+        controller=AsyncRetrying(
+            # Retry on HTTP errors and connection issues
+            retry=retry_if_exception_type((HTTPStatusError, ConnectionError)),
+            # Smart waiting: respects Retry-After headers, falls back to exponential backoff
+            wait=wait_retry_after(fallback_strategy=wait_exponential(multiplier=1, max=60), max_wait=300),
+            # Stop after 5 attempts
+            stop=stop_after_attempt(5),
+            # Re-raise the last exception if all retries fail
+            reraise=True,
+        ),
+        validate_response=should_retry_status,
+    )
+    return AsyncClient(transport=transport, params={"token": token, "workspace_id": workspace_id})
+
+
+def create_model(
+    token: str,
+    base_url: str,
+    workspace_id: str,
+    model: AnthropicModelName = "claude-4-sonnet-20250514",
+    run_id: Optional[str] = None,
+):
+    default_headers = {}
+    if run_id:
+        default_headers["X-Run-Id"] = run_id
+
+    client = AsyncAnthropic(
+        base_url=base_url,
+        http_client=create_retrying_client(token, workspace_id),
+        auth_token=token,
+        default_headers=default_headers,
+    )
+    return AnthropicModel(
+        model_name=model,
+        provider=AnthropicProvider(anthropic_client=client),
+    )
+
+
+model_costs = {
+    "input_cost_per_token": 3e-06,
+    "output_cost_per_token": 1.5e-05,
+}

{tinybird-0.0.1.dev280 → tinybird-0.0.1.dev282}/tinybird/tb/modules/agent/prompts.py

@@ -5,7 +5,6 @@ from typing import Any
 from pydantic_ai import format_as_xml
 
 from tinybird.prompts import (
-    copy_pipe_instructions,
     datasource_example,
     datasource_instructions,
     materialized_pipe_instructions,
@@ -35,6 +34,8 @@ available_commands = [
     "`tb sink ls`: List all sinks",
     "`tb workspace current`: Show the current workspace",
     "`tb workspace clear --yes`: Delete all resources in the workspace (Only available in Tinybird Local)",
+    "`tb local start --skip-new-version`: Start Tinybird Local container in non-interactive mode",
+    "`tb local restart --skip-new-version --yes`: Restart Tinybird Local container in non-interactive mode",
 ]
 
 plan_instructions = """
@@ -778,6 +779,37 @@ GCS_HMAC_SECRET {{ tb_secret("gcs_hmac_secret") }}
 ```
 """
 
+
+copy_pipe_instructions = """
+- Copy pipes should be created in the /copies folder.
+- In a .pipe file you can define how to export the result of a Pipe to a Data Source, optionally with a schedule.
+- Do not include `COPY_SCHEDULE` in the .pipe file unless is specifically requested by the user.
+- `COPY_SCHEDULE` is a cron expression that defines the schedule of the copy pipe.
+- `COPY_SCHEDULE` is optional and if not provided, the copy pipe will be executed only once.
+- `TARGET_DATASOURCE` is the name of the Data Source to export the result to.
+- `TYPE COPY` is the type of the pipe and it is mandatory for copy pipes.
+- If the copy pipe uses parameters, you must include the `%` character and a newline on top of every query to be able to use the parameters.
+- The content of the .pipe file must follow this format:
+
+<copy_pipe_example>
+DESCRIPTION Copy Pipe to export sales hour every hour to the sales_hour_copy Data Source
+
+NODE daily_sales
+SQL >
+    %
+    SELECT toStartOfDay(starting_date) day, country, sum(sales) as total_sales
+    FROM teams
+    WHERE
+    day BETWEEN toStartOfDay(now()) - interval 1 day AND toStartOfDay(now())
+    and country = {{ String(country, 'US')}}
+    GROUP BY day, country
+
+TYPE COPY
+TARGET_DATASOURCE sales_hour_copy
+COPY_SCHEDULE 0 * * * *
+</copy_pipe_example>
+"""
+
 agent_system_prompt = f"""
 You are a Tinybird Code, an agentic CLI that can help users to work with Tinybird.
 
@@ -867,6 +899,18 @@ where <scope> is one of the following: `TOKENS`, `ADMIN`, `ORG_DATASOURCES:READ`
 {sink_pipe_instructions}
 
 # Working with copy pipe files:
+
+## What are copy pipes?
+Copy pipes capture the result of a pipe at a moment in time and write the result into a target data source. 
+They can be run on a schedule, or executed on demand.
+
+## Use copy pipes for:
+- Event-sourced snapshots, such as change data capture (CDC).
+- Copy data from Tinybird to another location in Tinybird to experiment.
+- De-duplicate with snapshots.
+- Copy pipes should not be confused with materialized views. While materialized views continuously update as new events are inserted, copy pipes generate a single snapshot at a specific point in time.
+
+## Copy pipe instructions
 {copy_pipe_instructions}
 
 # Working with SQL queries:
@@ -913,6 +957,10 @@ where <scope> is one of the following: `TOKENS`, `ADMIN`, `ORG_DATASOURCES:READ`
 # When asked about the files in the project:
 - You can rely in your own context to answer the question.
 
+# When you need to copy/export data from the selected environment to a local file:
+- Use `explore_data` tool to export data from the selected environment to a local file.
+- Copy pipes do not copy data between environments, they are only used to copy data between data sources in the same environment.
+
 # Info
 Today is {datetime.now().strftime("%Y-%m-%d")}
 """

{tinybird-0.0.1.dev280 → tinybird-0.0.1.dev282}/tinybird/tb/modules/local.py

@@ -119,7 +119,13 @@ def remove() -> None:
     default=None,
     help="Path to the volumes directory. If not provided, the container data won't be persisted.",
 )
-def start(use_aws_creds: bool, volumes_path: str) -> None:
+@click.option(
+    "--skip-new-version",
+    default=False,
+    is_flag=True,
+    help="Skip pulling the latest Tinybird Local image. Use directly your current local image.",
+)
+def start(use_aws_creds: bool, volumes_path: str, skip_new_version: bool) -> None:
     """Start Tinybird Local"""
     if volumes_path is not None:
         absolute_path = Path(volumes_path).absolute()
@@ -128,7 +134,7 @@ def start(use_aws_creds: bool, volumes_path: str) -> None:
 
     click.echo(FeedbackManager.highlight(message="» Starting Tinybird Local..."))
     docker_client = get_docker_client()
-    start_tinybird_local(docker_client, use_aws_creds, volumes_path)
+    start_tinybird_local(docker_client, use_aws_creds, volumes_path, skip_new_version)
     click.echo(FeedbackManager.success(message="✓ Tinybird Local is ready!"))
 
 
@@ -144,7 +150,19 @@ def start(use_aws_creds: bool, volumes_path: str) -> None:
     default=None,
     help="Path to the volumes directory. If not provided, the container data won't be persisted.",
 )
-def restart(use_aws_creds: bool, volumes_path: str) -> None:
+@click.option(
+    "--skip-new-version",
+    default=False,
+    is_flag=True,
+    help="Skip pulling the latest Tinybird Local image. Use directly your current local image.",
+)
+@click.option(
+    "--yes",
+    default=False,
+    is_flag=True,
+    help="Skip the confirmation prompt. If provided, the container will be restarted without asking for confirmation.",
+)
+def restart(use_aws_creds: bool, volumes_path: str, skip_new_version: bool, yes: bool) -> None:
     """Restart Tinybird Local"""
     if volumes_path is not None:
         absolute_path = Path(volumes_path).absolute()
@@ -153,9 +171,10 @@ def restart(use_aws_creds: bool, volumes_path: str) -> None:
 
     click.echo(FeedbackManager.highlight(message="» Restarting Tinybird Local..."))
     docker_client = get_docker_client()
-    remove_tinybird_local(docker_client, volumes_path is not None)
+    persist_data = volumes_path is not None or yes
+    remove_tinybird_local(docker_client, persist_data)
     click.echo(FeedbackManager.info(message="✓ Tinybird Local stopped"))
-    start_tinybird_local(docker_client, use_aws_creds, volumes_path)
+    start_tinybird_local(docker_client, use_aws_creds, volumes_path, skip_new_version)
     click.echo(FeedbackManager.success(message="✓ Tinybird Local is ready!"))
 
 

{tinybird-0.0.1.dev280 → tinybird-0.0.1.dev282}/tinybird/tb/modules/local_common.py

@@ -220,29 +220,32 @@ def start_tinybird_local(
     docker_client: DockerClient,
     use_aws_creds: bool,
     volumes_path: Optional[str] = None,
+    skip_new_version: bool = False,
 ) -> None:
     """Start the Tinybird container."""
     pull_show_prompt = False
     pull_required = False
-    try:
-        local_image = docker_client.images.get(TB_IMAGE_NAME)
-        local_image_id = local_image.attrs["RepoDigests"][0].split("@")[1]
-        remote_image = docker_client.images.get_registry_data(TB_IMAGE_NAME)
-        pull_show_prompt = local_image_id != remote_image.id
-    except Exception:
-        pull_show_prompt = False
-        pull_required = True
-
-    if pull_show_prompt and click.confirm(
-        FeedbackManager.warning(message="△ New version detected, download? [y/N]:"),
-        show_default=False,
-        prompt_suffix="",
-    ):
-        click.echo(FeedbackManager.info(message="* Downloading latest version of Tinybird Local..."))
-        pull_required = True
-
-    if pull_required:
-        docker_client.images.pull(TB_IMAGE_NAME, platform="linux/amd64")
+
+    if not skip_new_version:
+        try:
+            local_image = docker_client.images.get(TB_IMAGE_NAME)
+            local_image_id = local_image.attrs["RepoDigests"][0].split("@")[1]
+            remote_image = docker_client.images.get_registry_data(TB_IMAGE_NAME)
+            pull_show_prompt = local_image_id != remote_image.id
+        except Exception:
+            pull_show_prompt = False
+            pull_required = True
+
+        if pull_show_prompt and click.confirm(
+            FeedbackManager.warning(message="△ New version detected, download? [y/N]:"),
+            show_default=False,
+            prompt_suffix="",
+        ):
+            click.echo(FeedbackManager.info(message="* Downloading latest version of Tinybird Local..."))
+            pull_required = True
+
+        if pull_required:
+            docker_client.images.pull(TB_IMAGE_NAME, platform="linux/amd64")
 
     environment = get_use_aws_creds() if use_aws_creds else {}
 

{tinybird-0.0.1.dev280 → tinybird-0.0.1.dev282}/tinybird/tb/modules/login_common.py

@@ -257,19 +257,31 @@ def open_url(url: str, *, new_tab: bool = False) -> bool:
 
     # 2. Inside WSL, prefer `wslview` if the user has it (wslu package).
     if _running_in_wsl() and shutil.which("wslview"):
-        subprocess.Popen(["wslview", url])
-        return True
+        try:
+            subprocess.Popen(["wslview", url])
+            return True
+        except Exception:
+            # wslview not found, continue to next fallback
+            pass
 
     # 3. Secondary WSL fallback use Windows **start** through cmd.exe.
     #    Empty "" argument is required so long URLs are not treated as a window title.
     if _running_in_wsl():
-        subprocess.Popen(["cmd.exe", "/c", "start", "", url])
-        return True
+        try:
+            subprocess.Popen(["cmd.exe", "/c", "start", "", url])
+            return True
+        except Exception:
+            # cmd.exe not found, continue to next fallback
+            pass
 
     # 4. Unix last-ditch fallback xdg-open (most minimal container images have it)
     if shutil.which("xdg-open"):
-        subprocess.Popen(["xdg-open", url])
-        return True
+        try:
+            subprocess.Popen(["xdg-open", url])
+            return True
+        except Exception:
+            # xdg-open not found, continue to next fallback
+            pass
 
     # 5. If everything failed, let the caller know.
     return False

{tinybird-0.0.1.dev280 → tinybird-0.0.1.dev282}/tinybird.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev280
+Version: 0.0.1.dev282
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird
@@ -22,7 +22,8 @@ Requires-Dist: humanfriendly~=8.2
 Requires-Dist: plotext==5.3.2
 Requires-Dist: prompt_toolkit==3.0.48
 Requires-Dist: pydantic~=2.11.7
-Requires-Dist: pydantic-ai-slim[anthropic]~=0.4.2
+Requires-Dist: pydantic-ai-slim[anthropic]~=0.5.0
+Requires-Dist: pydantic-ai-slim[retries]~=0.5.0
 Requires-Dist: pyperclip==1.8.2
 Requires-Dist: pyyaml<6.1,>=6.0
 Requires-Dist: requests<3,>=2.28.1

{tinybird-0.0.1.dev280 → tinybird-0.0.1.dev282}/tinybird.egg-info/requires.txt

@@ -13,7 +13,8 @@ humanfriendly~=8.2
 plotext==5.3.2
 prompt_toolkit==3.0.48
 pydantic~=2.11.7
-pydantic-ai-slim[anthropic]~=0.4.2
+pydantic-ai-slim[anthropic]~=0.5.0
+pydantic-ai-slim[retries]~=0.5.0
 pyperclip==1.8.2
 pyyaml<6.1,>=6.0
 requests<3,>=2.28.1

tinybird-0.0.1.dev280/tinybird/tb/modules/agent/models.py

@@ -1,35 +0,0 @@
-from typing import Optional
-
-from anthropic import AsyncAnthropic
-from httpx import AsyncClient
-from pydantic_ai.models.anthropic import AnthropicModel, AnthropicModelName
-from pydantic_ai.providers.anthropic import AnthropicProvider
-
-
-def create_model(
-    token: str,
-    base_url: str,
-    workspace_id: str,
-    model: AnthropicModelName = "claude-4-sonnet-20250514",
-    run_id: Optional[str] = None,
-):
-    default_headers = {}
-    if run_id:
-        default_headers["X-Run-Id"] = run_id
-
-    client = AsyncAnthropic(
-        base_url=base_url,
-        http_client=AsyncClient(params={"token": token, "workspace_id": workspace_id}),
-        auth_token=token,
-        default_headers=default_headers,
-    )
-    return AnthropicModel(
-        model_name=model,
-        provider=AnthropicProvider(anthropic_client=client),
-    )
-
-
-model_costs = {
-    "input_cost_per_token": 3e-06,
-    "output_cost_per_token": 1.5e-05,
-}