tinybird 0.0.1.dev255__py3-none-any.whl → 0.0.1.dev257__py3-none-any.whl

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

@@ -15,13 +15,13 @@ def plan(ctx: RunContext[TinybirdAgentContext], plan: str) -> str:
     """
     try:
         ctx.deps.thinking_animation.stop()
+        plan = plan.strip()
         click.echo(plan)
         confirmation = show_confirmation(
-            title="Do you want to implement the plan?", skip_confirmation=ctx.deps.dangerously_skip_permissions
+            title="Do you want to continue with the plan?", skip_confirmation=ctx.deps.dangerously_skip_permissions
         )
 
         if confirmation == "review":
-            click.echo()
             feedback = show_input(ctx.deps.workspace_name)
             ctx.deps.thinking_animation.start()
             return f"User did not confirm the proposed plan and gave the following feedback: {feedback}"

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

@@ -13,7 +13,7 @@ def request_endpoint(
     ctx: RunContext[TinybirdAgentContext],
     endpoint_name: str,
     params: Optional[dict[str, str]] = None,
-    cloud: bool = True,
+    cloud: bool = False,
 ):
     """Request an endpoint:
 
@@ -28,10 +28,10 @@ def request_endpoint(
     try:
         cloud_or_local = "cloud" if cloud else "local"
         ctx.deps.thinking_animation.stop()
-
+        with_params = f" with params {params}" if params else ""
         click.echo(
             FeedbackManager.highlight(
-                message=f"» Calling endpoint {endpoint_name} in {cloud_or_local} environment with params {params}"
+                message=f"» Calling endpoint {endpoint_name} in {cloud_or_local} environment{with_params}"
             )
         )
 
@@ -59,7 +59,8 @@ def request_endpoint(
         ctx.deps.thinking_animation.stop()
         click.echo(FeedbackManager.error(message=error))
         ctx.deps.thinking_animation.start()
-        if "not found" in error.lower() and cloud:
+        not_found_errors = ["not found", "does not exist"]
+        if any(not_found_error in error.lower() for not_found_error in not_found_errors) and cloud:
             return f"Error executing query: {error}. Please run the query against Tinybird local instead of cloud."
         else:
             return f"Error executing query: {error}. Please try again."

tinybird/tb/modules/agent/utils.py

@@ -39,7 +39,8 @@ class TinybirdAgentContext(BaseModel):
     deploy_project: Callable[[], None]
     deploy_check_project: Callable[[], None]
     mock_data: Callable[..., list[dict[str, Any]]]
-    append_data: Callable[..., None]
+    append_data_local: Callable[..., None]
+    append_data_cloud: Callable[..., None]
     analyze_fixture: Callable[..., dict[str, Any]]
     execute_query_cloud: Callable[..., dict[str, Any]]
     execute_query_local: Callable[..., dict[str, Any]]
@@ -55,6 +56,8 @@ class TinybirdAgentContext(BaseModel):
     token: str
     user_token: str
     host: str
+    local_host: str
+    local_token: str
 
 
 default_style = PromptStyle.from_dict(
@@ -141,7 +144,7 @@ class InquirerControl(FormattedTextControl):
                 tokens.append(
                     (
                         "",
-                        " \u276f " if selected else "   ",
+                        "\u276f " if selected else "  ",
                     )
                 )
                 if selected:
@@ -196,7 +199,7 @@ def prompt_question(message, **kwargs):
     def get_prompt_tokens():
         tokens = []
 
-        tokens.append(("class:question", " %s " % message))
+        tokens.append(("class:question", "%s " % message))
         if ic.answered:
             tokens.append(("class:answer", " " + ic.get_selection()[0]))
         else:

tinybird/tb/modules/cli.py

@@ -87,8 +87,10 @@ agent_mode_flag = os.environ.get("TB_AGENT_MODE", "false") == "true"
     hidden=True,
 )
 @click.option(
-    "--p",
-    help="Run agent in print mode with the provided prompt and exit.",
+    "--prompt",
+    "-p",
+    help="Run agent in prompt mode with the provided input and exit.",
+    hidden=True,
 )
 @click.version_option(version=VERSION)
 @click.pass_context
@@ -105,7 +107,7 @@ def cli(
     output: str,
     max_depth: int,
     dangerously_skip_permissions: bool,
-    p: Optional[str],
+    prompt: Optional[str] = None,
 ) -> None:
     """
     Use `OBFUSCATE_REGEX_PATTERN` and `OBFUSCATE_PATTERN_SEPARATOR` environment variables to define a regex pattern and a separator (in case of a single string with multiple regex) to obfuscate secrets in the CLI output.
@@ -166,7 +168,7 @@ def cli(
         user_token = os.environ.get("TB_USER_TOKEN", "")
 
     config = get_config(host, token, user_token=user_token, config_file=config_temp._path)
-    client = _get_tb_client(config.get("token", None), config["host"])
+    client = _get_tb_client(config.get("token", ""), config["host"])
 
     # Calculate project folder path properly
     tinyb_dir = os.path.dirname(config_temp._path)  # Directory containing .tinyb file
@@ -207,10 +209,10 @@ def cli(
     ctx.ensure_object(dict)["output"] = output
 
     is_agent_mode = agent_mode_flag and ctx.invoked_subcommand is None
-    is_print_mode = p is not None
+    is_prompt_mode = prompt is not None
 
-    if is_agent_mode or is_print_mode:
-        run_agent(config, project, dangerously_skip_permissions, prompt=p)
+    if is_agent_mode or is_prompt_mode:
+        run_agent(config, project, dangerously_skip_permissions, prompt=prompt)
 
 
 @cli.command(hidden=True)

{tinybird-0.0.1.dev255.dist-info → tinybird-0.0.1.dev257.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev255
+Version: 0.0.1.dev257
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird
@@ -21,7 +21,7 @@ Requires-Dist: GitPython~=3.1.32
 Requires-Dist: humanfriendly~=8.2
 Requires-Dist: prompt_toolkit==3.0.48
 Requires-Dist: pydantic~=2.11.7
-Requires-Dist: pydantic-ai-slim[anthropic]~=0.3.1
+Requires-Dist: pydantic-ai-slim[anthropic]~=0.4.2
 Requires-Dist: pyperclip==1.8.2
 Requires-Dist: pyyaml<6.1,>=6.0
 Requires-Dist: requests<3,>=2.28.1

{tinybird-0.0.1.dev255.dist-info → tinybird-0.0.1.dev257.dist-info}/RECORD

@@ -12,12 +12,12 @@ tinybird/syncasync.py,sha256=IPnOx6lMbf9SNddN1eBtssg8vCLHMt76SuZ6YNYm-Yk,27761
 tinybird/tornado_template.py,sha256=jjNVDMnkYFWXflmT8KU_Ssbo5vR8KQq3EJMk5vYgXRw,41959
 tinybird/ch_utils/constants.py,sha256=fPgZtwbr1ymxaW7uqVWHKmAbt7uGj3SxCCS3xsEMJqA,4151
 tinybird/ch_utils/engine.py,sha256=4X1B-iuhdW_mxKnX_m3iCsxgP9RPVgR75g7yH1vsJ6A,40851
-tinybird/datafile/common.py,sha256=J0Oydru3Nh3oCx66EEEHGxm2-r5_NfxvZxiFhBjgAH0,98428
+tinybird/datafile/common.py,sha256=naXznRSAceD0T05YNj0Fdk7k-6WfKA6dz2GT0tiHj9k,98505
 tinybird/datafile/exceptions.py,sha256=8rw2umdZjtby85QbuRKFO5ETz_eRHwUY5l7eHsy1wnI,556
 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=51Rg-OvKFWGEJgEjFVuc2XEUi5XT3bb5o8ZEUJ1bdhA,247
+tinybird/tb/__cli__.py,sha256=938aavDUj8x4pLfFvmzos4pwkLQIFF_1swhbm0m5fi4,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
@@ -25,7 +25,7 @@ tinybird/tb/config.py,sha256=mhMTGnMB5KcxGoh3dewIr2Jjsa6pHE183gCPAQWyp6o,3973
 tinybird/tb/modules/build.py,sha256=efD-vamK1NPaDo9R86Hn8be2DYoW0Hh5bZiH7knK5dk,7790
 tinybird/tb/modules/build_common.py,sha256=rWhemU8bk0ZE2eiwZDaTmV9cPabDGGlyc2WnRxfhT0M,12859
 tinybird/tb/modules/cicd.py,sha256=0KLKccha9IP749QvlXBmzdWv1On3mFwMY4DUcJlBxiE,7326
-tinybird/tb/modules/cli.py,sha256=1kErLFhxgMWldbE7P4-bkPUcug8ymRyXDHRG9-vGb_4,16755
+tinybird/tb/modules/cli.py,sha256=yLRYSk0t2WuyzBclb6HLEXHdRN8ktvypemWiTy5VmDc,16809
 tinybird/tb/modules/common.py,sha256=tj6DR2yOqMMQ0PILwFGXmMogxdrbQCgj36HdSM611rs,82657
 tinybird/tb/modules/config.py,sha256=gK7rgaWTDd4ZKCrNEg_Uemr26EQjqWt6TjyQKujxOws,11462
 tinybird/tb/modules/connection.py,sha256=-MY56NUAai6EMC4-wpi7bT0_nz_SA8QzTmHkV7HB1IQ,17810
@@ -68,29 +68,29 @@ 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=HtXB75QJ71YH21BdbWhx2Pb1c2KUFu1_3TfdVulCoh4,25349
+tinybird/tb/modules/agent/agent.py,sha256=_xH4No8K0cRWTRyLBR5M5p9lt-nNsPszhhQ3uaGYuN8,22264
 tinybird/tb/modules/agent/animations.py,sha256=4WOC5_2BracttmMCrV0H91tXfWcUzQHBUaIJc5FA7tE,3490
 tinybird/tb/modules/agent/banner.py,sha256=KX_e467uiy1gWOZ4ofTZt0GCFGQqHQ_8Ob27XLQqda0,3053
-tinybird/tb/modules/agent/memory.py,sha256=H6SJK--2L5C87B7AJd_jMqsq3sCvFvZwZXmajuT0GBE,1171
+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=qyAv3H1x9qctlYQSel0DHxLlRJM2_8HTg7M-foSoR0k,17567
-tinybird/tb/modules/agent/utils.py,sha256=5mUnc4LZATHLzQZThotNErzZdHQDwK3eur6W4NZULWA,26561
+tinybird/tb/modules/agent/prompts.py,sha256=ee0EGMIhcxDF2-0uAtI2ApsDk1AM_0vLTx3LZfjX760,26122
+tinybird/tb/modules/agent/utils.py,sha256=FEg9-TVCa4VN5wfz568pxO9xjgRQL94AJbPECdyvBsk,26648
 tinybird/tb/modules/agent/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tinybird/tb/modules/agent/tools/analyze.py,sha256=7oxJ3waCS24Qja_k5GUB59_XiHTG9pCewZogOXhH0cA,3495
-tinybird/tb/modules/agent/tools/append.py,sha256=ekG7OxOzPjjjAjYUONAfp_PWlzyLT1GDfQhECgO-DYY,5065
-tinybird/tb/modules/agent/tools/build.py,sha256=LhzJMx6tbxC7gogIrxhfKJc-SDgoSR-FC6IunfaCdn8,758
-tinybird/tb/modules/agent/tools/create_datafile.py,sha256=wcPcChACTIFKw0lKFTlhm0sWJKhQkMLPLnGNpKyeETA,2962
-tinybird/tb/modules/agent/tools/deploy.py,sha256=WrsSlaufKGOBx0S13uoMQQH2DnKue5LQ231Rx4RXh2I,1443
-tinybird/tb/modules/agent/tools/deploy_check.py,sha256=LBE8aENYvCEaxbVTVKMSI2NGGiHeh-y60_MPlfrcvFk,1331
-tinybird/tb/modules/agent/tools/diff_resource.py,sha256=euKo_mD9FZHC1X6yESv3D_CXMzBMB8EWHLeZjIRqjJg,2637
-tinybird/tb/modules/agent/tools/execute_query.py,sha256=hMRahWZkyP9qa-nMGzY0Z7MjDa9sb5O9V1evmA_V_w8,2702
+tinybird/tb/modules/agent/tools/append.py,sha256=4SSW1e8_2IMjXbzg8a7ljPXuhd6ugUv-3qZfT2QWZBk,5884
+tinybird/tb/modules/agent/tools/build.py,sha256=wcndqnxwK4lDbL_3vTdoSNXiD5P4YG_CPWykj5lInuE,756
+tinybird/tb/modules/agent/tools/create_datafile.py,sha256=QeL_1uS6_uDCIQjn94KAVqKhnyYp5YZD4ZmjXLbKwjM,4530
+tinybird/tb/modules/agent/tools/deploy.py,sha256=2rbmDOczXERvjPUJ7s3X9GWZuOpsPeBabyTcG8D3fu8,1320
+tinybird/tb/modules/agent/tools/deploy_check.py,sha256=u_l2iTrwUJDXe-M0IpgA672CAo_GWlCrdRcuUKXIC_Q,1132
+tinybird/tb/modules/agent/tools/diff_resource.py,sha256=_9xHcDzCTKk_E1wKQbuktVqV6U9sA0kqYaBxWvtliX0,2613
+tinybird/tb/modules/agent/tools/execute_query.py,sha256=h8be9_lhyMy7UZoCWrExigP-cbq3lLnlLHD9SzDGBCM,2703
 tinybird/tb/modules/agent/tools/explore.py,sha256=ihALc_kBcsjrKT3hZyicqyIowB0g_K3AtNNi-5uz9-8,412
-tinybird/tb/modules/agent/tools/get_endpoint_stats.py,sha256=E9yPi9LwnpsTyjFd8EaiSNvDGVPkFSNqp_tZxg_pWqs,1705
-tinybird/tb/modules/agent/tools/get_openapi_definition.py,sha256=9cQ-SUeB1NVhPJN1s8aQh9KQxqI9-DEEW1Ot5r2JbOk,1575
-tinybird/tb/modules/agent/tools/mock.py,sha256=4gEAPZCdTPo1w-fbryWiEx3hPXOK2ZfW0MpXY7smpcI,3408
-tinybird/tb/modules/agent/tools/plan.py,sha256=pr6LnItz6vlOeCG8GE459ExsrBEG0KLx-g02SZGNjXU,1217
+tinybird/tb/modules/agent/tools/get_endpoint_stats.py,sha256=LiEK6ToyPDW2aI8ijclzuwdYAcFmwH-TyjqdFEzQWAc,1689
+tinybird/tb/modules/agent/tools/get_openapi_definition.py,sha256=mjIVVXtgvTs5LzOR8Bp4jB1XhLVMysHrHXawkErFdt8,2282
+tinybird/tb/modules/agent/tools/mock.py,sha256=mHkDyorD51Je3AteAX_JOC9mlyW9Ca4PCkxQVLrIGQA,4068
+tinybird/tb/modules/agent/tools/plan.py,sha256=3_VBs5dc4nlCKH3TVzYldFBaEdFoW5EMzw9f3prMApk,1224
 tinybird/tb/modules/agent/tools/preview_datafile.py,sha256=Gbao_FxhXstnUnngVQxztpizjugyfx1rOXTkw7Yabls,858
-tinybird/tb/modules/agent/tools/request_endpoint.py,sha256=Jl64ln0Jspu_rmp3ycZabj-2IXkmWFSZxoCdcavRpQo,2687
+tinybird/tb/modules/agent/tools/request_endpoint.py,sha256=xNYIEFI9sfq3SjevYqJaHH1EP6uEX5HHrbKS6qo3Imo,2852
 tinybird/tb/modules/datafile/build.py,sha256=NFKBrusFLU0WJNCXePAFWiEDuTaXpwc0lHlOQWEJ43s,51117
 tinybird/tb/modules/datafile/build_common.py,sha256=2yNdxe49IMA9wNvl25NemY2Iaz8L66snjOdT64dm1is,4511
 tinybird/tb/modules/datafile/build_datasource.py,sha256=Ra8pVQBDafbFRUKlhpgohhTsRyp_ADKZJVG8Gd69idY,17227
@@ -111,8 +111,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.dev255.dist-info/METADATA,sha256=oudst3Nwc1lfUUD-RID7T4ncAgcIuWnv5_VNlSjK08c,1733
-tinybird-0.0.1.dev255.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev255.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev255.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev255.dist-info/RECORD,,
+tinybird-0.0.1.dev257.dist-info/METADATA,sha256=Fm23QkZZZv5impoV4to_Ba2OGnznHGMtLYuZK6mHVc4,1733
+tinybird-0.0.1.dev257.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev257.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev257.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev257.dist-info/RECORD,,