tinybird 0.0.1.dev237__py3-none-any.whl → 0.0.1.dev239__py3-none-any.whl

tinybird/sql_template.py

@@ -1388,6 +1388,7 @@ _namespace = {
     "close": None,
     "print": None,
     "input": None,
+    "id": None,
 }
 
 

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.dev237'
-__revision__ = '7aa96d2'
+__version__ = '0.0.1.dev239'
+__revision__ = '960f37b'

tinybird/tb/cli.py

@@ -1,3 +1,4 @@
+import tinybird.tb.modules.agent
 import tinybird.tb.modules.build
 import tinybird.tb.modules.cli
 import tinybird.tb.modules.common

tinybird/tb/client.py

@@ -1394,3 +1394,7 @@ class TinyB:
 
     def delete_tag(self, name: str):
         self._req(f"/v0/tags/{name}", method="DELETE")
+
+    def explore_data(self, prompt: str) -> str:
+        params = urlencode({"prompt": prompt, "host": self.host, "origin": "cli"})
+        return self._req(f"/v1/agents/explore?{params}")

tinybird/tb/modules/agent.py

@@ -0,0 +1,80 @@
+import os
+import sys
+from typing import Any, Dict, Optional
+
+import click
+from prompt_toolkit import PromptSession
+from prompt_toolkit.history import FileHistory
+from prompt_toolkit.styles import Style
+
+from tinybird.tb.client import TinyB
+from tinybird.tb.modules.common import _get_tb_client
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+
+
+def agent_banner():
+    # Define gradient colors (teal/turquoise range)
+    colors = [
+        "\033[38;2;0;128;128m",  # Teal
+        "\033[38;2;0;150;136m",  # Teal-ish
+        "\033[38;2;20;160;145m",  # Turquoise blend
+        "\033[38;2;40;170;155m",  # Light turquoise
+        "\033[38;2;60;180;165m",  # Lighter turquoise
+        "\033[38;2;80;190;175m",  # Very light turquoise
+    ]
+    reset = "\033[0m"
+
+    # The Tinybird Code ASCII art banner
+    banner = [
+        "  ████████╗██╗███╗   ██╗██╗   ██╗██████╗ ██╗██████╗ ██████╗     ██████╗ ██████╗ ██████╗ ███████╗",
+        "  ╚══██╔══╝██║████╗  ██║╚██╗ ██╔╝██╔══██╗██║██╔══██╗██╔══██╗   ██╔════╝██╔═══██╗██╔══██╗██╔════╝",
+        "     ██║   ██║██╔██╗ ██║ ╚████╔╝ ██████╔╝██║██████╔╝██║  ██║   ██║     ██║   ██║██║  ██║█████╗  ",
+        "     ██║   ██║██║╚██╗██║  ╚██╔╝  ██╔══██╗██║██╔══██╗██║  ██║   ██║     ██║   ██║██║  ██║██╔══╝  ",
+        "     ██║   ██║██║ ╚████║   ██║   ██████╔╝██║██║  ██║██████╔╝   ╚██████╗╚██████╔╝██████╔╝███████╗",
+        "     ╚═╝   ╚═╝╚═╝  ╚═══╝   ╚═╝   ╚═════╝ ╚═╝╚═╝  ╚═╝╚═════╝     ╚═════╝ ╚═════╝ ╚═════╝ ╚══════╝",
+    ]
+
+    # Print each line with a smooth horizontal gradient
+    for line in banner:
+        colored_line = ""
+        for j, char in enumerate(line):
+            # Skip coloring spaces
+            if char == " ":
+                colored_line += char
+                continue
+
+            # Calculate color index for a smooth gradient
+            color_index = min(int(j * len(colors) / len(line)), len(colors) - 1)
+            colored_line += f"{colors[color_index]}{char}"
+
+        click.echo(colored_line + reset)
+
+
+def explore_data(client: TinyB, prompt: str):
+    click.echo(FeedbackManager.highlight(message="\nExploring data...\n"))
+    result = client.explore_data(prompt)
+    click.echo(result)
+
+
+def run_agent_shell(config: Dict[str, Any]):
+    style = Style.from_dict({"prompt": "fg:#34D399 bold"})
+    history: Optional[FileHistory] = None
+    try:
+        history_file = os.path.expanduser("~/.tb_agent_history")
+        history = FileHistory(history_file)
+    except Exception:
+        pass
+    workspace_name = config.get("name", "No workspace found")
+    session: PromptSession = PromptSession(history=history)
+    user_input = session.prompt([("class:prompt", f"\ntb ({workspace_name}) » ")], style=style)
+    if user_input == "exit":
+        sys.exit(0)
+    else:
+        client = _get_tb_client(config.get("token", None), config["host"])
+        explore_data(client, user_input)
+        return run_agent_shell(config)
+
+
+def run_agent(config: Dict[str, Any]):
+    agent_banner()
+    run_agent_shell(config)

tinybird/tb/modules/cli.py

@@ -22,6 +22,7 @@ from tinybird.tb.client import (
     AuthException,
     AuthNoTokenException,
 )
