tinybird 0.0.1.dev237__tar.gz → 0.0.1.dev239__tar.gz

{tinybird-0.0.1.dev237 → tinybird-0.0.1.dev239}/PKG-INFO

@@ -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 → tinybird-0.0.1.dev239}/tinybird/sql_template.py

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

{tinybird-0.0.1.dev237 → tinybird-0.0.1.dev239}/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-0.0.1.dev237 → tinybird-0.0.1.dev239}/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-0.0.1.dev237 → tinybird-0.0.1.dev239}/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-0.0.1.dev239/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-0.0.1.dev237 → tinybird-0.0.1.dev239}/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-0.0.1.dev237 → tinybird-0.0.1.dev239}/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 → tinybird-0.0.1.dev239}/tinybird.egg-info/PKG-INFO

@@ -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 → tinybird-0.0.1.dev239}/tinybird.egg-info/SOURCES.txt

@@ -33,6 +33,7 @@ tinybird/tb/check_pypi.py
 tinybird/tb/cli.py
 tinybird/tb/client.py
 tinybird/tb/config.py
+tinybird/tb/modules/agent.py
 tinybird/tb/modules/build.py
 tinybird/tb/modules/cicd.py
 tinybird/tb/modules/cli.py