tinybird 0.0.1.dev2__tar.gz → 0.0.1.dev3__tar.gz

{tinybird-0.0.1.dev2 → tinybird-0.0.1.dev3}/PKG-INFO

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

{tinybird-0.0.1.dev2 → tinybird-0.0.1.dev3}/tinybird/__cli__.py

@@ -4,5 +4,5 @@ __description__ = 'Tinybird Command Line Tool'
 __url__ = 'https://www.tinybird.co/docs/cli/introduction.html'
 __author__ = 'Tinybird'
 __author_email__ = 'support@tinybird.co'
-__version__ = '0.0.1.dev2'
-__revision__ = 'bdbc260'
+__version__ = '0.0.1.dev3'
+__revision__ = '50bec29'

tinybird-0.0.1.dev3/tinybird/tb_cli_modules/llm.py

@@ -0,0 +1,31 @@
+from typing import List, Optional
+
+from openai import OpenAI
+from pydantic import BaseModel
+
+from tinybird.tb_cli_modules.prompts import create_project_prompt
+
+
+class DataFile(BaseModel):
+    name: str
+    content: str
+
+
+class DataProject(BaseModel):
+    datasources: List[DataFile]
+    pipes: List[DataFile]
+
+
+class LLM:
+    def __init__(self, model: Optional[str] = None, key: Optional[str] = None):
+        self.model = model
+        self.key = key
+
+    async def create_project(self, prompt: str) -> DataProject:
+        client = OpenAI(api_key=self.key)
+        completion = client.beta.chat.completions.parse(
+            model=self.model,
+            messages=[{"role": "system", "content": create_project_prompt}, {"role": "user", "content": prompt}],
+            response_format=DataProject,
+        )
+        return completion.choices[0].message.parsed or DataProject(datasources=[], pipes=[])

tinybird-0.0.1.dev3/tinybird/tb_cli_modules/local.py

@@ -0,0 +1,108 @@
+import time
+
+import click
+import requests
+
+import docker
+from tinybird.feedback_manager import FeedbackManager
+from tinybird.tb_cli_modules.common import CLIException
+from tinybird.tb_cli_modules.config import CLIConfig
+
+# TODO: Use the official Tinybird image once it's available 'tinybirdco/tinybird-local:latest'
+TB_IMAGE_NAME = "registry.gitlab.com/tinybird/analytics/tinybird-local-jammy-3.11:latest"
+TB_CONTAINER_NAME = "tinybird-local"
+TB_LOCAL_PORT = 80
+TB_LOCAL_HOST = f"http://localhost:{TB_LOCAL_PORT}"
+
+
+def start_tinybird_local(
+    docker_client,
+):
+    """Start the Tinybird container."""
+    containers = docker_client.containers.list(all=True, filters={"name": TB_CONTAINER_NAME})
+    if containers:
+        # Container `start` is idempotent. It's safe to call it even if the container is already running.
+        container = containers[0]
+        container.start()
+    else:
+        pull_required = False
+        try:
+            local_image = docker_client.images.get(TB_IMAGE_NAME)
+            local_image_id = local_image.attrs["RepoDigests"][0].split("@")[1]
+            remote_image = docker_client.images.get_registry_data(TB_IMAGE_NAME)
+            pull_required = local_image_id != remote_image.id
+        except Exception:
+            pull_required = True
+
+        if pull_required:
+            click.echo(
+                FeedbackManager.info(message="** Downloading latest version of Tinybird development environment...")
+            )
+            docker_client.images.pull(TB_IMAGE_NAME, platform="linux/amd64")
+
+        container = docker_client.containers.run(
+            TB_IMAGE_NAME,
+            name=TB_CONTAINER_NAME,
+            detach=True,
+            ports={"80/tcp": TB_LOCAL_PORT},
+            remove=False,
+            platform="linux/amd64",
+        )
+
+    click.echo(FeedbackManager.info(message="** Waiting for Tinybird development environment to be ready..."))
+    for attempt in range(10):
+        try:
+            run = container.exec_run("tb --no-version-warning sql 'SELECT 1 AS healthcheck' --format json").output
+            # dont parse the json as docker sometimes returns warning messages
+            # todo: rafa, make this rigth
+            if b'"healthcheck": 1' in run:
+                break
+            raise RuntimeError("Unexpected response from Tinybird")
+        except Exception:
+            if attempt == 9:  # Last attempt
+                raise CLIException("Tinybird local environment not ready yet. Please try again in a few seconds.")
+            time.sleep(5)  # Wait 5 seconds before retrying
+
+
+def get_docker_client():
+    """Check if Docker is installed and running."""
+    try:
+        client = docker.from_env()
+        client.ping()
+        return client
+    except Exception:
+        raise CLIException("Docker is not running or installed. Please ensure Docker is installed and running.")
+
+
+def stop_tinybird_local(docker_client):
+    """Stop the Tinybird container."""
+    try:
+        container = docker_client.containers.get(TB_CONTAINER_NAME)
+        container.stop()
+    except Exception:
+        pass
+
+
+def remove_tinybird_local(docker_client):
+    """Remove the Tinybird container."""
+    try:
+        container = docker_client.containers.get(TB_CONTAINER_NAME)
+        container.remove(force=True)
+    except Exception:
+        pass
+
+
+def set_up_tinybird_local(docker_client):
+    """Set up the Tinybird local environment."""
+    start_tinybird_local(docker_client)
+    return get_tinybird_local_client()
+
+
+def get_tinybird_local_client():
+    """Get a Tinybird client connected to the local environment."""
+    config = CLIConfig.get_project_config()
+    tokens = requests.get(f"{TB_LOCAL_HOST}/tokens").json()
+    token = tokens["workspace_admin_token"]
+    config.set_token(token)
+    config.set_host(TB_LOCAL_HOST)
+    return config.get_client(host=TB_LOCAL_HOST, token=token)

{tinybird-0.0.1.dev2 → tinybird-0.0.1.dev3}/tinybird.egg-info/PKG-INFO

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

{tinybird-0.0.1.dev2 → tinybird-0.0.1.dev3}/tinybird.egg-info/SOURCES.txt

@@ -36,6 +36,8 @@ tinybird/tb_cli_modules/datasource.py
 tinybird/tb_cli_modules/exceptions.py
 tinybird/tb_cli_modules/fmt.py
 tinybird/tb_cli_modules/job.py
+tinybird/tb_cli_modules/llm.py
+tinybird/tb_cli_modules/local.py
 tinybird/tb_cli_modules/pipe.py
 tinybird/tb_cli_modules/prompts.py
 tinybird/tb_cli_modules/regions.py