tinybird 0.0.1.dev259__py3-none-any.whl → 0.0.1.dev260__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.dev259'
-__revision__ = '5b5805f'
+__version__ = '0.0.1.dev260'
+__revision__ = 'bd0c45e'

tinybird/tb/modules/agent/agent.py

@@ -17,11 +17,11 @@ from tinybird.tb.modules.agent.animations import ThinkingAnimation
 from tinybird.tb.modules.agent.banner import display_banner
 from tinybird.tb.modules.agent.memory import clear_history, clear_messages, load_messages, save_messages
 from tinybird.tb.modules.agent.models import create_model, model_costs
-from tinybird.tb.modules.agent.prompts import agent_system_prompt, resources_prompt
+from tinybird.tb.modules.agent.prompts import agent_system_prompt, load_custom_project_rules, resources_prompt
 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
+from tinybird.tb.modules.agent.tools.create_datafile import create_datafile, 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
@@ -75,6 +75,12 @@ class TinybirdAgent:
             tools=[
                 Tool(preview_datafile, docstring_format="google", require_parameter_descriptions=True, takes_ctx=False),
                 Tool(create_datafile, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
+                Tool(
+                    rename_datafile_or_fixture,
+                    docstring_format="google",
+                    require_parameter_descriptions=True,
+                    takes_ctx=True,
+                ),
                 Tool(plan, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
                 Tool(build, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
                 Tool(deploy, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
@@ -170,7 +176,7 @@ class TinybirdAgent:
         )
 
     def run(self, user_prompt: str, config: dict[str, Any]) -> None:
-        user_prompt = f"{user_prompt}\n\n{resources_prompt(self.project)}"
+        user_prompt = f"{user_prompt}\n\n{load_custom_project_rules(self.project.folder)}"
         self.thinking_animation.start()
         result = self.agent.run_sync(
             user_prompt,
@@ -185,7 +191,7 @@ class TinybirdAgent:
         self._echo_usage(config, result)
 
     async def run_iter(self, user_prompt: str, config: dict[str, Any]) -> None:
-        user_prompt = f"{user_prompt}\n\n"
+        user_prompt = f"{user_prompt}\n\n{load_custom_project_rules(self.project.folder)}"
         self.thinking_animation.start()
         deps = self._build_agent_deps(config)
 

tinybird/tb/modules/agent/prompts.py

@@ -568,9 +568,10 @@ You have access to the following tools:
 13. `request_endpoint` - Request an endpoint against Tinybird Cloud or Local.
 14. `diff_resource` - Diff the content of a resource in Tinybird Cloud vs Tinybird Local vs Project local file.
 15. `create_tests` - Create tests for an endpoint.
+16. `rename_datafile_or_fixture` - Rename a datafile or fixture.
 
 # When creating or updating datafiles:
-1. Use `plan` tool to plan the creation or update of resources.
+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. Use `preview_datafile` tool to preview the content of a datafile.
 4. Without asking, use the `create_datafile` tool to create the datafile, because it will ask for confirmation before creating the file.
@@ -674,3 +675,12 @@ GCS: {gcs_connection_example}
 # Info
 Today is {datetime.now().strftime("%Y-%m-%d")}
 """
+
+
+def load_custom_project_rules(folder: str) -> str:
+    tinybird_rules = Path(folder).joinpath("TINYBIRD.md")
+
+    if not tinybird_rules.exists():
+        return ""
+
+    return f"# Custom Project Rulesd defined by the user\n\n{tinybird_rules.read_text()}"

tinybird/tb/modules/agent/tools/create_datafile.py

@@ -122,3 +122,59 @@ def create_file(ctx: RunContext[TinybirdAgentContext], path: str, content: str):
         f.write(content)
 
     return "ok"
+
+
+def rename_datafile_or_fixture(ctx: RunContext[TinybirdAgentContext], path: str, new_path: str):
+    """Renames a datafile or fixture.
+
+    Args:
+        path (str): The path to the file to rename. Required.
+        new_path (str): The new path to the file. Required.
+    """
+    try:
+        ctx.deps.thinking_animation.stop()
+        confirmation = show_confirmation(
+            title=f"Rename '{path}' to '{new_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"» Renaming file {path} to {new_path}..."))
+        new_path_full = Path(ctx.deps.folder) / new_path.removeprefix("/")
+
+        if new_path_full.exists():
+            click.echo(FeedbackManager.error(message=f"Error: File {new_path} already exists"))
+            ctx.deps.thinking_animation.start()
+            return f"Error: File {new_path} already exists"
+
+        parent_path = new_path_full.parent
+        parent_path.mkdir(parents=True, exist_ok=True)
+        os.rename(Path(ctx.deps.folder) / path.removeprefix("/"), new_path_full)
+        is_datafile = (".connection", ".datasource", ".pipe")
+
+        if new_path_full.suffix in is_datafile:
+            ctx.deps.build_project(test=False, silent=True)
+
+        click.echo(FeedbackManager.success(message=f"✓ {new_path} created"))
+        ctx.deps.thinking_animation.start()
+        return f"Renamed file from {path} to {new_path}"
+    except AgentRunCancelled as e:
+        raise e
+    except FileNotFoundError:
+        ctx.deps.thinking_animation.start()
+        click.echo(FeedbackManager.error(message=f"Error: File {path} not found"))
+        return f"Error: File {path} not found (double check the file path)"
+    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}"
+    except Exception as e:
+        ctx.deps.thinking_animation.stop()
+        click.echo(FeedbackManager.error(message=e))
+        ctx.deps.thinking_animation.start()
+        return f"Error renaming {path} to {new_path}: {e}"

{tinybird-0.0.1.dev259.dist-info → tinybird-0.0.1.dev260.dist-info}/METADATA

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

{tinybird-0.0.1.dev259.dist-info → tinybird-0.0.1.dev260.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=D0G5HjONd9vfaiba7rVUC32Pic_cEL9FIR6-Z3uB_2c,247
+tinybird/tb/__cli__.py,sha256=GooE8LYwW5AvSc5ky2GJcqRgW4eIKUmmqimo2SYppqE,247
 tinybird/tb/check_pypi.py,sha256=Gp0HkHHDFMSDL6nxKlOY51z7z1Uv-2LRexNTZSHHGmM,552
 tinybird/tb/cli.py,sha256=FdDFEIayjmsZEVsVSSvRiVYn_FHOVg_zWQzchnzfWho,1008
 tinybird/tb/client.py,sha256=pJbdkWMXGAqKseNAvdsRRnl_c7I-DCMB0dWCQnG82nU,54146
@@ -69,18 +69,18 @@ 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=1EGRxcIa3QTIO4EXrGAs8svBm4vlm3DiVCZEF5k7mFw,23773
+tinybird/tb/modules/agent/agent.py,sha256=r8mreWX8CdvzKp_BSCvWJHGQ5xcKee2aPVeOcLbDeMo,24121
 tinybird/tb/modules/agent/animations.py,sha256=4WOC5_2BracttmMCrV0H91tXfWcUzQHBUaIJc5FA7tE,3490
 tinybird/tb/modules/agent/banner.py,sha256=7f97PeCPW-oW9mReQn3D0by8mnDhoc0VbfebEPRPI7c,3070
 tinybird/tb/modules/agent/memory.py,sha256=O6Kumn9AyKxcTkhI45yjAUZ3ZIAibLOcNWoiEuLYeqY,3245
 tinybird/tb/modules/agent/models.py,sha256=LW1D27gjcd_jwFmghEzteCgToDfodX2B6B5S8BYbysw,735
-tinybird/tb/modules/agent/prompts.py,sha256=-qks8FvYb8_WpchyjjWi9k8tQEYTjX9fEynuFrFLbzA,27727
+tinybird/tb/modules/agent/prompts.py,sha256=pSjEeXzfN5_bFl0bF1mHcVva0A4mKBGXA7nBUUkVnGU,28057
 tinybird/tb/modules/agent/utils.py,sha256=tVymkZdDINcp5iLPrjEWEsoGmj4SjNbvgsXcYqsMrhU,26888
 tinybird/tb/modules/agent/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tinybird/tb/modules/agent/tools/analyze.py,sha256=fTKjMCq8ckvvqq8WqpE2aE_gO0XBM-eFPt75WTFC3ik,3565
 tinybird/tb/modules/agent/tools/append.py,sha256=b8nCO788MAk1ahITdNrrzMdUDhj-Y6KBwOCj4UmCQxg,5606
 tinybird/tb/modules/agent/tools/build.py,sha256=iSsdq42jpP8PC4SOUOLHYzuDrmqU3MDBYwhrT44tBFk,825
-tinybird/tb/modules/agent/tools/create_datafile.py,sha256=twsURhfIhQ5sjTwvsxR0rsERjfIcYxwDZpVNWG6MBPI,4423
+tinybird/tb/modules/agent/tools/create_datafile.py,sha256=habaRw-PXGFMuyWgqwcm4Lg4ataYhkyebmPUoCl10p8,6860
 tinybird/tb/modules/agent/tools/deploy.py,sha256=2hKj6LMYiDea6ventsBjE6ArECGIryTdo3X-LYo5oZI,1248
 tinybird/tb/modules/agent/tools/deploy_check.py,sha256=2Wr9hQfKPlhqhumOv5TNl_xFctvdq_DHZ2dI2h_LggY,1048
 tinybird/tb/modules/agent/tools/diff_resource.py,sha256=_9xHcDzCTKk_E1wKQbuktVqV6U9sA0kqYaBxWvtliX0,2613
@@ -113,8 +113,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.dev259.dist-info/METADATA,sha256=WTSSsjGRnfYv-YUjstmzMF58_YHbuYjg8OWOoghGaVg,1733
-tinybird-0.0.1.dev259.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev259.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev259.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev259.dist-info/RECORD,,
+tinybird-0.0.1.dev260.dist-info/METADATA,sha256=BREz0wuoONs5NhcgFYDlZXBWfTF3MLyADgy3eF6X168,1733
+tinybird-0.0.1.dev260.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev260.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev260.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev260.dist-info/RECORD,,