tinybird 0.0.1.dev288__tar.gz → 0.0.1.dev289__tar.gz

{tinybird-0.0.1.dev288 → tinybird-0.0.1.dev289}/PKG-INFO

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

{tinybird-0.0.1.dev288 → tinybird-0.0.1.dev289}/tinybird/datafile/common.py

@@ -1877,16 +1877,6 @@ def parse(
                 doc.shared_with += workspaces
                 total_workspaces += len(workspaces)
 
-        # Add warning once if any workspaces were found
-        if total_workspaces > 0:
-            warnings.append(
-                DatafileParseWarning(
-                    message=(
-                        f"{kwargs['cmd'].upper()} is not fully implemented in Forward and will be ignored while this warning is present."
-                    )
-                )
-            )
-
         # Validate that at least one workspace was provided
         if total_workspaces == 0:
             raise DatafileSyntaxError(

{tinybird-0.0.1.dev288 → tinybird-0.0.1.dev289}/tinybird/sql_toolset.py

@@ -144,7 +144,7 @@ def format_where_for_mutation_command(where_clause: str) -> str:
     return f"DELETE WHERE {quoted_condition[1:-1]}"
 
 
-@lru_cache(maxsize=2**13)
+@lru_cache(maxsize=2**15)
 def sql_get_used_tables_cached(
     sql: str,
     raising: bool = False,
@@ -301,7 +301,7 @@ def replacements_to_tuples(replacements: dict) -> dict:
     return parsed_replacements
 
 
-@lru_cache(maxsize=2**13)
+@lru_cache(maxsize=2**15)
 def replace_tables_chquery_cached(
     sql: str,
     sorted_replacements: Optional[tuple] = None,

{tinybird-0.0.1.dev288 → tinybird-0.0.1.dev289}/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.dev288'
-__revision__ = '6cd5b6c'
+__version__ = '0.0.1.dev289'
+__revision__ = 'ac20837'

{tinybird-0.0.1.dev288 → tinybird-0.0.1.dev289}/tinybird/tb/modules/agent/agent.py

@@ -50,7 +50,13 @@ from tinybird.tb.modules.agent.tools.get_endpoint_stats import get_endpoint_stat
 from tinybird.tb.modules.agent.tools.get_openapi_definition import get_openapi_definition
 from tinybird.tb.modules.agent.tools.plan import complete_plan, plan
 from tinybird.tb.modules.agent.tools.secret import create_or_update_secrets
-from tinybird.tb.modules.agent.utils import AgentRunCancelled, TinybirdAgentContext, show_confirmation, show_input
+from tinybird.tb.modules.agent.utils import (
+    AgentRunCancelled,
+    SubAgentRunCancelled,
+    TinybirdAgentContext,
+    show_confirmation,
+    show_input,
+)
 from tinybird.tb.modules.build_common import process as build_process
 from tinybird.tb.modules.common import (
     _analyze,
@@ -207,8 +213,11 @@ class TinybirdAgent:
             Returns:
                 str: The result of the command.
             """
-            result = self.command_agent.run(task, deps=ctx.deps, usage=ctx.usage)
-            return f"Result: {result.output}\nDo not repeat in your response the result again, because it is already displayed in the terminal."
+            try:
+                result = self.command_agent.run(task, deps=ctx.deps, usage=ctx.usage)
+                return f"Result: {result.output}\nDo not repeat in your response the result again, because it is already displayed in the terminal."
+            except SubAgentRunCancelled as e:
+                return f"User does not want to continue with the proposed solution. Reason: {e}"
 
         @self.agent.tool
         def explore_data(ctx: RunContext[TinybirdAgentContext], task: str) -> str:

{tinybird-0.0.1.dev288 → tinybird-0.0.1.dev289}/tinybird/tb/modules/agent/tools/run_command.py

@@ -3,7 +3,13 @@ import subprocess
 import click
 from pydantic_ai import RunContext
 
-from tinybird.tb.modules.agent.utils import AgentRunCancelled, TinybirdAgentContext, show_confirmation, show_input
+from tinybird.tb.modules.agent.utils import (
+    AgentRunCancelled,
+    SubAgentRunCancelled,
+    TinybirdAgentContext,
+    show_confirmation,
+    show_input,
+)
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 
@@ -24,7 +30,7 @@ def run_command(ctx: RunContext[TinybirdAgentContext], command: str):
         if confirmation == "review":
             feedback = show_input(ctx.deps.workspace_name)
             ctx.deps.thinking_animation.start()
-            return f"User did not confirm the command and gave the following feedback: {feedback}"
+            raise SubAgentRunCancelled(f"""User did not confirm the command `{command}`. Reason: {feedback}.""")
 
         click.echo(FeedbackManager.highlight(message=f"» Running command: {command}"))
         command = command.replace("tb", "tb --no-version-warning")
@@ -32,7 +38,7 @@ def run_command(ctx: RunContext[TinybirdAgentContext], command: str):
         click.echo(result.stdout)
         ctx.deps.thinking_animation.start()
         return result.stdout
-    except AgentRunCancelled as e:
+    except (AgentRunCancelled, SubAgentRunCancelled) as e:
         raise e
     except Exception as e:
         click.echo(FeedbackManager.error(message=f"Error running command: {e}"))

{tinybird-0.0.1.dev288 → tinybird-0.0.1.dev289}/tinybird/tb/modules/agent/utils.py

@@ -721,6 +721,12 @@ class AgentRunCancelled(Exception):
     pass
 
 
+class SubAgentRunCancelled(Exception):
+    """Exception raised when sub-agent cancels an operation"""
+
+    pass
+
+
 def show_confirmation(title: str, skip_confirmation: bool = False, show_review: bool = True) -> ConfirmationResult:
     if skip_confirmation:
         return "yes"

{tinybird-0.0.1.dev288 → tinybird-0.0.1.dev289}/tinybird/tb/modules/local_common.py

@@ -24,6 +24,7 @@ from tinybird.tb.modules.telemetry import add_telemetry_event
 TB_IMAGE_NAME = "tinybirdco/tinybird-local:latest"
 TB_CONTAINER_NAME = "tinybird-local"
 TB_LOCAL_PORT = int(os.getenv("TB_LOCAL_PORT", 7181))
+TB_LOCAL_CLICKHOUSE_INTERFACE_PORT = int(os.getenv("TB_LOCAL_CLICKHOUSE_INTERFACE_PORT", 7182))
 TB_LOCAL_HOST = re.sub(r"^https?://", "", os.getenv("TB_LOCAL_HOST", "localhost"))
 TB_LOCAL_ADDRESS = f"http://{TB_LOCAL_HOST}:{TB_LOCAL_PORT}"
 TB_LOCAL_DEFAULT_WORKSPACE_NAME = "Tinybird_Local_Testing"
@@ -269,7 +270,7 @@ def start_tinybird_local(
             TB_IMAGE_NAME,
             name=TB_CONTAINER_NAME,
             detach=True,
-            ports={"7181/tcp": TB_LOCAL_PORT},
+            ports={"7181/tcp": TB_LOCAL_PORT, "7182/tcp": TB_LOCAL_CLICKHOUSE_INTERFACE_PORT},
             remove=False,
             platform="linux/amd64",
             environment=environment,

{tinybird-0.0.1.dev288 → tinybird-0.0.1.dev289}/tinybird.egg-info/PKG-INFO

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