tinybird 0.0.1.dev8__tar.gz → 0.0.1.dev9__tar.gz

{tinybird-0.0.1.dev8 → tinybird-0.0.1.dev9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tinybird
-Version: 0.0.1.dev8
+Version: 0.0.1.dev9
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/cli/introduction.html
 Author: Tinybird

{tinybird-0.0.1.dev8 → tinybird-0.0.1.dev9}/tinybird/tb/modules/build.py

@@ -1,6 +1,8 @@
 import asyncio
 import os
 import random
+import subprocess
+import threading
 import time
 from pathlib import Path
 from typing import Any, Awaitable, Callable, List, Union
@@ -14,6 +16,7 @@ import tinybird.context as context
 from tinybird.client import TinyB
 from tinybird.config import FeatureFlags
 from tinybird.feedback_manager import FeedbackManager, bcolors
+from tinybird.syncasync import sync_to_async
 from tinybird.tb.modules.cli import cli
 from tinybird.tb.modules.common import (
     coro,
@@ -58,7 +61,7 @@ def watch_files(
         elapsed_time = time_end - time_start
         click.echo(
             FeedbackManager.success(message="\n✓ ")
-            + FeedbackManager.gray(message=f"Rebuild completed in {elapsed_time:.1f}s")
+            + FeedbackManager.gray(message=f"Rebuild completed in {elapsed_time:.1f}s\n")
         )
 
     event_handler = FileChangeHandler(filenames, lambda f: asyncio.run(process_wrapper(f)))
@@ -161,8 +164,23 @@ async def build(
     await build_once(filenames)
 
     if watch:
-        click.echo(FeedbackManager.highlight(message="◎ Watching for changes..."))
-        watch_files(filenames, process)
+        click.echo(FeedbackManager.highlight(message="◎ Watching for changes...\n"))
+        watcher_thread = threading.Thread(target=watch_files, args=(filenames, process), daemon=True)
+        watcher_thread.start()
+
+        # Main CLI loop
+        while True:
+            user_input = click.prompt("", prompt_suffix="")
+            if user_input.lower() == "exit":
+                break
+
+            if "tb build" in user_input:
+                click.echo(FeedbackManager.error(message="Build command is already running"))
+            else:
+                # Process the user command
+                await sync_to_async(subprocess.run, thread_sensitive=True)(user_input, shell=True, text=True)
+
+            click.echo(FeedbackManager.highlight(message="\n◎ Watching for changes...\n"))
 
 
 async def build_and_print_pipe(tb_client: TinyB, filename: str):

{tinybird-0.0.1.dev8 → tinybird-0.0.1.dev9}/tinybird/tb/modules/create.py

@@ -1,3 +1,4 @@
+import json
 import os
 from os import getcwd
 from pathlib import Path
@@ -39,6 +40,7 @@ from tinybird.tb.modules.local import (
     type=click.Path(exists=True, file_okay=False),
     help="Folder where datafiles will be placed",
 )
+@click.option("--rows", type=int, default=100, help="Number of events to send")
 @click.pass_context
 @coro
 async def create(
@@ -46,9 +48,9 @@ async def create(
     data: Optional[str],
     prompt: Optional[str],
     folder: Optional[str],
+    rows: int,
 ) -> None:
     """Initialize a new project."""
-    click.echo(FeedbackManager.gray(message="Setting up Tinybird Local"))
     folder = folder or getcwd()
     try:
         tb_client = get_tinybird_local_client()
@@ -72,7 +74,17 @@ async def create(
                 datasource_content = datasource_path.read_text()
                 has_json_path = "`json:" in datasource_content
                 if has_json_path:
-                    await llm.generate_sql_sample_data(tb_client, datasource_name, datasource_content)
+                    sql = await llm.generate_sql_sample_data(tb_client, datasource_name, datasource_content, rows)
+                    result = await tb_client.query(f"{sql} FORMAT JSON")
+                    data = result.get("data", [])
+                    max_rows_per_request = 100
+                    sent_rows = 0
+                    for i in range(0, len(data), max_rows_per_request):
+                        batch = data[i : i + max_rows_per_request]
+                        ndjson_data = "\n".join([json.dumps(row) for row in batch])
+                        await tb_client.datasource_events(datasource_name, ndjson_data)
+                        sent_rows += len(batch)
+                click.echo(f"Sent {sent_rows} rows to datasource '{datasource_name}'")
         click.echo(FeedbackManager.success(message="\n✓ Tinybird Local is ready!"))
     except Exception as e:
         click.echo(FeedbackManager.error(message=f"Error: {str(e)}"))

{tinybird-0.0.1.dev8 → tinybird-0.0.1.dev9}/tinybird.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tinybird
-Version: 0.0.1.dev8
+Version: 0.0.1.dev9
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/cli/introduction.html
 Author: Tinybird