tinybird 0.0.1.dev275__py3-none-any.whl → 0.0.1.dev276__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.dev275'
-__revision__ = '0c15fc3'
+__version__ = '0.0.1.dev276'
+__revision__ = '6716ae6'

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev275
+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.dev275.dist-info → tinybird-0.0.1.dev276.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=F6DuvogeaLQfXF18e87QuhWLJgcSX2Sq60BJKK8XzF8,247
+tinybird/tb/__cli__.py,sha256=bGmVAS-5X5ZYyTYKDq1fQ8vA4zppIypVRErGtzvDkAU,247
 tinybird/tb/check_pypi.py,sha256=Gp0HkHHDFMSDL6nxKlOY51z7z1Uv-2LRexNTZSHHGmM,552
 tinybird/tb/cli.py,sha256=FdDFEIayjmsZEVsVSSvRiVYn_FHOVg_zWQzchnzfWho,1008
 tinybird/tb/client.py,sha256=IQRaInDjOwr9Fzaz3_xXc3aUGqh94tM2lew7IZbB9eM,53733
@@ -42,7 +42,7 @@ tinybird/tb/modules/feedback_manager.py,sha256=xkZjwGOV50-oc0y9h58ei4MZ1mwOMOaN_
 tinybird/tb/modules/info.py,sha256=F5vY4kHS_kyO2uSBKac92HoOb447oDeRlzpwtAHTuKc,6872
 tinybird/tb/modules/infra.py,sha256=JE9oLIyF4bi_JBoe-BgZ5HhKp_lQgSihuSV1KIS02Qs,32709
 tinybird/tb/modules/job.py,sha256=wBsnu8UPTOha2rkLvucgmw4xYv73ubmui3eeSIF68ZM,3107
-tinybird/tb/modules/llm.py,sha256=CpTq2YAk88E8ENpQA94-mas3UDN1aqa--9Al8GdwQtc,1575
+tinybird/tb/modules/llm.py,sha256=fPBBCmM3KlCksLlgJkg4joDn6y3H5QjDzE-Pm4YNf7E,1782
 tinybird/tb/modules/llm_utils.py,sha256=nS9r4FAElJw8yXtmdYrx-rtI2zXR8qXfi1QqUDCfxvg,3469
 tinybird/tb/modules/local.py,sha256=tpiw_F_qOIp42h3kTBwTm5GQDyuVLF0QNF1jmB0zR94,6845
 tinybird/tb/modules/local_common.py,sha256=_WODjW3oPshgsZ1jDFFx2nr0zrLi3Gxz5dlahWPobM8,17464
@@ -69,7 +69,7 @@ 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=MjLpl2uNASKJiKH74U9HgvQk4h8gxii5u2KR1nJAXx4,31096
+tinybird/tb/modules/agent/agent.py,sha256=kPZEudIYL4pf4FXj_d7fWZvUi1f5BqG5fd-QWQueKsA,32976
 tinybird/tb/modules/agent/animations.py,sha256=4WOC5_2BracttmMCrV0H91tXfWcUzQHBUaIJc5FA7tE,3490
 tinybird/tb/modules/agent/banner.py,sha256=l6cO5Fi7lbVKp-GsBP8jf3IkjOWxg2jpAt9NBCy0WR8,4085
 tinybird/tb/modules/agent/command_agent.py,sha256=NTzgb9qnuG-gDpk87VijKs1UUMukJPaJI5UiZtRWUoQ,2864
@@ -117,8 +117,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.dev275.dist-info/METADATA,sha256=94PPiZAh3Vx9P9w5zSsse5_Dr9KV22caI0t7iM0hNsE,1763
-tinybird-0.0.1.dev275.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev275.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev275.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev275.dist-info/RECORD,,
+tinybird-0.0.1.dev276.dist-info/METADATA,sha256=EILhMFhFnfIsfN1iaV2ZG3Ex5EmbLNELsIXRrC1ODPo,1763
+tinybird-0.0.1.dev276.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev276.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev276.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev276.dist-info/RECORD,,