tinybird 0.0.1.dev266__tar.gz → 0.0.1.dev267__tar.gz

{tinybird-0.0.1.dev266 → tinybird-0.0.1.dev267}/PKG-INFO

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

{tinybird-0.0.1.dev266 → tinybird-0.0.1.dev267}/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.dev266'
-__revision__ = 'b4193e9'
+__version__ = '0.0.1.dev267'
+__revision__ = '8c3d4f6'

{tinybird-0.0.1.dev266 → tinybird-0.0.1.dev267}/tinybird/tb/modules/agent/agent.py

@@ -11,7 +11,6 @@ from typing import Any, Optional
 import click
 import humanfriendly
 from pydantic_ai import Agent, RunContext, Tool
-from pydantic_ai.agent import AgentRunResult
 from pydantic_ai.messages import ModelMessage, ModelRequest, UserPromptPart
 from requests import Response
 
@@ -255,7 +254,7 @@ class TinybirdAgent:
         save_messages(new_messages)
         self.thinking_animation.stop()
         click.echo(result.output)
-        self._echo_usage(config, result)
+        self.echo_usage(config)
 
     async def run_iter(self, user_prompt: str, config: dict[str, Any], model: Any) -> None:
         user_prompt = f"{user_prompt}\n\n{load_custom_project_rules(self.project.folder)}"
@@ -279,31 +278,28 @@ class TinybirdAgent:
             self.messages.extend(new_messages)
             save_messages(new_messages)
             self.thinking_animation.stop()
-            self._echo_usage(config, agent_run.result)
+            self.echo_usage(config)
 
-    def _echo_usage(self, config: dict[str, Any], result: AgentRunResult) -> None:
+    def echo_usage(self, config: dict[str, Any]) -> None:
         try:
             client = _get_tb_client(config["user_token"], config["host"])
             workspace_id = config.get("id", "")
             workspace = client.workspace(workspace_id, with_organization=True, version="v1")
             limits_data = client.organization_limits(workspace["organization"]["id"])
             ai_requests_limits = limits_data.get("limits", {}).get("ai_requests", {})
-            current_ai_requests = ai_requests_limits.get("quantity", 0)
-            max_ai_requests = ai_requests_limits.get("max", 0)
+            current_ai_requests = ai_requests_limits.get("quantity") or 0
+            max_ai_requests = ai_requests_limits.get("max") or 0
+            remaining_requests = max(max_ai_requests - current_ai_requests, 0)
+            current_ai_requests = min(max_ai_requests, current_ai_requests)
             if not max_ai_requests:
                 return
-            remaining_requests = max(max_ai_requests - current_ai_requests, 0)
             warning_threshold = max_ai_requests * 0.8
-            if current_ai_requests >= warning_threshold:
-                message_color = FeedbackManager.warning
-            else:
-                message_color = FeedbackManager.gray
-
-            current_ai_requests = min(max_ai_requests, current_ai_requests)
-
+            message_color = (
+                FeedbackManager.warning if current_ai_requests >= warning_threshold else FeedbackManager.gray
+            )
             click.echo(
                 message_color(
-                    message=f"{remaining_requests} agent requests left ({current_ai_requests}/{max_ai_requests}). This message is informative. Limits will be enforced soon."
+                    message=f"{remaining_requests} requests left ({current_ai_requests}/{max_ai_requests}). You can continue using Tinybird Code. Limits will be enforced soon."
                 )
             )
         except Exception:
