tinybird 0.0.1.dev274__tar.gz → 0.0.1.dev276__tar.gz

{tinybird-0.0.1.dev274 → tinybird-0.0.1.dev276}/PKG-INFO

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

{tinybird-0.0.1.dev274 → tinybird-0.0.1.dev276}/tinybird/feedback_manager.py

@@ -598,6 +598,12 @@ Ready? """
     warning_confirm_delete_branch = prompt_message("Do you want to remove '{branch}' Branch?")
     warning_confirm_delete_release = prompt_message("Do you want to remove Release {semver}?")
     warning_confirm_rollback_release = prompt_message("Do you want to rollback current Release {semver} to {rollback}?")
+    warning_confirm_on_demand_compute = warning_message(
+        "On-demand compute will incur additional costs beyond your regular usage.\n"
+        "This feature uses dedicated compute resources that are billed separately.\n"
+        "You can read more about the pricing at https://www.tinybird.co/docs/classic/work-with-data/process-and-copy/materialized-views#compute-compute-separation-for-populates\n\n"
+        "Do you want to proceed with on-demand compute?"
+    )
 
     warning_confirm_delete_token = prompt_message("Do you want to delete Token {token}?")
     warning_confirm_refresh_token = prompt_message("Do you want to refresh Token {token}?")

{tinybird-0.0.1.dev274 → tinybird-0.0.1.dev276}/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.dev274'
-__revision__ = 'c3d62b9'
+__version__ = '0.0.1.dev276'
+__revision__ = '6716ae6'

{tinybird-0.0.1.dev274 → tinybird-0.0.1.dev276}/tinybird/tb/modules/agent/agent.py

@@ -59,6 +59,7 @@ from tinybird.tb.modules.config import CLIConfig
 from tinybird.tb.modules.deployment_common import create_deployment
 from tinybird.tb.modules.exceptions import CLIBuildException, CLIDeploymentException, CLIMockException
 from tinybird.tb.modules.feedback_manager import FeedbackManager
+from tinybird.tb.modules.llm import LLM
 from tinybird.tb.modules.local_common import get_tinybird_local_client
 from tinybird.tb.modules.login_common import login
 from tinybird.tb.modules.mock_common import append_mock_data, create_mock_data
@@ -474,9 +475,13 @@ def run_agent(
                     click.echo(FeedbackManager.info(message="Message history cleared!"))
                     clear_messages()
                     continue
-                elif user_input.lower().startswith("select ") or user_input.lower().startswith("with "):
+                elif not is_natural_language(user_input, user_token, host):
                     query = f"SELECT * FROM ({user_input.strip()}) FORMAT JSON"
-                    result = execute_query_local(config, query=query)
+                    try:
+                        result = execute_query_local(config, query=query)
+                    except Exception as e:
+                        click.echo(FeedbackManager.error(message=f"Error executing query: {e}"))
+                        continue
                     stats = result["statistics"]
                     seconds = stats["elapsed"]
                     rows_read = humanfriendly.format_number(stats["rows_read"])
@@ -725,3 +730,46 @@ def get_pipe_data_test(client: TinyB, pipe_name: str, test_params: Optional[dict
         "pipeline": pipe_name,
     }
     return client._req_raw(f"""/v0/sql?{urllib.parse.urlencode(params)}&{test_params}""")
+
+
+def is_natural_language(user_input: str, user_token: str, host: str) -> bool:
+    stripped_input = user_input.strip().lower()
+    sql_keywords = ["select", "with"]
+    if not any(stripped_input.startswith(keyword) for keyword in sql_keywords):
+        return True
+
+    prompt = """Analyze the following text and determine if it's natural language or a SQL query.
+
+Respond with only "NATURAL" if it's natural language (like a question, request, or conversational text), or "SQL" if it's a SQL query (like SELECT, WITH, INSERT, UPDATE, DELETE, etc. statements).
+
+Examples:
+- "show me all users" -> NATURAL
+- "what are the top products?" -> NATURAL
+- "SELECT * FROM users" -> SQL
+- "WITH cte AS (SELECT...) SELECT..." -> SQL
+- "select count(*) from orders" -> SQL
+- "help me analyze the data" -> NATURAL"""
+
+    try:
+        thinking_animation = ThinkingAnimation()
+        thinking_animation.start()
+        llm = LLM(user_token=user_token, host=host)
+        response_text = llm.ask(
+            system_prompt=prompt,
+            prompt=f"Text: '{user_input}'",
+            feature="tb_agent_is_natural_language",
+            model="vertex_ai/gemini-2.0-flash-001",
+        )
+        thinking_animation.stop()
+        if "NATURAL" in response_text:
+            return True
+        elif "SQL" in response_text:
+            return False
+        else:
+            # If unclear, default to natural language to be safe
+            return True
+
+    except Exception:
+        # If the LLM call fails, fall back to simple heuristics
+        # Check if it starts with common SQL keywords
+        return not any(stripped_input.startswith(keyword) for keyword in sql_keywords)

{tinybird-0.0.1.dev274 → tinybird-0.0.1.dev276}/tinybird/tb/modules/cli.py

@@ -149,7 +149,7 @@ def cli(
     if user_token:
         config_temp.set_user_token(user_token)
     if token or host or user_token:
-        try_update_config_with_remote(config_temp, auto_persist=False, raise_on_errors=False)
+        try_update_config_with_remote(config_temp, auto_persist=True, raise_on_errors=False)
 
     # Overwrite token and host with env vars manually, without resorting to click.
     #

{tinybird-0.0.1.dev274 → tinybird-0.0.1.dev276}/tinybird/tb/modules/deployment_common.py

@@ -19,11 +19,24 @@ from tinybird.tb.modules.project import Project
 
 
 # TODO(eclbg): This should eventually end up in client.py, but we're not using it here yet.
-def api_fetch(url: str, headers: dict) -> dict:
-    r = requests.get(url, headers=headers)
-    if r.status_code == 200:
-        logging.debug(json.dumps(r.json(), indent=2))
-        return r.json()
+def api_fetch(url: str, headers: dict, max_retries: int = 3, backoff_factor: float = 0.5) -> dict:
+    retries = 0
+    while retries <= max_retries:
+        try:
+            r = requests.get(url, headers=headers)
+            if r.status_code == 200:
+                logging.debug(json.dumps(r.json(), indent=2))
+                return r.json()
+            else:
+                raise Exception(f"Request failed with status code {r.status_code}")
+        except Exception:
+            retries += 1
+            if retries > max_retries:
+                break
+
+            wait_time = backoff_factor * (2 ** (retries - 1))
+            time.sleep(wait_time)
+
     # Try to parse and print the error from the response
     try:
         result = r.json()
@@ -35,6 +48,7 @@ def api_fetch(url: str, headers: dict) -> dict:
         message = "Error parsing response from API"
         click.echo(FeedbackManager.error(message=message))
         sys_exit("deployment_error", message)
+
     return {}
 
 
@@ -48,6 +62,7 @@ def api_post(
     if r.status_code < 300:
         logging.debug(json.dumps(r.json(), indent=2))
         return r.json()
+
     # Try to parse and print the error from the response
     try:
         result = r.json()
@@ -61,6 +76,7 @@ def api_post(
         message = "Error parsing response from API"
         click.echo(FeedbackManager.error(message=message))
         sys_exit("deployment_error", message)
+
     return {}
 
 

{tinybird-0.0.1.dev274 → tinybird-0.0.1.dev276}/tinybird/tb/modules/llm.py

@@ -20,13 +20,15 @@ class LLM:
         self.host = host
         self.user_token = user_token
 
-    def ask(self, system_prompt: str, prompt: str, feature: Optional[str] = None) -> str:
+    def ask(self, system_prompt: str, prompt: str, feature: Optional[str] = None, model: Optional[str] = None) -> str:
         """
         Calls the model with the given prompt and returns the response.
 
         Args:
             system_prompt (str): The system prompt to send to the model.
             prompt (str): The user prompt to send to the model.
