tinybird 0.0.1.dev279__tar.gz → 0.0.1.dev280__tar.gz

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

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

{tinybird-0.0.1.dev279 → tinybird-0.0.1.dev280}/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.dev279'
-__revision__ = '9848d86'
+__version__ = '0.0.1.dev280'
+__revision__ = 'e419ff7'

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

@@ -14,7 +14,9 @@ from pydantic_ai import Agent, RunContext, Tool
 from pydantic_ai.messages import ModelMessage, ModelRequest, UserPromptPart
 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.modules.agent.animations import ThinkingAnimation
 from tinybird.tb.modules.agent.banner import display_banner
 from tinybird.tb.modules.agent.command_agent import CommandAgent
@@ -54,6 +56,7 @@ from tinybird.tb.modules.common import (
     echo_safe_humanfriendly_tables_format_pretty_table,
     get_region_from_host,
     get_regions,
+    update_cli,
 )
 from tinybird.tb.modules.config import CLIConfig
 from tinybird.tb.modules.deployment_common import create_deployment
@@ -368,6 +371,19 @@ class TinybirdAgent:
 def run_agent(
     config: dict[str, Any], project: Project, dangerously_skip_permissions: bool, prompt: Optional[str] = None
 ):
+    if not prompt:
+        latest_version = CheckPypi().get_latest_version()
+        if latest_version and "x.y.z" not in CURRENT_VERSION and latest_version != CURRENT_VERSION:
+            yes = click.confirm(
+                FeedbackManager.warning(
+                    message=f"New version available. {CURRENT_VERSION} -> {latest_version}. Do you want to update now? [Y/n]"
+                ),
+                show_default=False,
+                default=True,
+                prompt_suffix="",
+            )
+            if yes:
+                update_cli()
     click.echo(FeedbackManager.highlight(message="» Initializing Tinybird Code..."))
     token = config.get("token", None)
     host = config.get("host", None)

{tinybird-0.0.1.dev279 → tinybird-0.0.1.dev280}/tinybird/tb/modules/cli.py

@@ -121,7 +121,8 @@ def cli(
     if getenv_bool("TB_DISABLE_SSL_CHECKS", False):
         click.echo(FeedbackManager.warning_disabled_ssl_checks())
 
-    if not environ.get("PYTEST", None) and version_warning and not token:
+    is_agent_mode = ctx.invoked_subcommand is None
+    if not environ.get("PYTEST", None) and version_warning and not token and not is_agent_mode:
         latest_version = CheckPypi().get_latest_version()
         if latest_version:
             if "x.y.z" in CURRENT_VERSION:
@@ -204,7 +205,6 @@ def cli(
     ctx.ensure_object(dict)["env"] = get_target_env(cloud)
     ctx.ensure_object(dict)["output"] = output
 
-    is_agent_mode = ctx.invoked_subcommand is None
     is_prompt_mode = prompt is not None
 
     if is_agent_mode or is_prompt_mode:

{tinybird-0.0.1.dev279 → tinybird-0.0.1.dev280}/tinybird/tb/modules/common.py

@@ -9,6 +9,7 @@ import json
 import os
 import re
 import socket
+import subprocess
 import sys
 import time
 import uuid
@@ -2234,3 +2235,30 @@ def force_echo(string: str) -> None:
 
 def echo_json(data: Dict[str, Any], indent: Union[None, int, str] = None) -> None:
     force_echo(json.dumps(data, indent=indent))
+
+
+def update_cli() -> None:
+    click.echo(FeedbackManager.highlight(message="» Updating Tinybird CLI..."))
+
+    try:
+        process = subprocess.Popen(
+            ["uv", "tool", "upgrade", "tinybird"],
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            text=True,
+        )
+    except FileNotFoundError:
+        raise CLIException(
+            FeedbackManager.error(
+                message="Cannot find required tool: uv. Reinstall using: curl https://tinybird.co | sh"
+            )
+        )
+
+    stdout, stderr = process.communicate()
+    if "Nothing to upgrade" not in stdout + stderr:
+        for line in stdout.split("\n") + stderr.split("\n"):
+            if "Updated tinybird" in line:
+                click.echo(FeedbackManager.info(message=f"» {line}"))
+        click.echo(FeedbackManager.success(message="✓ Tinybird CLI updated"))
+    else:
+        click.echo(FeedbackManager.info(message="✓ Tinybird CLI is already up-to-date"))

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

@@ -1,4 +1,3 @@
-import subprocess
 from pathlib import Path
 
 import click
@@ -6,7 +5,7 @@ import requests
 
 from docker.client import DockerClient
 from tinybird.tb.modules.cli import cli
-from tinybird.tb.modules.exceptions import CLIException
+from tinybird.tb.modules.common import update_cli
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 from tinybird.tb.modules.local_common import (
     TB_CONTAINER_NAME,
@@ -42,33 +41,6 @@ def remove_tinybird_local(docker_client: DockerClient, persist_data: bool) -> No
         pass
 
 
-def update_cli() -> None:
-    click.echo(FeedbackManager.highlight(message="» Updating Tinybird CLI..."))
-
-    try:
-        process = subprocess.Popen(
-            ["uv", "tool", "upgrade", "tinybird"],
-            stdout=subprocess.PIPE,
-            stderr=subprocess.PIPE,
-            text=True,
-        )
-    except FileNotFoundError:
-        raise CLIException(
-            FeedbackManager.error(
-                message="Cannot find required tool: uv. Reinstall using: curl https://tinybird.co | sh"
-            )
-        )
-
-    stdout, stderr = process.communicate()
-    if "Nothing to upgrade" not in stdout + stderr:
-        for line in stdout.split("\n") + stderr.split("\n"):
-            if "Updated tinybird" in line:
-                click.echo(FeedbackManager.info(message=f"» {line}"))
-        click.echo(FeedbackManager.success(message="✓ Tinybird CLI updated"))
-    else:
-        click.echo(FeedbackManager.info(message="✓ Tinybird CLI is already up-to-date"))
-
-
 @cli.command()
 def update() -> None:
     """Update Tinybird CLI to the latest version."""

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

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