@@ -339,7 +335,7 @@ def run_agent(
                 workspace_id = cli_config.get("id", "")
                 workspace_name = cli_config.get("name", "")
 
-            if not token or not host or not user_token:
+            if not token or not host or not user_token or not workspace_id:
                 click.echo(
                     FeedbackManager.error(message="Tinybird Code requires authentication. Run 'tb login' first.")
                 )
@@ -383,8 +379,17 @@ def run_agent(
 
         # Interactive mode: show banner and enter interactive loop
         display_banner()
-        click.echo(FeedbackManager.info(message="Describe what you want to create and I'll help you build it"))
-        click.echo(FeedbackManager.info(message="Run /help for more commands"))
+        click.echo(
+            FeedbackManager.info(
+                message="""Tips for getting started:
+- Describe what you want to build or ask for specific resources.
+- Run tb commands directly without leaving interactive mode.
+- Create a TINYBIRD.md file to customize your interactions.
+"""
+            )
+        )
+        agent.echo_usage(config)
+        click.echo()
 
     except Exception as e:
         click.echo(FeedbackManager.error(message=f"Failed to initialize agent: {e}"))
@@ -431,11 +436,7 @@ def run_agent(
 
                     continue
                 elif user_input.lower() == "/help":
-                    click.echo("• Describe what you want to create: 'Create a user analytics system'")
-                    click.echo("• Ask for specific resources: 'Create a pipe to aggregate daily clicks'")
-                    click.echo("• Connect to external services: 'Set up a Kafka connection for events'")
-                    click.echo("• Type '/exit' or '/quit' to leave")
-
+                    subprocess.run(["tb", "--help"], check=True)
                     continue
                 elif user_input.strip() == "":
                     continue

tinybird-0.0.1.dev267/tinybird/tb/modules/agent/banner.py

@@ -0,0 +1,87 @@
+import os
+import sys
+
+import click
+
+
+def detect_terminal_capabilities():
+    """Detect terminal color and Unicode capabilities"""
+    # Check for true color support
+    colorterm = os.environ.get("COLORTERM", "").lower()
+    term = os.environ.get("TERM", "").lower()
+    term_program = os.environ.get("TERM_PROGRAM", "").lower()
+
+    # Known terminals with good true color support
+    modern_terminals = ["warp", "ghostty", "iterm2", "alacritty", "kitty", "hyper"]
+
+    # Check for true color support
+    has_truecolor = (
+        colorterm in ["truecolor", "24bit"]
+        or term_program in modern_terminals
+        or "truecolor" in term
+        or "24bit" in term
+    )
+
+    # Check if it's standard macOS Terminal
+    is_macos_terminal = term_program == "apple_terminal"
+
+    # Check for Unicode support (most modern terminals support this)
+    has_unicode = sys.stdout.encoding and "utf" in sys.stdout.encoding.lower()
+
+    return {
+        "truecolor": has_truecolor and not is_macos_terminal,
+        "unicode": has_unicode,
+        "is_macos_terminal": is_macos_terminal,
+    }
+
+
+def display_banner():
+    reset = "\033[0m"
+    capabilities = detect_terminal_capabilities()
+
+    click.echo("\n")
+
+    banner = [
+        "  ████████╗██╗███╗   ██╗██╗   ██╗██████╗ ██╗██████╗ ██████╗     ██████╗ ██████╗ ██████╗ ███████╗",
+        "  ╚══██╔══╝██║████╗  ██║╚██╗ ██╔╝██╔══██╗██║██╔══██╗██╔══██╗   ██╔════╝██╔═══██╗██╔══██╗██╔════╝",
+        "     ██║   ██║██╔██╗ ██║ ╚████╔╝ ██████╔╝██║██████╔╝██║  ██║   ██║     ██║   ██║██║  ██║█████╗  ",
+        "     ██║   ██║██║╚██╗██║  ╚██╔╝  ██╔══██╗██║██╔══██╗██║  ██║   ██║     ██║   ██║██║  ██║██╔══╝  ",
+        "     ██║   ██║██║ ╚████║   ██║   ██████╔╝██║██║  ██║██████╔╝   ╚██████╗╚██████╔╝██████╔╝███████╗",
+        "     ╚═╝   ╚═╝╚═╝  ╚═══╝   ╚═╝   ╚═════╝ ╚═╝╚═╝  ╚═╝╚═════╝     ╚═════╝ ╚═════╝ ╚═════╝ ╚══════╝",
+    ]
+
+    def rgb_to_ansi(r: int, g: int, b: int, use_truecolor: bool):
+        """Convert RGB values to ANSI escape code"""
+        if use_truecolor:
+            return f"\033[38;2;{r};{g};{b}m"
+        else:
+            # Convert to 8-bit color (256 color palette)
+            # Simple approximation: map RGB to 216-color cube + grayscale
+            if r == g == b:
+                # Grayscale
+                gray = int(r / 255 * 23) + 232
+                return f"\033[38;5;{gray}m"
+            else:
+                # Color cube (6x6x6)
+                r_idx = int(r / 255 * 5)
+                g_idx = int(g / 255 * 5)
+                b_idx = int(b / 255 * 5)
+                color_idx = 16 + (36 * r_idx) + (6 * g_idx) + b_idx
+                return f"\033[38;5;{color_idx}m"
+
+    # Define solid color (corresponding to #27f795)
+    solid_color = [39, 247, 149]  # #27f795 in RGB
+
+    # Print each line with solid color for all terminals
+    for line in banner:
+        colored_line = ""
+        color_code = rgb_to_ansi(*solid_color, use_truecolor=capabilities["truecolor"])  # type: ignore
+
+        for char in line:
+            if char == " ":
+                colored_line += char
+            else:
+                colored_line += f"{color_code}{char}"
+
+        click.echo(colored_line + reset)
+    click.echo()

{tinybird-0.0.1.dev266 → tinybird-0.0.1.dev267}/tinybird/tb/modules/cli.py

@@ -47,8 +47,6 @@ 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,
@@ -56,7 +54,7 @@ agent_mode_flag = os.environ.get("TB_AGENT_MODE", "false") == "true"
         "help_option_names": ["-h", "--help"],
         "max_content_width": shutil.get_terminal_size().columns - 10,
     },
-    invoke_without_command=agent_mode_flag,
+    invoke_without_command=True,
 )
 @click.option(
     "--debug/--no-debug",
@@ -208,7 +206,7 @@ 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
+    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.dev266 → tinybird-0.0.1.dev267}/tinybird/tb/modules/deployment_common.py

@@ -254,9 +254,12 @@ def create_deployment(
             elif verbose and f.get("level", "").upper() == "INFO":
                 feedback_func = FeedbackManager.info
                 feedback_icon = ""
+            else:
+                feedback_func = None
             resource = f.get("resource")
             resource_bit = f"{resource}: " if resource else ""
-            click.echo(feedback_func(message=f"{feedback_icon}{f.get('level')}: {resource_bit}{f.get('message')}"))
+            if feedback_func is not None:
+                click.echo(feedback_func(message=f"{feedback_icon}{f.get('level')}: {resource_bit}{f.get('message')}"))
 
         deploy_errors = deployment.get("errors")
         for deploy_error in deploy_errors:

{tinybird-0.0.1.dev266 → tinybird-0.0.1.dev267}/tinybird.egg-info/PKG-INFO

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

tinybird-0.0.1.dev266/tinybird/tb/modules/agent/banner.py

@@ -1,104 +0,0 @@
-import os
-import sys
-
-import click
-
-
-def detect_terminal_capabilities():
-    """Detect terminal color and Unicode capabilities"""
-    # Check for true color support
-    colorterm = os.environ.get("COLORTERM", "").lower()
-    term = os.environ.get("TERM", "").lower()
-    term_program = os.environ.get("TERM_PROGRAM", "").lower()
-
-    # Known terminals with good true color support
-    modern_terminals = ["warp", "ghostty", "iterm2", "alacritty", "kitty", "hyper"]
-
-    # Check for true color support
-    has_truecolor = (
-        colorterm in ["truecolor", "24bit"]
-        or term_program in modern_terminals
-        or "truecolor" in term
-        or "24bit" in term
-    )
-
-    # Check if it's standard macOS Terminal
-    is_macos_terminal = term_program == "apple_terminal"
-
-    # Check for Unicode support (most modern terminals support this)
-    has_unicode = sys.stdout.encoding and "utf" in sys.stdout.encoding.lower()
-
-    return {
-        "truecolor": has_truecolor and not is_macos_terminal,
-        "unicode": has_unicode,
-        "is_macos_terminal": is_macos_terminal,
-    }
-
-
-def display_banner():
-    reset = "\033[0m"
-    capabilities = detect_terminal_capabilities()
-
-    click.echo("\n")
-
-    # Choose banner based on Unicode support
-    if capabilities["unicode"]:
-        # Unicode box-drawing characters banner
-        banner = [
-            "  ████████╗██╗███╗   ██╗██╗   ██╗██████╗ ██╗██████╗ ██████╗     ██████╗ ██████╗ ██████╗ ███████╗",
-            "  ╚══██╔══╝██║████╗  ██║╚██╗ ██╔╝██╔══██╗██║██╔══██╗██╔══██╗   ██╔════╝██╔═══██╗██╔══██╗██╔════╝",
-            "     ██║   ██║██╔██╗ ██║ ╚████╔╝ ██████╔╝██║██████╔╝██║  ██║   ██║     ██║   ██║██║  ██║█████╗  ",
-            "     ██║   ██║██║╚██╗██║  ╚██╔╝  ██╔══██╗██║██╔══██╗██║  ██║   ██║     ██║   ██║██║  ██║██╔══╝  ",
-            "     ██║   ██║██║ ╚████║   ██║   ██████╔╝██║██║  ██║██████╔╝   ╚██████╗╚██████╔╝██████╔╝███████╗",
-            "     ╚═╝   ╚═╝╚═╝  ╚═══╝   ╚═╝   ╚═════╝ ╚═╝╚═╝  ╚═╝╚═════╝     ╚═════╝ ╚═════╝ ╚═════╝ ╚══════╝",
-        ]
-    else:
-        # ASCII fallback banner
-        banner = [
-            "  ████████T██I███N   ██N██   ██Y██████B ██I██████B ██████B     ██████C ██████O ██████D ███████E",
-            "  ╚══██╔══╝██║████╗  ██║╚██╗ ██╔╝██╔══██╗██║██╔══██╗██╔══██╗   ██╔════╝██╔═══██╗██╔══██╗██╔════╝",
-            "     ██║   ██║██╔██╗ ██║ ╚████╔╝ ██████╔╝██║██████╔╝██║  ██║   ██║     ██║   ██║██║  ██║█████╗  ",
-            "     ██║   ██║██║╚██╗██║  ╚██╔╝  ██╔══██╗██║██╔══██╗██║  ██║   ██║     ██║   ██║██║  ██║██╔══╝  ",
-            "     ██║   ██║██║ ╚████║   ██║   ██████╔╝██║██║  ██║██████╔╝   ╚██████╗╚██████╔╝██████╔╝███████╗",
-            "     ╚═╝   ╚═╝╚═╝  ╚═══╝   ╚═╝   ╚═════╝ ╚═╝╚═╝  ╚═╝╚═════╝     ╚═════╝ ╚═════╝ ╚═════╝ ╚══════╝",
-        ]
-
-    def interpolate_color(start_rgb, end_rgb, factor):
-        """Interpolate between two RGB colors"""
-        return [int(start_rgb[i] + (end_rgb[i] - start_rgb[i]) * factor) for i in range(3)]
-
-    def rgb_to_ansi(r: int, g: int, b: int, use_truecolor: bool):
-        """Convert RGB values to ANSI escape code"""
-        if use_truecolor:
-            return f"\033[38;2;{r};{g};{b}m"
-        else:
-            # Convert to 8-bit color (256 color palette)
-            # Simple approximation: map RGB to 216-color cube + grayscale
-            if r == g == b:
-                # Grayscale
-                gray = int(r / 255 * 23) + 232
-                return f"\033[38;5;{gray}m"
-            else:
-                # Color cube (6x6x6)
-                r_idx = int(r / 255 * 5)
-                g_idx = int(g / 255 * 5)
-                b_idx = int(b / 255 * 5)
-                color_idx = 16 + (36 * r_idx) + (6 * g_idx) + b_idx
-                return f"\033[38;5;{color_idx}m"
-
-    # Define solid color (corresponding to #27f795)
-    solid_color = [39, 247, 149]  # #27f795 in RGB
-
-    # Print each line with solid color for all terminals
-    for line in banner:
-        colored_line = ""
-        color_code = rgb_to_ansi(*solid_color, use_truecolor=capabilities["truecolor"])  # type: ignore
-
-        for char in line:
-            if char == " ":
-                colored_line += char
-            else:
-                colored_line += f"{color_code}{char}"
-
-        click.echo(colored_line + reset)
-    click.echo()