tinybird 0.0.1.dev241__py3-none-any.whl → 0.0.1.dev242__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.dev241'
-__revision__ = 'b44ab3b'
+__version__ = '0.0.1.dev242'
+__revision__ = '62ad714'

tinybird/tb/modules/agent/agent.py

@@ -22,6 +22,7 @@ from tinybird.prompts import (
     sink_pipe_instructions,
 )
 from tinybird.tb.client import TinyB
+from tinybird.tb.modules.agent.animations import ThinkingAnimation
 from tinybird.tb.modules.agent.banner import display_banner
 from tinybird.tb.modules.agent.memory import clear_history, load_history
 from tinybird.tb.modules.agent.models import create_model
@@ -125,7 +126,7 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
                 Tool(explore_data, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
                 Tool(preview_datafile, docstring_format="google", require_parameter_descriptions=True, takes_ctx=False),
                 Tool(create_datafile, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
-                Tool(plan, docstring_format="google", require_parameter_descriptions=True, takes_ctx=False),
+                Tool(plan, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
             ],
         )
 
@@ -135,18 +136,24 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
 
     def run(self, user_prompt: str) -> None:
         client = TinyB(token=self.token, host=self.host)
+
+        thinking_animation = ThinkingAnimation(message="Chirping", delay=0.15)
+        thinking_animation.start()
+
         result = self.agent.run_sync(
             user_prompt,
             deps=TinybirdAgentContext(
                 # context does not support the whole client, so we need to pass only the functions we need
                 explore_data=client.explore_data,
                 folder=self.folder,
+                thinking_animation=thinking_animation,
             ),
             message_history=self.messages,
         )
         new_messages = result.new_messages()
         self.messages.extend(new_messages)
-        click.echo("\n")
+        thinking_animation.stop()
+
         click.echo(result.output)
         click.echo("\n")
 

tinybird/tb/modules/agent/animations.py

@@ -0,0 +1,86 @@
+import random
+import sys
+import threading
+from time import sleep
+from typing import Optional
+
+import click
+
+
+class ThinkingAnimation:
+    """Thinking animation that shows changing sparkles as a prefix."""
+
+    def __init__(
+        self,
+        message: str = "Thinking",
+        delay: float = 0.15,
+        colors: bool = True,
+        dots: bool = True,
+    ):
+        self.message = message
+        self.delay = delay
+        self.running = False
+        self.thread: Optional[threading.Thread] = None
+        self.colors = colors and sys.stdout.isatty()
+        self.dots = dots
+
+        # Default sparkle characters
+        self.sparkle_chars = ["✧", "✦", "⋆", "✳", "✺", "✹", "*", "·"]
+
+        # ANSI color codes
+        self.colors_list = [
+            "\033[33m",  # Yellow
+            "\033[36m",  # Cyan
+            "\033[35m",  # Magenta
+            "\033[93m",  # Bright Yellow
+            "\033[96m",  # Bright Cyan
+        ]
+        self.reset_color = "\033[0m"
+
+    def start(self):
+        """Start the animation in a separate thread."""
+        self.running = True
+        self.thread = threading.Thread(target=self._run_animation)
+        self.thread.daemon = True
+
+        click.echo("\n")
+        self.thread.start()
+
+    def stop(self):
+        """Stop the animation."""
+        self.running = False
+        if self.thread:
+            self.thread.join()
+        # Clear the line and reset cursor position
+        sys.stdout.write("\r" + " " * (len(self.message) + 10) + "\r")
+        sys.stdout.flush()
+
+    def _run_animation(self):
+        """Run the animation until stopped."""
+        frame_count = 0
+        dots_count = 0
+
+        while self.running:
+            # Choose a random sparkle for this frame
+            sparkle = random.choice(self.sparkle_chars)
+
+            if self.colors:
+                color = random.choice(self.colors_list)
+                colored_sparkle = f"{color}{sparkle}{self.reset_color}"
+            else:
+                colored_sparkle = sparkle
+
+            # Handle dots animation if enabled
+            if self.dots:
+                dots_count = (frame_count // 4) % 4  # Change dots every 4 frames
+                dots = "." * dots_count
+                display_message = f"{self.message}{dots}"
+            else:
+                display_message = self.message
+
+            # Print the message with the prefix sparkle
+            sys.stdout.write(f"\r{colored_sparkle} {display_message}")
+            sys.stdout.flush()
+
+            sleep(self.delay)
+            frame_count += 1

tinybird/tb/modules/agent/tools/create_datafile.py

@@ -35,9 +35,11 @@ def create_datafile(ctx: RunContext[TinybirdAgentContext], resource: Datafile) -
         str: If the resource was created or not.
     """
     try:
+        ctx.deps.thinking_animation.stop()
         click.echo()
         click.echo(resource.content)
         confirmation = get_resource_confirmation(resource)
+        ctx.deps.thinking_animation.start()
 
         if not confirmation:
             return f"Resource {resource.pathname} was not created. User cancelled creation."

tinybird/tb/modules/agent/tools/plan.py

@@ -1,6 +1,7 @@
 import click
+from pydantic_ai import RunContext
 
-from tinybird.tb.modules.agent.utils import show_options
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_options
 
 
 def get_plan_confirmation() -> bool:
@@ -22,7 +23,7 @@ def get_plan_confirmation() -> bool:
         return False
 
 
-def plan(plan: str) -> str:
+def plan(ctx: RunContext[TinybirdAgentContext], plan: str) -> str:
     """Given a plan, ask the user for confirmation to implement it
 
     Args:
@@ -32,9 +33,11 @@ def plan(plan: str) -> str:
         str: If the plan was implemented or not.
     """
     try:
+        ctx.deps.thinking_animation.stop()
         click.echo()
         click.echo(plan)
         confirmation = get_plan_confirmation()
+        ctx.deps.thinking_animation.start()
 
         if not confirmation:
             return "Plan was not implemented. User cancelled implementation."

tinybird/tb/modules/agent/utils.py

@@ -20,6 +20,7 @@ from pydantic import BaseModel, Field
 class TinybirdAgentContext(BaseModel):
     explore_data: Callable[[str], str]
     folder: str
+    thinking_animation: Any
 
 
 default_style = Style.from_dict(

{tinybird-0.0.1.dev241.dist-info → tinybird-0.0.1.dev242.dist-info}/METADATA

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

{tinybird-0.0.1.dev241.dist-info → tinybird-0.0.1.dev242.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=QWlsNI8ondzuZlYrE_y6bh1Cfz6_5AfBIndh4z9wk60,247
+tinybird/tb/__cli__.py,sha256=WaMYFAckfNWnMzyIA-CatGh7AXWIKDHBGMB-QqtF2rw,247
 tinybird/tb/check_pypi.py,sha256=Gp0HkHHDFMSDL6nxKlOY51z7z1Uv-2LRexNTZSHHGmM,552
 tinybird/tb/cli.py,sha256=FdDFEIayjmsZEVsVSSvRiVYn_FHOVg_zWQzchnzfWho,1008
 tinybird/tb/client.py,sha256=pJbdkWMXGAqKseNAvdsRRnl_c7I-DCMB0dWCQnG82nU,54146
@@ -63,16 +63,17 @@ 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=nUvSfgi7OdrWXRxLjLHzepsrICiduAZcW7iFE0x5Q5A,10323
+tinybird/tb/modules/agent/agent.py,sha256=TWHU9gEPH7EZQguno1P_HZ5jO7v6QLj9UE1btgaAOxg,10570
+tinybird/tb/modules/agent/animations.py,sha256=z0MNLf8TnUO8qAjgYvth_wc9a9283pNVz1Z4jl15Ggs,2558
 tinybird/tb/modules/agent/banner.py,sha256=KX_e467uiy1gWOZ4ofTZt0GCFGQqHQ_8Ob27XLQqda0,3053
 tinybird/tb/modules/agent/memory.py,sha256=H6SJK--2L5C87B7AJd_jMqsq3sCvFvZwZXmajuT0GBE,1171
 tinybird/tb/modules/agent/models.py,sha256=mf8dRCdof6uEFZWh5xQ_D_FStk7eDds7qWRNSbDklUM,589
 tinybird/tb/modules/agent/prompts.py,sha256=j7nkx1E2BnPP6Ra718FvSDGRWucyFgSX07V9QAiaiGg,4864
-tinybird/tb/modules/agent/utils.py,sha256=BWbS8m8HYrbbm1OnH7kLYLxL37-q1yVKa1ObWw7klfc,13041
+tinybird/tb/modules/agent/utils.py,sha256=brW-2StkAy-X1LGDbjQYhOXMSifGwWJCSkPDDmo7SUk,13069
 tinybird/tb/modules/agent/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tinybird/tb/modules/agent/tools/create_datafile.py,sha256=06HK33h86T1d54zvnJXJz_1su3zIS7UJDPyBqhgo81c,1771
+tinybird/tb/modules/agent/tools/create_datafile.py,sha256=V5VMFvpne-LW1XzQY7AnSMiOkMALSRspEjtBqK1Glnk,1858
 tinybird/tb/modules/agent/tools/explore.py,sha256=ihALc_kBcsjrKT3hZyicqyIowB0g_K3AtNNi-5uz9-8,412
-tinybird/tb/modules/agent/tools/plan.py,sha256=acMbyjqI6cN7IwmMAC8-MPNSg0PEZVKGEk2NOY4gMi8,1162
+tinybird/tb/modules/agent/tools/plan.py,sha256=K9mXqCgLXbwXtyFFjPvrrwRwEm2rNnIMn73qazDTQts,1345
 tinybird/tb/modules/agent/tools/preview_datafile.py,sha256=e9q5fR0afApcrntzFrnuHmd10ex7MG_GM6T0Pwc9bRI,850
 tinybird/tb/modules/datafile/build.py,sha256=NFKBrusFLU0WJNCXePAFWiEDuTaXpwc0lHlOQWEJ43s,51117
 tinybird/tb/modules/datafile/build_common.py,sha256=2yNdxe49IMA9wNvl25NemY2Iaz8L66snjOdT64dm1is,4511
@@ -94,8 +95,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.dev241.dist-info/METADATA,sha256=urGByLwa4s7Sstte1j0r7UZ6iYFUmnpX4ujKWztx9II,1733
-tinybird-0.0.1.dev241.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev241.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev241.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev241.dist-info/RECORD,,
+tinybird-0.0.1.dev242.dist-info/METADATA,sha256=jcAeg7hLOB_0X5tNRd6Jchij6S62DjqZ3I4vHblBchk,1733
+tinybird-0.0.1.dev242.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev242.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev242.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev242.dist-info/RECORD,,