tinybird 0.0.1.dev248__py3-none-any.whl → 0.0.1.dev249__py3-none-any.whl

tinybird/ch_utils/engine.py

@@ -134,8 +134,9 @@ class TableDetails:
         _version = self.details.get("version", None)
         return _version
 
-    def is_replicated(self):
-        return "Replicated" in self.details.get("engine", None)
+    def is_replicated(self) -> bool:
+        engine: Optional[str] = self.details.get("engine", None)
+        return engine is not None and "Replicated" in engine
 
     def is_mergetree_family(self) -> bool:
         return self.engine is not None and "mergetree" in self.engine.lower()

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.dev248'
-__revision__ = '8eed30e'
+__version__ = '0.0.1.dev249'
+__revision__ = 'ef4e98c'

tinybird/tb/modules/agent/agent.py

@@ -1,3 +1,4 @@
+import shlex
 import subprocess
 import sys
 from datetime import datetime
@@ -60,14 +61,23 @@ from tinybird.tb.modules.project import Project
 
 
 class TinybirdAgent:
-    def __init__(self, token: str, host: str, project: Project, dangerously_skip_permissions: bool):
+    def __init__(
+        self,
+        token: str,
+        user_token: str,
+        host: str,
+        workspace_id: str,
+        project: Project,
+        dangerously_skip_permissions: bool,
+    ):
         self.token = token
+        self.user_token = user_token
         self.host = host
         self.dangerously_skip_permissions = dangerously_skip_permissions
         self.project = project
         self.messages: list[ModelMessage] = []
         self.agent = Agent(
-            model=create_model(token, host),
+            model=create_model(user_token, host, workspace_id),
             deps_type=TinybirdAgentContext,
             system_prompt=f"""
 You are a Tinybird Code, an agentic CLI that can help users to work with Tinybird.
@@ -221,9 +231,10 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
 def run_agent(config: dict[str, Any], project: Project, dangerously_skip_permissions: bool):
     token = config.get("token", None)
     host = config.get("host", None)
-
+    user_token = config.get("user_token", None)
+    workspace_id = config.get("id", None)
     try:
-        if not token or not host:
+        if not token or not host or not workspace_id or not user_token:
             yes = click.confirm(
                 FeedbackManager.warning(
                     message="Tinybird Code requires authentication. Do you want to authenticate now? [Y/n]"
@@ -238,16 +249,17 @@ def run_agent(config: dict[str, Any], project: Project, dangerously_skip_permiss
                 click.echo()
                 cli_config = CLIConfig.get_project_config()
                 token = cli_config.get_token()
+                user_token = cli_config.get_user_token()
                 host = cli_config.get_host()
 
-            if not token or not host:
+            if not token or not host or not user_token:
                 click.echo(
                     FeedbackManager.error(message="Tinybird Code requires authentication. Run 'tb login' first.")
                 )
                 return
 
         display_banner()
-        agent = TinybirdAgent(token, host, project, dangerously_skip_permissions)
+        agent = TinybirdAgent(token, user_token, host, workspace_id, project, dangerously_skip_permissions)
         click.echo()
         click.echo(FeedbackManager.info(message="Describe what you want to create and I'll help you build it"))
         click.echo(FeedbackManager.info(message="Run /help for more commands"))
@@ -273,7 +285,10 @@ def run_agent(config: dict[str, Any], project: Project, dangerously_skip_permiss
                         }
                     ),
                 )
-
+                if user_input.startswith("tb "):
+                    cmd_parts = shlex.split(user_input)
+                    subprocess.run(cmd_parts, check=True)
+                    continue
                 if user_input.lower() in ["/exit", "/quit"]:
                     click.echo(FeedbackManager.info(message="Goodbye!"))
                     break

tinybird/tb/modules/agent/models.py

@@ -7,11 +7,12 @@ from pydantic_ai.providers.anthropic import AnthropicProvider
 def create_model(
     token: str,
     base_url: str,
+    workspace_id: str,
     model: AnthropicModelName = "claude-4-sonnet-20250514",
 ):
     client = AsyncAnthropic(
         base_url=base_url,
-        http_client=AsyncClient(params={"token": token}),
+        http_client=AsyncClient(params={"token": token, "workspace_id": workspace_id}),
         auth_token=token,
     )
     return AnthropicModel(

tinybird/tb/modules/deployment.py

@@ -211,6 +211,7 @@ def deployment_ls(ctx: click.Context, include_deleted: bool) -> None:
     List all the deployments you have in the project.
     """
     client = ctx.ensure_object(dict)["client"]
+    output = ctx.ensure_object(dict)["output"]
 
     TINYBIRD_API_KEY = client.token
     HEADERS = {"Authorization": f"Bearer {TINYBIRD_API_KEY}"}
@@ -243,7 +244,27 @@ def deployment_ls(ctx: click.Context, include_deleted: bool) -> None:
         )
 
     table.reverse()
