tinybird 0.0.1.dev266__py3-none-any.whl → 0.0.1.dev267__py3-none-any.whl

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/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/tb/modules/agent/banner.py

@@ -41,31 +41,14 @@ def display_banner():
 
     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)]
+    banner = [
+        "  ████████╗██╗███╗   ██╗██╗   ██╗██████╗ ██╗██████╗ ██████╗     ██████╗ ██████╗ ██████╗ ███████╗",
+        "  ╚══██╔══╝██║████╗  ██║╚██╗ ██╔╝██╔══██╗██║██╔══██╗██╔══██╗   ██╔════╝██╔═══██╗██╔══██╗██╔════╝",
+        "     ██║   ██║██╔██╗ ██║ ╚████╔╝ ██████╔╝██║██████╔╝██║  ██║   ██║     ██║   ██║██║  ██║█████╗  ",
+        "     ██║   ██║██║╚██╗██║  ╚██╔╝  ██╔══██╗██║██╔══██╗██║  ██║   ██║     ██║   ██║██║  ██║██╔══╝  ",
+        "     ██║   ██║██║ ╚████║   ██║   ██████╔╝██║██║  ██║██████╔╝   ╚██████╗╚██████╔╝██████╔╝███████╗",
+        "     ╚═╝   ╚═╝╚═╝  ╚═══╝   ╚═╝   ╚═════╝ ╚═╝╚═╝  ╚═╝╚═════╝     ╚═════╝ ╚═════╝ ╚═════╝ ╚══════╝",
+    ]
 
     def rgb_to_ansi(r: int, g: int, b: int, use_truecolor: bool):
         """Convert RGB values to ANSI escape code"""

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/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.dist-info → tinybird-0.0.1.dev267.dist-info}/METADATA

@@ -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.dist-info → tinybird-0.0.1.dev267.dist-info}/RECORD

@@ -17,7 +17,7 @@ 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=24_fIgbVB3P7UF7JqFvY_zPB1-ZShXJhWbhwh4ZB7Jk,247
+tinybird/tb/__cli__.py,sha256=V9cRl0xsyU-hm_0lGu2QorOI9Mdck-XRwVjBngDbFZ8,247
 tinybird/tb/check_pypi.py,sha256=Gp0HkHHDFMSDL6nxKlOY51z7z1Uv-2LRexNTZSHHGmM,552
 tinybird/tb/cli.py,sha256=FdDFEIayjmsZEVsVSSvRiVYn_FHOVg_zWQzchnzfWho,1008
 tinybird/tb/client.py,sha256=OVNlCXI16kpD-Ph8BDhkyKbR68TdZcR_j4ng91Iiybg,54706
@@ -25,7 +25,7 @@ tinybird/tb/config.py,sha256=mhMTGnMB5KcxGoh3dewIr2Jjsa6pHE183gCPAQWyp6o,3973
 tinybird/tb/modules/build.py,sha256=efD-vamK1NPaDo9R86Hn8be2DYoW0Hh5bZiH7knK5dk,7790
 tinybird/tb/modules/build_common.py,sha256=dthlaDn_CuwZnedQcUi7iIdDoHWfSbbbGQwiDgNcmC0,13062
 tinybird/tb/modules/cicd.py,sha256=0KLKccha9IP749QvlXBmzdWv1On3mFwMY4DUcJlBxiE,7326
-tinybird/tb/modules/cli.py,sha256=qZIp8Ez4sunQ3_lIj1QGc3yOp8-SYfNHASHpE_7Ji6Y,16978
+tinybird/tb/modules/cli.py,sha256=2sw4W3sQyhAo474Mnq3hUN2QZUpEb9cU1NYnVB0qHl0,16877
 tinybird/tb/modules/common.py,sha256=tj6DR2yOqMMQ0PILwFGXmMogxdrbQCgj36HdSM611rs,82657
 tinybird/tb/modules/config.py,sha256=gK7rgaWTDd4ZKCrNEg_Uemr26EQjqWt6TjyQKujxOws,11462
 tinybird/tb/modules/connection.py,sha256=-MY56NUAai6EMC4-wpi7bT0_nz_SA8QzTmHkV7HB1IQ,17810