+from tinybird.tb.modules.agent import run_agent
 from tinybird.tb.modules.common import (
     CatchAuthExceptions,
     CLIException,
@@ -46,6 +47,8 @@ DEFAULT_PATTERNS: List[Tuple[str, Union[str, Callable[[str], str]]]] = [
 ]
 VERSION = f"{__cli__.__version__} (rev {__cli__.__revision__})"
 
+agent_mode_flag = os.environ.get("TB_AGENT_MODE", "false") == "true"
+
 
 @click.group(
     cls=CatchAuthExceptions,
@@ -53,6 +56,7 @@ VERSION = f"{__cli__.__version__} (rev {__cli__.__revision__})"
         "help_option_names": ["-h", "--help"],
         "max_content_width": shutil.get_terminal_size().columns - 10,
     },
+    invoke_without_command=agent_mode_flag,
 )
 @click.option(
     "--debug/--no-debug",
@@ -93,6 +97,7 @@ def cli(
     """
     Use `OBFUSCATE_REGEX_PATTERN` and `OBFUSCATE_PATTERN_SEPARATOR` environment variables to define a regex pattern and a separator (in case of a single string with multiple regex) to obfuscate secrets in the CLI output.
     """
+
     # We need to unpatch for our tests not to break
     if output != "human":
         __hide_click_output()
@@ -188,6 +193,11 @@ def cli(
     ctx.ensure_object(dict)["env"] = get_target_env(cloud)
     ctx.ensure_object(dict)["output"] = output
 
+    is_agent_mode = agent_mode_flag and ctx.invoked_subcommand is None
+
+    if is_agent_mode:
+        run_agent(config)
+
 
 @cli.command(hidden=True)
 @click.option("-f", "--force", is_flag=True, default=False, help="Override existing files")
@@ -365,7 +375,7 @@ def create_ctx_client(ctx: Context, config: Dict[str, Any], cloud: bool, staging
         "init",
     ]
     command = ctx.invoked_subcommand
-    if command in commands_without_ctx_client:
+    if not command or command in commands_without_ctx_client:
         return None
 
     commands_always_cloud = ["pull", "infra"]

tinybird/tb/modules/local.py

@@ -1,4 +1,5 @@
 import subprocess
+from pathlib import Path
 
 import click
 import requests
@@ -148,6 +149,11 @@ def remove() -> None:
 )
 def start(use_aws_creds: bool, volumes_path: str) -> None:
     """Start Tinybird Local"""
+    if volumes_path is not None:
+        absolute_path = Path(volumes_path).absolute()
+        absolute_path.mkdir(parents=True, exist_ok=True)
+        volumes_path = str(absolute_path)
+
     click.echo(FeedbackManager.highlight(message="» Starting Tinybird Local..."))
     docker_client = get_docker_client()
     start_tinybird_local(docker_client, use_aws_creds, volumes_path)
@@ -168,6 +174,11 @@ def start(use_aws_creds: bool, volumes_path: str) -> None:
 )
 def restart(use_aws_creds: bool, volumes_path: str) -> None:
     """Restart Tinybird Local"""
+    if volumes_path is not None:
+        absolute_path = Path(volumes_path).absolute()
+        absolute_path.mkdir(parents=True, exist_ok=True)
+        volumes_path = str(absolute_path)
+
     click.echo(FeedbackManager.highlight(message="» Restarting Tinybird Local..."))
     docker_client = get_docker_client()
     remove_tinybird_local(docker_client, volumes_path is not None)

{tinybird-0.0.1.dev237.dist-info → tinybird-0.0.1.dev239.dist-info}/METADATA

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

{tinybird-0.0.1.dev237.dist-info → tinybird-0.0.1.dev239.dist-info}/RECORD

@@ -5,7 +5,7 @@ tinybird/feedback_manager.py,sha256=1INQFfRfuMCb9lfB8KNf4r6qC2khW568hoHjtk-wshI,
 tinybird/git_settings.py,sha256=Sw_8rGmribEFJ4Z_6idrVytxpFYk7ez8ei0qHULzs3E,3934
 tinybird/prompts.py,sha256=dg8Q8Q29X69j-PPP8KGooMz7-M3FNXEYqC0Fj0LcUbw,45316
 tinybird/sql.py,sha256=BufnOgclQokDyihtuXesOwHBsebN6wRXIxO5wKRkOwE,48299
-tinybird/sql_template.py,sha256=WjsTBjpQLVBHGZbY2dZuhZUurFR-rbJ_KRRy5vx4Y5E,99967
+tinybird/sql_template.py,sha256=LChDztXUUrNO4Qukv2RMsdjQ-vhmepWiHVoX6yr140E,99983
 tinybird/sql_template_fmt.py,sha256=KUHdj5rYCYm_rKKdXYSJAE9vIyXUQLB0YSZnUXHeBlY,10196
 tinybird/sql_toolset.py,sha256=M2rpLYkgV2W8NnYEYPC1tJdpy4uZHXVF64NBSKLQka4,19549
 tinybird/syncasync.py,sha256=IPnOx6lMbf9SNddN1eBtssg8vCLHMt76SuZ6YNYm-Yk,27761
@@ -17,14 +17,15 @@ tinybird/datafile/exceptions.py,sha256=8rw2umdZjtby85QbuRKFO5ETz_eRHwUY5l7eHsy1w
 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=vmMQxvQ7ux44Mfpu0EDdpMaLEOM4WmWMZ3dbG8YeI5s,247
+tinybird/tb/__cli__.py,sha256=8mfKuN5jmJ9hrzhAZqnZA9ASZ0QXkYRjiGxO_zKzPhM,247
 tinybird/tb/check_pypi.py,sha256=Gp0HkHHDFMSDL6nxKlOY51z7z1Uv-2LRexNTZSHHGmM,552
-tinybird/tb/cli.py,sha256=_5hIbxDFdLjMq0GiXpdoMIz7TdTpQ0DHfAS9MKdlST4,975
-tinybird/tb/client.py,sha256=cHJky5SF-z_epBajNovo_gaIRdMsU5KXx7vtVl4sdgE,53957
+tinybird/tb/cli.py,sha256=FdDFEIayjmsZEVsVSSvRiVYn_FHOVg_zWQzchnzfWho,1008
+tinybird/tb/client.py,sha256=pJbdkWMXGAqKseNAvdsRRnl_c7I-DCMB0dWCQnG82nU,54146
 tinybird/tb/config.py,sha256=mhMTGnMB5KcxGoh3dewIr2Jjsa6pHE183gCPAQWyp6o,3973
+tinybird/tb/modules/agent.py,sha256=RE_QlGSq6MByEaOBUbr1dSr3X2bi09-QvzoxpxQtyPs,3900
 tinybird/tb/modules/build.py,sha256=MxGQY-KII4lYut-yOWGBIHFpg-m_cfHgpE7lwOzZCeQ,19871
 tinybird/tb/modules/cicd.py,sha256=0KLKccha9IP749QvlXBmzdWv1On3mFwMY4DUcJlBxiE,7326
-tinybird/tb/modules/cli.py,sha256=E91z3aw4Yziu93AxCjJyBfWnvh3tZk1zxkzRG1s5EhY,16036
+tinybird/tb/modules/cli.py,sha256=SgXaNjyX1ba3Uvmjpc-ViG1NLbscw394mH1vBj-A2JI,16335
 tinybird/tb/modules/common.py,sha256=jTTaDDHrZREt--032XhP6GkbfFwC79YJ5aH1Sl7bmbo,81925
 tinybird/tb/modules/config.py,sha256=gK7rgaWTDd4ZKCrNEg_Uemr26EQjqWt6TjyQKujxOws,11462
 tinybird/tb/modules/connection.py,sha256=-MY56NUAai6EMC4-wpi7bT0_nz_SA8QzTmHkV7HB1IQ,17810
@@ -42,7 +43,7 @@ tinybird/tb/modules/infra.py,sha256=JE9oLIyF4bi_JBoe-BgZ5HhKp_lQgSihuSV1KIS02Qs,
 tinybird/tb/modules/job.py,sha256=wBsnu8UPTOha2rkLvucgmw4xYv73ubmui3eeSIF68ZM,3107
 tinybird/tb/modules/llm.py,sha256=QbHRcMLgFmLKEh4zVb2ctR_5tIGUGdFJrAiRCDtMxDw,1572
 tinybird/tb/modules/llm_utils.py,sha256=nS9r4FAElJw8yXtmdYrx-rtI2zXR8qXfi1QqUDCfxvg,3469
-tinybird/tb/modules/local.py,sha256=HTLUFb8xqR8_PEFkvZV1UB-uk2L66_DAh6eflce6_FE,6446
+tinybird/tb/modules/local.py,sha256=tpiw_F_qOIp42h3kTBwTm5GQDyuVLF0QNF1jmB0zR94,6845
 tinybird/tb/modules/local_common.py,sha256=_WODjW3oPshgsZ1jDFFx2nr0zrLi3Gxz5dlahWPobM8,17464
 tinybird/tb/modules/login.py,sha256=glqj5RWH26AseEoBl8XfrSDEjQTdko17i_pVWOIMoGc,12497
 tinybird/tb/modules/logout.py,sha256=sniI4JNxpTrVeRCp0oGJuQ3yRerG4hH5uz6oBmjv724,1009
@@ -82,8 +83,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.dev237.dist-info/METADATA,sha256=Wx48Mkja_8CCOe1o8FJFVlz9M36gjsZHPKZK86YVQDo,1682
-tinybird-0.0.1.dev237.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev237.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev237.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev237.dist-info/RECORD,,
+tinybird-0.0.1.dev239.dist-info/METADATA,sha256=NtL4GLVWnmL5-eQn2I6DCzgQqGODeBofbbGVyPXOt90,1682
+tinybird-0.0.1.dev239.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev239.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev239.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev239.dist-info/RECORD,,