tinybird 0.0.1.dev248__tar.gz → 0.0.1.dev249__tar.gz

{tinybird-0.0.1.dev248 → tinybird-0.0.1.dev249}/PKG-INFO

@@ -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 → tinybird-0.0.1.dev249}/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-0.0.1.dev248 → tinybird-0.0.1.dev249}/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-0.0.1.dev248 → tinybird-0.0.1.dev249}/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-0.0.1.dev248 → tinybird-0.0.1.dev249}/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-0.0.1.dev248 → tinybird-0.0.1.dev249}/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 → tinybird-0.0.1.dev249}/tinybird.egg-info/PKG-INFO

@@ -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