-    echo_safe_humanfriendly_tables_format_smart_table(table, column_names=columns)
+
+    # Handle different output formats
+    if output == "json":
+        # Create JSON structure
+        deployments_json = []
+        for row in table:
+            deployments_json.append({"id": row[0], "status": row[1], "created_at": row[2]})
+        from tinybird.tb.modules.common import echo_json
+
+        echo_json({"deployments": deployments_json})
+    elif output == "csv":
+        # Create CSV output
+        csv_output = f"{columns[0]},{columns[1]},{columns[2]}\n"
+        for row in table:
+            csv_output += f"{row[0]},{row[1]},{row[2]}\n"
+        from tinybird.tb.modules.common import force_echo
+
+        force_echo(csv_output)
+    else:
+        # Default human-readable output
+        echo_safe_humanfriendly_tables_format_smart_table(table, column_names=columns)
 
 
 @deployment_group.command(name="promote")

{tinybird-0.0.1.dev248.dist-info → tinybird-0.0.1.dev249.dist-info}/METADATA

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

{tinybird-0.0.1.dev248.dist-info → tinybird-0.0.1.dev249.dist-info}/RECORD

@@ -11,13 +11,13 @@ tinybird/sql_toolset.py,sha256=M2rpLYkgV2W8NnYEYPC1tJdpy4uZHXVF64NBSKLQka4,19549
 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=X4tE9OrfaUy6kO9cqVEzyI9cDcmOF3IAssRRzsTsfEQ,40781
+tinybird/ch_utils/engine.py,sha256=4X1B-iuhdW_mxKnX_m3iCsxgP9RPVgR75g7yH1vsJ6A,40851
 tinybird/datafile/common.py,sha256=J0Oydru3Nh3oCx66EEEHGxm2-r5_NfxvZxiFhBjgAH0,98428
 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=XU7dQJoJUS04mQcPTcpDweJQbUllPEGhxSfkfVitr2Y,247
+tinybird/tb/__cli__.py,sha256=WAhvVHhg2X2uVpfHu3VLfJ2jxnMmru4kGnAeWrgBzQE,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
@@ -32,7 +32,7 @@ tinybird/tb/modules/connection.py,sha256=-MY56NUAai6EMC4-wpi7bT0_nz_SA8QzTmHkV7H
 tinybird/tb/modules/copy.py,sha256=dPZkcIDvxjJrlQUIvToO0vsEEEs4EYumbNV77-BzNoU,4404
 tinybird/tb/modules/create.py,sha256=pJxHXG69c9Z_21s-7VuJ3RZOF_nJU51LEwiAkvI3dZY,23251
 tinybird/tb/modules/datasource.py,sha256=cxq0VVjjidxq-v_JSIIAH7L90XNRctgNKsHRoQ_42OI,41632
-tinybird/tb/modules/deployment.py,sha256=Fw9wSNqmLBGCpKwmZsn3KPsy-6kmQzI8YzSdXWoDb6k,12046
+tinybird/tb/modules/deployment.py,sha256=c-iYgq9dgVsa3m3Oqn24CT3A1omw-p4nfVnUiUynatw,12795
 tinybird/tb/modules/deployment_common.py,sha256=Y0r3g-3d6AcihsVVa0OHer3ow3xHSV1VPskF1eI03KI,17644
 tinybird/tb/modules/deprecations.py,sha256=rrszC1f_JJeJ8mUxGoCxckQTJFBCR8wREf4XXXN-PRc,4507
 tinybird/tb/modules/dev_server.py,sha256=57FCKuWpErwYUYgHspYDkLWEm9F4pbvVOtMrFXX1fVU,10129
@@ -68,11 +68,11 @@ 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=8NBSwQuu3AagieLfbNOXSWeud3WGyThBuyPuzZTtxJQ,17932
+tinybird/tb/modules/agent/agent.py,sha256=dch7zQzdw_8KvORGDj73dGfiz03bGEISkdGYN_r9QQg,18519
 tinybird/tb/modules/agent/animations.py,sha256=z0MNLf8TnUO8qAjgYvth_wc9a9283pNVz1Z4jl15Ggs,2558
 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/models.py,sha256=Of74wcU8oX05ricTqmhMHVHfeYo_pQbnbCI_q3mlx5E,682
+tinybird/tb/modules/agent/models.py,sha256=LW1D27gjcd_jwFmghEzteCgToDfodX2B6B5S8BYbysw,735
 tinybird/tb/modules/agent/prompts.py,sha256=fZMTbTbq8SHWob8-wA5fQFnZ9lJa7Y_66_9JvJT3xuc,6818
 tinybird/tb/modules/agent/utils.py,sha256=7Y8bq_rZlqre8_OvLVjIvE8ZLOBpuKmXamyaNs02zzc,25231
 tinybird/tb/modules/agent/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -106,8 +106,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.dev248.dist-info/METADATA,sha256=Q2bYzjGQkmxcw_IutVCHU3qK6Au414Yse04ys2JugeA,1733
-tinybird-0.0.1.dev248.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev248.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev248.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev248.dist-info/RECORD,,
+tinybird-0.0.1.dev249.dist-info/METADATA,sha256=Mp5zwKJya4hnqnJMIPbvKeUKcal8q0tvZUcmCXD-Nsk,1733
+tinybird-0.0.1.dev249.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev249.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev249.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev249.dist-info/RECORD,,