tinybird 0.0.1.dev241__tar.gz → 0.0.1.dev242__tar.gz

{tinybird-0.0.1.dev241 → tinybird-0.0.1.dev242}/PKG-INFO

@@ -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 → tinybird-0.0.1.dev242}/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-0.0.1.dev241 → tinybird-0.0.1.dev242}/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-0.0.1.dev242/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-0.0.1.dev241 → tinybird-0.0.1.dev242}/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-0.0.1.dev241 → tinybird-0.0.1.dev242}/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-0.0.1.dev241 → tinybird-0.0.1.dev242}/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 → tinybird-0.0.1.dev242}/tinybird.egg-info/PKG-INFO

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

@@ -75,6 +75,7 @@ tinybird/tb/modules/workspace.py
 tinybird/tb/modules/workspace_members.py
 tinybird/tb/modules/agent/__init__.py
 tinybird/tb/modules/agent/agent.py
+tinybird/tb/modules/agent/animations.py
 tinybird/tb/modules/agent/banner.py
 tinybird/tb/modules/agent/memory.py
 tinybird/tb/modules/agent/models.py