@@ -33,7 +33,7 @@ tinybird/tb/modules/copy.py,sha256=dPZkcIDvxjJrlQUIvToO0vsEEEs4EYumbNV77-BzNoU,4
 tinybird/tb/modules/create.py,sha256=pJxHXG69c9Z_21s-7VuJ3RZOF_nJU51LEwiAkvI3dZY,23251
 tinybird/tb/modules/datasource.py,sha256=pae-ENeHYIF1HHYRSOziFC-2FPLUFa0KS60YpdlKCS8,41725
 tinybird/tb/modules/deployment.py,sha256=EDEVOqFk5fp0fvvs2jV0mT5POFd-i_8uZIUREwzAbEk,13199
-tinybird/tb/modules/deployment_common.py,sha256=g-2wnR63tlxmTObs6xRjZRgOOaX0fFn3Jq4x21qI6aw,18975
+tinybird/tb/modules/deployment_common.py,sha256=6Ws3YTtEyVl9Id7gyxXkEhe0GJpUJeibS-XGTtTA9DI,19076
 tinybird/tb/modules/deprecations.py,sha256=rrszC1f_JJeJ8mUxGoCxckQTJFBCR8wREf4XXXN-PRc,4507
 tinybird/tb/modules/dev_server.py,sha256=57FCKuWpErwYUYgHspYDkLWEm9F4pbvVOtMrFXX1fVU,10129
 tinybird/tb/modules/endpoint.py,sha256=ksRj6mfDb9Xv63PhTkV_uKSosgysHElqagg3RTt21Do,11958
@@ -69,9 +69,9 @@ tinybird/tb/modules/watch.py,sha256=No0bK1M1_3CYuMaIgylxf7vYFJ72lTJe3brz6xQ-mJo,
 tinybird/tb/modules/workspace.py,sha256=Q_8HcxMsNg8QG9aBlwcWS2umrDP5IkTIHqqz3sfmGuc,11341
 tinybird/tb/modules/workspace_members.py,sha256=5JdkJgfuEwbq-t6vxkBhYwgsiTDxF790wsa6Xfif9nk,8608
 tinybird/tb/modules/agent/__init__.py,sha256=i3oe3vDIWWPaicdCM0zs7D7BJ1W0k7th93ooskHAV00,54
-tinybird/tb/modules/agent/agent.py,sha256=aDMjkZUZVTW39HjsusBJkzc4TSVrD_clAQK2V-IJtE0,29020
+tinybird/tb/modules/agent/agent.py,sha256=vr2a5lSAhmJer1702gVmECUjbrjg2_MUYMlEy5o0iE4,28773
 tinybird/tb/modules/agent/animations.py,sha256=4WOC5_2BracttmMCrV0H91tXfWcUzQHBUaIJc5FA7tE,3490
-tinybird/tb/modules/agent/banner.py,sha256=IM7UtRqSoC6xFJ-TU33-1HrV6I2KVYQsVMeI4GgJsb0,6045
+tinybird/tb/modules/agent/banner.py,sha256=l6cO5Fi7lbVKp-GsBP8jf3IkjOWxg2jpAt9NBCy0WR8,4085
 tinybird/tb/modules/agent/command_agent.py,sha256=Q4M5qV9j3aETrXswoWZ02ZWQZhJm42HGHO1ymJrfDnw,2665
 tinybird/tb/modules/agent/memory.py,sha256=O6Kumn9AyKxcTkhI45yjAUZ3ZIAibLOcNWoiEuLYeqY,3245
 tinybird/tb/modules/agent/models.py,sha256=IAxqlnHy8c2OeSnDrrSp2Mg38W9_r0GsDM87Wv-YNfM,925
@@ -116,8 +116,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.dev266.dist-info/METADATA,sha256=GCalRxwS9i9RPYR0zmKe-vYpxFoMzjG7JoiUmWzv4aY,1733
-tinybird-0.0.1.dev266.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev266.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev266.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev266.dist-info/RECORD,,
+tinybird-0.0.1.dev267.dist-info/METADATA,sha256=EzrJCbpmOzBD7DT9HjC9ggTCcuXx-mSoyrJodo0JTRk,1733
+tinybird-0.0.1.dev267.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev267.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev267.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev267.dist-info/RECORD,,