tinybird 0.0.1.dev275__tar.gz → 0.0.1.dev277__tar.gz

{tinybird-0.0.1.dev275 → tinybird-0.0.1.dev277}/PKG-INFO

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

{tinybird-0.0.1.dev275 → tinybird-0.0.1.dev277}/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.dev277'
+__revision__ = 'f19b52a'

{tinybird-0.0.1.dev275 → tinybird-0.0.1.dev277}/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.dev275 → tinybird-0.0.1.dev277}/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 → tinybird-0.0.1.dev277}/tinybird/tb/modules/workspace.py

@@ -134,12 +134,6 @@ def workspace_current(ctx: Context):
 
 @workspace.command(name="create", short_help="Create a new Workspace for your Tinybird user")
 @click.argument("workspace_name", required=False)
-@click.option(
-    "--fork",
-    is_flag=True,
-    default=False,
-    help="When enabled, tb will share all data sources from the current workspace with the new one",
-)
 @click.option(
     "--organization-id",
     "organization_id",
@@ -151,7 +145,6 @@ def workspace_current(ctx: Context):
 def create_workspace(
     ctx: Context,
     workspace_name: str,
-    fork: bool,
     organization_id: Optional[str],
 ) -> None:
     config = CLIConfig.get_project_config()
@@ -165,6 +158,7 @@ def create_workspace(
 
     config = CLIConfig.get_project_config()
     user_token = config.get_user_token()
+    fork = False
 
     if not user_token:
         raise CLIWorkspaceException(

{tinybird-0.0.1.dev275 → tinybird-0.0.1.dev277}/tinybird.egg-info/PKG-INFO

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