tinybird 0.0.1.dev271__tar.gz → 0.0.1.dev273__tar.gz

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/PKG-INFO

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

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/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.dev271'
-__revision__ = 'aeab6b9'
+__version__ = '0.0.1.dev273'
+__revision__ = '9f4ea9e'

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/tinybird/tb/client.py

@@ -300,13 +300,10 @@ class TinyB:
         response = self._req(f"/v0/connectors?{urlencode(params)}")
         return response["connectors"]
 
-    def connections(self, connector: Optional[str] = None, skip_bigquery: Optional[bool] = False):
+    def connections(self, connector: Optional[str] = None):
         response = self._req("/v0/connectors")
         connectors = response["connectors"]
-        bigquery_connection = None
-        if not skip_bigquery:
-            bigquery_connection = self.bigquery_connection() if connector == "bigquery" or connector is None else None
-        connectors = [*connectors, bigquery_connection] if bigquery_connection else connectors
+
         if connector:
             return [
                 {
@@ -328,24 +325,9 @@ class TinyB:
                 **c["settings"],
             }
             for c in connectors
+            if c["service"] != "gcscheduler"
         ]
 
-    def bigquery_connection(self):
-        bigquery_resources = self.list_gcp_resources()
-        if len(bigquery_resources) == 0:
-            return None
-
-        gcp_account_details: Dict[str, Any] = self.get_gcp_service_account_details()
-        datasources = self.datasources()
-        bigquery_datasources = [ds["name"] for ds in datasources if ds["type"] == "bigquery"]
-        return {
-            "id": gcp_account_details["account"].split("@")[0],
-            "service": "bigquery",
-            "name": "bigquery",
-            "linkers": bigquery_datasources,
-            "settings": gcp_account_details,
-        }
-
     def get_datasource(self, ds_name: str, used_by: bool = False) -> Dict[str, Any]:
         params = {
             "attrs": "used_by" if used_by else "",
@@ -1037,10 +1019,9 @@ class TinyB:
         name_or_id: str,
         service: Optional[str] = None,
         key: Optional[str] = "name",
-        skip_bigquery: Optional[bool] = False,
     ) -> Optional[Dict[str, Any]]:
         return next(
-            (c for c in self.connections(connector=service, skip_bigquery=skip_bigquery) if c[key] == name_or_id),
+            (c for c in self.connections(connector=service) if c[key] == name_or_id),
             None,
         )
 

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/tinybird/tb/modules/agent/agent.py

@@ -37,7 +37,7 @@ from tinybird.tb.modules.agent.testing_agent import TestingAgent
 from tinybird.tb.modules.agent.tools.analyze import analyze_file, analyze_url
 from tinybird.tb.modules.agent.tools.append import append_file, append_url
 from tinybird.tb.modules.agent.tools.build import build
-from tinybird.tb.modules.agent.tools.create_datafile import create_datafile, rename_datafile_or_fixture
+from tinybird.tb.modules.agent.tools.datafile import create_datafile, remove_file, rename_datafile_or_fixture
 from tinybird.tb.modules.agent.tools.deploy import deploy
 from tinybird.tb.modules.agent.tools.deploy_check import deploy_check
 from tinybird.tb.modules.agent.tools.diff_resource import diff_resource
@@ -48,7 +48,13 @@ from tinybird.tb.modules.agent.tools.plan import 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.build_common import process as build_process
-from tinybird.tb.modules.common import _analyze, _get_tb_client, echo_safe_humanfriendly_tables_format_pretty_table
+from tinybird.tb.modules.common import (
+    _analyze,
+    _get_tb_client,
+    echo_safe_humanfriendly_tables_format_pretty_table,
+    get_region_from_host,
+    get_regions,
+)
 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
@@ -83,13 +89,15 @@ class TinybirdAgent:
             self.messages: list[ModelMessage] = get_last_messages_from_last_user_prompt()
         else:
             self.messages = []
