tinybird 0.0.1.dev240__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.dev240'
-__revision__ = '268407c'
+__version__ = '0.0.1.dev242'
+__revision__ = '62ad714'

tinybird/tb/modules/agent/agent.py

@@ -21,8 +21,9 @@ from tinybird.prompts import (
     s3_connection_example,
     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.client import TinybirdClient
 from tinybird.tb.modules.agent.memory import clear_history, load_history
 from tinybird.tb.modules.agent.models import create_model
 from tinybird.tb.modules.agent.prompts import datafile_instructions, plan_instructions, sql_instructions
@@ -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),
             ],
         )
 
@@ -134,17 +135,25 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
         return self.messages[-5:] if len(self.messages) > 5 else self.messages
 
     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(
-                client=TinybirdClient(token=self.token, host=self.host),
+                # 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/models.py

@@ -6,7 +6,7 @@ from pydantic_ai.providers.anthropic import AnthropicProvider
 
 def create_model(
     token: str,
-    base_url: str = "https://api.wadus3.aws.tinybird.co",
+    base_url: str,
     model: AnthropicModelName = "claude-4-sonnet-20250514",
 ):
     client = AsyncAnthropic(

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/explore.py

@@ -12,4 +12,4 @@ def explore_data(ctx: RunContext[TinybirdAgentContext], prompt: str):
     Returns:
         str: The result of the exploration.
     """
-    return ctx.deps.client.explore_data(prompt)
+    return ctx.deps.explore_data(prompt)

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

@@ -1,6 +1,6 @@
 import os
 from contextlib import contextmanager
-from typing import Any, List, Optional
+from typing import Any, Callable, List, Optional
 
 import click
 from prompt_toolkit.application import Application, get_app
@@ -16,12 +16,11 @@ from prompt_toolkit.shortcuts import PromptSession
 from prompt_toolkit.styles import Style
 from pydantic import BaseModel, Field
 
-from tinybird.tb.modules.agent.client import TinybirdClient
-
 
 class TinybirdAgentContext(BaseModel):
-    client: TinybirdClient
+    explore_data: Callable[[str], str]
     folder: str
+    thinking_animation: Any
 
 
 default_style = Style.from_dict(

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev240
+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.dev240.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=OB0M6HqPur6rsd93Mf9BRpLf5zvrZ_aq4HvYd7nAacc,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,17 +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=98X8rqUszArhOnrWWGAJRNwAK-me6Ql2UGNGFxcuc2E,10205
+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/client.py,sha256=ZNE_0xtbrfhXXRZ7nx3Ze0gHb1IK7zXRLTlGudAAuUg,1264
 tinybird/tb/modules/agent/memory.py,sha256=H6SJK--2L5C87B7AJd_jMqsq3sCvFvZwZXmajuT0GBE,1171
-tinybird/tb/modules/agent/models.py,sha256=jUEO7i_JwlQVZi6QEUeGyFJMFkPZ4ikyDZdei6s6f3M,628
+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=E785Sq_TLVBi1gaBR53YlHvyP_Kbh4TmRouYl3KHFpo,13080
+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/explore.py,sha256=iEVsi85CRPCPhWcl-fC1J4hYK22TfunxyqNu5pDxdDU,419
-tinybird/tb/modules/agent/tools/plan.py,sha256=acMbyjqI6cN7IwmMAC8-MPNSg0PEZVKGEk2NOY4gMi8,1162
+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=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
@@ -95,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.dev240.dist-info/METADATA,sha256=XT_HpNQ7KOTyQt5ICua7m_CWsFaKQGB0tMvg43YO4XM,1733
-tinybird-0.0.1.dev240.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev240.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev240.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev240.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,,

tinybird/tb/modules/agent/client.py

@@ -1,42 +0,0 @@
-from dataclasses import dataclass
-from typing import Any, Dict, Optional
-
-import httpx
-
-
-@dataclass
-class TinybirdClient:
-    def __init__(self, host: str, token: str):
-        self.host = host
-        self.token = token
-        self.client = httpx.Client(
-            timeout=30.0,
-            headers={"Accept": "application/json", "User-Agent": "Python/APIClient"},
-        )
-        self.insights: list[str] = []
-
-    def __aenter__(self):
-        return self
-
-    def __aexit__(self, exc_type, exc_val, exc_tb):
-        self.close()
-
-    def close(self):
-        """Close the underlying HTTP client."""
-        self.client.close()
-
-    def _get(self, endpoint: str, params: Optional[Dict[str, Any]] = None) -> str:
-        if params is None:
-            params = {}
-        params["token"] = self.token
-        url = f"{self.host}{endpoint}"
-        response = self.client.get(url, params=params)
-        try:
-            response.raise_for_status()
-        except Exception as e:
-            raise Exception(response.json().get("error", str(e))) from e
-        return response.text
-
-    def explore_data(self, prompt: str) -> str:
-        params = {"prompt": prompt, "host": self.host, "origin": "cli"}
-        return self._get("/v1/agents/explore", params)