+            feature (Optional[str]): The feature to send to the model.
+            model (Optional[str]): The model to use.
 
         Returns:
             str: The response from the language model.
@@ -36,6 +38,8 @@ class LLM:
         params = {"origin": "cli"}
         if feature:
             params["feature"] = feature
+        if model:
+            params["model"] = model
         cli_config = CLIConfig.get_project_config()
         workspace_id = cli_config.get("id")
         params_str = urllib.parse.urlencode(params)

{tinybird-0.0.1.dev274 → tinybird-0.0.1.dev276}/tinybird/tb_cli_modules/cli.py

@@ -643,6 +643,15 @@ async def push(
 ) -> None:
     """Push files to Tinybird."""
 
+    # Prompt for confirmation when on-demand compute is requested with populate
+    if (
+        populate
+        and on_demand_compute
+        and not click.confirm(FeedbackManager.warning_confirm_on_demand_compute(), default=False)
+    ):
+        click.echo("Operation cancelled.")
+        return
+
     ignore_sql_errors = FeatureFlags.ignore_sql_errors()
     context.disable_template_security_validation.set(True)
 

{tinybird-0.0.1.dev274 → tinybird-0.0.1.dev276}/tinybird/tb_cli_modules/pipe.py

@@ -253,6 +253,12 @@ async def pipe_populate(
     on_demand_compute: bool,
 ):
     """Populate the result of a Materialized Node into the target Materialized View"""
+
+    # Prompt for confirmation when on-demand compute is requested
+    if on_demand_compute and not click.confirm(FeedbackManager.warning_confirm_on_demand_compute(), default=False):
+        click.echo("Operation cancelled.")
+        return
+
     cl = create_tb_client(ctx)
 
     pipe = await cl.pipe(pipe_name)

{tinybird-0.0.1.dev274 → tinybird-0.0.1.dev276}/tinybird.egg-info/PKG-INFO

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