-
+        cli_config = CLIConfig.get_project_config()
+        regions = get_regions(cli_config)
         self.agent = Agent(
             model=create_model(user_token, host, workspace_id),
             deps_type=TinybirdAgentContext,
             system_prompt=agent_system_prompt,
             tools=[
                 Tool(create_datafile, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
+                Tool(remove_file, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
                 Tool(
                     rename_datafile_or_fixture,
                     docstring_format="google",
@@ -206,7 +214,27 @@ class TinybirdAgent:
 
         @self.agent.instructions
         def get_cloud_host(ctx: RunContext[TinybirdAgentContext]) -> str:
-            return f"Tinybird Cloud host: {ctx.deps.host}"
+            try:
+                region = get_region_from_host(ctx.deps.host, regions) or {
+                    "provider": "Unknown",
+                    "name": "Unknown",
+                }
+            except Exception as e:
+                click.echo(FeedbackManager.error(message=f"Error getting region info: {e}"))
+                region = {
+                    "provider": "Unknown",
+                    "name": "Unknown",
+                }
+
+            region_provider = region["provider"]
+            region_name = region["name"]
+            return f"""Tinybird Cloud info (region details):
+- API Host: {ctx.deps.host}
+- Workspace ID: {ctx.deps.workspace_id}
+- Workspace Name: {project.workspace_name} (in Tinybird Local the workspace name is the same because it is synced with Cloud)
+- Region provider: {region_provider}
+- Region name: {region_name}
+"""
 
         @self.agent.instructions
         def get_local_token(ctx: RunContext[TinybirdAgentContext]) -> str:
@@ -533,7 +561,7 @@ def build_project_test(
         raise CLIBuildException(build_error)
 
 
-def deploy_project(config: dict[str, Any], project: Project) -> None:
+def deploy_project(config: dict[str, Any], project: Project, allow_destructive_operations: bool = False) -> None:
     client = _get_tb_client(config["token"], config["host"])
     try:
         create_deployment(
@@ -542,7 +570,7 @@ def deploy_project(config: dict[str, Any], project: Project) -> None:
             config=config,
             wait=True,
             auto=True,
-            allow_destructive_operations=False,
+            allow_destructive_operations=allow_destructive_operations,
         )
     except SystemExit as e:
         raise CLIDeploymentException(e.args[0])

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/tinybird/tb/modules/agent/prompts.py

@@ -31,14 +31,17 @@ Steps:
 8. Copy: [name] - [description] - Depends on: [resources]
 9. Generate mock data: [datasource_name]
 10. Append existing fixture: [fixture_pathname] - Target: [datasource_name]
+11. Delete file: [file_pathname] - [reason for deletion]
 
 <dev_notes>
 You can skip steps where resources will not be created or updated.
 Always add 'Create secrets' as the FIRST step if any secrets/environment variables are required for the implementation. This step should include ALL required secrets at once.
 Always add 'Generate mock data' step if a landing datasource was created without providing a fixture file.
 Always add 'Append existing fixture' step if a landing datasource was created after providing a fixture file.
+Always add 'Delete file' step when removing datafiles, fixtures, or any other project files.
 Solve the specific user request, do not add extra steps that are not related to the user request.
 Reuse the existing resources if possible.
+If a plan only has one step, skip the plan and go directly to the next tool needed.
 </dev_notes>
 
 Resource dependencies:
@@ -735,7 +738,7 @@ You are an interactive CLI tool that helps users with data engineering tasks. Us
 
 # Tools
 You have access to the following tools:
-1. `create_datafile` - Create a datafile (datasource, endpoint, materialized, sink, copy, connection) in the project folder. Confirmation will be asked by the tool before creating the file.
+1. `datafile` - Create datafiles and remove files (datasource, endpoint, materialized, sink, copy, connection, fixtures) in the project folder. Confirmation will be asked by the tool before creating or removing the file.
 2. `plan` - Plan the creation or update of resources.
 3. `build` - Build the project.
 4. `deploy` - Deploy the project to Tinybird Cloud.
@@ -753,14 +756,14 @@ You have access to the following tools:
 16. `diff_resource` - Diff the content of a resource in Tinybird Cloud vs Tinybird Local vs Project local file.
 17. `rename_datafile_or_fixture` - Rename a datafile or fixture.
 
-# When creating or updating datafiles:
-1. Use `plan` tool to plan the creation, update or rename of resources.
-2. If the user confirms the plan, go from 3 to 7 steps until all the resources are created, updated or skipped.
-3. Without asking, use the `create_datafile` tool to create the datafile, because it will ask for confirmation before creating the file.
-4. Check the result of the `create_datafile` tool to see if the datafile was created successfully.
-5. If the datafile was created successfully, report the result to the user.
-6. If the datafile was not created, finish the process and just wait for a new user prompt.
-7. If the datafile was created successfully, but the built failed, try to fix the error and repeat the process.
+# When creating, updating, or deleting files:
+1. Use `plan` tool to plan the creation, update, rename, or deletion of resources.
+2. If the user confirms the plan, go from 3 to 7 steps until all the resources are created, updated, deleted, or skipped.
+3. Without asking, use the `create_datafile` or `remove_file` tool to create or remove the file, because it will ask for confirmation before creating or removing the file.
+4. Check the result of the `create_datafile` or `remove_file` tool to see if the file was created or removed successfully.
+5. If the file was created or removed successfully, report the result to the user.
+6. If the file was not created or removed, finish the process and just wait for a new user prompt.
+7. If the file was created or removed successfully, but the build failed, try to fix the error and repeat the process.
 
 # When creating a landing datasource given a .ndjson file:
 - If the user does not specify anything about the desired schema, create a schema like this (sorting key not needed in this case)

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/tinybird/tb/modules/agent/tools/append.py

@@ -1,4 +1,5 @@
 import os
+from typing import Optional
 
 import click
 from pydantic_ai import RunContext
@@ -8,6 +9,7 @@ from tinybird.tb.modules.agent.utils import (
     TinybirdAgentContext,
     copy_fixture_to_project_folder_if_needed,
     show_confirmation,
+    show_env_options,
     show_input,
 )
 from tinybird.tb.modules.common import echo_safe_humanfriendly_tables_format_pretty_table
@@ -15,14 +17,14 @@ from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 
 def append_file(
-    ctx: RunContext[TinybirdAgentContext], datasource_name: str, fixture_pathname: str, cloud: bool = False
+    ctx: RunContext[TinybirdAgentContext], datasource_name: str, fixture_pathname: str, cloud: Optional[bool] = None
 ) -> str:
     """Append a fixture file to a datasource
 
     Args:
         datasource_name: Name of the datasource to append fixture to
         fixture_pathname: Path to the fixture file to append
-        cloud: Whether to append the fixture to the cloud or local environment. Optional.
+        cloud: Whether to append the fixture to the cloud or local environment. If None (user didn't specify), will ask user to clarify. Defaults to local (False) in dangerous skip permissions mode.
 
     Returns:
         str: Message indicating the success or failure of the appending
@@ -37,6 +39,19 @@ def append_file(
 
         fixture_path = fixture_path_or_error
         fixture_pathname = os.path.relpath(fixture_path, ctx.deps.folder)
+
+        # Handle cloud parameter - ask user if uncertain and not in dangerous skip mode
+        if cloud is None:
+            if ctx.deps.dangerously_skip_permissions:
+                # Default to local when in dangerous skip mode
+                cloud = False
+            else:
+                # Ask the user to choose execution mode
+                cloud = show_env_options(ctx)
+                if cloud is None:
+                    ctx.deps.thinking_animation.start()
+                    return "Append operation cancelled by user."
+
         cloud_or_local = "Cloud" if cloud else "Local"
         confirmation = show_confirmation(
             title=f"Append fixture {fixture_pathname} to datasource '{datasource_name}' in Tinybird {cloud_or_local}?",
@@ -69,20 +84,33 @@ def append_file(
 
 
 def append_url(
-    ctx: RunContext[TinybirdAgentContext], datasource_name: str, fixture_url: str, cloud: bool = False
+    ctx: RunContext[TinybirdAgentContext], datasource_name: str, fixture_url: str, cloud: Optional[bool] = None
 ) -> str:
     """Append existing fixture to a datasource
 
     Args:
         datasource_name: Name of the datasource to append fixture to
         fixture_url: external url to the fixture file to append
-        cloud: Whether to append the fixture to the cloud or local environment. Optional.
+        cloud: Whether to append the fixture to the cloud or local environment. If None (user didn't specify), will ask user to clarify. Defaults to local (False) in dangerous skip permissions mode.
 
     Returns:
         str: Message indicating the success or failure of the appending
     """
     try:
         ctx.deps.thinking_animation.stop()
+
+        # Handle cloud parameter - ask user if uncertain and not in dangerous skip mode
+        if cloud is None:
+            if ctx.deps.dangerously_skip_permissions:
+                # Default to local when in dangerous skip mode
+                cloud = False
+            else:
+                # Ask the user to choose execution mode
+                cloud = show_env_options(ctx)
+                if cloud is None:
+                    ctx.deps.thinking_animation.start()
+                    return "Append operation cancelled by user."
+
         cloud_or_local = "Cloud" if cloud else "Local"
         confirmation = show_confirmation(
             title=f"Append URL {fixture_url} to datasource '{datasource_name}' in Tinybird {cloud_or_local}?",
@@ -115,10 +143,13 @@ def append_url(
 
 
 def handle_quarantine_error(
-    ctx: RunContext[TinybirdAgentContext], error_message: str, datasource_name: str, cloud: bool = False
+    ctx: RunContext[TinybirdAgentContext], error_message: str, datasource_name: str, cloud: Optional[bool] = None
 ) -> str:
     try:
         if "in quarantine" in error_message:
+            # Default to local if cloud is None for error handling
+            if cloud is None:
+                cloud = False
             cloud_or_local = "Cloud" if cloud else "Local"
             click.echo(FeedbackManager.highlight(message=f"» Looking for errors in {datasource_name}_quarantine..."))
             query = f"select * from {datasource_name}_quarantine order by insertion_date desc limit 5 FORMAT JSON"

tinybird-0.0.1.dev271/tinybird/tb/modules/agent/tools/create_datafile.py → tinybird-0.0.1.dev273/tinybird/tb/modules/agent/tools/datafile.py

@@ -191,3 +191,81 @@ def rename_datafile_or_fixture(ctx: RunContext[TinybirdAgentContext], path: str,
         click.echo(FeedbackManager.error(message=e))
         ctx.deps.thinking_animation.start()
         return f"Error renaming {path} to {new_path}: {e}"
+
+
+def remove_file(ctx: RunContext[TinybirdAgentContext], path: str) -> str:
+    """Removes a datafile or fixture from the project folder and rebuilds the project if needed
+
+    Args:
+        path (str): The path to the file to remove. Required.
+
+    Returns:
+        str: If the resource was removed successfully.
+    """
+    try:
+        ctx.deps.thinking_animation.stop()
+        path = path.removeprefix("/")
+        full_path = Path(ctx.deps.folder) / path
+
+        if not full_path.exists():
+            click.echo(FeedbackManager.error(message=f"Error: File {path} not found"))
+            ctx.deps.thinking_animation.start()
+            return f"Error: File {path} not found (double check the file path)"
+
+        confirmation = show_confirmation(
+            title=f"Delete '{path}'?",
+            skip_confirmation=ctx.deps.dangerously_skip_permissions,
+        )
+
+        if confirmation == "review":
+            feedback = show_input(ctx.deps.workspace_name)
+            ctx.deps.thinking_animation.start()
+            return f"User did not confirm the proposed changes and gave the following feedback: {feedback}"
+
+        click.echo(FeedbackManager.highlight(message=f"» Removing {path}..."))
+
+        # Check if it's a datafile that requires project rebuild
+        is_datafile = full_path.suffix in (".connection", ".datasource", ".pipe")
+
+        # Remove the file
+        full_path.unlink()
+
+        # Check for corresponding .sql file (for fixtures)
+        sql_file_path = full_path.with_suffix(".sql")
+        sql_file_removed = False
+        if sql_file_path.exists():
+            sql_file_path.unlink()
+            sql_file_removed = True
+
+        # Rebuild project if it's a datafile
+        if is_datafile:
+            ctx.deps.build_project(test=False, silent=True, load_fixtures=False)
+
+        success_message = f"✓ {path} removed"
+        if sql_file_removed:
+            success_message += f" (and {sql_file_path.name})"
+
+        click.echo(FeedbackManager.success(message=success_message))
+        ctx.deps.thinking_animation.start()
+
+        result_message = f"Removed {path}"
+        if sql_file_removed:
+            result_message += f" and {sql_file_path.name}"
+        if is_datafile:
+            result_message += ". Project built successfully."
+        else:
+            result_message += "."
+
+        return result_message
+    except AgentRunCancelled as e:
+        raise e
+    except CLIBuildException as e:
+        ctx.deps.thinking_animation.stop()
+        click.echo(FeedbackManager.error(message=e))
+        ctx.deps.thinking_animation.start()
+        return f"Error building project: {e}. If the error is related to another resource, fix it and try again."
+    except Exception as e:
+        ctx.deps.thinking_animation.stop()
+        click.echo(FeedbackManager.error(message=e))
+        ctx.deps.thinking_animation.start()
+        return f"Error removing {path}: {e}"

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/tinybird/tb/modules/agent/tools/deploy.py

@@ -6,13 +6,28 @@ from tinybird.tb.modules.exceptions import CLIDeploymentException
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 
-def deploy(ctx: RunContext[TinybirdAgentContext]) -> str:
-    """Deploy the project"""
+def deploy(ctx: RunContext[TinybirdAgentContext], allow_destructive_operations: bool = False) -> str:
+    """Deploy the project
+
+    Args:
+        allow_destructive_operations (bool): Set to true if a datasource, pipe or connection file has been deleted locally.
+                                           Optional. Default is False.
+
+    Returns:
+        str: The result of the deployment
+    """
     try:
         ctx.deps.thinking_animation.stop()
+
+        if allow_destructive_operations:
+            click.echo(
+                FeedbackManager.warning(message="Destructive operations flag is enabled due to a file deleted recently")
+            )
+
+        click.echo()
         confirmation = show_confirmation(
             title="Deploy the project?",
-            skip_confirmation=ctx.deps.dangerously_skip_permissions,
+            skip_confirmation=False,
         )
 
         if confirmation == "review":
@@ -21,7 +36,7 @@ def deploy(ctx: RunContext[TinybirdAgentContext]) -> str:
             return f"User did not confirm deployment and gave the following feedback: {feedback}"
 
         click.echo(FeedbackManager.highlight(message="» Deploying project..."))
-        ctx.deps.deploy_project()
+        ctx.deps.deploy_project(allow_destructive_operations=allow_destructive_operations)
         click.echo(FeedbackManager.success(message="✓ Project deployed successfully"))
         ctx.deps.thinking_animation.start()
         return "Project deployed successfully"

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/tinybird/tb/modules/agent/tools/execute_query.py

@@ -35,7 +35,7 @@ def execute_query(
     ctx: RunContext[TinybirdAgentContext],
     query: str,
     task: str,
-    cloud: bool = False,
+    cloud: Optional[bool] = None,
     script: Optional[str] = None,
     export_format: Optional[str] = None,
 ):
@@ -44,7 +44,7 @@ def execute_query(
     Args:
         query (str): The query to execute. Required.
         task (str): The purpose of the query. Required.
-        cloud (bool): Whether to execute the query on cloud or local. Optional.
+        cloud (bool): Whether to execute the query on cloud or local. If None (user didn't specify), will ask user to clarify. Defaults to local (False) in dangerous skip permissions mode.
         script (str): Python script using plotext to render the query results as a chart. The script will have access to 'data' (list of dicts), 'meta' (list of column info dicts), 'terminal_width' and 'terminal_height' variables. Always use plt.theme("clear") for transparent background and plt.plot_size(terminal_width, terminal_height) for proper sizing. For bar charts, use the simple versions: plt.simple_bar(), plt.simple_multiple_bar(), and plt.simple_stacked_bar(). Optional.
         export_format (str): The format to export the query results to. Options: csv, json, ndjson. Optional.
 
@@ -61,6 +61,19 @@ def execute_query(
             if query.lower().startswith(forbidden_command):
                 return f"Error executing query: {forbidden_command} is not allowed."
 
+        # Handle cloud parameter - ask user if uncertain and not in dangerous skip mode
+        if cloud is None:
+            if ctx.deps.dangerously_skip_permissions:
+                # Default to local when in dangerous skip mode
+                cloud = False
+            else:
+                # Ask the user to choose execution mode
+                from tinybird.tb.modules.agent.utils import show_env_options
+
+                cloud = show_env_options(ctx)
+                if cloud is None:
+                    return "Query execution cancelled by user."
+
         cloud_or_local = "cloud" if cloud else "local"
         ctx.deps.thinking_animation.stop()
 

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/tinybird/tb/modules/agent/tools/get_endpoint_stats.py

@@ -1,19 +1,21 @@
+from typing import Optional
+
 import click
 from pydantic_ai import RunContext
 
-from tinybird.tb.modules.agent.utils import TinybirdAgentContext
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_env_options
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 
 def get_endpoint_stats(
-    ctx: RunContext[TinybirdAgentContext], endpoint_name: str, interval_days: int = 1, cloud: bool = False
+    ctx: RunContext[TinybirdAgentContext], endpoint_name: str, interval_days: int = 1, cloud: Optional[bool] = None
 ):
     """Get stats for an endpoint:
 
     Args:
         endpoint_name (str): The name of the endpoint to get stats for. Required.
         interval_days (int): The number of days to get stats for. Optional.
-        cloud (bool): Whether to get stats from cloud or local. Optional.
+        cloud (bool): Whether to get stats from cloud or local. If None (user didn't specify), will ask user to clarify. Defaults to local (False) in dangerous skip permissions mode.
 
     Returns:
         str: The result of the stats.
@@ -25,6 +27,17 @@ def get_endpoint_stats(
         pipe_stats = "tinybird.pipe_stats"
         date_column = "date"
 
+    # Handle cloud parameter - ask user if uncertain and not in dangerous skip mode
+    if cloud is None:
+        if ctx.deps.dangerously_skip_permissions:
+            # Default to local when in dangerous skip mode
+            cloud = False
+        else:
+            # Ask the user to choose execution mode
+            cloud = show_env_options(ctx)
+            if cloud is None:
+                return "Get endpoint stats operation cancelled by user."
+
     days = "day" if interval_days == 1 else "days"
     cloud_or_local = "cloud" if cloud else "local"
     ctx.deps.thinking_animation.stop()

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/tinybird/tb/modules/agent/tools/get_openapi_definition.py

@@ -1,22 +1,35 @@
+from typing import Optional
+
 import click
 import requests
 from pydantic_ai import RunContext
 
-from tinybird.tb.modules.agent.utils import TinybirdAgentContext
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_env_options
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 
-def get_openapi_definition(ctx: RunContext[TinybirdAgentContext], endpoint_name: str, cloud: bool = False):
+def get_openapi_definition(ctx: RunContext[TinybirdAgentContext], endpoint_name: str, cloud: Optional[bool] = None):
     """Get the OpenAPI definition for an endpoint:
 
     Args:
         endpoint_name (str): The name of the endpoint to get the OpenAPI definition for. Required.
-        cloud (bool): Whether to get the OpenAPI definition from cloud or local. Optional.
+        cloud (bool): Whether to get the OpenAPI definition from cloud or local. If None (user didn't specify), will ask user to clarify. Defaults to local (False) in dangerous skip permissions mode.
 
     Returns:
         str: The OpenAPI definition for the endpoint.
     """
 
+    # Handle cloud parameter - ask user if uncertain and not in dangerous skip mode
+    if cloud is None:
+        if ctx.deps.dangerously_skip_permissions:
+            # Default to local when in dangerous skip mode
+            cloud = False
+        else:
+            # Ask the user to choose execution mode
+            cloud = show_env_options(ctx)
+            if cloud is None:
+                return "Get OpenAPI definition operation cancelled by user."
+
     cloud_or_local = "Cloud" if cloud else "Local"
     ctx.deps.thinking_animation.stop()
 

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/tinybird/tb/modules/agent/tools/request_endpoint.py

@@ -4,7 +4,7 @@ import click
 import humanfriendly
 from pydantic_ai import RunContext
 
-from tinybird.tb.modules.agent.utils import TinybirdAgentContext, limit_result_output
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext, limit_result_output, show_env_options
 from tinybird.tb.modules.common import echo_safe_humanfriendly_tables_format_pretty_table
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 
@@ -13,19 +13,29 @@ def request_endpoint(
     ctx: RunContext[TinybirdAgentContext],
     endpoint_name: str,
     params: Optional[dict[str, str]] = None,
-    cloud: bool = False,
+    cloud: Optional[bool] = None,
 ):
     """Request an endpoint:
 
     Args:
         endpoint_name (str): The name of the endpoint to request. Required.
         params (dict): The parameters to pass to the endpoint. Optional.
-        cloud (bool): Whether to request the endpoint on cloud or local. Optional.
+        cloud (bool): Whether to request the endpoint on cloud or local. If None (user didn't specify), will ask user to clarify. Defaults to local (False) in dangerous skip permissions mode.
 
     Returns:
         str: The result of the query.
     """
     try:
+        # Handle cloud parameter - ask user if uncertain and not in dangerous skip mode
+        if cloud is None:
+            if ctx.deps.dangerously_skip_permissions:
+                # Default to local when in dangerous skip mode
+                cloud = False
+            else:
+                # Ask the user to choose execution mode
+                cloud = show_env_options(ctx)
+                if cloud is None:
+                    return "Endpoint request cancelled by user."
         cloud_or_local = "cloud" if cloud else "local"
         ctx.deps.thinking_animation.stop()
         with_params = f" with params {params}" if params else ""

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/tinybird/tb/modules/agent/utils.py

@@ -43,7 +43,7 @@ class TinybirdAgentContext(BaseModel):
     explore_data: Callable[[str], str]
     build_project: Callable[..., None]
     build_project_test: Callable[..., None]
-    deploy_project: Callable[[], None]
+    deploy_project: Callable[..., None]
     deploy_check_project: Callable[[], None]
     mock_data: Callable[..., list[dict[str, Any]]]
     append_data_local: Callable[..., None]
@@ -743,6 +743,30 @@ def show_confirmation(title: str, skip_confirmation: bool = False, show_review:
         raise AgentRunCancelled(f"User cancelled the operation: {title}")
 
 
+def show_env_options(ctx: "RunContext[TinybirdAgentContext]") -> Optional[bool]:
+    """Show environment options for user to choose between local and cloud.
+
+    Args:
+        ctx: The run context containing TinybirdAgentContext
+
+    Returns:
+        bool: True for cloud, False for local, None if cancelled
+    """
+
+    ctx.deps.thinking_animation.stop()
+    click.echo(FeedbackManager.highlight(message="» Agent is uncertain about target environment"))
+
+    choice = show_options(
+        options=["Tinybird Local", f"Tinybird Cloud ({ctx.deps.host})"], title="Where should this query be executed?"
+    )
+    ctx.deps.thinking_animation.start()
+
+    if choice is None:
+        return None
+
+    return choice.startswith("Tinybird Cloud")
+
+
 def copy_fixture_to_project_folder_if_needed(
     ctx: RunContext[TinybirdAgentContext], fixture_pathname: str
 ) -> Union[Path, str]:

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/tinybird/tb/modules/connection.py

@@ -121,7 +121,7 @@ def connection_ls(ctx: Context, service: Optional[DataConnectorType] = None) ->
     obj: Dict[str, Any] = ctx.ensure_object(dict)
     client: TinyB = obj["client"]
 
-    connections = client.connections(connector=service, skip_bigquery=True)
+    connections = client.connections(connector=service)
     columns = []
     table = []
 

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/tinybird/tb/modules/datafile/pull.py

@@ -89,7 +89,7 @@ def folder_pull(
     try:
         datasources = client.datasources()
         pipes = client.pipes()
-        connections = client.connections(skip_bigquery=True)
+        connections = client.connections()
 
         write_files(
             resources=datasources,

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/tinybird.egg-info/PKG-INFO

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

{tinybird-0.0.1.dev271 → tinybird-0.0.1.dev273}/tinybird.egg-info/SOURCES.txt

@@ -95,7 +95,7 @@ tinybird/tb/modules/agent/tools/__init__.py
 tinybird/tb/modules/agent/tools/analyze.py
 tinybird/tb/modules/agent/tools/append.py
 tinybird/tb/modules/agent/tools/build.py
-tinybird/tb/modules/agent/tools/create_datafile.py
+tinybird/tb/modules/agent/tools/datafile.py
 tinybird/tb/modules/agent/tools/deploy.py
 tinybird/tb/modules/agent/tools/deploy_check.py
 tinybird/tb/modules/agent/tools/diff_resource.py